HaarFeatureGenerators.h

Go to the documentation of this file.

/* XBoost: Ada-Boost and Friends on Haar/ICF/HOG Features, Library and ToolBox
 *
 * Copyright (c) 2008-2014 Paolo Medici <medici@ce.unipr.it>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#ifndef _HAAR_FEATURE_GENERATORS_H
#define _HAAR_FEATURE_GENERATORS_H

#include "HaarFeatureGenerator.h"

typedef bool (* FeatureGenerator_t)(HaarFeature & f, int x, int y, int w, int h, unsigned int pw, unsigned int ph);

struct FeatureDescription {
   const char *name;
   FeatureGenerator_t proc;
 };  

extern const FeatureDescription featureDescription[];

extern const char iso_pattern[];
extern const char fav_pattern[];

extern const char standard_pattern[];

const FeatureDescription *findFeature(const char *str);

struct BaseHaarFeatureGenerator {
  public:
    
  unsigned int m_mw,m_mh;
  
  unsigned int m_minFeatArea;
  
  unsigned int m_minWidth;
  
  unsigned int m_step;

  std::vector<FeatureGenerator_t> m_feats;

  public:
  static const char *default_pattern;

  public:
    BaseHaarFeatureGenerator(const char *pattern);

  void SetGeometry(int mw, int mh);
      
  void SetFeatureMinArea(int minArea) { m_minFeatArea = minArea; }
  void SetFeatureMinWidth(int minWidth) { m_minWidth = minWidth; }
  
  void SetStep(int step) { m_step = step; }
  
  void SetBorder(int border) { } 
  
 bool generate(HaarFeature & f, int x, int y, int w, int h, int idx);

  unsigned int W() const { return m_mw; }
  unsigned int H() const { return m_mh; }
  unsigned int Size() const { return m_mw*m_mh; }

  unsigned int Feature() const { return m_feats.size(); }

};

class BruteForceFeatureGenerator: public BaseHaarFeatureGenerator, public HaarFeatureGenerator {
  unsigned int m_featCount; 
  
  std::vector<HaarFeature> m_features;  
  
public:
  
  BruteForceFeatureGenerator(const char *pattern) : BaseHaarFeatureGenerator(pattern), m_featCount(0) { }
  virtual ~BruteForceFeatureGenerator();
  
  void SetGeometry(int mw, int mh)
  {
    BaseHaarFeatureGenerator::SetGeometry(mw,mh);
    m_features.clear();
  }
  
   void SetFeatureMinArea(int minArea) { 
    BaseHaarFeatureGenerator::SetFeatureMinArea(minArea);
    m_features.clear();
   }
  
  void Reset();

  unsigned int Count() const { return m_features.size(); }
  
  bool Next(HaarFeature & out);
};

class RandomFeatureGenerator: public BaseHaarFeatureGenerator, public HaarFeatureGenerator {
  int m_cur;

  int m_featRand;
  
  public:
  
  typedef HaarFeature FeatureType; 
  
  public:
  RandomFeatureGenerator(const char *pattern, int nFeat) : BaseHaarFeatureGenerator(pattern), m_cur(0), m_featRand(nFeat) { } 
  virtual ~RandomFeatureGenerator();
  
  unsigned int Count() const { return m_featRand; }
  bool Next(HaarFeature & out);
  void Reset();
};

class JointSparseGranularFeatureGenerator: public HaarFeatureGenerator {

  int m_featRand;
  
  int m_cur;
  
  int m_width;
  int m_height;
  
  int m_minArea;
  
  int m_step;
  
  public:
  
  typedef HaarFeature FeatureType; 
  
  public:
    
  JointSparseGranularFeatureGenerator(int n) : m_featRand(n), m_cur(0), m_minArea(0), m_step(1) { }
  virtual ~JointSparseGranularFeatureGenerator();

  unsigned int Count() const { return m_featRand; }
  
  void SetGeometry(int mw, int mh) { m_width = mw; m_height = mh; }
      
  void SetFeatureMinArea(int minArea) { m_minArea = minArea; }
  
  void SetStep(int step) { m_step = step; }  

  void SetBorder(int border) { } 
  
  unsigned int W() const { return m_width; }
  unsigned int H() const { return m_height; }
  
  bool Next(HaarFeature & out);
  void Reset();  
};



#endif