#include "GameOfLife.h"

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <cilk\cilk.h>

using namespace std;

GameOfLife::GameOfLife(unsigned int height, unsigned int width, InitModel model)
{
    m_height = height;
    m_width = width;

    m_buffer = new unsigned char [m_width * m_height];
    m_nextIter = new unsigned char [m_width * m_height];

    // fill buffer
    switch (model)
    {
    case PULSAR:
        {
            if (width > 15 && height > 15)
            {
                memset(m_buffer, 0x0, m_width * m_height * sizeof(unsigned char));
                m_buffer[2 * m_width + 4] = 1;
                m_buffer[2 * m_width + 5] = 1;
                m_buffer[2 * m_width + 6] = 1;

                m_buffer[2 * m_width + 10] = 1;
                m_buffer[2 * m_width + 11] = 1;
                m_buffer[2 * m_width + 12] = 1;

                m_buffer[4 * m_width + 2] = 1;
                m_buffer[5 * m_width + 2] = 1;
                m_buffer[6 * m_width + 2] = 1;

                m_buffer[4 * m_width + 7] = 1;
                m_buffer[5 * m_width + 7] = 1;
                m_buffer[6 * m_width + 7] = 1;

                m_buffer[4 * m_width + 9] = 1;
                m_buffer[5 * m_width + 9] = 1;
                m_buffer[6 * m_width + 9] = 1;

                m_buffer[4 * m_width + 14] = 1;
                m_buffer[5 * m_width + 14] = 1;
                m_buffer[6 * m_width + 14] = 1;

                m_buffer[7 * m_width + 4] = 1;
                m_buffer[7 * m_width + 5] = 1;
                m_buffer[7 * m_width + 6] = 1;

                m_buffer[7 * m_width + 10] = 1;
                m_buffer[7 * m_width + 11] = 1;
                m_buffer[7 * m_width + 12] = 1;

                m_buffer[9 * m_width + 4] = 1;
                m_buffer[9 * m_width + 5] = 1;
                m_buffer[9 * m_width + 6] = 1;

                m_buffer[9 * m_width + 10] = 1;
                m_buffer[9 * m_width + 11] = 1;
                m_buffer[9 * m_width + 12] = 1;

                m_buffer[10 * m_width + 2] = 1;
                m_buffer[11 * m_width + 2] = 1;
                m_buffer[12 * m_width + 2] = 1;

                m_buffer[10 * m_width + 7] = 1;
                m_buffer[11 * m_width + 7] = 1;
                m_buffer[12 * m_width + 7] = 1;

                m_buffer[10 * m_width + 9] = 1;
                m_buffer[11 * m_width + 9] = 1;
                m_buffer[12 * m_width + 9] = 1;

                m_buffer[10 * m_width + 14] = 1;
                m_buffer[11 * m_width + 14] = 1;
                m_buffer[12 * m_width + 14] = 1;

                m_buffer[14 * m_width + 4] = 1;
                m_buffer[14 * m_width + 5] = 1;
                m_buffer[14 * m_width + 6] = 1;

                m_buffer[14 * m_width + 10] = 1;
                m_buffer[14 * m_width + 11] = 1;
                m_buffer[14 * m_width + 12] = 1;

                break;
            }
            else
            {
                cout << "Board too small, using RANDOM." << endl;
            }
        }

    case RANDOM:
        {
            srand(time(NULL));
            for (unsigned int i = 0; i < m_height; i++)
            {
                for (unsigned int j = 0; j < m_width; j++)
                {
                    m_buffer[i * m_width + j] = rand() % 2;
                }
            }
            break;
        }

    default:
        break;
    }
}


void GameOfLife::printIteration()
{
    for (int row = 0; row < m_height; row++)
    {
        for (int col = 0; col < m_width; col++)
        {
            char c;

            if (m_buffer[row * m_width + col] == 1)
                c = 0xb2;
            else
                c = 0x20;

            cout << c << " ";
        }
        cout << "\n";
    }
}


int GameOfLife::getNeighbourSum(unsigned int row, unsigned int col)
{
    int sum = 0;

    // TODO: PLACE CODE HERE
    for (int i=0; i<=2; i++)
        for (int j=0; j<=2; j++)
            if ((((row+i) * m_width + (col+j)) >=0) && (((row+i) * m_width + (col+j)) <= m_width*m_height))
                sum += m_buffer[(row+i)*m_width+(col+j)];

        sum -= m_buffer[row*m_width+col];
    return sum;
}


void GameOfLife::nextIterSerial()
{
    for (int row = 0; row < m_height; row++)
    {
        for (int col = 0; col < m_width; col++)
        {
            // TODO: IMPLEMENT getNeighbourSum (done)
            int neighbours = getNeighbourSum(row, col);
            if (m_buffer[row * m_width + col] == 1)
            {
                if ((neighbours == 2) || (neighbours == 3))
                {
                    m_nextIter[row * m_width + col] = 1;
                }
                else
                {
                    m_nextIter[row * m_width + col] = 0;
                }
            }
            else
            {
                if (neighbours == 3)
                {
                    m_nextIter[row * m_width + col] = 1;
                }
                else
                {
                    m_nextIter[row * m_width + col] = 0;
                }
            }
        }
    }

    // swtich buffers
    unsigned char* temp = m_buffer;
    m_buffer = m_nextIter;
    m_nextIter = temp;
}

void GameOfLife::tempSpawn(uint32 row)
{
    for (int col = 0; col < m_width; col++)
        {
            int neighbours = getNeighbourSum(row, col);
            if (m_buffer[row * m_width + col] == 1)
            {
                if ((neighbours == 2) || (neighbours == 3))
                {
                    m_nextIter[row * m_width + col] = 1;
                }
                else
                {
                    m_nextIter[row * m_width + col] = 0;
                }
            }
            else
            {
                if (neighbours == 3)
                {
                    m_nextIter[row * m_width + col] = 1;
                }
                else
                {
                    m_nextIter[row * m_width + col] = 0;
                }
            }
        }
}

void GameOfLife::nextIterParallel()
{
    // TODO: PLACE CODE HERE
    for (int row = 0; row < m_height; row++)
    {
        cilk_spawn tempSpawn(row);
    }
    cilk_sync;

    // swtich buffers
    unsigned char* temp = m_buffer;
    m_buffer = m_nextIter;
    m_nextIter = temp;
}


GameOfLife::~GameOfLife()
{
    delete [] m_buffer;
    delete [] m_nextIter;
}