#pragma once

#include <array>
#include <vector>

#include "framebuffer.h"
#include "pixelformats.h"

namespace kms
{

class DmabufFramebuffer : public Framebuffer
{
public:
	DmabufFramebuffer(Card& card, uint32_t width, uint32_t height, const std::string& format,
			  std::vector<int> fds, std::vector<uint32_t> pitches, std::vector<uint32_t> offsets, std::vector<uint64_t> modifiers = {});
	DmabufFramebuffer(Card& card, uint32_t width, uint32_t height, PixelFormat format,
			  std::vector<int> fds, std::vector<uint32_t> pitches, std::vector<uint32_t> offsets, std::vector<uint64_t> modifiers = {});
	~DmabufFramebuffer() override;

	uint32_t width() const override { return Framebuffer::width(); }
	uint32_t height() const override { return Framebuffer::height(); }

	PixelFormat format() const override { return m_format; }
	unsigned num_planes() const override { return m_num_planes; }

	uint32_t handle(unsigned plane) const { return m_planes.at(plane).handle; }
	uint32_t stride(unsigned plane) const override { return m_planes.at(plane).stride; }
	uint32_t size(unsigned plane) const override { return m_planes.at(plane).size; }
	uint32_t offset(unsigned plane) const override { return m_planes.at(plane).offset; }
	uint8_t* map(unsigned plane) override;
	int prime_fd(unsigned plane) override;

	void begin_cpu_access(CpuAccess access) override;
	void end_cpu_access() override;

private:
	struct FramebufferPlane {
		uint32_t handle;
		int prime_fd;
		uint32_t size;
		uint32_t stride;
		uint32_t offset;
		uint64_t modifier;
		uint8_t *map;
	};

	unsigned m_num_planes;
	std::array<FramebufferPlane, 4> m_planes;

	PixelFormat m_format;

	uint32_t m_sync_flags = 0;
};

}