anki-3d-engine/AnKi/Shaders/Include/MiscRendererTypes.h

// Copyright (C) 2009-present, Panagiotis Christopoulos Charitos and contributors.
// All rights reserved.
// Code licensed under the BSD License.
// http://www.anki3d.org/LICENSE

#pragma once

#include <AnKi/Shaders/Include/Common.h>

ANKI_BEGIN_NAMESPACE

/// Directional light (sun).
struct DirectionalLight
{
	Vec3 m_diffuseColor;
	F32 m_power;

	Vec3 m_direction;
	U32 m_shadowCascadeCount : 31; ///< If shadowCascadeCount is zero then it doesn't cast shadow.
	U32 m_active : 1;

	Vec4 m_shadowCascadeDistances;

	Mat4 m_textureMatrices[kMaxShadowCascades];

	Vec4 m_cascadeFarPlanes;

	Vec4 m_cascadePcfTexelRadius; ///< The radius of the PCF. In UV space.
};
static_assert(kMaxShadowCascades == 4u); // Because m_shadowCascadeDistances and m_cascadeFarPlanes is a Vec4

/// Common matrices and stuff.
struct CommonMatrices
{
	Mat3x4 m_cameraTransform;
	Mat3x4 m_view;
	Mat4 m_projection;
	Mat4 m_viewProjection;

	Mat4 m_jitter;
	Mat4 m_projectionJitter;
	Mat4 m_viewProjectionJitter;

	Mat4 m_invertedViewProjectionJitter; ///< To unproject in world space.
	Mat4 m_invertedViewProjection;
	Mat4 m_invertedProjectionJitter; ///< To unproject in view space.

	/// It's being used to reproject a clip space position of the current frame to the previous frame. Its value should be m_jitter *
	/// m_prevFrame.m_viewProjection * m_invertedViewProjectionJitter. At first it unprojects the current position to world space, all fine here. Then
	/// it projects to the previous frame as if the previous frame was using the current frame's jitter matrix.
	Mat4 m_reprojection;

	/// To unproject to view space. Jitter not considered.
	/// @code
	/// F32 z = m_unprojectionParameters.z / (m_unprojectionParameters.w + depth);
	/// Vec2 xy = ndc * m_unprojectionParameters.xy * z;
	/// pos = Vec3(xy, z);
	/// @endcode
	Vec4 m_unprojectionParameters;

	/// Part of the perspective projection matrix. Used in cheapPerspectiveUnprojection
	Vec4 m_projMat00_11_22_23;

	Vec2 m_jitterOffsetNdc;
	F32 m_near;
	F32 m_far;
};

enum class SkyType
{
	kSolidColor,
	kTextureWithEquirectangularMapping,
	kTextureWithEctahedronMapping,
};

struct Sky
{
	Vec3 m_solidColor;
	U32 m_type : 2; // One of SkyType
	U32 m_texture : 30;
};

struct IndirectDiffuseClipmapTextures
{
	U32 m_irradianceTexture : 16;
	U32 m_irradianceOctMapSize : 16;

	U32 m_radianceTexture : 16;
	U32 m_radianceOctMapSize : 16;

	U32 m_distanceMomentsTexture : 16;
	U32 m_distanceMomentsOctMapSize : 16;

	U32 m_averageIrradianceTexture : 16;
	U32 m_probeValidityTexture : 16;
};

struct IndirectDiffuseClipmapConstants
{
	UVec3 m_probeCounts;
	U32 m_totalProbeCount;

	Vec4 m_sizes[kIndirectDiffuseClipmapCount];

	Vec4 m_aabbMins[kIndirectDiffuseClipmapCount];

	Vec4 m_previousFrameAabbMins[kIndirectDiffuseClipmapCount];

	IndirectDiffuseClipmapTextures m_textures[kIndirectDiffuseClipmapCount];
};

struct LocalLightsGridConstants
{
	Vec3 m_volumeMin;
	F32 m_padding1;

	Vec3 m_volumeMax;
	F32 m_padding2;

	Vec3 m_cellSize;
	F32 m_padding3;

	UVec3 m_cellCounts;
	F32 m_padding4;
};

/// Common constants for all passes.
struct GlobalRendererConstants
{
	Vec2 m_renderingSize;
	F32 m_time;
	U32 m_frame;

	Vec4 m_nearPlaneWSpace;

	Vec3 m_cameraPosition;
	F32 m_reflectionProbesMipCount;

	UVec2 m_tileCounts;
	U32 m_zSplitCount;
	F32 m_zSplitCountOverFrustumLength; ///< m_zSplitCount/(far-near)

	Vec2 m_zSplitMagic; ///< It's the "a" and "b" of computeZSplitClusterIndex(). See there for details.
	U32 m_lightVolumeLastZSplit;
	U32 m_padding1;

	DirectionalLight m_directionalLight;

	CommonMatrices m_matrices;
	CommonMatrices m_previousMatrices;

	Sky m_sky;

	IndirectDiffuseClipmapConstants m_indirectDiffuseClipmaps;

	LocalLightsGridConstants m_localLightsGrid;
};

// RT shadows
struct RtShadowsDenoiseConstants
{
	Mat4 m_invViewProjMat;

	F32 m_time;
	U32 m_minSampleCount;
	U32 m_maxSampleCount;
	F32 m_padding2;
};

struct RtShadowsSbtBuildConstants
{
	U32 m_shaderHandleDwordSize;
	U32 m_sbtRecordDwordSize;
	U32 m_raygenHandleIndex; ///< Index to the handles buffer
	U32 m_missHandleIndex;
};

// Lens flare
struct LensFlareSprite
{
	Vec4 m_posScale; // xy: Position, zw: Scale
	Vec4 m_color;
	Vec4 m_depthPad3;
};

// Depth downscale
struct DepthDownscaleConstants
{
	Vec2 m_srcTexSizeOverOne;
	U32 m_threadgroupCount;
	U32 m_mipmapCount;
};

struct ReflectionConstants
{
	U32 m_ssrMaxIterations;
	U32 m_ssrStepIncrement;
	Vec2 m_roughnessCutoffToGiEdges;
};

struct PixelFailedSsr
{
	U32 m_pixel;
	U32 m_reflectionDirAndRoughness;
	U32 m_pdf_f16_rayDirT_f16;
};

// Vol fog
struct VolumetricFogConstants
{
	Vec3 m_fogDiffuse;
	F32 m_fogScatteringCoeff;

	F32 m_fogAbsorptionCoeff;
	F32 m_near;
	F32 m_far;
	F32 m_zSplitCountf;

	UVec3 m_volumeSize;
	F32 m_maxZSplitsToProcessf;
};

// Vol lighting
struct VolumetricLightingConstants
{
	F32 m_densityAtMinHeight;
	F32 m_densityAtMaxHeight;
	F32 m_minHeight;
	F32 m_oneOverMaxMinusMinHeight; // 1 / (maxHeight / minHeight)

	UVec3 m_volumeSize;
	F32 m_maxZSplitsToProcessf;
};

// SSAO
struct SsaoConstants
{
	F32 m_radius; ///< In meters.
	U32 m_sampleCount;
	Vec2 m_viewportSizef;

	Vec4 m_unprojectionParameters;

	F32 m_projectionMat00;
	F32 m_projectionMat11;
	F32 m_projectionMat22;
	F32 m_projectionMat23;

	Vec2 m_padding;
	F32 m_ssaoPower;
	U32 m_frameCount;

	Mat3x4 m_viewMat;

	Mat3x4 m_viewToWorldMat;
};

struct LodAndRenderableIndex
{
	U32 m_lod_2bit_renderableIndex_30bit;
};

ANKI_END_NAMESPACE