oops - now add termux/android DV patch
[goodguy/cinelerra.git] / cinelerra-5.1 / plugins / motion51 / motion51.h
1
2 /*
3  * CINELERRA
4  * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  */
21
22 #ifndef __MOTION_51_H__
23 #define __MOTION_51_H__
24
25 #include <math.h>
26 #include <stdint.h>
27 #include <string.h>
28
29 #include "affine.inc"
30 #include "bchash.inc"
31 #include "filexml.inc"
32 #include "keyframe.inc"
33 #include "loadbalance.h"
34 #include "motion51.inc"
35 #include "overlayframe.inc"
36 #include "pluginvclient.h"
37 #include "vframe.inc"
38
39 class FFT;
40 class Motion51Config;
41 class Motion51Main;
42 class Motion51ScanUnit;
43 class Motion51ScanPackage;
44 class Motion51Scan;
45 class Motion51VVFrame;
46
47 class Motion51Config
48 {
49 public:
50         Motion51Config();
51
52         void init();
53         int equivalent(Motion51Config &that);
54         void copy_from(Motion51Config &that);
55         void interpolate(Motion51Config &prev, Motion51Config &next,
56                 int64_t prev_frame, int64_t next_frame, int64_t current_frame);
57
58         float block_x, block_y, block_w, block_h;
59         float sample_r;
60         int sample_steps;
61         float horiz_limit, vert_limit, twist_limit;
62         float shake_fade, twist_fade;
63         int draw_vectors;
64
65         char tracking_file[BCTEXTLEN];
66         int tracking;
67 };
68
69
70 class Motion51Main : public PluginVClient
71 {
72 public:
73         Motion51Main(PluginServer *server);
74         ~Motion51Main();
75
76         int process_buffer(VFrame **frame, int64_t position, double frame_rate);
77         int is_multichannel();
78         int is_realtime();
79         void save_data(KeyFrame *keyframe);
80         void read_data(KeyFrame *keyframe);
81         void update_gui();
82         void reset_sample(int sz, double r);
83         void get_samples(VFrame *img, double *pix[3], double x, double y);
84         void centroid(double *pix[3], double *ctr_v, double *ctr_x, double *ctr_y);
85         void draw_vectors(VFrame *img);
86
87         int64_t get_ref_position();
88         void update_tracking_cache();
89         int load_tracking_cache(int64_t position);
90         void save_tracking_cache(int64_t position);
91
92         void match(VFrame *ref, VFrame *cur);
93         int transform_target(int use_opengl);
94         int handle_opengl();
95         VFrame* new_temp(VFrame *&tmp, VFrame *ref);
96
97         PLUGIN_CLASS_MEMBERS2(Motion51Config)
98
99         char cache_file[BCTEXTLEN];
100         FILE *cache_fp, *active_fp;
101         void set_tracking_path();
102         int open_cache_file();
103         void close_cache_file();
104         int load_cache_line();
105         int locate_cache_line(int64_t key);
106         int get_cache_line(int64_t key);
107         int put_cache_line(const char *line);
108         void update_cache_file();
109         void (*get_pixel)(double *pix[3], double rx, double ry,
110                 uint8_t **rows, int psz, int iw1, int ih1);
111
112         VFrame *out;
113         VFrame *out_frame;
114         int64_t out_position;
115         VFrame *ref_frame;
116         int64_t ref_position;
117         VFrame *tmp_frame;
118         AffineEngine *affine;
119         Motion51Scan *motion_scan;
120
121         char cache_line[BCSTRLEN];
122         int64_t cache_key, active_key;
123         int64_t tracking_position;
124         float out_w, out_h, out_r;
125         float dx, dy, dt;
126         double rx, ry, rr, rw, rh;
127         float current_dx, current_dy;
128         int x_steps, y_steps, r_steps;
129         double cir_r;  int cir_sz;
130         double *xpts, *ypts;
131         float total_dx, total_dy;
132         float total_angle;
133 };
134
135 class Motion51ScanPackage : public LoadPackage
136 {
137 public:
138         Motion51ScanPackage() {}
139         ~Motion51ScanPackage() {}
140         double x, y;
141 };
142 class Motion51ScanUnit : public LoadClient
143 {
144 public:
145         Motion51ScanUnit(Motion51Scan *server, Motion51Main *plugin);
146         ~Motion51ScanUnit();
147         void process_package(LoadPackage *package);
148
149         Motion51Scan *server;
150         Motion51Main *plugin;
151         FFT *fft;
152 };
153
154 class Motion51Scan : public LoadServer
155 {
156 public:
157         Motion51Scan(Motion51Main *plugin, int total_clients, int x_steps, int y_steps);
158         ~Motion51Scan();
159         friend class Motion51ScanUnit;
160
161         Motion51Main *plugin;
162         Mutex *result_lock;
163         FFT *fft;
164         VFrame *cur, *ref;
165
166         int x_steps, y_steps;
167         double bx, by, br, bw, bh;
168         int rpix_sz;
169         double *rpix[3], rpwr[3];
170         double rctr_v[3], rctr_x[3], rctr_y[3];
171         double *ref_real[3], *ref_imag[3];
172         double cor_value, value;
173         double dx_result, dy_result, dt_result;
174
175         void init_packages();
176         LoadClient *new_client() { return new Motion51ScanUnit(this, plugin); }
177         LoadPackage *new_package() { return new Motion51ScanPackage(); }
178         void scan(VFrame *ref, VFrame *cur, int sz);
179         double compare(double cx, double cy, double ct);
180 };
181
182 class Motion51VVFrame : public VFrame
183 {
184 public:
185         Motion51VVFrame(VFrame *vfrm, int n);
186         int draw_pixel(int x, int y);
187         int n;
188 };
189
190 #endif