/*
-----------------------------------------------------------------------------
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

You may use this sample code for anything you like, it is not covered by the
LGPL like the rest of the engine.
-----------------------------------------------------------------------------
*/

/**
    \file 
        Lighting.h
    \brief
        Shows lighting support in OGRE. Also demonstrates the use of billboards
        and controllers for automatic time-relative behaviour.
*/


#include "ExampleApplication.h"
#include "OgreIlluminationManager.h"
#include "SceneSerializer.h"

bool capture = false;
SceneNode* CharacterGroupOrigo;
AnimationState* mAnimState = 0;

bool PP_glowON = false;
bool PP_tonemapON = false;
bool PP_dofON = false;

Real lightpower = 32.0;
Real glowCut = 50.6;
Real glowTimeBlur = 0.6;
Real glowStrech = 1.8;
Real glowCombine = 0.5;
// Listener class for frame updates
class MoriaDemoFrameListener : public ExampleFrameListener
{
protected:
	
	unsigned long framecount;
	SceneNode* mainLightNode;
	Light* light;
	SceneNode* sysNode;
	
public:
    MoriaDemoFrameListener(RenderWindow* window, Camera* maincam, SceneNode* lightNode, Light* l, SceneNode* sysNode)
        :ExampleFrameListener(window, maincam) 
				 
    {
		mainLightNode = lightNode;
		light = l;
		framecount = 0;	
		mMoveSpeed /= 5.0;
		this->sysNode = sysNode;

		mStatsOn = false;
		showDebugOverlay(mStatsOn);

    }

    bool frameStarted(const FrameEvent& evt)
    {	
		if(framecount == 20)//some initialization problem
		{
			PP_tonemapON = true;
			PP_glowON = true;
			PP_dofON = true;
		    CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
															"GTP/Moria/PostProc/Glow", true);
			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
															"GTP/Moria/PostProc/ToneMap", true);
			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
															"GTP/Moria/PostProc/DoF", true);
		}
		
		float thisTime = (float) Root::getSingleton().getTimer()->getMilliseconds() / 2000.0f;
		thisTime= 0;
		//mainLightNode->setPosition(Vector3(2*cos(thisTime), 2, 2 * sin(thisTime)) + Vector3(5, 0, 5));
		Vector3 pos = light->getDerivedPosition();
		pos.normalise();
		pos *= -1;
		//light->setDirection( pos);
		
		framecount++;
	  
//	    mAnimState->addTime(evt.timeSinceLastFrame * 1.0);  
		

		//sysNode->setPosition(mCamera->getPosition() - Vector3(0,1.5,0));
		//sysNode->setPosition(FPSPlayer::thePlayer->getPosition() - Vector3(0,1.5,0));

        return ExampleFrameListener::frameStarted(evt);     
    }

	bool processUnbufferedKeyInput(const FrameEvent& evt)
    {
		if(capture)
		{
			if (mInputDevice->isKeyDown(KC_F2) && mTimeUntilNextToggle <= 0)
			{
				TexturePtr tex = TextureManager::getSingleton().getByName("SCREENSHOT");
				for(int i = 0; i< 30; i++)
					tex->getBuffer()->getRenderTarget()->update();
				tex->getBuffer()->getRenderTarget()->writeContentsToFile(String("screen") + StringConverter::toString(framecount) + ".tga");
				mTimeUntilNextToggle = 0.5;            
			}
		}
		if (mInputDevice->isKeyDown(KC_F3) && mTimeUntilNextToggle <= 0)
		{
				TexturePtr tex = TextureManager::getSingleton().getByName("Troll_SE_0_COLORCUBEMAP");
				for (int i = 0; i < 6; i++)
					tex->getBuffer(i)->getRenderTarget()->writeContentsToFile(String("face") + StringConverter::toString(i) + ".tga");
				mTimeUntilNextToggle = 0.5;            
		}
			

		if (mInputDevice->isKeyDown(KC_ADD) && mTimeUntilNextToggle <= 0)
		{
			if (mInputDevice->isKeyDown(KC_RSHIFT))
				lightpower /= 1.2;
			else
				lightpower *= 1.2;
			light->setPowerScale(lightpower);
			mTimeUntilNextToggle = 0.2;
				mWindow->setDebugText(String("LightPower ") + StringConverter::toString(lightpower));
		}
		if (mInputDevice->isKeyDown(KC_SUBTRACT) && mTimeUntilNextToggle <= 0)
		{
			if (mInputDevice->isKeyDown(KC_RSHIFT))
				CharacterGroupOrigo->translate(0,0,-evt.timeSinceLastFrame);
			else
				CharacterGroupOrigo->translate(0,0,evt.timeSinceLastFrame);			
		}
		if (mInputDevice->isKeyDown(KC_MULTIPLY) && mTimeUntilNextToggle <= 0)
		{
			if (mInputDevice->isKeyDown(KC_RSHIFT))
				glowTimeBlur /= 1.2;
			else
				glowTimeBlur *= 1.2;
			Ogre::Material* m = (Ogre::Material*) MaterialManager::getSingleton().getByName("GTP/Moria/PostProc/GlowCut").getPointer();
			m->getTechnique(0)->getPass(0)->getFragmentProgramParameters()->setNamedConstant("timeBlur", glowTimeBlur);
			mTimeUntilNextToggle = 0.2;
			mWindow->setDebugText(String("GlowTimeBlur ") + StringConverter::toString(glowTimeBlur));
		}
		if (mInputDevice->isKeyDown(KC_M))
		{
			if (mInputDevice->isKeyDown(KC_RSHIFT))
				glowStrech -= 0.01;
			else
				glowStrech += 0.01;
			Ogre::Material* m = (Ogre::Material*) MaterialManager::getSingleton().getByName("GTP/Moria/PostProc/GlowBlurH").getPointer();
			m->getTechnique(0)->getPass(0)->getFragmentProgramParameters()->setNamedConstant("Stretch", glowStrech);
			m = (Ogre::Material*) MaterialManager::getSingleton().getByName("GTP/Moria/PostProc/GlowBlurV").getPointer();
			m->getTechnique(0)->getPass(0)->getFragmentProgramParameters()->setNamedConstant("Stretch", glowStrech);
			mWindow->setDebugText(String("GlowStretch ") + StringConverter::toString(glowStrech));
		}
		if (mInputDevice->isKeyDown(KC_N))
		{
			if (mInputDevice->isKeyDown(KC_RSHIFT))
				glowCombine -= 0.01;
			else
				glowCombine += 0.01;
			Ogre::Material* m = (Ogre::Material*) MaterialManager::getSingleton().getByName("GTP/Moria/GlowFinalCombine").getPointer();
			m->getTechnique(0)->getPass(0)->getFragmentProgramParameters()->setNamedConstant("weight2", glowCombine);
			mTimeUntilNextToggle = 0.2;
			mWindow->setDebugText(String("GlowCombine ") + StringConverter::toString(glowCombine));
		}

		if (mInputDevice->isKeyDown(KC_3) && mTimeUntilNextToggle <= 0)
		{
			PP_glowON = !PP_glowON;
			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
														"GTP/Moria/PostProc/Glow", PP_glowON);
			mTimeUntilNextToggle = 0.2;
		}
		if (mInputDevice->isKeyDown(KC_4) && mTimeUntilNextToggle <= 0)
		{
			PP_tonemapON = !PP_tonemapON;

			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
														"GTP/PostProc/ToneMap", PP_tonemapON);
			
			mTimeUntilNextToggle = 0.2;
		}
		if (mInputDevice->isKeyDown(KC_5) && mTimeUntilNextToggle <= 0)
		{
			PP_dofON = !PP_dofON;
			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
														"GTP/Moria/PostProc/DoF", PP_dofON);
			mTimeUntilNextToggle = 0.2;
		}

		if (mInputDevice->isKeyDown(KC_I))
		{
			light->getParentSceneNode()->setPosition(
				light->getParentSceneNode()->getPosition()
				+ Vector3(10*evt.timeSinceLastFrame,0,0));
		}
		if (mInputDevice->isKeyDown(KC_K))
		{
			light->getParentSceneNode()->setPosition(
				light->getParentSceneNode()->getPosition()
				+ Vector3(-10*evt.timeSinceLastFrame,0,0));
		}
		if (mInputDevice->isKeyDown(KC_J))
		{
			light->getParentSceneNode()->setPosition(
				light->getParentSceneNode()->getPosition()
				+ Vector3(0,0,10*evt.timeSinceLastFrame));
		}
		if (mInputDevice->isKeyDown(KC_L))
		{
			light->getParentSceneNode()->setPosition(
				light->getParentSceneNode()->getPosition()
				+ Vector3(0,0,-10*evt.timeSinceLastFrame));
		}
		
	
		
		return ExampleFrameListener::processUnbufferedKeyInput(evt);		
	}

	bool processUnbufferedMouseInput(const FrameEvent& evt)
	{
		return ExampleFrameListener::processUnbufferedMouseInput(evt);
	}
};

/** Application class */
class MoriaDemoApplication : public ExampleApplication
{




protected:
	SceneNode* mainLightNode;
	Light* mainLight;
	SceneNode *pSysNode;
	
	void setupResources(void)
    {
        // Load resource paths from config file
        ConfigFile cf;
        cf.load("resources_moriaScene.cfg");

        // Go through all sections & settings in the file
        ConfigFile::SectionIterator seci = cf.getSectionIterator();

        String secName, typeName, archName;
        while (seci.hasMoreElements())
        {
            secName = seci.peekNextKey();
            ConfigFile::SettingsMultiMap *settings = seci.getNext();
            ConfigFile::SettingsMultiMap::iterator i;
            for (i = settings->begin(); i != settings->end(); ++i)
            {
                typeName = i->first;
                archName = i->second;
                ResourceGroupManager::getSingleton().addResourceLocation(
                    archName, typeName, secName);
            }
        }
    }
	void loadResources(void)
	{
		// Initialise, parse scripts etc
		ResourceGroupManager::getSingleton().initialiseResourceGroup("Bootstrap");
		ResourceGroupManager::getSingleton().initialiseResourceGroup("GTP_Basic");
		ResourceGroupManager::getSingleton().initialiseResourceGroup("General");
		ResourceGroupManager::getSingleton().initialiseResourceGroup("GTP_EnvMap");
		ResourceGroupManager::getSingleton().initialiseResourceGroup("PostProc");
		ResourceGroupManager::getSingleton().initialiseResourceGroup("GTP_Particles");
		ResourceGroupManager::getSingleton().initialiseResourceGroup("GTP_PathMap");
		ResourceGroupManager::getSingleton().initialiseResourceGroup("GTP_MoriaScene");	
		//ResourceGroupManager::getSingleton().initialiseResourceGroup("GTP_Diffuse");
		//ResourceGroupManager::getSingleton().initialiseResourceGroup("GTP_Caustic");
	}

	void createPostproc()
	{	
		if(!capture)
		{
			CompositorManager::getSingleton().addCompositor(mWindow->getViewport(0),
															"FireHeatCompositor");
			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
															"FireHeatCompositor", true);
			
			CompositorManager::getSingleton().addCompositor(mWindow->getViewport(0),
															"GTP/Moria/PostProc/DoF");
			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
															"GTP/Moria/PostProc/DoF", PP_dofON);

			CompositorManager::getSingleton().addCompositor(mWindow->getViewport(0),
															"GTP/Moria/PostProc/Glow");
			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
															"GTP/Moria/PostProc/Glow", PP_glowON);

			
				CompositorManager::getSingleton().addCompositor(mWindow->getViewport(0),
															"GTP/PostProc/ToneMap");
			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
															"GTP/PostProc/ToneMap", PP_tonemapON);
					
			/*CompositorManager::getSingleton().addCompositor(mWindow->getViewport(0),
															"GTP/PostProc/MotionBlur");
			CompositorManager::getSingleton().setCompositorEnabled(mWindow->getViewport(0),
															"GTP/PostProc/MotionBlur", false);*/
		}
		else
		{
			TexturePtr tex = TextureManager::getSingleton().getByName("SCREENSHOT");
			Viewport* vp = tex->getBuffer()->getRenderTarget()->getViewport(0);

			CompositorManager::getSingleton().addCompositor(vp,
															"FireHeatCompositor");
			CompositorManager::getSingleton().setCompositorEnabled(vp,
															"FireHeatCompositor", true);
			
			CompositorManager::getSingleton().addCompositor(vp,
															"GTP/Moria/PostProc/DoF");
			CompositorManager::getSingleton().setCompositorEnabled(vp,
															"GTP/Moria/PostProc/DoF", true);

			CompositorManager::getSingleton().addCompositor(vp,
															"GTP/Moria/PostProc/Glow");
			CompositorManager::getSingleton().setCompositorEnabled(vp,
															"GTP/Moria/PostProc/Glow", true);

			
				CompositorManager::getSingleton().addCompositor(vp,
															"GTP/PostProc/ToneMap");
			CompositorManager::getSingleton().setCompositorEnabled(vp,
															"GTP/PostProc/ToneMap", true);
		}
		
	}
    
	void chooseSceneManager()
	{
		mSceneMgr = mRoot->createSceneManager("OctreeSceneManager");
	}

	void createScene(void) ///TOWERS
    {

	
		//Overlay* mainOverlay = OverlayManager::getSingleton().getByName("MORIA/MainOverlay");
		//mainOverlay->show();
		if(capture)
		{
			TexturePtr texPtr = Ogre::TextureManager::getSingleton().createManual(	"SCREENSHOT", 
																					"default",
																					TEX_TYPE_2D,
																					2048,
																					1536,
																					0,
																					0,
																					PF_R8G8B8,
																					TU_RENDERTARGET);
			Texture* screenTexture = texPtr.getPointer();
			RenderTexture* rt = screenTexture->getBuffer()->getRenderTarget();
			Viewport* vp = rt->addViewport(mCamera);
			rt->setAutoUpdated(false);
			vp->setOverlaysEnabled(false);
		}

		createPostproc();

		OgreIlluminationManager::getSingleton().setMainCamera(mCamera);
		OgreIlluminationManager::getSingleton().setMainViewport(mWindow->getViewport(0));
		 OgreIlluminationManager::getSingleton().setFocusingSM(false);
	   OgreIlluminationManager::getSingleton().setShadowMapSize(128);
	   OgreIlluminationManager::getSingleton().setBlurShadowMap(true);
	   OgreIlluminationManager::getSingleton().setShadowMapMaterialName(Light::LT_SPOTLIGHT, "GTP/Basic/Distance_Normalized");//GameTools/ShadowMapDistance
	   OgreIlluminationManager::getSingleton().setShadowMapMaterialName(Light::LT_POINT, "GTP/Basic/Distance_Normalized");
	   OgreIlluminationManager::getSingleton().setUseLISPSM(false);
	   OgreIlluminationManager::getSingleton().joinSharedRuns(false);
				
		Root::getSingleton()._setCurrentSceneManager(mSceneMgr);
		mCamera->setPosition(0,12,20);
		//mCamera->setPosition(0,100,0);
		mCamera->lookAt(15,5,0);
		mCamera->setFOVy(Radian(Degree(65)));
		mCamera->setNearClipDistance(0.1);
		mCamera->setFarClipDistance(500);
		// Set ambient light
        mSceneMgr->setAmbientLight(ColourValue(0.1, 0.1, 0.1));
		mSceneMgr->setSkyBox(true, "Examples/EveningSkyBox", 500);
		
		SceneNode* rootNode = mSceneMgr->getRootSceneNode();
		
		SceneSerializer s(mSceneMgr);
		DataStreamPtr inputStream; 				
		inputStream = ResourceGroupManager::getSingleton().openResource("MoriaScene.level");	//towers2.level	
		s.parseScript(inputStream, ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME);
		s.parseEntryPoints("MoriaSceneEntryPoints.text");
		
	
		mainLight = mSceneMgr->createLight("MainLight");
		mainLight->setType(Light::LT_POINT);
		//mainLight->setCastShadows(true);
        mainLight->setDiffuseColour(ColourValue::White);
		mainLight->setSpecularColour(ColourValue::White);
		mainLightNode = rootNode->createChildSceneNode();
		
		mainLightNode->setPosition(0,9,-190);		
		mainLight->setPowerScale(30000);
		//mainLight->setDirection(1,0, 0);
		mainLightNode->attachObject(mainLight);
		//mainLight->setSpotlightRange(Degree(0),Degree(120));
		mainLight->setAttenuation(1000.0, 0, 0, 3);

		CharacterGroupOrigo = rootNode->createChildSceneNode("CharactersOrigo");
		ParticleSystem* mFireBall = mSceneMgr->createParticleSystem("FPS_PLAYER_FIREBALL", "Moria/FireBall");
			
		 Light* mFireBallLight = mSceneMgr->createLight("FireBallLight");
		 mFireBallLight->setType(Light::LT_POINT);
		 mFireBallLight->setCastShadows(true);
		 mFireBallLight->setDiffuseColour(ColourValue(0.95,0.98,0.8));
		 mFireBallLight->setPowerScale(1000);
		 mFireBallLight->setAttenuation(1000.0, 0, 0, 6);	
		 mFireBallLight->setPosition(0,1,0);
		 SceneNode* fireballNode = CharacterGroupOrigo ->createChildSceneNode("fireBallNode");
		 fireballNode->setPosition(-0.55, 8.74, -2.37 + 10);
		 fireballNode->attachObject(mFireBallLight);
		 fireballNode->attachObject(mFireBall);

	
		BillboardSet* bbs = mSceneMgr->createBillboardSet("bb", 1);
		bbs->setDefaultDimensions(2.3, 2.3);
		bbs->createBillboard(Vector3::ZERO, ColourValue::White);
		bbs->setMaterialName("Flare");
		mainLightNode->attachObject(bbs);

		Entity* troll = mSceneMgr->createEntity("Troll", "trollMed.mesh");
		troll->setCastShadows(true);
		SceneNode* trollNode = CharacterGroupOrigo ->createChildSceneNode("trollNode");
		trollNode->setPosition(0,0,10);
		trollNode->attachObject(troll);
		
		//troll->setMaterialName("MoriaScene/Troll");
		//trollNode->setScale(0.3,0.3,0.3);
		
		Entity* guard = mSceneMgr->createEntity("Guard", "guard.mesh");
		guard->setCastShadows(true);
		SceneNode* guardNode = CharacterGroupOrigo ->createChildSceneNode("guardNode");
		guardNode->setPosition(0,0,10);
		guardNode->attachObject(guard);
		 	
		
		OgreIlluminationManager::getSingleton().initTechniques();
   }
	
	void createPlane(Ogre::String _name, Ogre::String _material, Ogre::Vector3 _pos, Ogre::Vector2 _size, Ogre::Vector2 _subdivisions = Ogre::Vector2(1,1)) 
	{
		Plane _plane;
        _plane.normal = Vector3::UNIT_Y;
        _plane.d = 0;


        MeshManager::getSingleton().createPlane(_name + ".plane",
            ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME, 
            _plane,_size.x,_size.y,1,1,true,1,_subdivisions.x,_subdivisions.y,Vector3::UNIT_Z);
        
		Entity *_entity = mSceneMgr->createEntity(_name + ".entity", _name + ".plane");
        _entity->setMaterialName(_material);

		SceneNode* _node = mSceneMgr->getRootSceneNode()->createChildSceneNode(_name + ".node");
		_node->attachObject(_entity);
		_node->setPosition(_pos);

	}

	void createFrameListener(void)
    {
		// This is where we instantiate our own frame listener
		mFrameListener= new MoriaDemoFrameListener(mWindow,
			mCamera, mainLightNode, mainLight, pSysNode	);
        mFrameListener->setPriority(10);	
		mRoot->addFrameListener(mFrameListener);
		OgreIlluminationManager::getSingleton().setPriority(100);
	    mRoot->addFrameListener(&OgreIlluminationManager::getSingleton());
	  
    }

};