hilbert-os/libraries/daguerre/include/daguerre.hpp

#pragma once

#include <stdint.h>
#include <cstring>
#include <cstdio>

namespace daguerre {

  typedef uint32_t hilbert_color;

  struct rgb24 {
    uint8_t r;
    uint8_t g;
    uint8_t b;
  };

  static inline void default_overlay(hilbert_color &dest, const rgb24 &src) {
    dest = __euler_encode_color(src.r, src.g, src.b);
  }

  static inline void default_overlay(rgb24 &dest, const bool &src) {
    dest.r = src ? 255 : 0;
    dest.g = src ? 255 : 0;
    dest.b = src ? 255 : 0;
  }

  template <class color_t>
  class image {

  public:
    bool delete_buffer_on_destruct;
    color_t *buffer;
    unsigned width;
    unsigned height;
    unsigned pitch;//in sizeof(color_t)

    image()
    : delete_buffer_on_destruct(false), buffer(0), width(0), height(0),
      pitch(0) {}

    image(unsigned width, unsigned height)
    : delete_buffer_on_destruct(true), buffer(new color_t[width * height]),
      width(width), height(height), pitch(width) {}

    image(
      color_t *buffer, unsigned width, unsigned height, unsigned pitch,
      bool delete_buffer_on_destruct)
    : delete_buffer_on_destruct(delete_buffer_on_destruct), buffer(buffer),
      width(width), height(height), pitch(pitch) {}

    ~image() {
      if (delete_buffer_on_destruct && buffer)
        delete[] buffer;
    }

    image(const image<color_t> &other) = delete;

    image(image<color_t> &&other)
    : delete_buffer_on_destruct(other.delete_buffer_on_destruct),
      buffer(other.buffer), width(other.width), height(other.height),
      pitch(other.pitch) {
      other.buffer = 0;
    }

    image<color_t> &operator =(const image<color_t> &other) = delete;

    image<color_t> &operator =(image<color_t> &&other) {
      if (delete_buffer_on_destruct && buffer)
        delete[] buffer;
      delete_buffer_on_destruct = other.delete_buffer_on_destruct;
      buffer = other.buffer;
      width = other.width;
      height = other.height;
      pitch = other.pitch;
      other.buffer = 0;
      return *this;
    }

    color_t get(unsigned x, unsigned y) const {
      return buffer[y * pitch + x];
    }

    void set(unsigned x, unsigned y, const color_t &c) {
      buffer[y * pitch + x] = c;
    }

  };

  //it is assumed that the regions do not overlap in memory.
  //copies into [to_x, to_x + width) x [to_y, to_y + height)
  //from [from_x, from_x + width) x [from_y, from_y + height).
  template <class color_t>
  void copy_region(
    image<color_t> &to, unsigned to_x, unsigned to_y,
    const image<color_t> &from, unsigned from_x, unsigned from_y,
    unsigned width, unsigned height) {

    color_t *to_start = to.buffer + to.pitch * to_y + to_x;
    const color_t *from_start = from.buffer + from.pitch * from_y + from_x;

    for (unsigned y = 0; y < height; ++y)
      std::memcpy(
        to_start + to.pitch * y, from_start + from.pitch * y,
        width * sizeof(color_t));

  }

  //it is assumed that the regions do not overlap in memory.
  //copies into [to_x, to_x + width) x [to_y, to_y + height)
  //from [from_x, from_x + width) x [from_y, from_y + height).
  template <
    class to_color_t, class from_color_t,
    void overlay(to_color_t &dest, const from_color_t &src) = default_overlay>
  void overlay_region(
    image<to_color_t> &to, unsigned to_x, unsigned to_y,
    const image<from_color_t> &from, unsigned from_x, unsigned from_y,
    unsigned width, unsigned height) {

    to_color_t *to_start = to.buffer + to.pitch * to_y + to_x;
    const from_color_t *from_start =
      from.buffer + from.pitch * from_y + from_x;

    for (unsigned y = 0; y < height; ++y)
      for (unsigned x = 0; x < width; ++x)
        overlay(to_start[to.pitch * y + x], from_start[from.pitch * y + x]);

  }

  image<hilbert_color> get_hilbert_framebuffer();

  bool try_load_ppm(std::FILE *input, image<rgb24> &into);

  static inline bool try_load_ppm(const char *path, image<rgb24> &into) {
    std::FILE *f = std::fopen(path, "r");
    if (!f)
      return false;
    bool success = try_load_ppm(f, into);
    std::fclose(f);
    return success;
  }

  //TODO: unicode
  template <class color_t>
  class fixed_bitmap_font {

  public:
    unsigned width;
    unsigned height;
    image<color_t> glyphs[128];

    template <class target_color_t,
      void overlay(target_color_t &dest, const color_t &src) = default_overlay>
    void overlay_text(
      image<target_color_t> &target, unsigned x,
      unsigned y, const char *text) const {

      while (1) {
        uint8_t ch = (uint8_t)*text;
        if (ch == 0)
          return;
        if (ch < 128) {
          overlay_region<target_color_t, color_t, overlay>(
            target, x, y, glyphs[ch], 0, 0, width, height);
          x += width;
        }
        ++text;
      }

    }

  };

  bool try_load_psf(std::FILE *input, fixed_bitmap_font<bool> &into);

  static inline bool try_load_psf(
    const char *path, fixed_bitmap_font<bool> &into) {
    std::FILE *f = std::fopen(path, "r");
    if (!f)
      return false;
    bool success = try_load_psf(f, into);
    std::fclose(f);
    return success;
  }

}