#pragma once
#ifndef REGIONMAP_H
#define REGIONMAP_H

#include "map.h"
#include "tilemaptileselector.h"
#include "history.h"

#include <QStringList>
#include <QString>
#include <QVector>
#include <QList>
#include <QMap>
#include <QGraphicsScene>
#include <QGraphicsView>

#include <memory>
using std::shared_ptr;

class Project;

struct LayoutSquare
{
    LayoutSquare() : map_section("MAPSEC_NONE"), x(-1), y(-1), has_map(false) {}
    QString map_section;
    int x;
    int y;
    bool has_map;
};

struct MapSectionEntry
{
    QString name = "";
    int x = 0;
    int y = 0;
    int width = 1;
    int height = 1;
    bool valid = false;
};

class RegionMap : public QObject
{
    Q_OBJECT
public:
    RegionMap() = delete;
    RegionMap(Project *);

    ~RegionMap() {}

    Project *project = nullptr;

    bool loadMapData(poryjson::Json);
    bool loadTilemap(poryjson::Json);
    bool loadLayout(poryjson::Json);
    bool loadEntries();

    void setEntries(tsl::ordered_map<QString, MapSectionEntry> *entries) { this->region_map_entries = entries; }
    void setEntries(tsl::ordered_map<QString, MapSectionEntry> entries) { *(this->region_map_entries) = entries; }
    void clearEntries() { this->region_map_entries->clear(); }
    MapSectionEntry getEntry(QString section);
    void setEntry(QString section, MapSectionEntry entry);
    void removeEntry(QString section);

    void save();
    void saveTilemap();
    void saveLayout();

    void resizeTilemap(int width, int height, bool update = true);
    void resetSquare(int index);
    void clearLayout();
    void clearImage();
    void replaceSection(QString oldSection, QString newSection);
    void swapSections(QString secA, QString secB);

    unsigned getTileId(int index);
    shared_ptr<TilemapTile> getTile(int index);
    unsigned getTileId(int x, int y);
    shared_ptr<TilemapTile> getTile(int x, int y);
    bool squareHasMap(int index);
    QString squareMapSection(int index);
    void setSquareMapSection(int index, QString section);
    int squareX(int index);
    int squareY(int index);
    bool squareInLayout(int x, int y);
    int firstLayoutIndex() { return this->offset_left + this->offset_top * this->tilemap_width; }

    void setTileId(int index, unsigned id);
    void setTile(int index, TilemapTile &tile);
    void setTileData(int index, unsigned id, bool hFlip, bool vFlip, int palette);
    int getMapSquareIndex(int x, int y);

    QString getAlias() { return this->alias; }
    poryjson::Json::object config();
    
    QString palPath();
    QString pngPath();
    QString entriesPath() { return this->entries_path; }

    QByteArray getTilemap();
    void setTilemap(QByteArray newTilemap);

    QList<LayoutSquare> getLayout(QString layer);
    void setLayout(QString layer, QList<LayoutSquare> layout);

    bool layoutEnabled() { return this->layout_format != LayoutFormat::None; }

    QMap<QString, QList<LayoutSquare>> getAllLayouts();
    void setAllLayouts(QMap<QString, QList<LayoutSquare>> newLayouts);

    QStringList getLayers() { return this->layout_layers; }
    void setLayer(QString layer) { this->current_layer = layer; }
    QString getLayer() { return this->current_layer; }

    QString fixCase(QString);

    int padLeft() { return this->offset_left; }
    int padTop() { return this->offset_top; }
    int padRight() { return this->tilemap_width - this->layout_width - this->offset_left; }
    int padBottom() { return this->tilemap_height - this->layout_height - this->offset_top; }

    int tilemapWidth() { return this->tilemap_width; }
    int tilemapHeight() { return this->tilemap_height; }
    int tilemapSize() { return this->tilemap_width * this->tilemap_height; }
    int tilemapBytes();

    int layoutWidth() { return this->layout_width; }
    int layoutHeight() { return this->layout_height; }
    void setLayoutDimensions(int width, int height, bool update = true);

    int tilemapToLayoutIndex(int index);

    TilemapFormat tilemapFormat() { return this->tilemap_format; }

    int pixelWidth() { return this->tilemap_width * 8; }
    int pixelHeight() { return this->tilemap_height * 8; }

    QString fullPath(QString local);

    void commit(QUndoCommand *command);
    QUndoStack editHistory;

    void undo();
    void redo();

    void emitDisplay();

signals:
    void mapNeedsDisplaying();

private:
    // TODO: defaults needed?
    tsl::ordered_map<QString, MapSectionEntry> *region_map_entries = nullptr;

    QString alias = "";

    int tilemap_width;
    int tilemap_height;

    int layout_width;
    int layout_height;

    int offset_left;
    int offset_top;

    TilemapFormat tilemap_format;

    enum class LayoutFormat { None, Binary, CArray };
    LayoutFormat layout_format;

    QString tileset_path;
    QString tilemap_path;
    QString palette_path = "";

    QString entries_path;
    QString layout_path;

    QString layout_array_label;
    bool layout_uses_layers = false;
    QStringList layout_constants;
    QString layout_qualifiers;

#if QT_VERSION < QT_VERSION_CHECK(6, 0, 0)
    QVector<shared_ptr<TilemapTile>> tilemap;
#else
    QList<shared_ptr<TilemapTile>> tilemap;
#endif

    QStringList layout_layers;
    QString current_layer;
    QMap<QString, QList<LayoutSquare>> layouts;

    int get_tilemap_index(int x, int y);
    int get_layout_index(int x, int y);
};

#endif // REGIONMAP_H