#include <iostream>
#include <iomanip>
#include "animals.h"

using namespace std;

enum class StatusT:char { HIDDEN, FLIPPED, FOUND};

const int BOARD_SIZE = 4;
const char ROW_MAX = char ('A'+BOARD_SIZE-1);
const int GAME_SIZE =  BOARD_SIZE*BOARD_SIZE;
const string HIDDEN = "XXXXXXX";
const int WORD_SIZE = 9;

void InitBoard(AnimalT board[], StatusT display[]);
void ShowBoard(AnimalT board[], StatusT display[]);
void Shuffle(AnimalT board[]);
int GetPosition(AnimalT board[], StatusT display[], StatusT status);
void TakeTurn(AnimalT board[], StatusT display[], int & found);
int TwoToOne(int row, int col);

int main() {

   StatusT display[GAME_SIZE];
   AnimalT board[GAME_SIZE];
   int found = 0;

   // initializations
   srand48(time(nullptr));
   InitBoard(board, display);
   Shuffle(board);

   while (found < GAME_SIZE/2) {
       TakeTurn(board, display, found);
   }

   cout << endl;
   cout << "Congratulations, you win!" << endl;
   cout << endl;
   return 0;
}

void Hold(AnimalT board[], StatusT display[]) {
   string junk;

   cout << endl;
   ShowBoard(board, display);

   cout <<"Press any key to continue";
   getline(cin,junk);
   getline(cin,junk);

   return;
}

void TakeTurn(AnimalT board[], StatusT display[], int & found) {

     int first, second;

     first = GetPosition(board, display, StatusT::HIDDEN);
     display[first] = StatusT::FLIPPED;

     second = GetPosition(board,display, StatusT::HIDDEN);
     display[second] = StatusT::FLIPPED;

     cout << endl;
     if (board[first] == board[second]) {
        cout << "Match!" << endl;
     }  else {
        cout << "Miss " << endl;
     }

     Hold(board, display);

     if (board[first] == board[second]) {
         display[first] = StatusT::FOUND;
         display[second] = StatusT::FOUND;
	 found++;
     } else {
         display[first] = StatusT::HIDDEN;
         display[second] = StatusT::HIDDEN;
     }


     return;
}

int GetRow() {
    char inputValue;
    do {
         cout << "Enter a row letter (A-" << ROW_MAX << ")  => ";
	 cin >> inputValue;
    } while (inputValue < 'A' or inputValue >ROW_MAX ) ;
    return int(inputValue - 'A');
}

int GetCol() {
    int inputValue;
    do {
         cout << "Enter a collumn number (1-" << BOARD_SIZE << ") =>";
	 cin >> inputValue;
    } while (inputValue < 1 or inputValue > BOARD_SIZE);
    return inputValue -1;
}

int GetPosition(AnimalT board[], StatusT display[], StatusT status) {

     int row, col;
     int pos;

     do { 
         ShowBoard(board, display); 
	 cout << endl;
	 cout << "Enter a position" << endl;
	 cout << endl;

         row = GetRow();
         col = GetCol();

	 pos = TwoToOne(row,col);
     } while (display[pos] != status); 
     return pos;
}

void InitBoard(AnimalT board[], StatusT display[]){
    size_t i;

    for(i=0;i<GAME_SIZE;i++) {
        display[i] = StatusT::HIDDEN;
    }

    for(i=1;i<GAME_SIZE;i+= 2) {
        board[i] = RandomAnimalT();
	board[i-1] = board[i];
    }

    return;
}


void Shuffle(AnimalT board[]){
    int i;
    int swapPos;
    AnimalT tmp;


    for(i=0;i<GAME_SIZE;i++) {
        swapPos = drand48() * GAME_SIZE;
	if (swapPos != i) {
	    tmp = board[i];
	    board[i] = board[swapPos];
	    board[swapPos] = tmp;
	}
    }

    return;
}

// important, we will treat this like a 2d array for display purposes
// row * BOARD_SIZE + column
//     means the array   0 1 2 3 4 5 6 7 8 
//     will display as
//                       0 1 2      0*3+0   0*3+1    0*3+2
//                       3 4 6  or  1*3+0   1*3+1    1*3+2
//                       6 7 8      2*3+0   2*3+1    2*3+2
//
int TwoToOne(int row, int col) {
    return row*BOARD_SIZE + col;
}


void ShowBoard(AnimalT board[], StatusT display[]) {
    int row, col;
    int pos;
 
    // empty space in corner
    cout << setw(4) << " ";
    
    // col numnbers 
    for(col=0;col<BOARD_SIZE;col++) {
         cout << setw(WORD_SIZE) << col + 1;
    }
    cout << endl;

    // line under
    cout << setw(5) << "+";
    cout << setfill('-');
    for(col=0;col<BOARD_SIZE;col++) {
         cout << setw(WORD_SIZE) << "-";
    }
    cout << endl;
    cout << setfill(' ');

    // print the table.
    for(row=0;row<BOARD_SIZE;row++) {
       // row letter
       cout << setw(4) << char('A'+row) <<"|";       

       for(col=0;col<BOARD_SIZE;col++) {
           cout << setw(WORD_SIZE) ;
	   pos = TwoToOne(row,col);
           if (display[pos] == StatusT::FLIPPED) {
               cout << AnimalTToString(board[pos]);
           } else if (display[pos] == StatusT::HIDDEN) {
	       cout << HIDDEN ;
	   } else {
	       cout << " " ;
	   }
       }
       cout << endl;
    }
    return;
}