FeaturesConcatenator.h

// Features Concatenator base abstract class
// Written by Sergey Kosov in 2015 for Project X
#pragma once

#include "types.h"
#include "macroses.h"

namespace DirectGraphicalModels
{
    // ============================= Samples Concatenatenator Class =============================
    class CFeaturesConcatenator
    {
    public:
        DllExport CFeaturesConcatenator(word nFeatures) : m_nFeatures(nFeatures) {}
        DllExport virtual ~CFeaturesConcatenator(void) {}

        DllExport virtual word  getNumFeatures(void) const = 0;
        DllExport virtual void  concatenate(const Mat &featureVector1, const Mat &featureVector2, Mat &dst) const = 0;


    protected:
        word    m_nFeatures;        
    };


    // ============================= Simple Samples Concatenatenator Class =============================
    class CSimpleFeaturesConcatenator : public CFeaturesConcatenator
    {
    public:
        DllExport CSimpleFeaturesConcatenator(word nFeatures) : CFeaturesConcatenator(nFeatures) {}
        DllExport virtual ~CSimpleFeaturesConcatenator(void) {}
        DllExport virtual word  getNumFeatures(void) const { return 2 * m_nFeatures; }
        DllExport virtual void  concatenate(const Mat &featureVector1, const Mat &featureVector2, Mat &dst) const {
            // Assertions
            DGM_ASSERT(featureVector1.size() == featureVector2.size());
            DGM_ASSERT(featureVector1.type() == featureVector2.type());
            // DGM_ASSERT ( dst )

            for (word f = 0; f < m_nFeatures; f++) {
                dst.at<byte>(2 * f    , 0) = featureVector1.at<byte>(f, 0);
                dst.at<byte>(2 * f + 1, 0) = featureVector2.at<byte>(f, 0);
            }           
        }
    };

    
    // ============================= Simple Samples Concatenatenator Class =============================
    class CDiffFeaturesConcatenator : public CFeaturesConcatenator
    {
    public:
        DllExport CDiffFeaturesConcatenator(word nFeatures) : CFeaturesConcatenator(nFeatures) {}
        DllExport virtual ~CDiffFeaturesConcatenator(void) {}
        DllExport virtual word  getNumFeatures(void) const { return 1 * m_nFeatures; }
        DllExport virtual void  concatenate(const Mat &featureVector1, const Mat &featureVector2, Mat &dst) const {
            // Assertions
            DGM_ASSERT(featureVector1.size() == featureVector2.size());
            DGM_ASSERT(featureVector1.type() == featureVector2.type());
            // DGM_ASSERT ( dst )

            for (word f = 0; f < m_nFeatures; f++) {
                dst.at<byte>(1 * f, 0) = static_cast<byte>(MIN(255, MAX(0, 127.5 + 1.0 * featureVector1.at<byte>(f, 0) - 1.0 * featureVector2.at<byte>(f, 0))));
            }           
        }
    };

}