#include "shared/shared.h"
#include "vkpt.h"
#include "vk_util.h"

Macros
#define	BARRIER_COMPUTE(cmd_buf, img)

Functions
static void	create_image_views ()

static void	create_pipeline_layout ()

static void	create_pipelines ()

static void	create_descriptor_set ()

static void	update_descriptor_set ()

VkResult	vkpt_initialize_god_rays ()

VkResult	vkpt_destroy_god_rays ()

VkResult	vkpt_god_rays_create_pipelines ()

VkResult	vkpt_god_rays_destroy_pipelines ()

VkResult	vkpt_god_rays_update_images ()

VkResult	vkpt_god_rays_noop ()

void	vkpt_record_god_rays_trace_command_buffer (VkCommandBuffer command_buffer, int pass)

void	vkpt_record_god_rays_filter_command_buffer (VkCommandBuffer command_buffer)

void	vkpt_god_rays_prepare_ubo (QVKUniformBuffer_t ubo, const aabb_t world_aabb, const float proj, const float view, const float *shadowmap_viewproj, float shadowmap_depth_scale)

qboolean	vkpt_god_rays_enabled (const sun_light_t *sun_light)

Variables
static const uint32_t	THREAD_GROUP_SIZE = 16

static const uint32_t	FILTER_THREAD_GROUP_SIZE = 16

struct {
VkPipeline pipelines [2]

VkPipelineLayout pipeline_layout

VkDescriptorSetLayout descriptor_set_layout

VkDescriptorPool descriptor_pool

VkDescriptorSet descriptor_set

VkImageView shadow_image_view

VkSampler shadow_sampler

cvar_t * intensity

cvar_t * eccentricity

cvar_t * enable

}	god_rays

cvar_t *	physical_sky_space

Macro Definition Documentation

◆ BARRIER_COMPUTE

#define BARRIER_COMPUTE	(	cmd_buf,
		img
	)

Value:

    do { \
        VkImageSubresourceRange subresource_range = { \
            .aspectMask     = VK_IMAGE_ASPECT_COLOR_BIT, \
            .baseMipLevel   = 0, \
            .levelCount     = 1, \
            .baseArrayLayer = 0, \
            .layerCount     = 1 \
        }; \
        IMAGE_BARRIER(cmd_buf, \
                .image            = img, \
                .subresourceRange = subresource_range, \
                .srcAccessMask    = VK_ACCESS_SHADER_WRITE_BIT, \
                .dstAccessMask    = VK_ACCESS_SHADER_WRITE_BIT, \
                .oldLayout        = VK_IMAGE_LAYOUT_GENERAL, \
                .newLayout        = VK_IMAGE_LAYOUT_GENERAL, \
        ); \
    } while(0)

Definition at line 121 of file god_rays.c.

Function Documentation

◆ create_descriptor_set()

static void create_descriptor_set ( )

static

Definition at line 327 of file god_rays.c.

 {
     const VkDescriptorPoolSize pool_sizes[] = {
         { VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, 1 }
     };
  
     const VkDescriptorPoolCreateInfo pool_create_info = {
         .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO,
         .maxSets = 1,
         .poolSizeCount = LENGTH(pool_sizes),
         .pPoolSizes = pool_sizes
     };
  
     _VK(vkCreateDescriptorPool(qvk.device, &pool_create_info, NULL, &god_rays.descriptor_pool));
  
     const VkDescriptorSetAllocateInfo allocate_info = {
         .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_ALLOCATE_INFO,
         .descriptorPool = god_rays.descriptor_pool,
         .descriptorSetCount = 1,
         .pSetLayouts = &god_rays.descriptor_set_layout
     };
  
     _VK(vkAllocateDescriptorSets(qvk.device, &allocate_info, &god_rays.descriptor_set));
 }

Referenced by vkpt_god_rays_create_pipelines().

◆ create_image_views()

static void create_image_views ( )

static

Definition at line 227 of file god_rays.c.

 {
     god_rays.shadow_image_view = vkpt_shadow_map_get_view();
  
     VkSamplerReductionModeCreateInfoEXT redutcion_create_info = {
         .sType = VK_STRUCTURE_TYPE_SAMPLER_REDUCTION_MODE_CREATE_INFO_EXT,
         .reductionMode = VK_SAMPLER_REDUCTION_MODE_MIN_EXT
     };
  
     const VkSamplerCreateInfo sampler_create_info = {
         .sType = VK_STRUCTURE_TYPE_SAMPLER_CREATE_INFO,
         .pNext = &redutcion_create_info,
         .magFilter = VK_FILTER_LINEAR,
         .minFilter = VK_FILTER_LINEAR,
         .mipmapMode = VK_SAMPLER_MIPMAP_MODE_NEAREST,
         .addressModeU = VK_SAMPLER_ADDRESS_MODE_CLAMP_TO_BORDER,
         .addressModeV = VK_SAMPLER_ADDRESS_MODE_CLAMP_TO_BORDER,
         .addressModeW = VK_SAMPLER_ADDRESS_MODE_CLAMP_TO_BORDER,
         .borderColor = VK_BORDER_COLOR_FLOAT_OPAQUE_WHITE
     };
  
     _VK(vkCreateSampler(qvk.device, &sampler_create_info, NULL, &god_rays.shadow_sampler));
 }

Referenced by vkpt_god_rays_update_images().

◆ create_pipeline_layout()

static void create_pipeline_layout ( )

static

Definition at line 251 of file god_rays.c.

 {
     VkDescriptorSetLayoutBinding bindings[1] = { 0 };
     bindings[0].binding = 0;
     bindings[0].descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER;
     bindings[0].descriptorCount = 1;
     bindings[0].stageFlags = VK_SHADER_STAGE_COMPUTE_BIT;
     
     const VkDescriptorSetLayoutCreateInfo set_layout_create_info = {
         .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO,
         .bindingCount = LENGTH(bindings),
         .pBindings = bindings
     };
  
     _VK(vkCreateDescriptorSetLayout(qvk.device, &set_layout_create_info, NULL,
         &god_rays.descriptor_set_layout));
  
  
     VkDescriptorSetLayout desc_set_layouts[] = {
         god_rays.descriptor_set_layout,
         qvk.desc_set_layout_vertex_buffer,
         qvk.desc_set_layout_ubo,
         qvk.desc_set_layout_textures
     };
  
     VkPushConstantRange push_constant_range = {
         .stageFlags = VK_SHADER_STAGE_COMPUTE_BIT,
         .offset = 0,
         .size = sizeof(int),
     };
  
     const VkPipelineLayoutCreateInfo layout_create_info = {
         .sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO,
         .pSetLayouts = desc_set_layouts,
         .setLayoutCount = LENGTH(desc_set_layouts),
         .pushConstantRangeCount = 1,
         .pPushConstantRanges = &push_constant_range
     };
  
     _VK(vkCreatePipelineLayout(qvk.device, &layout_create_info, NULL,
         &god_rays.pipeline_layout));
 }

Referenced by vkpt_god_rays_create_pipelines().

◆ create_pipelines()

static void create_pipelines ( )

static

Definition at line 294 of file god_rays.c.

 {
     const VkPipelineShaderStageCreateInfo shader = {
         .sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO,
         .stage = VK_SHADER_STAGE_COMPUTE_BIT,
         .module = qvk.shader_modules[QVK_MOD_GOD_RAYS_COMP],
         .pName = "main"
     };
  
     const VkPipelineShaderStageCreateInfo filter_shader = {
         .sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO,
         .stage = VK_SHADER_STAGE_COMPUTE_BIT,
         .module = qvk.shader_modules[QVK_MOD_GOD_RAYS_FILTER_COMP],
         .pName = "main"
     };
  
     const VkComputePipelineCreateInfo pipeline_create_infos[2] = {
         {
             .sType = VK_STRUCTURE_TYPE_COMPUTE_PIPELINE_CREATE_INFO,
             .stage = shader,
             .layout = god_rays.pipeline_layout
         },
         {
             .sType = VK_STRUCTURE_TYPE_COMPUTE_PIPELINE_CREATE_INFO,
             .stage = filter_shader,
             .layout = god_rays.pipeline_layout
         },
     };
  
     _VK(vkCreateComputePipelines(qvk.device, VK_NULL_HANDLE, LENGTH(pipeline_create_infos), pipeline_create_infos,
         NULL, god_rays.pipelines));
 }

Referenced by vkpt_god_rays_create_pipelines().

◆ update_descriptor_set()

static void update_descriptor_set ( )

static

Definition at line 353 of file god_rays.c.

 {
     // if we end up here during init before we've called create_image_views(), punt --- we will be called again later
     if (god_rays.shadow_image_view == NULL)
         return;
  
     VkDescriptorImageInfo image_info = {
         .imageLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL,
         .imageView = god_rays.shadow_image_view,
         .sampler = god_rays.shadow_sampler
     };
     
     VkWriteDescriptorSet writes[1] = {
         {
             .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
             .dstSet = god_rays.descriptor_set,
             .descriptorCount = 1,
             .dstBinding = 0,
             .descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
             .pImageInfo = &image_info
         }
     };
  
     vkUpdateDescriptorSets(qvk.device, LENGTH(writes), writes, 0, NULL);
 }

Referenced by vkpt_god_rays_create_pipelines(), and vkpt_god_rays_update_images().

◆ vkpt_destroy_god_rays()

VkResult vkpt_destroy_god_rays ( )

Definition at line 65 of file god_rays.c.

 {
     vkDestroySampler(qvk.device, god_rays.shadow_sampler, NULL);
     vkDestroyDescriptorPool(qvk.device, god_rays.descriptor_pool, NULL);
  
     return VK_SUCCESS;
 }

◆ vkpt_god_rays_create_pipelines()

VkResult vkpt_god_rays_create_pipelines ( )

Definition at line 73 of file god_rays.c.

 {
     create_pipeline_layout();
     create_pipelines();
     create_descriptor_set();
     
     // this is a noop outside a shader reload
     update_descriptor_set();
  
     return VK_SUCCESS;
 }

◆ vkpt_god_rays_destroy_pipelines()

VkResult vkpt_god_rays_destroy_pipelines ( )

Definition at line 85 of file god_rays.c.

 {
     for (size_t i = 0; i < LENGTH(god_rays.pipelines); i++) {
         if (god_rays.pipelines[i]) {
             vkDestroyPipeline(qvk.device, god_rays.pipelines[i], NULL);
             god_rays.pipelines[i] = NULL;
         }
     }
  
     if (god_rays.pipeline_layout) {
         vkDestroyPipelineLayout(qvk.device, god_rays.pipeline_layout, NULL);
         god_rays.pipeline_layout = NULL;
     }
     
     if (god_rays.descriptor_set_layout) {
         vkDestroyDescriptorSetLayout(qvk.device, god_rays.descriptor_set_layout, NULL);
         god_rays.descriptor_set_layout = NULL;
     }
  
     return VK_SUCCESS;
 }

◆ vkpt_god_rays_enabled()

qboolean vkpt_god_rays_enabled ( const sun_light_t * sun_light )

Definition at line 379 of file god_rays.c.

 {
     return god_rays.enable->integer
         && god_rays.intensity->value > 0.f
         && sun_light->visible
         && !physical_sky_space->integer;  // god rays look weird in space because they also appear outside of the station
 }

Referenced by R_RenderFrame_RTX().

◆ vkpt_god_rays_noop()

VkResult vkpt_god_rays_noop ( )

Definition at line 116 of file god_rays.c.

 {
     return VK_SUCCESS;
 }

◆ vkpt_god_rays_prepare_ubo()

void vkpt_god_rays_prepare_ubo	(	QVKUniformBuffer_t *	ubo,
		const aabb_t *	world_aabb,
		const float *	proj,
		const float *	view,
		const float *	shadowmap_viewproj,
		float	shadowmap_depth_scale
	)

Definition at line 203 of file god_rays.c.

 {
     VectorAdd(world_aabb->mins, world_aabb->maxs, ubo->world_center);
     VectorScale(ubo->world_center, 0.5f, ubo->world_center);
     VectorSubtract(world_aabb->maxs, world_aabb->mins, ubo->world_size);
     VectorScale(ubo->world_size, 0.5f, ubo->world_half_size_inv);
     ubo->world_half_size_inv[0] = 1.f / ubo->world_half_size_inv[0];
     ubo->world_half_size_inv[1] = 1.f / ubo->world_half_size_inv[1];
     ubo->world_half_size_inv[2] = 1.f / ubo->world_half_size_inv[2];
     ubo->shadow_map_depth_scale = shadowmap_depth_scale;
  
     ubo->god_rays_intensity = max(0.f, god_rays.intensity->value);
     ubo->god_rays_eccentricity = god_rays.eccentricity->value;
  
     // Shadow parameters
     memcpy(ubo->shadow_map_VP, shadowmap_viewproj, 16 * sizeof(float));
 }

Referenced by R_RenderFrame_RTX().

◆ vkpt_god_rays_update_images()

VkResult vkpt_god_rays_update_images ( )

Definition at line 108 of file god_rays.c.

 {
     create_image_views();
     update_descriptor_set();
     return VK_SUCCESS;
 }

◆ vkpt_initialize_god_rays()

VkResult vkpt_initialize_god_rays ( )

Definition at line 51 of file god_rays.c.

 {
     memset(&god_rays, 0, sizeof(god_rays));
  
     VkPhysicalDeviceProperties properties;
     vkGetPhysicalDeviceProperties(qvk.physical_device, &properties);
  
     god_rays.intensity = Cvar_Get("gr_intensity", "2.0", 0);
     god_rays.eccentricity = Cvar_Get("gr_eccentricity", "0.75", 0);
     god_rays.enable = Cvar_Get("gr_enable", "1", 0);
  
     return VK_SUCCESS;
 }

◆ vkpt_record_god_rays_filter_command_buffer()

void vkpt_record_god_rays_filter_command_buffer ( VkCommandBuffer command_buffer )

Definition at line 169 of file god_rays.c.

 {
     BARRIER_COMPUTE(command_buffer, qvk.images[VKPT_IMG_PT_TRANSPARENT]);
  
     const VkImageSubresourceRange subresource_range = {
         .aspectMask = VK_IMAGE_ASPECT_COLOR_BIT,
         .levelCount = 1,
         .layerCount = 1
     };
  
     vkCmdBindPipeline(command_buffer, VK_PIPELINE_BIND_POINT_COMPUTE, god_rays.pipelines[1]);
  
     VkDescriptorSet desc_sets[] = {
         god_rays.descriptor_set,
         qvk.desc_set_vertex_buffer,
         qvk.desc_set_ubo,
         qvk_get_current_desc_set_textures(),
     };
  
     vkCmdBindDescriptorSets(command_buffer, VK_PIPELINE_BIND_POINT_COMPUTE, god_rays.pipeline_layout, 0, LENGTH(desc_sets),
         desc_sets, 0, NULL);
  
     int pass = 0;
     vkCmdPushConstants(command_buffer, god_rays.pipeline_layout, VK_SHADER_STAGE_COMPUTE_BIT, 0, sizeof(int), &pass);
  
     uint32_t group_size = FILTER_THREAD_GROUP_SIZE;
     uint32_t group_num_x = (qvk.extent_render.width / qvk.device_count + (group_size - 1)) / group_size;
     uint32_t group_num_y = (qvk.extent_render.height + (group_size - 1)) / group_size;
  
     vkCmdDispatch(command_buffer, group_num_x, group_num_y, 1);
  
     BARRIER_COMPUTE(command_buffer, qvk.images[VKPT_IMG_PT_TRANSPARENT]);
 }

Referenced by R_RenderFrame_RTX().

◆ vkpt_record_god_rays_trace_command_buffer()

void vkpt_record_god_rays_trace_command_buffer	(	VkCommandBuffer	command_buffer,
		int	pass
	)

Definition at line 140 of file god_rays.c.

 {
     BARRIER_COMPUTE(command_buffer, qvk.images[VKPT_IMG_PT_GODRAYS_THROUGHPUT_DIST]);
     BARRIER_COMPUTE(command_buffer, qvk.images[VKPT_IMG_ASVGF_COLOR]);
  
     vkCmdBindPipeline(command_buffer, VK_PIPELINE_BIND_POINT_COMPUTE, god_rays.pipelines[0]);
  
     VkDescriptorSet desc_sets[] = {
         god_rays.descriptor_set,
         qvk.desc_set_vertex_buffer,
         qvk.desc_set_ubo,
         qvk_get_current_desc_set_textures(),
     };
  
     vkCmdBindDescriptorSets(command_buffer, VK_PIPELINE_BIND_POINT_COMPUTE, god_rays.pipeline_layout, 0, LENGTH(desc_sets),
         desc_sets, 0, NULL);
  
     vkCmdPushConstants(command_buffer, god_rays.pipeline_layout, VK_SHADER_STAGE_COMPUTE_BIT, 0, sizeof(int), &pass);
  
     uint32_t group_size = THREAD_GROUP_SIZE;
     uint32_t group_num_x = (qvk.extent_render.width / (2 * qvk.device_count) + (group_size - 1)) / group_size;
     uint32_t group_num_y = (qvk.extent_render.height / 2 + (group_size - 1)) / group_size;
  
     vkCmdDispatch(command_buffer, group_num_x, group_num_y, 1);
  
     BARRIER_COMPUTE(command_buffer, qvk.images[VKPT_IMG_PT_GODRAYS_THROUGHPUT_DIST]);
     BARRIER_COMPUTE(command_buffer, qvk.images[VKPT_IMG_ASVGF_COLOR]);
 }

Referenced by R_RenderFrame_RTX().

Variable Documentation

◆ descriptor_pool

VkDescriptorPool descriptor_pool

Definition at line 32 of file god_rays.c.

◆ descriptor_set

VkDescriptorSet descriptor_set

Definition at line 33 of file god_rays.c.

◆ descriptor_set_layout

VkDescriptorSetLayout descriptor_set_layout

Definition at line 30 of file god_rays.c.

◆ eccentricity

cvar_t* eccentricity

Definition at line 39 of file god_rays.c.

◆ enable

cvar_t* enable

Definition at line 40 of file god_rays.c.

◆ FILTER_THREAD_GROUP_SIZE

const uint32_t FILTER_THREAD_GROUP_SIZE = 16

static

Definition at line 24 of file god_rays.c.

Referenced by vkpt_record_god_rays_filter_command_buffer().

◆ god_rays

struct { ... } god_rays

Referenced by create_descriptor_set(), create_image_views(), create_pipeline_layout(), create_pipelines(), update_descriptor_set(), vkpt_destroy_god_rays(), vkpt_god_rays_destroy_pipelines(), vkpt_god_rays_enabled(), vkpt_god_rays_prepare_ubo(), vkpt_initialize_god_rays(), vkpt_record_god_rays_filter_command_buffer(), and vkpt_record_god_rays_trace_command_buffer().

◆ intensity

cvar_t* intensity

Definition at line 38 of file god_rays.c.

Referenced by CL_AddPacketEntities(), and R_MarkLights_r().

◆ physical_sky_space

cvar_t* physical_sky_space

Definition at line 57 of file physical_sky.c.

Referenced by InitialiseSkyCVars(), vkpt_evaluate_sun_light(), vkpt_god_rays_enabled(), vkpt_physical_sky_endRegistration(), vkpt_physical_sky_record_cmd_buffer(), and vkpt_physical_sky_update_ubo().

◆ pipeline_layout

VkPipelineLayout pipeline_layout

Definition at line 29 of file god_rays.c.

◆ pipelines

VkPipeline pipelines[2]

Definition at line 28 of file god_rays.c.

Referenced by vkpt_vertex_buffer_create_pipelines().

◆ shadow_image_view

VkImageView shadow_image_view

Definition at line 35 of file god_rays.c.

◆ shadow_sampler

VkSampler shadow_sampler

Definition at line 36 of file god_rays.c.

◆ THREAD_GROUP_SIZE

const uint32_t THREAD_GROUP_SIZE = 16

static

Definition at line 23 of file god_rays.c.

Referenced by vkpt_record_god_rays_trace_command_buffer().

Macros

Functions

Variables

Macro Definition Documentation

◆ BARRIER_COMPUTE

Function Documentation

◆ create_descriptor_set()

◆ create_image_views()

◆ create_pipeline_layout()

◆ create_pipelines()

◆ update_descriptor_set()

◆ vkpt_destroy_god_rays()

◆ vkpt_god_rays_create_pipelines()

◆ vkpt_god_rays_destroy_pipelines()

◆ vkpt_god_rays_enabled()

◆ vkpt_god_rays_noop()

◆ vkpt_god_rays_prepare_ubo()

◆ vkpt_god_rays_update_images()

◆ vkpt_initialize_god_rays()

◆ vkpt_record_god_rays_filter_command_buffer()

◆ vkpt_record_god_rays_trace_command_buffer()

Variable Documentation

◆ descriptor_pool

◆ descriptor_set

◆ descriptor_set_layout

◆ eccentricity

◆ enable

◆ FILTER_THREAD_GROUP_SIZE

◆ god_rays

◆ intensity

◆ physical_sky_space

◆ pipeline_layout

◆ pipelines

◆ shadow_image_view

◆ shadow_sampler

◆ THREAD_GROUP_SIZE