11 #include "flutter/fml/cpu_affinity.h"
12 #include "flutter/fml/thread.h"
13 #include "flutter/fml/trace_event.h"
20 static std::shared_ptr<WaitSetEntry>
Create(vk::UniqueFence p_fence,
21 const fml::closure& p_callback) {
22 return std::shared_ptr<WaitSetEntry>(
30 is_signalled_ = device.getFenceStatus(fence_.get()) == vk::Result::eSuccess;
33 const vk::Fence&
GetFence()
const {
return fence_.get(); }
38 vk::UniqueFence fence_;
39 fml::ScopedCleanupClosure callback_;
40 bool is_signalled_ =
false;
42 WaitSetEntry(vk::UniqueFence p_fence,
const fml::closure& p_callback)
43 : fence_(
std::move(p_fence)),
44 callback_(fml::ScopedCleanupClosure{p_callback}) {}
46 WaitSetEntry(
const WaitSetEntry&) =
delete;
48 WaitSetEntry(WaitSetEntry&&) =
delete;
50 WaitSetEntry& operator=(
const WaitSetEntry&) =
delete;
52 WaitSetEntry& operator=(WaitSetEntry&&) =
delete;
55 FenceWaiterVK::FenceWaiterVK(std::weak_ptr<DeviceHolderVK> device_holder)
56 : device_holder_(
std::move(device_holder)) {
57 waiter_thread_ = std::make_unique<std::thread>([&]() {
Main(); });
62 waiter_thread_->join();
66 const fml::closure& callback) {
67 if (!fence || !callback) {
72 std::scoped_lock lock(wait_set_mutex_);
78 wait_set_cv_.notify_one();
83 std::vector<vk::Fence> fences;
84 for (
const auto& entry : set) {
85 if (!entry->IsSignalled()) {
86 fences.emplace_back(entry->GetFence());
92 void FenceWaiterVK::Main() {
93 fml::Thread::SetCurrentThreadName(
94 fml::Thread::ThreadConfig{
"IplrVkFenceWait"});
96 fml::RequestAffinity(fml::CpuAffinity::kEfficiency);
100 bool terminate =
false;
103 std::unique_lock lock(wait_set_mutex_);
106 wait_set_cv_.wait(lock,
107 [&]() {
return !wait_set_.empty() || terminate_; });
110 terminate = terminate_;
124 void FenceWaiterVK::WaitUntilEmpty() {
127 FML_DCHECK(terminate_) <<
"Fence waiter must be terminated.";
128 while (!wait_set_.empty() && Wait()) {
133 bool FenceWaiterVK::Wait() {
137 std::scoped_lock lock(wait_set_mutex_);
138 wait_set = wait_set_;
141 using namespace std::literals::chrono_literals;
144 auto device_holder = device_holder_.lock();
145 if (!device_holder) {
149 const auto& device = device_holder->GetDevice();
155 if (fences.empty()) {
159 auto result = device.waitForFences(
163 std::chrono::nanoseconds{100ms}.count());
164 if (!(result == vk::Result::eSuccess || result == vk::Result::eTimeout)) {
165 VALIDATION_LOG <<
"Fence waiter encountered an unexpected error. Tearing "
166 "down the waiter thread.";
173 TRACE_EVENT0(
"impeller",
"CheckFenceStatus");
174 for (
auto& entry : wait_set) {
175 entry->UpdateSignalledStatus(device);
185 static constexpr
auto is_signalled = [](
const auto& entry) {
186 return entry->IsSignalled();
188 std::scoped_lock lock(wait_set_mutex_);
191 std::copy_if(wait_set_.begin(), wait_set_.end(),
192 std::back_inserter(erased_entries), is_signalled);
194 std::remove_if(wait_set_.begin(), wait_set_.end(), is_signalled),
199 TRACE_EVENT0(
"impeller",
"ClearSignaledFences");
201 erased_entries.clear();
209 std::scoped_lock lock(wait_set_mutex_);
212 wait_set_cv_.notify_one();