Class ImageReader

src/include/image/image_reader.hpp:35–68

};

class ImageReader {
public:
	explicit ImageReader(size_t max_cached_per_size = 16);
	~ImageReader();

	bool load_image(const std::string& filename, image_data_t& out_image);
	bool load_image_base64(const std::string& base64_string, image_data_t& out_image);
	bool resize_image(const image_data_t& input, int target_width, int target_height, image_data_t& output);
	bool reorder_hwc_to_chw(const image_data_t& input, image_data_t& output);
	bool save_png(const std::string& filename, const image_data_t& image);

	void recycle(image_data_t& image);

private:
	static void initialize_ffmpeg();
	static bool parse_image_header(const uint8_t* data, size_t size, int& codec_id);
	bool decode_bytes(const uint8_t* data, size_t size, image_data_t& out_image);
	bool ensure_decode_resources(int codec_id);
	void reset_decode_resources();

	ImageMemoryPool memory_pool_;

	struct AVCodecContext* codec_ctx_ = nullptr;
	struct AVPacket* packet_ = nullptr;
	struct AVFrame* frame_ = nullptr;
	struct AVFrame* rgb_frame_ = nullptr;
	struct SwsContext* sws_ctx_ = nullptr;

	int cached_codec_id_ = -1;
	int sws_src_w_ = 0;
	int sws_src_h_ = 0;
	int sws_src_fmt_ = -1;
	int sws_src_range_ = -1;
};

bool save_image(const std::string& filename, const bytes& image);