Painter.hpp

#pragma once
#include <score/model/Skin.hpp>
 
#include <QFlags>
#include <QGradient>
#include <QGraphicsItem>
#include <QGraphicsSceneMouseEvent>
#include <QGraphicsSceneWheelEvent>
#include <QPainter>
#include <QPolygon>
 
#include <avnd/concepts/painter.hpp>
#include <avnd/wrappers/colors.hpp>
 
#include <algorithm>
#include <cmath>
 
namespace oscr
{
struct rgba_color
{
  uint8_t r, g, b, a;
};
struct QPainterAdapter
{
  QPainter& painter;
  QGraphicsItem& item;
  QPainterPath path;
  score::Skin& skin = score::Skin::instance();
 
  void begin_path() { path = QPainterPath{}; }
  void close_path() { path.closeSubpath(); }
  void stroke() { painter.strokePath(path, painter.pen()); }
  void fill() { painter.fillPath(path, painter.brush()); }
  void update() { item.update(); }
 
  void move_to(double x, double y) { path.moveTo(x, y); }
  void line_to(double x, double y) { path.lineTo(x, y); }
  void arc_to(double x, double y, double w, double h, double start, double length)
  {
    path.arcTo(x, y, w, h, start, length);
  }
 
  void cubic_to(double c1x, double c1y, double c2x, double c2y, double endx, double endy)
  {
    path.cubicTo(c1x, c1y, c2x, c2y, endx, endy);
  }
 
  void quad_to(double x1, double y1, double x2, double y2)
  {
    path.quadTo(x1, y1, x2, y2);
  }
 
  void translate(double x, double y) { painter.translate(x, y); }
  void scale(double x, double y) { painter.scale(x, y); }
  void rotate(double a) { painter.rotate(a); }
  void reset_transform() { painter.resetTransform(); }
 
  // Colors:
  void unset_stroke() { painter.setPen(score::Skin::instance().NoPen); }
 
  rgba_color to_rgba(auto c)
  {
    const auto standard_color = avnd::get_color(c);
    static constexpr auto convert = [](QColor c) constexpr -> rgba_color {
      const auto rgba = c.rgba();
      return {
          (uint8_t)qRed(rgba),
          (uint8_t)qGreen(rgba),
          (uint8_t)qBlue(rgba),
          (uint8_t)qAlpha(rgba),
      };
    };
 
    using enum avnd::color_type;
    switch(standard_color)
    {
      case dark:
        return convert(skin.Dark.main.brush.color());
      case darker:
        return convert(skin.HalfDark.main.brush.color());
      case mid:
        return convert(skin.Gray.main.brush.color());
      case lighter:
        return convert(skin.HalfLight.main.brush.color());
      case light:
        return convert(skin.Light.main.brush.color());
 
      case background_darker:
        return convert(skin.Background2.darker300.brush.color());
      case background_dark:
        return convert(skin.Background2.darker.brush.color());
      case background_mid:
        return convert(skin.Background2.main.brush.color());
      case background_light:
        return convert(skin.Background2.lighter.brush.color());
      case background_lighter:
        return convert(skin.Background2.lighter180.brush.color());
 
      case runtime_value_dark:
        return convert(skin.Base4.darker.brush.color());
      case runtime_value_mid:
        return convert(skin.Base4.main.brush.color());
      case runtime_value_light:
        return convert(skin.Base4.lighter.brush.color());
 
      case editable_value_dark:
        return convert(skin.Emphasis2.darker.brush.color());
      case editable_value_mid:
        return convert(skin.Emphasis2.main.brush.color());
      case editable_value_light:
        return convert(skin.Emphasis2.lighter.brush.color());
    }
 
    return {};
  }
 
  void set_stroke_color(rgba_color c)
  {
    QPen p = painter.pen();
    p.setColor(QColor::fromRgb(c.r, c.g, c.b, c.a));
    painter.setPen(p);
  }
 
  template <typename E>
    requires std::is_enum_v<E>
  void set_stroke_color(E c)
  {
    set_stroke_color(to_rgba(c));
  }
 
  void set_stroke_width(double w)
  {
    QPen p = painter.pen();
    p.setWidth(w);
    painter.setPen(p);
  }
 
  void set_fill_color(rgba_color c)
  {
    painter.setBrush(QColor::fromRgb(c.r, c.g, c.b, c.a));
  }
 
  template <typename E>
    requires std::is_enum_v<E>
  void set_fill_color(E c)
  {
    set_fill_color(to_rgba(c));
  }
 
  void set_linear_gradient(
      double x1, double y1, double x2, double y2, rgba_color c1, rgba_color c2)
  {
    QLinearGradient gradient(QPointF(x1, y1), QPointF(x2, y2));
    gradient.setColorAt(0, QColor::fromRgb(c1.r, c1.g, c1.b, c1.a));
    gradient.setColorAt(1, QColor::fromRgb(c2.r, c2.g, c2.b, c2.a));
    painter.setBrush(gradient);
  }
 
  void set_radial_gradient(double cx, double cy, double cr, rgba_color c1, rgba_color c2)
  {
    QRadialGradient gradient(cx, cy, cr);
    gradient.setColorAt(0, QColor::fromRgb(c1.r, c1.g, c1.b, c1.a));
    gradient.setColorAt(1, QColor::fromRgb(c2.r, c2.g, c2.b, c2.a));
    painter.setBrush(gradient);
  }
 
  void set_conical_gradient(double x, double y, double a, rgba_color c1, rgba_color c2)
  {
    QConicalGradient gradient(x, y, a);
    gradient.setColorAt(0, QColor::fromRgb(c1.r, c1.g, c1.b, c1.a));
    gradient.setColorAt(1, QColor::fromRgb(c2.r, c2.g, c2.b, c2.a));
    painter.setBrush(gradient);
  }
 
  template <typename E>
    requires std::is_enum_v<E>
  void
  set_linear_gradient(double x1, double y1, double x2, double y2, E c1, rgba_color c2)
  {
    set_linear_gradient(x1, y1, x2, y2, to_rgba(c1), c2);
  }
 
  template <typename E>
    requires std::is_enum_v<E>
  void
  set_linear_gradient(double x1, double y1, double x2, double y2, rgba_color c1, E c2)
  {
    set_linear_gradient(x1, y1, x2, y2, c1, to_rgba(c2));
  }
 
  template <typename E1, typename E2>
    requires std::is_enum_v<E1> && std::is_enum_v<E2>
  void set_linear_gradient(double x1, double y1, double x2, double y2, E1 c1, E2 c2)
  {
    set_linear_gradient(x1, y1, x2, y2, to_rgba(c1), to_rgba(c2));
  }
 
  template <typename E>
    requires std::is_enum_v<E>
  void
  set_radial_gradient(double x1, double y1, double x2, double y2, E c1, rgba_color c2)
  {
    set_radial_gradient(x1, y1, x2, y2, to_rgba(c1), c2);
  }
 
  template <typename E>
    requires std::is_enum_v<E>
  void
  set_radial_gradient(double x1, double y1, double x2, double y2, rgba_color c1, E c2)
  {
    set_radial_gradient(x1, y1, x2, y2, c1, to_rgba(c2));
  }
 
  template <typename E1, typename E2>
    requires std::is_enum_v<E1> && std::is_enum_v<E2>
  void set_radial_gradient(double x1, double y1, double x2, double y2, E1 c1, E2 c2)
  {
    set_radial_gradient(x1, y1, x2, y2, to_rgba(c1), to_rgba(c2));
  }
 
  template <typename E>
    requires std::is_enum_v<E>
  void
  set_conical_gradient(double x1, double y1, double x2, double y2, E c1, rgba_color c2)
  {
    set_conical_gradient(x1, y1, x2, y2, to_rgba(c1), c2);
  }
 
  template <typename E>
    requires std::is_enum_v<E>
  void
  set_conical_gradient(double x1, double y1, double x2, double y2, rgba_color c1, E c2)
  {
    set_conical_gradient(x1, y1, x2, y2, c1, to_rgba(c2));
  }
 
  template <typename E1, typename E2>
    requires std::is_enum_v<E1> && std::is_enum_v<E2>
  void set_conical_gradient(double x1, double y1, double x2, double y2, E1 c1, E2 c2)
  {
    set_conical_gradient(x1, y1, x2, y2, to_rgba(c1), to_rgba(c2));
  }
 
  // Text:
  void set_font(std::string_view f)
  {
    auto font = painter.font();
    font.setFamily(QString::fromUtf8(f.data(), f.size()));
    painter.setFont(font);
  }
 
  void set_font_size(double f)
  {
    auto font = painter.font();
    font.setPixelSize(std::max(1, (int)std::lround(f)));
    painter.setFont(font);
  }
 
  void draw_text(double x, double y, std::string_view str)
  {
    path.addText(x, y, painter.font(), QString::fromUtf8(str.data(), str.size()));
  }
 
  void draw_text(double x, double y, double w, double h, std::string_view str)
  {
    QRectF rect{x, y, w, h};
    const auto& m = painter.fontMetrics();
    auto txt = QString::fromUtf8(str.data(), str.size());
    auto text_rect = m.boundingRect(txt);
    auto text_half_diagonal = (text_rect.center() - text_rect.topLeft()) / 2.;
    auto pos = rect.center() + text_half_diagonal;
    path.addText(pos.x(), pos.y(), painter.font(), txt);
  }
 
  // Drawing
  void draw_line(double x1, double y1, double x2, double y2)
  {
    path.moveTo(x1, y1);
    path.lineTo(x2, y2);
  }
 
  //          x1, y1, x2 , y2, x3, y3
  void draw_triangle(double x1, double y1, double x2, double y2, double x3, double y3)
  {
    path.moveTo(x1, y1);
    path.lineTo(x2, y2);
    path.lineTo(x3, y3);
    path.lineTo(x1, y1);
    painter.drawPath(path);
  }
 
  //          x , y , w  , h
  void draw_rect(double x, double y, double w, double h) { path.addRect(x, y, w, h); }
 
  //                  x , y , w  , h
  void draw_rounded_rect(double x, double y, double w, double h, double r)
  {
    path.addRoundedRect(x, y, w, h, r, r);
  }
 
  //            x , y , filename
  void draw_pixmap(double x, double y, const QString& str)
  {
    painter.drawPixmap(x, y, str);
  }
 
  //             x , y , w  , h
  void draw_ellipse(double x, double y, double w, double h)
  {
    path.addEllipse(x, y, w, h);
  }
 
  //            cx, cy, radius
  void draw_circle(double cx, double cy, double cr)
  {
    path.addEllipse(QPointF{cx, cy}, cr, cr);
  }
 
  //          tab, count
  void draw_polygon(const double* tab, int count)
  {
    QPolygonF poly;
    double x, y;
    for(int i = 0; i < count * 2; i += 2)
    {
      x = tab[i];
      y = tab[i + 1];
      poly << QPointF(x, y);
    }
    path.addPolygon(poly);
  }
 
  void draw_bytes(
      int x, int y, int w, int h, const unsigned char* image, int img_w, int img_h,
      bool smooth = false)
  {
    auto img = QImage(image, img_w, img_h, QImage::Format_RGBA8888);
    auto prev = painter.renderHints() & QPainter::SmoothPixmapTransform;
    painter.setRenderHint(QPainter::SmoothPixmapTransform, smooth);
    painter.drawImage(
        QRect(x, y, w, h), img, QRect(0, 0, img_w, img_h), Qt::ImageConversionFlags{});
 
    painter.setRenderHint(QPainter::SmoothPixmapTransform, prev);
  }
  void draw_bytes(
      int x, int y, int w, int h, const float* image, int img_w, int img_h,
      bool smooth = false)
  {
    auto img
        = QImage((const unsigned char*)image, img_w, img_h, QImage::Format_RGBA32FPx4);
    auto prev = painter.renderHints() & QPainter::SmoothPixmapTransform;
    painter.setRenderHint(QPainter::SmoothPixmapTransform, smooth);
    painter.drawImage(
        QRect(x, y, w, h), img, QRect(0, 0, img_w, img_h), Qt::ImageConversionFlags{});
 
    painter.setRenderHint(QPainter::SmoothPixmapTransform, prev);
  }
};
static_assert(avnd::painter<QPainterAdapter>);
 
template <typename Item, typename Control = void>
class CustomItem : public QGraphicsItem
{
public:
  // Item may be T::item_type or T&
  using item_type = std::decay_t<Item>;
 
  // // Case T::item_type
  // explicit CustomItem() { init(); }
 
  // Case T&
  explicit CustomItem(Item t)
      : impl{t}
  {
    this->setFlag(QGraphicsItem::ItemClipsToShape);
    this->setFlag(QGraphicsItem::ItemClipsChildrenToShape);
 
    if constexpr(requires { this->impl.update = [this] {}; })
    {
      this->impl.update = [this] { this->update(); };
    }
  }
 
  QRectF boundingRect() const override
  {
    return {0., 0., item_type::width(), item_type::height()};
  }
 
  void paint(QPainter* painter, const QStyleOptionGraphicsItem* option, QWidget* widget)
      override
  {
    auto& skin = score::Skin::instance();
    painter->setRenderHint(QPainter::Antialiasing, true);
    painter->setPen(skin.Dark.main.pen1);
    // Views set DontSavePainterState: else the previous item's font leaks
    painter->setFont(skin.Medium8Pt);
    impl.paint(QPainterAdapter{*painter, *this, {}});
    painter->setRenderHint(QPainter::Antialiasing, false);
  }
 
  struct custom_mouse_event
  {
    enum button
    {
      no_button,
      left = (1 << 1),
      right = (1 << 2),
      middle = (1 << 3)
    };
    friend button& operator|=(button& lhs, button rhs) noexcept
    {
      return (
          enum button&)(reinterpret_cast<std::underlying_type_t<enum button>&>(lhs) |= reinterpret_cast<std::underlying_type_t<enum button>&>(rhs));
    }
    enum modifier
    {
      no_modifier,
      shift = (1 << 1),
      ctrl = (1 << 2),
      alt = (1 << 3),
      meta = (1 << 4)
    };
    friend modifier& operator|=(modifier& lhs, modifier rhs) noexcept
    {
      return (
          enum modifier&)(reinterpret_cast<std::underlying_type_t<enum modifier>&>(lhs) |= reinterpret_cast<std::underlying_type_t<enum modifier>&>(rhs));
    }
 
    float x, y;
 
    enum button button = {};
    enum button held_buttons = {};
    enum modifier modifiers = {};
  };
 
protected:
  static custom_mouse_event make_event(QGraphicsSceneMouseEvent* event) noexcept
  {
    custom_mouse_event p;
 
    p.x = event->pos().x();
    p.y = event->pos().y();
 
    if(event->button() == Qt::LeftButton)
      p.button = custom_mouse_event::left;
    else if(event->button() == Qt::RightButton)
      p.button = custom_mouse_event::right;
    else if(event->button() == Qt::MiddleButton)
      p.button = custom_mouse_event::middle;
 
    if(event->buttons() & Qt::LeftButton)
      p.held_buttons |= p.left;
    if(event->buttons() & Qt::RightButton)
      p.held_buttons |= p.right;
    if(event->buttons() & Qt::MiddleButton)
      p.held_buttons |= p.middle;
 
    if(event->modifiers() & Qt::ShiftModifier)
      p.modifiers |= p.shift;
    if(event->modifiers() & Qt::AltModifier)
      p.modifiers |= p.alt;
    if(event->modifiers() & Qt::ControlModifier)
      p.modifiers |= p.ctrl;
    if(event->modifiers() & Qt::MetaModifier)
      p.modifiers |= p.meta;
 
    return p;
  }
 
  void mousePressEvent(QGraphicsSceneMouseEvent* event) override
  {
    if constexpr(requires { impl.mouse_press(0, 0); })
    {
      if(impl.mouse_press(event->pos().x(), event->pos().y()))
        event->accept();
    }
    else if constexpr(requires { impl.mouse_press(custom_mouse_event{}); })
    {
      if(impl.mouse_press(make_event(event)))
        event->accept();
    }
    else
    {
      event->ignore();
    }
    update();
  }
 
  void mouseMoveEvent(QGraphicsSceneMouseEvent* event) override
  {
    if constexpr(requires { impl.mouse_move(0, 0); })
    {
      impl.mouse_move(event->pos().x(), event->pos().y());
      event->accept();
    }
    else if constexpr(requires { impl.mouse_move(custom_mouse_event{}); })
    {
      impl.mouse_move(make_event(event));
      event->accept();
    }
    else
    {
      event->ignore();
    }
    update();
  }
 
  void mouseReleaseEvent(QGraphicsSceneMouseEvent* event) override
  {
    if constexpr(requires { impl.mouse_release(0, 0); })
    {
      impl.mouse_release(event->pos().x(), event->pos().y());
      event->accept();
    }
    else if constexpr(requires { impl.mouse_release(custom_mouse_event{}); })
    {
      impl.mouse_release(make_event(event));
      event->accept();
    }
    else
    {
      event->ignore();
    }
    update();
  }
 
  void wheelEvent(QGraphicsSceneWheelEvent* event) override
  {
    if constexpr(requires { impl.wheel(0., 0., 0.); })
    {
      impl.wheel(event->pos().x(), event->pos().y(), event->delta());
      event->accept();
      update();
    }
    else
    {
      event->ignore();
    }
  }
 
protected:
  Item impl;
};
 
template <typename Item>
class CustomControl : public CustomItem<Item>
{
public:
  OngoingCommandDispatcher cmd;
  explicit CustomControl(
      Item item_init, Process::ControlInlet& ctl, const score::DocumentContext& ctx)
      : CustomItem<Item>{item_init}
      , cmd{ctx.commandStack}
  {
    if constexpr(requires { this->impl.transaction; })
    {
      this->impl.transaction.start = [] {
 
      };
      this->impl.transaction.update = [this, &ctl](const auto& value) {
        auto val = oscr::to_ossia_value(value);
        cmd.submit<Process::SetControlValue>(ctl, val);
 
        this->impl.value = value; //impl.value_to_control(Control{}
        this->update();
      };
      this->impl.transaction.commit = [this] { cmd.commit(); };
      this->impl.transaction.rollback = [this] { cmd.rollback(); };
    }
  }
};
}