#pragma once

#include "av.hpp"

#include <memory>
#include <string>
#include <vector>
#include <functional>

// Basic image representation
// 3 bytes per pixel (rgb), so size of data is width*height*3
struct raw_rgb_image {
	std::vector<uint8_t, av::allocator<uint8_t>> data;
	av::frame frame;

	raw_rgb_image();
	raw_rgb_image(int W, int H);

	int width() const;
	int height() const;
	AVPixelFormat format() const;
};

// dumps image in ppm format
void save_as_ppm(raw_rgb_image const & image, std::string const & filename);

// opens an image in its own format
av::frame open_image(std::string const & filename);

// crops to the bottom right square (cheap operation)
void crop_to_square(av::frame & frame);
av::frame crop_to_square(av::frame const & frame);
av::frame crop_to_square(av::frame && frame);

// converts and resizes
raw_rgb_image to_raw_rgb_image(av::frame const & frame, int new_width, int new_height);

// Legacy
inline raw_rgb_image open_as_rgb(const std::string &filename){
	return to_raw_rgb_image(crop_to_square(open_image(filename)), 512, 512);
}

// apply function to every tile, fun :: Column, Row, Image -> Void
void apply_to_tiles(std::string const & filename, int h_tiles, int v_tiles, std::function<void(int, int, av::frame const &)> fun);

// does what you think it does
void save_as_jpg(av::frame const & frame, std::string const & filename);