1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  *
16  * The Original Code is Copyright (C) 2005 Blender Foundation.
17  * All rights reserved.
18  */
19 
20 #include "../node_shader_util.h"
21 
22 /* **************** Layer Weight ******************** */
23 
24 static bNodeSocketTemplate sh_node_layer_weight_in[] = {
25     {SOCK_FLOAT, N_("Blend"), 0.5f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f},
26     {SOCK_VECTOR, N_("Normal"), 0.0f, 0.0f, 0.0f, 1.0f, -1.0f, 1.0f, PROP_NONE, SOCK_HIDE_VALUE},
27     {-1, ""},
28 };
29 
30 static bNodeSocketTemplate sh_node_layer_weight_out[] = {
31     {SOCK_FLOAT, N_("Fresnel"), 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f},
32     {SOCK_FLOAT, N_("Facing"), 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f},
33     {-1, ""},
34 };
35 
node_shader_gpu_layer_weight(GPUMaterial * mat,bNode * node,bNodeExecData * UNUSED (execdata),GPUNodeStack * in,GPUNodeStack * out)36 static int node_shader_gpu_layer_weight(GPUMaterial *mat,
37                                         bNode *node,
38                                         bNodeExecData *UNUSED(execdata),
39                                         GPUNodeStack *in,
40                                         GPUNodeStack *out)
41 {
42   if (!in[1].link) {
43     in[1].link = GPU_builtin(GPU_VIEW_NORMAL);
44   }
45   else {
46     GPU_link(
47         mat, "direction_transform_m4v3", in[1].link, GPU_builtin(GPU_VIEW_MATRIX), &in[1].link);
48   }
49 
50   return GPU_stack_link(mat, node, "node_layer_weight", in, out, GPU_builtin(GPU_VIEW_POSITION));
51 }
52 
node_shader_exec_layer_weight(void * UNUSED (data),int UNUSED (thread),bNode * UNUSED (node),bNodeExecData * UNUSED (execdata),bNodeStack ** UNUSED (in),bNodeStack ** UNUSED (out))53 static void node_shader_exec_layer_weight(void *UNUSED(data),
54                                           int UNUSED(thread),
55                                           bNode *UNUSED(node),
56                                           bNodeExecData *UNUSED(execdata),
57                                           bNodeStack **UNUSED(in),
58                                           bNodeStack **UNUSED(out))
59 {
60 }
61 
62 /* node type definition */
register_node_type_sh_layer_weight(void)63 void register_node_type_sh_layer_weight(void)
64 {
65   static bNodeType ntype;
66 
67   sh_node_type_base(&ntype, SH_NODE_LAYER_WEIGHT, "Layer Weight", NODE_CLASS_INPUT, 0);
68   node_type_socket_templates(&ntype, sh_node_layer_weight_in, sh_node_layer_weight_out);
69   node_type_init(&ntype, NULL);
70   node_type_storage(&ntype, "", NULL, NULL);
71   node_type_gpu(&ntype, node_shader_gpu_layer_weight);
72   node_type_exec(&ntype, NULL, NULL, node_shader_exec_layer_weight);
73 
74   nodeRegisterType(&ntype);
75 }
76