Image.h

// Copyright 2022 ShenMian
// License(Apache-2.0)
 
#pragma once
 
#include <cstdint>
#include <filesystem>
#include <math/math.hpp>
#include <vector>
 
class Image
{
public:
    Image() = default;
 
    Image(const std::filesystem::path& path);
 
    Image(const void* data, size_t sizeBytes, const Vector2i& size, int channels);
 
    void load(const std::filesystem::path& path);
 
    void load(const void* data, size_t sizeBytes, const Vector2i& size, int channels);
 
    void save(const std::filesystem::path& path) const;
 
    void set_pixel(const Vector4f& color, const Vector2i& pos);
 
    Vector4f get_pixel(const Vector2i& pos) const;
 
    void flip_horizontally() noexcept;
 
    void flip_vertically() noexcept;
 
    uint8_t* data() noexcept;
 
    const uint8_t* data() const noexcept;
 
    size_t size_bytes() const noexcept;
 
    Vector2i size() const noexcept;
 
    int channel_count() const noexcept;
 
private:
    std::vector<uint8_t> data_;
    Vector2i             size_;
    int                  channels_ = 0;
};