/*
Copyright (C) 2004 Parallel Realities

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/

#include "mapEditor.h"

void drawMap(int mapX, int mapY)
{
	SDL_Rect r;
	int tile = 0;

	for (int x = 0 ; x < 20 ; x++)
	{
		for (int y = 0 ; y < 15 ; y++)
		{
			r.x = x * BRICKSIZE;
			r.y = y * BRICKSIZE;
			r.w = r.h = BRICKSIZE - 1;
			
			tile = map.data[x + mapX][y + mapY];
			
			if (tile == MAP_WATERANIM)
			{
				tile = graphics.getWaterAnim();
			}

			if (tile > 0)
			{
				graphics.blit(graphics.tile[tile], r.x, r.y, graphics.screen, false);
				
				if ((tile >= MAP_NORESET) && (tile < MAP_DECORATION))
				{
					graphics.drawRect(r.x, r.y, 32, 4, graphics.yellow, graphics.screen);
				}
			}
		}
	}
}

void showMap(int *mapX, int *mapY)
{
	SDL_FillRect(graphics.screen, NULL, graphics.black);

	engine.keyState[SDLK_SPACE] = 0;
	int moveTimer = 0;

	while (true)
	{
		graphics.updateScreen();
		engine.getInput();

		if (engine.keyState[SDLK_SPACE])
			break;
			
		for (int x = 0 ; x < MAPWIDTH ; x++)
		{
			for (int y = 0 ; y < MAPHEIGHT ; y++)
			{
				switch (map.data[x][y])
				{
					case 0:
						graphics.putPixel(x, y, 9, graphics.screen);
						break;
					case 1:
						graphics.putPixel(x, y,graphics.blue, graphics.screen);
						break;
					case 2:
						graphics.putPixel(x, y, graphics.green, graphics.screen);
						break;
					case 3:
						graphics.putPixel(x, y, graphics.red, graphics.screen);
						break;
					default:
						graphics.putPixel(x, y, graphics.white, graphics.screen);
						break;
				}
			}
		}

		if (moveTimer == 0)
		{
			if (engine.keyState[SDLK_LEFT])
				*mapX -= 1;

			if (engine.keyState[SDLK_RIGHT])
				*mapX += 1;

			if (engine.keyState[SDLK_UP])
				*mapY -= 1;

			if (engine.keyState[SDLK_DOWN])
				*mapY += 1;
		}

		moveTimer--;
		Math::limitInt(&moveTimer, 0, 60);

		graphics.drawRect(*mapX, *mapY, 20, 1, graphics.green, graphics.screen);
		graphics.drawRect(*mapX, *mapY, 1, 15, graphics.green, graphics.screen);
		graphics.drawRect(*mapX, *mapY + 15, 20, 1, graphics.green, graphics.screen);
		graphics.drawRect(*mapX + 20, *mapY, 1, 15, graphics.green, graphics.screen);
	}

	engine.keyState[SDLK_SPACE] = 0;
}

int nextBlock(int current, int dir)
{
	if ((current + dir) > 255)
		return current;

	if ((current + dir) < 0)
		return current;

	int wanted = current;

	while (true)
	{
		wanted += dir;

		if (wanted < 0) return current;
		if (wanted > 255) return current;

		if (graphics.tile[wanted])
			return wanted;
	}
}

void drawEnemies()
{
	Entity *enemy = (Entity*)map.enemyList.getHead();

	int x, y, absX, absY;

	while (enemy->next != NULL)
	{
		enemy = (Entity*)enemy->next;

		if (enemy->owner != enemy)
		{
			enemy->face = enemy->owner->face;
			(enemy->face == 0) ? enemy->x = enemy->owner->x + enemy->tx : enemy->x = enemy->owner->x + -enemy->tx;
			enemy->y = enemy->owner->y + enemy->ty;
		}

		x = (int)(enemy->x - engine.playerPosX);
		y = (int)(enemy->y - engine.playerPosY);

		absX = abs(x);
		absY = abs(y);

		if ((absX < 800) && (absY < 600))
		{
			graphics.blit(enemy->getFaceImage(), x, y, graphics.screen, false);
			enemy->animate();

			if ((!(enemy->flags & ENT_WEIGHTLESS)) && (!(enemy->flags & ENT_FLIES)) && (!(enemy->flags & ENT_SWIMS)))
				enemy->applyGravity();
		}
	}
}

void deleteEnemy(int x, int y)
{
	x *= BRICKSIZE;
	y *= BRICKSIZE;

	Entity *enemy = (Entity*)map.enemyList.getHead();
	Entity *previous = enemy;

	while (enemy->next != NULL)
	{
		enemy = (Entity*)enemy->next;

		if ((enemy->x == x) && (enemy->y == y))
		{
			map.enemyList.remove(previous, enemy);
			enemy = previous;
		}
		else
		{
			previous = enemy;
		}
	}
}

void saveMap(char *name)
{
	FILE *fp = fopen(name, "wb");
	Entity *enemy = (Entity*)map.enemyList.getHead();
	String *str;

	if (fp)
	{
		for (int y = 0 ; y < MAPHEIGHT ; y++)
		{
			for (int x = 0 ; x < MAPWIDTH ; x++)
			{
				fprintf(fp, "%d ", map.data[x][y]);
			}
			fprintf(fp, "\n");
		}

		str = stringHead->next;
		while (str != NULL)
		{
			fprintf(fp, "%s", str->string);
			str = str->next;
		}
		
		while (enemy->next != NULL)
		{
			enemy = (Entity*)enemy->next;
			fprintf(fp, "EMH ENEMY \"%s\" %d %d\n", enemy->name, (int)enemy->x, (int)enemy->y);
		}

		fprintf(fp, "@EOF@\n");

		fclose(fp);
	}
	
	printf("Saved %s (%d)\n", name, SDL_GetTicks());
}

void collectMapData()
{
	FILE *fp = fopen(game.mapName, "rb");
	if (!fp)
		return;

	char string[1024];
	String *str;

	for (int y = 0 ; y < 300 ; y++)
		fgets(string, 1024, fp);

	while (true)
	{
		fgets(string, 1024, fp);
		printf("Read: %s", string);

		if (strstr(string, "@EOF@"))
			break;

		if (!strstr(string, " ENEMY \""))
		{
			str = new String;
			strcpy(str->string, string);
			stringTail->next = str;
			stringTail = str;
		}
		
		if (strstr(string, "TILESET gfx/ancient"))
		{
			for (int x = 0 ; x < MAPWIDTH ; x++)
			{
				for (int y = 0 ; y < MAPHEIGHT ; y++)
				{
					if ((map.data[x][y] >= 9) && (map.data[x][y] <= 20) && ((rand() % 2) == 0))
						map.data[x][y] = Math::rrand(9, 20);
				}
			}
		}
		
		if (strstr(string, "TILESET gfx/caves"))
		{
			for (int x = 0 ; x < MAPWIDTH ; x++)
			{
				for (int y = 0 ; y < MAPHEIGHT ; y++)
				{
					if ((map.data[x][y] >= 9) && (map.data[x][y] <= 20))
						map.data[x][y] = Math::rrand(9, 12);
				}
			}
		}
	}

	fclose(fp);
}

void newMap(char *name)
{
	FILE *fp = fopen(name, "wb");

	if (fp)
	{
		for (int y = 0 ; y < MAPHEIGHT ; y++)
		{
			for (int x = 0 ; x < MAPWIDTH ; x++)
			{
				fprintf(fp, "%d ", map.data[x][y]);
			}
			fprintf(fp, "\n");
		}
		
		fprintf(fp, "EMH STAGENAME \"unnamed\"\n");

		fprintf(fp, "EMH TILESET gfx/grasslands\n");
		fprintf(fp, "EMH BACKGROUND gfx/grasslands/stone.jpg\n");
		fprintf(fp, "EMH MUSIC music/viking.mod\n");

		fprintf(fp, "EMH ALPHATILES 1 2 3 200 201 202 203 204 244 245 246 -1\n");

		fprintf(fp, "EMH START 10 10\n");

		fprintf(fp, "@EOF@\n");
		
		fclose(fp);
	}
	else
	{
		exit(1);
	}
}

void addTileDecoration()
{
	printf("Adding Tile Decoration...\n");

	for (int y = 1 ; y < MAPHEIGHT ; y++)
	{
		for (int x = 0 ; x < MAPWIDTH ; x++)
		{
			if ((map.data[x][y] == 9) && (map.data[x][y - 1] == MAP_AIR))
			{
				if (rand() % 4)
					map.data[x][y - 1] = 241;
			}
		}
	}

	for (int y = 0 ; y < MAPHEIGHT ; y++)
	{
		for (int x = 0 ; x < MAPWIDTH ; x++)
		{
			if (map.data[x][y] == 241)
			{
				if ((rand() % 3) == 0)
					map.data[x][y] = 242;
			}
		}
	}
	
	engine.keyState[SDLK_F1] = 0;
}

void fillHorizontal(int block, int x, int y)
{
	bool moveLeft = true;
	bool moveRight = true;
	bool ok = true;
	
	int left = x;
	int right = x;
	
	if (map.data[x][y] == 0)
	{
		map.data[x][y] = block;
		
		while (ok)
		{
			if (moveLeft)
			{
				left--;
				if (left < 0) 
				{
					left = 0;
				}
			}
			
			if (map.data[left][y] == 0)
			{
				map.data[left][y] = block;
			}
			else
			{
				moveLeft = false;
			}
			
			if (moveRight)
			{
				right++;
				
				if (right >= MAPWIDTH)
				{
					right = MAPWIDTH - 1;
				}
			}
			
			if (map.data[right][y] == 0)
			{
				map.data[right][y] = block;
			}
			else
			{
				moveRight = false;
			}
			
			if ((!moveLeft) && (!moveRight))
			{
				ok = false;
			}
		}
	}
}

int main(int argc, char *argv[])
{
	if (argc < 2)
	{
		printf("Usage: mapeditor <filename>\n\n");
		exit(1);
	}

	atexit(cleanup);
	
	engine.useAudio = 0;

	initSystem();

	FILE *fp = fopen(argv[1], "rb");
	if (!fp)
		newMap(argv[1]);
	else
		fclose(fp);
	
	game.setMapName(argv[1]);
	
	loadResources();
	
	collectMapData();
	
	int mapX, mapY, allowMove, x, y;
	mapX = mapY = allowMove = x = y = 0;
	
	int editing = 0;
	int currentMonster = 0;
	int currentItem = 0;
	int currentBlock = 1;

	SDL_Rect r;

	int MOVESPEED = 5;
	
	char string[255];
	
	unsigned int frameLimit = SDL_GetTicks() + 16;

	while (true)
	{
		engine.getInput();
		engine.doPause();
		engine.doFrameLoop();

		graphics.updateScreen();
		graphics.animateSprites();
		graphics.drawBackground();

		engine.setPlayerPosition((mapX * 32) + 320, (mapY * 32) + 240, map.limitLeft, map.limitRight, map.limitUp, map.limitDown);

		drawMap(mapX, mapY);
		doTrains();
		doSwitches();
		doItems();
		doBullets();
		doMIAs();
		drawEnemies();
		doObstacles();
		doEffects();
		doParticles();
		doTeleporters();
		doLineDefs();
		doTraps();

		x = engine.getMouseX();
		y = engine.getMouseY();

		x /= BRICKSIZE;
		y /= BRICKSIZE;

		r.x = x * BRICKSIZE;
		r.y = y * BRICKSIZE;
		r.w = BRICKSIZE;
		r.h = BRICKSIZE;

		switch (editing)
		{
			case 0:
				graphics.drawRect(r.x - 1, r.y - 1, 34, 1, graphics.yellow, graphics.screen);
				graphics.drawRect(r.x - 1, r.y - 1, 1, 34, graphics.yellow, graphics.screen);
				graphics.drawRect(r.x + 32, r.y - 1, 1, 34, graphics.yellow, graphics.screen);
				graphics.drawRect(r.x - 1, r.y + 32, 34, 1, graphics.yellow, graphics.screen);
				graphics.blit(graphics.tile[currentBlock], r.x, r.y, graphics.screen, false);
				if (engine.mouseLeft)
					map.data[mapX + x][mapY + y] = currentBlock;
				break;
			case 1:
				graphics.blit(defEnemy[currentMonster].getFaceImage(), r.x, r.y, graphics.screen, false);
				if (engine.mouseLeft)
				{
					addEnemy(defEnemy[currentMonster].name, (mapX + x) * BRICKSIZE, (mapY + y) * BRICKSIZE, 0);
					engine.mouseLeft = 0;
				}
				break;
			case 2:
				graphics.blit(defItem[currentItem].getFaceImage(), r.x, r.y, graphics.screen, false);
				if (engine.mouseLeft)
				{
					addItem(defItem[currentItem].id, defItem[currentItem].name, (mapX + x) * BRICKSIZE, (mapY + y) * BRICKSIZE, defItem[currentItem].sprite[0]->name, 0, defItem[currentItem].value, 0, true);
					engine.mouseLeft = 0;
				}
				break;
		}

		if (engine.mouseRight)
		{
			if (editing == 0) map.data[mapX + x][mapY + y] = MAP_AIR;
			if (editing == 1) deleteEnemy(mapX + x, mapY + y);
		}

		allowMove--;
		if (allowMove < 1) allowMove = 0;

		if (allowMove == 0)
		{
			if (engine.keyState[SDLK_UP]) {mapY--; allowMove = MOVESPEED;}
			if (engine.keyState[SDLK_DOWN]) {mapY++; allowMove = MOVESPEED;}
			if (engine.keyState[SDLK_LEFT]) {mapX--; allowMove = MOVESPEED;}
			if (engine.keyState[SDLK_RIGHT]) {mapX++; allowMove = MOVESPEED;}

			if (engine.keyState[SDLK_PAGEDOWN]) {mapY += 10; allowMove = MOVESPEED;}
			if (engine.keyState[SDLK_PAGEUP]) {mapY -= 10; allowMove = MOVESPEED;}

			if (engine.keyState[SDLK_1]) editing = 0;
			if (engine.keyState[SDLK_2]) editing = 1;
			if (engine.keyState[SDLK_3]) editing = 2;
			
			if (engine.keyState[SDLK_0]) fillHorizontal(currentBlock, mapX + x, mapY + y);

			if (engine.keyState[SDLK_F1]) addTileDecoration();

			if (engine.keyState[SDLK_ESCAPE]) break;
		}

		if (engine.keyState[SDLK_PERIOD])
		{
			switch (editing)
			{
				case 0:
					currentBlock = nextBlock(currentBlock, 1);
					break;
				case 1:
					currentMonster++;
					break;
				case 2:
					currentItem++;
					break;
			}

			engine.keyState[SDLK_PERIOD] = 0;
		}

		if (engine.keyState[SDLK_COMMA])
		{
			switch (editing)
			{
				case 0:
					currentBlock = nextBlock(currentBlock, -1);
					break;
				case 1:
					currentMonster--;
					break;
				case 2:
					currentItem--;
					break;
			}

			engine.keyState[SDLK_COMMA] = 0;
		}

		Math::limitInt(&currentMonster, 0, MAX_ENEMIES - 1);
		Math::limitInt(&currentItem, 0, MAX_ITEMS - 1);

		if (defEnemy[currentMonster].sprite[0] == NULL)
			currentMonster = 0;
			
		if (defEnemy[currentItem].sprite[0] == NULL)
			currentItem = 0;

		if (engine.keyState[SDLK_SPACE]) {showMap(&mapX, &mapY);}

		if (engine.keyState[SDLK_s]) {saveMap(game.mapName); engine.keyState[SDLK_s] = 0;}

		if (mapX < 0) mapX = 0;
		if (mapY < 0) mapY = 0;
		if (mapX > MAPWIDTH - 40) mapX = MAPWIDTH - 40;
		if (mapY > MAPHEIGHT - 30) mapY = MAPHEIGHT - 30;

		if (editing == 0)
			sprintf(string, "Index : %d:%d ; Screen %d:%d ; Tile %d", mapX + x, mapY + y, (mapX + x) * BRICKSIZE, (mapY + y) * BRICKSIZE, currentBlock);
		else if (editing == 1)
			sprintf(string, "Index : %d:%d ; Screen %d:%d ; %s", mapX + x, mapY + y, (mapX + x) * BRICKSIZE, (mapY + y) * BRICKSIZE, defEnemy[currentMonster].name);
		else if (editing == 2)
			sprintf(string, "Index : %d:%d ; Screen %d:%d ; %s", mapX + x, mapY + y, (mapX + x) * BRICKSIZE, (mapY + y) * BRICKSIZE, defEnemy[currentItem].name);

		r.x = 0;
		r.w = 640;
		r.h = 20;

		if (mapY < MAPHEIGHT - 30)
			r.y = 460;
		else
			r.y = 0;

		SDL_FillRect(graphics.screen, &r, graphics.black);

		graphics.drawString(string, 320, r.y + 5, true, graphics.screen);
		
		while (SDL_GetTicks() < frameLimit){}
		frameLimit = SDL_GetTicks() + 16;
	}

	String *str = stringHead->next;
	String *str2;
	while (str != NULL)
	{
		str2 = str;
		printf("Deleting %s", str->string);
		str = str->next;
		delete str2;
	}

	return 0;
}