#ifndef MAZEGEN_H
#define MAZEGEN_H

#include "mazegen.h"

#include <stdlib.h>
#include <iostream>
#include <ctime>
#include <time.h>
//#include <windows.h>
//#include <conio.h>
#include <stack>
#include <fstream>
//#include "stdafx.h" //http://sourceforge.net/p/wpbdc/website/ci/master/tree/Judge/StdAfx.h

using namespace std;

#define MSIZE 11

void ClearScreen(){
    printf("\033[2J\033[1;1H"); // clear screen
}

// CELL STRUCTURE
struct Cell
{
    bool visited;
    bool top_wall;
    bool bot_wall;
    bool left_wall;
    bool right_wall;
    char display;
};

// INITIALIZE MAZE
void Initialize(Cell Level[][MSIZE]) {
    for(int i=0; i<MSIZE; i++) {
        for(int j=0; j<MSIZE; j++) {
            Level[i][j].display = '*';
            Level[i][j].visited = false;
            Level[i][j].top_wall = true;
            Level[i][j].bot_wall = true;
            Level[i][j].left_wall = true;
            Level[i][j].right_wall = true;
        }
    }
    for(int i=1; i<MSIZE-1; i++) {
        for(int j=1; j<MSIZE-1; j++) {
            // Border Cells have fewer accessible walls
            Level[1][j].top_wall = false;
            Level[MSIZE-2][j].bot_wall = false;
            Level[i][1].left_wall = false;
            Level[i][MSIZE-2].right_wall = false;
        }
    }
}

// REDRAW MAZE
void Redraw(Cell Level[][MSIZE]) {
    for(int i=0; i<MSIZE; i++) {
        cout << endl;
        for(int j=0; j< MSIZE; j++)
            cout << " " << Level[i][j].display;
    }
}

// GENERATE MAZE
void GenerateMaze(Cell Level[][MSIZE], int &posX, int &posY, int &goalX, int &goalY) {
    srand((unsigned)time(NULL));                                                                            // Pick random start cell
    int random = 0;
    int randomX = ((2*rand())+1)%(MSIZE-1);                                          // Generate a random odd number between 1 and MSIZE
    int randomY = ((2*rand())+1)%(MSIZE-1);                                          // Generate a random odd number between 1 and MSIZE
    posX = randomX;                                                                 // Save player's initial X position
    posY = randomY;                                                                 // Save player's initial Y position
    int visitedCells = 1;
    int totalCells = ((MSIZE-1)/2)*((MSIZE-1)/2);
    int percent = 0;
    stack<int> back_trackX, back_trackY;                                            // Stack is used to trace the reverse path

    Level[randomY][randomX].display = 'S';                                          // Set S as the start cell
    Level[randomY][randomX].visited = true;                                         // Set start cell as visited;

    while(visitedCells < totalCells)
    {
        if(((Level[randomY-2][randomX].visited == false) && (Level[randomY][randomX].top_wall == true && Level[randomY-2][randomX].bot_wall == true)) ||
           ((Level[randomY+2][randomX].visited == false) && (Level[randomY][randomX].bot_wall == true && Level[randomY+2][randomX].top_wall == true)) ||
           ((Level[randomY][randomX-2].visited == false) && (Level[randomY][randomX].left_wall == true && Level[randomY][randomX-2].right_wall == true)) ||
           ((Level[randomY][randomX+2].visited == false) && (Level[randomY][randomX].right_wall == true && Level[randomY][randomX+2].left_wall == true)))
        {
            random = (rand() % 4) + 1;              // Pick a random wall 1-4 to knock down

            // GO UP
            if((random == 1) && (randomY > 1)) {
                if(Level[randomY-2][randomX].visited == false) {        // If not visited
                    Level[randomY-1][randomX].display = ' ';        // Delete display
                    Level[randomY-1][randomX].visited = true;       // Mark cell as visited
                    Level[randomY][randomX].top_wall = false;       // Knock down wall

                    back_trackX.push(randomX);                      // Push X for back track
                    back_trackY.push(randomY);                      // Push Y for back track

                    randomY -= 2;                                   // Move to next cell
                    Level[randomY][randomX].visited = true;         // Mark cell moved to as visited
                    Level[randomY][randomX].display = ' ';          // Update path
                    Level[randomY][randomX].bot_wall = false;       // Knock down wall
                    visitedCells++;                                 // Increase visitedCells counter
                }
                else
                    continue;
            }

                // GO DOWN
            else if((random == 2) && (randomY < MSIZE-2)) {
                if(Level[randomY+2][randomX].visited == false) {        // If not visited
                    Level[randomY+1][randomX].display = ' ';        // Delete display
                    Level[randomY+1][randomX].visited = true;       // Mark cell as visited
                    Level[randomY][randomX].bot_wall = false;       // Knock down wall

                    back_trackX.push(randomX);                      // Push X for back track
                    back_trackY.push(randomY);                      // Push Y for back track

                    randomY += 2;                                   // Move to next cell
                    Level[randomY][randomX].visited = true;         // Mark cell moved to as visited
                    Level[randomY][randomX].display = ' ';          // Update path
                    Level[randomY][randomX].top_wall = false;       // Knock down wall
                    visitedCells++;                                 // Increase visitedCells counter
                }
                else
                    continue;
            }

                // GO LEFT
            else if((random == 3) && (randomX > 1)) {
                if(Level[randomY][randomX-2].visited == false) {        // If not visited
                    Level[randomY][randomX-1].display = ' ';        // Delete display
                    Level[randomY][randomX-1].visited = true;       // Mark cell as visited
                    Level[randomY][randomX].left_wall = false;      // Knock down wall

                    back_trackX.push(randomX);                      // Push X for back track
                    back_trackY.push(randomY);                      // Push Y for back track

                    randomX -= 2;                                   // Move to next cell
                    Level[randomY][randomX].visited = true;         // Mark cell moved to as visited
                    Level[randomY][randomX].display = ' ';          // Update path
                    Level[randomY][randomX].right_wall = false;     // Knock down wall
                    visitedCells++;                                 // Increase visitedCells counter
                }
                else
                    continue;
            }

                // GO RIGHT
            else if((random == 4) && (randomX < MSIZE-2)) {
                if(Level[randomY][randomX+2].visited == false) {        // If not visited
                    Level[randomY][randomX+1].display = ' ';        // Delete display
                    Level[randomY][randomX+1].visited = true;       // Mark cell as visited
                    Level[randomY][randomX].right_wall = false;     // Knock down wall

                    back_trackX.push(randomX);                      // Push X for back track
                    back_trackY.push(randomY);                      // Push Y for back track

                    randomX += 2;                                   // Move to next cell
                    Level[randomY][randomX].visited = true;         // Mark cell moved to as visited
                    Level[randomY][randomX].display = ' ';          // Update path
                    Level[randomY][randomX].left_wall = false;      // Knock down wall
                    visitedCells++;                                 // Increase visitedCells counter
                }
                else
                    continue;
            }

            percent = (visitedCells*100/totalCells*100)/100;                // Progress in percentage
            cout << endl << "       Generating a Random Maze... " << percent << "%" << endl;
        }
        else {
            randomX = back_trackX.top();
            back_trackX.pop();

            randomY = back_trackY.top();
            back_trackY.pop();
        }

        ClearScreen();
        Redraw(Level);
    }

    goalX = randomX;
    goalY = randomY;
    Level[goalY][goalX].display = 'E';
    ClearScreen();
    Redraw(Level);
    cout << endl << "\a\t   Complete!" << endl;
}

#endif // MAZEGEN_H