#include "GLTB/commandline.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(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(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(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(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(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(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 "";		
	}
}