IntegralChannelFeatureGenerators.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 _INTEGRAL_CHANNEL_FEATURE_GENERATORS_H
#define _INTEGRAL_CHANNEL_FEATURE_GENERATORS_H

#include "IntegralChannelFeatureGenerator.h"
#include "Types.h"

struct BaseIntegralChannelFeatureGenerator: public IntegralChannelFeatureGenerator {
public:

    unsigned int m_mw, m_mh;

    unsigned int m_nchannel;

    unsigned int m_minFeatArea;

    unsigned int m_minWidth;

    unsigned int m_step;

    unsigned int m_border;

public:

    typedef IntegralChannelFeature FeatureType;

public:
    BaseIntegralChannelFeatureGenerator() : m_border(0) { }
    ~BaseIntegralChannelFeatureGenerator();

    void SetGeometry(unsigned int mw, unsigned int mh, unsigned int n);

    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) {
        m_border = border;
    }

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

    void SetParams(int nchn, int w,int h)
    {
        m_nchannel = nchn;
        m_mw = w;
        m_mh = h;

    }


};

class BruteForceIntegralChannelFeatureGenerator: public BaseIntegralChannelFeatureGenerator {
    unsigned int m_featCount; 

    std::vector< rect> m_rects;

public:

    BruteForceIntegralChannelFeatureGenerator() : BaseIntegralChannelFeatureGenerator(), m_featCount(0) { }
    virtual ~BruteForceIntegralChannelFeatureGenerator();

    void Reset();

    unsigned int Count() const {
        return m_rects.size() * m_nchannel;
    }

    bool Next(IntegralChannelFeature & out);
};

class RandomIntegralChannelFeatureGenerator: public BaseIntegralChannelFeatureGenerator {
    int m_cur;

    int m_featRand;

public:

    RandomIntegralChannelFeatureGenerator(int nFeat) : BaseIntegralChannelFeatureGenerator(), m_cur(0), m_featRand(nFeat) { }
    virtual ~RandomIntegralChannelFeatureGenerator();

    unsigned int Count() const {
        return m_featRand;
    }

    bool Next(IntegralChannelFeature & out);

    void Reset();

};

#endif