Grid.cpp

#include "Grid.h"
#include <windows.h>
#include <iostream>
#include <cstdlib>

using namespace std;

Grid::Grid(int w, int h) : width(w), height(h) {
    cells.resize(height, vector<char>(width, ' '));
    initializeGrid();
}

void Grid::initializeGrid() {
    // Create walls around the perimeter
    for (int i = 0; i < height; i++) {
        for (int j = 0; j < width; j++) {
            if (i == 0 || i == height - 1 || j == 0 || j == width - 1) {
                cells[i][j] = '#';
            }
        }
    }
    
    // Add some internal walls
    for (int i = 5; i < 15; i++) {
        if (width/2 < width) cells[i][width/2] = '#';
    }
    for (int j = 5; j < width/2 && j < width; j++) {
        if (7 < height) cells[7][j] = '#';
    }
    
    // Create exits
    exits.push_back(Position(1, 0));
    exits.push_back(Position(width-2, 0));
    exits.push_back(Position(width-1, height/2));
    
    for (const auto& exit : exits) {
        if (isValid(exit)) {
            cells[exit.y][exit.x] = 'E';
        }
    }
}

bool Grid::isValid(const Position& pos) const {
    return pos.x >= 0 && pos.x < width && pos.y >= 0 && pos.y < height;
}

char Grid::getCell(const Position& pos) const {
    if (!isValid(pos)) return '#';
    return cells[pos.y][pos.x];
}

void Grid::setCell(const Position& pos, char value) {
    if (isValid(pos)) {
        cells[pos.y][pos.x] = value;
    }
}

bool Grid::isWalkable(const Position& pos) const {
    if (!isValid(pos)) return false;
    char cell = getCell(pos);
    return cell == ' ' || cell == 'E';
}

vector<Position> Grid::getNeighbors(const Position& pos) const {
    vector<Position> neighbors;
    int dx[] = {0, 1, 0, -1};
    int dy[] = {-1, 0, 1, 0};
    
    for (int i = 0; i < 4; i++) {
        Position neighbor(pos.x + dx[i], pos.y + dy[i]);
        if (isValid(neighbor)) {
            neighbors.push_back(neighbor);
        }
    }
    return neighbors;
}

const vector<Position>& Grid::getExits() const {
    return exits;
}

void Grid::clearScreen() {
    system("cls");
}

void Grid::display() const {
    COORD coord = {0, 0};
    SetConsoleCursorPosition(GetStdHandle(STD_OUTPUT_HANDLE), coord);
    
    for (int i = 0; i < height; i++) {
        for (int j = 0; j < width; j++) {
            cout << cells[i][j];
        }
        cout << endl;
    }
}