#ifndef YAZE_APP_EDITOR_GRAPHICS_EDITOR_H
#define YAZE_APP_EDITOR_GRAPHICS_EDITOR_H

#include <stack>

#include "absl/status/status.h"
#include "app/editor/editor.h"
#include "app/editor/graphics/palette_editor.h"
#include "app/gfx/bitmap.h"
#include "app/gfx/snes_tile.h"
#include "app/gui/canvas.h"
#include "app/gui/modules/asset_browser.h"
#include "app/rom.h"
#include "app/zelda3/overworld/overworld.h"
#include "imgui/imgui.h"
#include "imgui_memory_editor.h"

namespace yaze {
namespace editor {

// "99973","A3D80",

const std::string kSuperDonkeyTiles[] = {
    "97C05", "98219", "9871E", "98C00", "99084", "995AF", "99DE0", "9A27E",
    "9A741", "9AC31", "9B07E", "9B55C", "9B963", "9BB99", "9C009", "9C4B4",
    "9C92B", "9CDD6", "9D2C2", "9E037", "9E527", "9EA56", "9EF65", "9FCD1",
    "A0193", "A059E", "A0B17", "A0FB6", "A14A5", "A1988", "A1E66", "A232B",
    "A27F0", "A2B6E", "A302C", "A3453", "A38CA", "A42BB", "A470C", "A4BA9",
    "A5089", "A5385", "A5742", "A5BCC", "A6017", "A6361", "A66F8"};

const std::string kSuperDonkeySprites[] = {
    "A8E5D", "A9435", "A9934", "A9D83", "AA2F1", "AA6D4", "AABE4", "AB127",
    "AB65A", "ABBDD", "AC38D", "AC797", "ACCC8", "AD0AE", "AD245", "AD554",
    "ADAAC", "ADECC", "AE453", "AE9D2", "AEF40", "AF3C9", "AF92E", "AFE9D",
    "B03D2", "B09AC", "B0F0C", "B1430", "B1859", "B1E01", "B229A", "B2854",
    "B2D27", "B31D7", "B3B58", "B40B5", "B45A5", "B4D64", "B5031", "B555F",
    "B5F30", "B6858", "B70DD", "B7526", "B79EC", "B7C83", "B80F7", "B85CC",
    "B8A3F", "B8F97", "B94F2", "B9A20", "B9E9A", "BA3A2", "BA8F6", "BACDC",
    "BB1F9", "BB781", "BBCCA", "BC26D", "BC7D4", "BCBB0", "BD082", "BD5FC",
    "BE115", "BE5C2", "BEB63", "BF0CB", "BF607", "BFA55", "BFD71", "C017D",
    "C0567", "C0981", "C0BA7", "C116D", "C166A", "C1FE0", "C24CE", "C2B19"};

/**
 * @class GraphicsEditor
 * @brief Allows the user to edit graphics sheets from the game or view
 * prototype graphics.
 *
 * The GraphicsEditor class is responsible for providing functionality to edit
 * graphics sheets from the game or view prototype graphics of Link to the Past
 * from the CGX, SCR, and OBJ formats. It provides various methods to update
 * different components of the graphics editor, such as the graphics edit tab,
 * link graphics view, and prototype graphics viewer. It also includes import
 * functions for different file formats, as well as other utility functions for
 * drawing toolsets, palette controls, clipboard imports, experimental features,
 * and memory editor.
 */
class GraphicsEditor : public Editor {
 public:
  explicit GraphicsEditor(Rom* rom = nullptr) : rom_(rom) { 
    type_ = EditorType::kGraphics; 
  }

  void Initialize() override;
  absl::Status Load() override;
  absl::Status Save() override { return absl::UnimplementedError("Save"); }
  absl::Status Update() override;
  absl::Status Cut() override { return absl::UnimplementedError("Cut"); }
  absl::Status Copy() override { return absl::UnimplementedError("Copy"); }
  absl::Status Paste() override { return absl::UnimplementedError("Paste"); }
  absl::Status Undo() override { return absl::UnimplementedError("Undo"); }
  absl::Status Redo() override { return absl::UnimplementedError("Redo"); }
  absl::Status Find() override { return absl::UnimplementedError("Find"); }
  
  // Set the ROM pointer
  void set_rom(Rom* rom) { rom_ = rom; }
  
  // Get the ROM pointer
  Rom* rom() const { return rom_; }

 private:
  enum class GfxEditMode {
    kSelect,
    kPencil,
    kFill,
  };

  // Graphics Editor Tab
  absl::Status UpdateGfxEdit();
  absl::Status UpdateGfxSheetList();
  absl::Status UpdateGfxTabView();
  absl::Status UpdatePaletteColumn();
  void DrawGfxEditToolset();

  // Link Graphics Edit Tab
  absl::Status UpdateLinkGfxView();

  // Prototype Graphics Viewer
  absl::Status UpdateScadView();

  // Import Functions
  absl::Status DrawCgxImport();
  absl::Status DrawScrImport();
  absl::Status DrawFileImport();
  absl::Status DrawObjImport();
  absl::Status DrawTilemapImport();

  // Other Functions
  absl::Status DrawToolset();
  absl::Status DrawPaletteControls();
  absl::Status DrawClipboardImport();
  absl::Status DrawExperimentalFeatures();
  absl::Status DrawMemoryEditor();

  absl::Status DecompressImportData(int size);
  absl::Status DecompressSuperDonkey();

  // Member Variables
  ImVec4 current_color_;
  uint16_t current_sheet_ = 0;
  uint8_t tile_size_ = 0x01;
  std::set<uint16_t> open_sheets_;
  std::set<uint16_t> child_window_sheets_;
  std::stack<uint16_t> release_queue_;
  uint64_t edit_palette_group_name_index_ = 0;
  uint64_t edit_palette_group_index_ = 0;
  uint64_t edit_palette_index_ = 0;
  uint64_t edit_palette_sub_index_ = 0;
  float sheet_scale_ = 2.0f;
  float current_scale_ = 4.0f;

  // Prototype Graphics Viewer
  int current_palette_ = 0;
  uint64_t current_offset_ = 0;
  uint64_t current_size_ = 0;
  uint64_t current_palette_index_ = 0;
  int current_bpp_ = 0;
  int scr_mod_value_ = 0;

  uint64_t num_sheets_to_load_ = 1;
  uint64_t bin_size_ = 0;
  uint64_t clipboard_offset_ = 0;
  uint64_t clipboard_size_ = 0;

  bool refresh_graphics_ = false;
  bool open_memory_editor_ = false;
  bool gfx_loaded_ = false;
  bool is_open_ = false;
  bool super_donkey_ = false;
  bool col_file_ = false;
  bool cgx_loaded_ = false;
  bool scr_loaded_ = false;
  bool obj_loaded_ = false;
  bool tilemap_loaded_ = false;

  std::string file_path_ = "";
  std::string col_file_path_ = "";
  std::string col_file_name_ = "";
  std::string cgx_file_path_ = "";
  std::string cgx_file_name_ = "";
  std::string scr_file_path_ = "";
  std::string scr_file_name_ = "";
  std::string obj_file_path_ = "";
  std::string tilemap_file_path_ = "";
  std::string tilemap_file_name_ = "";

  gui::GfxSheetAssetBrowser asset_browser_;

  GfxEditMode gfx_edit_mode_ = GfxEditMode::kSelect;

  Rom temp_rom_;
  Rom tilemap_rom_;
  zelda3::Overworld overworld_{&temp_rom_};
  MemoryEditor cgx_memory_editor_;
  MemoryEditor col_memory_editor_;
  PaletteEditor palette_editor_;
  std::vector<uint8_t> import_data_;
  std::vector<uint8_t> graphics_buffer_;
  std::vector<uint8_t> decoded_cgx_;
  std::vector<uint8_t> cgx_data_;
  std::vector<uint8_t> extra_cgx_data_;
  std::vector<SDL_Color> decoded_col_;
  std::vector<uint8_t> scr_data_;
  std::vector<uint8_t> decoded_scr_data_;
  gfx::Bitmap cgx_bitmap_;
  gfx::Bitmap scr_bitmap_;
  gfx::Bitmap bin_bitmap_;
  gfx::Bitmap link_full_sheet_;
  std::array<gfx::Bitmap, kNumGfxSheets> gfx_sheets_;
  std::array<gfx::Bitmap, kNumLinkSheets> link_sheets_;

  gfx::PaletteGroup col_file_palette_group_;
  gfx::SnesPalette z3_rom_palette_;
  gfx::SnesPalette col_file_palette_;
  gfx::SnesPalette link_palette_;
  gui::Canvas import_canvas_;
  gui::Canvas scr_canvas_;
  gui::Canvas super_donkey_canvas_;
  gui::Canvas graphics_bin_canvas_;
  gui::Canvas current_sheet_canvas_{"CurrentSheetCanvas", ImVec2(0x80, 0x20),
                                    gui::CanvasGridSize::k8x8};
  gui::Canvas link_canvas_{
      "LinkCanvas",
      ImVec2(gfx::kTilesheetWidth * 4, gfx::kTilesheetHeight * 0x10 * 4),
      gui::CanvasGridSize::k16x16};
  absl::Status status_;

  Rom* rom_;
};

}  // namespace editor
}  // namespace yaze

#endif  // YAZE_APP_EDITOR_GRAPHICS_EDITOR_H