#pragma once

#include <vector>

#include "vulkan_buffer.h"
#include "vulkan_common.h"

#include "../../main/rendering/image.h"

namespace progressia::desktop {

class Image : public VkObjectWrapper {
  public:
    VkImage vk;
    VkImageView view;
    VkFormat format;

    Image(VkImage, VkImageView, VkFormat);
    virtual ~Image();
};

class ManagedImage : public Image {

  public:
    VkDeviceMemory memory;
    Vulkan &vulkan;

    struct State {
        VkImageLayout layout;
        VkAccessFlags accessMask;
        VkPipelineStageFlags stageMask;
    };

  private:
    State state;

  public:
    ManagedImage(std::size_t width, std::size_t height, VkFormat format,
                 VkImageAspectFlags aspect, VkImageUsageFlags usage,
                 Vulkan &vulkan);
    ~ManagedImage();

    void transition(State);
};

class Texture : public ManagedImage {

  public:
    VkSampler sampler;
    VkDescriptorSet descriptorSet;

    Texture(const main::Image &src, Vulkan &vulkan);
    ~Texture();

    void bind();
};

} // namespace progressia::desktop