1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
/*-----------------------------------------------------------------
LOG
GEM - Graphics Environment for Multimedia
Copyright (c) 1997-1999 Mark Danks. mark@danks.org
Copyright (c) Günther Geiger. geiger@epy.co.at
Copyright (c) 2001-2002 IOhannes m zmoelnig. forum::für::umläute. IEM. zmoelnig@iem.kug.ac.at
Copyright (c) 2002 James Tittle & Chris Clepper
For information on usage and redistribution, and for a DISCLAIMER OF ALL
WARRANTIES, see the file, "GEM.LICENSE.TERMS" in this distribution.
-----------------------------------------------------------------*/
#ifndef INCLUDE_PIX_OPENCV_OPTICALFLOW_H_
#define INCLUDE_PIX_OPENCV_OPTICALFLOW_H_
#include "opencv2/opencv.hpp"
#include "Base/GemBase.h"
#include "Gem/Exception.h"
#include "Gem/State.h"
#include "Base/GemPixObj.h"
#include "RTE/MessageCallbacks.h"
#include <stdio.h>
/*-----------------------------------------------------------------
-------------------------------------------------------------------
CLASS
pix_opencv_opticalflow
apply a simple flow method to compute optical flow
KEYWORDS
pix
DESCRIPTION
-----------------------------------------------------------------*/
//~class Colorcode
//~{
//~// from colorcode.cpp by Daniel Scharstein, 4/2007
//~// http://vision.middlebury.edu/flow/data/
//~#define MAXCOLS 60
//~
//~public:
//~Colorcode();
//~static cv::Vec3b computeColor(float fx, float fy);
//~
//~private:
//~int ncols;
//~int colorwheel[MAXCOLS][3];
//~
//~
//~void setcols(int r, int g, int b, int k);
//~
//~void makecolorwheel();
//~};
static cv::Vec3b computeColor(float fx, float fy);
inline bool isFlowCorrect(cv::Point2f u);
class GEM_EXTERN pix_opencv_opticalflow : public GemPixObj
{
CPPEXTERN_HEADER(pix_opencv_opticalflow, GemPixObj)
public:
//////////
// Constructor
pix_opencv_opticalflow();
protected:
//////////
// Destructor
virtual ~pix_opencv_opticalflow();
//////////
// Do the processing
virtual void processRGBAImage(imageStruct &image);
// Messages handling
void gainMess(double arg);
void normalizeMess(double arg);
private:
cv::Mat prev, flow;
t_float m_gain;
bool m_normalize;
cv::Mat m_prev, m_curr, m_flow;
t_outlet *m_dataout_middle; // contour outlet
t_outlet *m_dataout_right; // info outlet
};
#endif // for header file
|
