f18310d351d9405a7ccc3c44dc2484723b25317f
[blender.git] / source / blender / nodes / shader / node_shader_util.h
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 /** \file blender/nodes/shader/node_shader_util.h
29  *  \ingroup nodes
30  */
31
32
33 #ifndef NODE_SHADER_UTIL_H_
34 #define NODE_SHADER_UTIL_H_
35
36 #include <math.h>
37 #include <float.h>
38 #include <string.h>
39
40 #include "MEM_guardedalloc.h"
41
42 #include "DNA_color_types.h"
43 #include "DNA_customdata_types.h"
44 #include "DNA_ID.h"
45 #include "DNA_image_types.h"
46 #include "DNA_material_types.h"
47 #include "DNA_node_types.h"
48 #include "DNA_object_types.h"
49 #include "DNA_scene_types.h"
50 #include "DNA_texture_types.h"
51
52 #include "BKE_blender.h"
53 #include "BKE_colortools.h"
54 #include "BKE_global.h"
55 #include "BKE_image.h"
56 #include "BKE_main.h"
57 #include "BKE_material.h"
58 #include "BKE_node.h"
59 #include "BKE_texture.h"
60
61 #include "BKE_library.h"
62
63 #include "NOD_shader.h"
64 #include "node_util.h"
65
66 #include "BLI_math.h"
67 #include "BLI_blenlib.h"
68 #include "BLI_rand.h"
69 #include "BLI_threads.h"
70 #include "BLI_utildefines.h"
71
72 #include "IMB_imbuf_types.h"
73 #include "IMB_imbuf.h"
74
75 #include "RE_pipeline.h"
76 #include "RE_render_ext.h"
77 #include "RE_shader_ext.h"
78
79 #include "GPU_material.h"
80
81 /* ********* exec data struct, remains internal *********** */
82
83 typedef struct ShaderCallData {
84         ShadeInput *shi;                /* from render pipe */
85         ShadeResult *shr;               /* from render pipe */
86
87         TexResult *texres;
88         float *co;
89         float *dxt, *dyt;
90 } ShaderCallData;
91
92 /* output socket defines */
93 #define GEOM_OUT_GLOB   0
94 #define GEOM_OUT_LOCAL  1
95 #define GEOM_OUT_VIEW   2
96 #define GEOM_OUT_ORCO   3
97 #define GEOM_OUT_UV             4
98 #define GEOM_OUT_NORMAL 5
99 #define GEOM_OUT_VCOL   6
100 #define GEOM_OUT_FRONTBACK      7
101
102
103 /* input socket defines */
104 #define MAT_IN_COLOR    0
105 #define MAT_IN_SPEC             1
106 #define MAT_IN_REFL             2
107 #define MAT_IN_NORMAL   3
108 #define MAT_IN_MIR              4
109 #define MAT_IN_AMB              5
110 #define MAT_IN_EMIT     6
111 #define MAT_IN_SPECTRA  7
112 #define MAT_IN_RAY_MIRROR       8
113 #define MAT_IN_ALPHA    9
114 #define MAT_IN_TRANSLUCENCY     10
115 #define NUM_MAT_IN              11      /* for array size */
116
117 /* output socket defines */
118 #define MAT_OUT_COLOR           0
119 #define MAT_OUT_ALPHA           1
120 #define MAT_OUT_NORMAL  2
121 #define MAT_OUT_DIFFUSE 3
122 #define MAT_OUT_SPEC            4
123 #define MAT_OUT_AO              5
124
125
126 extern void node_ID_title_cb(void *node_v, void *unused_v);
127 void nodestack_get_vec(float *in, short type_in, bNodeStack *ns);
128
129 void node_gpu_stack_from_data(struct GPUNodeStack *gs, int type, struct bNodeStack *ns);
130 void node_data_from_gpu_stack(struct bNodeStack *ns, struct GPUNodeStack *gs);
131
132 void ntreeExecGPUNodes(struct bNodeTreeExec *exec, struct GPUMaterial *mat, int do_outputs);
133
134 #endif