image.hpp

Go to the documentation of this file.

#pragma once
 
#include <optional>
#include <vector>
 
#include "fmt/format.h"
 
#include "porytiles2/domain/models/rgba32.hpp"
#include "porytiles2/xcut/panic/panic.hpp"
 
namespace porytiles2 {
 
template <typename PixelType>
class Image {
  public:
    Image() : Image(0, 0) {}
 
    explicit Image(std::size_t width, std::size_t height) : pixels_(width * height), width_{width}, height_{height} {}
 
    Image(std::size_t width, std::size_t height, std::vector<Rgba32> palette)
        : pixels_(width * height), palette_(std::move(palette)), width_{width}, height_{height}
    {
    }
 
    [[nodiscard]] PixelType at(std::size_t i) const
    {
        if (const auto s = size(); i >= s) {
            panic(fmt::format("index {} out of bounds for image size {}", i, s));
        }
        return pixels_[i];
    }
 
    [[nodiscard]] PixelType at(std::size_t row, std::size_t col) const
    {
        if (col >= width_) {
            panic(fmt::format("col {} out of bounds for image width {}", col, width_));
        }
        if (row >= height_) {
            panic(fmt::format("row {} out of bounds for image height {}", row, height_));
        }
        return pixels_[row * width_ + col];
    }
 
    void set(std::size_t i, PixelType pixel)
    {
        if (const auto s = size(); i >= s) {
            panic(fmt::format("index {} out of bounds for image size {}", i, s));
        }
        pixels_[i] = pixel;
    }
 
    void set(std::size_t row, std::size_t col, PixelType pixel)
    {
        if (col >= width_) {
            panic(fmt::format("col {} out of bounds for image width {}", col, width_));
        }
        if (row >= height_) {
            panic(fmt::format("row {} out of bounds for image height {}", row, height_));
        }
        pixels_[row * width_ + col] = pixel;
    }
 
    [[nodiscard]] std::size_t width() const
    {
        return width_;
    }
 
    [[nodiscard]] std::size_t height() const
    {
        return height_;
    }
 
    [[nodiscard]] std::size_t size() const
    {
        return pixels_.size();
    }
 
    [[nodiscard]] const std::optional<std::vector<Rgba32>> &palette() const
    {
        return palette_;
    }
 
  private:
    std::vector<PixelType> pixels_;
    // TODO: is there a better way to handle this?
    std::optional<std::vector<Rgba32>> palette_;
    std::size_t width_;
    std::size_t height_;
};
 
} // namespace porytiles2