griefith/test2
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
93bb7cbd91d784792fa332a40d624cf5c4e5cebf
test2/source/blender/draw/tests/eevee_test.cc
Jeroen Bakker c2c1d6f67b Metal: EEVEE Next shadow unit test enablement 
Ensure correct SSBO bindings are present for shadow tests.
Metal validation errors occur if SSBO bindings that are expected are
not bound. In this case, we can bind empty SSBOs, but these should
be of the correct type for the tests.

Also adding missing zero-initializations for required members within
LightData. Without these, unit tests fail with various issues including
prevalence of OOB reads.

Co-authored-by: Michael Parkin-White <mparkinwhite@apple.com>
Pull Request: https://projects.blender.org/blender/blender/pulls/109645
2023-07-03 14:24:34 +02:00

1293 lines
44 KiB
C++
Raw Blame History

/* SPDX-FileCopyrightText: 2023 Blender Foundation
*
* SPDX-License-Identifier: Apache-2.0 */
#include "testing/testing.h"
#include "BKE_context.h"
#include "BKE_idtype.h"
#include "BKE_main.h"
#include "BKE_node.hh"
#include "BKE_object.h"
#include "DEG_depsgraph.h"
#include "RNA_define.h"
#include "GPU_batch.h"
#include "draw_shader.h"
#include "draw_testing.hh"
#include "engines/eevee_next/eevee_instance.hh"
namespace blender::draw {
using namespace blender::eevee;
/* Replace with template version that is not GPU only. */
using ShadowPageCacheBuf = draw::StorageArrayBuffer<uint2, SHADOW_MAX_PAGE, false>;
using ShadowTileDataBuf = draw::StorageArrayBuffer<ShadowTileDataPacked, SHADOW_MAX_TILE, false>;
static void test_eevee_shadow_shift_clear()
{
GPU_render_begin();
ShadowTileMapDataBuf tilemaps_data = {"tilemaps_data"};
ShadowTileDataBuf tiles_data = {"tiles_data"};
ShadowTileMapClipBuf tilemaps_clip = {"tilemaps_clip"};
ShadowPageCacheBuf pages_cached_data_ = {"pages_cached_data_"};
int tiles_index = 1;
int tile_lod0 = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 5;
int tile_lod1 = tile_lod0 + square_i(SHADOW_TILEMAP_RES);
{
ShadowTileMapData tilemap = {};
tilemap.tiles_index = tiles_index * SHADOW_TILEDATA_PER_TILEMAP;
tilemap.grid_shift = int2(SHADOW_TILEMAP_RES);
tilemap.projection_type = SHADOW_PROJECTION_CUBEFACE;
tilemaps_data.append(tilemap);
tilemaps_data.push_update();
}
{
ShadowTileData tile;
tile.page = uint2(1, 2);
tile.is_used = true;
tile.do_update = true;
tiles_data[tile_lod0] = shadow_tile_pack(tile);
tile.page = uint2(3, 4);
tile.is_used = false;
tile.do_update = false;
tiles_data[tile_lod1] = shadow_tile_pack(tile);
tiles_data.push_update();
}
GPUShader *sh = GPU_shader_create_from_info_name("eevee_shadow_tilemap_init");
PassSimple pass("Test");
pass.shader_set(sh);
pass.bind_ssbo("tilemaps_buf", tilemaps_data);
pass.bind_ssbo("tilemaps_clip_buf", tilemaps_clip);
pass.bind_ssbo("tiles_buf", tiles_data);
pass.bind_ssbo("pages_cached_buf", pages_cached_data_);
pass.dispatch(int3(1, 1, tilemaps_data.size()));
Manager manager;
manager.submit(pass);
GPU_memory_barrier(GPU_BARRIER_BUFFER_UPDATE);
tilemaps_data.read();
tiles_data.read();
EXPECT_EQ(tilemaps_data[0].grid_offset, int2(0));
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_lod0]).page, uint2(1, 2));
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_lod0]).is_used, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_lod0]).do_update, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_lod1]).page, uint2(3, 4));
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_lod1]).is_used, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_lod1]).do_update, true);
GPU_shader_free(sh);
DRW_shaders_free();
GPU_render_end();
}
DRAW_TEST(eevee_shadow_shift_clear)
static void test_eevee_shadow_shift()
{
GPU_render_begin();
ShadowTileMapDataBuf tilemaps_data = {"tilemaps_data"};
ShadowTileDataBuf tiles_data = {"tiles_data"};
ShadowTileMapClipBuf tilemaps_clip = {"tilemaps_clip"};
ShadowPageCacheBuf pages_cached_data_ = {"pages_cached_data_"};
{
ShadowTileMapData tilemap = {};
tilemap.tiles_index = 0;
tilemap.clip_data_index = 0;
tilemap.grid_shift = int2(-1, 2);
tilemap.projection_type = SHADOW_PROJECTION_CLIPMAP;
tilemaps_data.append(tilemap);
tilemaps_data.push_update();
}
{
ShadowTileData tile = shadow_tile_unpack(ShadowTileDataPacked(SHADOW_NO_DATA));
for (auto x : IndexRange(SHADOW_TILEMAP_RES)) {
for (auto y : IndexRange(SHADOW_TILEMAP_RES)) {
tile.is_allocated = true;
tile.is_rendered = true;
tile.do_update = true;
tile.page = uint2(x, y);
tiles_data[x + y * SHADOW_TILEMAP_RES] = shadow_tile_pack(tile);
}
}
tiles_data.push_update();
}
GPUShader *sh = GPU_shader_create_from_info_name("eevee_shadow_tilemap_init");
PassSimple pass("Test");
pass.shader_set(sh);
pass.bind_ssbo("tilemaps_buf", tilemaps_data);
pass.bind_ssbo("tilemaps_clip_buf", tilemaps_clip);
pass.bind_ssbo("tiles_buf", tiles_data);
pass.bind_ssbo("pages_cached_buf", pages_cached_data_);
pass.dispatch(int3(1, 1, tilemaps_data.size()));
Manager manager;
manager.submit(pass);
GPU_memory_barrier(GPU_BARRIER_BUFFER_UPDATE);
tilemaps_data.read();
tiles_data.read();
EXPECT_EQ(tilemaps_data[0].grid_offset, int2(0));
EXPECT_EQ(shadow_tile_unpack(tiles_data[0]).page, uint2(SHADOW_TILEMAP_RES - 1, 2));
EXPECT_EQ(shadow_tile_unpack(tiles_data[0]).do_update, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[0]).is_rendered, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[0]).is_allocated, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[1]).page, uint2(0, 2));
EXPECT_EQ(shadow_tile_unpack(tiles_data[1]).do_update, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[1]).is_rendered, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[1]).is_allocated, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[0 + SHADOW_TILEMAP_RES * 2]).page,
uint2(SHADOW_TILEMAP_RES - 1, 4));
EXPECT_EQ(shadow_tile_unpack(tiles_data[0 + SHADOW_TILEMAP_RES * 2]).do_update, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[0 + SHADOW_TILEMAP_RES * 2]).is_rendered, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[0 + SHADOW_TILEMAP_RES * 2]).is_allocated, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[1 + SHADOW_TILEMAP_RES * 2]).page, uint2(0, 4));
EXPECT_EQ(shadow_tile_unpack(tiles_data[1 + SHADOW_TILEMAP_RES * 2]).do_update, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[1 + SHADOW_TILEMAP_RES * 2]).is_rendered, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[1 + SHADOW_TILEMAP_RES * 2]).is_allocated, true);
GPU_shader_free(sh);
DRW_shaders_free();
GPU_render_end();
}
DRAW_TEST(eevee_shadow_shift)
static void test_eevee_shadow_tag_update()
{
GPU_render_begin();
using namespace blender::math;
StorageVectorBuffer<uint, 128> past_casters_updated = {"PastCastersUpdated"};
StorageVectorBuffer<uint, 128> curr_casters_updated = {"CurrCastersUpdated"};
Manager manager;
{
/* Simulate 1 object moving and 1 object static with changing resource index. */
float4x4 obmat = float4x4::identity();
float4x4 obmat2 = from_loc_rot_scale<float4x4>(
float3(1.0f), Quaternion::identity(), float3(0.5f));
float3 half_extent = float3(0.24f, 0.249f, 0.001f);
{
manager.begin_sync();
ResourceHandle hdl = manager.resource_handle(obmat, float3(0.5f, 0.5f, -1.0f), half_extent);
manager.resource_handle(obmat2);
manager.end_sync();
past_casters_updated.append(hdl.resource_index());
past_casters_updated.push_update();
}
{
manager.begin_sync();
manager.resource_handle(obmat2);
ResourceHandle hdl = manager.resource_handle(obmat, float3(-1.0f, 0.5f, -1.0f), half_extent);
manager.end_sync();
curr_casters_updated.append(hdl.resource_index());
curr_casters_updated.push_update();
}
}
ShadowTileMapDataBuf tilemaps_data = {"tilemaps_data"};
ShadowTileDataBuf tiles_data = {"tiles_data"};
tiles_data.clear_to_zero();
{
ShadowTileMap tilemap(0 * SHADOW_TILEDATA_PER_TILEMAP);
tilemap.sync_cubeface(float4x4::identity(), 0.01f, 1.0f, Z_NEG, 0.0f);
tilemaps_data.append(tilemap);
}
{
ShadowTileMap tilemap(1 * SHADOW_TILEDATA_PER_TILEMAP);
tilemap.sync_orthographic(float4x4::identity(), int2(0), 1, 0.0f, SHADOW_PROJECTION_CLIPMAP);
tilemaps_data.append(tilemap);
}
tilemaps_data.push_update();
GPUShader *sh = GPU_shader_create_from_info_name("eevee_shadow_tag_update");
PassSimple pass("Test");
pass.shader_set(sh);
pass.bind_ssbo("tilemaps_buf", tilemaps_data);
pass.bind_ssbo("tiles_buf", tiles_data);
pass.bind_ssbo("bounds_buf", &manager.bounds_buf.previous());
pass.bind_ssbo("resource_ids_buf", past_casters_updated);
pass.dispatch(int3(past_casters_updated.size(), 1, tilemaps_data.size()));
pass.bind_ssbo("bounds_buf", &manager.bounds_buf.current());
pass.bind_ssbo("resource_ids_buf", curr_casters_updated);
pass.dispatch(int3(curr_casters_updated.size(), 1, tilemaps_data.size()));
manager.submit(pass);
GPU_memory_barrier(GPU_BARRIER_BUFFER_UPDATE);
tiles_data.read();
/** The layout of these expected strings is Y down. */
StringRefNull expected_lod0 =
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"xxxx----------------xxxxxxxx----"
"xxxx----------------xxxxxxxx----"
"xxxx----------------xxxxxxxx----"
"xxxx----------------xxxxxxxx----"
"xxxx----------------xxxxxxxx----"
"xxxx----------------xxxxxxxx----"
"xxxx----------------xxxxxxxx----"
"xxxx----------------xxxxxxxx----"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------";
StringRefNull expected_lod1 =
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"xx--------xxxx--"
"xx--------xxxx--"
"xx--------xxxx--"
"xx--------xxxx--"
"----------------"
"----------------";
StringRefNull expected_lod2 =
"--------"
"--------"
"--------"
"--------"
"--------"
"x----xx-"
"x----xx-"
"--------";
StringRefNull expected_lod3 =
"----"
"----"
"x-xx"
"x-xx";
StringRefNull expected_lod4 =
"--"
"xx";
StringRefNull expected_lod5 = "x";
const uint lod0_len = SHADOW_TILEMAP_LOD0_LEN;
const uint lod1_len = SHADOW_TILEMAP_LOD1_LEN;
const uint lod2_len = SHADOW_TILEMAP_LOD2_LEN;
const uint lod3_len = SHADOW_TILEMAP_LOD3_LEN;
const uint lod4_len = SHADOW_TILEMAP_LOD4_LEN;
const uint lod5_len = SHADOW_TILEMAP_LOD5_LEN;
auto stringify_result = [&](uint start, uint len) -> std::string {
std::string result = "";
for (auto i : IndexRange(start, len)) {
result += (shadow_tile_unpack(tiles_data[i]).do_update) ? "x" : "-";
}
return result;
};
EXPECT_EQ(stringify_result(0, lod0_len), expected_lod0);
EXPECT_EQ(stringify_result(lod0_len, lod1_len), expected_lod1);
EXPECT_EQ(stringify_result(lod0_len + lod1_len, lod2_len), expected_lod2);
EXPECT_EQ(stringify_result(lod0_len + lod1_len + lod2_len, lod3_len), expected_lod3);
EXPECT_EQ(stringify_result(lod0_len + lod1_len + lod2_len + lod3_len, lod4_len), expected_lod4);
EXPECT_EQ(stringify_result(lod0_len + lod1_len + lod2_len + lod3_len + lod4_len, lod5_len),
expected_lod5);
GPU_shader_free(sh);
DRW_shaders_free();
GPU_render_end();
}
DRAW_TEST(eevee_shadow_tag_update)
static void test_eevee_shadow_free()
{
GPU_render_begin();
ShadowTileMapDataBuf tilemaps_data = {"tilemaps_data"};
ShadowTileDataBuf tiles_data = {"tiles_data"};
ShadowPageHeapBuf pages_free_data = {"PagesFreeBuf"};
ShadowPageCacheBuf pages_cached_data = {"PagesCachedBuf"};
ShadowPagesInfoDataBuf pages_infos_data = {"PagesInfosBuf"};
int tiles_index = 1;
int tile_orphaned_cached = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 5;
int tile_orphaned_allocated = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 6;
int tile_used_cached = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 260;
int tile_used_allocated = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 32;
int tile_used_unallocated = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 64;
int tile_unused_cached = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 9;
int tile_unused_allocated = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 8;
int page_free_count = SHADOW_MAX_PAGE - 6;
for (uint i : IndexRange(2, page_free_count)) {
uint2 page = {i % SHADOW_PAGE_PER_ROW, i / SHADOW_PAGE_PER_ROW};
pages_free_data[i] = page.x | (page.y << 16u);
}
pages_free_data.push_update();
pages_infos_data.page_free_count = page_free_count;
pages_infos_data.page_alloc_count = 0;
pages_infos_data.page_cached_next = 2u;
pages_infos_data.page_cached_start = 0u;
pages_infos_data.page_cached_end = 2u;
pages_infos_data.push_update();
for (uint i : IndexRange(pages_cached_data.size())) {
pages_cached_data[i] = uint2(-1, -1);
}
pages_cached_data[0] = uint2(0, tile_orphaned_cached);
pages_cached_data[1] = uint2(1, tile_used_cached);
pages_cached_data.push_update();
{
ShadowTileData tile;
/* is_orphaned = true */
tile.is_used = false;
tile.do_update = true;
tile.is_cached = true;
tile.is_allocated = false;
tiles_data[tile_orphaned_cached] = shadow_tile_pack(tile);
tile.is_cached = false;
tile.is_allocated = true;
tiles_data[tile_orphaned_allocated] = shadow_tile_pack(tile);
/* is_orphaned = false */
tile.do_update = false;
tile.is_used = true;
tile.is_cached = true;
tile.is_allocated = false;
tiles_data[tile_used_cached] = shadow_tile_pack(tile);
tile.is_cached = false;
tile.is_allocated = true;
tiles_data[tile_used_allocated] = shadow_tile_pack(tile);
tile.is_cached = false;
tile.is_allocated = false;
tiles_data[tile_used_unallocated] = shadow_tile_pack(tile);
tile.is_used = false;
tile.is_cached = true;
tile.is_allocated = false;
tiles_data[tile_unused_cached] = shadow_tile_pack(tile);
tile.is_cached = false;
tile.is_allocated = true;
tiles_data[tile_unused_allocated] = shadow_tile_pack(tile);
tiles_data.push_update();
}
{
ShadowTileMapData tilemap = {};
tilemap.tiles_index = tiles_index * SHADOW_TILEDATA_PER_TILEMAP;
tilemaps_data.append(tilemap);
tilemaps_data.push_update();
}
GPUShader *sh = GPU_shader_create_from_info_name("eevee_shadow_page_free");
PassSimple pass("Test");
pass.shader_set(sh);
pass.bind_ssbo("tilemaps_buf", tilemaps_data);
pass.bind_ssbo("tiles_buf", tiles_data);
pass.bind_ssbo("pages_infos_buf", pages_infos_data);
pass.bind_ssbo("pages_free_buf", pages_free_data);
pass.bind_ssbo("pages_cached_buf", pages_cached_data);
pass.dispatch(int3(1, 1, tilemaps_data.size()));
Manager manager;
manager.submit(pass);
GPU_memory_barrier(GPU_BARRIER_BUFFER_UPDATE);
tiles_data.read();
pages_infos_data.read();
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_orphaned_cached]).is_cached, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_orphaned_cached]).is_allocated, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_orphaned_allocated]).is_cached, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_orphaned_allocated]).is_allocated, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_used_cached]).is_cached, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_used_cached]).is_allocated, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_used_allocated]).is_cached, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_used_allocated]).is_allocated, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_used_unallocated]).is_cached, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_used_unallocated]).is_allocated, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_unused_cached]).is_cached, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_unused_cached]).is_allocated, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_unused_allocated]).is_cached, true);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_unused_allocated]).is_allocated, false);
EXPECT_EQ(pages_infos_data.page_alloc_count, 1);
EXPECT_EQ(pages_infos_data.page_free_count, page_free_count + 2);
EXPECT_EQ(pages_infos_data.page_cached_next, 3);
EXPECT_EQ(pages_infos_data.page_cached_end, 2);
GPU_shader_free(sh);
DRW_shaders_free();
GPU_render_end();
}
DRAW_TEST(eevee_shadow_free)
class TestDefrag {
private:
ShadowTileDataBuf tiles_data = {"tiles_data"};
ShadowPageHeapBuf pages_free_data = {"PagesFreeBuf"};
ShadowPageCacheBuf pages_cached_data = {"PagesCachedBuf"};
ShadowPagesInfoDataBuf pages_infos_data = {"PagesInfosBuf"};
StorageBuffer<DispatchCommand> clear_dispatch_buf;
ShadowStatisticsBuf statistics_buf = {"statistics_buf"};
public:
TestDefrag(int allocation_count,
int descriptor_offset,
StringRefNull descriptor,
StringRefNull expect)
{
for (uint i : IndexRange(SHADOW_MAX_PAGE)) {
uint2 page = {i % SHADOW_PAGE_PER_ROW, i / SHADOW_PAGE_PER_ROW};
pages_free_data[i] = page.x | (page.y << 16u);
}
for (uint i : IndexRange(tiles_data.size())) {
tiles_data[i] = 0;
}
int free_count = SHADOW_MAX_PAGE;
int tile_index = 0;
for (uint i : IndexRange(pages_cached_data.size())) {
pages_cached_data[i] = uint2(-1, -1);
}
int cached_index = descriptor_offset;
int hole_count = 0;
int inserted_count = 0;
ShadowTileData tile = {};
tile.is_cached = true;
for (char c : descriptor) {
switch (c) {
case 'c':
tile.cache_index = cached_index++ % SHADOW_MAX_PAGE;
pages_cached_data[tile.cache_index] = uint2(pages_free_data[--free_count], tile_index);
tiles_data[tile_index++] = shadow_tile_pack(tile);
break;
case 'f':
pages_cached_data[cached_index++ % SHADOW_MAX_PAGE] = uint2(-1, -1);
hole_count++;
break;
case 'i':
tile.cache_index = (cached_index + inserted_count++) % SHADOW_MAX_PAGE;
pages_cached_data[tile.cache_index] = uint2(pages_free_data[--free_count], tile_index);
tiles_data[tile_index++] = shadow_tile_pack(tile);
break;
default:
break;
}
}
pages_infos_data.page_alloc_count = allocation_count;
pages_infos_data.page_cached_next = cached_index + inserted_count;
pages_infos_data.page_free_count = free_count;
pages_infos_data.page_cached_start = descriptor_offset;
pages_infos_data.page_cached_end = cached_index;
tiles_data.push_update();
pages_infos_data.push_update();
pages_free_data.push_update();
pages_cached_data.push_update();
GPUShader *sh = GPU_shader_create_from_info_name("eevee_shadow_page_defrag");
PassSimple pass("Test");
pass.shader_set(sh);
pass.bind_ssbo("tiles_buf", tiles_data);
pass.bind_ssbo("pages_infos_buf", pages_infos_data);
pass.bind_ssbo("pages_free_buf", pages_free_data);
pass.bind_ssbo("pages_cached_buf", pages_cached_data);
pass.bind_ssbo("statistics_buf", statistics_buf);
pass.bind_ssbo("clear_dispatch_buf", clear_dispatch_buf);
pass.dispatch(int3(1, 1, 1));
Manager manager;
manager.submit(pass);
GPU_memory_barrier(GPU_BARRIER_BUFFER_UPDATE);
tiles_data.read();
pages_cached_data.read();
pages_infos_data.read();
std::string result = "";
int expect_cached_len = 0;
for (auto i : IndexRange(descriptor_offset, descriptor.size())) {
if (pages_cached_data[i % SHADOW_MAX_PAGE].y != -1) {
result += 'c';
expect_cached_len++;
}
else {
result += 'f';
}
}
EXPECT_EQ(expect, result);
allocation_count = min_ii(allocation_count, SHADOW_MAX_PAGE);
int additional_pages = max_ii(0, allocation_count - free_count);
int expected_free_count = max_ii(free_count, allocation_count);
int expected_start = descriptor_offset + hole_count + additional_pages;
int result_cached_len = pages_infos_data.page_cached_end - pages_infos_data.page_cached_start;
if (expected_start > SHADOW_MAX_PAGE) {
expected_start -= SHADOW_MAX_PAGE;
}
EXPECT_EQ(expected_free_count, pages_infos_data.page_free_count);
EXPECT_EQ(expected_start, pages_infos_data.page_cached_start);
EXPECT_EQ(expect_cached_len, result_cached_len);
EXPECT_EQ(pages_infos_data.page_cached_end, pages_infos_data.page_cached_next);
GPU_shader_free(sh);
DRW_shaders_free();
}
};
static void test_eevee_shadow_defrag()
{
TestDefrag(0, 0, "cfi", "fcc");
TestDefrag(0, 0, "fci", "fcc");
TestDefrag(0, 47, "ccfcffccfcfciiiii", "fffffcccccccccccc");
TestDefrag(10, SHADOW_MAX_PAGE - 5, "ccfcffccfcfciiiii", "fffffcccccccccccc");
TestDefrag(SHADOW_MAX_PAGE - 8, 30, "ccfcffccfcfciiiii", "fffffffffcccccccc");
TestDefrag(SHADOW_MAX_PAGE - 4, 30, "ccfcffccfcfciiiii", "fffffffffffffcccc");
/* Over allocation but should not crash. */
TestDefrag(SHADOW_MAX_PAGE + 4, 30, "ccfcffccfcfciiiii", "fffffffffffffffff");
}
DRAW_TEST(eevee_shadow_defrag)
class TestAlloc {
private:
ShadowTileMapDataBuf tilemaps_data = {"tilemaps_data"};
ShadowTileDataBuf tiles_data = {"tiles_data"};
ShadowPageHeapBuf pages_free_data = {"PagesFreeBuf"};
ShadowPageCacheBuf pages_cached_data = {"PagesCachedBuf"};
ShadowPagesInfoDataBuf pages_infos_data = {"PagesInfosBuf"};
ShadowStatisticsBuf statistics_buf = {"statistics_buf"};
public:
TestAlloc(int page_free_count)
{
GPU_render_begin();
int tiles_index = 1;
for (uint i : IndexRange(0, page_free_count)) {
uint2 page = {i % SHADOW_PAGE_PER_ROW, i / SHADOW_PAGE_PER_ROW};
pages_free_data[i] = page.x | (page.y << 16u);
}
pages_free_data.push_update();
pages_cached_data.push_update();
pages_infos_data.page_free_count = page_free_count;
pages_infos_data.page_alloc_count = 1;
pages_infos_data.page_cached_next = 0u;
pages_infos_data.page_cached_start = 0u;
pages_infos_data.page_cached_end = 0u;
pages_infos_data.view_count = 0u;
pages_infos_data.page_size = 256u;
pages_infos_data.push_update();
int tile_allocated = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 5;
int tile_free = tiles_index * SHADOW_TILEDATA_PER_TILEMAP + 6;
{
ShadowTileData tile;
tile.is_used = true;
tile.do_update = false;
tile.is_cached = false;
tile.is_allocated = false;
tiles_data[tile_free] = shadow_tile_pack(tile);
tile.is_cached = false;
tile.is_allocated = true;
tiles_data[tile_allocated] = shadow_tile_pack(tile);
tiles_data.push_update();
}
{
ShadowTileMapData tilemap = {};
tilemap.tiles_index = tiles_index * SHADOW_TILEDATA_PER_TILEMAP;
tilemaps_data.append(tilemap);
tilemaps_data.push_update();
}
GPUShader *sh = GPU_shader_create_from_info_name("eevee_shadow_page_allocate");
PassSimple pass("Test");
pass.shader_set(sh);
pass.bind_ssbo("tilemaps_buf", tilemaps_data);
pass.bind_ssbo("tiles_buf", tiles_data);
pass.bind_ssbo("pages_infos_buf", pages_infos_data);
pass.bind_ssbo("pages_free_buf", pages_free_data);
pass.bind_ssbo("pages_cached_buf", pages_cached_data);
pass.bind_ssbo("statistics_buf", statistics_buf);
pass.dispatch(int3(1, 1, tilemaps_data.size()));
Manager manager;
manager.submit(pass);
GPU_memory_barrier(GPU_BARRIER_BUFFER_UPDATE);
tiles_data.read();
pages_infos_data.read();
bool alloc_success = page_free_count >= 1;
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_free]).do_update, alloc_success);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_free]).is_allocated, alloc_success);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_allocated]).do_update, false);
EXPECT_EQ(shadow_tile_unpack(tiles_data[tile_allocated]).is_allocated, true);
EXPECT_EQ(pages_infos_data.page_free_count, page_free_count - 1);
GPU_shader_free(sh);
DRW_shaders_free();
GPU_render_end();
}
};
static void test_eevee_shadow_alloc()
{
TestAlloc(SHADOW_MAX_PAGE);
TestAlloc(1);
TestAlloc(0);
}
DRAW_TEST(eevee_shadow_alloc)
static void test_eevee_shadow_finalize()
{
GPU_render_begin();
ShadowTileMapDataBuf tilemaps_data = {"tilemaps_data"};
ShadowTileDataBuf tiles_data = {"tiles_data"};
ShadowPageHeapBuf pages_free_data = {"PagesFreeBuf"};
ShadowPageCacheBuf pages_cached_data = {"PagesCachedBuf"};
ShadowPagesInfoDataBuf pages_infos_data = {"PagesInfosBuf"};
ShadowStatisticsBuf statistics_buf = {"statistics_buf"};
ShadowTileMapClipBuf tilemaps_clip = {"tilemaps_clip"};
const uint lod0_len = SHADOW_TILEMAP_LOD0_LEN;
const uint lod1_len = SHADOW_TILEMAP_LOD1_LEN;
const uint lod2_len = SHADOW_TILEMAP_LOD2_LEN;
const uint lod3_len = SHADOW_TILEMAP_LOD3_LEN;
const uint lod4_len = SHADOW_TILEMAP_LOD4_LEN;
const uint lod0_ofs = 0;
const uint lod1_ofs = lod0_len;
const uint lod2_ofs = lod1_ofs + lod1_len;
const uint lod3_ofs = lod2_ofs + lod2_len;
const uint lod4_ofs = lod3_ofs + lod3_len;
const uint lod5_ofs = lod4_ofs + lod4_len;
for (auto i : IndexRange(SHADOW_TILEDATA_PER_TILEMAP)) {
tiles_data[i] = 0;
}
{
ShadowTileData tile;
tile.is_used = true;
tile.is_allocated = true;
tile.page = uint2(1, 0);
tile.do_update = false;
tiles_data[lod0_ofs] = shadow_tile_pack(tile);
tile.page = uint2(2, 0);
tile.do_update = false;
tiles_data[lod1_ofs] = shadow_tile_pack(tile);
tile.page = uint2(3, 0);
tile.do_update = true;
tiles_data[lod2_ofs] = shadow_tile_pack(tile);
tile.page = uint2(4, 0);
tile.do_update = false;
tiles_data[lod3_ofs] = shadow_tile_pack(tile);
tile.page = uint2(5, 0);
tile.do_update = true;
tiles_data[lod4_ofs] = shadow_tile_pack(tile);
tile.page = uint2(6, 0);
tile.do_update = true;
tiles_data[lod5_ofs] = shadow_tile_pack(tile);
tile.page = uint2(7, 0);
tile.do_update = true;
tiles_data[lod0_ofs + 8] = shadow_tile_pack(tile);
tiles_data.push_update();
}
{
ShadowTileMapData tilemap = {};
tilemap.tiles_index = 0;
tilemap.projection_type = SHADOW_PROJECTION_CUBEFACE;
tilemaps_data.append(tilemap);
tilemaps_data.push_update();
}
{
pages_infos_data.page_free_count = -5;
pages_infos_data.page_alloc_count = 0;
pages_infos_data.page_cached_next = 0u;
pages_infos_data.page_cached_start = 0u;
pages_infos_data.page_cached_end = 0u;
pages_infos_data.view_count = 0u;
pages_infos_data.page_size = 256u;
pages_infos_data.push_update();
}
Texture tilemap_tx = {"tilemap_tx"};
tilemap_tx.ensure_2d(GPU_R32UI,
int2(SHADOW_TILEMAP_RES),
GPU_TEXTURE_USAGE_HOST_READ | GPU_TEXTURE_USAGE_SHADER_READ |
GPU_TEXTURE_USAGE_SHADER_WRITE);
tilemap_tx.clear(uint4(0));
Texture render_map_tx = {"ShadowRenderMap",
GPU_R32UI,
GPU_TEXTURE_USAGE_HOST_READ | GPU_TEXTURE_USAGE_SHADER_READ |
GPU_TEXTURE_USAGE_SHADER_WRITE | GPU_TEXTURE_USAGE_MIP_SWIZZLE_VIEW,
int2(SHADOW_TILEMAP_RES),
1, /* Only one layer for the test. */
nullptr,
SHADOW_TILEMAP_LOD + 1};
render_map_tx.ensure_mip_views();
View shadow_multi_view = {"ShadowMultiView", 64, true};
StorageBuffer<DispatchCommand> clear_dispatch_buf;
StorageArrayBuffer<uint, SHADOW_MAX_PAGE> clear_page_buf = {"clear_page_buf"};
GPUShader *sh = GPU_shader_create_from_info_name("eevee_shadow_tilemap_finalize");
PassSimple pass("Test");
pass.shader_set(sh);
pass.bind_ssbo("tilemaps_buf", tilemaps_data);
pass.bind_ssbo("tiles_buf", tiles_data);
pass.bind_ssbo("pages_infos_buf", pages_infos_data);
pass.bind_image("tilemaps_img", tilemap_tx);
pass.bind_ssbo("view_infos_buf", shadow_multi_view.matrices_ubo_get());
pass.bind_ssbo("clear_dispatch_buf", clear_dispatch_buf);
pass.bind_ssbo("clear_page_buf", clear_page_buf);
pass.bind_ssbo("statistics_buf", statistics_buf);
pass.bind_ssbo("tilemaps_clip_buf", tilemaps_clip);
pass.bind_image("render_map_lod0_img", render_map_tx.mip_view(0));
pass.bind_image("render_map_lod1_img", render_map_tx.mip_view(1));
pass.bind_image("render_map_lod2_img", render_map_tx.mip_view(2));
pass.bind_image("render_map_lod3_img", render_map_tx.mip_view(3));
pass.bind_image("render_map_lod4_img", render_map_tx.mip_view(4));
pass.bind_image("render_map_lod5_img", render_map_tx.mip_view(5));
pass.dispatch(int3(1, 1, tilemaps_data.size()));
Manager manager;
manager.submit(pass);
GPU_memory_barrier(GPU_BARRIER_BUFFER_UPDATE | GPU_BARRIER_TEXTURE_UPDATE);
{
uint *pixels = tilemap_tx.read<uint32_t>(GPU_DATA_UINT);
std::string result = "";
for (auto y : IndexRange(SHADOW_TILEMAP_RES)) {
for (auto x : IndexRange(SHADOW_TILEMAP_RES)) {
result += std::to_string(shadow_tile_unpack(pixels[y * SHADOW_TILEMAP_RES + x]).page.x);
}
}
MEM_SAFE_FREE(pixels);
/** The layout of these expected strings is Y down. */
StringRefNull expected_pages =
"12334444755555556666666666666666"
"22334444555555556666666666666666"
"33334444555555556666666666666666"
"33334444555555556666666666666666"
"44444444555555556666666666666666"
"44444444555555556666666666666666"
"44444444555555556666666666666666"
"44444444555555556666666666666666"
"55555555555555556666666666666666"
"55555555555555556666666666666666"
"55555555555555556666666666666666"
"55555555555555556666666666666666"
"55555555555555556666666666666666"
"55555555555555556666666666666666"
"55555555555555556666666666666666"
"55555555555555556666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666"
"66666666666666666666666666666666";
EXPECT_EQ(expected_pages, result);
}
{
auto stringify_lod = [](Span<uint> data) -> std::string {
std::string result = "";
for (auto x : data) {
result += (x == 0xFFFFFFFFu) ? '-' : '0' + (x % 10);
}
return result;
};
/** The layout of these expected strings is Y down. */
StringRefNull expected_lod0 =
"--------7-----------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------";
StringRefNull expected_lod1 =
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------"
"----------------";
StringRefNull expected_lod2 =
"3-------"
"--------"
"--------"
"--------"
"--------"
"--------"
"--------"
"--------";
StringRefNull expected_lod3 =
"----"
"----"
"----"
"----";
StringRefNull expected_lod4 =
"5-"
"--";
StringRefNull expected_lod5 = "6";
uint *pixels_lod0 = render_map_tx.read<uint32_t>(GPU_DATA_UINT, 0);
uint *pixels_lod1 = render_map_tx.read<uint32_t>(GPU_DATA_UINT, 1);
uint *pixels_lod2 = render_map_tx.read<uint32_t>(GPU_DATA_UINT, 2);
uint *pixels_lod3 = render_map_tx.read<uint32_t>(GPU_DATA_UINT, 3);
uint *pixels_lod4 = render_map_tx.read<uint32_t>(GPU_DATA_UINT, 4);
uint *pixels_lod5 = render_map_tx.read<uint32_t>(GPU_DATA_UINT, 5);
EXPECT_EQ(stringify_lod(Span<uint>(pixels_lod0, lod0_len)), expected_lod0);
EXPECT_EQ(stringify_lod(Span<uint>(pixels_lod1, lod1_len)), expected_lod1);
EXPECT_EQ(stringify_lod(Span<uint>(pixels_lod2, lod2_len)), expected_lod2);
EXPECT_EQ(stringify_lod(Span<uint>(pixels_lod3, lod3_len)), expected_lod3);
EXPECT_EQ(stringify_lod(Span<uint>(pixels_lod4, lod4_len)), expected_lod4);
EXPECT_EQ(stringify_lod(Span<uint>(pixels_lod5, 1)), expected_lod5);
MEM_SAFE_FREE(pixels_lod0);
MEM_SAFE_FREE(pixels_lod1);
MEM_SAFE_FREE(pixels_lod2);
MEM_SAFE_FREE(pixels_lod3);
MEM_SAFE_FREE(pixels_lod4);
MEM_SAFE_FREE(pixels_lod5);
}
pages_infos_data.read();
EXPECT_EQ(pages_infos_data.page_free_count, 0);
EXPECT_EQ(pages_infos_data.view_count, 1);
GPU_shader_free(sh);
DRW_shaders_free();
GPU_render_end();
}
DRAW_TEST(eevee_shadow_finalize)
static void test_eevee_shadow_page_mask()
{
GPU_render_begin();
ShadowTileMapDataBuf tilemaps_data = {"tilemaps_data"};
ShadowTileDataBuf tiles_data = {"tiles_data"};
{
ShadowTileMap tilemap(0);
tilemap.sync_cubeface(float4x4::identity(), 0.01f, 1.0f, Z_NEG, 0.0f);
tilemaps_data.append(tilemap);
}
const uint lod0_len = SHADOW_TILEMAP_LOD0_LEN;
const uint lod1_len = SHADOW_TILEMAP_LOD1_LEN;
const uint lod2_len = SHADOW_TILEMAP_LOD2_LEN;
const uint lod3_len = SHADOW_TILEMAP_LOD3_LEN;
const uint lod4_len = SHADOW_TILEMAP_LOD4_LEN;
const uint lod5_len = SHADOW_TILEMAP_LOD5_LEN;
const uint lod0_ofs = 0;
const uint lod1_ofs = lod0_ofs + lod0_len;
const uint lod2_ofs = lod1_ofs + lod1_len;
const uint lod3_ofs = lod2_ofs + lod2_len;
const uint lod4_ofs = lod3_ofs + lod3_len;
const uint lod5_ofs = lod4_ofs + lod4_len;
{
ShadowTileData tile;
/* Init all LOD to true. */
for (auto i : IndexRange(SHADOW_TILEDATA_PER_TILEMAP)) {
tile.is_used = true;
tiles_data[i] = shadow_tile_pack(tile);
}
/* Init all of LOD0 to false. */
for (auto i : IndexRange(square_i(SHADOW_TILEMAP_RES))) {
tile.is_used = false;
tiles_data[i] = shadow_tile_pack(tile);
}
/* Bottom Left of the LOD0 to true. */
for (auto y : IndexRange((SHADOW_TILEMAP_RES / 2) + 1)) {
for (auto x : IndexRange((SHADOW_TILEMAP_RES / 2) + 1)) {
tile.is_used = true;
tiles_data[x + y * SHADOW_TILEMAP_RES] = shadow_tile_pack(tile);
}
}
/* All Bottom of the LOD0 to true. */
for (auto x : IndexRange(SHADOW_TILEMAP_RES)) {
tile.is_used = true;
tiles_data[x] = shadow_tile_pack(tile);
}
/* Bottom Left of the LOD1 to false. */
/* Should still cover bottom LODs since it is itself fully masked */
for (auto y : IndexRange((SHADOW_TILEMAP_RES / 8))) {
for (auto x : IndexRange((SHADOW_TILEMAP_RES / 8))) {
tile.is_used = false;
tiles_data[x + y * (SHADOW_TILEMAP_RES / 2) + lod0_len] = shadow_tile_pack(tile);
}
}
/* Top right Center of the LOD1 to false. */
/* Should un-cover 1 LOD2 tile. */
{
int x = SHADOW_TILEMAP_RES / 4;
int y = SHADOW_TILEMAP_RES / 4;
tile.is_used = false;
tiles_data[x + y * (SHADOW_TILEMAP_RES / 2) + lod0_len] = shadow_tile_pack(tile);
}
tiles_data.push_update();
}
tilemaps_data.push_update();
GPUShader *sh = GPU_shader_create_from_info_name("eevee_shadow_page_mask");
PassSimple pass("Test");
pass.shader_set(sh);
pass.bind_ssbo("tilemaps_buf", tilemaps_data);
pass.bind_ssbo("tiles_buf", tiles_data);
pass.dispatch(int3(1, 1, tilemaps_data.size()));
Manager manager;
manager.submit(pass);
GPU_memory_barrier(GPU_BARRIER_BUFFER_UPDATE);
tiles_data.read();
/** The layout of these expected strings is Y down. */
StringRefNull expected_lod0 =
"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"xxxxxxxxxxxxxxxxx---------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------"
"--------------------------------";
StringRefNull expected_lod1 =
"--------xxxxxxxx"
"--------xxxxxxxx"
"--------xxxxxxxx"
"--------xxxxxxxx"
"--------xxxxxxxx"
"--------xxxxxxxx"
"--------xxxxxxxx"
"--------xxxxxxxx"
"xxxxxxxx-xxxxxxx"
"xxxxxxxxxxxxxxxx"
"xxxxxxxxxxxxxxxx"
"xxxxxxxxxxxxxxxx"
"xxxxxxxxxxxxxxxx"
"xxxxxxxxxxxxxxxx"
"xxxxxxxxxxxxxxxx"
"xxxxxxxxxxxxxxxx";
StringRefNull expected_lod2 =
"--------"
"--------"
"--------"
"--------"
"----x---"
"--------"
"--------"
"--------";
StringRefNull expected_lod3 =
"----"
"----"
"----"
"----";
StringRefNull expected_lod4 =
"--"
"--";
StringRefNull expected_lod5 = "-";
auto stringify_result = [&](uint start, uint len) -> std::string {
std::string result = "";
for (auto i : IndexRange(start, len)) {
result += (shadow_tile_unpack(tiles_data[i]).is_used) ? "x" : "-";
}
return result;
};
EXPECT_EQ(stringify_result(lod0_ofs, lod0_len), expected_lod0);
EXPECT_EQ(stringify_result(lod1_ofs, lod1_len), expected_lod1);
EXPECT_EQ(stringify_result(lod2_ofs, lod2_len), expected_lod2);
EXPECT_EQ(stringify_result(lod3_ofs, lod3_len), expected_lod3);
EXPECT_EQ(stringify_result(lod4_ofs, lod4_len), expected_lod4);
EXPECT_EQ(stringify_result(lod5_ofs, lod5_len), expected_lod5);
GPU_shader_free(sh);
DRW_shaders_free();
GPU_render_end();
}
DRAW_TEST(eevee_shadow_page_mask)
static void test_eevee_surfel_list()
{
GPU_render_begin();
StorageArrayBuffer<int> list_start_buf = {"list_start_buf"};
StorageVectorBuffer<Surfel> surfel_buf = {"surfel_buf"};
CaptureInfoBuf capture_info_buf = {"capture_info_buf"};
SurfelListInfoBuf list_info_buf = {"list_info_buf"};
/**
* Simulate surfels on a 2x2 projection grid covering [0..2] on the Z axis.
*/
{
Surfel surfel;
/* NOTE: Expected link assumes linear increasing processing order [0->5]. But this is
* multithreaded and we can't know the execution order in advance. */
/* 0: Project to (1, 0) = list 1. Unsorted Next = -1; Next = -1; Previous = 3. */
surfel.position = {1.1f, 0.1f, 0.1f};
surfel_buf.append(surfel);
/* 1: Project to (1, 0) = list 1. Unsorted Next = 0; Next = 2; Previous = -1. */
surfel.position = {1.1f, 0.2f, 0.5f};
surfel_buf.append(surfel);
/* 2: Project to (1, 0) = list 1. Unsorted Next = 1; Next = 3; Previous = 1. */
surfel.position = {1.1f, 0.3f, 0.3f};
surfel_buf.append(surfel);
/* 3: Project to (1, 0) = list 1. Unsorted Next = 2; Next = 0; Previous = 2. */
surfel.position = {1.2f, 0.4f, 0.2f};
surfel_buf.append(surfel);
/* 4: Project to (1, 1) = list 3. Unsorted Next = -1; Next = -1; Previous = -1. */
surfel.position = {1.0f, 1.0f, 0.5f};
surfel_buf.append(surfel);
/* 5: Project to (0, 1) = list 2. Unsorted Next = -1; Next = -1; Previous = -1. */
surfel.position = {0.1f, 1.1f, 0.5f};
surfel_buf.append(surfel);
surfel_buf.push_update();
}
{
capture_info_buf.surfel_len = surfel_buf.size();
capture_info_buf.push_update();
}
{
list_info_buf.ray_grid_size = int2(2);
list_info_buf.list_max = list_info_buf.ray_grid_size.x * list_info_buf.ray_grid_size.y;
list_info_buf.push_update();
}
{
list_start_buf.resize(ceil_to_multiple_u(list_info_buf.list_max, 4u));
list_start_buf.push_update();
GPU_storagebuf_clear(list_start_buf, -1);
}
/* Top-down view. */
View view = {"RayProjectionView"};
view.sync(float4x4::identity(), math::projection::orthographic<float>(0, 2, 0, 2, 0, 1));
GPUShader *sh_build = GPU_shader_create_from_info_name("eevee_surfel_list_build");
GPUShader *sh_sort = GPU_shader_create_from_info_name("eevee_surfel_list_sort");
PassSimple pass("Build_and_Sort");
pass.shader_set(sh_build);
pass.bind_ssbo("list_start_buf", list_start_buf);
pass.bind_ssbo("surfel_buf", surfel_buf);
pass.bind_ssbo("capture_info_buf", capture_info_buf);
pass.bind_ssbo("list_info_buf", list_info_buf);
pass.dispatch(int3(1, 1, 1));
pass.barrier(GPU_BARRIER_SHADER_STORAGE);
pass.shader_set(sh_sort);
pass.bind_ssbo("list_start_buf", list_start_buf);
pass.bind_ssbo("surfel_buf", surfel_buf);
pass.bind_ssbo("list_info_buf", list_info_buf);
pass.dispatch(int3(1, 1, 1));
pass.barrier(GPU_BARRIER_BUFFER_UPDATE);
Manager manager;
manager.submit(pass, view);
list_start_buf.read();
surfel_buf.read();
/* NOTE: All of these are unstable by definition (atomic + multi-thread).
* But should be consistent since we only dispatch one thread-group. */
/* Expect last added surfel index. It is the list start index before sorting. */
Vector<int> expect_list_start = {-1, 3, 5, 4};
/* Expect surfel list. */
Vector<int> expect_link_next = {-1, +2, +3, +0, -1, -1};
Vector<int> expect_link_prev = {+3, -1, +1, +2, -1, -1};
Vector<int> link_next, link_prev;
for (auto &surfel : Span<Surfel>(surfel_buf.data(), surfel_buf.size())) {
link_next.append(surfel.next);
link_prev.append(surfel.prev);
}
#if 0 /* Useful for debugging */
// Span<int>(list_start_buf.data(), expect_list_start.size()).print_as_lines("list_start");
// link_next.as_span().print_as_lines("link_next");
// link_prev.as_span().print_as_lines("link_prev");
#endif
EXPECT_EQ_ARRAY(list_start_buf.data(), expect_list_start.data(), expect_list_start.size());
EXPECT_EQ_ARRAY(link_next.data(), expect_link_next.data(), expect_link_next.size());
EXPECT_EQ_ARRAY(link_prev.data(), expect_link_prev.data(), expect_link_prev.size());
GPU_shader_free(sh_build);
GPU_shader_free(sh_sort);
DRW_shaders_free();
GPU_render_end();
}
DRAW_TEST(eevee_surfel_list)
} // namespace blender::draw
Reference in New Issue View Git Blame Copy Permalink