Keying Screen node from tomato branch
[blender.git] / source / blender / compositor / operations / COM_KeyingScreenOperation.h
1 /*
2  * Copyright 2012, Blender Foundation.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * Contributor:
19  *              Jeroen Bakker
20  *              Monique Dewanchand
21  *              Sergey Sharybin
22  */
23
24
25 #ifndef _COM_KeyingScreenOperation_h
26 #define _COM_KeyingScreenOperation_h
27
28 #include <string.h>
29
30 #include "COM_NodeOperation.h"
31
32 #include "DNA_scene_types.h"
33 #include "DNA_movieclip_types.h"
34
35 #include "BLI_listbase.h"
36
37 extern "C" {
38         #include "BLI_voronoi.h"
39 }
40
41 /**
42   * Class with implementation of green screen gradient rasterization
43   */
44 class KeyingScreenOperation : public NodeOperation {
45 protected:
46         typedef struct TriangulationData {
47                 VoronoiTriangulationPoint *triangulated_points;
48                 int (*triangles)[3];
49                 int triangulated_points_total, triangles_total;
50         } TriangulationData;
51
52         MovieClip *movieClip;
53         int framenumber;
54         TriangulationData *cachedTriangulation;
55         char trackingObject[64];
56
57         /**
58           * Determine the output resolution. The resolution is retrieved from the Renderer
59           */
60         void determineResolution(unsigned int resolution[], unsigned int preferredResolution[]);
61
62         TriangulationData *buildVoronoiTriangulation();
63
64  public:
65         KeyingScreenOperation();
66
67         void initExecution();
68         void deinitExecution();
69
70         void *initializeTileData(rcti *rect, MemoryBuffer **memoryBuffers);
71
72         void setMovieClip(MovieClip *clip) {this->movieClip = clip;}
73         void setTrackingObject(char *object) {strncpy(this->trackingObject, object, sizeof(this->trackingObject));}
74         void setFramenumber(int framenumber) {this->framenumber = framenumber;}
75
76         void executePixel(float *color, int x, int y, MemoryBuffer *inputBuffers[], void *data);
77 };
78
79 #endif