#include <iostream.h>
#include "ttt.h"

void Welcome (char[][3]);
bool check_valid(char[][3]);
bool Play_Again();
int Get_Players(int&);
int Get_Level();

void Welcome (char Board[][3]) 
{
   char game;
   bool valid=false;
   cout<<"Welcome to Alan's Tic-Tac-Toe Game"<<endl;
   while (!valid) {
      cout<<"Do you want to play the (S)tandard game or (C)ustom setup? ";
      cin>>game;
      if (game=='s' || game=='S') {
         int i,x;
         for (i=0; i<3; i++)
            for (x=0; x<3; x++)
               Board[i][x]='-';
         valid=true;
      }
      else if (game=='c' || game=='C') {
         cout<<"X=X, O=O, -=blank"<<endl;
         cout<<"Enter 3 characters separated by spaces for the first row  : ";
         int i;
         for (i=0; i<3; i++)
            cin>>Board[0][i];
         cout<<"Enter 3 characters separated by spaces for the second row : ";
         for (i=0; i<3; i++)
            cin>>Board[1][i];
         cout<<"Enter 3 characters separated by spaces for the third row  : ";
         for (i=0; i<3; i++)
            cin>>Board[2][i];
         if (check_valid(Board))
            valid=true;
         }
      else 
         cout<<"Invalid choice."<<endl;
   }
}


int Get_Level ()
{
   char diff='0';
   while (diff=='0') {
      cout<<"Choose difficulty level (1/2/3) : ";
      cin>>diff;
      if (diff=='1') return EASY;
      if (diff=='2') return MEDIUM;
      if (diff=='3') return HARD;
      diff='0';
      }
   return ERROR;
}

bool check_valid(char Board[][3])
{
   int i,x;
   int checksum=0;
   for (i=0; i<3; i++)
      for (x=0; x<3; x++)
         if (Board[i][x] == 'X')
            checksum+=1;
         else if (Board[i][x] == 'O')
            checksum-=1;
   if (checksum == 0 || checksum == 1)
      return true;
   cout<<"Invalid setup.  Try again."<<endl;
   return false;
}  


bool Play_Again()
{
   char again;
   cout<<"Do you want to play again? (Y/N) ";
   cin>>again;
   if (again=='Y' || again=='y') 
      return true;
   cout<<"Thanks for playing!"<<endl<<endl;
   return false;
}

int Get_Players(int& players)
{
   char num_players;
   bool valid=false;
   while (!valid) {
      valid=true;
      cout<<"How many players (0/1/2) ? ";
      cin>>num_players;
      if (num_players=='2')
         players=2;
      else if (num_players=='1')
         players=1;
      else if (num_players=='0')
         players=0;
      else valid=false;
   }
   if (players==1) {
      cout<<"(C)omputer or (P)layer goes first? ";
      cin>>num_players;
      if (num_players=='P' || num_players=='p') {
         cout<<"Ok, you go first"<<endl;
         return PLAYER; 
      }
      return COMPUTER;
   }
   return PLAYER;   
  }

int main()
{
   int x_wins=0;
   int o_wins=0;
   int cat_wins=0;
   bool CONTINUE = true;
   char init2[3][3];
   int Players;
   char err='E';
   int Level;
   int first;
   while (err=='E') {
      Welcome(init2);
      first=Get_Players(Players);
      if (Players==2)
         Level=3;
      else
         Level=Get_Level();
      TicTacToe Board(init2,Level,Players,first);
      err=Board.Evaluate();
      if (err=='E')
        cout<<"Error in board set-up, re-enter data."<<endl;
      }
   while (CONTINUE) {
      char init[3][3]=init2;
      char win;
      int turn=1;
      bool game_over=false;
      TicTacToe Board(init, Level, Players, first);
      while (!game_over) {
         Board.Display();
         win=Board.Evaluate();
         switch (win) {
            case 'X' : game_over=true;
                       cout<<"X wins!!"<<endl; 
                       x_wins++;
                       break; 
            case 'O' : game_over=true;
                       cout<<"O wins!!"<<endl;
                       o_wins++;
                       break;
            case 'C' : game_over=true;
                       cout<<"CAT wins!"<<endl;
                       cat_wins++;
                       break; 
            case 'E' : game_over=true;
                       cout<<"Error...call Alan!"<<endl;
                       break;
            default  : break;
            }
         if (!game_over)
            Board.Make_Move(turn);
      }
      cout<<"X-"<<x_wins<<"   O-"<<o_wins<<"   CAT-"<<cat_wins<<endl;
      CONTINUE=Play_Again();
}
}