/**

 * test/demos/alpha_beta.c

 * Copyright (C) 2008 Markus Broeker

 */

#include <stdio.h>

#include <stdlib.h>

#include <limits.h>

#define BOARD_SIZE 9

#define EMPTY 32

struct Move {

    int target;

    int value;

};

typedef struct Move Move;

struct Node {

    Move *data;

    struct Node *prev;

};

typedef struct Node Node;

Move *bestMove;

Node *stack_end;

char board[BOARD_SIZE] = {

    EMPTY, EMPTY, EMPTY,

    EMPTY, EMPTY, EMPTY,

    EMPTY, EMPTY, EMPTY,

};

int desired_depth = -1;

int estimateFunction ();

int max_alpha_beta (int, int, int);

int min_alpha_beta (int, int, int);

void print (void);

/**

 * push: push a move onto the stack

 */

void push (Move * move)

{

    Node *actual;

    if ((actual = malloc (sizeof (Node))) == NULL) {

        perror ("MALLOC");

        exit (EXIT_FAILURE);

    }

    actual->data = move;

    actual->prev = stack_end;

    stack_end = actual;

}

/**

 * undo: revert the last move on the stack

 */

void undo ()

{

    Node *actual;

    if (stack_end == NULL) {

        printf ("Stack underrun\n");

        return;

    }

    board[stack_end->data->target] = EMPTY;

    actual = stack_end->prev;

    free (stack_end);

    stack_end = actual;

}

/**

 * validMove returns true(1) when the inspected field is EMPTY

 */

int validMove (int t)

{

    if (board[t] == EMPTY)

        return 1;

    return 0;

}

/**

 * Returns a Score of +-50 for 3 in a Row

 */

int estimateFunction ()

{

    if (board[0] == 'X' && board[1] == 'X' && board[2] == 'X')

        return -50;

    if (board[0] == 'O' && board[1] == 'O' && board[2] == 'O')

        return 50;

    if (board[3] == 'X' && board[4] == 'X' && board[5] == 'X')

        return -50;

    if (board[3] == 'O' && board[4] == 'O' && board[5] == 'O')

        return 50;

    if (board[6] == 'X' && board[7] == 'X' && board[8] == 'X')

        return -50;

    if (board[6] == 'O' && board[7] == 'O' && board[8] == 'O')

        return 50;

    if (board[0] == 'X' && board[3] == 'X' && board[6] == 'X')

        return -50;

    if (board[0] == 'O' && board[3] == 'O' && board[6] == 'O')

        return 50;

    if (board[1] == 'X' && board[4] == 'X' && board[7] == 'X')

        return -50;

    if (board[1] == 'O' && board[4] == 'O' && board[7] == 'O')

        return 50;

    if (board[2] == 'X' && board[5] == 'X' && board[8] == 'X')

        return -50;

    if (board[2] == 'O' && board[5] == 'O' && board[8] == 'O')

        return 50;

    if (board[0] == 'X' && board[4] == 'X' && board[8] == 'X')

        return -50;

    if (board[0] == 'O' && board[4] == 'O' && board[8] == 'O')

        return 50;

    if (board[2] == 'X' && board[4] == 'X' && board[6] == 'X')

        return -50;

    if (board[2] == 'O' && board[4] == 'O' && board[6] == 'O')

        return 50;

    return 0;

}

/**

 * this function visualizes the board

 */

void print ()

{

    int i;

    printf ("       1   2   3  \n     +---+---+---+\n");

    for (i = 0; i < 3; i++) {

        printf ("  %d  | ", i + 1);

        printf ("%c", board[3 * i]);

        printf (" | ");

        printf ("%c", board[3 * i + 1]);

        printf (" | ");

        printf ("%c", board[3 * i + 2]);

        printf (" | \n");

        if (i != 2) {

            printf ("     +---+---+---+\n");

        } else {

            printf ("     +---+---+---+\n");

        }

    }

    if (stack_end->data != NULL)

        printf ("Score: %3d\n", stack_end->data->value);

}

/**

 * simulates and returns the next possible move for a Player or null, when no more moves are possible.

 */

Move *simulate (int t, int player)

{

    Move *move;

    while (t < BOARD_SIZE) {

        if (!validMove (t)) {

            t++;

            continue;

        }

        if ((move = malloc (sizeof (Move))) == NULL) {

            perror ("MALLOC");

            exit (EXIT_FAILURE);

        }

        if (player == 0)

            board[t] = 'O';

        else

            board[t] = 'X';

        move->target = t;

        move->value = estimateFunction ();

        push (move);

        return move;

    }

    return NULL;

}

/**

 * AlphaBeta-Algorithm: the Maximizer

 */

int max_alpha_beta (int depth, int alpha, int beta)

{

    int moveValue;

    int t;

    Move *move;

    if (desired_depth == -1)

        desired_depth = depth;

    t = 0;

    while ((move = simulate (t, 0)) != NULL) {

        /*

         * particular move from black

         */

        t = move->target;

        if (depth == 0 || (estimateFunction () * estimateFunction ()) == 2500)

            moveValue = move->value;

        else                    /* best next move */

            moveValue = min_alpha_beta (depth - 1, alpha, beta);

        print ();

        undo ();

        t++;

        if (moveValue >= beta)

            return beta;

        if (moveValue > alpha) {

            alpha = moveValue;

            if (depth == desired_depth)

                bestMove = move;

        }

    }

    return alpha;

}

/**

 * AlphaBeta-Algorithm: the Minimizer

 */

int min_alpha_beta (int depth, int alpha, int beta)

{

    int moveValue;

    int t;

    Move *move;

    t = 0;

    while ((move = simulate (t, 1)) != NULL) {

        /*

         * particular move from white

         */

        t = move->target;

        if (depth == 0 || (estimateFunction () * estimateFunction ()) == 2500)

            moveValue = move->value;

        else                    /* best next move */

            moveValue = max_alpha_beta (depth - 1, alpha, beta);

        undo ();

        t++;

        if (moveValue <= alpha)

            return alpha;

        if (moveValue < beta)

            beta = moveValue;

    }

    return beta;

}

/**

 * returns true when no more fields are empty

 */

int game_over ()

{

    int i;

    for (i = 0; i < BOARD_SIZE; i++) {

        if (board[i] == EMPTY)

            return 0;

    }

    return 1;

}

int main (int argc, char **argv)

{

    Node *actual;

    int i, t, value;

    int depth;

    if (argc == 2)

        depth = atoi (argv[1]);

    else

        depth = 1;

    if ((actual = malloc (sizeof (Node))) == NULL) {

        perror ("MALLOC");

        return EXIT_FAILURE;

    }

    actual->data = NULL;

    actual->prev = NULL;

    stack_end = actual;

    printf ("\tTic Tac Toe\n");

    for (i = 0; i < 5; i++) {

        printf ("Enter a number: ");

        while (scanf ("%d", &t) < 1) {

            printf ("Enter a number: ");

            t = getchar ();

        }

        if (!validMove (--t)) {

            i--;

            continue;

        }

        board[t] = 'X';

        bestMove = NULL;

        value = max_alpha_beta (depth, SHRT_MIN, SHRT_MAX);

        if (bestMove == NULL) {

            break;

        }

        board[bestMove->target] = 'O';

        print ();

        if ((estimateFunction () * estimateFunction ()) == 2500) {

            return EXIT_SUCCESS;

        }

        if (game_over ())

            break;

    }

    print ();

    return EXIT_SUCCESS;

}