DawnAsyncWait.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_DawnAsyncWait_DEFINED
#define skgpu_graphite_DawnAsyncWait_DEFINED
 
#include "include/core/SkTypes.h"
 
#include "webgpu/webgpu_cpp.h"  // NO_G3_REWRITE
 
#include <atomic>
#include <functional>
 
namespace skgpu::graphite {
 
class DawnSharedContext;
 
class DawnAsyncWait {
public:
    DawnAsyncWait(const DawnSharedContext*);
 
    // Returns true if the wait has been signaled and false otherwise. If the Context allows
    // yielding, then this function yields execution to the event loop where Dawn's asynchronous
    // tasks get scheduled and returns as soon as the loop yields the execution back to the caller.
    // Otherwise, it just checks.
    bool yieldAndCheck() const;
 
    // Returns true if it is legal to call busyWait().
    bool mayBusyWait() const;
 
    // Busy-waits until this wait has been signaled. May only be called if the Context allows
    // yielding.
    // TODO(armansito): This could benefit from a timeout in the case the wait never gets signaled.
    void busyWait() const;
 
    // Marks this wait as resolved. Once called, all calls to `yieldAndCheck` and `busyWait` will
    // return true immediately.
    void signal() { fSignaled = true; }
 
    // Resets this object into its unsignaled state.
    void reset() { fSignaled = false; }
 
private:
    const DawnSharedContext* fSharedContext;
    std::atomic_bool fSignaled;
};
 
template <typename T> class DawnAsyncResult {
public:
    DawnAsyncResult(const DawnSharedContext* sharedContext) : fSync(sharedContext) {}
 
    ~DawnAsyncResult() {
        if (fSync.mayBusyWait()) {
            fSync.busyWait();
        }
        SkASSERT(fSync.yieldAndCheck());
    }
 
    void set(const T& result) {
        fResult = result;
        fSync.signal();
    }
 
    const T* getIfReady() const { return fSync.yieldAndCheck() ? &fResult : nullptr; }
 
    const T& waitAndGet() const {
        // If fSync is already signaled, the wait will return immediately.
        fSync.busyWait();
        return fResult;
    }
 
private:
    DawnAsyncWait fSync;
    T fResult;
};
 
} // namespace skgpu::graphite
 
#endif // skgpu_graphite_DawnAsyncWait_DEFINED