#include<stdio.h>
#include<stdlib.h>
#include<curses.h>
#include<signal.h>
#include<time.h>
#include<unistd.h>
#define maxiter 500

static void finish(int sig);


 int main(int argc, char* argv[]){
 int s;
 int player=0;
 int action[maxiter][2];
 int reward[maxiter][2];
 int error[2]={1,1};
 char buffer[200];
 int iter;
 float reward_matrix[2][2][2]=
      {{{0.25,0.75},{0.75,0.25}},{{0.75,0.25},{0.25,0.75}}};
 int reward_buffer[2];
 int action_buffer;
 int key[2][2]={{'a','s'},{'k','l'}};
 int r1,r2;
 int income[2]={0,0};
 /* Saving Data */

 FILE *fp; 
 fp=fopen("result1.txt","wt");

 signal(SIGINT, finish); // Arrange interrupts to terminate
 initscr();              // Initialize screen
 keypad(stdscr, TRUE);   // enable Keypad
 nonl();                 // NL != CR/NL
 cbreak();               // Take input char one at a time, don't wait /n
 noecho();               // Don't echo input
 srand(time(NULL));

 move(1,1);
 addstr("*************************************************************************************************************");
 sprintf(buffer,"Press A for front or S for rear\t\t\t\t\t Press K for front or L for rear");
 move(3,1);
 addstr(buffer);
 move(5,1);
 addstr("*************************************************************************************************************");
 move(40,1);
 addstr("*************************************************************************************************************");
 
for (iter=0;iter<maxiter;iter++){
   /* Keyboard Input From Each Player */
   
   do{
     
     action_buffer=getch();
     for (player=0;player<=1;player++){
       if(error[player]==1){
	 if (player==0)move(20,1);
	 else move(20,90);
	 addstr("                 "); 
	//addstr("waiting for input");
       }
     }
     for (player=0;player<=1;player++){
       if (error[player]==1){
 
	 if (action_buffer==key[player][0]) {
	   action[iter][player]=0;
	   error[player]=0;
	   if (player==0) move(20,1);
	   else move(20,90);
	   addstr("*****             ");
	 }
	 else if (action_buffer==key[player][1]) {
	   action[iter][player]=1;
	   error[player]=0;
	   if (player==0) move(20,1);
	   else move(20,90);
	   addstr("*****            ");
	 }
	 else {
	  error[player]=1;
	   
	 }
       }
     }
   }while (error[0]==1 || error[1]==1);
   error[0]=1;
   error[1]=1;
   /* Determine Reward Stochastically */
   player=0;
   if (rand() <= RAND_MAX *reward_matrix[player][action[iter][0]][action[iter][1]])
     reward_buffer[player]=1;
   else reward_buffer[player]=0;
                   
   player=1;
   if (rand() <= RAND_MAX * reward_matrix[player][action[iter][0]][action[iter][1]])
     reward_buffer[player]=1;
   else reward_buffer[player]=0;
   r1=reward_buffer[0];
   r2=reward_buffer[1];
   reward[iter][0]=reward_buffer[0];
   reward[iter][1]=reward_buffer[1];
   if (r1==0 && r2==0){
	     move(15,1);
	     printw("You lose\t\t\t\t\t\t\t\t\t\tYou lose   ");
	     refresh();
	     sleep(1);
	     move(15,1);
	     printw("New round\t\t\t\t\t\t\t\t\t\tNew round");
	     move(20,1);
	     printw("                    \t\t\t\t\t\t\t\t                   ");
		
	     refresh();
   }
   if (r1==0 && r2==1){
     		move(15,1);
		printw("You lose\t\t\t\t\t\t\t\t\t\tYou win    ");
		refresh();
		sleep(1);
		move(15,1);
		printw("New round\t\t\t\t\t\t\t\t\t\tNew round");
		move(20,1);
		printw("                    \t\t\t\t\t\t\t\t                     ");
	
		refresh();
	
   }
   if (r1==1 && r2==0){
		move(15,1);
		printw("You win\t\t\t\t\t\t\t\t\t\tYou lose   ");
		refresh();
		sleep(1);
		move(15,1);
		printw("New round\t\t\t\t\t\t\t\t\t\tNew round");
		move(20,1);
		printw("                    \t\t\t\t\t\t\t\t                   ");
	
		refresh();
   }
   if (r1==1 && r2==1){
		move(15,1);
		printw("You win\t\t\t\t\t\t\t\t\t\tYou win    ");
		refresh();
		sleep(1);
		move(15,1);
		printw("New round\t\t\t\t\t\t\t\t\t\tNew round");
		move(20,1);
		printw("                    \t\t\t\t\t\t\t\t                    ");
		refresh();
		
   }
	fprintf(fp,"%d\t\t%d\t%d\t\t%d\t%d\n",iter,action[iter][0],reward[iter][0],action[iter][1],reward[iter][1]);
	income[0]+=reward[iter][0];
	income[1]+=reward[iter][1];
}
  

 move(20,45);
 addstr("Game Over");
 move(21,45);
 printw("player1 =%d  player2= %d",income[0],income[1]); 
getch(); 
 echo();
  nl();
 endwin();
 
 fprintf(fp,"player1=%d player2=%d\n",income[0],income[1]);
 fclose(fp);
 return 0; 
}
 
static void finish(int sig){
  endwin();
  exit(0);
}