#include "display.hpp"

#include "config.h"

using namespace blit;

SurfaceInfo cur_surf_info;

bool fb_double_buffer = true;

#if defined(BUILD_LOADER) || defined(BLIT_BOARD_PIMORONI_PICOVISION)
uint16_t *screen_fb = nullptr;
static uint32_t max_fb_size = 0;
static Size max_fb_bounds(DISPLAY_WIDTH, DISPLAY_HEIGHT);
#else
uint16_t screen_fb[FRAMEBUFFER_SIZE];
static const uint32_t max_fb_size = FRAMEBUFFER_SIZE * sizeof(uint16_t);
#endif

static const Size lores_screen_size(DISPLAY_WIDTH / 2, DISPLAY_HEIGHT / 2);
static const Size hires_screen_size(DISPLAY_WIDTH, DISPLAY_HEIGHT);

ScreenMode cur_screen_mode = ScreenMode::lores;

int get_display_page_size() {
  return max_fb_size / 2;
}

// blit api

SurfaceInfo &set_screen_mode(ScreenMode mode) {
  SurfaceTemplate temp{nullptr, {0, 0}, mode == ScreenMode::hires_palette ? PixelFormat::P : DEFAULT_SCREEN_FORMAT};

  // may fail for hires/palette
  if(set_screen_mode_format(mode, temp)) {
    cur_surf_info.data = temp.data;
    cur_surf_info.palette = temp.palette;
  }

  return cur_surf_info;
}

bool set_screen_mode_format(ScreenMode new_mode, SurfaceTemplate &new_surf_template) {
  new_surf_template.data = (uint8_t *)screen_fb;

  if(new_surf_template.format == (PixelFormat)-1)
    new_surf_template.format = DEFAULT_SCREEN_FORMAT;

  int min_buffers = 1;

  switch(new_mode) {
    case ScreenMode::lores:
      if(new_surf_template.bounds.empty())
        new_surf_template.bounds = lores_screen_size;
      else
        new_surf_template.bounds /= 2;

#ifdef BUILD_LOADER
      if(new_surf_template.bounds.w > max_fb_bounds.w / 2)
        new_surf_template.bounds.w = max_fb_bounds.w / 2;
#endif
      min_buffers = 2;
      break;
    case ScreenMode::hires:
    case ScreenMode::hires_palette:
      if(new_surf_template.bounds.empty())
        new_surf_template.bounds = hires_screen_size;

#ifdef BUILD_LOADER
      if(new_surf_template.bounds.w > max_fb_bounds.w)
        new_surf_template.bounds.w = max_fb_bounds.w;
#endif
      break;
  }

  // check the framebuffer is large enough for mode
  auto fb_size = uint32_t(new_surf_template.bounds.area()) * pixel_format_stride[int(new_surf_template.format)];
// TODO: more generic "doesn't have a framebuffer"?
#ifndef BLIT_BOARD_PIMORONI_PICOVISION
  if(max_fb_size < fb_size * min_buffers)
    return false;
#endif

  if(!display_mode_supported(new_mode, new_surf_template))
    return false;

  fb_double_buffer = fb_size * 2 <= max_fb_size;
  if(!fb_double_buffer)
    screen.data = new_surf_template.data;

  cur_surf_info.bounds = new_surf_template.bounds;
  cur_surf_info.format = new_surf_template.format;

  display_mode_changed(new_mode, new_surf_template);

  cur_screen_mode = new_mode;

#ifdef BUILD_LOADER
  screen = Surface(new_surf_template.data, new_surf_template.format, new_surf_template.bounds);
#endif

  return true;
}

void set_screen_palette(const Pen *colours, int num_cols) {

}

void set_framebuffer(uint8_t *data, uint32_t max_size, Size max_bounds) {
#if defined(BUILD_LOADER) && !defined(BLIT_BOARD_PIMORONI_PICOVISION)
  screen_fb = (uint16_t *)data;
  screen.data = data;
  max_fb_size = max_size;
  max_fb_bounds = max_bounds;
#endif
}