VulkanPipelineWarping.cpp

/* ----------------------
* Copyright 2023 Université Libre de Bruxelles(ULB), Universidad Politécnica de Madrid(UPM), CREAL, Deutsches Zentrum für Luft - und Raumfahrt(DLR)
 
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at < http://www.apache.org/licenses/LICENSE-2.0>
 
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissionsand
* limitations under the License.
---------------------- */
 
#include "VulkanPipelineWarping.h"
#include"Shader.h"
#include"VulkanContext.h"
#include"VulkanHelperFunction.h"
#include"VulkanRenderPass.h"
#include"VulkanWrapper.h"
 
 
 
VulkanPipelineWarping::VulkanPipelineWarping(VulkanContext* context, VulkanRenderPass* renderPass, InputProvider * inputP, VulkanWrapper* wraps, int inputCamId)
{
    this->wrapper = wraps;
    this->context = context;
    this->renderPass = renderPass;
    this->colorAttachmentNb = 3;
    this->inputId = inputCamId;
    this->subPass = 2 * inputCamId + 0;
 
    bufferController = std::make_unique<BuffersControllerWarping>(context, renderPass, this, inputP, wraps, inputCamId);
 
    
    auto activation = wrapper->getCameraActivation();
    for (int i = 0; i <= inputCamId; i++) {
        if (!activation[i]) {
            inputCamId++;
            this->inputId++;
        }
    }
 
    
 
    auto streamParam = inputP->enumerateStreamsParameters()[inputCamId];
    colorSampler = streamParam.colorSampler;
    depthSampler = streamParam.depthSampler;
 
    
}
void VulkanPipelineWarping::init()
{
    //createRenderPass();
    createDescriptorSetLayout();
    createGraphicsPipeline();
 
    //for the buffer Controller
    bufferController->init();
    
 
}
 
void VulkanPipelineWarping::cleanUp()
{
    context->device.destroy(descriptorSetLayout);
    bufferController->cleanUp();
    context->device.destroyPipeline(pipeline);
    context->device.destroyPipelineLayout(pipelineLayout);
    
}
 
 
 
void VulkanPipelineWarping::createDescriptorSetLayout()
{
    vk::DescriptorSetLayoutBinding depthSamplerLayoutBinding(
        1,                          //binding
        vk::DescriptorType::eCombinedImageSampler, //descriptor type
        1,                              //descriptorCount
        vk::ShaderStageFlagBits::eVertex | vk::ShaderStageFlagBits::eFragment,  //stageFlags
        &depthSampler               //pimmutable sampler
    );
    vk::DescriptorSetLayoutBinding uboVertCamLayoutBinding(
        2,                          //binding
        vk::DescriptorType::eUniformBuffer, //descriptor type
        1,                              //descriptorCount
        vk::ShaderStageFlagBits::eVertex,   //stageFlags
        nullptr             //pimmutable sampler
    );
    vk::DescriptorSetLayoutBinding uboVertLayoutBinding(
        3,                          //binding
        vk::DescriptorType::eUniformBuffer, //descriptor type
        1,                              //descriptorCount
        vk::ShaderStageFlagBits::eVertex,   //stageFlags
        nullptr             //pimmutable sampler
    );
    vk::DescriptorSetLayoutBinding uboGeomLayoutBinding(
        4,                          //binding
        vk::DescriptorType::eUniformBuffer, //descriptor type
        1,                              //descriptorCount
        vk::ShaderStageFlagBits::eGeometry, //stageFlags
        nullptr             //pimmutable sampler
    );
    vk::DescriptorSetLayoutBinding uboFragLayoutBinding(
        5,                          //binding
        vk::DescriptorType::eUniformBuffer, //descriptor type
        1,                              //descriptorCount
        vk::ShaderStageFlagBits::eFragment, //stageFlags
        nullptr             //pimmutable sampler
    );
    vk::DescriptorSetLayoutBinding samplerLayoutBinding(
        6,
        vk::DescriptorType::eCombinedImageSampler,
        1,
        vk::ShaderStageFlagBits::eFragment,
        &colorSampler
    );
 
    std::array<vk::DescriptorSetLayoutBinding, 6> bindings = { depthSamplerLayoutBinding, uboVertCamLayoutBinding,uboVertLayoutBinding, uboGeomLayoutBinding, uboFragLayoutBinding, samplerLayoutBinding };
 
    vk::DescriptorSetLayoutCreateInfo layoutInfo(vk::DescriptorSetLayoutCreateFlags(), static_cast<uint32_t>(bindings.size()), bindings.data());
 
    descriptorSetLayout =  context->device.createDescriptorSetLayout(layoutInfo);
 
}
 
void VulkanPipelineWarping::update(std::span<InputProvider::StreamFrameInfo> infos, int view)
{
    bufferController->update(infos, view);
}
 
 
 
 
void VulkanPipelineWarping::createGraphicsPipeline()
{
    //ShadersList shaders = ShadersList::getInstance();
    //shaders.loadShaders();
    /*
    vk::ShaderModule vertShaderModule = createShaderModule(shaders("shader.vert"));
    vk::ShaderModule fragShaderModule = createShaderModule(shaders("shader.frag"));
 
    std::array<vk::PipelineShaderStageCreateInfo, 2> pipelineShaderStageCreateInfos = {
      vk::PipelineShaderStageCreateInfo(
        vk::PipelineShaderStageCreateFlags(), vk::ShaderStageFlagBits::eVertex, vertShaderModule, "main"), //main is entry point of the shader
      vk::PipelineShaderStageCreateInfo(
        vk::PipelineShaderStageCreateFlags(), vk::ShaderStageFlagBits::eFragment, fragShaderModule, "main")
    };
    */
    vk::ShaderModule vertShaderModule = createShaderModule(ShadersList::getInstance()("synthesis.vert"));
    vk::ShaderModule geomShaderModule = createShaderModule(ShadersList::getInstance()("synthesis.geom"));
    vk::ShaderModule fragShaderModule = createShaderModule(ShadersList::getInstance()("synthesis.frag"));
 
    std::vector<vk::PipelineShaderStageCreateInfo> pipelineShaderStageCreateInfos = {
      vk::PipelineShaderStageCreateInfo(
        vk::PipelineShaderStageCreateFlags(), vk::ShaderStageFlagBits::eVertex, vertShaderModule, "main"), //main is entry point of the shader
      vk::PipelineShaderStageCreateInfo(
        vk::PipelineShaderStageCreateFlags(), vk::ShaderStageFlagBits::eGeometry, geomShaderModule, "main"),
        vk::PipelineShaderStageCreateInfo(
        vk::PipelineShaderStageCreateFlags(), vk::ShaderStageFlagBits::eFragment, fragShaderModule, "main")
    };
 
 
    this->createPipeline(pipelineShaderStageCreateInfos);
 
    context->device.destroyShaderModule(vertShaderModule);
    context->device.destroyShaderModule(geomShaderModule);
    context->device.destroyShaderModule(fragShaderModule);
}