Update to 2.0.0 tree from current Fremantle build
[opencv] / src / cvaux / vs / blobtrackanalysis.cpp
diff --git a/src/cvaux/vs/blobtrackanalysis.cpp b/src/cvaux/vs/blobtrackanalysis.cpp
new file mode 100644 (file)
index 0000000..8be1fe7
--- /dev/null
@@ -0,0 +1,128 @@
+/*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
+//                For Open Source Computer Vision Library
+//
+// 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"
+
+/*======================= FILTER LIST SHELL =====================*/
+typedef struct DefTrackAnalyser
+{
+    CvBlob                  blob;
+    CvBlobTrackAnalysisOne* pFilter;
+    int                     m_LastFrame;
+    int                     state;
+} DefTrackAnalyser;
+
+class CvBlobTrackAnalysisList : public CvBlobTrackAnalysis
+{
+protected:
+    CvBlobTrackAnalysisOne* (*m_CreateAnalysis)();
+    CvBlobSeq               m_TrackAnalyserList;
+    int                     m_Frame;
+public:
+    CvBlobTrackAnalysisList(CvBlobTrackAnalysisOne* (*create)()):m_TrackAnalyserList(sizeof(DefTrackAnalyser))
+    {
+        m_Frame = 0;
+        m_CreateAnalysis = create;
+        SetModuleName("List");
+    }
+    ~CvBlobTrackAnalysisList()
+    {
+        int i;
+        for(i=m_TrackAnalyserList.GetBlobNum(); i>0; --i)
+        {
+            DefTrackAnalyser* pF = (DefTrackAnalyser*)m_TrackAnalyserList.GetBlob(i-1);
+            pF->pFilter->Release();
+        }
+    };
+    virtual void    AddBlob(CvBlob* pBlob)
+    {
+        DefTrackAnalyser* pF = (DefTrackAnalyser*)m_TrackAnalyserList.GetBlobByID(CV_BLOB_ID(pBlob));
+        if(pF == NULL)
+        {   /* Create new filter: */
+            DefTrackAnalyser F;
+            F.state = 0;
+            F.blob = pBlob[0];
+            F.m_LastFrame = m_Frame;
+            F.pFilter = m_CreateAnalysis();
+            m_TrackAnalyserList.AddBlob((CvBlob*)&F);
+            pF = (DefTrackAnalyser*)m_TrackAnalyserList.GetBlobByID(CV_BLOB_ID(pBlob));
+        }
+
+        assert(pF);
+        pF->blob = pBlob[0];
+        pF->m_LastFrame = m_Frame;
+    };
+    virtual void    Process(IplImage* pImg, IplImage* pFG)
+    {
+        int i;
+        for(i=m_TrackAnalyserList.GetBlobNum(); i>0; --i)
+        {
+            DefTrackAnalyser* pF = (DefTrackAnalyser*)m_TrackAnalyserList.GetBlob(i-1);
+            if(pF->m_LastFrame == m_Frame)
+            {   /* Process: */
+                int ID = CV_BLOB_ID(pF);
+                pF->state = pF->pFilter->Process(&(pF->blob), pImg, pFG);
+                CV_BLOB_ID(pF) = ID;
+            }
+            else
+            {   /* Delete blob filter: */
+                pF->pFilter->Release();
+                m_TrackAnalyserList.DelBlob(i-1);
+            }
+        } /* Next blob. */
+        m_Frame++;
+    };
+    float GetState(int BlobID)
+    {
+        DefTrackAnalyser* pF = (DefTrackAnalyser*)m_TrackAnalyserList.GetBlobByID(BlobID);
+        return pF?pF->state:0.f;
+    };
+    void    Release(){delete this;};
+
+}; /* CvBlobTrackAnalysisList */
+
+CvBlobTrackAnalysis* cvCreateBlobTrackAnalysisList(CvBlobTrackAnalysisOne* (*create)())
+{
+    return (CvBlobTrackAnalysis*) new CvBlobTrackAnalysisList(create);
+}
+
+/* ======================== Analyser modules ============================= */
+