qzdoom/src/gl/shaders/gl_ambientshader.h

#ifndef __GL_AMBIENTSHADER_H
#define __GL_AMBIENTSHADER_H

#include <memory>

#include "gl_shaderprogram.h"

class FLinearDepthShader
{
public:
	void Bind();

	FBufferedUniformSampler DepthTexture;
	FBufferedUniformSampler ColorTexture;

	struct UniformBlock
	{
		int SampleIndex;
		float LinearizeDepthA;
		float LinearizeDepthB;
		float InverseDepthRangeA;
		float InverseDepthRangeB;
		float Padding0, Padding1, Padding2;
		FVector2 Scale;
		FVector2 Offset;

		static std::vector<UniformFieldDesc> Desc()
		{
			return
			{
				{ "SampleIndex", UniformType::Int, offsetof(UniformBlock, SampleIndex) },
				{ "LinearizeDepthA", UniformType::Float, offsetof(UniformBlock, LinearizeDepthA) },
				{ "LinearizeDepthB", UniformType::Float, offsetof(UniformBlock, LinearizeDepthB) },
				{ "InverseDepthRangeA", UniformType::Float, offsetof(UniformBlock, InverseDepthRangeA) },
				{ "InverseDepthRangeB", UniformType::Float, offsetof(UniformBlock, InverseDepthRangeB) },
				{ "Padding0", UniformType::Float, offsetof(UniformBlock, Padding0) },
				{ "Padding1", UniformType::Float, offsetof(UniformBlock, Padding1) },
				{ "Padding2", UniformType::Float, offsetof(UniformBlock, Padding2) },
				{ "Scale", UniformType::Vec2, offsetof(UniformBlock, Scale) },
				{ "Offset", UniformType::Vec2, offsetof(UniformBlock, Offset) }
			};
		}
	};
	
	ShaderUniforms<UniformBlock, POSTPROCESS_BINDINGPOINT> Uniforms;

private:
	std::unique_ptr<FShaderProgram> mShader;
	bool mMultisample = false;
};

class FSSAOShader
{
public:
	void Bind();

	FBufferedUniformSampler DepthTexture;
	FBufferedUniformSampler NormalTexture;
	FBufferedUniformSampler RandomTexture;

	struct UniformBlock
	{
		FVector2 UVToViewA;
		FVector2 UVToViewB;
		FVector2 InvFullResolution;
		float NDotVBias;
		float NegInvR2;
		float RadiusToScreen;
		float AOMultiplier;
		float AOStrength;
		int SampleIndex;
		float Padding0, Padding1;
		FVector2 Scale;
		FVector2 Offset;

		static std::vector<UniformFieldDesc> Desc()
		{
			return
			{
				{ "UVToViewA", UniformType::Vec2, offsetof(UniformBlock, UVToViewA) },
				{ "UVToViewB", UniformType::Vec2, offsetof(UniformBlock, UVToViewB) },
				{ "InvFullResolution", UniformType::Vec2, offsetof(UniformBlock, InvFullResolution) },
				{ "NDotVBias", UniformType::Float, offsetof(UniformBlock, NDotVBias) },
				{ "NegInvR2", UniformType::Float, offsetof(UniformBlock, NegInvR2) },
				{ "RadiusToScreen", UniformType::Float, offsetof(UniformBlock, RadiusToScreen) },
				{ "AOMultiplier", UniformType::Float, offsetof(UniformBlock, AOMultiplier) },
				{ "AOStrength", UniformType::Float, offsetof(UniformBlock, AOStrength) },
				{ "SampleIndex", UniformType::Int, offsetof(UniformBlock, SampleIndex) },
				{ "Padding0", UniformType::Float, offsetof(UniformBlock, Padding0) },
				{ "Padding1", UniformType::Float, offsetof(UniformBlock, Padding1) },
				{ "Scale", UniformType::Vec2, offsetof(UniformBlock, Scale) },
				{ "Offset", UniformType::Vec2, offsetof(UniformBlock, Offset) },
			};
		}
	};

	ShaderUniforms<UniformBlock, POSTPROCESS_BINDINGPOINT> Uniforms;

private:
	enum Quality
	{
		Off,
		LowQuality,
		MediumQuality,
		HighQuality,
		NumQualityModes
	};

	FString GetDefines(int mode, bool multisample);

	std::unique_ptr<FShaderProgram> mShader;
	Quality mCurrentQuality = Off;
	bool mMultisample = false;
};

class FDepthBlurShader
{
public:
	void Bind(bool vertical);

	FBufferedUniformSampler AODepthTexture[2];

	struct UniformBlock
	{
		float BlurSharpness;
		float PowExponent;
		FVector2 InvFullResolution;

		static std::vector<UniformFieldDesc> Desc()
		{
			return
			{
				{ "BlurSharpness", UniformType::Float, offsetof(UniformBlock, BlurSharpness) },
				{ "PowExponent", UniformType::Float, offsetof(UniformBlock, PowExponent) },
				{ "InvFullResolution", UniformType::Vec2, offsetof(UniformBlock, InvFullResolution) }
			};
		}
	};

	ShaderUniforms<UniformBlock, POSTPROCESS_BINDINGPOINT> Uniforms[2];

private:
	FShaderProgram mShader[2];
};

class FSSAOCombineShader
{
public:
	void Bind();

	FBufferedUniformSampler AODepthTexture;
	FBufferedUniformSampler SceneFogTexture;

	struct UniformBlock
	{
		int SampleCount;
		int Padding0, Padding1, Padding2;
		FVector2 Scale;
		FVector2 Offset;

		static std::vector<UniformFieldDesc> Desc()
		{
			return
			{
				{ "SampleCount", UniformType::Int, offsetof(UniformBlock, SampleCount) },
				{ "Padding0", UniformType::Int, offsetof(UniformBlock, Padding0) },
				{ "Padding1", UniformType::Int, offsetof(UniformBlock, Padding1) },
				{ "Padding2", UniformType::Int, offsetof(UniformBlock, Padding2) },
				{ "Scale", UniformType::Vec2, offsetof(UniformBlock, Scale) },
				{ "Offset", UniformType::Vec2, offsetof(UniformBlock, Offset) }
			};
		}
	};

	ShaderUniforms<UniformBlock, POSTPROCESS_BINDINGPOINT> Uniforms;

private:
	std::unique_ptr<FShaderProgram> mShader;
	bool mMultisample = false;
};

#endif