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 * Copyright 2016, Blender Foundation.
17 */
18
19 /** \file
20 * \ingroup draw_engine
21 *
22 * All specific data handler for Objects, Lights, ViewLayers, ...
23 */
24
25 #include "DRW_render.h"
26
27 #include "BLI_ghash.h"
28 #include "BLI_memblock.h"
29
30 #include "BKE_duplilist.h"
31 #include "BKE_modifier.h"
32 #include "BKE_object.h"
33
34 #include "DEG_depsgraph_query.h"
35
36 #include "GPU_vertex_buffer.h"
37
38 #include "eevee_lightcache.h"
39 #include "eevee_private.h"
40
41 /* Motion Blur data. */
42
eevee_motion_blur_mesh_data_free(void * val)43 static void eevee_motion_blur_mesh_data_free(void *val)
44 {
45 EEVEE_GeometryMotionData *geom_mb = (EEVEE_GeometryMotionData *)val;
46 EEVEE_HairMotionData *hair_mb = (EEVEE_HairMotionData *)val;
47 switch (geom_mb->type) {
48 case EEVEE_MOTION_DATA_HAIR:
49 for (int j = 0; j < hair_mb->psys_len; j++) {
50 for (int i = 0; i < ARRAY_SIZE(hair_mb->psys[0].hair_pos); i++) {
51 GPU_VERTBUF_DISCARD_SAFE(hair_mb->psys[j].hair_pos[i]);
52 }
53 for (int i = 0; i < ARRAY_SIZE(hair_mb->psys[0].hair_pos); i++) {
54 DRW_TEXTURE_FREE_SAFE(hair_mb->psys[j].hair_pos_tx[i]);
55 }
56 }
57 break;
58
59 case EEVEE_MOTION_DATA_MESH:
60 for (int i = 0; i < ARRAY_SIZE(geom_mb->vbo); i++) {
61 GPU_VERTBUF_DISCARD_SAFE(geom_mb->vbo[i]);
62 }
63 break;
64 }
65 MEM_freeN(val);
66 }
67
eevee_object_key_hash(const void * key)68 static uint eevee_object_key_hash(const void *key)
69 {
70 EEVEE_ObjectKey *ob_key = (EEVEE_ObjectKey *)key;
71 uint hash = BLI_ghashutil_ptrhash(ob_key->ob);
72 hash = BLI_ghashutil_combine_hash(hash, BLI_ghashutil_ptrhash(ob_key->parent));
73 for (int i = 0; i < MAX_DUPLI_RECUR; i++) {
74 if (ob_key->id[i] != 0) {
75 hash = BLI_ghashutil_combine_hash(hash, BLI_ghashutil_inthash(ob_key->id[i]));
76 }
77 else {
78 break;
79 }
80 }
81 return hash;
82 }
83
84 /* Return false if equal. */
eevee_object_key_cmp(const void * a,const void * b)85 static bool eevee_object_key_cmp(const void *a, const void *b)
86 {
87 EEVEE_ObjectKey *key_a = (EEVEE_ObjectKey *)a;
88 EEVEE_ObjectKey *key_b = (EEVEE_ObjectKey *)b;
89
90 if (key_a->ob != key_b->ob) {
91 return true;
92 }
93 if (key_a->parent != key_b->parent) {
94 return true;
95 }
96 if (memcmp(key_a->id, key_b->id, sizeof(key_a->id)) != 0) {
97 return true;
98 }
99 return false;
100 }
101
EEVEE_motion_blur_data_init(EEVEE_MotionBlurData * mb)102 void EEVEE_motion_blur_data_init(EEVEE_MotionBlurData *mb)
103 {
104 if (mb->object == NULL) {
105 mb->object = BLI_ghash_new(eevee_object_key_hash, eevee_object_key_cmp, "EEVEE Object Motion");
106 }
107 if (mb->geom == NULL) {
108 mb->geom = BLI_ghash_new(BLI_ghashutil_ptrhash, BLI_ghashutil_ptrcmp, "EEVEE Mesh Motion");
109 }
110 }
111
EEVEE_motion_blur_data_free(EEVEE_MotionBlurData * mb)112 void EEVEE_motion_blur_data_free(EEVEE_MotionBlurData *mb)
113 {
114 if (mb->object) {
115 BLI_ghash_free(mb->object, MEM_freeN, MEM_freeN);
116 mb->object = NULL;
117 }
118 if (mb->geom) {
119 BLI_ghash_free(mb->geom, NULL, eevee_motion_blur_mesh_data_free);
120 mb->geom = NULL;
121 }
122 }
123
EEVEE_motion_blur_object_data_get(EEVEE_MotionBlurData * mb,Object * ob,bool hair)124 EEVEE_ObjectMotionData *EEVEE_motion_blur_object_data_get(EEVEE_MotionBlurData *mb,
125 Object *ob,
126 bool hair)
127 {
128 if (mb->object == NULL) {
129 return NULL;
130 }
131
132 EEVEE_ObjectKey key, *key_p;
133 /* Small hack to avoid another comparisson. */
134 key.ob = (Object *)((char *)ob + hair);
135 DupliObject *dup = DRW_object_get_dupli(ob);
136 if (dup) {
137 key.parent = DRW_object_get_dupli_parent(ob);
138 memcpy(key.id, dup->persistent_id, sizeof(key.id));
139 }
140 else {
141 key.parent = key.ob;
142 memset(key.id, 0, sizeof(key.id));
143 }
144
145 EEVEE_ObjectMotionData *ob_step = BLI_ghash_lookup(mb->object, &key);
146 if (ob_step == NULL) {
147 key_p = MEM_mallocN(sizeof(*key_p), __func__);
148 memcpy(key_p, &key, sizeof(*key_p));
149
150 ob_step = MEM_callocN(sizeof(EEVEE_ObjectMotionData), __func__);
151
152 BLI_ghash_insert(mb->object, key_p, ob_step);
153 }
154 return ob_step;
155 }
156
motion_blur_deform_data_get(EEVEE_MotionBlurData * mb,Object * ob,bool hair)157 static void *motion_blur_deform_data_get(EEVEE_MotionBlurData *mb, Object *ob, bool hair)
158 {
159 if (mb->geom == NULL) {
160 return NULL;
161 }
162 DupliObject *dup = DRW_object_get_dupli(ob);
163 void *key;
164 if (dup) {
165 key = dup->ob;
166 }
167 else {
168 key = ob;
169 }
170 /* Only use data for object that have no modifiers. */
171 if (!BKE_object_is_modified(DRW_context_state_get()->scene, ob)) {
172 key = ob->data;
173 }
174 key = (char *)key + (int)hair;
175 EEVEE_GeometryMotionData *geom_step = BLI_ghash_lookup(mb->geom, key);
176 if (geom_step == NULL) {
177 if (hair) {
178 EEVEE_HairMotionData *hair_step;
179 /* Ugly, we allocate for each modifiers and just fill based on modifier index in the list. */
180 int psys_len = (ob->type != OB_HAIR) ? BLI_listbase_count(&ob->modifiers) : 1;
181 hair_step = MEM_callocN(sizeof(EEVEE_HairMotionData) + sizeof(hair_step->psys[0]) * psys_len,
182 __func__);
183 hair_step->psys_len = psys_len;
184 geom_step = (EEVEE_GeometryMotionData *)hair_step;
185 geom_step->type = EEVEE_MOTION_DATA_HAIR;
186 }
187 else {
188 geom_step = MEM_callocN(sizeof(EEVEE_GeometryMotionData), __func__);
189 geom_step->type = EEVEE_MOTION_DATA_MESH;
190 }
191 BLI_ghash_insert(mb->geom, key, geom_step);
192 }
193 return geom_step;
194 }
195
EEVEE_motion_blur_geometry_data_get(EEVEE_MotionBlurData * mb,Object * ob)196 EEVEE_GeometryMotionData *EEVEE_motion_blur_geometry_data_get(EEVEE_MotionBlurData *mb, Object *ob)
197 {
198 return motion_blur_deform_data_get(mb, ob, false);
199 }
200
EEVEE_motion_blur_hair_data_get(EEVEE_MotionBlurData * mb,Object * ob)201 EEVEE_HairMotionData *EEVEE_motion_blur_hair_data_get(EEVEE_MotionBlurData *mb, Object *ob)
202 {
203 return motion_blur_deform_data_get(mb, ob, true);
204 }
205
206 /* View Layer data. */
207
EEVEE_view_layer_data_free(void * storage)208 void EEVEE_view_layer_data_free(void *storage)
209 {
210 EEVEE_ViewLayerData *sldata = (EEVEE_ViewLayerData *)storage;
211
212 /* Lights */
213 MEM_SAFE_FREE(sldata->lights);
214 DRW_UBO_FREE_SAFE(sldata->light_ubo);
215 DRW_UBO_FREE_SAFE(sldata->shadow_ubo);
216 GPU_FRAMEBUFFER_FREE_SAFE(sldata->shadow_fb);
217 DRW_TEXTURE_FREE_SAFE(sldata->shadow_cube_pool);
218 DRW_TEXTURE_FREE_SAFE(sldata->shadow_cascade_pool);
219 for (int i = 0; i < 2; i++) {
220 MEM_SAFE_FREE(sldata->shcasters_buffers[i].bbox);
221 MEM_SAFE_FREE(sldata->shcasters_buffers[i].update);
222 }
223
224 if (sldata->fallback_lightcache) {
225 EEVEE_lightcache_free(sldata->fallback_lightcache);
226 sldata->fallback_lightcache = NULL;
227 }
228
229 /* Probes */
230 MEM_SAFE_FREE(sldata->probes);
231 DRW_UBO_FREE_SAFE(sldata->probe_ubo);
232 DRW_UBO_FREE_SAFE(sldata->grid_ubo);
233 DRW_UBO_FREE_SAFE(sldata->planar_ubo);
234 DRW_UBO_FREE_SAFE(sldata->common_ubo);
235
236 DRW_UBO_FREE_SAFE(sldata->renderpass_ubo.combined);
237 DRW_UBO_FREE_SAFE(sldata->renderpass_ubo.diff_color);
238 DRW_UBO_FREE_SAFE(sldata->renderpass_ubo.diff_light);
239 DRW_UBO_FREE_SAFE(sldata->renderpass_ubo.spec_color);
240 DRW_UBO_FREE_SAFE(sldata->renderpass_ubo.spec_light);
241 DRW_UBO_FREE_SAFE(sldata->renderpass_ubo.emit);
242 DRW_UBO_FREE_SAFE(sldata->renderpass_ubo.environment);
243
244 if (sldata->material_cache) {
245 BLI_memblock_destroy(sldata->material_cache, NULL);
246 sldata->material_cache = NULL;
247 }
248 }
249
EEVEE_view_layer_data_get(void)250 EEVEE_ViewLayerData *EEVEE_view_layer_data_get(void)
251 {
252 return (EEVEE_ViewLayerData *)DRW_view_layer_engine_data_get(&draw_engine_eevee_type);
253 }
254
eevee_view_layer_init(EEVEE_ViewLayerData * sldata)255 static void eevee_view_layer_init(EEVEE_ViewLayerData *sldata)
256 {
257 sldata->common_ubo = GPU_uniformbuf_create(sizeof(sldata->common_data));
258 }
259
EEVEE_view_layer_data_ensure_ex(struct ViewLayer * view_layer)260 EEVEE_ViewLayerData *EEVEE_view_layer_data_ensure_ex(struct ViewLayer *view_layer)
261 {
262 EEVEE_ViewLayerData **sldata = (EEVEE_ViewLayerData **)DRW_view_layer_engine_data_ensure_ex(
263 view_layer, &draw_engine_eevee_type, &EEVEE_view_layer_data_free);
264
265 if (*sldata == NULL) {
266 *sldata = MEM_callocN(sizeof(**sldata), "EEVEE_ViewLayerData");
267 eevee_view_layer_init(*sldata);
268 }
269
270 return *sldata;
271 }
272
EEVEE_view_layer_data_ensure(void)273 EEVEE_ViewLayerData *EEVEE_view_layer_data_ensure(void)
274 {
275 EEVEE_ViewLayerData **sldata = (EEVEE_ViewLayerData **)DRW_view_layer_engine_data_ensure(
276 &draw_engine_eevee_type, &EEVEE_view_layer_data_free);
277
278 if (*sldata == NULL) {
279 *sldata = MEM_callocN(sizeof(**sldata), "EEVEE_ViewLayerData");
280 eevee_view_layer_init(*sldata);
281 }
282
283 return *sldata;
284 }
285
286 /* Object data. */
287
eevee_object_data_init(DrawData * dd)288 static void eevee_object_data_init(DrawData *dd)
289 {
290 EEVEE_ObjectEngineData *eevee_data = (EEVEE_ObjectEngineData *)dd;
291 eevee_data->shadow_caster_id = -1;
292 eevee_data->need_update = false;
293 eevee_data->geom_update = false;
294 }
295
EEVEE_object_data_get(Object * ob)296 EEVEE_ObjectEngineData *EEVEE_object_data_get(Object *ob)
297 {
298 if (ELEM(ob->type, OB_LIGHTPROBE, OB_LAMP)) {
299 return NULL;
300 }
301 return (EEVEE_ObjectEngineData *)DRW_drawdata_get(&ob->id, &draw_engine_eevee_type);
302 }
303
EEVEE_object_data_ensure(Object * ob)304 EEVEE_ObjectEngineData *EEVEE_object_data_ensure(Object *ob)
305 {
306 BLI_assert(!ELEM(ob->type, OB_LIGHTPROBE, OB_LAMP));
307 return (EEVEE_ObjectEngineData *)DRW_drawdata_ensure(&ob->id,
308 &draw_engine_eevee_type,
309 sizeof(EEVEE_ObjectEngineData),
310 eevee_object_data_init,
311 NULL);
312 }
313
314 /* Light probe data. */
315
eevee_lightprobe_data_init(DrawData * dd)316 static void eevee_lightprobe_data_init(DrawData *dd)
317 {
318 EEVEE_LightProbeEngineData *ped = (EEVEE_LightProbeEngineData *)dd;
319 ped->need_update = false;
320 }
321
EEVEE_lightprobe_data_get(Object * ob)322 EEVEE_LightProbeEngineData *EEVEE_lightprobe_data_get(Object *ob)
323 {
324 if (ob->type != OB_LIGHTPROBE) {
325 return NULL;
326 }
327 return (EEVEE_LightProbeEngineData *)DRW_drawdata_get(&ob->id, &draw_engine_eevee_type);
328 }
329
EEVEE_lightprobe_data_ensure(Object * ob)330 EEVEE_LightProbeEngineData *EEVEE_lightprobe_data_ensure(Object *ob)
331 {
332 BLI_assert(ob->type == OB_LIGHTPROBE);
333 return (EEVEE_LightProbeEngineData *)DRW_drawdata_ensure(&ob->id,
334 &draw_engine_eevee_type,
335 sizeof(EEVEE_LightProbeEngineData),
336 eevee_lightprobe_data_init,
337 NULL);
338 }
339
340 /* Light data. */
341
eevee_light_data_init(DrawData * dd)342 static void eevee_light_data_init(DrawData *dd)
343 {
344 EEVEE_LightEngineData *led = (EEVEE_LightEngineData *)dd;
345 led->need_update = true;
346 }
347
EEVEE_light_data_get(Object * ob)348 EEVEE_LightEngineData *EEVEE_light_data_get(Object *ob)
349 {
350 if (ob->type != OB_LAMP) {
351 return NULL;
352 }
353 return (EEVEE_LightEngineData *)DRW_drawdata_get(&ob->id, &draw_engine_eevee_type);
354 }
355
EEVEE_light_data_ensure(Object * ob)356 EEVEE_LightEngineData *EEVEE_light_data_ensure(Object *ob)
357 {
358 BLI_assert(ob->type == OB_LAMP);
359 return (EEVEE_LightEngineData *)DRW_drawdata_ensure(&ob->id,
360 &draw_engine_eevee_type,
361 sizeof(EEVEE_LightEngineData),
362 eevee_light_data_init,
363 NULL);
364 }
365
366 /* World data. */
367
eevee_world_data_init(DrawData * dd)368 static void eevee_world_data_init(DrawData *dd)
369 {
370 EEVEE_WorldEngineData *wed = (EEVEE_WorldEngineData *)dd;
371 wed->dd.recalc |= 1;
372 }
373
EEVEE_world_data_get(World * wo)374 EEVEE_WorldEngineData *EEVEE_world_data_get(World *wo)
375 {
376 return (EEVEE_WorldEngineData *)DRW_drawdata_get(&wo->id, &draw_engine_eevee_type);
377 }
378
EEVEE_world_data_ensure(World * wo)379 EEVEE_WorldEngineData *EEVEE_world_data_ensure(World *wo)
380 {
381 return (EEVEE_WorldEngineData *)DRW_drawdata_ensure(&wo->id,
382 &draw_engine_eevee_type,
383 sizeof(EEVEE_WorldEngineData),
384 eevee_world_data_init,
385 NULL);
386 }
387