// 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-2007 Emergent Game Technologies.
//      All Rights Reserved.
//
// Emergent Game Technologies, Chapel Hill, North Carolina 27517
// http://www.emergent.net

// Precompiled Header
#include "NiD3DRendererPCH.h"

#include "NiDX9ErrorShader.h"

NiImplementRTTI(NiDX9ErrorShader, NiD3DShader);

//---------------------------------------------------------------------------
NiDX9ErrorShader::NiDX9ErrorShader()
{
    //  Create the local stages and passes we will use...
    CreateStagesAndPasses();

    SetName("NiDX9ErrorShader");

    // This is the best (and only) implementation of this shader
    m_bIsBestImplementation = true;
}
//---------------------------------------------------------------------------
NiDX9ErrorShader::~NiDX9ErrorShader()
{
    m_kPasses.RemoveAll();
}
//---------------------------------------------------------------------------
bool NiDX9ErrorShader::Initialize()
{
    if (m_bInitialized)
        return true;

    if (NiD3DShader::Initialize())
    {
        CreateShaderDeclaration();
        return true;
    }

    return false;
}
//---------------------------------------------------------------------------
unsigned int NiDX9ErrorShader::UpdatePipeline(
    NiGeometry* pkGeometry, const NiSkinInstance* pkSkin, 
    NiGeometryData::RendererData* pkRendererData, 
    const NiPropertyState* pkState, const NiDynamicEffectState* pkEffects,
    const NiTransform& kWorld, const NiBound& kWorldBound)
{
    // Create a new, bright random color
    unsigned char ucR = (unsigned int)(NiUnitRandom() * 127) + 128;
    unsigned char ucG = (unsigned int)(NiUnitRandom() * 127) + 128;
    unsigned char ucB = (unsigned int)(NiUnitRandom() * 127) + 128;

    D3DCOLOR kColor = D3DCOLOR_ARGB(255, ucR, ucG, ucB);
   
    m_pkPass->SetRenderState(D3DRS_TEXTUREFACTOR, kColor, true);

    m_pkPass->SetRenderState(D3DRS_SPECULARENABLE, false, true);
    m_pkPass->SetRenderState(D3DRS_ALPHATESTENABLE, false, true);
    m_pkPass->SetRenderState(D3DRS_ALPHABLENDENABLE, false, true);
    m_pkPass->SetRenderState(D3DRS_STENCILENABLE, false, true);

    return NiD3DShader::UpdatePipeline(pkGeometry, pkSkin, pkRendererData, 
        pkState, pkEffects, kWorld, kWorldBound);
}
//---------------------------------------------------------------------------
unsigned int NiDX9ErrorShader::SetupTransformations(NiGeometry* pkGeometry, 
    const NiSkinInstance* pkSkin, 
    const NiSkinPartition::Partition* pkPartition, 
    NiGeometryData::RendererData* pkRendererData, 
    const NiPropertyState* pkState, const NiDynamicEffectState* pkEffects, 
    const NiTransform& kWorld, const NiBound& kWorldBound)
{
    // Add random jitter to world transform
    NiTransform kNewWorld = kWorld;

    float fBound = kWorldBound.GetRadius();
    float fRange = 0.1f * fBound;
    kNewWorld.m_Translate.x += NiSymmetricRandom() * fRange;
    kNewWorld.m_Translate.y += NiSymmetricRandom() * fRange;
    kNewWorld.m_Translate.z += NiSymmetricRandom() * fRange;

    // Pass NULL for partition so, even if the object itself is skinned,
    // it will be rendered as unskinned.
    return NiD3DShader::SetupTransformations(pkGeometry, NULL, NULL,
        pkRendererData, pkState, pkEffects, kNewWorld, kWorldBound);
}
//---------------------------------------------------------------------------
bool NiDX9ErrorShader::CreateStagesAndPasses()
{
    NiD3DPassPtr spPass = NiD3DPass::CreateNewPass();
    NIASSERT(spPass);
    m_pkPass = spPass;

    NiD3DTextureStagePtr spStage = NiD3DTextureStage::CreateNewStage();
    NIASSERT(spStage);
    NiD3DTextureStage* pkStage = spStage;

    pkStage->SetStage(0);
    pkStage->SetTexture(0);
    pkStage->SetStageState(D3DTSS_COLOROP, D3DTOP_SELECTARG1);
    pkStage->SetStageState(D3DTSS_COLORARG1, D3DTA_TFACTOR);

    pkStage->SetStageState(D3DTSS_ALPHAOP, D3DTOP_SELECTARG1);
    pkStage->SetStageState(D3DTSS_ALPHAARG1, D3DTA_TFACTOR);
    pkStage->SetStageState(D3DTSS_TEXCOORDINDEX, 0);

    pkStage->SetSamplerState(NiD3DRenderState::NISAMP_ADDRESSU, 
        D3DTADDRESS_CLAMP);
    pkStage->SetSamplerState(NiD3DRenderState::NISAMP_ADDRESSV, 
        D3DTADDRESS_CLAMP);

    pkStage->SetSamplerState(NiD3DRenderState::NISAMP_MAGFILTER, 
        D3DTEXF_LINEAR);
    pkStage->SetSamplerState(NiD3DRenderState::NISAMP_MINFILTER, 
        D3DTEXF_LINEAR);
    pkStage->SetSamplerState(NiD3DRenderState::NISAMP_MIPFILTER, 
        D3DTEXF_NONE);

    pkStage->SetStageState(D3DTSS_TEXTURETRANSFORMFLAGS, D3DTTFF_DISABLE);

    spPass->AppendStage(pkStage);
    m_kPasses.SetAt(0, spPass);
    m_uiPassCount = 1;

    return true;
}
//---------------------------------------------------------------------------
bool NiDX9ErrorShader::CreateShaderDeclaration()
{
    m_spShaderDecl = NiDX9ShaderDeclaration::Create(m_pkD3DRenderer, 1);
    m_spShaderDecl->SetEntry(0,
        NiShaderDeclaration::SHADERPARAM_NI_POSITION,
        NiShaderDeclaration::SPTYPE_FLOAT3, 0);
    return true;
}
//---------------------------------------------------------------------------