sqwarmed/sdk_src/public/shaderlib/BaseShader.h

286 lines
9.9 KiB
C++

//===== Copyright © 1996-2005, Valve Corporation, All rights reserved. ======//
//
// Purpose:
//
// $NoKeywords: $
// This is what all shaders inherit from.
//===========================================================================//
#ifndef BASESHADER_H
#define BASESHADER_H
#ifdef _WIN32
#pragma once
#endif
#include "materialsystem/IShader.h"
#include "materialsystem/imaterialvar.h"
#include "materialsystem/ishaderapi.h"
#include "materialsystem/imaterialsystemhardwareconfig.h"
//-----------------------------------------------------------------------------
// Forward declarations
//-----------------------------------------------------------------------------
class IMaterialVar;
class CPerInstanceContextData;
//-----------------------------------------------------------------------------
// Standard material vars
//-----------------------------------------------------------------------------
// Note: if you add to these, add to s_StandardParams in CBaseShader.cpp
enum ShaderMaterialVars_t
{
FLAGS = 0,
FLAGS_DEFINED, // mask indicating if the flag was specified
FLAGS2,
FLAGS_DEFINED2,
COLOR,
ALPHA,
BASETEXTURE,
FRAME,
BASETEXTURETRANSFORM,
FLASHLIGHTTEXTURE,
FLASHLIGHTTEXTUREFRAME,
COLOR2,
SRGBTINT,
NUM_SHADER_MATERIAL_VARS
};
// Alpha belnd mode enums. Moved from basevsshader
enum BlendType_t
{
// no alpha blending
BT_NONE = 0,
// src * srcAlpha + dst * (1-srcAlpha)
// two passes for HDR:
// pass 1:
// color: src * srcAlpha + dst * (1-srcAlpha)
// alpha: srcAlpha * zero + dstAlpha * (1-srcAlpha)
// pass 2:
// color: none
// alpha: srcAlpha * one + dstAlpha * one
//
BT_BLEND,
// src * one + dst * one
// one pass for HDR
BT_ADD,
// Why do we ever use this instead of using premultiplied alpha?
// src * srcAlpha + dst * one
// two passes for HDR
// pass 1:
// color: src * srcAlpha + dst * one
// alpha: srcAlpha * one + dstAlpha * one
// pass 2:
// color: none
// alpha: srcAlpha * one + dstAlpha * one
BT_BLENDADD
};
//-----------------------------------------------------------------------------
// Base class for shaders, contains helper methods.
//-----------------------------------------------------------------------------
class CBaseShader : public IShader
{
public:
// constructor
CBaseShader();
// Methods inherited from IShader
virtual char const* GetFallbackShader( IMaterialVar** params ) const { return 0; }
virtual int GetParamCount( ) const;
virtual const ShaderParamInfo_t& GetParamInfo( int paramIndex ) const;
virtual void InitShaderParams( IMaterialVar** ppParams, const char *pMaterialName );
virtual void InitShaderInstance( IMaterialVar** ppParams, IShaderInit *pShaderInit, const char *pMaterialName, const char *pTextureGroupName );
virtual void DrawElements( IMaterialVar **params, int nModulationFlags, IShaderShadow* pShaderShadow, IShaderDynamicAPI* pShaderAPI,
VertexCompressionType_t vertexCompression, CBasePerMaterialContextData **pContext, CBasePerInstanceContextData** pInstanceDataPtr );
virtual int ComputeModulationFlags( IMaterialVar** params, IShaderDynamicAPI* pShaderAPI );
virtual bool NeedsPowerOfTwoFrameBufferTexture( IMaterialVar **params, bool bCheckSpecificToThisFrame = true ) const;
virtual bool NeedsFullFrameBufferTexture( IMaterialVar **params, bool bCheckSpecificToThisFrame = true ) const;
virtual bool IsTranslucent( IMaterialVar **params ) const;
public:
// These functions must be implemented by the shader
virtual void OnInitShaderParams( IMaterialVar** ppParams, const char *pMaterialName ) {}
virtual void OnInitShaderInstance( IMaterialVar** ppParams, IShaderInit *pShaderInit, const char *pMaterialName ) = 0;
virtual void OnDrawElements( IMaterialVar **params, IShaderShadow* pShaderShadow, IShaderDynamicAPI* pShaderAPI, VertexCompressionType_t vertexCompression, CBasePerMaterialContextData **pContextDataPtr ) = 0;
// Sets the default shadow state
void SetInitialShadowState( );
// Draws a snapshot
void Draw( bool bMakeActualDrawCall = true );
// Are we currently taking a snapshot?
bool IsSnapshotting() const;
// Methods related to building per-instance ("PI_") command buffers
void PI_BeginCommandBuffer();
void PI_EndCommandBuffer();
void PI_SetPixelShaderAmbientLightCube( int nFirstRegister );
void PI_SetPixelShaderLocalLighting( int nFirstRegister );
void PI_SetPixelShaderAmbientLightCubeLuminance( int nFirstRegister );
void PI_SetPixelShaderGlintDamping( int nFirstRegister );
void PI_SetVertexShaderAmbientLightCube( /*int nFirstRegister*/ );
void PI_SetModulationPixelShaderDynamicState( int nRegister );
void PI_SetModulationPixelShaderDynamicState_LinearColorSpace_LinearScale( int nRegister, float scale );
void PI_SetModulationPixelShaderDynamicState_LinearScale( int nRegister, float scale );
void PI_SetModulationPixelShaderDynamicState_LinearScale_ScaleInW( int nRegister, float scale );
void PI_SetModulationPixelShaderDynamicState_LinearColorSpace( int nRegister );
void PI_SetModulationPixelShaderDynamicState_Identity( int nRegister );
void PI_SetModulationVertexShaderDynamicState( void );
void PI_SetModulationVertexShaderDynamicState_LinearScale( float flScale );
// Gets at the current materialvar flags
int CurrentMaterialVarFlags() const;
// Gets at the current materialvar2 flags
int CurrentMaterialVarFlags2() const;
// Finds a particular parameter (works because the lowest parameters match the shader)
int FindParamIndex( const char *pName ) const;
// Are we using graphics?
bool IsUsingGraphics();
// Are we using editor materials?
bool CanUseEditorMaterials() const;
// Loads a texture
void LoadTexture( int nTextureVar );
// Loads a bumpmap
void LoadBumpMap( int nTextureVar );
// Loads a cubemap
void LoadCubeMap( int nTextureVar );
// get the shaderapi handle for a texture. BE CAREFUL WITH THIS.
ShaderAPITextureHandle_t GetShaderAPITextureBindHandle( int nTextureVar, int nFrameVar, int nTextureChannel = 0 );
ShaderAPITextureHandle_t GetShaderAPITextureBindHandle( ITexture *pTexture, int nFrame, int nTextureChannel = 0 );
// Binds a texture
void BindTexture( Sampler_t sampler1, Sampler_t sampler2, int nTextureVar, int nFrameVar = -1 );
void BindTexture( Sampler_t sampler1, int nTextureVar, int nFrameVar = -1 );
void BindTexture( Sampler_t sampler1, ITexture *pTexture, int nFrame = 0 );
void BindTexture( Sampler_t sampler1, Sampler_t sampler2, ITexture *pTexture, int nFrame = 0 );
// Bind vertex texture
void BindVertexTexture( VertexTextureSampler_t vtSampler, int nTextureVar, int nFrame = 0 );
// Is the texture translucent?
bool TextureIsTranslucent( int textureVar, bool isBaseTexture );
// Is the color var white?
bool IsWhite( int colorVar );
// Helper methods for fog
void FogToOOOverbright( void );
void FogToWhite( void );
void FogToBlack( void );
void FogToGrey( void );
void FogToFogColor( void );
void DisableFog( void );
void DefaultFog( void );
// Helpers for alpha blending
void EnableAlphaBlending( ShaderBlendFactor_t src, ShaderBlendFactor_t dst );
void DisableAlphaBlending();
void SetBlendingShadowState( BlendType_t nMode );
void SetNormalBlendingShadowState( int textureVar = -1, bool isBaseTexture = true );
void SetAdditiveBlendingShadowState( int textureVar = -1, bool isBaseTexture = true );
void SetDefaultBlendingShadowState( int textureVar = -1, bool isBaseTexture = true );
void SingleTextureLightmapBlendMode( );
// Helpers for color modulation
bool IsAlphaModulating();
// Helpers for HDR
bool IsHDREnabled( void );
bool UsingFlashlight( IMaterialVar **params ) const;
bool UsingEditor( IMaterialVar **params ) const;
void ApplyColor2Factor( float *pColorOut ) const; // (*pColorOut) *= COLOR2
private:
// This is a per-instance state which is handled completely by the system
void PI_SetSkinningMatrices();
void PI_SetVertexShaderLocalLighting( );
FORCEINLINE void SetFogMode( ShaderFogMode_t fogMode );
protected:
static IMaterialVar **s_ppParams;
static const char *s_pTextureGroupName; // Current material's texture group name.
static IShaderShadow *s_pShaderShadow;
static IShaderDynamicAPI *s_pShaderAPI;
static IShaderInit *s_pShaderInit;
private:
static int s_nPassCount;
static int s_nModulationFlags;
static CPerInstanceContextData** s_pInstanceDataPtr;
template <class T> friend class CBaseCommandBufferBuilder;
};
//-----------------------------------------------------------------------------
// Gets at the current materialvar flags
//-----------------------------------------------------------------------------
inline int CBaseShader::CurrentMaterialVarFlags() const
{
return s_ppParams[FLAGS]->GetIntValue();
}
//-----------------------------------------------------------------------------
// Gets at the current materialvar2 flags
//-----------------------------------------------------------------------------
inline int CBaseShader::CurrentMaterialVarFlags2() const
{
return s_ppParams[FLAGS2]->GetIntValue();
}
//-----------------------------------------------------------------------------
// Are we currently taking a snapshot?
//-----------------------------------------------------------------------------
inline bool CBaseShader::IsSnapshotting() const
{
return (s_pShaderShadow != NULL);
}
//-----------------------------------------------------------------------------
// Is the color var white?
//-----------------------------------------------------------------------------
inline bool CBaseShader::IsWhite( int colorVar )
{
if (colorVar < 0)
return true;
if (!s_ppParams[colorVar]->IsDefined())
return true;
float color[3];
s_ppParams[colorVar]->GetVecValue( color, 3 );
return (color[0] >= 1.0f) && (color[1] >= 1.0f) && (color[2] >= 1.0f);
}
#endif // BASESHADER_H