// EMERGENT GAME TECHNOLOGIES PROPRIETARY INFORMATION
//
// This software is supplied under the terms of a license agreement or
// nondisclosure agreement with Emergent Game Technologies and may not 
// be copied or disclosed except in accordance with the terms of that 
// agreement.
//
//      Copyright (c) 1996-2008 Emergent Game Technologies.
//      All Rights Reserved.
//
// Emergent Game Technologies, Chapel Hill, North Carolina 27517
// http://www.emergent.net

#ifndef NID3DSHADERCONSTANTMAP_H
#define NID3DSHADERCONSTANTMAP_H

#include <NiRenderObject.h>
#include <NiShaderConstantMap.h>
#include <NiTArray.h>
#include "NiD3DDefines.h"
#include "NiD3DError.h"
#include "NiDX9RenderState.h"
#include "NiShaderConstantMapEntry.h"

class NiD3DShaderProgram;

class NID3D_ENTRY NiD3DShaderConstantMap : public NiShaderConstantMap
{
public:
    NiD3DShaderConstantMap(NiGPUProgram::ProgramType eType);
    virtual ~NiD3DShaderConstantMap();

    //*** Access functions
    inline bool GetModified();
    inline void SetModified(bool bModified);
    inline NiD3DError GetLastError();

    //*** Entry access/manipulation functions
    inline unsigned int GetEntryCount();

    virtual NiShaderError AddEntry(const char* pszKey, unsigned int uiFlags,
        unsigned int uiExtra, unsigned int uiShaderRegister, 
        unsigned int uiRegisterCount, const char* pszVariableName = NULL, 
        unsigned int uiDataSize = 0, unsigned int uiDataStride = 0, 
        const void* pvDataSource = NULL, bool bCopyData = false);
    virtual NiShaderError AddPredefinedEntry(const char* pszKey, 
        unsigned int uiExtra, unsigned int uiShaderRegister, 
        const char* pszVariableName);
    virtual NiShaderError AddAttributeEntry(const char* pszKey, 
        unsigned int uiFlags, unsigned int uiExtra, 
        unsigned int uiShaderRegister, unsigned int uiRegisterCount, 
        const char* pszVariableName, unsigned int uiDataSize, 
        unsigned int uiDataStride, const void* pvDataSource, 
        bool bCopyData = true);
    virtual NiShaderError AddConstantEntry(const char* pszKey, 
        unsigned int uiFlags, unsigned int uiExtra, 
        unsigned int uiShaderRegister, unsigned int uiRegisterCount, 
        const char* pszVariableName, unsigned int uiDataSize, 
        unsigned int uiDataStride, const void* pvDataSource, 
        bool bCopyData = true);
    virtual NiShaderError AddGlobalEntry(const char* pszKey, 
        unsigned int uiFlags, unsigned int uiExtra, 
        unsigned int uiShaderRegister, unsigned int uiRegisterCount, 
        const char* pszVariableName, unsigned int uiDataSize, 
        unsigned int uiDataStride, const void* pvDataSource, 
        bool bCopyData = false);
    virtual NiShaderError AddOperatorEntry(const char* pszKey, 
        unsigned int uiFlags, unsigned int uiExtra, 
        unsigned int uiShaderRegister, unsigned int uiRegisterCount, 
        const char* pszVariableName);
    virtual NiShaderError AddObjectEntry(const char* pszKey,
        unsigned int uiShaderRegister, const char* pszVariableName,
        unsigned int uiObjectIndex,
        NiShaderAttributeDesc::ObjectType eObjectType);

    virtual NiD3DError RemoveEntry(const char* pszKey);
    virtual NiShaderConstantMapEntry* GetEntry(const char* pszKey);
    virtual NiShaderConstantMapEntry* GetEntryAtIndex(
        unsigned int uiIndex);

    //*** Pipeline functions
    virtual NiD3DError SetShaderConstants(
        const NiRenderCallContext& kRCC,
        NiD3DShaderProgram* pkShaderProgram, 
        bool bGlobal);

    // *** begin Emergent internal use only ***
    void SetD3DRenderer(NiD3DRenderer* pkD3DRenderer);
    virtual unsigned int GetEntryIndex(const char* pszKey);

    static void SetupTextureTransformMatrix(D3DXMATRIX& kResult, 
        const NiMatrix3* pkTexMatrix = NULL, bool bInv = false, 

        bool bTrans = false, unsigned int uiTexCoordSize = 4);

    static bool GetDynamicEffectData(
        void* pvData, 
        unsigned int uiDataSize,
        ObjectMappings eMapping, 
        NiDynamicEffect* pkDynEffect,
        const NiRenderCallContext& kRCC);
    // *** end Emergent internal use only ***

protected:
    void SetD3DDevice(D3DDevicePtr pkD3DDevice);
    void SetD3DRenderState(NiD3DRenderState* pkRS);

    virtual NiD3DError InsertEntry(NiShaderConstantMapEntry* pkEntry);
    virtual NiD3DError SetupPredefinedEntry(
        NiShaderConstantMapEntry* pkEntry);
    virtual NiD3DError SetDefinedConstant(
        const NiRenderCallContext& kRCC,
        NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry);
    virtual NiD3DError SetConstantConstant(
        const NiRenderCallContext& kRCC,
        NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry);
    virtual NiD3DError SetAttributeConstant(
        const NiRenderCallContext& kRCC,
        NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry, 
        bool bGlobal,
        NiSCMExtraData* pkSCMExtraData);
    virtual NiD3DError SetGlobalConstant(
        const NiRenderCallContext& kRCC,
        NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry);
    virtual NiD3DError SetOperatorConstant(
        const NiRenderCallContext& kRCC,
        NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry);
    virtual NiD3DError SetupObjectEntry(NiShaderConstantMapEntry* pkEntry);
    virtual NiD3DError SetObjectConstant(
        const NiRenderCallContext& kRCC,
        NiD3DShaderProgram* pkShaderProgram,
        NiShaderConstantMapEntry* pkEntry,
        NiDynamicEffect* pkDynEffect);

    // Attribute Mapping Helper functions
    const void* MapAttributeValue(NiShaderConstantMapEntry* pkEntry,
        NiExtraData* pkExtra);
    const void* MapAttributeArrayValue(NiShaderConstantMapEntry* pkEntry,
        NiExtraData* pkExtra);

    // Constant Mapping Helper functions
    NiShaderError SetShaderConstantArray(NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry, const float * pfData);

    bool ComputeRestridingValues(NiShaderConstantMapEntry* pkEntry, 
        unsigned int & uiRegistersPerElement, 
        unsigned int & uiItemsPerRegister);

    const void* MapConstantValue(NiShaderConstantMapEntry* pkEntry);

    const void* MapConstantArray(NiShaderConstantMapEntry* pkEntry, 
        const float * pfSource);

    static float ms_vTaylorSin[4];
    static float ms_vTaylorCos[4];

    NiTPrimitiveArray<float> m_afArrayMapping;
    static float ms_vMappingValue[4];
    static D3DXMATRIX ms_mMappingValue;

    static BOOL ms_bMappingValue;
    static int ms_aiMappingValue[4];

    NiTObjectArray<NiShaderConstantMapEntryPtr> m_aspEntries;
    bool m_bModified;

    NiD3DShaderProgram* m_pkLastShaderProgram;
    NiD3DError m_eLastError;

    D3DDevicePtr m_pkD3DDevice;
    NiD3DRenderer* m_pkD3DRenderer;
    NiD3DRenderState* m_pkD3DRenderState;

    static D3DXALIGNEDMATRIX ms_kMatrices[2];
    static float ms_afVector4[2][4];
    static D3DXALIGNEDMATRIX ms_kMatrixResult;
    static float ms_fVector4Result[4];

    unsigned int SetupDefinedConstantValue(
        const NiRenderCallContext& kRCC,
        unsigned int uiOperatorNum, 
        NiShaderConstantMapEntry* pkEntry);
    unsigned int SetupConstantConstantValue(
        const NiRenderCallContext& kRCC,
        unsigned int uiOperatorNum, 
        NiShaderConstantMapEntry* pkEntry);
    unsigned int SetupAttributeConstantValue(
        const NiRenderCallContext& kRCC,
        unsigned int uiOperatorNum, 
        NiShaderConstantMapEntry* pkEntry);
    unsigned int SetupGlobalConstantValue(
        const NiRenderCallContext& kRCC,
        unsigned int uiOperatorNum, 
        NiShaderConstantMapEntry* pkEntry);

    NiD3DError PerformOperatorMultiply(NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry,
        NiShaderAttributeDesc::AttributeType eType1, 
        NiShaderAttributeDesc::AttributeType eType2, bool bInverse, 
        bool bTranspose);
    NiD3DError PerformOperatorDivide(NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry,
        NiShaderAttributeDesc::AttributeType eType1, 
        NiShaderAttributeDesc::AttributeType eType2, bool bInverse, 
        bool bTranspose);
    NiD3DError PerformOperatorAdd(NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry,
        NiShaderAttributeDesc::AttributeType eType1, 
        NiShaderAttributeDesc::AttributeType eType2, bool bInverse, 
        bool bTranspose);
    NiD3DError PerformOperatorSubtract(NiD3DShaderProgram* pkShaderProgram, 
        NiShaderConstantMapEntry* pkEntry,
        NiShaderAttributeDesc::AttributeType eType1, 
        NiShaderAttributeDesc::AttributeType eType2, bool bInverse, 
        bool bTranspose);

    static const float ms_fDirLightDistance;
    static float ms_afObjectData[16];
};

typedef NiPointer<NiD3DShaderConstantMap> NiD3DShaderConstantMapPtr;

#include "NiD3DShaderConstantMap.inl"

#endif  //#ifndef NID3DSHADERCONSTANTMAP_H