svn merge -r 37306:39975 https://svn.blender.org/svnroot/bf-blender/trunk/blender
[blender.git] / source / blender / nodes / texture / nodes / node_texture_hueSatVal.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): Juho Vepsäläinen
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  */
29
30 /** \file blender/nodes/texture/nodes/node_texture_hueSatVal.c
31  *  \ingroup texnodes
32  */
33
34
35 #include "node_texture_util.h"
36 #include "NOD_texture.h"
37
38
39 static bNodeSocketTemplate inputs[]= {
40         { SOCK_FLOAT, 1, "Hue",        0.0f, 0.0f, 0.0f, 0.0f, -0.5f, 0.5f, PROP_NONE },
41         { SOCK_FLOAT, 1, "Saturation", 1.0f, 0.0f, 0.0f, 0.0f,  0.0f, 2.0f, PROP_FACTOR },
42         { SOCK_FLOAT, 1, "Value",      1.0f, 0.0f, 0.0f, 0.0f,  0.0f, 2.0f, PROP_FACTOR },
43         { SOCK_FLOAT, 1, "Factor",     1.0f, 0.0f, 0.0f, 0.0f,  0.0f, 1.0f, PROP_FACTOR },
44         { SOCK_RGBA,  1, "Color",      0.8f, 0.8f, 0.8f, 1.0f },
45         { -1, 0, "" }
46 };
47 static bNodeSocketTemplate outputs[]= {
48         { SOCK_RGBA, 0, "Color" },
49         { -1, 0, "" }
50 };
51
52 static void do_hue_sat_fac(bNode *UNUSED(node), float *out, float hue, float sat, float val, float *in, float fac)
53 {
54         if(fac != 0 && (hue != 0.5f || sat != 1 || val != 1)) {
55                 float col[3], hsv[3], mfac= 1.0f - fac;
56                 
57                 rgb_to_hsv(in[0], in[1], in[2], hsv, hsv+1, hsv+2);
58                 hsv[0]+= (hue - 0.5f);
59                 if(hsv[0]>1.0f) hsv[0]-=1.0f; else if(hsv[0]<0.0f) hsv[0]+= 1.0f;
60                 hsv[1]*= sat;
61                 if(hsv[1]>1.0f) hsv[1]= 1.0f; else if(hsv[1]<0.0f) hsv[1]= 0.0f;
62                 hsv[2]*= val;
63                 if(hsv[2]>1.0f) hsv[2]= 1.0f; else if(hsv[2]<0.0f) hsv[2]= 0.0f;
64                 hsv_to_rgb(hsv[0], hsv[1], hsv[2], col, col+1, col+2);
65                 
66                 out[0]= mfac*in[0] + fac*col[0];
67                 out[1]= mfac*in[1] + fac*col[1];
68                 out[2]= mfac*in[2] + fac*col[2];
69         }
70         else {
71                 QUATCOPY(out, in);
72         }
73 }
74
75 static void colorfn(float *out, TexParams *p, bNode *node, bNodeStack **in, short thread)
76 {       
77         float hue = tex_input_value(in[0], p, thread);
78         float sat = tex_input_value(in[1], p, thread);
79         float val = tex_input_value(in[2], p, thread);
80         float fac = tex_input_value(in[3], p, thread);
81         
82         float col[4];
83         tex_input_rgba(col, in[4], p, thread);
84         
85         hue += 0.5f; /* [-.5, .5] -> [0, 1] */
86         
87         do_hue_sat_fac(node, out, hue, sat, val, col, fac);
88         
89         out[3] = col[3];
90 }
91
92 static void exec(void *data, bNode *node, bNodeStack **in, bNodeStack **out)
93 {
94         tex_output(node, in, out[0], &colorfn, data);
95 }
96
97 void register_node_type_tex_hue_sat(ListBase *lb)
98 {
99         static bNodeType ntype;
100         
101         node_type_base(&ntype, TEX_NODE_HUE_SAT, "Hue Saturation Value", NODE_CLASS_OP_COLOR, NODE_OPTIONS);
102         node_type_socket_templates(&ntype, inputs, outputs);
103         node_type_size(&ntype, 150, 80, 250);
104         node_type_exec(&ntype, exec);
105         
106         nodeRegisterType(lb, &ntype);
107 }