root/trunk/opencv/src/cvaux/vs/bgfg_estimation.cpp

/*M///////////////////////////////////////////////////////////////////////////////////////
//
//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
//
//  By downloading, copying, installing or using the software you agree to this license.

//  If you do not agree to this license, do not download, install,
//  copy or use the software.

////
//                        Intel License Agreement

//
// Copyright (C) 2000, Intel Corporation, all rights reserved.
// Third party copyrights are property of their respective owners.
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
//   * Redistribution's of source code must retain the above copyright notice,

//     this list of conditions and the following disclaimer.
//
//   * Redistribution's in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//
//   * The name of Intel Corporation may not be used to endorse or promote products
//     derived from this software without specific prior written permission.
//
// This software is provided by the copyright holders and contributors "as is" and
// any express or implied warranties, including, but not limited to, the implied
// warranties of merchantability and fitness for a particular purpose are disclaimed.
// In no event shall the Intel Corporation or contributors be liable for any direct,
// indirect, incidental, special, exemplary, or consequential damages
// (including, but not limited to, procurement of substitute goods or services;
// loss of use, data, or profits; or business interruption) however caused
// and on any theory of liability, whether in contract, strict liability,
// or tort (including negligence or otherwise) arising in any way out of
// the use of this software, even if advised of the possibility of such damage.
//
//M*/
#include "_cvaux.h"
//Function cvCreateBGStatModel creates and returns initialized BG model
// parameters:
//      first_frame   - frame from video sequence
//      model_type – type of BG model (CV_BG_MODEL_MOG, CV_BG_MODEL_FGD,…)
//      parameters  - (optional) if NULL the default parameters of the algorithm will be used
CvBGStatModel* cvCreateBGStatModel( IplImage* first_frame, int model_type, void* params )
{
    CvBGStatModel* bg_model = NULL;
    if( model_type == CV_BG_MODEL_FGD || model_type == CV_BG_MODEL_FGD_SIMPLE )

        bg_model = cvCreateFGDStatModel( first_frame, (CvFGDStatModelParams*)params );

    else if( model_type == CV_BG_MODEL_MOG )
        bg_model = cvCreateGaussianBGModel( first_frame, (CvGaussBGStatModelParams*)params );
    return bg_model;
}
/* FOREGROUND DETECTOR INTERFACE */
class CvFGDetectorBase:public CvFGDetector
{
protected:
    CvBGStatModel*  m_pFG;
    int             m_FGType;
    void*           m_pFGParam; /* foreground params */
    CvFGDStatModelParams        m_ParamFGD;
    CvGaussBGStatModelParams    m_ParamMOG;
    char*                       m_SaveName;
    char*                       m_LoadName;
public:
    virtual void SaveState(CvFileStorage* )
    {
        if( m_FGType == CV_BG_MODEL_FGD || m_FGType == CV_BG_MODEL_FGD_SIMPLE )
        {
            if( m_SaveName ) /* File name is not empty */
            {
                //cvSaveStatModel(m_SaveName, (CvFGDStatModel*)m_pFG);
            }
        }
    };
    virtual void LoadState(CvFileStorage* , CvFileNode* )
    {
        if( m_FGType == CV_BG_MODEL_FGD || m_FGType == CV_BG_MODEL_FGD_SIMPLE )
        {
            if( m_LoadName ) /* File name is not empty */
            {
                //cvRestoreStatModel(m_LoadName, (CvFGDStatModel*)m_pFG);
            }
        }
    };
    CvFGDetectorBase(int type, void* param)
    {
        m_pFG = NULL;
        m_FGType = type;
        m_pFGParam = param;
        if( m_FGType == CV_BG_MODEL_FGD || m_FGType == CV_BG_MODEL_FGD_SIMPLE )
        {
            if(m_pFGParam)
            {
              m_ParamFGD = *(CvFGDStatModelParams*)m_pFGParam;
            }
            else
            {
                m_ParamFGD.Lc = CV_BGFG_FGD_LC;
                m_ParamFGD.N1c = CV_BGFG_FGD_N1C;
                m_ParamFGD.N2c = CV_BGFG_FGD_N2C;
                m_ParamFGD.Lcc = CV_BGFG_FGD_LCC;
                m_ParamFGD.N1cc = CV_BGFG_FGD_N1CC;
                m_ParamFGD.N2cc = CV_BGFG_FGD_N2CC;
                m_ParamFGD.delta = CV_BGFG_FGD_DELTA;
                m_ParamFGD.alpha1 = CV_BGFG_FGD_ALPHA_1;
                m_ParamFGD.alpha2 = CV_BGFG_FGD_ALPHA_2;
                m_ParamFGD.alpha3 = CV_BGFG_FGD_ALPHA_3;
                m_ParamFGD.T = CV_BGFG_FGD_T;
                m_ParamFGD.minArea = CV_BGFG_FGD_MINAREA;
            }
            AddParam("LC",&m_ParamFGD.Lc);
            AddParam("alpha1",&m_ParamFGD.alpha1);
            AddParam("alpha2",&m_ParamFGD.alpha2);
            AddParam("alpha3",&m_ParamFGD.alpha3);
            AddParam("N1c",&m_ParamFGD.N1c);
            AddParam("N2c",&m_ParamFGD.N2c);
            AddParam("N1cc",&m_ParamFGD.N1cc);
            AddParam("N2cc",&m_ParamFGD.N2cc);
            m_SaveName = 0;
            m_LoadName = 0;
            AddParam("SaveName",&m_SaveName);
            AddParam("LoadName",&m_LoadName);
        }
        else if( m_FGType == CV_BG_MODEL_MOG )
        {
            if(m_pFGParam)
            {
                m_ParamMOG = *(CvGaussBGStatModelParams*)m_pFGParam;
            }
            else
            {
                m_ParamMOG.win_size = CV_BGFG_MOG_WINDOW_SIZE;
                m_ParamMOG.bg_threshold = CV_BGFG_MOG_BACKGROUND_THRESHOLD;
                m_ParamMOG.std_threshold = CV_BGFG_MOG_STD_THRESHOLD;
                m_ParamMOG.weight_init = CV_BGFG_MOG_WEIGHT_INIT;
                m_ParamMOG.variance_init = CV_BGFG_MOG_SIGMA_INIT*CV_BGFG_MOG_SIGMA_INIT;
                m_ParamMOG.minArea = CV_BGFG_MOG_MINAREA;
                m_ParamMOG.n_gauss = CV_BGFG_MOG_NGAUSSIANS;
            }
            AddParam("NG",&m_ParamMOG.n_gauss);
        }
    };
    ~CvFGDetectorBase()
    {
        if(m_pFG)cvReleaseBGStatModel( &m_pFG );
    }
    void ParamUpdate()
    {
        if(m_pFG)cvReleaseBGStatModel( &m_pFG );
    }
    inline IplImage* GetMask()
    {
        return m_pFG?m_pFG->foreground:NULL;
    };
    /* process current image */
    virtual void    Process(IplImage* pImg)
    {
        if(m_pFG == NULL)
        {
            void* param = m_pFGParam;
            if( m_FGType == CV_BG_MODEL_FGD || m_FGType == CV_BG_MODEL_FGD_SIMPLE 
            {
                param = &m_ParamFGD;
            }
            else if( m_FGType == CV_BG_MODEL_MOG )
            {
                param = &m_ParamMOG;
            }
            m_pFG = cvCreateBGStatModel(
                pImg,
                m_FGType,
                param);
            LoadState(0, 0);
        }
        else
        {
            cvUpdateBGStatModel( pImg, m_pFG );
        }
    };
    /* release foreground detector */
    virtual void    Release()
    {
        SaveState(0);
        if(m_pFG)cvReleaseBGStatModel( &m_pFG );
    };
};
CvFGDetector* cvCreateFGDetectorBase(int type, void *param)
{
    return (CvFGDetector*) new CvFGDetectorBase(type, param);
}