screenshot.h

#define STB_IMAGE_WRITE_IMPLEMENTATION
#include "stb_image_write.h"

#include <kope/util/align.h>

static kope_g5_buffer screenshot_buffer;

static void screenshot_take(kope_g5_device *device, kope_g5_command_list *list, kope_g5_texture *framebuffer, int width, int height) {
	uint32_t row_bytes = kope_g5_device_align_texture_row_bytes(device, width * 4);

	kope_g5_buffer_parameters buffer_parameters;
	buffer_parameters.size = row_bytes * height;
	buffer_parameters.usage_flags = KOPE_G5_BUFFER_USAGE_CPU_READ;
	kope_g5_device_create_buffer(device, &buffer_parameters, &screenshot_buffer);

	kope_g5_image_copy_texture source = {0};
	source.texture = framebuffer;
	source.mip_level = 0;

	kope_g5_image_copy_buffer destination = {0};
	destination.buffer = &screenshot_buffer;
	destination.bytes_per_row = row_bytes;

	kope_g5_command_list_copy_texture_to_buffer(list, &source, &destination, width, height, 1);

	kope_g5_device_execute_command_list(device, list);

	kope_g5_device_wait_until_idle(device);

	uint8_t *pixels = (uint8_t *)kope_g5_buffer_lock_all(&screenshot_buffer);

	stbi_write_png("test.png", width, height, 4, pixels, row_bytes);

	kope_g5_buffer_unlock(&screenshot_buffer);

	exit(0);
}