CommonFeatureExtractor.h

// A common feature extraction class, which unites other feature extraction algorithms
// Writtem by Sergey G. Kosov in 2015 for Project X
#pragma once

#include "IFeatureExtractor.h"
#include "Coordinate.h"
#include "Intensity.h"
#include "HSV.h"
#include "Gradient.h"
#include "NDVI.h"
#include "Distance.h"
#include "HOG.h"
#include "SIFT.h"
#include "Variance.h"
#include "Scale.h"
#include "SparseCoding.h"
#include "GlobalFeatureExtractor.h"
#include "macroses.h"

namespace DirectGraphicalModels { namespace fex
{
    enum ChannelsRGB {
        CH_BLUE,            
        CH_GREEN,           
        CH_RED              
    };

    enum ChannelsHSV {
        CH_HUE,             
        CH_SATURATION,      
        CH_VALUE            
    };

    
    // ================================ Common Feature Extractor Class ==============================
    class CCommonFeatureExtractor : public ILocalFeatureExtractor
    {
    public:
        DllExport CCommonFeatureExtractor(const Mat &img) : ILocalFeatureExtractor(img) {}
        DllExport virtual ~CCommonFeatureExtractor(void) {}

        DllExport Mat virtual get(void) const { return m_img; }

        DllExport CGlobalFeatureExtractor toGlobal(void) const { return CGlobalFeatureExtractor(m_img); }
        DllExport CCommonFeatureExtractor getCoordinate(coordinateType type = COORDINATE_ORDINATE) const { return CCommonFeatureExtractor(CCoordinate::get(m_img, type)); }
        DllExport CCommonFeatureExtractor getIntensity(cv::Scalar weight = CV_RGB(0.333, 0.333, 0.333)) const { return CCommonFeatureExtractor(CIntensity::get(m_img, weight)); }
        DllExport CCommonFeatureExtractor getHSV(void) const { return CCommonFeatureExtractor(CHSV::get(m_img)); }
        DllExport CCommonFeatureExtractor getHue(void) const { return CCommonFeatureExtractor(CHSV::get(m_img)).getChannel(CH_HUE); }
        DllExport CCommonFeatureExtractor getSaturation(void) const { return CCommonFeatureExtractor(CHSV::get(m_img)).getChannel(CH_SATURATION); }
        DllExport CCommonFeatureExtractor getBrightness(void) const { return CCommonFeatureExtractor(CHSV::get(m_img)).getChannel(CH_VALUE); }
        DllExport CCommonFeatureExtractor getGradient(float mid = GRADIENT_MAX_VALUE) const { return CCommonFeatureExtractor(CGradient::get(m_img, mid)); }
        DllExport CCommonFeatureExtractor getNDVI(byte midPoint = 127) const  { return CCommonFeatureExtractor(CNDVI::get(m_img, midPoint)); }
        DllExport CCommonFeatureExtractor getDistance(byte threshold = 16, double multiplier = 4.0) const { return CCommonFeatureExtractor(CDistance::get(m_img, threshold, multiplier)); }
        DllExport CCommonFeatureExtractor getHOG(int nBins = 9, SqNeighbourhood nbhd = sqNeighbourhood(5)) const { return CCommonFeatureExtractor(CHOG::get(m_img, nBins, nbhd)); }
        DllExport CCommonFeatureExtractor getSIFT() const { return CCommonFeatureExtractor(CSIFT::get(m_img)); }
        DllExport CCommonFeatureExtractor getVariance(SqNeighbourhood nbhd = sqNeighbourhood(5)) const { return CCommonFeatureExtractor(CVariance::get(m_img, nbhd)); }
        DllExport CCommonFeatureExtractor getSparseCoding(const Mat &D, SqNeighbourhood nbhd = sqNeighbourhood(3)) const { return CCommonFeatureExtractor(CSparseCoding::get(m_img, D, nbhd)); }
        DllExport CCommonFeatureExtractor reScale(SqNeighbourhood nbhd = sqNeighbourhood(5)) const { return CCommonFeatureExtractor(CScale::get(m_img, nbhd)); }
        DllExport CCommonFeatureExtractor invert(void) const;
        DllExport CCommonFeatureExtractor blur(int R = 2) const;
        DllExport CCommonFeatureExtractor autoContrast(void) const;
        DllExport CCommonFeatureExtractor thresholding(byte threshold) const;
        DllExport CCommonFeatureExtractor getChannel(int channel) const;
    };
} }