The Machine Perception Toolbox

[Introduction]- [News]- [Download]- [Screenshots]- [Manual (pdf)]- [Forums]- [API Reference]- [Repository ]

 

mpisearch/windows/FilterSrc/FilterInterface.h

Go to the documentation of this file.

#ifndef __FILTERINTERFACE_H__
#define __FILTERINTERFACE_H__

#include <windows.h>
#include "mpisearchFaceDetector.h"
#include "Images/color.h"

class FilterInterface
{
public:

        FilterInterface();    // Constructor
        ~FilterInterface();     // Destructor

        int runInterface(BYTE *pData, int imgWidth, int imgHeight);
        void printFrame(RGBTRIPLE *prgb, int cxImage, int cyImage);

private:

//      MPISearchBinary m_Detector;
        MPISearchFaceDetector m_Detector;

        FaceBoxList m_faces;
        FaceBoxList m_lastFace;
        RImage<float> m_pixels;
        MPColorTools colorT;
        int m_failedCycles;
        bool m_failedSearch;

// ================================================================

        inline void GreyScaleFlipVideo(int width, int height, RGBTRIPLE *rgbpix, RImage<float> & pixels)
        {
                RGBTRIPLE *source;
                unsigned char *sp;
                float *dest;

                // convert image to grayscale and out of directshow coordinates 
                for (int y=0; y < height; y++) {
                        source = rgbpix + width * ((height-1) - y);
                        sp = (unsigned char *) source;
                        dest = pixels.array + width*y;
                        for (int x = 0; x < width; x++) {
                                *dest++ = (float)(sp[2] + sp[1]*6 + sp[0]*3);
                                sp += 3;
                        }
                }

        } 

// ================================================================

        inline void DrawBoxes(RGBTRIPLE *prgb, int cxImage, int cyImage)
        {  
                bool firstface = true;
                RGBCOLOR color;
                double alpha;
                FFImage<RGBTRIPLE> rgbimage(prgb, cxImage, cyImage);

                if(m_failedSearch) alpha = exp(-m_failedCycles*0.25);
                else alpha = 1;
                list<Square>::iterator face = m_lastFace.begin();
                list<Square>::iterator last_face = m_lastFace.end();
                for ( ; face != last_face; ++face) {
                        //print face box
                        if(firstface) color = RGBRED;
                        else color = RGBBLUE;
                        Square fa = (*face);
                        colorT.DrawRGBBox(rgbimage, fa.x, fa.y, fa.size, fa.size, color, alpha);

                        firstface = false;
                }
        }


}; // FilterInterface

#endif __FILTERINTERFACE_H__

---