svn merge -r 12391:12419 https://svn.blender.org/svnroot/bf-blender/trunk/blender
[blender.git] / source / blender / nodes / intern / CMP_util.h
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (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 Foundation,
18  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19  *
20  * The Original Code is Copyright (C) 2006 Blender Foundation.
21  * All rights reserved.
22  *
23  * The Original Code is: all of this file.
24  *
25  * Contributor(s): none yet.
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  */
29
30 #ifndef CMP_NODE_UTILS_H_
31 #define CMP_NODE_UTILS_H_
32
33 #include <stdlib.h>
34 #include <string.h>
35 #include <math.h>
36
37 #include "MEM_guardedalloc.h"
38
39 #include "DNA_camera_types.h" /* qdn: defocus node, need camera info */
40 #include "DNA_action_types.h"
41 #include "DNA_color_types.h"
42 #include "DNA_ipo_types.h"
43 #include "DNA_ID.h"
44 #include "DNA_image_types.h"
45 #include "DNA_material_types.h"
46 #include "DNA_node_types.h"
47 #include "DNA_object_types.h"
48 #include "DNA_scene_types.h"
49 #include "DNA_space_types.h"
50 #include "DNA_screen_types.h"
51 #include "DNA_texture_types.h"
52 #include "DNA_userdef_types.h"
53
54 #include "BKE_blender.h"
55 #include "BKE_colortools.h"
56 #include "BKE_global.h"
57 #include "BKE_image.h"
58 #include "BKE_main.h"
59 #include "BKE_material.h"
60 #include "BKE_texture.h"
61 #include "BKE_utildefines.h"
62 #include "BKE_library.h"
63 #include "BKE_object.h"
64
65 #include "../CMP_node.h"
66 #include "node_util.h"
67
68 #include "BIF_gl.h"
69 #include "BIF_glutil.h"
70 #include "BIF_interface.h"
71 #include "BIF_interface_icons.h"
72 #include "BIF_language.h"
73 #include "BIF_mywindow.h"
74 #include "BIF_previewrender.h"
75 #include "BIF_resources.h"
76 #include "BIF_screen.h"
77 #include "BIF_space.h"
78
79 #include "BLI_arithb.h"
80 #include "BLI_blenlib.h"
81 #include "BLI_rand.h"
82 #include "BLI_threads.h"
83
84 #include "IMB_imbuf_types.h"
85 #include "IMB_imbuf.h"
86
87 #include "BSE_drawipo.h"
88 #include "BSE_node.h"
89 #include "BSE_view.h"
90
91 #include "RE_pipeline.h"
92 #include "RE_shader_ext.h"
93 #include "RE_render_ext.h"
94
95 #include "butspace.h"
96 #include "blendef.h"
97 #include "mydevice.h"
98
99 /* *************************** operations support *************************** */
100
101 /* general signal that's in output sockets, and goes over the wires */
102 typedef struct CompBuf {
103         float *rect;
104         int x, y, xrad, yrad;
105         short type, malloc;
106         rcti disprect;          /* cropped part of image */
107         int xof, yof;           /* relative to center of target image */
108         
109         void (*rect_procedural)(struct CompBuf *, float *, float, float);
110         bNode *node;            /* only in use for procedural bufs */
111         
112         struct CompBuf *next, *prev;    /* for pass-on, works nicer than reference counting */
113 } CompBuf;
114
115 /* defines also used for pixel size */
116 #define CB_RGBA         4
117 #define CB_VEC4         4
118 #define CB_VEC3         3
119 #define CB_VEC2         2
120 #define CB_VAL          1
121
122 /* defines for RGBA channels */
123 #define CHAN_R  0
124 #define CHAN_G  1
125 #define CHAN_B  2
126 #define CHAN_A  3
127
128
129
130 CompBuf *alloc_compbuf(int sizex, int sizey, int type, int alloc);
131 CompBuf *dupalloc_compbuf(CompBuf *cbuf);
132 CompBuf *pass_on_compbuf(CompBuf *cbuf);
133 void free_compbuf(CompBuf *cbuf);
134 void print_compbuf(char *str, CompBuf *cbuf);
135
136 CompBuf *get_cropped_compbuf(rcti *drect, float *rectf, int rectx, int recty, int type);
137 CompBuf *scalefast_compbuf(CompBuf *inbuf, int newx, int newy);
138 CompBuf *typecheck_compbuf(CompBuf *inbuf, int type);
139 float *compbuf_get_pixel(CompBuf *cbuf, float *rectf, int x, int y, int xrad, int yrad);
140
141 /* **************************************************** */
142
143 /* Pixel-to-Pixel operation, 1 Image in, 1 out */
144 void composit1_pixel_processor(bNode *node, CompBuf *out, CompBuf *src_buf, float *src_col,
145                                                                           void (*func)(bNode *, float *, float *), 
146                                                                           int src_type);
147 /* Pixel-to-Pixel operation, 2 Images in, 1 out */
148 void composit2_pixel_processor(bNode *node, CompBuf *out, CompBuf *src_buf, float *src_col,
149                                                                           CompBuf *fac_buf, float *fac, void (*func)(bNode *, float *, float *, float *), 
150                                                                           int src_type, int fac_type);
151
152 /* Pixel-to-Pixel operation, 3 Images in, 1 out */
153 void composit3_pixel_processor(bNode *node, CompBuf *out, CompBuf *src1_buf, float *src1_col, CompBuf *src2_buf, float *src2_col, 
154                                                                           CompBuf *fac_buf, float *fac, void (*func)(bNode *, float *, float *, float *, float *), 
155                                                                           int src1_type, int src2_type, int fac_type);
156
157 /* Pixel-to-Pixel operation, 4 Images in, 1 out */
158 void composit4_pixel_processor(bNode *node, CompBuf *out, CompBuf *src1_buf, float *src1_col, CompBuf *fac1_buf, float *fac1, 
159                                                                           CompBuf *src2_buf, float *src2_col, CompBuf *fac2_buf, float *fac2, 
160                                                                           void (*func)(bNode *, float *, float *, float *, float *, float *), 
161                                                                           int src1_type, int fac1_type, int src2_type, int fac2_type);
162
163 CompBuf *valbuf_from_rgbabuf(CompBuf *cbuf, int channel);
164 void generate_preview(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 convolve(CompBuf* dst, CompBuf* in1, CompBuf* in2);
180
181 extern void node_ID_title_cb(void *node_v, void *unused_v);
182
183
184 /* utility functions used by glare, tonemap and lense distortion */
185 /* soms macros for color handling */
186 typedef float fRGB[4];
187 /* clear color */
188 #define fRGB_clear(c) { c[0]=c[1]=c[2]=0.f; }
189 /* copy c2 to c1 */
190 #define fRGB_copy(c1, c2) { c1[0]=c2[0];  c1[1]=c2[1];  c1[2]=c2[2]; }
191 /* add c2 to c1 */
192 #define fRGB_add(c1, c2) { c1[0]+=c2[0];  c1[1]+=c2[1];  c1[2]+=c2[2]; }
193 /* subtract c2 from c1 */
194 #define fRGB_sub(c1, c2) { c1[0]-=c2[0];  c1[1]-=c2[1];  c1[2]-=c2[2]; }
195 /* multiply c by float value s */
196 #define fRGB_mult(c, s) { c[0]*=s;  c[1]*=s;  c[2]*=s; }
197 /* multiply c2 by s and add to c1 */
198 #define fRGB_madd(c1, c2, s) { c1[0]+=c2[0]*s;  c1[1]+=c2[1]*s;  c1[2]+=c2[2]*s; }
199 /* multiply c2 by color c1 */
200 #define fRGB_colormult(c, cs) { c[0]*=cs[0];  c[1]*=cs[1];  c[2]*=cs[2]; }
201 /* multiply c2 by color c3 and add to c1 */
202 #define fRGB_colormadd(c1, c2, c3) { c1[0]+=c2[0]*c3[0];  c1[1]+=c2[1]*c3[1];  c1[2]+=c2[2]*c3[2]; }
203 /* multiply c2 by color rgb, rgb as separate arguments */
204 #define fRGB_rgbmult(c, r, g, b) { c[0]*=(r);  c[1]*=(g);  c[2]*=(b); }
205 /* swap colors c1 & c2 */
206 #define fRGB_swap(c1, c2) { float _t=c1[0];  c1[0]=c2[0];  c2[0]=_t;\
207                                   _t=c1[1];  c1[1]=c2[1];  c2[1]=_t;\
208                                   _t=c1[2];  c1[2]=c2[2];  c2[2]=_t; }
209
210 void qd_getPixel(CompBuf* src, int x, int y, float* col);
211 void qd_setPixel(CompBuf* src, int x, int y, float* col);
212 void qd_addPixel(CompBuf* src, int x, int y, float* col);
213 void qd_multPixel(CompBuf* src, int x, int y, float f);
214 void qd_getPixelLerpWrap(CompBuf* src, float u, float v, float* col);
215 void qd_getPixelLerp(CompBuf* src, float u, float v, float* col);
216 void qd_getPixelLerpChan(CompBuf* src, float u, float v, int chan, float* out);
217 CompBuf* qd_downScaledCopy(CompBuf* src, int scale);
218 void IIR_gauss(CompBuf* src, float sigma, int chan, int xy);
219 /* end utility funcs */
220
221 #endif