// Copyright (C) 2022 Jérôme "Lynix" Leclercq (lynix680@gmail.com)
// This file is part of the "Nazara Engine - Utility module"
// For conditions of distribution and use, see copyright notice in Config.hpp

#include <Nazara/Utility/UniformBuffer.hpp>
#include <Nazara/Core/Error.hpp>
#include <Nazara/Core/ErrorFlags.hpp>
#include <Nazara/Utility/Config.hpp>
#include <Nazara/Utility/Debug.hpp>

namespace Nz
{
	UniformBuffer::UniformBuffer(std::shared_ptr<Buffer> buffer)
	{
		ErrorFlags(ErrorMode::ThrowException, true);
		Reset(std::move(buffer));
	}

	UniformBuffer::UniformBuffer(std::shared_ptr<Buffer> buffer, UInt32 offset, UInt32 size)
	{
		ErrorFlags(ErrorMode::ThrowException, true);
		Reset(std::move(buffer), offset, size);
	}

	UniformBuffer::UniformBuffer(UInt32 length, DataStorage storage, BufferUsageFlags usage)
	{
		ErrorFlags(ErrorMode::ThrowException, true);
		Reset(length, storage, usage);
	}

	bool UniformBuffer::Fill(const void* data, UInt32 offset, UInt32 size)
	{
		NazaraAssert(m_buffer && m_buffer->IsValid(), "Invalid buffer");
		NazaraAssert(m_startOffset + offset + size <= m_endOffset, "Exceeding virtual buffer size");

		return m_buffer->Fill(data, m_startOffset + offset, size);
	}

	void* UniformBuffer::Map(BufferAccess access, UInt32 offset, UInt32 size)
	{
		NazaraAssert(m_buffer && m_buffer->IsValid(), "Invalid buffer");
		NazaraAssert(m_startOffset + offset + size <= m_endOffset, "Exceeding virtual buffer size");

		return m_buffer->Map(access, offset, size);
	}

	void* UniformBuffer::Map(BufferAccess access, UInt32 offset, UInt32 size) const
	{
		NazaraAssert(m_buffer && m_buffer->IsValid(), "Invalid buffer");
		NazaraAssert(m_startOffset + offset + size <= m_endOffset, "Exceeding virtual buffer size");

		return m_buffer->Map(access, offset, size);
	}

	void UniformBuffer::Reset()
	{
		m_buffer.reset();
	}

	void UniformBuffer::Reset(std::shared_ptr<Buffer> buffer)
	{
		NazaraAssert(buffer && buffer->IsValid(), "Invalid buffer");

		Reset(buffer, 0, buffer->GetSize());
	}

	void UniformBuffer::Reset(std::shared_ptr<Buffer> buffer, UInt32 offset, UInt32 size)
	{
		NazaraAssert(buffer && buffer->IsValid(), "Invalid buffer");
		NazaraAssert(buffer->GetType() == BufferType::Uniform, "Buffer must be an uniform buffer");
		NazaraAssert(size > 0, "Invalid size");
		NazaraAssert(offset + size > buffer->GetSize(), "Virtual buffer exceed buffer bounds");

		m_buffer = buffer;
		m_endOffset = offset + size;
		m_startOffset = offset;
	}

	void UniformBuffer::Reset(UInt32 size, DataStorage storage, BufferUsageFlags usage)
	{
		m_endOffset = size;
		m_startOffset = 0;

		m_buffer = std::make_shared<Buffer>(BufferType::Uniform, m_endOffset, storage, usage);
	}

	void UniformBuffer::Reset(const UniformBuffer& uniformBuffer)
	{
		m_buffer = uniformBuffer.m_buffer;
		m_endOffset = uniformBuffer.m_endOffset;
		m_startOffset = uniformBuffer.m_startOffset;
	}

	void UniformBuffer::Unmap() const
	{
		m_buffer->Unmap();
	}
}