/* Copyright (c) 2021-2025, NVIDIA CORPORATION. All rights reserved.
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * 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.
 */

#pragma once

#include "core/hpp_image.h"
#include "core/hpp_image_view.h"
#include <functional>
#include <memory>

namespace vkb
{
namespace core
{
class HPPDevice;
}

namespace rendering
{
/**
 * @brief Description of render pass attachments.
 * HPPAttachment descriptions can be used to automatically create render target images.
 */
struct HPPAttachment
{
	HPPAttachment() = default;

	HPPAttachment(vk::Format format, vk::SampleCountFlagBits samples, vk::ImageUsageFlags usage) :
	    format{format},
	    samples{samples},
	    usage{usage}
	{}

	vk::Format              format         = vk::Format::eUndefined;
	vk::SampleCountFlagBits samples        = vk::SampleCountFlagBits::e1;
	vk::ImageUsageFlags     usage          = vk::ImageUsageFlagBits::eSampled;
	vk::ImageLayout         initial_layout = vk::ImageLayout::eUndefined;
};

/**
 * @brief HPPRenderContext is a transcoded version of vkb::RenderContext from vulkan to vulkan-hpp.
 *
 * See vkb::RenderContext for documentation
 */
class HPPRenderTarget
{
  public:
	using CreateFunc = std::function<std::unique_ptr<HPPRenderTarget>(core::HPPImage &&)>;

	static const CreateFunc DEFAULT_CREATE_FUNC;

	HPPRenderTarget(std::vector<core::HPPImage> &&images);

	HPPRenderTarget(std::vector<core::HPPImageView> &&image_views);

	HPPRenderTarget(const HPPRenderTarget &) = delete;

	HPPRenderTarget(HPPRenderTarget &&) = delete;

	HPPRenderTarget &operator=(const HPPRenderTarget &other) noexcept = delete;

	HPPRenderTarget &operator=(HPPRenderTarget &&other) noexcept = delete;

	const vk::Extent2D                    &get_extent() const;
	const std::vector<core::HPPImageView> &get_views() const;
	const std::vector<HPPAttachment>      &get_attachments() const;

	/**
	 * @brief Sets the current input attachments overwriting the current ones
	 *        Should be set before beginning the render pass and before starting a new subpass
	 * @param input Set of attachment reference number to use as input
	 */
	void                         set_input_attachments(std::vector<uint32_t> &input);
	const std::vector<uint32_t> &get_input_attachments() const;

	/**
	 * @brief Sets the current output attachments overwriting the current ones
	 *        Should be set before beginning the render pass and before starting a new subpass
	 * @param output Set of attachment reference number to use as output
	 */
	void                         set_output_attachments(std::vector<uint32_t> &output);
	const std::vector<uint32_t> &get_output_attachments() const;
	void                         set_layout(uint32_t attachment, vk::ImageLayout layout);
	vk::ImageLayout              get_layout(uint32_t attachment) const;

  private:
	vkb::core::DeviceCpp const     &device;
	vk::Extent2D                    extent;
	std::vector<core::HPPImage>     images;
	std::vector<core::HPPImageView> views;
	std::vector<HPPAttachment>      attachments;
	std::vector<uint32_t>           input_attachments  = {};         // By default there are no input attachments
	std::vector<uint32_t>           output_attachments = {0};        // By default the output attachments is attachment 0
};
}        // namespace rendering
}        // namespace vkb