Merging r50470 through r50477 from trunk into soc-2011-tomato
[blender.git] / source / blender / nodes / composite / node_composite_util.h
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
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  * The Original Code is Copyright (C) 2006 Blender Foundation.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/nodes/composite/node_composite_util.h
29  *  \ingroup nodes
30  */
31
32
33 #ifndef __NODE_COMPOSITE_UTIL_H__
34 #define __NODE_COMPOSITE_UTIL_H__
35
36 #include <stdlib.h>
37 #include <string.h>
38 #include <math.h>
39
40 #include "MEM_guardedalloc.h"
41
42 #include "DNA_camera_types.h" /* qdn: defocus node, need camera info */
43 #include "DNA_color_types.h"
44 #include "DNA_ID.h"
45 #include "DNA_image_types.h"
46 #include "DNA_material_types.h"
47 #include "DNA_movieclip_types.h"
48 #include "DNA_node_types.h"
49 #include "DNA_object_types.h"
50 #include "DNA_scene_types.h"
51 #include "DNA_texture_types.h"
52
53 #include "BLI_math.h"
54 #include "BLI_blenlib.h"
55 #include "BLI_rand.h"
56 #include "BLI_threads.h"
57 #include "BLI_utildefines.h"
58 #include "BLI_utildefines.h"
59
60 #include "BLF_translation.h"
61
62 #include "BKE_blender.h"
63 #include "BKE_camera.h"
64 #include "BKE_colortools.h"
65 #include "BKE_global.h"
66 #include "BKE_image.h"
67 #include "BKE_main.h"
68 #include "BKE_material.h"
69 #include "BKE_movieclip.h"
70 #include "BKE_node.h"
71 #include "BKE_texture.h"
72 #include "BKE_tracking.h"
73
74 #include "BKE_library.h"
75 #include "BKE_object.h"
76
77 #include "node_util.h"
78
79 #include "IMB_imbuf_types.h"
80 #include "IMB_imbuf.h"
81
82 #include "RE_pipeline.h"
83 #include "RE_shader_ext.h"
84 #include "RE_render_ext.h"
85
86 /* only for forward declarations */
87 #include "NOD_composite.h"
88
89 #define CMP_SCALE_MAX   12000
90
91 #ifdef WITH_COMPOSITOR_LEGACY
92
93 /* *************************** operations support *************************** */
94
95 /* general signal that's in output sockets, and goes over the wires */
96 typedef struct CompBuf {
97         float *rect;
98         int x, y, xrad, yrad;
99         short type, malloc;
100         rcti disprect;          /* cropped part of image */
101         int xof, yof;           /* relative to center of target image */
102         
103         void (*rect_procedural)(struct CompBuf *, float *, float, float);
104         float procedural_size[3], procedural_offset[3];
105         int procedural_type;
106         bNode *node;            /* only in use for procedural bufs */
107         
108         struct CompBuf *next, *prev;    /* for pass-on, works nicer than reference counting */
109 } CompBuf;
110
111 /* defines also used for pixel size */
112 #define CB_RGBA         4
113 #define CB_VEC4         4
114 #define CB_VEC3         3
115 #define CB_VEC2         2
116 #define CB_VAL          1
117
118 /* defines for RGBA channels */
119 #define CHAN_R  0
120 #define CHAN_G  1
121 #define CHAN_B  2
122 #define CHAN_A  3
123
124
125
126 CompBuf *alloc_compbuf(int sizex, int sizey, int type, int alloc);
127 CompBuf *dupalloc_compbuf(CompBuf *cbuf);
128 CompBuf *pass_on_compbuf(CompBuf *cbuf);
129 void free_compbuf(CompBuf *cbuf);
130 void print_compbuf(char *str, CompBuf *cbuf);
131 void compbuf_set_node(struct CompBuf *cbuf, struct bNode *node);
132
133 CompBuf *get_cropped_compbuf(rcti *drect, float *rectf, int rectx, int recty, int type);
134 CompBuf *scalefast_compbuf(CompBuf *inbuf, int newx, int newy);
135 CompBuf *typecheck_compbuf(CompBuf *inbuf, int type);
136 void typecheck_compbuf_color(float *out, float *in, int outtype, int intype);
137
138 /* **************************************************** */
139
140 float *compbuf_get_pixel(CompBuf *cbuf, float *defcol, float *use, int x, int y, int xrad, int yrad);
141
142 /* Pixel-to-Pixel operation, 1 Image in, 1 out */
143 void composit1_pixel_processor(bNode *node, CompBuf *out, CompBuf *src_buf, float *src_col,
144                                                                           void (*func)(bNode *, float *, float *), 
145                                                                           int src_type);
146 /* Pixel-to-Pixel operation, 2 Images in, 1 out */
147 void composit2_pixel_processor(bNode *node, CompBuf *out, CompBuf *src_buf, float *src_col,
148                                                                           CompBuf *fac_buf, float *fac, void (*func)(bNode *, float *, float *, float *), 
149                                                                           int src_type, int fac_type);
150
151 /* Pixel-to-Pixel operation, 3 Images in, 1 out */
152 void composit3_pixel_processor(bNode *node, CompBuf *out, CompBuf *src1_buf, float *src1_col, CompBuf *src2_buf, float *src2_col, 
153                                                                           CompBuf *fac_buf, float *fac, void (*func)(bNode *, float *, float *, float *, float *), 
154                                                                           int src1_type, int src2_type, int fac_type);
155
156 /* Pixel-to-Pixel operation, 4 Images in, 1 out */
157 void composit4_pixel_processor(bNode *node, CompBuf *out, CompBuf *src1_buf, float *src1_col, CompBuf *fac1_buf, float *fac1, 
158                                                                           CompBuf *src2_buf, float *src2_col, CompBuf *fac2_buf, float *fac2, 
159                                                                           void (*func)(bNode *, float *, float *, float *, float *, float *), 
160                                                                           int src1_type, int fac1_type, int src2_type, int fac2_type);
161
162 CompBuf *valbuf_from_rgbabuf(CompBuf *cbuf, int channel);
163 void valbuf_to_rgbabuf(CompBuf *valbuf, CompBuf *cbuf, int channel);
164 void generate_preview(void *data, bNode *node, CompBuf *stackbuf);
165
166 void do_copy_rgba(bNode *node, float *out, float *in);
167 void do_copy_rgb(bNode *node, float *out, float *in);
168 void do_copy_value(bNode *node, float *out, float *in);
169 void do_copy_a_rgba(bNode *node, float *out, float *in, float *fac);
170
171 void do_rgba_to_yuva(bNode *node, float *out, float *in);
172 void do_rgba_to_hsva(bNode *node, float *out, float *in);
173 void do_rgba_to_ycca(bNode *node, float *out, float *in);
174 void do_yuva_to_rgba(bNode *node, float *out, float *in);
175 void do_hsva_to_rgba(bNode *node, float *out, float *in);
176 void do_ycca_to_rgba(bNode *node, float *out, float *in);
177
178 void gamma_correct_compbuf(CompBuf *img, int inversed);
179 void premul_compbuf(CompBuf *img, int inversed);
180 void convolve(CompBuf* dst, CompBuf* in1, CompBuf* in2);
181
182 extern void node_ID_title_cb(void *node_v, void *unused_v);
183
184
185 /* utility functions used by glare, tonemap and lens distortion */
186 /* soms macros for color handling */
187 typedef float fRGB[4];
188 /* multiply c2 by color rgb, rgb as separate arguments */
189 #define fRGB_rgbmult(c, r, g, b) { c[0]*=(r);  c[1]*=(g);  c[2]*=(b); } (void)0
190
191 void qd_getPixel(CompBuf* src, int x, int y, float* col);
192 void qd_setPixel(CompBuf* src, int x, int y, float* col);
193 void qd_addPixel(CompBuf* src, int x, int y, float* col);
194 void qd_multPixel(CompBuf* src, int x, int y, float f);
195 void qd_getPixelLerpWrap(CompBuf* src, float u, float v, float* col);
196 void qd_getPixelLerp(CompBuf* src, float u, float v, float* col);
197 void qd_getPixelLerpChan(CompBuf* src, float u, float v, int chan, float* out);
198 CompBuf* qd_downScaledCopy(CompBuf* src, int scale);
199 void IIR_gauss(CompBuf* src, float sigma, int chan, int xy);
200 /* end utility funcs */
201
202 /* transformations */
203
204 CompBuf* node_composit_transform(CompBuf *cbuf, float x, float y, float angle, float scale, int filter_type);
205 float *node_composit_get_float_buffer(RenderData *rd, ImBuf *ibuf, int *alloc);
206
207 #endif
208
209 #endif  /* WITH_COMPOSITOR_LEGACY */