build version update
[goodguy/history.git] / cinelerra-5.1 / plugins / gaborobj / gaborobj.h
1 /*
2  * CINELERRA
3  * Copyright (C) 1997-2014 Adam Williams <broadcast at earthling dot net>
4  * 
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  * 
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  * 
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18  */
19
20 #ifndef GABOROBJ_H
21 #define GABOROBJ_H
22
23 #include "pluginvclient.h"
24 #include "loadbalance.h"
25 #include "mutex.inc"
26
27 #include "opencv2/core/types.hpp"
28 #include "opencv2/core/mat.hpp"
29 #include "opencv2/calib3d.hpp"
30 #include "opencv2/video/video.hpp"
31
32 #include <vector>
33
34 using namespace std;
35 using namespace cv;
36
37
38 class GaborObjConfig;
39 class GaborObj;
40
41
42 class GaborObjConfig
43 {
44 public:
45         GaborObjConfig();
46
47         int equivalent(GaborObjConfig &that);
48         void copy_from(GaborObjConfig &that);
49         void interpolate(GaborObjConfig &prev, GaborObjConfig &next, 
50                 long prev_frame, long next_frame, long current_frame);
51         void limits();
52 };
53
54 class GaborObjFilterPackage : public LoadPackage
55 {
56 public:
57         GaborObjFilterPackage();
58         int i;
59 };
60
61 class GaborObjFilterEngine : public LoadServer
62 {
63 public:
64         GaborObjFilterEngine(GaborObj *plugin, int cpus);
65         ~GaborObjFilterEngine();
66
67         void init_packages();
68         LoadClient* new_client();
69         LoadPackage* new_package();
70
71         GaborObj *plugin;
72 };
73
74 class GaborObjFilterUnit : public LoadClient
75 {
76 public:
77         GaborObjFilterUnit(GaborObjFilterEngine *engine, GaborObj *plugin);
78         ~GaborObjFilterUnit();
79
80         void process_package(LoadPackage *pkg);
81         void process_filter(GaborObjFilterPackage *pkg);
82
83         GaborObjFilterEngine *engine;
84         GaborObj *plugin;
85 };
86
87 class GaborObj : public PluginVClient
88 {
89 public:
90         GaborObj(PluginServer *server);
91         ~GaborObj();
92         PLUGIN_CLASS_MEMBERS2(GaborObjConfig)
93         int is_realtime();
94         void update_gui();
95         void save_data(KeyFrame *keyframe);
96         void read_data(KeyFrame *keyframe);
97         int process_buffer(VFrame *frame, int64_t start_position, double frame_rate);
98         void to_mat(Mat &mat, int mcols, int mrows,
99                 VFrame *inp, int ix,int iy, int mcolor_model);
100         void from_mat(VFrame *out, int ox, int oy, int ow, int oh,
101                 Mat &mat, int mcolor_model);
102
103         int width, height, color_model;
104         VFrame *input, *output, *accum;
105         ::Mutex *remap_lock;
106
107         Mat next_img;
108         vector<Mat> filters;
109         GaborObjFilterEngine *gabor_engine;
110 };
111
112 #endif