QueueManager.h

Go to the documentation of this file.

/*
 * Copyright 2022 Google LLC
 *
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */
 
#ifndef skgpu_graphite_QueueManager_DEFINED
#define skgpu_graphite_QueueManager_DEFINED
 
#include "include/core/SkRefCnt.h"
#include "include/gpu/graphite/GraphiteTypes.h"
#include "include/private/base/SkDeque.h"
#include "include/private/base/SkTArray.h"
#include "src/core/SkTHash.h"
 
#include <memory>
#include <vector>
 
namespace skgpu::graphite {
 
class Buffer;
class CommandBuffer;
class Context;
class GpuWorkSubmission;
struct InsertRecordingInfo;
class ResourceProvider;
class SharedContext;
class Task;
class UploadBufferManager;
 
class QueueManager {
public:
    virtual ~QueueManager();
 
    // Adds the commands from the passed in Recording to the current CommandBuffer
    [[nodiscard]] bool addRecording(const InsertRecordingInfo&, Context*);
 
    // Adds the commands from the passed in Task to the current CommandBuffer
    [[nodiscard]] bool addTask(Task*, Context*);
 
    // Adds a proc that will be called when the current CommandBuffer is submitted and finishes
    [[nodiscard]] bool addFinishInfo(const InsertFinishInfo&,
                                     ResourceProvider*,
                                     SkSpan<const sk_sp<Buffer>> buffersToAsyncMap = {});
 
    [[nodiscard]] bool submitToGpu();
    [[nodiscard]] bool hasUnfinishedGpuWork();
    void checkForFinishedWork(SyncToCpu);
 
#if defined(GRAPHITE_TEST_UTILS)
    virtual void startCapture() {}
    virtual void stopCapture() {}
#endif
 
    void returnCommandBuffer(std::unique_ptr<CommandBuffer>);
 
    virtual void tick() const {}
 
    void addUploadBufferManagerRefs(UploadBufferManager*);
 
protected:
    QueueManager(const SharedContext* sharedContext);
 
    using OutstandingSubmission = std::unique_ptr<GpuWorkSubmission>;
 
    const SharedContext* fSharedContext;
    std::unique_ptr<CommandBuffer> fCurrentCommandBuffer;
 
private:
    virtual std::unique_ptr<CommandBuffer> getNewCommandBuffer(ResourceProvider*) = 0;
    virtual OutstandingSubmission onSubmitToGpu() = 0;
 
    bool setupCommandBuffer(ResourceProvider*);
 
    SkDeque fOutstandingSubmissions;
 
    std::vector<std::unique_ptr<CommandBuffer>> fAvailableCommandBuffers;
 
    skia_private::THashMap<uint32_t, uint32_t> fLastAddedRecordingIDs;
};
 
} // namespace skgpu::graphite
 
#endif // skgpu_graphite_QueueManager_DEFINED