source-engine/materialsystem/stdshaders/sky_hdr_dx9.cpp
FluorescentCIAAfricanAmerican 3bf9df6b27 1
2020-04-22 12:56:21 -04:00

298 lines
9.2 KiB
C++

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $Header: $
// $NoKeywords: $
//=============================================================================//
#include "BaseVSShader.h"
#include "sky_vs20.inc"
#include "sky_ps20.inc"
#include "sky_ps20b.inc"
#include "sky_hdr_compressed_ps20.inc"
#include "sky_hdr_compressed_ps20b.inc"
#include "sky_hdr_compressed_rgbs_ps20.inc"
#include "sky_hdr_compressed_rgbs_ps20b.inc"
#include "convar.h"
static ConVar mat_use_compressed_hdr_textures( "mat_use_compressed_hdr_textures", "1" );
DEFINE_FALLBACK_SHADER( Sky, Sky_HDR_DX9 )
BEGIN_VS_SHADER( Sky_HDR_DX9, "Help for Sky_HDR_DX9 shader" )
BEGIN_SHADER_PARAMS
SHADER_PARAM( HDRBASETEXTURE, SHADER_PARAM_TYPE_TEXTURE, "", "base texture when running with HDR enabled" )
SHADER_PARAM( HDRCOMPRESSEDTEXTURE, SHADER_PARAM_TYPE_TEXTURE, "", "base texture (compressed) for hdr compression method A" )
SHADER_PARAM( HDRCOMPRESSEDTEXTURE0, SHADER_PARAM_TYPE_TEXTURE, "", "compressed base texture0 for hdr compression method B" )
SHADER_PARAM( HDRCOMPRESSEDTEXTURE1, SHADER_PARAM_TYPE_TEXTURE, "", "compressed base texture1 for hdr compression method B" )
SHADER_PARAM( HDRCOMPRESSEDTEXTURE2, SHADER_PARAM_TYPE_TEXTURE, "", "compressed base texture2 for hdr compression method B" )
SHADER_PARAM_OVERRIDE( COLOR, SHADER_PARAM_TYPE_VEC3, "[ 1 1 1]", "color multiplier", SHADER_PARAM_NOT_EDITABLE )
END_SHADER_PARAMS
SHADER_FALLBACK
{
if( g_pHardwareConfig->GetDXSupportLevel() < 90 || g_pHardwareConfig->GetHDRType() == HDR_TYPE_NONE )
{
return "Sky_DX9";
}
return 0;
}
SHADER_INIT_PARAMS()
{
SET_FLAGS( MATERIAL_VAR_NOFOG );
SET_FLAGS( MATERIAL_VAR_IGNOREZ );
}
SHADER_INIT
{
// First figure out if sampler zero wants to be sRGB
int nSamplerZeroFlags = 0;
if ( (params[HDRCOMPRESSEDTEXTURE]->IsDefined()) && mat_use_compressed_hdr_textures.GetBool() )
{
nSamplerZeroFlags = 0;
}
else
{
if (params[HDRCOMPRESSEDTEXTURE0]->IsDefined())
{
nSamplerZeroFlags = 0;
}
else
{
nSamplerZeroFlags = TEXTUREFLAGS_SRGB;
if ( params[HDRBASETEXTURE]->IsDefined() && params[HDRBASETEXTURE]->IsTexture() )
{
ITexture *txtr=params[HDRBASETEXTURE]->GetTextureValue();
ImageFormat fmt=txtr->GetImageFormat();
if ( ( fmt == IMAGE_FORMAT_RGBA16161616F ) || ( fmt == IMAGE_FORMAT_RGBA16161616 ) )
{
nSamplerZeroFlags = 0;
}
}
}
}
// Next, figure out which texture will be on sampler zero
int nSampler0 = HDRCOMPRESSEDTEXTURE;
if ( params[HDRCOMPRESSEDTEXTURE]->IsDefined() && mat_use_compressed_hdr_textures.GetBool() )
{
nSampler0 = HDRCOMPRESSEDTEXTURE;
}
else
{
if ( params[HDRCOMPRESSEDTEXTURE0]->IsDefined() )
{
nSampler0 = HDRCOMPRESSEDTEXTURE0;
}
else
{
nSampler0 = HDRBASETEXTURE;
}
}
// Load the appropriate textures, making sure that the texture set on sampler 0 is sRGB if necessary
if ( params[HDRCOMPRESSEDTEXTURE]->IsDefined() && (mat_use_compressed_hdr_textures.GetBool() ) )
{
LoadTexture( HDRCOMPRESSEDTEXTURE, HDRCOMPRESSEDTEXTURE == nSampler0 ? nSamplerZeroFlags : 0 );
}
else
{
if (params[HDRCOMPRESSEDTEXTURE0]->IsDefined())
{
LoadTexture( HDRCOMPRESSEDTEXTURE0, HDRCOMPRESSEDTEXTURE0 == nSampler0 ? nSamplerZeroFlags : 0 );
if ( params[HDRCOMPRESSEDTEXTURE1]->IsDefined() )
{
LoadTexture( HDRCOMPRESSEDTEXTURE1, HDRCOMPRESSEDTEXTURE1 == nSampler0 ? nSamplerZeroFlags : 0 );
}
if ( params[HDRCOMPRESSEDTEXTURE2]->IsDefined())
{
LoadTexture( HDRCOMPRESSEDTEXTURE2, HDRCOMPRESSEDTEXTURE2 == nSampler0 ? nSamplerZeroFlags : 0 );
}
}
else
{
if ( params[HDRBASETEXTURE]->IsDefined() )
{
LoadTexture( HDRBASETEXTURE, HDRBASETEXTURE == nSampler0 ? nSamplerZeroFlags : 0 );
}
}
}
}
SHADER_DRAW
{
SHADOW_STATE
{
SetInitialShadowState();
// pShaderShadow->EnableAlphaWrites( true );
pShaderShadow->EnableTexture( SHADER_SAMPLER0, true );
pShaderShadow->VertexShaderVertexFormat( VERTEX_POSITION, 1, NULL, 0 );
DECLARE_STATIC_VERTEX_SHADER( sky_vs20 );
SET_STATIC_VERTEX_SHADER( sky_vs20 );
if ( (params[HDRCOMPRESSEDTEXTURE]->IsDefined()) &&
mat_use_compressed_hdr_textures.GetBool() )
{
pShaderShadow->EnableSRGBRead(SHADER_SAMPLER0,false);
if( g_pHardwareConfig->SupportsPixelShaders_2_b() )
{
DECLARE_STATIC_PIXEL_SHADER( sky_hdr_compressed_rgbs_ps20b );
SET_STATIC_PIXEL_SHADER( sky_hdr_compressed_rgbs_ps20b );
}
else
{
DECLARE_STATIC_PIXEL_SHADER( sky_hdr_compressed_rgbs_ps20 );
SET_STATIC_PIXEL_SHADER( sky_hdr_compressed_rgbs_ps20 );
}
}
else
{
if (params[HDRCOMPRESSEDTEXTURE0]->IsDefined())
{
pShaderShadow->EnableTexture( SHADER_SAMPLER1, true );
pShaderShadow->EnableTexture( SHADER_SAMPLER2, true );
pShaderShadow->EnableSRGBRead(SHADER_SAMPLER0,false);
pShaderShadow->EnableSRGBRead(SHADER_SAMPLER1,false);
pShaderShadow->EnableSRGBRead(SHADER_SAMPLER2,false);
if( g_pHardwareConfig->SupportsPixelShaders_2_b() )
{
DECLARE_STATIC_PIXEL_SHADER( sky_hdr_compressed_ps20b );
SET_STATIC_PIXEL_SHADER( sky_hdr_compressed_ps20b );
}
else
{
DECLARE_STATIC_PIXEL_SHADER( sky_hdr_compressed_ps20 );
SET_STATIC_PIXEL_SHADER( sky_hdr_compressed_ps20 );
}
}
else
{
ITexture *txtr=params[HDRBASETEXTURE]->GetTextureValue();
ImageFormat fmt=txtr->GetImageFormat();
if ((fmt==IMAGE_FORMAT_RGBA16161616F) || (fmt==IMAGE_FORMAT_RGBA16161616))
pShaderShadow->EnableSRGBRead(SHADER_SAMPLER0,false);
else
pShaderShadow->EnableSRGBRead(SHADER_SAMPLER0,true);
if( g_pHardwareConfig->SupportsPixelShaders_2_b() )
{
DECLARE_STATIC_PIXEL_SHADER( sky_ps20b );
SET_STATIC_PIXEL_SHADER( sky_ps20b );
}
else
{
DECLARE_STATIC_PIXEL_SHADER( sky_ps20 );
SET_STATIC_PIXEL_SHADER( sky_ps20 );
}
}
}
// we are writing linear values from this shader.
pShaderShadow->EnableSRGBWrite( true );
pShaderShadow->EnableAlphaWrites( true );
}
DYNAMIC_STATE
{
DECLARE_DYNAMIC_VERTEX_SHADER( sky_vs20 );
SET_DYNAMIC_VERTEX_SHADER( sky_vs20 );
// Texture coord transform
SetVertexShaderTextureTransform( VERTEX_SHADER_SHADER_SPECIFIC_CONST_1, BASETEXTURETRANSFORM );
float c0[4]={1,1,1,1};
if (params[COLOR]->IsDefined())
{
memcpy(c0,params[COLOR]->GetVecValue(),3*sizeof(float));
}
if (
params[HDRCOMPRESSEDTEXTURE]->IsDefined() &&
mat_use_compressed_hdr_textures.GetBool()
)
{
// set up data needs for pixel shader interpolation
ITexture *txtr=params[HDRCOMPRESSEDTEXTURE]->GetTextureValue();
float w=txtr->GetActualWidth();
float h=txtr->GetActualHeight();
float FUDGE=0.01/max(w,h); // per ATI
float c1[4]={(float)(0.5/w-FUDGE), (float)(0.5/h-FUDGE), w, h };
pShaderAPI->SetVertexShaderConstant( VERTEX_SHADER_SHADER_SPECIFIC_CONST_0, c1);
BindTexture( SHADER_SAMPLER0, HDRCOMPRESSEDTEXTURE, FRAME );
c0[0]*=8.0;
c0[1]*=8.0;
c0[2]*=8.0;
if( g_pHardwareConfig->SupportsPixelShaders_2_b() )
{
DECLARE_DYNAMIC_PIXEL_SHADER( sky_hdr_compressed_rgbs_ps20b );
SET_DYNAMIC_PIXEL_SHADER_COMBO( WRITE_DEPTH_TO_DESTALPHA, pShaderAPI->ShouldWriteDepthToDestAlpha() );
SET_DYNAMIC_PIXEL_SHADER( sky_hdr_compressed_rgbs_ps20b );
}
else
{
DECLARE_DYNAMIC_PIXEL_SHADER( sky_hdr_compressed_rgbs_ps20 );
SET_DYNAMIC_PIXEL_SHADER( sky_hdr_compressed_rgbs_ps20 );
}
}
else
{
float c1[4]={0,0,0,0};
pShaderAPI->SetVertexShaderConstant( VERTEX_SHADER_SHADER_SPECIFIC_CONST_0, c1);
if (params[HDRCOMPRESSEDTEXTURE0]->IsDefined() )
{
BindTexture( SHADER_SAMPLER0, HDRCOMPRESSEDTEXTURE0, FRAME );
BindTexture( SHADER_SAMPLER1, HDRCOMPRESSEDTEXTURE1, FRAME );
BindTexture( SHADER_SAMPLER2, HDRCOMPRESSEDTEXTURE2, FRAME );
if( g_pHardwareConfig->SupportsPixelShaders_2_b() )
{
DECLARE_DYNAMIC_PIXEL_SHADER( sky_hdr_compressed_ps20b );
SET_DYNAMIC_PIXEL_SHADER_COMBO( WRITE_DEPTH_TO_DESTALPHA, pShaderAPI->ShouldWriteDepthToDestAlpha() );
SET_DYNAMIC_PIXEL_SHADER( sky_hdr_compressed_ps20b );
}
else
{
DECLARE_DYNAMIC_PIXEL_SHADER( sky_hdr_compressed_ps20 );
SET_DYNAMIC_PIXEL_SHADER( sky_hdr_compressed_ps20 );
}
}
else
{
BindTexture( SHADER_SAMPLER0, HDRBASETEXTURE, FRAME );
ITexture *txtr=params[HDRBASETEXTURE]->GetTextureValue();
ImageFormat fmt=txtr->GetImageFormat();
if (
(fmt==IMAGE_FORMAT_RGBA16161616) ||
( (fmt==IMAGE_FORMAT_RGBA16161616F) &&
(g_pHardwareConfig->GetHDRType()==HDR_TYPE_INTEGER))
)
{
c0[0]*=16.0;
c0[1]*=16.0;
c0[2]*=16.0;
}
if( g_pHardwareConfig->SupportsPixelShaders_2_b() )
{
DECLARE_DYNAMIC_PIXEL_SHADER( sky_ps20b );
SET_DYNAMIC_PIXEL_SHADER_COMBO( WRITE_DEPTH_TO_DESTALPHA, pShaderAPI->ShouldWriteDepthToDestAlpha() );
SET_DYNAMIC_PIXEL_SHADER( sky_ps20b );
}
else
{
DECLARE_DYNAMIC_PIXEL_SHADER( sky_ps20 );
SET_DYNAMIC_PIXEL_SHADER( sky_ps20 );
}
}
}
pShaderAPI->SetPixelShaderConstant( 0, c0, 1 );
}
Draw( );
}
END_SHADER