GxRenderShader.h

Go to the documentation of this file.

//===========================================================================
//===========================================================================
#pragma once
 
GX_CORE_NAMESPACE_BEGIN()
 
//===========================================================================
//===========================================================================
class GxRenderShader : public GxRenderResource
{
    //-------------------------------------------------------------
    //-------------------------------------------------------------
public:
    GX_RTTI_ABSTRACT_CLASS(GxRenderShader, GxRenderResource)
    // ClassBaseReference継承クラス用禁止宣言
    GX_PROHIBIT_CLASS_BASE_REFERENCE(GxRenderShader)
 
    friend class GxRenderDevice;
 
    enum class REGISTER_SET_TYPE
    {
        BOOL,       
        INT4,       
        FLOAT4,     
        SAMPLER,    
        MAX,        
    };
 
    enum class PARAMETER_TYPE
    {
        VOID_POINTER,   
        BOOL,           
        INT,            
        FLOAT,          
        STRING,         
        TEXTURE,        
        TEXTURE_1D,     
        TEXTURE_2D,     
        TEXTURE_3D,     
        TEXTURE_CUBE,   
        SAMPLER,        
        SAMPLER_1D,     
        SAMPLER_2D,     
        SAMPLER_3D,     
        SAMPLER_CUBE,   
        PIXEL_SHADER,   
        VERTEX_SHADER,  
        PIXEL_FRAGMENT, 
        VERTEX_FRAGMENT,
        MAX,            
    };
 
    struct GxConstantRegisterParameter
    {
        static constexpr u32 PARAMETER_NAME_MAX = GxMath::VALUE_64;
 
        GX_HANDLE               _handle;                        
        PARAMETER_TYPE          _type;                          
#if GX_USE_OPENGL
        u32                     _bindingLocationIndex;          
#endif //GX_USE_OPENGL
        u32                     _registerIndex;                 
        u32                     _registerCount;                 
        GX_CHAR                 _name[PARAMETER_NAME_MAX];      
    };
 
    struct GxRegisterSet
    {
        GxConstantRegisterParameter*    _pConstantRegisterParameter;        
        u32                             _constantRegisterParameterCount;    
        u32                             _constantRegisterStart;             
        u32                             _constantRegisterCount;             
    };
 
protected:
#if defined(_NX64)
    static constexpr u32 INVALID_POOL_OFFSET = static_cast<u32>(-1);
#endif //_NX64
 
    //-------------------------------------------------------------
    //-------------------------------------------------------------
public:
    void cleanup(void) override;
 
protected:
    virtual b32 initialize(const void* pAssembledCode, u32 assembledCodeSize, u32 constantRegisterParameterCount, u32 usageInputSemantics, u32 constantBufferSize) = 0;
    b32 initializeParameter(u32 constantRegisterParameterCount, u32 usageInputSemantics, u32 constantBufferSize);
 
#if defined(_NX64)
    void createShader(void* pShaderData, u32 shaderDataSize, void* pShaderControl, GxNativeShaderStageBits shaderStage);
#endif //_NX64
 
    GxRenderShader(void);
 
    //-------------------------------------------------------------
    //-------------------------------------------------------------
public:
    constexpr u32   getUsageInputSemantics(void) const { return _usageInputSemantics; }
    constexpr u32   getParameterCount(void) const { return _constantRegisterParameterCount; }
    constexpr u32   getConstantBufferSize(void) const { return _constantBufferSize; }
 
    constexpr void* getBuffer(void) const { return _pBuffer; }
    constexpr u32 getBufferSize(void) const { return _bufferSize; }
 
    constexpr u32 getParameterCount(REGISTER_SET_TYPE type) const { return _registerSet[static_cast<s32>(type)]._constantRegisterParameterCount; }
    constexpr GX_HANDLE getParameterHandle(REGISTER_SET_TYPE type, u32 index) const { return _registerSet[static_cast<s32>(type)]._pConstantRegisterParameter[index]._handle; }
    constexpr PARAMETER_TYPE getParameterType(REGISTER_SET_TYPE type, u32 index) const { return _registerSet[static_cast<s32>(type)]._pConstantRegisterParameter[index]._type; }
#if GX_USE_OPENGL
    constexpr u32 getParameterBindLocationIndex(REGISTER_SET_TYPE type, u32 index) const { return _registerSet[static_cast<s32>(type)]._pConstantRegisterParameter[index]._bindingLocationIndex; }
#endif //GX_USE_OPENGL
    constexpr u32 getParameterRegisterIndex(REGISTER_SET_TYPE type, u32 index) const { return _registerSet[static_cast<s32>(type)]._pConstantRegisterParameter[index]._registerIndex; }
    constexpr u32 getParameterRegisterCount(REGISTER_SET_TYPE type, u32 index) const { return _registerSet[static_cast<s32>(type)]._pConstantRegisterParameter[index]._registerCount; }
    constexpr GX_CSTR getParameterName(REGISTER_SET_TYPE type, u32 index) const { return _registerSet[static_cast<s32>(type)]._pConstantRegisterParameter[index]._name; }
 
    constexpr u32 getRegisterStart(REGISTER_SET_TYPE type) const { return _registerSet[static_cast<s32>(type)]._constantRegisterStart; }
    constexpr u32 getRegisterCount(REGISTER_SET_TYPE type) const { return _registerSet[static_cast<s32>(type)]._constantRegisterCount; }
    virtual void    getAssembledCode(void* pBuffer, u32 size) const;
    virtual u32     getAssembledCodeSize(void) const;
 
    // (DX11 など解放できるプラットフォームのみ)
    // isStoreDevelopMemory はバッファを ALLOCATOR_DEVELOP に移動します
    // これは開発中にシェーダーパッケージを保存する際に必要になります。
    GX_FORCE_INLINE virtual void releaseDiscardBuffer(b32 isStoreDevelopMemory = true) {}
 
#if defined(_NX64)
    GX_FORCE_INLINE GxNativeShader* getShader(void) { return &_shader; }
#elif defined(_PS5)
    constexpr GxNativeShader* getShader(void) const { return _pShader; }
#endif  // _PS5
 
#if GX_SHADER_DEBUG
    constexpr GX_CSTR getPath(void) const { return _path.getString(); }
    GX_FORCE_INLINE void setPath(GxString path) { _path = path; }
#endif //GX_SHADER_DEBUG
 
    //-------------------------------------------------------------
    //-------------------------------------------------------------
public:
    virtual b32         map(GxRegisterSet*& pRegisterSet, void*& pConstantRegisterParameterTable, u32 mappingType = 0);
    virtual void        unmap(void);
 
#if defined(_WINDOWS) || defined(_PS4) || defined(_PS5) || defined(_XBOX_XS)
    GX_FORCE_INLINE void    writeConstantBuffer(u32 startRegister, const f32* pFloatArray, u32 count);
    void                    updateConstantBuffer(void);
#if GX_USE_DX11
    ID3D11Buffer*           getConstantBuffer(void) const { return _pD3D11ConstantBuffer; }
#endif //GX_USE_DX11
#endif  // _WINDOWS || _PS4 || _PS5 || _XBOX_XS
 
    //-------------------------------------------------------------
    //-------------------------------------------------------------
protected:
    GxRegisterSet               _registerSet[static_cast<s32>(REGISTER_SET_TYPE::MAX)];     
    void*                       _pConstantRegisterParameterTable;   
    u32                         _constantRegisterParameterCount;    
    u32                         _usageInputSemantics;               
    u32                         _constantBufferSize;                
    u32                         _mappingType;                       
    void*                       _pBuffer;                           
    u32                         _bufferSize;                        
 
#if GX_USE_DX11
    ID3D11Buffer*               _pD3D11ConstantBuffer;              
#endif //GX_USE_DX11
#if GX_USE_DX11 || GX_USE_DX12 || defined(_PS5)
    GxVector4*                  _pConstantBufferNow;                
    GxVector4*                  _pConstantBufferSet;                
#elif defined(_NX64)
    u32                         _poolOffset;                        
#endif  // _NX64
#if defined(_NX64)
    GxNativeShader              _shader;                            
#elif defined(_PS5)
    GxNativeShader*             _pShader;                           
#endif //_PS5
#if GX_SHADER_DEBUG
    GxString                    _path;                              
#endif //GX_SHADER_DEBUG
 
};
 
#if defined(_WINDOWS) || defined(_PS4) || defined(_PS5) || defined(_XBOX_XS)
//---------------------------------------------------------------------------
//  定数バッファ書き込み
//---------------------------------------------------------------------------
void GxRenderShader::writeConstantBuffer(u32 startRegister, const f32* pFloatArray, u32 count)
{
    GX_MEMCPY(&_pConstantBufferNow[startRegister], pFloatArray, sizeof(GxVector4) * count);
}
#endif  //_WINDOWS || _PS4 || _PS5 || _XBOX_XS
 
GX_CORE_NAMESPACE_END()