vk_upload_worker.h

Go to the documentation of this file.

#pragma once
 
#include <condition_variable>
#include <cstddef>
#include <deque>
#include <functional>
#include <mutex>
#include <thread>
#include <vector>
 
namespace tachyon::vulkan
{

    class UploadWorker
    {
    public:
        static constexpr size_t kInlineBytes = 32 * 1024;
 
        UploadWorker();
        ~UploadWorker();
        UploadWorker(const UploadWorker&) = delete;
        UploadWorker& operator=(const UploadWorker&) = delete;

        void Copy(void* destination, const void* source, size_t size);
        void Copy(void* destination, std::vector<std::byte>&& source);
        void Then(std::function<void()> job);
        void Wait();
        bool Busy() const;
 
    private:
        struct Job
        {
            void*                  destination = nullptr;
            const void*            source      = nullptr;
            size_t                 size        = 0;
            std::vector<std::byte> owned;
            std::function<void()>  then;
        };
        void Push(Job&& job);
        void Run();
 
        mutable std::mutex      m_mutex;
        std::condition_variable m_wake;  
        std::condition_variable m_done;  
        std::deque<Job>         m_jobs;
        bool                    m_running = false; 
        bool                    m_stop    = false;
        std::thread             m_thread;
    };

}