source: trunk/c++_tools/classifier/FeatureSelector.h @ 624

// $Id$

#ifndef _theplu_classifier_featureselector_
#define _theplu_classifier_featureselector_

#include <list>
#include <vector>

namespace theplu {
namespace classifier {
  class MatrixLookup;
  class MatrixLookupWeighted;
  class Target;

  ///
  /// @brief Interface class for FeatureSelection
  ///
  class FeatureSelector
  {
  public:
    /// 
    /// @brief Default Constructor 
    /// 
    FeatureSelector(size_t, size_t);

    /// 
    /// Cleaning upp all objects that have beed returned from FeatureSelector.
    ///
    /// @brief Destructor
    /// 
    virtual ~FeatureSelector(void);

    ///
    /// @return MatrixLookup containing only selected features. 
    ///
    const MatrixLookup& get(const MatrixLookup& data);

    ///
    /// @return MatrixLookupWeighted containing only selected features. 
    ///
    const MatrixLookupWeighted& get(const MatrixLookupWeighted& data);

    ///
    ///
    ///
    inline const std::vector<size_t> features(void) const { return features_; }

    ///
    /// Uses @a matrix to select features.
    ///
    virtual void update(const MatrixLookup& matrix, const Target& target)=0;

    ///
    /// Uses @a matrix to select features.
    ///
    virtual void update(const MatrixLookupWeighted& matrix, 
                        const Target& target)=0;

  protected:
    std::vector<size_t> features_;
    size_t first_;
    size_t N_;

  private:
    std::list<const MatrixLookup*> garbage_;
    std::list<const MatrixLookupWeighted*> garbage_weighted_;

  };

} // end of namespace classifier
} // end of namespace theplu

#endif