1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/viz/test/fake_external_begin_frame_source.h"
#include "base/functional/bind.h"
#include "base/location.h"
#include "base/logging.h"
#include "base/memory/raw_ptr.h"
#include "base/task/single_thread_task_runner.h"
#include "base/time/time.h"
#include "components/viz/test/begin_frame_args_test.h"
namespace viz {
FakeExternalBeginFrameSource::FakeExternalBeginFrameSource(
double refresh_rate,
bool tick_automatically)
: BeginFrameSource(kNotRestartableId),
tick_automatically_(tick_automatically),
milliseconds_per_frame_(1000.0 / refresh_rate) {
DETACH_FROM_SEQUENCE(sequence_checker_);
}
FakeExternalBeginFrameSource::~FakeExternalBeginFrameSource() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
}
void FakeExternalBeginFrameSource::SetPaused(bool paused) {
if (paused != paused_) {
paused_ = paused;
std::set<raw_ptr<BeginFrameObserver, SetExperimental>> observers(
observers_);
for (BeginFrameObserver* obs : observers) {
obs->OnBeginFrameSourcePausedChanged(paused_);
}
}
}
void FakeExternalBeginFrameSource::AddObserver(BeginFrameObserver* obs) {
DCHECK(obs);
DCHECK(observers_.find(obs) == observers_.end());
bool observers_was_empty = observers_.empty();
observers_.insert(obs);
obs->OnBeginFrameSourcePausedChanged(paused_);
if (observers_was_empty && tick_automatically_) {
PostTestOnBeginFrame();
} else if (current_args_.IsValid()) {
const BeginFrameArgs& last_args = obs->LastUsedBeginFrameArgs();
if (!last_args.IsValid() ||
last_args.frame_time < current_args_.frame_time) {
current_args_.type = BeginFrameArgs::MISSED;
obs->OnBeginFrame(current_args_);
}
}
if (client_)
client_->OnAddObserver(obs);
}
void FakeExternalBeginFrameSource::RemoveObserver(BeginFrameObserver* obs) {
DCHECK(obs);
DCHECK(observers_.find(obs) != observers_.end());
observers_.erase(obs);
if (observers_.empty())
begin_frame_task_.Cancel();
if (client_)
client_->OnRemoveObserver(obs);
}
void FakeExternalBeginFrameSource::DidFinishFrame(BeginFrameObserver* obs) {
pending_frames_[obs]--;
}
BeginFrameArgs FakeExternalBeginFrameSource::CreateBeginFrameArgs(
BeginFrameArgs::CreationLocation location,
const base::TickClock* now_src) {
return CreateBeginFrameArgsForTesting(location, source_id(),
next_begin_frame_number_++, now_src);
}
BeginFrameArgs FakeExternalBeginFrameSource::CreateBeginFrameArgs(
BeginFrameArgs::CreationLocation location) {
return CreateBeginFrameArgsForTesting(location, source_id(),
next_begin_frame_number_++);
}
BeginFrameArgs FakeExternalBeginFrameSource::CreateBeginFrameArgsWithGenerator(
base::TimeTicks frame_time,
base::TimeTicks next_frame_time,
base::TimeDelta vsync_interval) {
return begin_frame_args_generator_.GenerateBeginFrameArgs(
source_id(), frame_time, next_frame_time, vsync_interval);
}
void FakeExternalBeginFrameSource::TestOnBeginFrame(
const BeginFrameArgs& args) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
current_args_ = args;
std::set<raw_ptr<BeginFrameObserver, SetExperimental>> observers(observers_);
for (BeginFrameObserver* obs : observers) {
pending_frames_[obs]++;
obs->OnBeginFrame(current_args_);
}
if (tick_automatically_)
PostTestOnBeginFrame();
}
void FakeExternalBeginFrameSource::PostTestOnBeginFrame() {
begin_frame_task_.Reset(
base::BindOnce(&FakeExternalBeginFrameSource::TestOnBeginFrame,
weak_ptr_factory_.GetWeakPtr(),
CreateBeginFrameArgs(BEGINFRAME_FROM_HERE)));
base::SingleThreadTaskRunner::GetCurrentDefault()->PostDelayedTask(
FROM_HERE, begin_frame_task_.callback(),
base::Milliseconds(milliseconds_per_frame_));
next_begin_frame_number_++;
}
bool FakeExternalBeginFrameSource::AllFramesDidFinish() {
bool found_pending_frames = false;
for (auto const& entry : pending_frames_) {
if (entry.second != 0) {
LOG(WARNING) << "Observer " << entry.first << " has " << entry.second
<< " pending frame(s)";
found_pending_frames = true;
}
}
return !found_pending_frames;
}
} // namespace viz
|
