FaceTrackerNode.h

Go to the documentation of this file.

/*
-----------------------------------------------------------------------------
This source file is part of FRAPPER
research.animationsinstitut.de
sourceforge.net/projects/frapper

Copyright (c) 2008-2009 Filmakademie Baden-Wuerttemberg, Institute of Animation 

This program 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; version 2.1 of the License.

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 Lesser General Public License for more details.

You should have received a copy of the GNU Lesser 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, or go to
http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html
-----------------------------------------------------------------------------
*/


#ifndef FACETRACKERNODE_H
#define FACETRACKERNODE_H

#include "opencvColorInput.h"
#include "ParameterGroup.h"
#include <QtCore/QVariant>
#include <QtCore/QTime>
#include <QtCore/QThread>
#include <QtCore/QMutex>
#include <QtCore/QTimer>

#include <vector>

#include "Face.h"
#include "Helper.h"

namespace FaceTrackerNode {
using namespace Frapper;
using namespace cv;

class FaceTrackerNode : public opencvColorInput  //, public QThread
{

        Q_OBJECT

public: 

        FaceTrackerNode ( QString name, ParameterGroup *parameterRoot );

public: 

    ~FaceTrackerNode ();

public: 
        
    //virtual void run();

public slots:

private slots:

        void flipPicture();

        void probability();

        void switchTime();

        void switchTimeRand();
    
        void randomLook();

        void focusPointSpeed();

        void focusPointMoveLimit();

        void switchUserNow();

        void minSize();

        void maxSize();

        void drawDebugInformation();

        virtual void processInputMatrix();

private: 





        void detectFaces(cv::Mat image);

        void drawDetectedFaces(cv::Mat image);

        void drawDebugInfo(cv::Mat image);

    void decreaseProbabilities();

    void processAttractors();

    void animateFocusPoint();



private: 
    
    QMutex m_mutex;

    QTimer m_timer;

    QTime m_timestamp;
    
    QTime m_lastTimestamp;

    int m_detectedObjectsCount;
    
    std::vector<Face *> m_faces;

    std::vector<Attractor> m_attractors;

    int m_attractorIndex;

    int m_attractorIndexOld;
    
    int m_justSwitched;

    QTime m_focusTimeLimit;

    QTime m_smallFocusTimeLimit;

        vector<cv::Rect> m_detectedObjects;

        cv::CascadeClassifier *m_cascadeClassifier;

    ParameterGroup *m_configGroup;

    ParameterGroup *m_outputGroup;

    PointNDC m_focusPoint;
    
    PointNDC m_focusPointTarget;

        unsigned int delay;

        unsigned int attractorCount;

        double m_probability;

        bool m_drawDebugInformation;

        bool m_flipHorizontal;

        int m_switchTime;

        int m_switchTimeRand;

        int m_randomLook;

        double m_focusPointSpeed;

        double m_focusPointMoveLimit;

        bool m_switchUserNow;

        double m_minSize;

        double m_maxSize;

        int m_lastNumberofAttractos;
        
};

} // namespace FaceTrackerNode 

#endif