Seele/src/Engine/Graphics/Metal/Texture.h

#pragma once
#include "Graphics.h"
#include "Graphics/Texture.h"


namespace Seele {
namespace Metal {
class TextureBase {
  public:
    TextureBase(PGraphics graphics, MTL::TextureType type, const TextureCreateInfo& createInfo, Gfx::QueueType& owner,
                MTL::Texture* existingImage = nullptr);
    virtual ~TextureBase();
    uint32 getWidth() const { return width; }
    uint32 getHeight() const { return height; }
    uint32 getDepth() const { return depth; }
    constexpr MTL::Texture* getTexture() const { return texture; }
    constexpr Gfx::SeImageLayout getLayout() const { return layout; }
    void setLayout(Gfx::SeImageLayout val) { layout = val; }
    constexpr Gfx::SeFormat getFormat() const { return format; }
    constexpr Gfx::SeSampleCountFlags getNumSamples() const { return samples; }
    constexpr uint32 getMipLevels() const { return mipLevels; }
    void executeOwnershipBarrier(Gfx::QueueType newOwner) { currentOwner = newOwner; }
    void executePipelineBarrier(Gfx::SeAccessFlags srcAccess, Gfx::SePipelineStageFlags srcStage, Gfx::SeAccessFlags dstAccess,
                                Gfx::SePipelineStageFlags dstStage);
    void changeLayout(Gfx::SeImageLayout newLayout, Gfx::SeAccessFlags srcAccess, Gfx::SePipelineStageFlags srcStage,
                      Gfx::SeAccessFlags dstAccess, Gfx::SePipelineStageFlags dstStage);
    void download(uint32 mipLevel, uint32 arrayLayer, uint32 face, Array<uint8>& buffer);

  protected:
    // Updates via reference
    Gfx::QueueType& currentOwner;
    PGraphics graphics;
    uint32 width;
    uint32 height;
    uint32 depth;
    uint32 arrayCount;
    uint32 layerCount;
    uint32 mipLevels;
    uint32 samples;
    Gfx::SeFormat format;
    Gfx::SeImageUsageFlags usage;
    MTL::Texture* texture;
    MTL::TextureType type;
    Gfx::SeImageLayout layout;
    uint8 ownsImage;
    friend class Graphics;
};
DEFINE_REF(TextureBase)
class Texture2D : public Gfx::Texture2D, public TextureBase {
  public:
    Texture2D(PGraphics graphics, const TextureCreateInfo& createInfo, MTL::Texture* exisitingTexture = nullptr);
    virtual ~Texture2D();
    virtual uint32 getWidth() const override { return width; }
    virtual uint32 getHeight() const override { return height; }
    virtual uint32 getDepth() const override { return depth; }
    virtual Gfx::SeFormat getFormat() const override { return format; }
    virtual Gfx::SeSampleCountFlags getNumSamples() const override { return samples; }
    virtual uint32 getMipLevels() const override { return mipLevels; }
    virtual void changeLayout(Gfx::SeImageLayout newLayout, Gfx::SeAccessFlags srcAccess, Gfx::SePipelineStageFlags srcStage,
                              Gfx::SeAccessFlags dstAccess, Gfx::SePipelineStageFlags dstStage) override;
    virtual void download(uint32 mipLevel, uint32 arrayLayer, uint32 face, Array<uint8>& buffer) override;

  protected:
    // Inherited via QueueOwnedResource
    virtual void executeOwnershipBarrier(Gfx::QueueType newOwner) override;
    virtual void executePipelineBarrier(Gfx::SeAccessFlags srcAccess, Gfx::SePipelineStageFlags srcStage, Gfx::SeAccessFlags dstAccess,
                                        Gfx::SePipelineStageFlags dstStage) override;
};
DEFINE_REF(Texture2D)
class Texture3D : public Gfx::Texture3D, public TextureBase {
  public:
    Texture3D(PGraphics graphics, const TextureCreateInfo& createInfo);
    virtual ~Texture3D();
    virtual uint32 getWidth() const override { return width; }
    virtual uint32 getHeight() const override { return height; }
    virtual uint32 getDepth() const override { return depth; }
    virtual Gfx::SeFormat getFormat() const override { return format; }
    virtual Gfx::SeSampleCountFlags getNumSamples() const override { return samples; }
    virtual uint32 getMipLevels() const override { return mipLevels; }
    virtual void changeLayout(Gfx::SeImageLayout newLayout, Gfx::SeAccessFlags srcAccess, Gfx::SePipelineStageFlags srcStage,
                              Gfx::SeAccessFlags dstAccess, Gfx::SePipelineStageFlags dstStage) override;
    virtual void download(uint32 mipLevel, uint32 arrayLayer, uint32 face, Array<uint8>& buffer) override;

  protected:
    // Inherited via QueueOwnedResource
    virtual void executeOwnershipBarrier(Gfx::QueueType newOwner) override;
    virtual void executePipelineBarrier(Gfx::SeAccessFlags srcAccess, Gfx::SePipelineStageFlags srcStage, Gfx::SeAccessFlags dstAccess,
                                        Gfx::SePipelineStageFlags dstStage) override;
};
DEFINE_REF(Texture3D)
class TextureCube : public Gfx::TextureCube, TextureBase {
  public:
    TextureCube(PGraphics graphics, const TextureCreateInfo& createInfo);
    virtual ~TextureCube();
    virtual uint32 getWidth() const override { return width; }
    virtual uint32 getHeight() const override { return height; }
    virtual uint32 getDepth() const override { return depth; }
    virtual Gfx::SeFormat getFormat() const override { return format; }
    virtual Gfx::SeSampleCountFlags getNumSamples() const override { return samples; }
    virtual uint32 getMipLevels() const override { return mipLevels; }
    virtual void changeLayout(Gfx::SeImageLayout newLayout, Gfx::SeAccessFlags srcAccess, Gfx::SePipelineStageFlags srcStage,
                              Gfx::SeAccessFlags dstAccess, Gfx::SePipelineStageFlags dstStage) override;
    virtual void download(uint32 mipLevel, uint32 arrayLayer, uint32 face, Array<uint8>& buffer) override;

  protected:
    // Inherited via QueueOwnedResource
    virtual void executeOwnershipBarrier(Gfx::QueueType newOwner) override;
    virtual void executePipelineBarrier(Gfx::SeAccessFlags srcAccess, Gfx::SePipelineStageFlags srcStage, Gfx::SeAccessFlags dstAccess,
                                        Gfx::SePipelineStageFlags dstStage) override;
};
DEFINE_REF(TextureCube)
} // namespace Metal
} // namespace Seele