#include <imgui.h>
#include "pch.h"
#include "WorldLevel.h"

#include <algorithm>
#include <iostream>
#include <ostream>

#include "Collision.h"
#include "colors.h"
#include "GroundTileTypeManager.h"
#include "utils.h"
#include "GridSystem/WorldTile.h"
#include "Gui/Screens/ScreenManager.h"


class GroundTileType;
WorldLevel::WorldLevel(Camera* camera, Rectf viewport): Level(camera),
                                                        m_gridManager(WorldGridManager()),
                                                        m_player(Player { Vector2f { 0, 100 }, TextureManager::GetInstance() }),
                                                        m_mousePos { 0, 0 },
                                                        m_viewport(viewport),
                                                        m_screenManager(ScreenManager::GetInstance()) {
	InitializeGroundTiles();
	// The grid is 34 x 50 big, the top center is 0,0 in world coords
	for (int x { 0 }; x < WORLD_WIDTH; ++x) {
		for (int y { 0 }; y < WORLD_HEIGHT; ++y) {
			const int actualX = x - WORLD_WIDTH / 2;
			Vector2f pos = Vector2f { float(actualX * TILE_WIDTH), -float(y * TILE_HEIGHT) - TILE_HEIGHT };
			GroundTileType* type = getRandomGroundTile();
			m_gridManager.SetTileAtIndex(x, y, new WorldTile { pos, type, TextureManager::GetInstance(), &m_gridManager });
		}
	}
	for (int x { 0 }; x < WORLD_WIDTH; ++x) {
		m_gridManager.GetTileAtIndex(x, 0)->SetTileType(GroundTileTypeManager::GetInstance()->AIR);
		m_gridManager.GetTileAtIndex(x, 1)->SetTileType(GroundTileTypeManager::GetInstance()->GRASS);
	}
	m_refeulBuilding = new Building { "buildings/fuelStation.png", Vector2f { -700, -52 }, TextureManager::GetInstance() };
	m_gridManager.GetTileAtWorldPos(Vector2f { -700, -50 })->SetTileType(GroundTileTypeManager::GetInstance()->HARD_LEFT);
	m_gridManager.GetTileAtWorldPos(Vector2f { -650, -50 })->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f { -600, -50 })->SetTileType(GroundTileTypeManager::GetInstance()->HARD_RIGHT);

	m_mineralBuilding = new Building { "buildings/mineralStation.png", Vector2f { -350, -52 }, TextureManager::GetInstance() };
	m_gridManager.GetTileAtWorldPos(Vector2f {-400, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_LEFT);
	m_gridManager.GetTileAtWorldPos(Vector2f {-350, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {-300, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {-250, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {-200, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {-150, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_RIGHT);

	m_junkBuilding = new Building { "buildings/junkStation.png", Vector2f { 250, -52 }, TextureManager::GetInstance() };
	m_gridManager.GetTileAtWorldPos(Vector2f {200, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_LEFT);
	m_gridManager.GetTileAtWorldPos(Vector2f {250, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {300, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {350, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {400, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {450, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_RIGHT);

	m_repairBuilding = new Building { "buildings/repairStation.png", Vector2f { 700, -52 }, TextureManager::GetInstance() };
	m_gridManager.GetTileAtWorldPos(Vector2f {650, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_LEFT);
	m_gridManager.GetTileAtWorldPos(Vector2f {700, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {750, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_MIDDLE);
	m_gridManager.GetTileAtWorldPos(Vector2f {800, -50})->SetTileType(GroundTileTypeManager::GetInstance()->HARD_RIGHT);


}
WorldLevel::~WorldLevel() {
	delete m_refeulBuilding;
	delete m_mineralBuilding;
	delete m_junkBuilding;
	delete m_repairBuilding;
	//delete m_pTextTexture;
}
void WorldLevel::Update(float elapsedSec) {
	m_fps = 1 / elapsedSec;

	int mouseX, mouseY;
	SDL_GetMouseState(&mouseX, &mouseY);
	m_mousePos = Vector2f { float(mouseX), float(mouseY) };
	m_mousePos = m_pCamera->TransformMouse(m_mousePos);

	// for (size_t x { 0 }; x < WORLD_WIDTH; ++x) {
	// 	for (size_t y { 0 }; y < WORLD_HEIGHT; ++y) {
	// 		m_gridManager.GetTileAtIndex(x, y)->m_Hightlight = false;
	// 	}
	// }
	
	for (size_t x { 0 }; x < WORLD_WIDTH; ++x) {
		for (size_t y { 0 }; y < WORLD_HEIGHT; ++y) {
			m_gridManager.GetTileAtIndex(x, y)->Update(m_pCamera);
			if (m_gridManager.GetTileAtIndex(x, y)->GetCollisionRect().Contains(m_mousePos)) {
				m_pSelectedTile = m_gridManager.GetTileAtIndex(x, y);
				//selectedTile->m_Hightlight = true;
			}
		}
	}
	if (m_pSelectedTile != nullptr) {
		if (utils::isMouseDown(SDL_BUTTON_LEFT)) {
			m_pSelectedTile->SetTileType(GroundTileTypeManager::GetInstance()->AIR);
		}
	}

	//Get the diogonal tiles of the selected tile
	// if (m_pSelectedTile != nullptr) {
	// 	surroundingTiles surroundingTiles = m_gridManager.GetSurroundingTiles(m_pSelectedTile);
	// 	TileDirection direction = TileDirection::TopMiddle;
	// 	const std::array<TileDirection, 8> directions = { TileDirection::TopLeft, TileDirection::TopMiddle, TileDirection::TopRight, TileDirection::MiddleLeft,
	// 		TileDirection::MiddleRight, TileDirection::BottomLeft, TileDirection::BottomMiddle, TileDirection::BottomRight };
	//
	// 	for (int i = 0; i < 8; i++) {
	// 		direction = directions[i];
	// 		if (surroundingTiles.GetTile(direction) != nullptr) {
	// 			if (surroundingTiles.GetTile(direction)->GetTileType() != GroundTileTypeManager::GetInstance()->AIR) {
	// 				//surroundingTiles.GetTile(direction)->m_Hightlight = true;
	// 			}
	// 		}
	// 	}
	// }

	m_player.Update(elapsedSec, *this);

	//Move the camera when the player gets to the edge
	if(m_FollowPlayer) {
		Vector2f playerPos = m_player.GetPosition();
		Vector2f newCameraPos = m_pCamera->GetPosition();
		if (playerPos.x < newCameraPos.x + 50) {
			newCameraPos.x = playerPos.x - 50;
		}
		if (playerPos.x > newCameraPos.x + m_viewport.width - 100) {
			newCameraPos.x = playerPos.x - m_viewport.width + 100;
		}
		if (playerPos.y < newCameraPos.y + 50) {
			newCameraPos.y = playerPos.y - 50;
		}
		if (playerPos.y > newCameraPos.y + m_viewport.height - 50) {
			newCameraPos.y = playerPos.y - m_viewport.height + 50;
		}
		m_pCamera->SetPosition(newCameraPos);
	}		

	Screen* screen = m_screenManager->GetCurrentScreen();
	if (screen != nullptr) {
		screen->Update(elapsedSec);
	}

	//Vector2f playerPos = m_player.GetPosition();
	//Vector2f newCameraPos = playerPos;
	//m_pCamera->SetPosition(newCameraPos);

	//place the player in the center of the camera
	//m_pCamera->SetPosition(Vector2f{playerPos.x - m_viewport.width / 2, playerPos.y - m_viewport.height / 2});
}
void WorldLevel::Draw() const {
	m_pCamera->BeginRendering();

	for (Collision::CollisionRect rect : m_Rects) {
		utils::DrawRect(rect.pos, rect.size.x, rect.size.y);
	}

	utils::SetColor(Colors::WHITE);
	utils::FillEllipse(m_mousePos, 2, 2);

	for (size_t x { 0 }; x < WORLD_WIDTH; ++x) {
		for (size_t y { 0 }; y < WORLD_HEIGHT; ++y) {
			m_gridManager.GetTileAtIndex(x, y)->Draw();
		}
	}

	utils::SetColor(Colors::MAGENTA);
	utils::FillEllipse(-5, -5, 5, 5);

	if (m_pSelectedTile != nullptr) {
		m_pSelectedTile->Draw();
	}

	m_refeulBuilding->Draw();
	m_mineralBuilding->Draw();
	m_junkBuilding->Draw();
	m_repairBuilding->Draw();
	m_player.Draw();

	utils::SetColor(Colors::GREEN);
	utils::DrawArrow(Vector2f{0, 0}, m_mousePos);

	m_pCamera->EndRendering();

	utils::FillRect(utils::GetMousePos(), 10, 10);

	utils::DrawRect(50, 50, m_viewport.width - 100,  m_viewport.height - 100);

	const Screen* screen = m_screenManager->GetCurrentScreen();
	if (screen != nullptr) {
		screen->Draw();
	}
}
void WorldLevel::MouseMove(const Vector2f& mousePos) {
	m_mousePos = mousePos;
}

void WorldLevel::ProcessImGui() {
	ImGui::Begin("Selected Tile");
	std::string tileType = "None";
	if (m_pSelectedTile != nullptr) {
		switch (m_pSelectedTile->GetTileType()->getType()) {
		case GroundTileTypes::Air:
			tileType = "Air";
			break;
		case GroundTileTypes::Dirt:
			tileType = "Dirt";
			break;
		case GroundTileTypes::Iron:
			tileType = "Iron";
			break;
		case GroundTileTypes::Hard:
			tileType = "Hard";
			break;
		case GroundTileTypes::Stone:
			tileType = "Stone";
			break;
		default:
			tileType = "Unknown";
			break;
		}
	}
	ImGui::Text("Selected Tile: %s", tileType.c_str());
	ImGui::End();
	if (ImGui::BeginMainMenuBar()) {
		if (ImGui::BeginMenu("Properties")) {
			if (ImGui::MenuItem("TextureManager Info")) {
				m_ShowTextureManagerWindow = !m_ShowTextureManagerWindow;
			}
			if (ImGui::MenuItem("Camera Info")) {
				m_ShowCameraWindow = !m_ShowCameraWindow;
			}
			if (ImGui::MenuItem("Player Info")) {
				m_ShowPlayerInfo = !m_ShowPlayerInfo;
			}
			ImGui::EndMenu();
		}
		if (ImGui::BeginMenu("Screens")) {
			if (ImGui::MenuItem("Open Fuel screen")) {
				ScreenManager::GetInstance()->OpenScreen(ScreenManager::m_FuelScreen);
			}


			if (ImGui::MenuItem("Open Sell screen")) {
				ScreenManager::GetInstance()->OpenScreen(ScreenManager::m_SellScreen);
			}

			if (ImGui::MenuItem("Close Screen")) {
				ScreenManager::GetInstance()->CloseScreen();
			}

			ImGui::EndMenu();
		}
		const Vector2f screenPos = utils::GetMousePos();
		const std::string mousePos = "Mouse Pos: (" + std::to_string(screenPos.x) + ", " + std::to_string(screenPos.y) + ")";
		if (ImGui::BeginMenu(mousePos.c_str())) {
			ImGui::EndMenu();
		}

		if (ImGui::BeginMenu(std::to_string(m_fps).c_str())) {
			ImGui::EndMenu();
		}

		ImGui::EndMainMenuBar();
	}

	if (m_ShowTextureManagerWindow) {
		ImGui::Begin("Texture Manager", &m_ShowTextureManagerWindow, ImGuiWindowFlags_AlwaysAutoResize);
		ImGui::Text("Texture loading Count:");
		ImGui::SameLine();
		ImGui::TextColored(ImVec4(1.0f, 0.0f, 1.0f, 1.0f), std::to_string(Texture::m_TextureCounter).c_str());
		ImGui::End();
	}

	if (m_ShowCameraWindow) {
		ImGui::Begin("Camera", &m_ShowCameraWindow, ImGuiWindowFlags_AlwaysAutoResize);
		ImGui::Text("Camera Position: (%f, %f)", m_pCamera->GetPosition().x, m_pCamera->GetPosition().y);
		ImGui::Text("Camera Scale: %f", m_pCamera->GetScale());
		ImGui::Text("Is Right Mouse Down: %s", utils::isMouseDown(0) ? "true" : "false");
		if (ImGui::Button("Reset Camera")) {
			m_pCamera->SetPosition(Vector2f { -m_viewport.width / 2, -m_viewport.height / 2 });
			m_pCamera->SetScale(1.0f);
		}
		ImGui::Checkbox("Follow Player", &m_FollowPlayer);
		ImGui::End();
	}

	if (m_ShowPlayerInfo) {
		ImGui::Begin("Player Info", &m_ShowPlayerInfo, ImGuiWindowFlags_AlwaysAutoResize);
		ImGui::Text("Player Position: (%f, %f)", m_player.GetPosition().x, m_player.GetPosition().y);
		ImGui::Text("Player Velocity: (%f, %f)", m_player.GetVelocity().x, m_player.GetVelocity().y);
		ImGui::End();
		m_player.ProcessImGui();
	}
}