#define OLC_PGE_APPLICATION
#include "pixelGameEngine.h"
#define OLC_PGEX_SPLASHSCREEN
#include "splash.h"
#define OLC_SOUNDWAVE
#include "soundwaveEngine.h"
#include "tiles.h"
#include "references.h"
#include "states.h"

#define WIDTH 256
#define HEIGHT 224
#define CAMERA_MOVESPD 5
#define TILEMAP_SIZE_X 512
#define TILEMAP_SIZE_Y 512
#define TILEMAP_EDITOR_DRAW_MULT 0.4375
#define TILEMAP_EDITOR_TILESIZE (32*TILEMAP_EDITOR_DRAW_MULT)

using namespace olc;

class Map{
	public:
		std::string filename;
		std::string l2filename;
		std::string l3filename;
		std::string l4filename;
		Decal*tileset;
		Map(std::string fname,std::string layer2_fname,std::string layer3_fname,std::string layer4_fname,Decal*tileset) {
			this->filename=fname;
			this->l2filename=layer2_fname;
			this->l3filename=layer3_fname;
			this->l4filename=layer4_fname;
			this->tileset=tileset;
		}
};

class Animation{
	public:
		Decal*spr;
		int frames=1;
		int width=0;
		Animation(Decal*spr,int width){
			this->frames=spr->sprite->width/width;
			this->width=width;
			this->spr=spr;
		}
};

class Object{
	public:
		Animation*spr;
		vd2d pos;
		int frameIndex=0;
		int frameCount=0;
		int animationSpd=12; //How many frames to wait between each frame.
		std::string name;
		vd2d scale={1,1};
		Pixel color=WHITE;
		vd2d originPoint={0,0};
		bool drawn=false;
		//animationSpd is how long to wait before switching frames.
		Object(std::string name,vi2d pos,Animation*spr,vi2d scale={1,1},Pixel color=WHITE,int animationSpd=1) {
			this->spr=spr;
			this->pos=pos;
			this->name=name;
			this->scale=scale;
			this->color=color;
			this->animationSpd=animationSpd;
			this->originPoint={spr->spr->sprite->width/2,spr->spr->sprite->height-4};
		}
};

class SeasonI : public PixelGameEngine
{
public:
	SeasonI()
	{
		sAppName = "Season I: Winters of Loneliness";

	}

public:
	int frameCount=0;
	float elapsedTime=0;
	const float TARGET_RATE = 1/60.0;
	int MAP_WIDTH;
	int MAP_HEIGHT;
	Map*CURRENT_MAP;
	Map*MAP_ONETT;
	vd2d cameraPos = {0,0};
	state::GameState GAME_STATE = state::EDITOR;
	vi2d SELECTED_TILE={0,0};
	vi2d HIGHLIGHTED_TILE={0,0};
	int EDITING_LAYER=2;

	bool MOUSE_PRESSED_DOWN=false,MOUSE_DOWN=false,MOUSE_RELEASED=false; //TODO Implement Mouse things.

	std::vector<std::vector<TILE*>> MAP; //The foreground layer.
	std::vector<std::vector<TILE*>> MAP2;
	std::vector<std::vector<TILE*>> MAP3;
	std::vector<std::vector<TILE*>> MAP4;
	std::map<Reference,Decal*> SPRITES;
	std::map<Reference,Animation*> ANIMATIONS;
	std::map<int,Object*> OBJ_INFO;
	std::vector<Object*> OBJECTS;

	bool OnUserCreate() override
	{
		for (int i=1;i<5;i++) {
			CreateLayer();
			EnableLayer(i,true);
		}
		SetPixelMode(Pixel::ALPHA);
		ConsoleCaptureStdOut(true);
		// Called once at the start, so create things here

		SetupAnimations();
		SetupObjectInfo();

		MAP_ONETT=new Map("map0","map0_2","map0_3","map0_4",SPRITES[TILESET1]);

		//OBJ_INFO["PLAYER"]=PLAYER_ANIMATION;

		LoadMap(MAP_ONETT);

		return true;
	}

	bool OnUserUpdate(float fElapsedTime) override
	{
		elapsedTime+=fElapsedTime;
		while (elapsedTime>TARGET_RATE) {
			elapsedTime-=TARGET_RATE;
			updateGame();
		}
		keyUpdates();
		SetDrawTarget(nullptr);
		Clear(BLANK);
		SetDrawTarget(1);
		Clear(BLANK);
		SetDrawTarget(2);
		Clear(BLANK);
		SetDrawTarget(3);
		Clear(BLANK);
		SetDrawTarget(4);
		Clear(BLANK);
		drawGame();
		return true;
	}

	void updateGame(){
		frameCount++;
		for (auto&obj:OBJECTS) {
			if (obj->frameCount++>obj->animationSpd) {
				obj->frameCount=0;
				obj->frameIndex++;
			}
		}

		switch (GAME_STATE) {
			case state::TILE_SELECT:{
				if (!TabHeld()) {
					GAME_STATE=state::EDITOR;
				}
				if (GetMouse(0).bHeld) {
					int selectedTileX=GetMouseX()/TILEMAP_EDITOR_TILESIZE;
					int selectedTileY=GetMouseY()/TILEMAP_EDITOR_TILESIZE;
					if (selectedTileX*TILEMAP_EDITOR_DRAW_MULT>=0&&selectedTileX*TILEMAP_EDITOR_DRAW_MULT<TILEMAP_SIZE_X&&selectedTileY*TILEMAP_EDITOR_DRAW_MULT<TILEMAP_SIZE_Y&&selectedTileY*TILEMAP_EDITOR_DRAW_MULT>=0) {
						SELECTED_TILE={selectedTileX,selectedTileY};
					}
				}
			}break;
			case state::EDITOR:{
				if (TabHeld()) {
					GAME_STATE=state::TILE_SELECT;
				}
				if (GetKey(K1).bHeld) {
					EDITING_LAYER=1;
				}
				if (GetKey(K2).bHeld) {
					EDITING_LAYER=2;
				}
				if (GetKey(K3).bHeld) {
					EDITING_LAYER=3;
				}
				if (GetKey(K4).bHeld) {
					EDITING_LAYER=4;
				}
				if (UpHeld()) {
					cameraPos.y-=CAMERA_MOVESPD;
				}
				if (RightHeld()) {
					cameraPos.x+=CAMERA_MOVESPD;
				}
				if (LeftHeld()) {
					cameraPos.x-=CAMERA_MOVESPD;
				}
				if (DownHeld()) {
					cameraPos.y+=CAMERA_MOVESPD;
				}
				if (GetKey(I).bHeld) {
					OBJECTS[0]->pos.y-=1;
				}
				if (GetKey(K).bHeld) {
					OBJECTS[0]->pos.y+=1;
				}
				if (GetKey(J).bHeld) {
					OBJECTS[0]->pos.x-=1;
				}
				if (GetKey(L).bHeld) {
					OBJECTS[0]->pos.x+=1;
				}
				int selectedTileX=(GetMouseX()+cameraPos.x)/32;
				int selectedTileY=(GetMouseY()+cameraPos.y)/32;
				if (selectedTileX<MAP_WIDTH&&selectedTileY<MAP_HEIGHT&&selectedTileX>=0&&selectedTileY>=0) {
					HIGHLIGHTED_TILE={selectedTileX,selectedTileY};
				}
				if (GetMouse(0).bHeld) {
					switch (EDITING_LAYER) {
						case 1:{
							TILE*tile=MAP[HIGHLIGHTED_TILE.y][HIGHLIGHTED_TILE.x];
							tile->tileX=SELECTED_TILE.x;
							tile->tileY=SELECTED_TILE.y;
						}break;
						case 2:{
							TILE*tile=MAP2[HIGHLIGHTED_TILE.y][HIGHLIGHTED_TILE.x];
							tile->tileX=SELECTED_TILE.x;
							tile->tileY=SELECTED_TILE.y;
						}break;
						case 3:{
							TILE*tile=MAP3[HIGHLIGHTED_TILE.y][HIGHLIGHTED_TILE.x];
							tile->tileX=SELECTED_TILE.x;
							tile->tileY=SELECTED_TILE.y;
						}break;
						case 4:{
							TILE*tile=MAP4[HIGHLIGHTED_TILE.y][HIGHLIGHTED_TILE.x];
							tile->tileX=SELECTED_TILE.x;
							tile->tileY=SELECTED_TILE.y;
						}break;
					}
				}
			}break;
		}
	}

	void keyUpdates() {
		switch (GAME_STATE) {
			case state::TILE_SELECT:{
				if (UpPressed()) {
					SELECTED_TILE.y=SELECTED_TILE.y-1;
					if (SELECTED_TILE.y<0) {
						SELECTED_TILE.y=TILEMAP_SIZE_Y/32-1;
					}
				}
				if (RightPressed()) {
					SELECTED_TILE.x=(SELECTED_TILE.x+1)%(TILEMAP_SIZE_X/32);
				}
				if (LeftPressed()) {
					SELECTED_TILE.x=SELECTED_TILE.x-1;
					if (SELECTED_TILE.x<0) {
						SELECTED_TILE.x=TILEMAP_SIZE_X/32-1;
					}
				}
				if (DownPressed()) {
					SELECTED_TILE.y=(SELECTED_TILE.y+1)%(TILEMAP_SIZE_Y/32);
				}
			}break;
		}
	}

	void drawGame(){
		for (auto&obj:OBJECTS) {
			obj->drawn=false;
		}
		switch (GAME_STATE) {
			case state::EDITOR:{
				for (int y=-1;y<HEIGHT/32+2;y++) {
					int yTileOffset = cameraPos.y/32;
					for (auto&obj:OBJECTS) {
						if (!obj->drawn&&obj->pos.y+obj->originPoint.y>(y+yTileOffset)*32&&obj->pos.y+obj->originPoint.y<=(y+yTileOffset+1)*32) {
							obj->drawn=true;
							SetDrawTarget(2);
							DrawPartialDecal(obj->pos-cameraPos,obj->spr->spr,{(obj->frameIndex%obj->spr->frames)*obj->spr->width,0},{obj->spr->width,obj->spr->spr->sprite->height},obj->scale,obj->color);
						}
					}
					for (int x=-1;x<WIDTH/32+2;x++) {
						int xTileOffset = cameraPos.x/32;
						if (x+xTileOffset>=0&&x+xTileOffset<MAP_WIDTH&&y+yTileOffset>=0&&y+yTileOffset<MAP_HEIGHT) {
							SetDrawTarget(4);
							DrawPartialDecal({x*32-fmod(cameraPos.x,32),y*32-fmod(cameraPos.y,32)},SPRITES[TILESET1],{MAP4[y+yTileOffset][x+xTileOffset]->tileX*32,MAP4[y+yTileOffset][x+xTileOffset]->tileY*32},{32,32});
							SetDrawTarget(3);
							DrawPartialDecal({x*32-fmod(cameraPos.x,32),y*32-fmod(cameraPos.y,32)},SPRITES[TILESET1],{MAP3[y+yTileOffset][x+xTileOffset]->tileX*32,MAP3[y+yTileOffset][x+xTileOffset]->tileY*32},{32,32});
							SetDrawTarget(2);
							DrawPartialDecal({x*32-fmod(cameraPos.x,32),y*32-fmod(cameraPos.y,32)},SPRITES[TILESET1],{MAP2[y+yTileOffset][x+xTileOffset]->tileX*32,MAP2[y+yTileOffset][x+xTileOffset]->tileY*32},{32,32});
							SetDrawTarget(1);
							DrawPartialDecal({x*32-fmod(cameraPos.x,32),y*32-fmod(cameraPos.y,32)},SPRITES[TILESET1],{MAP[y+yTileOffset][x+xTileOffset]->tileX*32,MAP[y+yTileOffset][x+xTileOffset]->tileY*32},{32,32});
						}
					}
				}
				SetDrawTarget(nullptr);
				DrawRectDecal((HIGHLIGHTED_TILE)*32-cameraPos,{32,32},YELLOW);
				DrawStringPropDecal({2,2},"Editing Layer "+std::to_string(EDITING_LAYER));
			}break;
			case state::TILE_SELECT:{
				//14x14 pixels per tile.
				DrawDecal({0,0},SPRITES[TILESET1],{TILEMAP_EDITOR_DRAW_MULT,TILEMAP_EDITOR_DRAW_MULT});
				DrawRectDecal(SELECTED_TILE*(TILEMAP_EDITOR_TILESIZE),{TILEMAP_EDITOR_TILESIZE,TILEMAP_EDITOR_TILESIZE},RED);
			}break;
		}
	};

	void LoadMap(Map*map) {
		std::ifstream f("assets/maps/"+map->filename);
		std::ifstream f2("assets/maps/"+map->l2filename);
		std::ifstream f3("assets/maps/"+map->l3filename);
		std::ifstream f4("assets/maps/"+map->l4filename);
		for (int i=0;i<MAP.size();i++) {
			for (int j=0;j<MAP[i].size();j++) {
				delete MAP[i][j];
			}
			MAP[i].clear();
		}
		for (int i=0;i<MAP2.size();i++) {
			for (int j=0;j<MAP2[i].size();j++) {
				delete MAP2[i][j];
			}
			MAP2[i].clear();
		}
		for (int i=0;i<MAP3.size();i++) {
			for (int j=0;j<MAP3[i].size();j++) {
				delete MAP3[i][j];
			}
			MAP3[i].clear();
		}
		for (int i=0;i<MAP4.size();i++) {
			for (int j=0;j<MAP4[i].size();j++) {
				delete MAP4[i][j];
			}
			MAP4[i].clear();
		}
		MAP_WIDTH=-1;
		MAP_HEIGHT=-1;
		MAP.clear();
		std::string data;
		while (f.good()) {
			f>>data;
			if (MAP_WIDTH==-1) {
				MAP_WIDTH=data.length();
				printf("Map Width: %d\n",MAP_WIDTH);
			}
			if (data.find("OBJECT")!=std::string::npos) {
				int marker=data.find_first_of(';');
				int lastMarker=marker;
				std::stringstream split1(data.substr(6,marker-6));
				marker=data.find_first_of(';',marker+1);
				std::stringstream split2(data.substr(lastMarker+1,marker-lastMarker-1));
				lastMarker=marker;
				marker=data.find_first_of(';',marker+1);
				std::stringstream split3(data.substr(lastMarker+1,marker-lastMarker-1));
				lastMarker=marker;

				double x,y;

				split1>>x;
				split2>>y;

				int id;
				split3>>id;

				Object*newObj = new Object(OBJ_INFO[id]->name,{x,y},OBJ_INFO[id]->spr);
				OBJECTS.push_back(newObj);
				//printf("Object: %s %s %s\n",split1.c_str(),split2.c_str(),split3.c_str());
			} else {
				std::vector<TILE*> tiles;
				for (int i=0;i<data.length();i++) {
					unsigned char tileX,tileY;
					tileX=data[i]&0x0F;
					tileY=(data[i]&0xF0)>>4;
					tiles.push_back(new TILE(tileX,tileY));
				}
				MAP.push_back(tiles);
				MAP_HEIGHT++;
			}
		}
		while (f2.good()) {
			f2>>data;
			if (data.find("OBJECT")!=std::string::npos) {
				int marker=data.find_first_of(';');
				int lastMarker=marker;
				std::stringstream split1(data.substr(6,marker-6));
				marker=data.find_first_of(';',marker+1);
				std::stringstream split2(data.substr(lastMarker+1,marker-lastMarker-1));
				lastMarker=marker;
				marker=data.find_first_of(';',marker+1);
				std::stringstream split3(data.substr(lastMarker+1,marker-lastMarker-1));
				lastMarker=marker;

				double x,y;

				split1>>x;
				split2>>y;

				int id;
				split3>>id;

				Object*newObj = new Object(OBJ_INFO[id]->name,{x,y},OBJ_INFO[id]->spr);
				OBJECTS.push_back(newObj);
				//printf("Object: %s %s %s\n",split1.c_str(),split2.c_str(),split3.c_str());
			} else {
				std::vector<TILE*> tiles;
				for (int i=0;i<data.length();i++) {
					unsigned char tileX,tileY;
					tileX=data[i]&0x0F;
					tileY=(data[i]&0xF0)>>4;
					tiles.push_back(new TILE(tileX,tileY));
				}
				MAP2.push_back(tiles);
			}
		}
		while (f3.good()) {
			f3>>data;
			if (data.find("OBJECT")!=std::string::npos) {
				int marker=data.find_first_of(';');
				int lastMarker=marker;
				std::stringstream split1(data.substr(6,marker-6));
				marker=data.find_first_of(';',marker+1);
				std::stringstream split2(data.substr(lastMarker+1,marker-lastMarker-1));
				lastMarker=marker;
				marker=data.find_first_of(';',marker+1);
				std::stringstream split3(data.substr(lastMarker+1,marker-lastMarker-1));
				lastMarker=marker;

				double x,y;

				split1>>x;
				split2>>y;

				int id;
				split3>>id;

				Object*newObj = new Object(OBJ_INFO[id]->name,{x,y},OBJ_INFO[id]->spr);
				OBJECTS.push_back(newObj);
				//printf("Object: %s %s %s\n",split1.c_str(),split2.c_str(),split3.c_str());
			} else {
				std::vector<TILE*> tiles;
				for (int i=0;i<data.length();i++) {
					unsigned char tileX,tileY;
					tileX=data[i]&0x0F;
					tileY=(data[i]&0xF0)>>4;
					tiles.push_back(new TILE(tileX,tileY));
				}
				MAP3.push_back(tiles);
			}
		}
		while (f4.good()) {
			f4>>data;
			if (data.find("OBJECT")!=std::string::npos) {
				int marker=data.find_first_of(';');
				int lastMarker=marker;
				std::stringstream split1(data.substr(6,marker-6));
				marker=data.find_first_of(';',marker+1);
				std::stringstream split2(data.substr(lastMarker+1,marker-lastMarker-1));
				lastMarker=marker;
				marker=data.find_first_of(';',marker+1);
				std::stringstream split3(data.substr(lastMarker+1,marker-lastMarker-1));
				lastMarker=marker;

				double x,y;

				split1>>x;
				split2>>y;

				int id;
				split3>>id;

				Object*newObj = new Object(OBJ_INFO[id]->name,{x,y},OBJ_INFO[id]->spr);
				OBJECTS.push_back(newObj);
				//printf("Object: %s %s %s\n",split1.c_str(),split2.c_str(),split3.c_str());
			} else {
				std::vector<TILE*> tiles;
				for (int i=0;i<data.length();i++) {
					unsigned char tileX,tileY;
					tileX=data[i]&0x0F;
					tileY=(data[i]&0xF0)>>4;
					tiles.push_back(new TILE(tileX,tileY));
				}
				MAP4.push_back(tiles);
			}
		}
		printf("Loaded map %s.\n",map->filename.c_str());
		f.close();
		f2.close();
		f3.close();
	}

	Decal*CreateSprite(std::string spriteName) {
		return new Decal(new Sprite("assets/"+spriteName));
	}

	void SetupAnimations() {
		SPRITES[PLAYER] = CreateSprite("player.png");
		ANIMATIONS[PLAYER] = new Animation(SPRITES[PLAYER],32);
		SPRITES[TILESET1] = CreateSprite("terrainmap.png");
	}

	void SetupObjectInfo() {
		OBJ_INFO[PLAYER]=new Object("player",{0,0},ANIMATIONS[PLAYER]);
	}



	bool TabHeld(){
		return GetKey(TAB).bHeld;
	}
	bool UpPressed(){
		return GetKey(W).bPressed||GetKey(UP).bPressed||GetKey(NP8).bPressed||MOUSE_PRESSED_DOWN&&GetMouseY()<=HEIGHT-128+32&&GetMouseY()>=HEIGHT-128&&GetMouseX()<=HEIGHT-128;
	}
	bool DownPressed(){
		return GetKey(S).bPressed||GetKey(DOWN).bPressed||GetKey(NP5).bPressed||GetKey(NP2).bPressed||MOUSE_PRESSED_DOWN&&GetMouseY()<=HEIGHT&&GetMouseY()>=HEIGHT-32&&GetMouseX()<=HEIGHT-128;
	}
	bool LeftPressed(){
		return GetKey(A).bPressed||GetKey(LEFT).bPressed||GetKey(NP4).bPressed||MOUSE_PRESSED_DOWN&&GetMouseX()<=32&&GetMouseX()>=0&&GetMouseY()>=HEIGHT-128;
	}
	bool RightPressed(){
		return GetKey(D).bPressed||GetKey(RIGHT).bPressed||GetKey(NP6).bPressed||MOUSE_PRESSED_DOWN&&GetMouseX()<=128&&GetMouseX()>=96&&GetMouseY()>=HEIGHT-128;
	}
	bool UpHeld(){
		return GetKey(W).bHeld||GetKey(UP).bHeld||GetKey(NP8).bHeld||MOUSE_DOWN&&GetMouseY()<=HEIGHT-128+32&&GetMouseY()>=HEIGHT-128&&GetMouseX()<=HEIGHT-128;
	}
	bool DownHeld(){
		return GetKey(S).bHeld||GetKey(DOWN).bHeld||GetKey(NP5).bHeld||GetKey(NP2).bHeld||MOUSE_DOWN&&GetMouseY()<=HEIGHT&&GetMouseY()>=HEIGHT-32&&GetMouseX()<=HEIGHT-128;
	}
	bool LeftHeld(){
		return GetKey(A).bHeld||GetKey(LEFT).bHeld||GetKey(NP4).bHeld||MOUSE_DOWN&&GetMouseX()<=32&&GetMouseX()>=0&&GetMouseY()>=HEIGHT-128;
	}
	bool RightHeld(){
		return GetKey(D).bHeld||GetKey(RIGHT).bHeld||GetKey(NP6).bHeld||MOUSE_DOWN&&GetMouseX()<=128&&GetMouseX()>=96&&GetMouseY()>=HEIGHT-128;
	}
	bool UpReleased(){
		return GetKey(W).bReleased||GetKey(UP).bReleased||GetKey(NP8).bReleased||MOUSE_RELEASED&&GetMouseY()<=HEIGHT-128+32&&GetMouseY()>=HEIGHT-128&&GetMouseX()<=HEIGHT-128;
	}
	bool DownReleased(){
		return GetKey(S).bReleased||GetKey(DOWN).bReleased||GetKey(NP5).bReleased||GetKey(NP2).bReleased||MOUSE_RELEASED&&GetMouseY()<=HEIGHT&&GetMouseY()>=HEIGHT-32&&GetMouseX()<=HEIGHT-128;
	}
	bool LeftReleased(){
		return GetKey(A).bReleased||GetKey(LEFT).bReleased||GetKey(NP4).bReleased||MOUSE_RELEASED&&GetMouseX()<=32&&GetMouseX()>=0&&GetMouseY()>=HEIGHT-128;
	}
	bool RightReleased(){
		return GetKey(D).bReleased||GetKey(RIGHT).bReleased||GetKey(NP6).bReleased||MOUSE_RELEASED&&GetMouseX()<=128&&GetMouseX()>=96&&GetMouseY()>=HEIGHT-128;
	}
};


int main()
{
	SeasonI demo;
	if (demo.Construct(WIDTH, HEIGHT, 4, 4))
		demo.Start();

	return 0;
}