Image.h

#pragma once
 
#include "Common.h"
 
#include "Device.h"
#include "Log.h"
 
namespace Mandrill
{
    class Image
    {
    public:
        MANDRILL_NON_COPYABLE(Image)
 
        
        MANDRILL_API Image(ptr<Device> pDevice, uint32_t width, uint32_t height, uint32_t depth, uint32_t mipLevels,
                           VkSampleCountFlagBits samples, VkFormat format, VkImageTiling tiling,
                           VkImageUsageFlags usage, VkMemoryPropertyFlags properties);
 
        MANDRILL_API Image(ptr<Device> pDevice, uint32_t width, uint32_t height, uint32_t depth, uint32_t mipLevels,
                           VkSampleCountFlagBits samples, VkFormat format, VkImageTiling tiling,
                           VkImageUsageFlags usage, VkDeviceMemory memory, VkDeviceSize offset);
 
        MANDRILL_API ~Image();
 
        MANDRILL_API void createImageView(VkImageAspectFlags aspectFlags);
 
        MANDRILL_API VkImage getImage() const
        {
            return mImage;
        }
 
        MANDRILL_API void setImageView(VkImageView imageView)
        {
            mImageView = imageView;
        }
 
        MANDRILL_API VkImageView getImageView() const
        {
            return mImageView;
        }
 
        MANDRILL_API VkDeviceMemory getMemory() const
        {
            return mMemory;
        }
 
        MANDRILL_API VkImageUsageFlags getUsage() const
        {
            return mUsage;
        }
 
        MANDRILL_API VkMemoryPropertyFlags getProperties() const
        {
            return mProperties;
        }
 
        MANDRILL_API void* getHostMap() const
        {
            if (!(mProperties & VK_MEMORY_PROPERTY_HOST_COHERENT_BIT)) {
                Log::Error("Unable to access host map of buffer that is not host coherent.");
                return nullptr;
            }
            return mpHostMap;
        }
 
        MANDRILL_API VkFormat getFormat() const
        {
            return mFormat;
        }
 
        MANDRILL_API uint32_t getWidth() const
        {
            return mWidth;
        }
 
        MANDRILL_API uint32_t getHeight() const
        {
            return mHeight;
        }
 
        MANDRILL_API uint32_t getDepth() const
        {
            return mDepth;
        }
 
        MANDRILL_API uint32_t getMipLevels() const
        {
            return mMipLevels;
        }
 
    private:
        ptr<Device> mpDevice;
 
        VkImage mImage;
        VkImageView mImageView;
 
        VkImageUsageFlags mUsage;
        VkMemoryPropertyFlags mProperties;
 
        VkDeviceMemory mMemory;
        bool mOwnMemory;
        void* mpHostMap;
 
        uint32_t mWidth;
        uint32_t mHeight;
        uint32_t mDepth;
 
        uint32_t mMipLevels;
        VkFormat mFormat;
        VkImageTiling mTiling;
    };
} // namespace Mandrill