#include <sstream>

#include "GLTB/commandline.h"
#include "GLTB/exception.h"

namespace gltb {
	void CommandLine::setArgs(int argc, char *argv[])
	{
		for(int i=0;i<argc;i++) {
			parameters.push_back(argv[i]);
			visited.push_back(false);
		}
	}
	
	bool CommandLine::checkPresence(std::string argument)
	{
		for(unsigned int i=0;i<parameters.size();i++) {
			if(parameters[i]==argument) {
				visited[i]=true;
				return true;
			}
		}
		return false;
	} 
	
	std::string CommandLine::getStringOption(std::string argument)
	{
		for(unsigned int i=0;i<parameters.size()-1;i++) {
			if(parameters[i]==argument) {
				visited[i]=true;
				visited[i+1]=true;
				return parameters[i+1];
			}
		}
		return "";
	}

#ifdef USE_GLM
	glm::vec2 CommandLine::getVec2Option(std::string argument, bool &isPresent)
	{
		for(unsigned int i = 0; i < parameters.size() - 2; i++) {
			if(parameters[i] == argument) {
				for(unsigned int j = i; j < i + 3; j++) {
					visited[j] = true;
				}

				glm::vec2 parameter(atof(parameters[i + 1].c_str()), atof(parameters[i + 2].c_str()));
				return parameter;
			}
		}
		return glm::vec2(0);
	}

	glm::vec2 CommandLine::getVec2Option(std::string argument)
	{
		bool isPresent;
		return getVec2Option(argument, isPresent);
	}

	glm::vec3 CommandLine::getVec3Option(std::string argument, bool &isPresent)
	{
		for(unsigned int i = 0; i < parameters.size() - 3; i++) {
			if(parameters[i] == argument) {
				for(unsigned int j = i; j < i + 4; j++) {
					visited[j] = true;
				}

				glm::vec3 parameter(atof(parameters[i + 1].c_str()), atof(parameters[i + 2].c_str()), atof(parameters[i + 3].c_str()));
				return parameter;
			}
		}
		return glm::vec3(0);
	}

	glm::vec3 CommandLine::getVec3Option(std::string argument)
	{
		bool isPresent;
		return getVec3Option(argument, isPresent);
	}

	glm::vec4 CommandLine::getVec4Option(std::string argument, bool &isPresent)
	{
		for(unsigned int i = 0; i < parameters.size() - 4; i++) {
			if(parameters[i] == argument) {
				for(unsigned int j = i; j < i + 5; j++) {
					visited[j] = true;
				}

				glm::vec4 parameter(atof(parameters[i + 1].c_str()), atof(parameters[i + 2].c_str()), atof(parameters[i + 3].c_str()), atof(parameters[i + 4].c_str()));
				return parameter;
			}
		}
		return glm::vec4(0);
	}

	glm::vec4 CommandLine::getVec4Option(std::string argument)
	{
		bool isPresent;
		return getVec4Option(argument, isPresent);
	}

	glm::ivec2 CommandLine::getIVec2Option(std::string argument, bool &isPresent)
	{
		for(unsigned int i = 0; i < parameters.size() - 2; i++) {
			if(parameters[i] == argument) {
				for(unsigned int j = i; j < i + 3; j++) {
					visited[j] = true;
				}

				glm::ivec2 parameter(atoi(parameters[i + 1].c_str()), atoi(parameters[i + 2].c_str()));
				return parameter;
			}
		}
		return glm::ivec2(0);
	}

	glm::ivec2 CommandLine::getIVec2Option(std::string argument)
	{
		bool isPresent;
		return getIVec2Option(argument, isPresent);
	}

	glm::ivec3 CommandLine::getIVec3Option(std::string argument, bool &isPresent)
	{
		for(unsigned int i = 0; i < parameters.size() - 3; i++) {
			if(parameters[i] == argument) {
				for(unsigned int j = i; j < i + 4; j++) {
					visited[j] = true;
				}

				glm::ivec3 parameter(atoi(parameters[i + 1].c_str()), atoi(parameters[i + 2].c_str()), atoi(parameters[i + 3].c_str()));
				return parameter;
			}
		}
		return glm::ivec3(0);
	}

	glm::ivec3 CommandLine::getIVec3Option(std::string argument)
	{
		bool isPresent;
		return getIVec3Option(argument, isPresent);
	}

	glm::ivec4 CommandLine::getIVec4Option(std::string argument, bool &isPresent)
	{
		for(unsigned int i = 0; i < parameters.size() - 4; i++) {
			if(parameters[i] == argument) {
				for(unsigned int j = i; j < i + 5; j++) {
					visited[j] = true;
				}

				glm::ivec4 parameter(atoi(parameters[i + 1].c_str()), atoi(parameters[i + 2].c_str()), atoi(parameters[i + 3].c_str()), atoi(parameters[i + 4].c_str()));
				return parameter;
			}
		}
		return glm::ivec4(0);
	}

	glm::ivec4 CommandLine::getIVec4Option(std::string argument)
	{
		bool isPresent;
		return getIVec4Option(argument, isPresent);
	}
#endif
	
	int CommandLine::getNumUnvisitedParameters() {
		int count=0;
		for(unsigned int i=0;i<parameters.size();i++) {
			if(!visited[i]) {
				count++;
			}
		}
		return count;
	}
	
	std::string CommandLine::getUnvisitedParameter(int numParameter) {
		int count=0;
		for(unsigned int i=0;i<parameters.size();i++) {
			if(!visited[i]) {
				if(count==numParameter) {
					return parameters[i];
				}
				count++;
			}
		}
		return "";		
	}

	void CommandLineParser::registerArgument(RefPtr<CommandLineArgument> argument)
	{
		arguments.push_back(argument);
		argumentHandled.push_back(false);
	}

	void CommandLineParser::parse(int argc, const char *argv[])
	{
		for(int i = 1; i < argc; i++) {
			bool isArgument = false;

			for(size_t j = 0; j < arguments.size(); j++) {
				if(argv[i] == arguments[j]->getName()) {
					if(argumentHandled[j] == true && arguments[j]->onlyOnceAllowed()) {
						throw Exception("command line argument " + arguments[j]->getName() + " present multiple times on the command line", "gltb::CommandLineParser::parse()");
					}

					if(i > argc - (arguments[j]->getNumParameters() + 1)) {
						std::ostringstream exceptionMessage;
						exceptionMessage << "command line argument " << arguments[j]->getName() << " expects " << arguments[j]->getNumParameters() << " parameters, but only " << argc - (arguments[j]->getNumParameters() + 1) - i << " are present";
						throw Exception(exceptionMessage.str(), "gltb::CommandLineParser::parse()");
					}

					std::vector<std::string> parameters;

					for(int k = 0; k < arguments[j]->getNumParameters(); k++) {
						parameters.push_back(argv[i + k + 1]);
					}
					arguments[j]->_addParameters(parameters);
					i += arguments[j]->getNumParameters();

					argumentHandled[j] = true;
					isArgument = true;
					break;
				}
			}

			if(!isArgument) {
				unhandledArguments.push_back(argv[i]);
			}
		}
	}
}