#include <cstdint>
#include <bit>

#include "holly/texture_memory_alloc2.hpp"
#include "holly/isp_tsp.hpp"
#include "holly/ta_fifo_polygon_converter.hpp"
#include "holly/ta_parameter.hpp"
#include "holly/ta_global_parameter.hpp"
#include "holly/ta_vertex_parameter.hpp"
#include "holly/ta_bits.hpp"
#include "holly/background.hpp"
#include "holly/region_array.hpp"
#include "holly/holly.hpp"
#include "holly/core.hpp"
#include "holly/core_bits.hpp"
#include "holly/video_output.hpp"

#include "sh7091/store_queue.hpp"
#include "sh7091/serial.hpp"
#include "sh7091/vbr.hpp"
#include "sh7091/sh7091.hpp"
#include "sh7091/sh7091_bits.hpp"

#include "systembus.hpp"
#include "systembus_bits.hpp"

#include "font/font.h"
#include "font/dejavusansmono/dejavusansmono.data.h"

#include "palette.hpp"

#include "printf.h"
#include "runner.h"

#include "maple/maple.hpp"
#include "maple/maple_host_command_writer.hpp"
#include "maple/maple_bus_bits.hpp"
#include "maple/maple_bus_commands.hpp"
#include "maple/maple_bus_ft0.hpp"

static ft0::data_transfer::data_format maple_ft0_data[4];

static uint8_t send_buf[1024] __attribute__((aligned(32)));
static uint8_t recv_buf[1024] __attribute__((aligned(32)));

static bool maple_wait = false;

void do_get_condition()
{
  if (maple_wait)
    maple::dma_wait_complete();

  auto writer = maple::host_command_writer(send_buf, recv_buf);

  writer.append_command_all_ports<maple::device_request, maple::device_status>(false);

  using command_type = maple::get_condition;
  using response_type = maple::data_transfer<ft0::data_transfer::data_format>;

  auto [host_command, host_response]
    = writer.append_command_all_ports<command_type, response_type>();

  host_command->bus_data.data_fields.function_type = std::byteswap(function_type::controller);

  maple::dma_start(send_buf, writer.send_offset,
                   recv_buf, writer.recv_offset);
  maple_wait = true;

  for (uint8_t port = 0; port < 4; port++) {
    auto& bus_data = host_response[port].bus_data;
    maple_ft0_data[port].digital_button = 0;
    if (bus_data.command_code != response_type::command_code) {
      return;
    }
    auto& data_fields = bus_data.data_fields;
    if ((std::byteswap(data_fields.function_type) & function_type::controller) == 0) {
      return;
    }

    maple_ft0_data[port].digital_button = data_fields.data.digital_button;
    maple_ft0_data[port].analog_coordinate_axis[0] = data_fields.data.analog_coordinate_axis[0];
    maple_ft0_data[port].analog_coordinate_axis[1] = data_fields.data.analog_coordinate_axis[1];
    maple_ft0_data[port].analog_coordinate_axis[2] = data_fields.data.analog_coordinate_axis[2];
    maple_ft0_data[port].analog_coordinate_axis[3] = data_fields.data.analog_coordinate_axis[3];
    maple_ft0_data[port].analog_coordinate_axis[4] = data_fields.data.analog_coordinate_axis[4];
    maple_ft0_data[port].analog_coordinate_axis[5] = data_fields.data.analog_coordinate_axis[5];
  }
}

struct vertex {
  float x;
  float y;
  float z;
  float u;
  float v;
};

const struct vertex strip_vertices[4] = {
  // [ position       ]  [ uv coordinates ]
  { 0.f,  1.f,  0.f, 0.f, 1.f, },
  { 0.f,  0.f,  0.f, 0.f, 0.f, },
  { 1.f,  1.f,  0.f, 1.f, 1.f, },
  { 1.f,  0.f,  0.f, 1.f, 0.f, },
};
constexpr uint32_t strip_length = (sizeof (strip_vertices)) / (sizeof (struct vertex));


void glyph_start(const uint32_t texture_width, uint32_t texture_height)
{
  const uint32_t parameter_control_word = para_control::para_type::polygon_or_modifier_volume
                                        | para_control::list_type::translucent
                                        | obj_control::col_type::packed_color
                                        | obj_control::texture;

  const uint32_t isp_tsp_instruction_word = isp_tsp_instruction_word::depth_compare_mode::greater
                                          | isp_tsp_instruction_word::culling_mode::no_culling;

  const uint32_t tsp_instruction_word = tsp_instruction_word::src_alpha_instr::src_alpha
                                      | tsp_instruction_word::dst_alpha_instr::one
                                      | tsp_instruction_word::fog_control::no_fog
                                      | tsp_instruction_word::use_alpha
                                      | tsp_instruction_word::texture_u_size::from_int(texture_width)
                                      | tsp_instruction_word::texture_v_size::from_int(texture_height);

  const uint32_t texture_address = texture_memory_alloc::texture.start;
  const uint32_t texture_control_word = texture_control_word::pixel_format::_8bpp_palette
                                      | texture_control_word::scan_order::twiddled
                                      | texture_control_word::texture_address(texture_address / 8);

  *reinterpret_cast<ta_global_parameter::polygon_type_0 *>(store_queue) =
    ta_global_parameter::polygon_type_0(parameter_control_word,
                                        isp_tsp_instruction_word,
                                        tsp_instruction_word,
                                        texture_control_word,
                                        0, // data_size_for_sort_dma
                                        0  // next_address_for_sort_dma
                                        );
  sq_transfer_32byte(ta_fifo_polygon_converter);
}

int32_t transform_char(const uint32_t texture_width,
                       const uint32_t texture_height,
                       const uint32_t first_char_code,
                       const glyph * glyphs,
                       const char c,
                       int32_t horizontal_advance,
                       int32_t vertical_advance)
{
  auto& glyph = glyphs[c - first_char_code];

  for (uint32_t i = 0; i < strip_length; i++) {
    float x = strip_vertices[i].x;
    float y = strip_vertices[i].y;
    float z = strip_vertices[i].z;

    x *= glyph.bitmap.width;
    y *= glyph.bitmap.height;
    x += (float)(horizontal_advance + glyph.metrics.horiBearingX) / 64.0;
    y += (float)(vertical_advance - glyph.metrics.horiBearingY) / 64.0;
    z = 1.f / (z + 10.f);

    float u = strip_vertices[i].u;
    float v = strip_vertices[i].v;
    u *= glyph.bitmap.width;
    v *= glyph.bitmap.height;
    u += glyph.bitmap.x;
    v += glyph.bitmap.y;
    u = u / static_cast<float>(texture_width);
    v = v / static_cast<float>(texture_height);

    bool end_of_strip = i == strip_length - 1;
    *reinterpret_cast<ta_vertex_parameter::polygon_type_3 *>(store_queue) =
      ta_vertex_parameter::polygon_type_3(polygon_vertex_parameter_control_word(end_of_strip),
                                          x, y, z,
                                          u, v,
                                          0, // base_color
                                          0  // offset_color
                                          );
    sq_transfer_32byte(ta_fifo_polygon_converter);
  }

  return glyph.metrics.horiAdvance;
}

void transfer_scene(const struct font * font,
                    const struct glyph * glyphs)
{
  glyph_start(font->texture_width, font->texture_height);

  int32_t horizontal_advance = font->face_metrics.max_advance / 5; // 26.6 fixed point
  int32_t vertical_advance = font->face_metrics.height; // 26.6 fixed point

  for (int i = 0; i < global_output_buffer.buf_ix; i++) {
    char c = global_output_buffer.buf[i];
    if (c < 0x20 && c > 0x7f) {
      continue;
    } else if (c == '\n') {
      horizontal_advance = font->face_metrics.max_advance / 5;
      vertical_advance += font->face_metrics.height;
    } else {
      horizontal_advance += transform_char(font->texture_width,
                                           font->texture_height,
                                           font->first_char_code,
                                           glyphs,
                                           c,
                                           horizontal_advance,
                                           vertical_advance);
    }
  }
}

void copy_font(const uint8_t * src,
               const uint32_t z_curve_max_ix)
{
  auto texture = reinterpret_cast<volatile uint32_t *>(&texture_memory64[texture_memory_alloc::texture.start / 4]);

  for (uint32_t i = 0; i < z_curve_max_ix / 4; i++) {
    texture[i] = reinterpret_cast<const uint32_t *>(src)[i];
  }
}

constexpr uint32_t ta_alloc = ta_alloc_ctrl::pt_opb::no_list
                            | ta_alloc_ctrl::tm_opb::no_list
                            | ta_alloc_ctrl::t_opb::_32x4byte
                            | ta_alloc_ctrl::om_opb::no_list
                            | ta_alloc_ctrl::o_opb::no_list;

constexpr int render_passes = 1;
constexpr struct opb_size opb_size[render_passes] = {
  {
    .opaque = 0,
    .opaque_modifier = 0,
    .translucent = 32 * 4,
    .translucent_modifier = 0,
    .punch_through = 0
  }
};

static int ta;
static int core;

const struct font * font;
const struct glyph * glyphs;

constexpr int framebuffer_width = 640;
constexpr int framebuffer_height = 480;
constexpr int tile_width = framebuffer_width / 32;
constexpr int tile_height = framebuffer_height / 32;

static uint32_t frame = 0;

struct runner_state runner_state = {0};

void render()
{
  if (core >= 0) {
    // core = 0  ; core = 1
    // ta = 1    ; ta = 0
    core_wait_end_of_render_video();
    frame += 1;
    holly.FB_R_SOF1 = texture_memory_alloc::framebuffer[core].start;
  }

  // core = -2 ; core = 1 ; core = 0
  // ta = -1   ; ta = 0   ; ta = 1
  core += 1;
  ta += 1;
  if (core > 1) core = 0;
  if (ta > 1) ta = 0;

  if (core >= 0) {
    // core = 1 ; core = 0
    // ta = 0   ; ta = 1
    ta_wait_translucent_list();

    core_start_render2(texture_memory_alloc::region_array[core].start,
                       texture_memory_alloc::isp_tsp_parameters[core].start,
                       texture_memory_alloc::background[core].start,
                       texture_memory_alloc::framebuffer[core].start,
                       framebuffer_width);
  }

  // core = -1 ; core = 1 ; core = 0
  // ta = 0    ; ta = 0   ; ta = 1
  ta_polygon_converter_init2(texture_memory_alloc::isp_tsp_parameters[ta].start,
                             texture_memory_alloc::isp_tsp_parameters[ta].end,
                             texture_memory_alloc::object_list[ta].start,
                             texture_memory_alloc::object_list[ta].end,
                             opb_size[0].total(),
                             ta_alloc,
                             tile_width,
                             tile_height);
  ///
  transfer_scene(font, glyphs);
  if (runner_state.want_render) {
    runner_render(&runner_state, font, glyphs, maple_ft0_data);
  }

  *reinterpret_cast<ta_global_parameter::end_of_list *>(store_queue) =
    ta_global_parameter::end_of_list(para_control::para_type::end_of_list);
  sq_transfer_32byte(ta_fifo_polygon_converter);
}

void vbr600()
{
  /*
  serial::string("vbr600\n");
  serial::string("expevt ");
  serial::integer<uint16_t>(sh7091.CCN.EXPEVT);
  serial::string("intevt ");
  serial::integer<uint16_t>(sh7091.CCN.INTEVT);
  serial::string("tra ");
  serial::integer<uint16_t>(sh7091.CCN.TRA);

  serial::string("istnrm ");
  serial::integer<uint32_t>(system.ISTNRM & system.IML6NRM);
  */

  do_get_condition();

  render();

  // reset v_blank_in interrupt
  system.ISTNRM = istnrm::v_blank_in_interrupt;

  return;
}

void vbr400()
{
  serial::string("vbr400\n");
  while (1);
}

void vbr100()
{
  serial::string("vbr100\n");
  serial::string("expevt ");
  serial::integer<uint16_t>(sh7091.CCN.EXPEVT);
  serial::string("intevt ");
  serial::integer<uint16_t>(sh7091.CCN.INTEVT);
  serial::string("tra ");
  serial::integer<uint16_t>(sh7091.CCN.TRA);
  while (1);
}

void interrupt_init()
{
  sh7091.CCN.INTEVT = 0;
  sh7091.CCN.EXPEVT = 0;

  system.IML2NRM = 0;
  system.IML2ERR = 0;
  system.IML2EXT = 0;

  system.IML4NRM = 0;
  system.IML4ERR = 0;
  system.IML4EXT = 0;

  system.IML6NRM = 0;
  system.IML6ERR = 0;
  system.IML6EXT = 0;

  // enable v-blank-in interrupt at IRL level 6
  system.IML6NRM = istnrm::v_blank_in_interrupt;

  const uint32_t vbr = reinterpret_cast<uint32_t>(&__vbr_link_start) - 0x100;
  // set VBR
  asm volatile ("ldc %0,vbr"
		:
		: "r" (vbr));

  uint32_t sr;
  // read current SR
  asm volatile ("stc sr,%0"
		: "=r" (sr));

  // unset interrupt block bit
  sr &= ~sh::sr::bl;
  // enable interrupt level 6
  sr &= ~sh::sr::imask(15);
  sr |= sh::sr::imask(0b0100);

  // set new SR
  asm volatile ("ldc %0,sr"
		:
		: "r" (sr));
}

int main()
{
  serial::init(0);
  serial::string("framebuffer: ");
  serial::integer<uint32_t>((uint32_t)&texture_memory32[texture_memory_alloc::framebuffer[0].start / 4]);

  holly.SOFTRESET = softreset::pipeline_soft_reset
		  | softreset::ta_soft_reset;
  holly.SOFTRESET = 0;

  core_init();

  region_array_multipass(tile_width,
			 tile_height,
			 opb_size,
			 render_passes,
			 texture_memory_alloc::region_array[0].start,
			 texture_memory_alloc::object_list[0].start);
  region_array_multipass(tile_width,
			 tile_height,
			 opb_size,
			 render_passes,
			 texture_memory_alloc::region_array[1].start,
			 texture_memory_alloc::object_list[1].start);

  background_parameter2(texture_memory_alloc::background[0].start,
			0xff220033);
  background_parameter2(texture_memory_alloc::background[1].start,
			0xff220033);

  font = reinterpret_cast<const struct font *>(&_binary_font_dejavusansmono_dejavusansmono_data_start);
  glyphs = reinterpret_cast<const struct glyph *>(&font[1]);
  auto texture = reinterpret_cast<const uint8_t *>(&glyphs[font->glyph_count]);

  copy_font(texture, font->max_z_curve_ix);
  palette_data<256>();

  video_output::set_mode_vga();

  ta = -1;
  core = -2;

  interrupt_init();

  uint32_t done_frame = 0;
  bool done = false;
  //int last_state = -1;
  while (true) {
    if (!done && runner_tick(&runner_state)) {
      done = true;
      done_frame = frame;
    }
    if (done && (frame == done_frame + 3))
      break;
    /*
    if (runner_state.tick != last_state) {
      last_state = runner_state.tick;
      serial::integer<uint32_t>(runner_state.tick);
    }
    */
  }

  serial::string("return\n");
  serial::string("return\n");
  serial::string("return\n");
  serial::string("return\n");
}