Image2D.hpp

#pragma once
 
#include "PixelBullet/Filesystem/Resources/Resource.hpp"
#include "PixelBullet/Filesystem/VirtualPath.hpp"
#include "PixelBullet/Graphics/Images/Image.hpp"
 
#include <filesystem>
#include <typeindex>
 
namespace PixelBullet
{
    class Node;
 
    class Image2D : public Image, public Resource
    {
    public:
        static std::shared_ptr<Image2D> Create(const Node& node);
 
        static std::shared_ptr<Image2D> Create(const VirtualPath& filename, VkFilter filter = VK_FILTER_LINEAR,
                                               VkSamplerAddressMode addressMode = VK_SAMPLER_ADDRESS_MODE_REPEAT,
                                               bool anisotropic = true, bool mipmap = true);
 
        explicit Image2D(const VirtualPath& filename, VkFilter filter = VK_FILTER_LINEAR,
                         VkSamplerAddressMode addressMode = VK_SAMPLER_ADDRESS_MODE_REPEAT, bool anisotropic = true,
                         bool mipmap = true, bool load = true);
 
        explicit Image2D(const glm::uvec2& extent, VkFormat format = VK_FORMAT_R8G8B8A8_UNORM,
                         VkImageLayout layout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL,
                         VkImageUsageFlags usage = VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT | VK_IMAGE_USAGE_STORAGE_BIT,
                         VkFilter filter = VK_FILTER_LINEAR,
                         VkSamplerAddressMode addressMode = VK_SAMPLER_ADDRESS_MODE_CLAMP_TO_EDGE,
                         VkSampleCountFlagBits samples = VK_SAMPLE_COUNT_1_BIT, bool anisotropic = false,
                         bool mipmap = false);
 
        explicit Image2D(std::unique_ptr<Bitmap>&& bitmap, VkFormat format = VK_FORMAT_R8G8B8A8_UNORM,
                         VkImageLayout layout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL,
                         VkImageUsageFlags usage = VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT | VK_IMAGE_USAGE_STORAGE_BIT,
                         VkFilter filter = VK_FILTER_LINEAR,
                         VkSamplerAddressMode addressMode = VK_SAMPLER_ADDRESS_MODE_CLAMP_TO_EDGE,
                         VkSampleCountFlagBits samples = VK_SAMPLE_COUNT_1_BIT, bool anisotropic = false,
                         bool mipmap = false);
 
        void SetPixels(const uint8_t* pixels, uint32_t layerCount, uint32_t baseArrayLayer);
 
        std::type_index GetTypeIndex() const override
        {
            return typeid(Image2D);
        }
 
        const VirtualPath& GetFilename() const
        {
            return m_Filename;
        }
        bool IsAnisotropic() const
        {
            return m_Anisotropic;
        }
        bool IsMipmap() const
        {
            return m_Mipmap;
        }
        uint32_t GetComponents() const
        {
            return m_Components;
        }
 
        friend const Node& operator>>(const Node& node, Image2D& image);
        friend Node& operator<<(Node& node, const Image2D& image);
 
    private:
        void Load(std::unique_ptr<Bitmap> loadBitmap = nullptr);
 
    private:
        VirtualPath m_Filename;
 
        bool m_Anisotropic;
        bool m_Mipmap;
        uint32_t m_Components = 0;
    };
} // namespace PixelBullet