// 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 NID3DSHADER_H
#define NID3DSHADER_H

#include "NiD3DShaderInterface.h"
#include "NiD3DRenderStateGroup.h"
#include "NiD3DTextureStage.h"
#include "NiD3DPass.h"
#include <NiSCMExtraData.h>
#include <NiTPool.h>
#include <NiTextureEffect.h>

// NiD3DShader
// Shader derived for D3D-based renderers. (DX9/Xenon)

class NID3D_ENTRY NiD3DShader : public NiD3DShaderInterface
{
    NiDeclareRTTI;
public:
    NiD3DShader();
    virtual ~NiD3DShader();

    virtual bool IsInitialized();

    // Override these functions to implement a custom pipeline...
    // Initialize function
    virtual bool Initialize();

    virtual unsigned int PreProcessPipeline(const NiRenderCallContext& kRCC);
    virtual unsigned int UpdatePipeline(const NiRenderCallContext& kRCC);
    virtual unsigned int FirstPass();
    virtual unsigned int SetupRenderingPass(const NiRenderCallContext& kRCC);
    virtual unsigned int SetupTransformations(const NiRenderCallContext& kRCC);
    virtual unsigned int SetupShaderPrograms(const NiRenderCallContext& kRCC);
    virtual unsigned int PreRenderSubmesh(const NiRenderCallContext& kRCC);
    virtual unsigned int PostRenderSubmesh(const NiRenderCallContext& kRCC);
    virtual unsigned int NextPass();
    virtual unsigned int PostProcessPipeline(const NiRenderCallContext& kRCC);

    // Used during UpdatePipeline
    virtual void ResetPasses();

    static void InitializePools();
    static void ShutdownPools();

    // This function is called by NiMaterialInstance::GetCurrentShader, and
    // is used to verify the geometry is usable for the shader, as well
    // as to 'process' the geometry, adding what is needed.
    virtual bool SetupGeometry(NiRenderObject* pkGeometry, 
        NiMaterialInstance* pkMaterialInstance);

    // This function sets up an NiSCMExtraData object on the geometry that
    // caches extra data lookups for shaders. It is declared static so that
    // applications can call it if necessary when removing and adding extra
    // data.
    static void SetupSCMExtraData(NiD3DShader* pkD3DShader,
        NiRenderObject* pkGeometry);
    inline void ResetSCMExtraData(NiRenderObject* pkGeometry);

    // Allow for complete shader reconstruction
    virtual void DestroyRendererData();
    virtual void RecreateRendererData();

    // Allows for the shader to use the
    // NiD3DRenderState::UpdateRenderState function
    inline bool GetUsesNiRenderState() const;
    virtual bool SetUsesNiRenderState(bool bUses);

    // Allows for the shader to use the Ni###LightManager::SetState
    // function to set lights on the device
    inline bool GetUsesNiLightState() const;
    inline void SetUsesNiLightState(bool bUses);

    inline bool GetConvertBlendIndicesToD3DColor() const;
    inline void SetConvertBlendIndicesToD3DColor(bool bConvert);

    // Query shader for presence of shader programs
    virtual bool GetVSPresentAllPasses() const;
    virtual bool GetVSPresentAnyPass() const;
    virtual bool GetPSPresentAllPasses() const;
    virtual bool GetPSPresentAnyPass() const;

    inline void UseWorldSpaceSphereMaps(bool bWorldSpace);

    // *** begin Emergent internal use only

    static void _SDMInit();
    static void _SDMShutdown();

    // *** end Emergent internal use only

    static NiFixedString ms_kNDLShaderMapName;

protected:
    // Projected texture transform packing functions
    static void PackDynamicEffect(
        const NiTextureEffect* pkTextureEffect,
        NiD3DTextureStage* pkStage, 
        bool bSave, 
        NiD3DRenderer* pkD3DRenderer);

    typedef void (*DynamicEffectPacker)(
        const NiMatrix3& kWorldMat, 
        const NiPoint3& kWorldTrans, 
        NiD3DTextureStage* pkStage, 
        bool bSave, 
        NiD3DRenderer* pkD3DRenderer);

    static void PackWorldParallelEffect(
        const NiMatrix3& kWorldMat,
        const NiPoint3& kWorldTrans,
        NiD3DTextureStage* pkStage,
        bool bSave,
        NiD3DRenderer* pkD3DRenderer);
    static void PackWorldPerspectiveEffect(
        const NiMatrix3& kWorldMat,
        const NiPoint3& kWorldTrans,
        NiD3DTextureStage* pkStage,
        bool bSave,
        NiD3DRenderer* pkD3DRenderer);
    static void PackWorldSphereEffect(
        const NiMatrix3& kWorldMat,
        const NiPoint3& kWorldTrans,
        NiD3DTextureStage* pkStage,
        bool bSave,
        NiD3DRenderer* pkD3DRenderer);
    static void PackCameraSphereEffect(
        const NiMatrix3& kWorldMat,
        const NiPoint3& kWorldTrans,
        NiD3DTextureStage* pkStage,
        bool bSave,
        NiD3DRenderer* pkD3DRenderer);
    static void PackSpecularCubeEffect(
        const NiMatrix3& kWorldMat,
        const NiPoint3& kWorldTrans,
        NiD3DTextureStage* pkStage,
        bool bSave,
        NiD3DRenderer* pkD3DRenderer);
    static void PackDiffuseCubeEffect(
        const NiMatrix3& kWorldMat,
        const NiPoint3& kWorldTrans,
        NiD3DTextureStage* pkStage,
        bool bSave,
        NiD3DRenderer* pkD3DRenderer);

    static DynamicEffectPacker ms_apfnDynEffectPackers[
        NiTextureEffect::NUM_COORD_GEN];

    bool m_bUsesNiRenderState;
    bool m_bUsesNiLightState;
    bool m_bConvertBlendIndicesToD3DColor;

    // 'Global' render state settings for the shader
    NiD3DRenderStateGroup* m_pkRenderStateGroup;
    //  'Global' pixel shader mapping for the shader
    NiD3DShaderConstantMapPtr m_spPixelConstantMap;
    //  'Global' vertex shader mapping for the shader
    NiD3DShaderConstantMapPtr m_spVertexConstantMap;

    // Passes
    unsigned int m_uiCurrentPass;
    unsigned int m_uiPassCount;
    NiD3DPassPtr m_spCurrentPass;
    NiTObjectArray<NiD3DPassPtr> m_kPasses;
};

typedef NiPointer<NiD3DShader> NiD3DShaderPtr;

#include "NiD3DShader.inl"

#endif  //#ifndef NID3DSHADER_H