converted more mixed tab/space indentations to tabs. only whitespace changes.
[blender.git] / source / blender / nodes / intern / CMP_nodes / CMP_valToRgb.c
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, 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 /** \file blender/nodes/intern/CMP_nodes/CMP_valToRgb.c
31  *  \ingroup cmpnodes
32  */
33
34
35 #include "../CMP_util.h"
36
37
38 /* **************** VALTORGB ******************** */
39 static bNodeSocketType cmp_node_valtorgb_in[]= {
40         {       SOCK_VALUE, 1, "Fac",                   0.5f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f},
41         {       -1, 0, ""       }
42 };
43 static bNodeSocketType cmp_node_valtorgb_out[]= {
44         {       SOCK_RGBA, 0, "Image",                  0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 1.0f},
45         {       SOCK_VALUE, 0, "Alpha",                 1.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f},
46         {       -1, 0, ""       }
47 };
48
49 static void do_colorband_composit(bNode *node, float *out, float *in)
50 {
51         do_colorband(node->storage, in[0], out);
52 }
53
54 static void node_composit_exec_valtorgb(void *UNUSED(data), bNode *node, bNodeStack **in, bNodeStack **out)
55 {
56         /* stack order in: fac */
57         /* stack order out: col, alpha */
58         
59         if(out[0]->hasoutput==0 && out[1]->hasoutput==0) 
60                 return;
61         
62         if(node->storage) {
63                 /* input no image? then only color operation */
64                 if(in[0]->data==NULL) {
65                         do_colorband(node->storage, in[0]->vec[0], out[0]->vec);
66                 }
67                 else {
68                         /* make output size of input image */
69                         CompBuf *cbuf= in[0]->data;
70                         CompBuf *stackbuf= alloc_compbuf(cbuf->x, cbuf->y, CB_RGBA, 1); /* allocs */
71                         
72                         composit1_pixel_processor(node, stackbuf, in[0]->data, in[0]->vec, do_colorband_composit, CB_VAL);
73                         
74                         out[0]->data= stackbuf;
75                         
76                         if(out[1]->hasoutput)
77                                 out[1]->data= valbuf_from_rgbabuf(stackbuf, CHAN_A);
78
79                 }
80         }
81 }
82
83 static void node_composit_init_valtorgb(bNode* node)
84 {
85         node->storage= add_colorband(1);
86 }
87
88 void register_node_type_cmp_valtorgb(ListBase *lb)
89 {
90         static bNodeType ntype;
91
92         node_type_base(&ntype, CMP_NODE_VALTORGB, "ColorRamp", NODE_CLASS_CONVERTOR, NODE_OPTIONS,
93                 cmp_node_valtorgb_in, cmp_node_valtorgb_out);
94         node_type_size(&ntype, 240, 200, 300);
95         node_type_init(&ntype, node_composit_init_valtorgb);
96         node_type_storage(&ntype, "ColorBand", node_free_standard_storage, node_copy_standard_storage);
97         node_type_exec(&ntype, node_composit_exec_valtorgb);
98
99         nodeRegisterType(lb, &ntype);
100 }
101
102
103
104 /* **************** RGBTOBW ******************** */
105 static bNodeSocketType cmp_node_rgbtobw_in[]= {
106         {       SOCK_RGBA, 1, "Image",                  0.8f, 0.8f, 0.8f, 1.0f, 0.0f, 1.0f},
107         {       -1, 0, ""       }
108 };
109 static bNodeSocketType cmp_node_rgbtobw_out[]= {
110         {       SOCK_VALUE, 0, "Val",                   0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 1.0f},
111         {       -1, 0, ""       }
112 };
113
114 static void do_rgbtobw(bNode *UNUSED(node), float *out, float *in)
115 {
116         out[0]= in[0]*0.35f + in[1]*0.45f + in[2]*0.2f;
117 }
118
119 static void node_composit_exec_rgbtobw(void *UNUSED(data), bNode *node, bNodeStack **in, bNodeStack **out)
120 {
121         /* stack order out: bw */
122         /* stack order in: col */
123         
124         if(out[0]->hasoutput==0)
125                 return;
126         
127         /* input no image? then only color operation */
128         if(in[0]->data==NULL) {
129                 do_rgbtobw(node, out[0]->vec, in[0]->vec);
130         }
131         else {
132                 /* make output size of input image */
133                 CompBuf *cbuf= in[0]->data;
134                 CompBuf *stackbuf= alloc_compbuf(cbuf->x, cbuf->y, CB_VAL, 1); /* allocs */
135                 
136                 composit1_pixel_processor(node, stackbuf, in[0]->data, in[0]->vec, do_rgbtobw, CB_RGBA);
137                 
138                 out[0]->data= stackbuf;
139         }
140 }
141
142 void register_node_type_cmp_rgbtobw(ListBase *lb)
143 {
144         static bNodeType ntype;
145         
146         node_type_base(&ntype, CMP_NODE_RGBTOBW, "RGB to BW", NODE_CLASS_CONVERTOR, 0,
147                                    cmp_node_rgbtobw_in, cmp_node_rgbtobw_out);
148         node_type_size(&ntype, 80, 40, 120);
149         node_type_exec(&ntype, node_composit_exec_rgbtobw);
150         
151         nodeRegisterType(lb, &ntype);
152 }