F:/My Projects/LSEngine/Modules/LSGraphicsLib/Src/OpenGLShaders/LSGOpenGlShaderProgramManager.h

#ifndef __LSG_OPENGLSHADERPROGRAMMANAGER_H__
#define __LSG_OPENGLSHADERPROGRAMMANAGER_H__

#include "../LSGGraphicsLib.h"
#include "../ShaderLanguage/LSGShaderBase.h"
#include "../ShaderLanguage/LSGShaderTranslationProcessorBase.h"
#include "../VertexBuffer/LSGVertexBufferBase.h"
#include "LSGOpenGlCompiledShader.h"
#include "Map/LSTLMap.h"

#if defined( LSG_OGL )

namespace lsg {

        class COpenGlShaderProgramManager {
        public :
                // == Types.
                typedef struct LSG_ATTRIBUTE_USAGE_LOCS {
                        GLint                                                                                                           iLoc[LSG_VBM_MAX_USAGES];
                } * LPLSG_ATTRIBUTE_USAGE_LOCS, * const LPCLSG_ATTRIBUTE_USAGE_LOCS;

                class COpenGlShaderProgram {
                public :
                        // == Various constructors.
                        LSE_CALLCTOR                                                                                            COpenGlShaderProgram();
                        LSE_CALLCTOR                                                                                            ~COpenGlShaderProgram();


                        // == Functions.
                        LSVOID LSE_CALL                                                                                         AddShader( COpenGlCompiledShader * _pogcsShader );

                        COpenGlCompiledShader * LSE_CALL                                                        GetShader( CShaderLanguageParser::LSG_SHADER_TYPES _stShaderType );

                        const COpenGlCompiledShader * LSE_CALL                                          GetShader( CShaderLanguageParser::LSG_SHADER_TYPES _stShaderType ) const;

                        LSBOOL LSE_CALL                                                                                         Compile( const CShaderTranslationProcessorBase::LSG_UNIFORM_LIST * _pulUniforms,
                                const CShaderTranslationProcessorBase::LSG_ATTRIBUTE_LIST * _palAttributes,
                                CString &_sError );

                        LSVOID LSE_CALL                                                                                         Activate();

                        LSUINT32 LSE_CALL                                                                                       GetRef() const;

                        LSUINT32 LSE_CALL                                                                                       GetId() const;

                        const LSG_ATTRIBUTE_USAGE_LOCS * LSE_CALL                                       GetAttributeLocs() const;

                        CShaderBase::LSG_HANDLE LSE_CALL                                                        GetConstantByName( const LSCHAR * _pcName );

                        LSBOOL LSE_CALL                                                                                         SetFloat( CShaderBase::LSG_HANDLE _hHandle, LSFLOAT _fValue );

                        LSBOOL LSE_CALL                                                                                         SetFloatArray( CShaderBase::LSG_HANDLE _hHandle, const LSFLOAT * _pfValues, LSUINT32 _ui32Total );

                        LSBOOL LSE_CALL                                                                                         SetInt( CShaderBase::LSG_HANDLE _hHandle, LSINT32 _iValue );

                        LSBOOL LSE_CALL                                                                                         SetIntArray( CShaderBase::LSG_HANDLE _hHandle, const LSINT32 * _piValues, LSUINT32 _ui32Total );

                        LSBOOL LSE_CALL                                                                                         SetBoolArray( CShaderBase::LSG_HANDLE _hHandle, const LSBOOL * _pbValues, LSUINT32 _ui32Total );

                        LSBOOL LSE_CALL                                                                                         SetVector2( CShaderBase::LSG_HANDLE _hHandle, const CVector2 &_vValue );

                        LSBOOL LSE_CALL                                                                                         SetVector2Array( CShaderBase::LSG_HANDLE _hHandle, const CVector2 * _pvValues, LSUINT32 _ui32Total );

                        LSBOOL LSE_CALL                                                                                         SetVector3( CShaderBase::LSG_HANDLE _hHandle, const CVector3 &_vValue );

                        LSBOOL LSE_CALL                                                                                         SetVector3Array( CShaderBase::LSG_HANDLE _hHandle, const CVector3 * _pvValues, LSUINT32 _ui32Total );

                        LSBOOL LSE_CALL                                                                                         SetVector4( CShaderBase::LSG_HANDLE _hHandle, const CVector4 &_vValue );

                        LSBOOL LSE_CALL                                                                                         SetVector4Array( CShaderBase::LSG_HANDLE _hHandle, const CVector4 * _pvValues, LSUINT32 _ui32Total );

                        LSBOOL LSE_CALL                                                                                         SetMatrix2x2( CShaderBase::LSG_HANDLE _hHandle, const CMatrix2x2 &_mValue );

                        LSBOOL LSE_CALL                                                                                         SetMatrix3x3( CShaderBase::LSG_HANDLE _hHandle, const CMatrix3x3 &_mValue );

                        LSBOOL LSE_CALL                                                                                         SetMatrix4x4( CShaderBase::LSG_HANDLE _hHandle, const CMatrix4x4 &_mValue );

                        LSBOOL LSE_CALL                                                                                         SetMatrix4x4Array( CShaderBase::LSG_HANDLE _hHandle, const CMatrix4x4 * _pmValues, LSUINT32 _ui32Total );


                protected :
                        // == Types.
                        typedef struct LSG_UNIFORM_MAP {
                                LSUINTPTR                                                                                               uiptrApiId;

                                LSUINT32                                                                                                ui32DataPtr;

                                LSBOOL                                                                                                  bDirty;

                                LSUINT32                                                                                                ui32Name;

                                CShaderLanguageParser::LSG_SHADER_TYPES                                 stType;
                        } * LPLSG_UNIFORM_MAP, * const LPCLSG_UNIFORM_MAP;


                        // == Members.
                        COpenGlCompiledShader *                                                                         m_pogcsShader[CShaderLanguageParser::LSG_ST_TOTAL];

                        LSG_ATTRIBUTE_USAGE_LOCS *                                                                      m_paulLocs;

                        GLuint                                                                                                          m_uiProgram;

                        LSUINT32                                                                                                        m_ui32Ref;

                        CShaderTranslationProcessorBase::LSG_UNIFORM_LIST                       m_ulUniforms[CShaderLanguageParser::LSG_ST_TOTAL];

                        LSUINT32                                                                                                        m_ui32NextUniformStart;

                        CVectorPoD<LSUINT8, LSUINT32>                                                           m_vUniformBuffer;

                        CVectorPoD<LSG_UNIFORM_MAP>                                                                     m_vUniformDataMap;

                        CVectorPoD<LSUINT8, LSUINT32>                                                           m_vTempBuffer;


                        // == Functions.
                        LSVOID LSE_CALL                                                                                         IncRef();

                        LSVOID LSE_CALL                                                                                         DecRef();

                        static LSUINT32 LSE_CALL                                                                        ShaderSemanticToVertexAttribute( LSUINT32 _ui32Semantic );

                private :
                        friend class                                                                                            COpenGlShaderProgramManager;
                };


                // == Functions.
                static LSVOID LSE_CALL                                                                                  SetSharing( LSBOOL _bVal );

                static COpenGlShaderProgram * LSE_CALL                                                  CreateShaderProgram( COpenGlCompiledShader * _pogcsVertexShader,
                        COpenGlCompiledShader * _pogcsPixelShader, COpenGlCompiledShader * _pogcsGeometryShader,
                        const CShaderTranslationProcessorBase::LSG_UNIFORM_LIST * _pulUniforms,
                        const CShaderTranslationProcessorBase::LSG_ATTRIBUTE_LIST * _palAttributes,
                        CString &_sError );

                static LSVOID LSE_CALL                                                                                  DeleteShaderProgram( COpenGlShaderProgram * _pogspProgram );

                static LSUINT32 LSE_CALL                                                                                DeleteUnreferencedShaderPrograms();


        protected :
                // == Types.
                typedef struct LSG_SHADER_PROGRAM_KEY {
                        LSUINT32                                                                                                        ui32VertexId;

                        LSUINT32                                                                                                        ui32PixelId;

                        LSUINT32                                                                                                        ui32GeometryId;


                        // == Various constructors.
                        LSE_CALLCTOR                                                                                            LSG_SHADER_PROGRAM_KEY();
                        
                        // == Operators.
                        LSBOOL LSE_CALL                                                                                         operator < ( const LSG_SHADER_PROGRAM_KEY &_spkOther ) const;

                        LSBOOL LSE_CALL                                                                                         operator == ( const LSG_SHADER_PROGRAM_KEY &_spkOther ) const;
                } * LPLSG_SHADER_PROGRAM_KEY, * LPCLSG_SHADER_PROGRAM_KEY;


                // == Members.
                // Is sharing enabled?  Defaults to true.  Tools/editors should set this to false at program start.
                static LSBOOL                                                                                                   m_bSharing;

                // Map of shader programs.
                static CMap<LSG_SHADER_PROGRAM_KEY, COpenGlShaderProgram *>             m_mShaders;


                // == Functions.
                static LSG_SHADER_PROGRAM_KEY LSE_CALL                                                  CreateKey( const COpenGlShaderProgram &_ogspProgram );
        };

}       // namespace lsg

#endif  // #if defined( LSG_OGL )

#endif  // __LSG_OPENGLSHADERPROGRAMMANAGER_H__