UploadContext.hpp

Go to the documentation of this file.

#pragma once
#include <Async/Future.hpp>
#include <Bits/Format.hpp>
#include <RHICore/Device.hpp>
#include <RHICore/Resource.hpp>
#include "StagingBuffer.hpp"
namespace Foundation::Rendering
{
    using namespace RHI;
    class UploadContext
    {
        RHIDevice* mDevice;
        Allocator* mAllocator;
 
        RHIDeviceQueue* mQueue;
        RHIDeviceScopedObjectHandle<RHICommandPool> mCommandPool;
        Vector<RHICommandPoolScopedHandle<RHICommandList>> mCommandLists;
        RHIDeviceScopedObjectHandle<RHIDeviceFence> mFence;
        StagingBuffer mStagingBuffer;
 
        Async::Mutex mMutex;
    public:
        UploadContext(RHIDevice* device, Allocator* allocator, size_t stagingBudget = 16_MB);
        // !! TODO: Bound checks!
        void Upload(RHIBuffer* dst, Span<const char> data, size_t dstOffset = 0, size_t alignment = 4,
                               RHIResourceAccess dst_access = RHIResourceAccessBits::ShaderRead,
                               RHIPipelineStage dst_stage = RHIPipelineStageBits::AllGraphics);
        void Upload(RHITexture* dst, Span<const char> data,
                               RHITextureSubresourceRange range,
                               RHICommandList::CopyImageRegion region,
                               RHIResourceAccess dst_access = RHIResourceAccessBits::ShaderRead,
                               RHIPipelineStage dst_stage = RHIPipelineStageBits::AllGraphics,
                               RHITextureLayout dst_layout = RHITextureLayout::ShaderReadOnly);
        void Upload(RHITexture* dst, Span<const char> data, uint32_t mipLevel = 0, uint32_t arrayLayer = 0,
                               RHITextureAspectFlag aspect = RHITextureAspectFlagBits::Color,
                               RHIResourceAccess dst_access = RHIResourceAccessBits::ShaderRead,
                               RHIPipelineStage dst_stage = RHIPipelineStageBits::AllGraphics,
                               RHITextureLayout dst_layout = RHITextureLayout::ShaderReadOnly);
        void SubmitAndWait();
        ~UploadContext();
    };
} // namespace Foundation::Rendering