VulkanSimulation/include/vulkan/buffer.hpp

#pragma once

#include <vulkan/vulkan_core.h>
#include <stdexcept>
#include "vertex.hpp"
#include "vk_mem_alloc.h"
#include <memory>
#include <optional>

using std::unique_ptr;
using std::shared_ptr;
using std::make_shared;
using std::make_unique;
using std::optional;

class Image;

class Buffer {
public:
	Buffer(VkDeviceSize bufferSize, VkBufferUsageFlags bufferUsage,
					VmaMemoryUsage memoryUsage, VmaAllocationCreateFlags vmaAllocationFlags);
	unique_ptr<Buffer> appended(void* data, VkDeviceSize appendSize, VkCommandBuffer commandBuffer) const;
	unique_ptr<Buffer> replaced(void* data, VkDeviceSize replaceSize, VkCommandBuffer commandBuffer) const;
	virtual ~Buffer();
	VkBuffer handle = VK_NULL_HANDLE;
	VmaAllocation allocation = VK_NULL_HANDLE;
	VmaAllocationInfo allocationInfo {};
	VkDeviceSize size;
	void setName(const std::string& newName);
	template <typename T>
	T& access(){
		return *reinterpret_cast<T*>(allocationInfo.pMappedData);
	}
	void copyTo(Buffer* buffer) const;
	void copyTo(Image* image) const;
	void setData(void* data, VkDeviceSize offset, VkDeviceSize dataSize, VkCommandBuffer commandBuffer=VK_NULL_HANDLE);
	shared_ptr<Buffer> getStagingBuffer();
	Buffer(const Buffer& other) = delete;
	Buffer& operator =(const Buffer& other) = delete;
private:
	optional<shared_ptr<Buffer>> stagingBufferOptional;
	std::string name;
	VkBufferUsageFlags bufferUsageFlags;
	VmaMemoryUsage memoryUsage;
	VmaAllocationCreateFlags allocationCreateFlags;
};