/*

* Copyright (C) 2020 by Xu Xing (xu.xing@outlook.com)

*

* This code is licensed under the MIT license (MIT)

* (http://opensource.org/licenses/MIT)

*/

#include "ComputeCopyImageOp.h"

ComputeCopyImageOp::ComputeCopyImageOp() {}

ComputeCopyImageOp::~ComputeCopyImageOp() {}

ComputeCopyImageOp::ComputeCopyImageOp(const InitParams &init_params)

: ComputeOp(init_params) {}

void ComputeCopyImageOp::execute() {

// Prepare storage buffers.

const VkDeviceSize bufferSize =

(params_.inputWidth * params_.inputHeight) * sizeof(uint32_t);

const VkDeviceSize filterBufferSize =

(params_.filterWidth * params_.filterHeight) * sizeof(uint32_t);

const VkDeviceSize outputBufferSize =

(params_.outputWidth * params_.outputHeight) * sizeof(uint32_t);

// Copy input data to VRAM using a staging buffer.

{

createBufferWithData(VK_BUFFER_USAGE_TRANSFER_SRC_BIT |

VK_BUFFER_USAGE_TRANSFER_DST_BIT,

VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT, &hostBuffer_,

&hostMemory_, bufferSize, params_.computeInput.data());

createDeviceImage(image_, params_.inputWidth, params_.inputHeight);

createSampler(image_, sampler_, view_);

copyHostBufferToDeviceImage(image_, hostBuffer_, params_.inputWidth,

params_.inputHeight);

VkDeviceMemory testMemory;

VkBuffer testBuffer;

createBufferWithData(VK_BUFFER_USAGE_TRANSFER_SRC_BIT |

VK_BUFFER_USAGE_TRANSFER_DST_BIT,

VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT, &testBuffer,

&testMemory, bufferSize);

copyDeviceImageToHostBuffer(image_, params_.computeOutput.data(),

bufferSize, params_.inputWidth,

params_.inputHeight);

}

vkQueueWaitIdle(queue_);

// TODO: exit clean up.

}