873d24116a1744ade415d3e87db0da553065be70
[blender.git] / source / blender / draw / engines / workbench / workbench_materials.c
1 /*
2  * Copyright 2016, Blender Foundation.
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  * Contributor(s): Blender Institute
19  *
20  */
21
22 /** \file workbench_materials.c
23  *  \ingroup draw_engine
24  */
25
26 #include "workbench_private.h"
27 #include "GPU_shader.h"
28
29 /* *********** STATIC *********** */
30 static struct {
31         struct GPUShader *depth_sh;
32
33         /* Solid flat mode */
34         struct GPUShader *solid_flat_sh;
35
36         /* Solid studio mode */
37         struct GPUShader *solid_studio_sh;
38
39 } e_data = {NULL};
40
41 /* Shaders */
42 extern char datatoc_solid_flat_frag_glsl[];
43 extern char datatoc_solid_studio_frag_glsl[];
44 extern char datatoc_workbench_vert_glsl[];
45 extern char datatoc_workbench_studio_vert_glsl[];
46 extern char datatoc_workbench_diffuse_lib_glsl[];
47
48 /* Functions */
49 static uint get_material_hash(const float color[3])
50 {
51         uint r = (uint)(color[0] * 512);
52         uint g = (uint)(color[1] * 512);
53         uint b = (uint)(color[2] * 512);
54
55         return r + g * 4096 + b * 4096 * 4096;
56 }
57
58 WORKBENCH_MaterialData *workbench_get_or_create_solid_flat_material_data(WORKBENCH_Data *vedata, const float color[3])
59 {
60         WORKBENCH_StorageList *stl = vedata->stl;
61         WORKBENCH_PassList *psl = vedata->psl;
62         WORKBENCH_PrivateData *wpd = stl->g_data;
63
64         uint hash = get_material_hash(color);
65         WORKBENCH_MaterialData *material;
66
67         material = BLI_ghash_lookup(wpd->material_hash, SET_UINT_IN_POINTER(hash));
68         if (material == NULL) {
69                 material = MEM_mallocN(sizeof(WORKBENCH_MaterialData), "WORKBENCH_MaterialData");
70                 material->shgrp = DRW_shgroup_create(e_data.solid_flat_sh, psl->solid_pass);
71                 material->color[0] = color[0];
72                 material->color[1] = color[1];
73                 material->color[2] = color[2];
74                 DRW_shgroup_uniform_vec3(material->shgrp, "color", material->color, 1);
75                 BLI_ghash_insert(wpd->material_hash, SET_UINT_IN_POINTER(hash), material);
76         }
77         return material;
78 }
79
80 WORKBENCH_MaterialData *workbench_get_or_create_solid_studio_material_data(WORKBENCH_Data *vedata, const float color[3])
81 {
82         WORKBENCH_StorageList *stl = vedata->stl;
83         WORKBENCH_PassList *psl = vedata->psl;
84         WORKBENCH_PrivateData *wpd = stl->g_data;
85
86         uint hash = get_material_hash(color);
87         WORKBENCH_MaterialData *material;
88
89         material = BLI_ghash_lookup(wpd->material_hash, SET_UINT_IN_POINTER(hash));
90         if (material == NULL) {
91                 material = MEM_mallocN(sizeof(WORKBENCH_MaterialData), "WORKBENCH_MaterialData");
92                 material->shgrp = DRW_shgroup_create(e_data.solid_studio_sh, psl->solid_pass);
93                 material->color[0] = color[0];
94                 material->color[1] = color[1];
95                 material->color[2] = color[2];
96                 DRW_shgroup_uniform_vec3(material->shgrp, "color", material->color, 1);
97                 BLI_ghash_insert(wpd->material_hash, SET_UINT_IN_POINTER(hash), material);
98         }
99         return material;
100 }
101
102 void workbench_materials_engine_init(void)
103 {
104         if (!e_data.depth_sh) {
105                 /* Depth pass */
106                 e_data.depth_sh = DRW_shader_create_3D_depth_only();
107
108                 /* Solid flat */
109                 e_data.solid_flat_sh = DRW_shader_create(datatoc_workbench_vert_glsl, NULL, datatoc_solid_flat_frag_glsl, "\n");
110                 e_data.solid_studio_sh = DRW_shader_create(datatoc_workbench_studio_vert_glsl, NULL, datatoc_solid_studio_frag_glsl, datatoc_workbench_diffuse_lib_glsl);
111         }
112 }
113
114 void workbench_materials_engine_finish(void)
115 {
116         DRW_SHADER_FREE_SAFE(e_data.solid_flat_sh);
117         DRW_SHADER_FREE_SAFE(e_data.solid_studio_sh);
118 }
119
120 void workbench_materials_cache_init(WORKBENCH_Data *vedata)
121 {
122         WORKBENCH_StorageList *stl = vedata->stl;
123         WORKBENCH_PassList *psl = vedata->psl;
124         WORKBENCH_PrivateData *wpd = stl->g_data;
125
126         wpd->depth_shgrp = DRW_shgroup_create(e_data.depth_sh, psl->depth_pass);
127         wpd->material_hash = BLI_ghash_ptr_new("Workbench material_hash");
128 }
129
130 void workbench_materials_cache_finish(WORKBENCH_Data *vedata)
131 {
132         WORKBENCH_StorageList *stl = vedata->stl;
133         WORKBENCH_PrivateData *wpd = stl->g_data;
134         BLI_ghash_free(wpd->material_hash, NULL, MEM_freeN);
135 }