#include "CellularAutomaton.h"

CellularAutomaton::CellularAutomaton(int width, int height) : m_fieldWidth(width), m_fieldHeight(height)
{
	field.resize(m_fieldHeight, std::vector<int>(m_fieldWidth, 0));
	fieldNextState.resize(m_fieldHeight, std::vector<int>(m_fieldWidth, 0));

	//initializeRandom();
    initializeManual();
}


void CellularAutomaton::initializeRandom()
{
    std::random_device rd;
    std::mt19937 gen(rd());
    std::uniform_int_distribution<> dis(0, 1);

    for (int y = 0; y < m_fieldHeight; ++y)
    {
        for (int x = 0; x < m_fieldWidth; ++x)
        {
            field[y][x] = dis(gen);
        }
    }
}

void CellularAutomaton::initializeManual()
{
    std::cout << "Введите начальное состояние поля (" << m_fieldWidth << " x " << m_fieldHeight << ").\n";
    std::cout << "Введите 0 или 1 для каждой клетки.\n";

    for (int y = 0; y < m_fieldHeight; ++y)
    {
        for (int x = 0; x < m_fieldWidth; ++x)
        {
            std::cout << "Клетка (" << x << ", " << y << "): ";
            int cellValue = inputNumber(0, 1);

            field[y][x] = cellValue;
        }
    }
}

int CellularAutomaton::getCellState(int x, int y) const
{
    if (x < 0 || x >= m_fieldWidth || y < 0 || y >= m_fieldHeight)
    {
        // Единичные граничные условия: за пределами поля клетка считается равной 1
        return 1;
    }
    return field[y][x];
}

int CellularAutomaton::getNeighborhoodIndex(int x, int y) const
{
    int s0 = getCellState(x, y);
    int s1 = getCellState(x, y - 1);
    int s2 = getCellState(x, y + 1);
    int s3 = getCellState(x - 1, y);
    int s4 = getCellState(x + 1, y);

    int index = (s0 << 4) | (s1 << 3) | (s2 << 2) | (s3 << 1) | s4;

    return index;
}

void CellularAutomaton::update()
{
    for (int y = 0; y < m_fieldHeight; ++y)
    {
        for (int x = 0; x < m_fieldWidth; ++x)
        {
            int neighborhood = getNeighborhoodIndex(x, y);
            fieldNextState[y][x] = (functionValues >> neighborhood) & 1;
        }
    }

    field.swap(fieldNextState);
}

void CellularAutomaton::displayField() const
{
    for (const auto& row : field)
    {
        for (const auto& cell : row)
        {
            std::cout << (cell ? '1' : '0') << ' ';
        }
        std::cout << '\n';
    }
}