/*
 * Copyright 2022-2023 Frederico de Oliveira Linhares
 *
 * 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 permissions and
 * limitations under the License.
 */

#include "graphics_pipeline_2d_wired_layout.hpp"

#include <array>

#include "../core.hpp"
#include "graphics_pipeline_2d_solid_layout.hpp"
#include "uniform_data_object.hpp"

namespace
{

void
load_pipeline(void *obj)
{
  auto self = static_cast<VK::GraphicsPipeline2DWiredLayout*>(obj);

  std::array<VkDescriptorSetLayout, 1> set_layouts{
    cg_core.vk_descriptor_set_layout->view
  };

  std::array<VkPushConstantRange, 2> push_constants;
  push_constants[0].stageFlags = VK_SHADER_STAGE_VERTEX_BIT;
  push_constants[0].offset = 0;
  push_constants[0].size = sizeof(VK::UDOVector4D);

  push_constants[1].stageFlags = VK_SHADER_STAGE_FRAGMENT_BIT;
  push_constants[1].offset = sizeof(VK::UDOVector4D);
  push_constants[1].size = sizeof(VK::UDOVector3D);

  VkPipelineLayoutCreateInfo pipeline_layout_info{};
  pipeline_layout_info.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO;
  pipeline_layout_info.setLayoutCount = set_layouts.size();
  pipeline_layout_info.pSetLayouts = set_layouts.data();
  pipeline_layout_info.pushConstantRangeCount = push_constants.size();
  pipeline_layout_info.pPushConstantRanges = push_constants.data();

  if(vkCreatePipelineLayout(
       cg_core.vk_device_with_swapchain->device, &pipeline_layout_info,
       nullptr, &self->pipeline) != VK_SUCCESS)
    throw CommandError{"Failed to create Vulkan pipeline layout."};
}

void
unload_pipeline(void *obj)
{
  auto self = static_cast<VK::GraphicsPipeline2DWiredLayout*>(obj);

  vkDestroyPipelineLayout(
    cg_core.vk_device_with_swapchain->device, self->pipeline, nullptr);
}

void
load_render_pass(void *obj)
{
  auto self = static_cast<VK::GraphicsPipeline2DWiredLayout*>(obj);

  std::array<VkAttachmentDescription, 1> attachments{};
  // Color attachment.
  attachments[0].flags = 0;
  attachments[0].format = cg_core.vk_swapchain->image_format;
  attachments[0].samples = VK_SAMPLE_COUNT_1_BIT;
  attachments[0].loadOp = VK_ATTACHMENT_LOAD_OP_LOAD;
  attachments[0].storeOp = VK_ATTACHMENT_STORE_OP_STORE;
  attachments[0].stencilLoadOp = VK_ATTACHMENT_LOAD_OP_DONT_CARE;
  attachments[0].stencilStoreOp = VK_ATTACHMENT_STORE_OP_DONT_CARE;
  attachments[0].initialLayout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL;
  attachments[0].finalLayout = VK_IMAGE_LAYOUT_PRESENT_SRC_KHR;

  VkAttachmentReference color_attachment_ref{};
  color_attachment_ref.attachment = 0;
  color_attachment_ref.layout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL;

  VkSubpassDescription subpass{};
  subpass.flags = 0;
  subpass.pipelineBindPoint = VK_PIPELINE_BIND_POINT_GRAPHICS;
  subpass.inputAttachmentCount = 0;
  subpass.pInputAttachments = nullptr;
  subpass.colorAttachmentCount = 1;
  subpass.pColorAttachments = &color_attachment_ref;
  subpass.pResolveAttachments = nullptr;
  subpass.pDepthStencilAttachment = nullptr;
  subpass.preserveAttachmentCount = 0;
  subpass.pPreserveAttachments = nullptr;

  VkSubpassDependency dependency{};
  dependency.srcSubpass = VK_SUBPASS_EXTERNAL;
  dependency.dstSubpass = 0;
  dependency.srcStageMask = VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT |
    VK_PIPELINE_STAGE_EARLY_FRAGMENT_TESTS_BIT;
  dependency.srcAccessMask = 0;
  dependency.dstStageMask = VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT |
    VK_PIPELINE_STAGE_EARLY_FRAGMENT_TESTS_BIT;
  dependency.dstAccessMask = VK_ACCESS_COLOR_ATTACHMENT_READ_BIT |
    VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_WRITE_BIT |
    VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT;

  VkRenderPassCreateInfo render_pass_info{};
  render_pass_info.sType = VK_STRUCTURE_TYPE_RENDER_PASS_CREATE_INFO;
  render_pass_info.pNext = nullptr;
  render_pass_info.flags = 0;
  render_pass_info.attachmentCount = attachments.size();
  render_pass_info.pAttachments = attachments.data();
  render_pass_info.subpassCount = 1;
  render_pass_info.pSubpasses = &subpass;
  render_pass_info.dependencyCount = 1;
  render_pass_info.pDependencies = &dependency;

  if(vkCreateRenderPass(
       cg_core.vk_device_with_swapchain->device, &render_pass_info,
       nullptr, &self->render_pass) != VK_SUCCESS)
    throw CommandError{"Failed to create Vulkan Render Pass 2D."};
}

static void
unload_render_pass(void *obj)
{
  auto self = static_cast<VK::GraphicsPipeline2DWiredLayout*>(obj);

  vkDestroyRenderPass(
    cg_core.vk_device_with_swapchain->device, self->render_pass, nullptr);
}

const CommandChain loader{
  {&load_pipeline, &unload_pipeline},
  {&load_render_pass, &unload_render_pass}
};

}

namespace VK
{

GraphicsPipeline2DWiredLayout::GraphicsPipeline2DWiredLayout()
{
  loader.execute(this);
}

GraphicsPipeline2DWiredLayout::~GraphicsPipeline2DWiredLayout()
{
  loader.revert(this);
}

}