Browse Source

vulkan: make api design more senseful

niki 3 years ago
parent
commit
c51b12a177

+ 8 - 8
src/modules/graphics/vulkan/Buffer.cpp

@@ -88,14 +88,14 @@ void Buffer::unmap(size_t usedoffset, size_t usedsize) {
 void Buffer::copyTo(love::graphics::Buffer* dest, size_t sourceoffset, size_t destoffset, size_t size) {
 	Graphics* vgfx = (Graphics*)gfx;
 
-	vgfx->queueDatatransfer([buffer = buffer, dest = dest, sourceoffset, destoffset, size](VkCommandBuffer commandBuffer){
-		VkBufferCopy bufferCopy{};
-		bufferCopy.srcOffset = sourceoffset;
-		bufferCopy.dstOffset = destoffset;
-		bufferCopy.size = size;
-
-		vkCmdCopyBuffer(commandBuffer, buffer, (VkBuffer) dest->getHandle(), 1, &bufferCopy);
-	}, nullptr);
+	auto commandBuffer = vgfx->getDataTransferCommandBuffer();
+
+	VkBufferCopy bufferCopy{};
+	bufferCopy.srcOffset = sourceoffset;
+	bufferCopy.dstOffset = destoffset;
+	bufferCopy.size = size;
+
+	vkCmdCopyBuffer(commandBuffer, buffer, (VkBuffer) dest->getHandle(), 1, &bufferCopy);
 }
 
 } // vulkan

+ 2 - 5
src/modules/graphics/vulkan/Graphics.cpp

@@ -555,11 +555,8 @@ const PFN_vkCmdPushDescriptorSetKHR Graphics::getVkCmdPushDescriptorSetKHRFuncti
 	return vkCmdPushDescriptorSet;
 }
 
-void Graphics::queueDatatransfer(std::function<void(VkCommandBuffer)> command, std::function<void()> cleanUp) {
-	command(dataTransferCommandBuffers.at(currentFrame));
-	if (cleanUp) {
-		cleanUpFunctions.at(currentFrame).push_back(std::move(cleanUp));
-	}
+VkCommandBuffer Graphics::getDataTransferCommandBuffer() {
+	return dataTransferCommandBuffers.at(currentFrame);
 }
 
 void Graphics::queueCleanUp(std::function<void()> cleanUp) {

+ 1 - 1
src/modules/graphics/vulkan/Graphics.h

@@ -135,7 +135,7 @@ public:
 	GraphicsReadback* newReadbackInternal(ReadbackMethod method, love::graphics::Buffer* buffer, size_t offset, size_t size, data::ByteData* dest, size_t destoffset) override { return nullptr;  };
 	GraphicsReadback* newReadbackInternal(ReadbackMethod method, love::graphics::Texture* texture, int slice, int mipmap, const Rect& rect, image::ImageData* dest, int destx, int desty) { return nullptr; }
 
-	void queueDatatransfer(std::function<void(VkCommandBuffer)> command, std::function<void()> cleanUp);
+	VkCommandBuffer getDataTransferCommandBuffer();
 	void queueCleanUp(std::function<void()> cleanUp);
 
 	VkCommandBuffer beginSingleTimeCommands();

+ 77 - 78
src/modules/graphics/vulkan/Texture.cpp

@@ -56,10 +56,11 @@ bool Texture::loadVolatile() {
 	if (vmaCreateImage(allocator, &imageInfo, &imageAllocationCreateInfo, &textureImage, &textureImageAllocation, nullptr) != VK_SUCCESS) {
 		throw love::Exception("failed to create image");
 	}
+
+	auto commandBuffer = vgfx->getDataTransferCommandBuffer();
+
 	// fixme: we should use VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL as the default image layout instead of VK_IMAGE_LAYOUT_GENERAL.
-	vgfx->queueDatatransfer([textureImage = textureImage, layerCount = layerCount](VkCommandBuffer commandBuffer){
-		Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_UNDEFINED, VK_IMAGE_LAYOUT_GENERAL, 0, 1, 0, layerCount);
-	}, nullptr);
+	Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_UNDEFINED, VK_IMAGE_LAYOUT_GENERAL, 0, 1, 0, layerCount);
 
 	if (data) {
 		for (int slice = 0; slice < layerCount; slice++) {
@@ -221,90 +222,88 @@ void Texture::uploadByteData(PixelFormat pixelformat, const void* data, size_t s
 
 	memcpy(allocInfo.pMappedData, data, size);
 
-	auto command = [buffer = stagingBuffer, image = textureImage, r = r, level = level, slice = slice](VkCommandBuffer commandBuffer) {
-		VkBufferImageCopy region{};
-		region.bufferOffset = 0;
-		region.bufferRowLength = 0;
-		region.bufferImageHeight = 0;
-
-		region.imageSubresource.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT;
-		region.imageSubresource.mipLevel = level;
-		region.imageSubresource.baseArrayLayer = slice;
-		region.imageSubresource.layerCount = 1;
-
-		region.imageOffset = { r.x, r.y, 0 };
-		region.imageExtent = {
-			static_cast<uint32_t>(r.w),
-			static_cast<uint32_t>(r.h), 1
-		};
-
-		Vulkan::cmdTransitionImageLayout(commandBuffer, image, VK_IMAGE_LAYOUT_GENERAL, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, level, 1, slice, 1);
-
-		vkCmdCopyBufferToImage(
-			commandBuffer,
-			buffer,
-			image,
-			VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
-			1,
-			&region
-		);
-
-		Vulkan::cmdTransitionImageLayout(commandBuffer, image, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, VK_IMAGE_LAYOUT_GENERAL, level, 1, slice, 1);
-	};
+	VkBufferImageCopy region{};
+	region.bufferOffset = 0;
+	region.bufferRowLength = 0;
+	region.bufferImageHeight = 0;
 
-	auto cleanUp = [allocator = allocator, stagingBuffer, vmaAllocation]() {
-		vmaDestroyBuffer(allocator, stagingBuffer, vmaAllocation);
+	region.imageSubresource.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT;
+	region.imageSubresource.mipLevel = level;
+	region.imageSubresource.baseArrayLayer = slice;
+	region.imageSubresource.layerCount = 1;
+
+	region.imageOffset = { r.x, r.y, 0 };
+	region.imageExtent = {
+		static_cast<uint32_t>(r.w),
+		static_cast<uint32_t>(r.h), 1
 	};
 
-	vgfx->queueDatatransfer(command, cleanUp);
+	auto commandBuffer = vgfx->getDataTransferCommandBuffer();
+
+	Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_GENERAL, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, level, 1, slice, 1);
+
+	vkCmdCopyBufferToImage(
+		commandBuffer,
+		stagingBuffer,
+		textureImage,
+		VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
+		1,
+		&region
+	);
+
+	Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, VK_IMAGE_LAYOUT_GENERAL, level, 1, slice, 1);
+
+	vgfx->queueCleanUp([allocator = allocator, stagingBuffer, vmaAllocation]() {
+		vmaDestroyBuffer(allocator, stagingBuffer, vmaAllocation);
+	});
 }
 
 void Texture::copyFromBuffer(graphics::Buffer* source, size_t sourceoffset, int sourcewidth, size_t size, int slice, int mipmap, const Rect& rect) {
-	vgfx->queueDatatransfer([source, textureImage = textureImage, rect=rect, sourceoffset, sourcewidth, size, slice, mipmap](VkCommandBuffer commandBuffer){
-		VkImageSubresourceLayers layers{};
-		layers.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT;
-		layers.mipLevel = mipmap;
-		layers.baseArrayLayer = slice;
-		layers.layerCount = 1;
-
-		VkBufferImageCopy region{};
-		region.bufferOffset = sourceoffset;
-		region.bufferRowLength = sourcewidth;
-		region.bufferImageHeight = 1;
-		region.imageSubresource = layers;
-		region.imageExtent.width = static_cast<uint32_t>(rect.w);
-		region.imageExtent.height = static_cast<uint32_t>(rect.h);
-
-		Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_GENERAL, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL);
-
-		vkCmdCopyBufferToImage(commandBuffer, (VkBuffer)source->getHandle(), textureImage, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, 1, &region);
-
-		Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, VK_IMAGE_LAYOUT_GENERAL);
-	}, nullptr);
+	auto commandBuffer = vgfx->getDataTransferCommandBuffer();
+
+	VkImageSubresourceLayers layers{};
+	layers.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT;
+	layers.mipLevel = mipmap;
+	layers.baseArrayLayer = slice;
+	layers.layerCount = 1;
+
+	VkBufferImageCopy region{};
+	region.bufferOffset = sourceoffset;
+	region.bufferRowLength = sourcewidth;
+	region.bufferImageHeight = 1;
+	region.imageSubresource = layers;
+	region.imageExtent.width = static_cast<uint32_t>(rect.w);
+	region.imageExtent.height = static_cast<uint32_t>(rect.h);
+
+	Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_GENERAL, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL);
+
+	vkCmdCopyBufferToImage(commandBuffer, (VkBuffer)source->getHandle(), textureImage, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, 1, &region);
+
+	Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, VK_IMAGE_LAYOUT_GENERAL);
 }
 
 void Texture::copyToBuffer(graphics::Buffer* dest, int slice, int mipmap, const Rect& rect, size_t destoffset, int destwidth, size_t size) {
-	vgfx->queueDatatransfer([dest, textureImage = textureImage, rect=rect, destoffset, destwidth, size, slice, mipmap](VkCommandBuffer commandBuffer){
-		VkImageSubresourceLayers layers{};
-		layers.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT;
-		layers.mipLevel = mipmap;
-		layers.baseArrayLayer = slice;
-		layers.layerCount = 1;
-
-		VkBufferImageCopy region{};
-		region.bufferOffset = destoffset;
-		region.bufferRowLength = destwidth;
-		region.bufferImageHeight = 1;
-		region.imageSubresource = layers;
-		region.imageExtent.width = static_cast<uint32_t>(rect.w);
-		region.imageExtent.height = static_cast<uint32_t>(rect.h);
-
-		Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_GENERAL, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL);
-
-		vkCmdCopyImageToBuffer(commandBuffer, textureImage, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL, (VkBuffer) dest->getHandle(), 1, &region);
-
-		Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL, VK_IMAGE_LAYOUT_GENERAL);
-	}, nullptr);
+	auto commandBuffer = vgfx->getDataTransferCommandBuffer();
+
+	VkImageSubresourceLayers layers{};
+	layers.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT;
+	layers.mipLevel = mipmap;
+	layers.baseArrayLayer = slice;
+	layers.layerCount = 1;
+
+	VkBufferImageCopy region{};
+	region.bufferOffset = destoffset;
+	region.bufferRowLength = destwidth;
+	region.bufferImageHeight = 1;
+	region.imageSubresource = layers;
+	region.imageExtent.width = static_cast<uint32_t>(rect.w);
+	region.imageExtent.height = static_cast<uint32_t>(rect.h);
+
+	Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_GENERAL, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL);
+
+	vkCmdCopyImageToBuffer(commandBuffer, textureImage, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL, (VkBuffer) dest->getHandle(), 1, &region);
+
+	Vulkan::cmdTransitionImageLayout(commandBuffer, textureImage, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL, VK_IMAGE_LAYOUT_GENERAL);
 }
 
 } // vulkan