30171eea9e2b4d6c384f8989dadfb76b199aa291
[blender.git] / source / blender / nodes / shader / nodes / node_shader_tex_musgrave.c
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) 2005 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 #include "../node_shader_util.h"
29
30 /* **************** MUSGRAVE ******************** */
31
32 static bNodeSocketTemplate sh_node_tex_musgrave_in[] = {
33         {       SOCK_VECTOR, 1, N_("Vector"),           0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, PROP_NONE, SOCK_HIDE_VALUE},
34         {       SOCK_FLOAT, 1, N_("Scale"),                     5.0f, 0.0f, 0.0f, 0.0f, -1000.0f, 1000.0f},
35         {       SOCK_FLOAT, 1, N_("Detail"),            2.0f, 0.0f, 0.0f, 0.0f, 0.0f, 16.0f},
36         {       SOCK_FLOAT, 1, N_("Dimension"),         2.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1000.0f},
37         {       SOCK_FLOAT, 1, N_("Lacunarity"),        1.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1000.0f},
38         {       SOCK_FLOAT, 1, N_("Offset"),            0.0f, 0.0f, 0.0f, 0.0f, -1000.0f, 1000.0f},
39         {       SOCK_FLOAT, 1, N_("Gain"),                      1.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1000.0f},
40         {       -1, 0, ""       }
41 };
42
43 static bNodeSocketTemplate sh_node_tex_musgrave_out[] = {
44         {       SOCK_RGBA, 0, N_("Color"),              0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, PROP_NONE, SOCK_NO_INTERNAL_LINK},
45         {       SOCK_FLOAT, 0, N_("Fac"),               0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, PROP_FACTOR, SOCK_NO_INTERNAL_LINK},
46         {       -1, 0, ""       }
47 };
48
49 static void node_shader_init_tex_musgrave(bNodeTree *UNUSED(ntree), bNode *node)
50 {
51         NodeTexMusgrave *tex = MEM_callocN(sizeof(NodeTexMusgrave), "NodeTexMusgrave");
52         BKE_texture_mapping_default(&tex->base.tex_mapping, TEXMAP_TYPE_POINT);
53         BKE_texture_colormapping_default(&tex->base.color_mapping);
54         tex->musgrave_type = SHD_MUSGRAVE_FBM;
55
56         node->storage = tex;
57 }
58
59 static int node_shader_gpu_tex_musgrave(GPUMaterial *mat, bNode *node, bNodeExecData *UNUSED(execdata), GPUNodeStack *in, GPUNodeStack *out)
60 {
61         if (!in[0].link) {
62                 in[0].link = GPU_attribute(CD_ORCO, "");
63                 GPU_link(mat, "generated_from_orco", in[0].link, &in[0].link);
64         }
65
66         node_shader_gpu_tex_mapping(mat, node, in, out);
67
68         NodeTexMusgrave *tex = (NodeTexMusgrave *)node->storage;
69         float type = tex->musgrave_type;
70
71         return GPU_stack_link(mat, node, "node_tex_musgrave", in, out, GPU_constant(&type));
72 }
73
74 /* node type definition */
75 void register_node_type_sh_tex_musgrave(void)
76 {
77         static bNodeType ntype;
78
79         sh_node_type_base(&ntype, SH_NODE_TEX_MUSGRAVE, "Musgrave Texture", NODE_CLASS_TEXTURE, 0);
80         node_type_socket_templates(&ntype, sh_node_tex_musgrave_in, sh_node_tex_musgrave_out);
81         node_type_size_preset(&ntype, NODE_SIZE_MIDDLE);
82         node_type_init(&ntype, node_shader_init_tex_musgrave);
83         node_type_storage(&ntype, "NodeTexMusgrave", node_free_standard_storage, node_copy_standard_storage);
84         node_type_gpu(&ntype, node_shader_gpu_tex_musgrave);
85
86         nodeRegisterType(&ntype);
87 }