#pragma once

#include <sstream>

#include "IEngine.h"
#include "IResourceFactory.h"

#include "../Resources/RenderTarget/SwapChain/SwapChain.h"
#include "../Resources/RenderTarget/RenderTargetTexture/RenderTargetTexture.h"
#include "../Resources/GeometryBuffer/GeometryBuffer.h"
#include "../Resources/Shader/Shader.h"
#include "../Resources/Shader/InternalShader.h"
#include "../Resources/RenderCommandNode/RenderCommandNode.h"
#include "../Resources/ShaderParameterCollection/ShaderParameterCollection.h"
#include "../Resources/Texture/Texture.h"
#include "../Resources/Transformation/Transformation.h"
#include "../Resources/TransformationAnimation/TransformationAnimation.h"
#include "../Resources/Camera/OrthographicCamera.h"
#include "../Resources/Camera/PerspectiveCamera.h"
#include "../Resources/Bone/Bone.h"
#include "../Resources/Bone/BoneController.h"
#include "../Window/Window.h"


class ResourceFactory
	: public IResourceFactory
{
public:
	ResourceFactory(IEngine& argEngine);
	virtual ~ResourceFactory();

	virtual ITransformationAnimation& CreateTransformationAnimation();
	virtual void DeleteTransformationAnimation(ITransformationAnimation& argValue, bool argRemoveOnly = false);

	virtual IWindow& CreateSwapChainWindow();
	virtual void DeleteSwapChainWindow(IWindow& argValue, bool argRemoveOnly = false);

	virtual IRenderCommandNode& CreateRenderCommandNode();
	virtual void DeleteRenderCommandNode(IRenderCommandNode& argValue, bool argRemoveOnly = false);

	virtual IShader* FindShader(const string8& argId);
	virtual IShader& CreateOrFindShader(const string8& argId = "");
	virtual void DeleteShader(IShader& argValue, bool argRemoveOnly = false);

	virtual IInternalShader* FindInternalShader(const string8& argId);
	virtual IInternalShader& CreateOrFindInternalShader(const string8& argId = "");
	virtual void DeleteInternalShader(IInternalShader& argValue, bool argRemoveOnly = false);

	virtual ISwapChain* FindSwapChain(const string8& argId);
	virtual ISwapChain& CreateOrFindSwapChain(const string8& argId = "");
	virtual void DeleteSwapChain(ISwapChain& argValue, bool argRemoveOnly = false);

	virtual IRenderTargetTexture* FindRenderTargetTexture(const string8& argId);
	virtual IRenderTargetTexture& CreateOrFindRenderTargetTexture(const string8& argId = "");
	virtual void DeleteRenderTargetTexture(IRenderTargetTexture& argValue, bool argRemoveOnly = false);

	virtual IGeometryBuffer* FindGeometryBuffer(const string8& argId);
	virtual IGeometryBuffer& CreateOrFindGeometryBuffer(const string8& argId = "");
	virtual void DeleteGeometryBuffer(IGeometryBuffer& argValue, bool argRemoveOnly = false);

	virtual IShaderParameterCollection* FindShaderParameterCollection(const string8& argId);
	virtual IShaderParameterCollection& CreateOrFindShaderParameterCollection(const string8& argId = "");
	virtual void DeleteShaderParameterCollection(IShaderParameterCollection& argValue, bool argRemoveOnly = false);

	virtual ITexture* FindTexture(const string8& argId);
	virtual ITexture& CreateOrFindTexture(const string8& argId = "");
	virtual void DeleteTexture(ITexture& argValue, bool argRemoveOnly = false);

	virtual ITransformation* FindTransformation(const string8& argId);
	virtual ITransformation& CreateOrFindTransformation(const string8& argId = "");
	virtual void DeleteTransformation(ITransformation& argValue, bool argRemoveOnly = false);

	virtual IPerspectiveCamera* FindPerspectiveCamera(const string8& argId);
	virtual IPerspectiveCamera& CreateOrFindPerspectiveCamera(const string8& argId = "");
	virtual void DeletePerspectiveCamera(IPerspectiveCamera& argValue, bool argRemoveOnly = false);

	virtual IOrthographicCamera* FindOrthographicCamera(const string8& argId);
	virtual IOrthographicCamera& CreateOrFindOrthographicCamera(const string8& argId = "");
	virtual void DeleteOrthographicCamera(IOrthographicCamera& argValue, bool argRemoveOnly = false);

	virtual IBone* FindBone(const string8& argId);
	virtual IBone& CreateOrFindBone(const string8& argId = "");
	virtual void DeleteBone(IBone& argValue, bool argRemoveOnly = false);

	virtual IBoneController* FindBoneController(const string8& argId);
	virtual IBoneController& CreateOrFindBoneController(const string8& argId = "");
	virtual void DeleteBoneController(IBoneController& argValue, bool argRemoveOnly = false);

	virtual bool HasCreatedResource() { return m_LastCallCreatedResource; }


protected:

	template<typename A, typename B> void DeleteInternal(std::map<string8, A*>& argPool, B& argWhich, bool argRemoveOnly)
	{
		A* valueA = dynamic_cast<A*>(&argWhich);

		for (std::map<string8, A*>::iterator iter = argPool.begin(); iter != argPool.end(); ++iter)
			if (iter->second == valueA)
			{
				ICommandUser* commandUser = dynamic_cast<ICommandUser*>(iter->second);
				if (commandUser != NULL)
					std::cout << "Deleting \"" << std::green << iter->first << std::white << "\" [" << commandUser->get_UserName() << "]" << std::endl;

				argPool.erase(iter);
				if (!argRemoveOnly)
					delete valueA;

				return;
			}
	}

	template<typename T> void ClearInternal(std::map<string8, T*>& argPool)
	{
		while (!argPool.empty())
		{
			ICommandUser* commandUser = dynamic_cast<ICommandUser*>(argPool.begin()->second);
			if (commandUser != NULL)
				std::cout << "Deleting \"" << std::green << argPool.begin()->first << std::white << "\" [" << commandUser->get_UserName() << "]" << std::endl;

			T* valueA = argPool.begin()->second;
			argPool.erase(argPool.begin());
			delete valueA;
		}
	}

	template<typename T> void ClearInternal(std::vector<T*>& argPool)
	{
		while (!argPool.empty())
		{
			delete *argPool.begin();
			argPool.erase(argPool.begin());
		}
	}

	template<typename T> T* FindInternal(std::map<string8, T*>& argPool, const string8& argId)
	{
		m_LastCallCreatedResource = false;
		if (argPool.find(argId) == argPool.end())
			return NULL;
		return argPool[argId];
	}

	template<typename T> string8 CreateUniqueName(std::map<string8, T*>& argPool, const string8& argPrefix = "")
	{
		int NameIndex = argPool.size();
		std::stringstream CurrentName;
		do
		{
			CurrentName.flush();
			CurrentName << argPrefix << NameIndex++;
		}
		while (FindInternal(argPool, CurrentName.str()) != NULL);

		return CurrentName.str();
	}

	template<typename T> T* CreateOrFindInternal(std::map<string8, T*>& argPool, const string8& argId)
	{
		std::string currentId = argId;
		if (currentId.empty())
			currentId = CreateUniqueName(argPool);

		T* instance = this->FindInternal(argPool, currentId);
		if (instance != NULL)
			return instance;
		m_LastCallCreatedResource = true;
		
		instance = new T(m_Engine);

		ICommandUser* commandUser = dynamic_cast<ICommandUser*>(instance);
		if (commandUser != NULL)
			std::cout << "Creating \"" << std::green << currentId << std::white << "\" [" << commandUser->get_UserName() << "]" << std::endl;

		argPool[currentId] = instance;
		return instance;
	}

private:
	IEngine& m_Engine;

	std::map<string8, InternalShader*> m_InternalShaderPool;
	std::map<string8, Shader*> m_ShaderPool;
	std::map<string8, GeometryBuffer*> m_GeometryBufferPool;
	std::map<string8, SwapChain*> m_SwapChainPool;
	std::map<string8, RenderTargetTexture*> m_RenderTargetTexturePool;
	std::map<string8, ShaderParameterCollection*> m_ShaderParameterCollectionPool;
	std::map<string8, Texture*> m_TexturePool;
	std::map<string8, Transformation*> m_TransformationPool;
	std::map<string8, OrthographicCamera*> m_OrthographicCameraPool;
	std::map<string8, PerspectiveCamera*> m_PerspectiveCameraPool;
	std::map<string8, Bone*> m_BonePool;
	std::map<string8, BoneController*> m_BoneControllerPool;
	
	std::vector<IRenderCommandNode*> m_RenderCommandNodes;
	std::vector<IWindow*> m_Windows;
	std::vector<ITransformationAnimation*> m_TransformationAnimations;

	bool m_LastCallCreatedResource;
};