blob: 1536d1c1e0e2fb8a8aa5610252ee00c75d10891f [file] [log] [blame]
/*
* Copyright 2020 Google LLC
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef FlushFinishTracker_DEFINED
#define FlushFinishTracker_DEFINED
#include "include/core/SkRefCnt.h"
#include "include/gpu/GpuTypes.h"
class GrDirectContext;
#ifdef SK_GRAPHITE_ENABLED
namespace skgpu::graphite { class Context; }
#endif
namespace sk_gpu_test {
class FlushFinishTracker : public SkRefCnt {
public:
static void FlushFinished(void* finishedContext) {
auto tracker = static_cast<FlushFinishTracker*>(finishedContext);
tracker->setFinished();
tracker->unref();
}
static void FlushFinishedResult(void* finishedContext, skgpu::CallbackResult) {
FlushFinished(finishedContext);
}
FlushFinishTracker(GrDirectContext* context) : fContext(context) {}
#ifdef SK_GRAPHITE_ENABLED
FlushFinishTracker(skgpu::graphite::Context* context) : fGraphiteContext(context) {}
#endif
void setFinished() { fIsFinished = true; }
void waitTillFinished();
private:
GrDirectContext* fContext = nullptr;
#ifdef SK_GRAPHITE_ENABLED
skgpu::graphite::Context* fGraphiteContext = nullptr;
#endif
// Currently we don't have the this bool be atomic cause all current uses of this class happen
// on a single thread. In other words we call flush, checkAsyncWorkCompletion, and
// waitTillFinished all on the same thread. If we ever want to support the flushing and waiting
// to happen on different threads then we should make this atomic.
bool fIsFinished = false;
};
} //namespace sk_gpu_test
#endif