Move the sources to trunk
[opencv] / filters / CamShift / CamShiftF.h
1 /*M///////////////////////////////////////////////////////////////////////////////////////
2 //
3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
4 //
5 //  By downloading, copying, installing or using the software you agree to this license.
6 //  If you do not agree to this license, do not download, install,
7 //  copy or use the software.
8 //
9 //
10 //                        Intel License Agreement
11 //                For Open Source Computer Vision Library
12 //
13 // Copyright (C) 2000, Intel Corporation, all rights reserved.
14 // Third party copyrights are property of their respective owners.
15 //
16 // Redistribution and use in source and binary forms, with or without modification,
17 // are permitted provided that the following conditions are met:
18 //
19 //   * Redistribution's of source code must retain the above copyright notice,
20 //     this list of conditions and the following disclaimer.
21 //
22 //   * Redistribution's in binary form must reproduce the above copyright notice,
23 //     this list of conditions and the following disclaimer in the documentation
24 //     and/or other materials provided with the distribution.
25 //
26 //   * The name of Intel Corporation may not be used to endorse or promote products
27 //     derived from this software without specific prior written permission.
28 //
29 // This software is provided by the copyright holders and contributors "as is" and
30 // any express or implied warranties, including, but not limited to, the implied
31 // warranties of merchantability and fitness for a particular purpose are disclaimed.
32 // In no event shall the Intel Corporation or contributors be liable for any direct,
33 // indirect, incidental, special, exemplary, or consequential damages
34 // (including, but not limited to, procurement of substitute goods or services;
35 // loss of use, data, or profits; or business interruption) however caused
36 // and on any theory of liability, whether in contract, strict liability,
37 // or tort (including negligence or otherwise) arising in any way out of
38 // the use of this software, even if advised of the possibility of such damage.
39 //
40 //M*/
41
42 #pragma once
43
44 #include "CV.hpp"
45
46 class CCamShiftF : public CTransformFilter,
47             public ICamShift,
48             public ISpecifyPropertyPages
49 {
50
51 public:
52
53     static CUnknown * WINAPI CreateInstance(LPUNKNOWN punk, HRESULT *phr);
54
55     // Reveals ICamShift & ISpecifyPropertyPages
56     STDMETHODIMP NonDelegatingQueryInterface(REFIID riid, void ** ppv);
57
58     DECLARE_IUNKNOWN;
59
60     HRESULT Transform(IMediaSample *pIn, IMediaSample *pOut);
61     HRESULT CheckInputType(const CMediaType *mtIn);
62     HRESULT CheckTransform(const CMediaType *mtIn,const CMediaType *mtOut);
63     HRESULT GetMediaType(int iPosition, CMediaType *pMediaType);
64     HRESULT DecideBufferSize(IMemAllocator *pAlloc,
65                              ALLOCATOR_PROPERTIES *pProperties);
66
67     // ICamShift methods
68
69     STDMETHODIMP SetParams( CvCamShiftParams* params );
70     STDMETHODIMP GetParams( CvCamShiftParams* params );
71
72     STDMETHODIMP StartTracking();
73     STDMETHODIMP StopTracking();
74
75     // ISpecifyPropertyPages method
76
77     STDMETHODIMP GetPages(CAUUID *pPages);
78
79 private:
80
81     // Constructor
82     CCamShiftF(TCHAR *tszName, LPUNKNOWN punk, HRESULT *phr);
83     void  CheckBackProject( IplImage* image );
84     void  ApplyCamShift( IplImage* image, bool initialize );
85     void  DrawCross( IplImage* image );
86
87     // Look after the transformation of media samples
88     BOOL CanChangeCamShiftLevel(const CMediaType *pMediaType) const;
89     HRESULT Copy(IMediaSample *pSource, IMediaSample *pDest) const;
90     HRESULT Transform(IMediaSample *pMediaSample);
91
92     // Non interface locking critical section
93     CCritSec m_CamShiftLock;
94     CvCamShiftTracker m_cCamShift;
95     CvCamShiftParams m_params;
96
97     CvRect m_object;
98     bool IsTracking;
99     bool IsInit;
100
101 }; // CCamShift
102
103
104 /* End of file. */