/*---------------------------------------------------------------------------*
  Project:  NintendoWare
  File:     gfx_ShaderProgram.cpp
  
  Copyright (C)2009-2011 Nintendo/HAL Laboratory, Inc.  All rights reserved.
  
  These coded instructions, statements, and computer programs contain proprietary
  information of Nintendo and/or its licensed developers and are protected by
  national and international copyright laws. They may not be disclosed to third
  parties or copied or duplicated in any form, in whole or in part, without the
  prior written consent of Nintendo.
  
  The content herein is highly confidential and should be handled accordingly.
  
  $Revision: 31311 $
 *---------------------------------------------------------------------------*/

#include "precompiled.h"

#include <nw/gfx/gfx_ShaderProgram.h>

#include <nw/os/os_Memory.h>
#include <nw/ut/ut_Foreach.h>
#include <nw/gfx/gfx_ShaderBinaryInfo.h>

namespace nw
{
namespace gfx
{

//----------------------------------------
ShaderProgram::ShaderProgram(os::IAllocator* allocator)
: GfxObject(allocator),
#if defined(NW_GFX_PROGRAM_OBJECT_ENABLED)
  m_UniformLocation(NULL),
#endif
  m_ProgramObject(0),
  m_GeometryShaderMode(GEOMETRY_SHADER_MODE_UNINITIALIZED)
{
    const u32 REG_VERTEX_UNIFORM_BOOL = 0x2B0;
    const u32 REG_GEOMETRY_UNIFORM_BOOL = 0x280;
    const u32 REG_GEOMETRY_DISABLE = 0x244;
    
    const u32 HEADER_VERTEX   = internal::MakeCommandHeader(REG_VERTEX_UNIFORM_BOOL, 5, true, 0xF);
    const u32 HEADER_GEOMETRY = internal::MakeCommandHeader(REG_GEOMETRY_UNIFORM_BOOL, 5, true, 0xF);
    const u32 HEADER_DISABLE_GEOMETRY = internal::MakeCommandHeader(REG_GEOMETRY_DISABLE, 1, false, 0x1);
    
    m_VertexIntUniforms[0] = 0x7fff0000; // b0-b15
    m_VertexIntUniforms[1] = HEADER_VERTEX,
    m_VertexIntUniforms[2] = 0x00000000; // i0
    m_VertexIntUniforms[3] = 0x00000000; // i1
    m_VertexIntUniforms[4] = 0x00000000; // i2
    m_VertexIntUniforms[5] = 0x00000000; // i3

    m_GeometryIntUniforms[0] = 0x7fff0000; // b0-b15
    m_GeometryIntUniforms[1] = HEADER_GEOMETRY,
    m_GeometryIntUniforms[2] = 0x00000000; // i0
    m_GeometryIntUniforms[3] = 0x00000000; // i1
    m_GeometryIntUniforms[4] = 0x00000000; // i2
    m_GeometryIntUniforms[5] = 0x00000000; // i3
    
    m_DisableGeometry[0] = 0;
    m_DisableGeometry[1] = HEADER_DISABLE_GEOMETRY;
}

//----------------------------------------
ShaderProgram::~ShaderProgram()
{
}

//----------------------------------------
void
ShaderProgram::ActivateDescription(ResShaderProgramDescription shaderProgramDescription)
{
    // まずジオメトリシェーダのモード切替。
    ResBinaryShader shader = ResBinaryShader( shaderProgramDescription.GetOwnerShaderData() );
    bool hasGeometry = (shaderProgramDescription.GetGeometryShaderIndex() >= 0);
    
    bool skipShaderBinary = false;
    bool skipShaderMode   = false;
    
    if ( m_Description.IsValid() )
    {
        bool hasGeometryPrev = (m_Description.GetGeometryShaderIndex() >= 0);
        if ( hasGeometry == hasGeometryPrev )
        {
            skipShaderMode = true;
        }
        
        ResBinaryShader shaderPrev = ResBinaryShader( m_Description.GetOwnerShaderData() );
        if ( shader == shaderPrev )
        {
            skipShaderBinary = true;
        }
    }
    
    if ( ! skipShaderMode )
    {
        ActivateShaderProgramMode( hasGeometry );
    }
    
    if ( ! skipShaderBinary )
    {
        NW_NULL_ASSERT( shader.ref().m_CommandCache );
        internal::NWUseCmdlist( shader.ref().m_CommandCache, shader.ref().m_CommandCacheSize );
    }
    
    NW_NULL_ASSERT( shaderProgramDescription.ref().m_CommandCache );
    internal::NWUseCmdlist( shaderProgramDescription.ref().m_CommandCache, shaderProgramDescription.ref().m_CommandCacheSize );
    
#if defined(NW_GFX_PROGRAM_OBJECT_ENABLED)
    m_UniformLocation = static_cast<ShaderUniformLocation*>(shaderProgramDescription.GetUniformLocation());
    m_ProgramObject = shaderProgramDescription.GetProgramObject();
#endif
    
    const ShaderBinaryInfo* shaderBinaryInfo = shader.GetShaderBinaryInfo();

    m_VertexIntUniforms[0]       |= shaderBinaryInfo->GetBoolConstant( shaderProgramDescription.GetVertexShaderIndex() );
    
    if ( hasGeometry )
    {
        m_GeometryIntUniforms[0] |= shaderBinaryInfo->GetBoolConstant( shaderProgramDescription.GetGeometryShaderIndex() );
    }
    
    m_Description = shaderProgramDescription;
}

//----------------------------------------
void
ShaderProgram::ActivateShaderProgramMode(bool useGeometry)
{
    nw::gfx::internal::SetupShaderProgramMode( useGeometry );
    
    m_GeometryShaderMode = useGeometry ? GEOMETRY_SHADER_MODE_USED : GEOMETRY_SHADER_MODE_NONE;
    m_DisableGeometry[0] = useGeometry ? 1 : 0;
}

} // namespace gfx
} // namespace nw