#include "ray_resources.h"
#include "ray_pass.h"

#define LIST_SCENE_BINDINGS(X) \
	X(1, tlas, VK_DESCRIPTOR_TYPE_ACCELERATION_STRUCTURE_KHR, 1) \
	X(2, ubo, VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, 1) \
	X(3, kusochki, VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 1) \
	X(4, indices, VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 1) \
	X(5, vertices, VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 1) \
	X(6, all_textures, VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, MAX_TEXTURES) \
	X(7, lights, VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 1) \
	X(8, light_clusters, VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 1) \

#define LIST_COMMON_BINDINGS(X) \
	LIST_SCENE_BINDINGS(X) \
	RAY_LIGHT_DIRECT_INPUTS(X)

// FIXME more conservative shader stages
#define INIT_BINDING(index, name, type, count) \
	{ \
		.binding = index, \
		.descriptorType = type, \
		.descriptorCount = count, \
		.stageFlags = VK_SHADER_STAGE_RAYGEN_BIT_KHR | VK_SHADER_STAGE_CLOSEST_HIT_BIT_KHR | VK_SHADER_STAGE_ANY_HIT_BIT_KHR, \
	},

#define INIT_IMAGE(index, name, ...) INIT_BINDING(index, name, VK_DESCRIPTOR_TYPE_STORAGE_IMAGE, 1)

static const VkDescriptorSetLayoutBinding bindings[] = {
	LIST_SCENE_BINDINGS(INIT_BINDING)
	RAY_LIGHT_DIRECT_INPUTS(INIT_IMAGE)

	// FIXME it's an artifact that point and poly outputs have same bindings indices
	RAY_LIGHT_DIRECT_POLY_OUTPUTS(INIT_IMAGE)
};

#undef INIT_IMAGE
#undef INIT_BINDING

static const int semantics_poly[] = {
#define IN(index, name, ...) (RayResource_##name + 1),
#define OUT(index, name, ...) -(RayResource_##name + 1),
	LIST_COMMON_BINDINGS(IN)
	RAY_LIGHT_DIRECT_POLY_OUTPUTS(OUT)
#undef IN
#undef OUT
};

static const int semantics_point[] = {
#define IN(index, name, ...) (RayResource_##name + 1),
#define OUT(index, name, ...) -(RayResource_##name + 1),
	LIST_COMMON_BINDINGS(IN)
	RAY_LIGHT_DIRECT_POINT_OUTPUTS(OUT)
#undef IN
#undef OUT
};

const ray_pass_layout_t light_direct_poly_layout_fixme = {
	.bindings = bindings,
	.bindings_semantics = semantics_poly,
	.bindings_count = COUNTOF(bindings),
	.push_constants = {0},
};

const ray_pass_layout_t light_direct_point_layout_fixme = {
	.bindings = bindings,
	.bindings_semantics = semantics_point,
	.bindings_count = COUNTOF(bindings),
	.push_constants = {0},
};