#include "bullet.hpp"
#include "enemy.hpp"
#include "player.hpp"
#include "boss.hpp"
#include "powerup.hpp"
#include "definitions.hpp"
#include "functions.hpp"
#include "wall.hpp"

using std::cout;
using std::endl; 
using std::min;

int ENEMY_NUM=1;

int main(int argc,char** argv)
{
	if(argc>1){
		if(strcmp(argv[1],"--uber-debug")!=0 && strcmp(argv[1],"--info")!=0 && strcmp(argv[1],"--autowin")!=0){
			cout<<"Usage: "<<argv[0]<<" [--info]"<<endl;
			return 1;
		}
	
		if(strcmp(argv[1],"--info")==0){
			cout<<endl<<"Curses Invaders 4.4"<<endl;
			cout<<endl<<"Game developed by Giacomo Parolini (jp) and Enrico Guiraud (blue) in years 2010-2013."<<endl;
			cout<<"Source code is available under request to jp@lcm.mi.infn.it (it's quite ugly, I warn you ;-) )"<<endl;
			cout<<"Report any bug to the same mail address."<<endl<<endl;
			return 1;
		}
	}
	
  	typedef std::list<bullet> b_list;
  	typedef std::list<enemy> e_list;
  	typedef std::vector<wall> w_vec;
  	typedef std::list<rocket> r_list;
	srand(time(NULL));
	
	//OBJECTS INIZIALIZATION

	player player1;						//player is automatically created in [C/2][R]
	e_list enemies;						//vector of enemies
	b_list bullets;                				//list of player's bullets
	b_list bombs;						//list of enemies' bombs
	b_list powerups;					//list of powerups
	r_list rockets;						//list of rockets
	int refresh_time;         				//how long the program waits before refreshing the screen
	int chflag=0;						//cheat flag: 0=normal, 1=cheats allowed, 2=special mode activated, 3=both special mode and cheats.
	boss boss1;
	WINDOW *Score,*BossHP;
	int commands[CMD_NUM]={'a','d','w','s','p',' ','l','m','q'};	//0:left,1:right,2:up,3:down,4:pause,5:shoot1,6:shoot2,7:mute,8:quit
	
	//PARAMETERS/UTILITIES
	double shootrate;				  	//probability of an enemy shooting a bomb 
	int command;						//keyboard input
	int score=0;					   	//score: gain +100 when an enemy is destroyed and +50 when a bomb is destroyed
	char level='1';						//difficulty level
	//bool sound=false;					
	
	/*//PTHREAD STUFF
	pthread_t thread[THREADS_NUM];
	int bkgd_music;			//#0
	int enshoot_sound;		//#1
	int win_theme;			//#2
	int lose_theme;			//#3
	int shoot_sound;		//#4
	
	if(argc>1){		//./invaders -m/m/mute -> the game starts with no sounds.
		if(strcmp(argv[1],"-s")==0 || strcmp(argv[1],"s")==0 || strcmp(argv[1],"sound")==0)
			sound=true;
		else{
			cout<<"usage: "<<argv[0]<<" [-s]/[s]/[sound]"<<endl;
			return -1;
		}
	}*/
	
	//NCURSES STUFF
	initscr();
	curs_set(0);
	noecho();	
	nodelay(NULL, true);
	cbreak();
	keypad(stdscr,TRUE);
	start_color();
	init_pair(0,COLOR_WHITE,COLOR_BLACK);
	init_pair(1,COLOR_GREEN,COLOR_BLACK);	//PLAYER
	init_pair(2,COLOR_RED,COLOR_BLACK);	//ENEMY
	init_pair(3,COLOR_YELLOW,COLOR_BLACK);	//BULLETS-POWERUPS
	init_pair(4,COLOR_MAGENTA,COLOR_BLACK);	//WALLS
	init_pair(5,COLOR_CYAN,COLOR_BLACK);	//BOSS
	init_pair(6,COLOR_BLUE,COLOR_BLACK);	//BOMBS
	init_pair(7,COLOR_RED,COLOR_RED);	
	init_pair(8,COLOR_GREEN,COLOR_GREEN);
	init_pair(9,COLOR_BLACK,COLOR_GREEN);
	
	//////////PROGRAM START: creation of game objects and set of parameters
	
	WALLS_NUM=2;
	
	choose_level(shootrate, refresh_time, boss1, level, ENEMY_NUM, commands);     	//choose difficulty level and set game parameters and boss
	load_enemies(enemies,ENEMY_NUM);
	player1.set_commands(commands);
	erase();
	
	if(atoi(&level)==1) WALLS_NUM=3;
	
	
	w_vec walls(WALLS_NUM);
	
	Score=newwin(3,10,R/3,C+3);
	BossHP=newwin(3,15,R/3-3,C+3);
	
	int i=0;
	for(w_vec::iterator it=walls.begin(); it!=walls.end(); ++it, ++i)				//creating walls (in a quite symmetric pattern)
		it->create((i+1)*(C/(3*walls.size()+1))+i*(2*C/(3*walls.size()+1)),2*R/3,(int)min(6,2*C/(3*(int)(walls.size())+1)),2,3);

	/////////ENTERING MAIN LOOP
	
	/*if(sound)
		bkgd_music = pthread_create(&thread[0],NULL,pmusic,NULL); 	//executing background music in parallel thread*/
	
	if(argc>1){
		if(strcmp(argv[1],"--uber-debug")==0){ player1.weaponclass=5; player1.weaponclassrkt=5; player1.rocketlauncher=true; player1.length=MAX_LENGTH; }
		else if(strcmp(argv[1],"--autowin")==0){ Victory(boss1.name,score,level,0); endwin(); return 1; } 
	}
	
	while(1)
	{
		napms(refresh_time);					//ncurses sleep function (ms)
		timeout(0);
	    	command = getch();
	    	              
	    	if(command!=ERR)                        		//READING INPUT
	    		command = tolower(command);     
		    		           
	    	if(command == commands[8]){ 						//q = exit game
	    		endwin();
	    		//pkill_music(thread[0]);
	    		cout<<lightgreen<<"Game exited correctly."<<none<<endl;
	    		return 1;
	    	}  

	    	if(command == commands[4])
	    		pause_game(chflag);							//p = pause game

		//=====CHEATS================================================
		
	    	if(chflag==1 || chflag==3)
			get_cheat(command,player1,enemies,shootrate);
	    	if(chflag!=2 && chflag!=3)
			get_SpecialMode(command,chflag,boss1,enemies,bullets,bombs,powerups,rockets);
			
		//======END CHEATS===============================================
			    
	    	if(command == commands[0] || command == KEY_LEFT || command == commands[1] || command == KEY_RIGHT ||
		((chflag==1 || chflag==3) && (command == commands[2]|| command == KEY_UP || command == commands[3] || command == KEY_DOWN))){   		//a or d = move player (if cheats are triggered, also w and s)
	    		//for(int i=player1.x;i<player1.x+player1.length;i++) //mvaddch(player1.y,i,' ');	
	    		player1.next_pos(command);	
	    	}	    
		    		
		if(command==commands[5] || command==commands[6]){                		//spacebar = shoot!
			//if(sound) shoot_sound = pthread_create(&thread[4],NULL,pshoot_sound,NULL);
			player1.shoot(bullets);
			if(player1.rocketlauncher) player1.shootrkt(rockets);
		}
		
		/*if(command==commands[7]){				//mute/unmute
			if(sound) pkill_music(thread[0]);
			else	bkgd_music = pthread_create(&thread[0],NULL,pmusic,NULL); 
			sound=!sound;
		}*/
		
		if(enemyalive(enemies))					//if there's at least one enemy alive:
		{
			int n_enemies=0;
			e_list::const_iterator e_end = enemies.end();
			
			for(e_list::iterator it=enemies.begin(); it!=e_end; ++it)
				if(it->alive) n_enemies++;				//counts alive enemies
		
			for(e_list::iterator it=enemies.begin(); it!=e_end; ++it)         
			{
				//mvaddch(it->y,it->x,' ');
				it->next_pos();                                        //evaluate new positions
				if(it->alive && (double)rand()/RAND_MAX<(shootrate*ENEMY_NUM/n_enemies)){     //try a bomb-dropping
						//if(sound) shoot_sound = pthread_create(&thread[1],NULL,penshoot_sound,NULL);
						it->shoot(bombs);
				}
			}
		}
		else boss1.alive = true;					//if no enemy is alive, boss spawns
				
		if(boss1.alive)							//if boss is alive, moves and shoots
		{
			for(int i=0;i<boss1.width;i++) //mvaddch(boss1.y,boss1.x+i,' ');
			for(int j=0;j<boss1.height;j++){
				//mvaddch(boss1.y+j,boss1.x,' ');
				//mvaddch(boss1.y+j,boss1.x+boss1.width-1,' ');
			}
			boss1.next_pos();
			if((double)rand()/RAND_MAX<shootrate*25.){		//boss's shootrate is 25 times the one of normal enemies
				//if(sound) shoot_sound = pthread_create(&thread[1],NULL,penshoot_sound,NULL);
				boss1.shoot(bombs);	
			}
		}
		
		if(!bullets.empty())						//evaluate new bullets' positions
			for(b_list::iterator it=bullets.begin(); it!=bullets.end(); ++it){
				//mvaddch(it->y,it->x,' ');
				it->next_pos();
			}
				
		bullets.unique();      					//remove duplicates of bullets (i.e. player can't shoot enemies)


		if(!rockets.empty())						//evaluate new rockets' positions
			for(r_list::iterator it=rockets.begin(); it!=rockets.end(); ++it){
				//mvaddch(it->y,it->x,' ');
				it->next_pos();
			}

		rockets.unique();

		if(!bombs.empty())						//evaluate new bombs' positions
			for(b_list::iterator it=bombs.begin(); it!=bombs.end(); ++it){
				//mvaddch(it->y,it->x,' ');
				it->next_pos();
			}
				  	
		bombs.unique();						//remove duplicates of bombs

		if(!powerups.empty())						//new powerups' positions
			for(b_list::iterator it=powerups.begin(); it!=powerups.end(); ++it){
				//mvaddch(it->y,it->x,' ');
				it->next_pos();
			}
				
		powerups.unique(); 	 //FIXME: THIS ONLY WORKS ON CONSECUTIVE ELEMENTS OF THE LIST!! we should at least sort powerups before calling unique()
		
		interactions(player1,bullets,bombs,enemies,walls,powerups,rockets,boss1,score);

		///////ENDGAME CHECKS               
		if(boss1.health<1)  					//YOU WON!!
		{
			/*if(sound){
				pkill_music(thread[0]);	
				win_theme = pthread_create(&thread[2],NULL,pwin_theme,NULL);
			}*/
				
			Victory(boss1.name,score,level,chflag);
			
			WINDOW *replay;
			
			replay=newwin(3,25,26,20);
			box(replay,ACS_VLINE,ACS_HLINE);
		
			if(playagain(replay))                           //playagain() returns true if player wants to play again, false otherwise
			{
				delwin(replay);
				timeout(500);
				score=0;
				erase();
				refresh();
				choose_level(shootrate, refresh_time, boss1, level, ENEMY_NUM, commands);
				load_enemies(enemies,ENEMY_NUM);
				reset(player1, enemies, boss1, bullets, bombs, walls, powerups,rockets, level,chflag); // reset box, player, enemy and deletes all bullets and bombs
				erase();
				refresh();
				continue;  
			}
			else { 
				endwin();
				cout<<lightgreen<<"Game exited correctly."<<none<<endl;
				return 0;
			}	
		}
				
			
		if(gameover(player1,bombs))
		{
			Defeat(score);
			
			WINDOW* replay;
		
			replay=newwin(3,25,17,20);
			box(replay,ACS_VLINE,ACS_HLINE);
			
			if(playagain(replay))                                        //playagain() returns true if player wants to play again, false otherwise
			{
				delwin(replay);
				timeout(500);
				score=0;
				erase();
				refresh();
				choose_level(shootrate, refresh_time, boss1, level, ENEMY_NUM, commands);
				load_enemies(enemies,ENEMY_NUM);
				reset(player1, enemies, boss1, bullets, bombs, walls, powerups,rockets,level,chflag);       // reset box, player, enemy and delete all bullets and bombs
				erase();
				refresh();
				continue;  
			}
			else {
				endwin();
				cout<<lightgreen<<"Game exited correctly."<<none<<endl;
				return 0;
			}
		}
		
		draw(player1,bullets,bombs,enemies,walls,powerups,rockets,boss1);
		write_score(Score,score);
	
		if(boss1.alive){
			write_bosshp(BossHP,boss1.health,boss1.healthmax,boss1.name);
			wrefresh(BossHP);
		}
		
		wrefresh(Score);
		refresh();
	}  
////////END OF MAIN LOOP

	endwin();
	
	return -1;  //program never really reaches here, but compiler is happy
}