OmniAlgoVRFilterDetection.hh

Go to the documentation of this file.

/*
 *  Copyright (C) Massimo Cora' 2006 <maxcvs@email.it>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#ifndef __OMNIALGO_VRFILTER_DETECTION_H__
#define __OMNIALGO_VRFILTER_DETECTION_H__


#include "../OmniConfig.hh"
#include "../OmniConversion.hh"
#include "../OmniGCardConverter.hh"
#include "../OmniFastLookupTable.hh"
#include "../OmniCallback.hh"

// the testing stuff
#include "../testing/CvFaceDetect.hh"
#include "../testing/CvFaceTrack.hh"
#include "../testing/CvVRFilter.hh"

#include <cxtypes.h>
#include <cv.h>
#include <highgui.h>

template <class ConverterT, class AngleT, class ImageT, class PointT>
class OmniAlgoVRFilterDetection {
public:
        OmniAlgoVRFilterDetection( OmniConversion<ConverterT, AngleT, ImageT, PointT> *conversion_klass,
                                                        IplImage *first_frame );
        virtual ~OmniAlgoVRFilterDetection();
        
        void loop_next_frame( IplImage* frame );
        
        void loop_stop();

        void register_image_callbacks( on_create_image_context_cb* create_image_context,
                                                                   on_image_do_action_cb* image_show,
                                                                   on_destroy_image_context_cb* destroy_image_context,
                                                                   void* callback_data
                                                                  );
        
        void register_all_image_callbacks( on_create_image_context_cb* create_image_context,
                                                                           on_image_do_action_cb* image_show,
                                                                           on_destroy_image_context_cb* destroy_image_context,
                                                                           void* callback_data_on_image_do_action
                                                                          );    

private:                // functions
        on_create_image_context_cb* create_image_context;
        
        on_image_do_action_cb* image_show;
        
        on_destroy_image_context_cb* destroy_image_context;
                
private:                // data
        OmniConversion<ConverterT, AngleT, ImageT, PointT> *_conv;
        CvFaceTrack<ConverterT, AngleT> *_facetracking;
        CvFaceDetect *_facedetect;
        int _frames_count;
        
        void *_callback_data_image_cb;
        
        bool _is_running;       
};





template <class ConverterT, class AngleT, class ImageT, class PointT>
OmniAlgoVRFilterDetection<ConverterT, AngleT, ImageT, PointT>::
OmniAlgoVRFilterDetection( OmniConversion<ConverterT, AngleT, ImageT, PointT> *conversion_klass,
                                                   IplImage *first_frame ) : _conv( conversion_klass )
{
        
        _is_running = false;
        _facedetect = new CvFaceDetect();
        _facetracking = new CvFaceTrack<ConverterT, AngleT>( _conv );
        
        _callback_data_image_cb = NULL;
        _frames_count = 0;
        
        // set the default callbacks. They can be changed with register_callbacks () call.
        register_image_callbacks( omnistuff_create_image_context, omnistuff_image_do_action, 
                                                          omnistuff_destroy_image_context, NULL );
        
        _is_running = true;
        
}

template <class ConverterT, class AngleT, class ImageT, class PointT>
OmniAlgoVRFilterDetection<ConverterT, AngleT, ImageT, PointT>::
~OmniAlgoVRFilterDetection()
{
        delete _conv;
        delete _facedetect;
        delete _facetracking;
}

template <class ConverterT, class AngleT, class ImageT, class PointT>
void OmniAlgoVRFilterDetection<ConverterT, AngleT, ImageT, PointT>::loop_next_frame( IplImage* frame )
{
        if ( !_is_running )
                return; 


        // perform a check every other frame.
        if ( _frames_count %2 == 0) {
                _frames_count = 0;
                _facedetect->cvFaceDetect( frame,  DRAW_NONE );
//              _facedetect->cvFaceDetect( frame,  DRAW_ALL );
                        
                _facetracking->cvAddPoint2Track( _facedetect->faces );
        }

        _facetracking->cvTrackNextFrame( frame );       
        _frames_count++;
}

template <class ConverterT, class AngleT, class ImageT, class PointT>
void OmniAlgoVRFilterDetection<ConverterT, AngleT, ImageT, PointT>::loop_stop()
{
        _is_running = false;
}


template <class ConverterT, class AngleT, class ImageT, class PointT>
void OmniAlgoVRFilterDetection<ConverterT, AngleT, ImageT, PointT>::
register_image_callbacks( on_create_image_context_cb* create_image_context,
                                                  on_image_do_action_cb* image_show,
                                                  on_destroy_image_context_cb* destroy_image_context,
                                                  void* callback_data_on_image_cb )
{
        this->create_image_context = create_image_context;
        this->image_show = image_show;
        this->destroy_image_context = destroy_image_context;
        this->_callback_data_image_cb = callback_data_on_image_cb;
}


template <class ConverterT, class AngleT, class ImageT, class PointT>
void OmniAlgoVRFilterDetection<ConverterT, AngleT, ImageT, PointT>::register_all_image_callbacks( 
                                                                                on_create_image_context_cb* create_image_context,
                                                                                on_image_do_action_cb* image_show,
                                                                                on_destroy_image_context_cb* destroy_image_context,
                                                                                void* callback_data_on_image_cb )
{
        // instances.
        _facetracking->register_image_callbacks( create_image_context, image_show, 
                                                          destroy_image_context, callback_data_on_image_cb );
}

#endif  /* __OMNIALGO_VRFILTER_DETECTION_H__ */

---