#pragma once

#include "drivers/st7789/st7789.hpp"
#include "libraries/pico_graphics/pico_graphics.hpp"

namespace pimoroni {

  class PicoDisplay : public PicoGraphics {
  public:
    static const int WIDTH = 240;
    static const int HEIGHT = 135;
    static const int PORTRAIT_WIDTH  = 135;
    static const int PORTRAIT_HEIGHT = 240;
    static const uint8_t A = 12;
    static const uint8_t B = 13;
    static const uint8_t X = 14;
    static const uint8_t Y = 15;
    static const uint8_t LED_R = 6;
    static const uint8_t LED_G = 7;
    static const uint8_t LED_B = 8;

    void *__fb;
  private:
    ST7789 screen;

  public:
    PicoDisplay(void *buf);
    PicoDisplay(void *buf, int width, int height);

    void init();
    void update();
    void set_backlight(uint8_t brightness);
    void set_led(uint8_t r, uint8_t g, uint8_t b);
    bool is_pressed(uint8_t button);
    void flip();
  };

}