#include "olcPGEX_TransformedView.h"
#include "Crawler.h"
#include "olcUTIL_Camera2D.h"
#include "DamageNumber.h"
#include "DEFINES.h"

//192x192
const vi2d WINDOW_SIZE={24*8,24*8};
std::map<AnimationState,Animate2D::FrameSequence>ANIMATION_DATA;
std::vector<Monster>MONSTER_LIST;
std::vector<MonsterSpawner>SPAWNER_LIST;
std::vector<DamageNumber>DAMAGENUMBER_LIST;
Crawler*game;

Crawler::Crawler()
{
	sAppName = "Crawler Concept";
	game=this;
}

bool Crawler::OnUserCreate(){
	//Initialize Camera.
	camera=Camera2D{WINDOW_SIZE};
	camera.SetMode(olc::utils::Camera2D::Mode::LazyFollow);
	camera.SetTarget(player.GetPos());
	camera.SetWorldBoundary({0,0},WORLD_SIZE*24);
	camera.EnableWorldBoundary(true);

	//Graphics
	GFX_Pl_Sheet.Load("assets/nico-warrior.png");
	GFX_Slime_Sheet.Load("assets/slime.png");
	GFX_Circle.Load("assets/circle.png");
	GFX_Effect_GroundSlam_Back.Load("assets/ground-slam-attack-back.png");
	GFX_Effect_GroundSlam_Front.Load("assets/ground-slam-attack-front.png");

	//Animations
	InitializeAnimations();

	player.AddAnimation(AnimationState::WALK_N);
	player.AddAnimation(AnimationState::WALK_E);
	player.AddAnimation(AnimationState::WALK_S);
	player.AddAnimation(AnimationState::WALK_W);
	player.AddAnimation(AnimationState::IDLE_N);
	player.AddAnimation(AnimationState::IDLE_E);
	player.AddAnimation(AnimationState::IDLE_S);
	player.AddAnimation(AnimationState::IDLE_W);
	view=TileTransformedView{GetScreenSize(),{1,1}};

	player.SetPos({4*24,4*24});
	player.UpdateAnimation(AnimationState::IDLE_S);


	SPAWNER_LIST.push_back(MonsterSpawner({336,96},4*24,{{{MonsterName::SLIME_BLUE,{-32,-40}},{MonsterName::SLIME_GREEN,{64,20}}}}));
	std::vector<std::pair<MonsterName,vf2d>>circleSpawn;
	for(int i=0;i<12;i++){
		float angle=(2*PI)*(i/12.f);
		switch(i%3){
			case 0:{
				circleSpawn.push_back({MonsterName::SLIME_BLUE,{cos(angle)*32,sin(angle)*32}});
			}break;
			case 1:{
				circleSpawn.push_back({MonsterName::SLIME_GREEN,{cos(angle)*32,sin(angle)*32}});
			}break;
			case 2:{
				circleSpawn.push_back({MonsterName::SLIME_RED,{cos(angle)*32,sin(angle)*32}});
			}break;
		}
	}
	SPAWNER_LIST.push_back(MonsterSpawner({540,96},4*24,circleSpawn));

	return true;
}

bool Crawler::OnUserUpdate(float fElapsedTime){
	HandleUserInput(fElapsedTime);
	UpdateEffects(fElapsedTime);
	player.Update(fElapsedTime);
	for(Monster&m:MONSTER_LIST){
		m.Update(fElapsedTime);
	}
	UpdateCamera(fElapsedTime);
	RenderWorld(fElapsedTime);
	RenderHud();
	return true;
}

void Crawler::InitializeAnimations(){
	Animate2D::FrameSequence pl_walk_s{0.2};
	pl_walk_s.AddFrame({&GFX_Pl_Sheet,{vi2d{0,0}*24,{24,24}}});
	pl_walk_s.AddFrame({&GFX_Pl_Sheet,{vi2d{1,0}*24,{24,24}}});
	pl_walk_s.AddFrame({&GFX_Pl_Sheet,{vi2d{0,0}*24,{24,24}}});
	pl_walk_s.AddFrame({&GFX_Pl_Sheet,{vi2d{2,0}*24,{24,24}}});
	ANIMATION_DATA[AnimationState::WALK_S]=pl_walk_s;
	Animate2D::FrameSequence pl_walk_e{0.2};
	pl_walk_e.AddFrame({&GFX_Pl_Sheet,{vi2d{0,3}*24,{24,24}}});
	pl_walk_e.AddFrame({&GFX_Pl_Sheet,{vi2d{1,3}*24,{24,24}}});
	pl_walk_e.AddFrame({&GFX_Pl_Sheet,{vi2d{0,3}*24,{24,24}}});
	pl_walk_e.AddFrame({&GFX_Pl_Sheet,{vi2d{2,3}*24,{24,24}}});
	ANIMATION_DATA[AnimationState::WALK_E]=pl_walk_e;
	Animate2D::FrameSequence pl_walk_w{0.2};
	pl_walk_w.AddFrame({&GFX_Pl_Sheet,{vi2d{0,2}*24,{24,24}}});
	pl_walk_w.AddFrame({&GFX_Pl_Sheet,{vi2d{1,2}*24,{24,24}}});
	pl_walk_w.AddFrame({&GFX_Pl_Sheet,{vi2d{0,2}*24,{24,24}}});
	pl_walk_w.AddFrame({&GFX_Pl_Sheet,{vi2d{2,2}*24,{24,24}}});
	ANIMATION_DATA[AnimationState::WALK_W]=pl_walk_w;
	Animate2D::FrameSequence pl_walk_n{0.2};
	pl_walk_n.AddFrame({&GFX_Pl_Sheet,{vi2d{0,1}*24,{24,24}}});
	pl_walk_n.AddFrame({&GFX_Pl_Sheet,{vi2d{1,1}*24,{24,24}}});
	pl_walk_n.AddFrame({&GFX_Pl_Sheet,{vi2d{0,1}*24,{24,24}}});
	pl_walk_n.AddFrame({&GFX_Pl_Sheet,{vi2d{2,1}*24,{24,24}}});
	ANIMATION_DATA[AnimationState::WALK_N]=pl_walk_n;
	Animate2D::FrameSequence pl_idle_s;
	pl_idle_s.AddFrame({&GFX_Pl_Sheet,{vi2d{0,0}*24,{24,24}}});
	ANIMATION_DATA[AnimationState::IDLE_S]=pl_idle_s;
	Animate2D::FrameSequence pl_idle_e;
	pl_idle_e.AddFrame({&GFX_Pl_Sheet,{vi2d{0,3}*24,{24,24}}});
	ANIMATION_DATA[AnimationState::IDLE_E]=pl_idle_e;
	Animate2D::FrameSequence pl_idle_w;
	pl_idle_w.AddFrame({&GFX_Pl_Sheet,{vi2d{0,2}*24,{24,24}}});
	ANIMATION_DATA[AnimationState::IDLE_W]=pl_idle_w;
	Animate2D::FrameSequence pl_idle_n;
	pl_idle_n.AddFrame({&GFX_Pl_Sheet,{vi2d{0,1}*24,{24,24}}});
	ANIMATION_DATA[AnimationState::IDLE_N]=pl_idle_n;
	//Load slime animations.
	for(int slime=0;slime<4;slime++){
		for(int state=0;state<5;state++){
			Animate2D::FrameSequence anim;
			if(state==4){//These are death animations.
				anim=Animate2D::FrameSequence(0.1f,Animate2D::Style::OneShot);
			}
			for (int frame=0;frame<10;frame++){
				anim.AddFrame({&GFX_Slime_Sheet,{vi2d{frame,state+5*slime}*24,{24,24}}});
			}
			ANIMATION_DATA[AnimationState(AnimationState::GREEN_SLIME_IDLE+state+slime*5)]=anim;
		}
	}
	Animate2D::FrameSequence effect_groundslam_back(0.02f,Animate2D::Style::OneShot),effect_groundslam_front(0.02f,Animate2D::Style::OneShot);
	for(int i=0;i<5;i++){
		effect_groundslam_back.AddFrame({&GFX_Effect_GroundSlam_Back,{{i*64,0},{64,64}}});
		effect_groundslam_front.AddFrame({&GFX_Effect_GroundSlam_Front,{{i*64,0},{64,64}}});
	}
	ANIMATION_DATA[AnimationState::GROUND_SLAM_ATTACK_BACK]=effect_groundslam_back;
	ANIMATION_DATA[AnimationState::GROUND_SLAM_ATTACK_FRONT]=effect_groundslam_front;
}

bool Crawler::LeftHeld(){
	return GetKey(LEFT).bHeld|GetKey(A).bHeld;
}
bool Crawler::RightHeld(){
	return GetKey(RIGHT).bHeld|GetKey(D).bHeld;
}
bool Crawler::UpHeld(){
	return GetKey(UP).bHeld|GetKey(W).bHeld;
}
bool Crawler::DownHeld(){
	return GetKey(DOWN).bHeld|GetKey(S).bHeld;
}
bool Crawler::LeftReleased(){
	return GetKey(LEFT).bReleased|GetKey(A).bReleased;
}
bool Crawler::RightReleased(){
	return GetKey(RIGHT).bReleased|GetKey(D).bReleased;
}
bool Crawler::UpReleased(){
	return GetKey(UP).bReleased|GetKey(W).bReleased;
}
bool Crawler::DownReleased(){
	return GetKey(DOWN).bReleased|GetKey(S).bReleased;
}

void Crawler::HandleUserInput(float fElapsedTime){
	bool setIdleAnimation=true;
	if(GetKey(SPACE).bPressed&&player.GetState()==State::NORMAL&&player.GetGroundSlamCooldown()==0){
		player.Spin(Player::GROUND_SLAM_SPIN_TIME,14*PI);
	}
	if(RightHeld()){
		if(player.GetPos().x+12+fElapsedTime*100*player.GetMoveSpdMult()<WORLD_SIZE.x*24){
			player.SetX(player.GetX()+fElapsedTime*100*player.GetMoveSpdMult());
		} else {
			player.SetX(WORLD_SIZE.x*24-12);
		}
		player.SetFacingDirection(RIGHT);
		player.UpdateAnimation(AnimationState::WALK_E);
		setIdleAnimation=false;
	}
	if(LeftHeld()){
		if(player.GetPos().x-12+fElapsedTime*100*player.GetMoveSpdMult()>0){
			player.SetX(player.GetX()-fElapsedTime*100*player.GetMoveSpdMult());
		} else {
			player.SetX(12);
		}
		if(setIdleAnimation){
			player.SetFacingDirection(LEFT);
			player.UpdateAnimation(AnimationState::WALK_W);
		}
		setIdleAnimation=false;
	}
	if(UpHeld()){
		if(player.GetPos().y-12+fElapsedTime*100*player.GetMoveSpdMult()>0){
			player.SetY(player.GetY()-fElapsedTime*100*player.GetMoveSpdMult());
		} else {
			player.SetY(12);
		}
		if(setIdleAnimation){
			player.SetFacingDirection(UP);
			player.UpdateAnimation(AnimationState::WALK_N);
		}
		setIdleAnimation=false;
	}
	if(DownHeld()){
		if(player.GetPos().y+12+fElapsedTime*100*player.GetMoveSpdMult()<WORLD_SIZE.y*24){
			player.SetY(player.GetY()+fElapsedTime*100*player.GetMoveSpdMult());
		} else {
			player.SetY(WORLD_SIZE.y*24-12);
		}
		if(setIdleAnimation){
			player.SetFacingDirection(DOWN);
			player.UpdateAnimation(AnimationState::WALK_S);
		}
		setIdleAnimation=false;
	}

	if(UpReleased()){
		player.SetLastReleasedMovementKey(UP);
		if(RightHeld()){
			player.UpdateAnimation(AnimationState::WALK_E);
		} else
		if(DownHeld()){
			player.UpdateAnimation(AnimationState::WALK_S);
		} else
		if(LeftHeld()){
			player.UpdateAnimation(AnimationState::WALK_W);
		}
	}
	if(RightReleased()){
		player.SetLastReleasedMovementKey(RIGHT);
		if(UpHeld()){
			player.UpdateAnimation(AnimationState::WALK_N);
		} else
		if(DownHeld()){
			player.UpdateAnimation(AnimationState::WALK_S);
		} else
		if(LeftHeld()){
			player.UpdateAnimation(AnimationState::WALK_W);
		}
	}
	if(LeftReleased()){
		player.SetLastReleasedMovementKey(LEFT);
		if(RightHeld()){
			player.UpdateAnimation(AnimationState::WALK_E);
		} else
		if(DownHeld()){
			player.UpdateAnimation(AnimationState::WALK_S);
		} else
		if(UpHeld()){
			player.UpdateAnimation(AnimationState::WALK_N);
		}
	}
	if(DownReleased()){
		player.SetLastReleasedMovementKey(DOWN);
		if(RightHeld()){
			player.UpdateAnimation(AnimationState::WALK_E);
		} else
		if(UpHeld()){
			player.UpdateAnimation(AnimationState::WALK_N);
		} else
		if(LeftHeld()){
			player.UpdateAnimation(AnimationState::WALK_W);
		}
	}

	if(setIdleAnimation){
		switch(player.GetLastReleasedMovementKey()){
			case UP:{
				player.UpdateAnimation(AnimationState::IDLE_N);
			}break;
			case DOWN:{
				player.UpdateAnimation(AnimationState::IDLE_S);
			}break;
			case LEFT:{
				player.UpdateAnimation(AnimationState::IDLE_W);
			}break;
			case RIGHT:{
				player.UpdateAnimation(AnimationState::IDLE_E);
			}break;
			default:{
				player.UpdateAnimation(AnimationState::IDLE_S);
			}
		}
	} else {
		//We have moved.
		player.Moved();
	}
}

void Crawler::UpdateCamera(float fElapsedTime){
	camera.Update(fElapsedTime);
	view.SetWorldOffset(vi2d(camera.GetViewPosition()));
}

void Crawler::UpdateEffects(float fElapsedTime){
	for(std::vector<Effect>::iterator it=backgroundEffects.begin();it!=backgroundEffects.end();++it){
		Effect&e=*it;
		if(!e.Update(fElapsedTime)){
			it=backgroundEffects.erase(it);
			if(it==backgroundEffects.end()){
				break;
			}
		}
	}
	for(std::vector<Effect>::iterator it=foregroundEffects.begin();it!=foregroundEffects.end();++it){
		Effect&e=*it;
		if(!e.Update(fElapsedTime)){
			it=foregroundEffects.erase(it);
			if(it==foregroundEffects.end()){
				break;
			}
		}
	}
}
void Crawler::HurtEnemies(vf2d pos,float radius,int damage){
	for(Monster&m:MONSTER_LIST){
		if(geom2d::overlaps(geom2d::circle(pos,radius),geom2d::circle(m.GetPos(),12*m.GetSizeMult()))){
			m.Hurt(damage);
		}
	}
}

void Crawler::RenderWorld(float fElapsedTime){
	Clear({100,180,100});
	for (int x = view.GetTopLeftTile().x/24-1; x <= view.GetBottomRightTile().x/24; x++){
		for (int y = view.GetTopLeftTile().y/24-1; y <= view.GetBottomRightTile().y/24; y++){
			view.DrawRect(vi2d{x,y}*24,{24,24},VERY_DARK_GREY);
		}
	}
	std::vector<Monster>monstersBefore,monstersAfter;
	Player&pl=player;
	std::copy_if(MONSTER_LIST.begin(),MONSTER_LIST.end(),std::back_inserter(monstersBefore),[&pl](Monster&m){return m.GetPos().y<pl.GetPos().y;});
	std::copy_if(MONSTER_LIST.begin(),MONSTER_LIST.end(),std::back_inserter(monstersAfter),[&pl](Monster&m){return m.GetPos().y>=pl.GetPos().y;});
	std::sort(monstersBefore.begin(),monstersBefore.end(),[](Monster&m1,Monster&m2){return m1.GetPos().y<m2.GetPos().y;});
	std::sort(monstersAfter.begin(),monstersAfter.end(),[](Monster&m1,Monster&m2){return m1.GetPos().y<m2.GetPos().y;});
	if(player.GetZ()>0){
		vf2d shadowScale=vf2d{8/3.f,1}/std::max(1.f,player.GetZ()/4);
		view.DrawDecal(player.GetPos()-vf2d{3,3}*shadowScale/2+vf2d{0,6},GFX_Circle.Decal(),shadowScale);
	}
	for(Effect&e:backgroundEffects){
		e.Draw();
	}
	for(Monster&m:monstersBefore){
		view.DrawPartialDecal(m.GetPos()-vf2d{12,12}*m.GetSizeMult(),m.GetFrame().GetSourceImage()->Decal(),m.GetFrame().GetSourceRect().pos,m.GetFrame().GetSourceRect().size,vf2d(m.GetSizeMult(),m.GetSizeMult()));
	}
	view.DrawPartialRotatedDecal(player.GetPos()+vf2d{0,-player.GetZ()},player.GetFrame().GetSourceImage()->Decal(),player.GetSpinAngle(),{12,12},player.GetFrame().GetSourceRect().pos,player.GetFrame().GetSourceRect().size,vf2d(player.GetSizeMult(),player.GetSizeMult()));
	for(Monster&m:monstersAfter){
		view.DrawPartialDecal(m.GetPos()-vf2d{12,12}*m.GetSizeMult(),m.GetFrame().GetSourceImage()->Decal(),m.GetFrame().GetSourceRect().pos,m.GetFrame().GetSourceRect().size,vf2d(m.GetSizeMult(),m.GetSizeMult()));
	}
	for(Effect&e:foregroundEffects){
		e.Draw();
	}
	for(std::vector<DamageNumber>::iterator it=DAMAGENUMBER_LIST.begin();it!=DAMAGENUMBER_LIST.end();++it){
		DamageNumber&dn=*it;
		dn.lifeTime+=fElapsedTime;
		if(dn.lifeTime>1){
			it=DAMAGENUMBER_LIST.erase(it);
			if(it==DAMAGENUMBER_LIST.end()){
				break;
			}
		} else {
			if(dn.lifeTime<DamageNumber::MOVE_UP_TIME){
				dn.pos.y-=20*fElapsedTime;
			}
			std::string text=std::to_string(dn.damage);
			view.DrawStringPropDecal(dn.pos-GetTextSizeProp(text)/2,text,DARK_RED);
		}
	}
}

void Crawler::RenderHud(){
	if(player.GetGroundSlamCooldown()>0){
		FillRectDecal({10,ScreenHeight()-22.f},{64,6},BLACK);
		FillRectDecal({11,ScreenHeight()-21.f},{62,4},DARK_GREY);
		GradientFillRectDecal({10,ScreenHeight()-22.f},{(player.GetGroundSlamCooldown()/Player::GROUND_SLAM_COOLDOWN)*64,6},VERY_DARK_RED,VERY_DARK_RED,DARK_RED,DARK_RED);
		DrawRotatedStringPropDecal(vf2d{8,ScreenHeight()-20.f}+vf2d{1,1},"G R O U N D  S L A M",-PI/64,{0,0},BLACK,{0.4,0.4});
		DrawRotatedStringPropDecal({8,ScreenHeight()-20.f},"G R O U N D  S L A M",-PI/64,{0,0},WHITE,{0.4,0.4});
	}
}

void Crawler::AddEffect(Effect foreground,Effect background){
	foregroundEffects.push_back(foreground);
	backgroundEffects.push_back(background);
}

vf2d Crawler::GetWorldMousePos(){
	return GetMousePos()+view.GetWorldOffset();
}

int main()
{
	Crawler demo;
	if (demo.Construct(WINDOW_SIZE.x, WINDOW_SIZE.y, 4, 4))
		demo.Start();

	return 0;
}