/*
-----------------------------------------------------------------------------
This source file is part of OGRE
    (Object-oriented Graphics Rendering Engine)
For the latest info, see http://www.ogre3d.org/

Copyright (c) 2000-2005 The OGRE Team
Also see acknowledgements in Readme.html

This program is free software you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the Free Software
Foundation either version 2 of the License, or (at your option) any later
version.

This program is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License along with
this program if not, write to the Free Software Foundation, Inc., 59 Temple
Place - Suite 330, Boston, MA 02111-1307, USA, or go to
http://www.gnu.org/copyleft/lesser.txt.
-----------------------------------------------------------------------------
*/
#include "OgreD3D9HLSLProgram.h"
#include "OgreGpuProgramManager.h"
#include "OgreStringConverter.h"
#include "OgreD3D9GpuProgram.h"
#include "OgreGpuProgram.h"
#include "OgreRoot.h"
#include "OgreRenderSystem.h"
#include "OgreRenderSystemCapabilities.h"

#ifdef GAMETOOLS_ILLUMINATION_MODULE
#include "OgreFileSystem.h"
#endif

namespace Ogre {
    //-----------------------------------------------------------------------
    D3D9HLSLProgram::CmdEntryPoint D3D9HLSLProgram::msCmdEntryPoint;
    D3D9HLSLProgram::CmdTarget D3D9HLSLProgram::msCmdTarget;
#ifdef GAMETOOLS_ILLUMINATION_MODULE
	D3D9HLSLProgram::CmdOptimalization D3D9HLSLProgram::msCmdOptimalization;
	D3D9HLSLProgram::CmdFlowControl D3D9HLSLProgram::msCmdFlowControl;
#endif
    //-----------------------------------------------------------------------
    //-----------------------------------------------------------------------
    void D3D9HLSLProgram::loadFromSource(void)
    {
        LPD3DXBUFFER errors = 0;
		
		#ifdef GAMETOOLS_ILLUMINATION_MODULE
			mCompilerFlags = NULL;
			
			if(!mOptimalization)
				mCompilerFlags = D3DXSHADER_SKIPOPTIMIZATION;
			if(mFlowControl == 1)
				mCompilerFlags = mCompilerFlags|D3DXSHADER_PREFER_FLOW_CONTROL;
			if(mFlowControl == -1)
				mCompilerFlags = mCompilerFlags|D3DXSHADER_AVOID_FLOW_CONTROL;

			HRESULT hr = D3DXCompileShader(
				mSource.c_str(),
				static_cast<UINT>(mSource.length()),
				NULL, //no preprocessor defines
				NULL, //no includes
				mEntryPoint.c_str(),
				mTarget.c_str(),
				mCompilerFlags, 
				&mpMicroCode,
				&errors,
				&mpConstTable);
		#else
			// Compile & assemble into microcode
			HRESULT hr = D3DXCompileShader(
				mSource.c_str(),
				static_cast<UINT>(mSource.length()),
				NULL, //no preprocessor defines
				NULL, //no includes
				mEntryPoint.c_str(),
				mTarget.c_str(),
				NULL, // no compile flags
				&mpMicroCode,
				&errors,
				&mpConstTable);
		#endif

        if (FAILED(hr))
        {
            String message = "Cannot assemble D3D9 high-level shader " + mName + " Errors:\n" +
                static_cast<const char*>(errors->GetBufferPointer());
            errors->Release();
            OGRE_EXCEPT(hr, message,
                "D3D9HLSLProgram::loadFromSource");
        }


    }	
#ifdef GAMETOOLS_ILLUMINATION_MODULE
	//-----------------------------------------------------------------------
	void D3D9HLSLProgram::loadHighLevelImpl(void)
    {
        if (mLoadFromFile)
        	loadFromFile();
        else
			loadFromSource();        
    }
    void D3D9HLSLProgram::loadFromFile(void)
	{
		DataStreamPtr stream = 
                ResourceGroupManager::getSingleton().openResource(
                    mFilename, mGroup, true, this);

		mSource = "";

		while(!stream->eof())
		{
			String line = stream->getLine();
			if(line.find("#include")!= line.npos)
			{
				int firstocc = line.find("\"",0);
				int secondocc = line.find("\"",firstocc+1);
				String filename = line.substr(firstocc+1,secondocc - firstocc - 1);

				DataStreamPtr includedstream = ResourceGroupManager::getSingleton().openResource(
                    filename, mGroup, true, 0);
				while(!includedstream->eof())
				{
					line = includedstream->getLine();
					mSource.append(line);
				}
			}
			else
				mSource.append(line);
			mSource.append("\n");
		}
//		mSource.append(0);
	//	mSource = stream->getAsString();
		loadFromSource();
		/*
		 LPD3DXBUFFER errors = 0;
		
			mCompilerFlags = NULL;
			
			if(!mOptimalization)
				mCompilerFlags = D3DXSHADER_SKIPOPTIMIZATION;
			if(mFlowControl == 1)
				mCompilerFlags = mCompilerFlags|D3DXSHADER_PREFER_FLOW_CONTROL;
			if(mFlowControl == -1)
				mCompilerFlags = mCompilerFlags|D3DXSHADER_AVOID_FLOW_CONTROL;
			
			FileInfoListPtr fileInfos = ResourceGroupManager::getSingleton().findResourceFileInfo(mGroup, mFilename);
			FileInfo FI = fileInfos->at(0);
			String fullPathName = ((FileSystemArchive*)FI.archive)->getTmpPath()+"\\"+
								((FileSystemArchive*)FI.archive)->getBasePath() +"\\"+
								mFilename;
		    String Fname2 = "C:\GameToolsSVN\Ogre\Ogre1.2\Samples\Media\DeferredShadingMedia\DeferredShading\material\hlsl\vs.hlsl";
			HRESULT hr = D3DXCompileShaderFromFile(
				fullPathName.c_str(),
				NULL, //no preprocessor defines
				NULL, //no includes
				mEntryPoint.c_str(),
				mTarget.c_str(),
				mCompilerFlags, 
				&mpMicroCode,
				&errors,
				&mpConstTable);		

        if (FAILED(hr))
        {
            String message = "Cannot assemble D3D9 high-level shader " + mName + " Errors:\n" +
                static_cast<const char*>(errors->GetBufferPointer());
            errors->Release();
            OGRE_EXCEPT(hr, message,
                "D3D9HLSLProgram::loadFromSource");
        }*/
	}
#endif
    //-----------------------------------------------------------------------
    void D3D9HLSLProgram::createLowLevelImpl(void)
    {
        // Create a low-level program, give it the same name as us
        mAssemblerProgram = 
            GpuProgramManager::getSingleton().createProgramFromString(
                mName, 
                mGroup,
                "",// dummy source, since we'll be using microcode
                mType, 
                mTarget);
        static_cast<D3D9GpuProgram*>(mAssemblerProgram.get())->setExternalMicrocode(mpMicroCode);

    }
    //-----------------------------------------------------------------------
    void D3D9HLSLProgram::unloadHighLevelImpl(void)
    {
        SAFE_RELEASE(mpMicroCode);
        SAFE_RELEASE(mpConstTable);

    }
    //-----------------------------------------------------------------------
    void D3D9HLSLProgram::populateParameterNames(GpuProgramParametersSharedPtr params)
    {
        // Derive parameter names from const table
        assert(mpConstTable && "Program not loaded!");
        // Get contents of the constant table
        D3DXCONSTANTTABLE_DESC desc;
        HRESULT hr = mpConstTable->GetDesc(&desc);

        if (FAILED(hr))
        {
            OGRE_EXCEPT(Exception::ERR_INTERNAL_ERROR, 
                "Cannot retrieve constant descriptions from HLSL program.", 
                "D3D9HLSLProgram::populateParameterNames");
        }
        // Iterate over the constants
        for (unsigned int i = 0; i < desc.Constants; ++i)
        {
            // Recursively descend through the structure levels
            // Since D3D9 has no nice 'leaf' method like Cg (sigh)
            processParamElement(NULL, "", i, params);
        }

        
    }
    //-----------------------------------------------------------------------
    void D3D9HLSLProgram::processParamElement(D3DXHANDLE parent, String prefix, 
        unsigned int index, GpuProgramParametersSharedPtr params)
    {
        D3DXHANDLE hConstant = mpConstTable->GetConstant(parent, index);

        // Since D3D HLSL doesn't deal with naming of array and struct parameters
        // automatically, we have to do it by hand

        D3DXCONSTANT_DESC desc;
        unsigned int numParams = 1;
        HRESULT hr = mpConstTable->GetConstantDesc(hConstant, &desc, &numParams);
        if (FAILED(hr))
        {
            OGRE_EXCEPT(Exception::ERR_INTERNAL_ERROR, 
                "Cannot retrieve constant description from HLSL program.", 
                "D3D9HLSLProgram::processParamElement");
        }

        String paramName = desc.Name;
        // trim the odd '$' which appears at the start of the names in HLSL
        if (paramName.at(0) == '$')
            paramName.erase(paramName.begin());

        // If it's an array, elements will be > 1
        for (unsigned int e = 0; e < desc.Elements; ++e)
        {
            if (desc.Class == D3DXPC_STRUCT)
            {
                // work out a new prefix for nested members, if it's an array, we need an index
                if (desc.Elements > 1)
                    prefix = prefix + paramName + "[" + StringConverter::toString(e) + "].";
                else
                    prefix = prefix + paramName + ".";
                // Cascade into struct
                for (unsigned int i = 0; i < desc.StructMembers; ++i)
                {
                    processParamElement(hConstant, prefix, i, params);
                }
            }
            else
            {
                // Process params
                if (desc.Type == D3DXPT_FLOAT || desc.Type == D3DXPT_INT || desc.Type == D3DXPT_BOOL)
                {
                    size_t paramIndex = desc.RegisterIndex;
                    String name = prefix + paramName;
                    // If this is an array, need to append element index
                    if (desc.Elements > 1)
                        name += "[" + StringConverter::toString(e) + "]";
                    
                    params->_mapParameterNameToIndex(name, paramIndex);
                    // setup constant definition
                    // is it float or int
                    GpuProgramParameters::ElementType elementType = GpuProgramParameters::ET_INT;
                    if (desc.Type == D3DXPT_FLOAT)
                        elementType = GpuProgramParameters::ET_REAL;
                    params->addConstantDefinition(name, paramIndex, 0, elementType);
                }
            }
        }
            
    }
    //-----------------------------------------------------------------------
    D3D9HLSLProgram::D3D9HLSLProgram(ResourceManager* creator, const String& name, 
        ResourceHandle handle, const String& group, bool isManual, 
        ManualResourceLoader* loader)
        : HighLevelGpuProgram(creator, name, handle, group, isManual, loader)
        , mpMicroCode(NULL), mpConstTable(NULL)
    {
		#ifdef GAMETOOLS_ILLUMINATION_MODULE
			mOptimalization = true;
			mFlowControl = 0;
			mCompilerFlags = NULL;
		#endif

        if (createParamDictionary("D3D9HLSLProgram"))
        {
            setupBaseParamDictionary();
            ParamDictionary* dict = getParamDictionary();

            dict->addParameter(ParameterDef("entry_point", 
                "The entry point for the HLSL program.",
                PT_STRING),&msCmdEntryPoint);
            dict->addParameter(ParameterDef("target", 
                "Name of the assembler target to compile down to.",
                PT_STRING),&msCmdTarget);
			#ifdef GAMETOOLS_ILLUMINATION_MODULE
				dict->addParameter(ParameterDef("optimalization", 
					"Boolean to turn optimalization on/off",
					PT_STRING),&msCmdOptimalization);
				dict->addParameter(ParameterDef("flow_control", 
					"Flow control: default, prefer, avoid.",
					PT_STRING),&msCmdFlowControl);
			#endif
        }
        
    }
    //-----------------------------------------------------------------------
    D3D9HLSLProgram::~D3D9HLSLProgram()
    {
        // have to call this here reather than in Resource destructor
        // since calling virtual methods in base destructors causes crash
        if (mIsLoaded)
        {
            unload();
        }
        else
        {
            unloadHighLevel();
        }
    }
    //-----------------------------------------------------------------------
    bool D3D9HLSLProgram::isSupported(void) const
    {
		// If skeletal animation is being done, we need support for UBYTE4
		if (isSkeletalAnimationIncluded() && 
			!Root::getSingleton().getRenderSystem()->getCapabilities()
				->hasCapability(RSC_VERTEX_FORMAT_UBYTE4))
		{
			return false;
		}

        return GpuProgramManager::getSingleton().isSyntaxSupported(mTarget);
    }
    //-----------------------------------------------------------------------
    GpuProgramParametersSharedPtr D3D9HLSLProgram::createParameters(void)
    {
        // Call superclass
        GpuProgramParametersSharedPtr params = HighLevelGpuProgram::createParameters();

        // D3D HLSL uses column-major matrices
        params->setTransposeMatrices(true);

        return params;
    }
    //-----------------------------------------------------------------------
    void D3D9HLSLProgram::setTarget(const String& target)
    {
        mTarget = target;
    }

    //-----------------------------------------------------------------------
    const String& D3D9HLSLProgram::getLanguage(void) const
    {
        static const String language = "hlsl";

        return language;
    }

	#ifdef GAMETOOLS_ILLUMINATION_MODULE
		void D3D9HLSLProgram::setOptimalization(const String& optimalization)
		{
			mOptimalization = StringConverter::parseBool(optimalization);			
		}
		const String& D3D9HLSLProgram::getOptimalization() const
		{
			return StringConverter::toString(mOptimalization);
		}
		void D3D9HLSLProgram::setFlowControl(const String& control)
		{
			if(control == "default")
				mFlowControl = 0;
			else if(control == "avoid")
				mFlowControl = -1;
			else if(control == "prefer")
				mFlowControl = 1;	
		}
		const String& D3D9HLSLProgram::getFlowControl() const
		{
			static const String def = "default";
			static const String pref = "prefer";
			static const String av = "avoid";
			if(mFlowControl == 0)
				return def;
			if(mFlowControl == -1)
				return av;
			if(mFlowControl == 1)
				return pref;		
		}
	#endif

    //-----------------------------------------------------------------------
    //-----------------------------------------------------------------------
    String D3D9HLSLProgram::CmdEntryPoint::doGet(const void *target) const
    {
        return static_cast<const D3D9HLSLProgram*>(target)->getEntryPoint();
    }
    void D3D9HLSLProgram::CmdEntryPoint::doSet(void *target, const String& val)
    {
        static_cast<D3D9HLSLProgram*>(target)->setEntryPoint(val);
    }
    //-----------------------------------------------------------------------
    String D3D9HLSLProgram::CmdTarget::doGet(const void *target) const
    {
        return static_cast<const D3D9HLSLProgram*>(target)->getTarget();
    }
    void D3D9HLSLProgram::CmdTarget::doSet(void *target, const String& val)
    {
        static_cast<D3D9HLSLProgram*>(target)->setTarget(val);
    }

	#ifdef GAMETOOLS_ILLUMINATION_MODULE
		String D3D9HLSLProgram::CmdOptimalization::doGet(const void *target) const
		{
			return static_cast<const D3D9HLSLProgram*>(target)->getOptimalization();
		}
		void D3D9HLSLProgram::CmdOptimalization::doSet(void *target, const String& val)
		{
			static_cast<D3D9HLSLProgram*>(target)->setOptimalization(val);
		}
		String D3D9HLSLProgram::CmdFlowControl::doGet(const void *target) const
		{
			return static_cast<const D3D9HLSLProgram*>(target)->getFlowControl();
		}
		void D3D9HLSLProgram::CmdFlowControl::doSet(void *target, const String& val)
		{
			static_cast<D3D9HLSLProgram*>(target)->setFlowControl(val);
		}
	#endif

}