/home/marcvs/Desktop/working/pa-materiale/pa/codeBase/C++/include/Maze.h

#ifndef __MAZE_H__
#define __MAZE_H__

#pragma once

#include "DisjointSets.h"

class Maze {
private:
        int ** cell;
        int width, height;
        
        DisjointSets sets;
        
        void allocate() {
                cell = new int* [height];
                for (int i = 0; i < height; i++) {
                        cell[i] = new int[width];
                        for (int j = 0; j < width; j++) {
                                cell[i][j] = 1;
                        }
                }
        }

        inline void merge_rooms_of(int x1, int y1, int x2, int y2) {
                sets.merge_sets_of(x1 * width + y1, x2 * width + y2);
        }

public:
        Maze(int width, int height) : width(width), height(height), sets(width*height) { 
                allocate();
        }

        ~Maze() {
                for (int i = 0; i < height; i++) {
                        delete[] cell[i];
                }
                delete[] cell;
        }

        void clear_wall(int x, int y) {
                cell[x][y] = 0;
                if (x > 0 && cell[x-1][y] == 0) merge_rooms_of(x - 1, y, x, y);
                if (y > 0 && cell[x][y-1] == 0) merge_rooms_of(x, y - 1, x, y);
                if (x+1 < height && cell[x+1][y] == 0) merge_rooms_of(x + 1, y, x, y);
                if (y+1 < width  && cell[x][y+1] == 0) merge_rooms_of(x, y + 1, x, y);
        }
        
        inline bool same_room(int x1, int y1, int x2, int y2) {
                return sets.same_set(x1 * width + y1, x2 * width + y2);
        }
        
        inline int get_room_ID(int x, int y) {
                return sets.get_ID(x * width + y);
        }
        
        inline int get_width() {
                return width;
        }
        
        inline int get_height() {
                return height;
        }
        
        friend std::ostream& operator << (std::ostream &, const Maze &);
};

#endif