Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(109)

Side by Side Diff: media/cast/test/cast_benchmarks.cc

Issue 308713005: Cast: Synthetic benchmark tool. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: merge + minor fix Created 6 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « media/cast/receiver/video_decoder.cc ('k') | media/cast/test/fake_single_thread_task_runner.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 //
5 // This program benchmarks the theoretical throughput of the cast library.
6 // It runs using a fake clock, simulated network and fake codecs. This allows
7 // tests to run much faster than real time.
8 // To run the program, run:
9 // $ ./out/Release/cast_benchmarks | tee benchmarkoutput.asc
10 // This may take a while, when it is done, you can view the data with
11 // meshlab by running:
12 // $ meshlab benchmarkoutput.asc
13 // After starting meshlab, turn on Render->Show Axis. The red axis will
14 // represent bandwidth (in megabits) the blue axis will be packet drop
15 // (in percent) and the green axis will be latency (in milliseconds).
16 //
17 // This program can also be used for profiling. On linux it has
18 // built-in support for this. Simply set the environment variable
19 // PROFILE_FILE before running it, like so:
20 // $ export PROFILE_FILE=cast_benchmark.profile
21 // Then after running the program, you can view the profile with:
22 // $ pprof ./out/Release/cast_benchmarks $PROFILE_FILE --gv
23
24 #include <math.h>
25 #include <stdint.h>
26
27 #include <map>
28 #include <vector>
29
30 #include "base/at_exit.h"
31 #include "base/bind.h"
32 #include "base/bind_helpers.h"
33 #include "base/command_line.h"
34 #include "base/debug/profiler.h"
35 #include "base/stl_util.h"
36 #include "base/strings/string_number_conversions.h"
37 #include "base/strings/stringprintf.h"
38 #include "base/test/simple_test_tick_clock.h"
39 #include "base/threading/thread.h"
40 #include "base/time/tick_clock.h"
41 #include "media/base/audio_bus.h"
42 #include "media/base/video_frame.h"
43 #include "media/cast/cast_config.h"
44 #include "media/cast/cast_environment.h"
45 #include "media/cast/cast_receiver.h"
46 #include "media/cast/cast_sender.h"
47 #include "media/cast/logging/simple_event_subscriber.h"
48 #include "media/cast/test/fake_single_thread_task_runner.h"
49 #include "media/cast/test/skewed_single_thread_task_runner.h"
50 #include "media/cast/test/skewed_tick_clock.h"
51 #include "media/cast/test/utility/audio_utility.h"
52 #include "media/cast/test/utility/default_config.h"
53 #include "media/cast/test/utility/test_util.h"
54 #include "media/cast/test/utility/udp_proxy.h"
55 #include "media/cast/test/utility/video_utility.h"
56 #include "media/cast/transport/cast_transport_config.h"
57 #include "media/cast/transport/cast_transport_defines.h"
58 #include "media/cast/transport/cast_transport_sender.h"
59 #include "media/cast/transport/cast_transport_sender_impl.h"
60 #include "testing/gtest/include/gtest/gtest.h"
61
62 namespace media {
63 namespace cast {
64
65 namespace {
66
67 static const int64 kStartMillisecond = INT64_C(1245);
68 static const int kAudioChannels = 2;
69 static const int kVideoHdWidth = 1280;
70 static const int kVideoHdHeight = 720;
71 static const int kTargetDelay = 300;
72
73 // The tests are commonly implemented with |kFrameTimerMs| RunTask function;
74 // a normal video is 30 fps hence the 33 ms between frames.
75 static const int kFrameTimerMs = 33;
76
77 void UpdateCastTransportStatus(transport::CastTransportStatus status) {
78 bool result = (status == transport::TRANSPORT_AUDIO_INITIALIZED ||
79 status == transport::TRANSPORT_VIDEO_INITIALIZED);
80 EXPECT_TRUE(result);
81 }
82
83 void AudioInitializationStatus(CastInitializationStatus status) {
84 EXPECT_EQ(STATUS_AUDIO_INITIALIZED, status);
85 }
86
87 void VideoInitializationStatus(CastInitializationStatus status) {
88 EXPECT_EQ(STATUS_VIDEO_INITIALIZED, status);
89 }
90
91 void IgnoreRawEvents(const std::vector<PacketEvent>& packet_events) {
92 }
93
94 } // namespace
95
96 // Shim that turns forwards packets from a test::PacketPipe to a
97 // PacketReceiverCallback.
98 class LoopBackPacketPipe : public test::PacketPipe {
99 public:
100 LoopBackPacketPipe(const transport::PacketReceiverCallback& packet_receiver)
101 : packet_receiver_(packet_receiver) {}
102
103 virtual ~LoopBackPacketPipe() {}
104
105 // PacketPipe implementations.
106 virtual void Send(scoped_ptr<transport::Packet> packet) OVERRIDE {
107 packet_receiver_.Run(packet.Pass());
108 }
109
110 private:
111 transport::PacketReceiverCallback packet_receiver_;
112 };
113
114 // Class that sends the packet direct from sender into the receiver with the
115 // ability to drop packets between the two.
116 // TODO(hubbe): Break this out and share code with end2end_unittest.cc
117 class LoopBackTransport : public transport::PacketSender {
118 public:
119 explicit LoopBackTransport(scoped_refptr<CastEnvironment> cast_environment)
120 : cast_environment_(cast_environment) {}
121
122 void SetPacketReceiver(
123 const transport::PacketReceiverCallback& packet_receiver,
124 const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
125 base::TickClock* clock) {
126 scoped_ptr<test::PacketPipe> loopback_pipe(
127 new LoopBackPacketPipe(packet_receiver));
128 if (packet_pipe_) {
129 packet_pipe_->AppendToPipe(loopback_pipe.Pass());
130 } else {
131 packet_pipe_ = loopback_pipe.Pass();
132 }
133 packet_pipe_->InitOnIOThread(task_runner, clock);
134 }
135
136 virtual bool SendPacket(transport::PacketRef packet,
137 const base::Closure& cb) OVERRIDE {
138 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN));
139 scoped_ptr<Packet> packet_copy(new Packet(packet->data));
140 packet_pipe_->Send(packet_copy.Pass());
141 return true;
142 }
143
144 void SetPacketPipe(scoped_ptr<test::PacketPipe> pipe) {
145 // Append the loopback pipe to the end.
146 pipe->AppendToPipe(packet_pipe_.Pass());
147 packet_pipe_ = pipe.Pass();
148 }
149
150 private:
151 scoped_refptr<CastEnvironment> cast_environment_;
152 scoped_ptr<test::PacketPipe> packet_pipe_;
153 };
154
155 // Wraps a CastTransportSender and records some statistics about
156 // the data that goes through it.
157 class CastTransportSenderWrapper : public transport::CastTransportSender {
158 public:
159 // Takes ownership of |transport|.
160 void Init(CastTransportSender* transport,
161 uint64* encoded_video_bytes,
162 uint64* encoded_audio_bytes) {
163 transport_.reset(transport);
164 encoded_video_bytes_ = encoded_video_bytes;
165 encoded_audio_bytes_ = encoded_audio_bytes;
166 }
167
168 virtual void InitializeAudio(
169 const transport::CastTransportAudioConfig& config) OVERRIDE {
170 transport_->InitializeAudio(config);
171 }
172
173 virtual void InitializeVideo(
174 const transport::CastTransportVideoConfig& config) OVERRIDE {
175 transport_->InitializeVideo(config);
176 }
177
178 virtual void SetPacketReceiver(
179 const transport::PacketReceiverCallback& packet_receiver) OVERRIDE {
180 transport_->SetPacketReceiver(packet_receiver);
181 }
182
183 virtual void InsertCodedAudioFrame(
184 const transport::EncodedFrame& audio_frame) OVERRIDE {
185 *encoded_audio_bytes_ += audio_frame.data.size();
186 transport_->InsertCodedAudioFrame(audio_frame);
187 }
188
189 virtual void InsertCodedVideoFrame(
190 const transport::EncodedFrame& video_frame) OVERRIDE {
191 *encoded_video_bytes_ += video_frame.data.size();
192 transport_->InsertCodedVideoFrame(video_frame);
193 }
194
195 virtual void SendRtcpFromRtpSender(uint32 packet_type_flags,
196 uint32 ntp_seconds,
197 uint32 ntp_fraction,
198 uint32 rtp_timestamp,
199 const transport::RtcpDlrrReportBlock& dlrr,
200 uint32 sending_ssrc,
201 const std::string& c_name) OVERRIDE {
202 transport_->SendRtcpFromRtpSender(packet_type_flags,
203 ntp_seconds,
204 ntp_fraction,
205 rtp_timestamp,
206 dlrr,
207 sending_ssrc,
208 c_name);
209 }
210
211 // Retransmission request.
212 virtual void ResendPackets(
213 bool is_audio,
214 const MissingFramesAndPacketsMap& missing_packets,
215 bool cancel_rtx_if_not_in_list) OVERRIDE {
216 transport_->ResendPackets(
217 is_audio, missing_packets, cancel_rtx_if_not_in_list);
218 }
219
220 private:
221 scoped_ptr<transport::CastTransportSender> transport_;
222 uint64* encoded_video_bytes_;
223 uint64* encoded_audio_bytes_;
224 };
225
226 struct MeasuringPoint {
227 MeasuringPoint(double bitrate_, double latency_, double percent_packet_drop_)
228 : bitrate(bitrate_),
229 latency(latency_),
230 percent_packet_drop(percent_packet_drop_) {}
231 bool operator<=(const MeasuringPoint& other) const {
232 return bitrate >= other.bitrate && latency <= other.latency &&
233 percent_packet_drop <= other.percent_packet_drop;
234 }
235 bool operator>=(const MeasuringPoint& other) const {
236 return bitrate <= other.bitrate && latency >= other.latency &&
237 percent_packet_drop >= other.percent_packet_drop;
238 }
239
240 std::string AsString() const {
241 return base::StringPrintf(
242 "%f Mbit/s %f ms %f %% ", bitrate, latency, percent_packet_drop);
243 }
244
245 double bitrate;
246 double latency;
247 double percent_packet_drop;
248 };
249
250 class RunOneBenchmark {
251 public:
252 RunOneBenchmark()
253 : start_time_(),
254 task_runner_(new test::FakeSingleThreadTaskRunner(&testing_clock_)),
255 testing_clock_sender_(new test::SkewedTickClock(&testing_clock_)),
256 task_runner_sender_(
257 new test::SkewedSingleThreadTaskRunner(task_runner_)),
258 testing_clock_receiver_(new test::SkewedTickClock(&testing_clock_)),
259 task_runner_receiver_(
260 new test::SkewedSingleThreadTaskRunner(task_runner_)),
261 cast_environment_sender_(new CastEnvironment(
262 scoped_ptr<base::TickClock>(testing_clock_sender_).Pass(),
263 task_runner_sender_,
264 task_runner_sender_,
265 task_runner_sender_)),
266 cast_environment_receiver_(new CastEnvironment(
267 scoped_ptr<base::TickClock>(testing_clock_receiver_).Pass(),
268 task_runner_receiver_,
269 task_runner_receiver_,
270 task_runner_receiver_)),
271 receiver_to_sender_(cast_environment_receiver_),
272 sender_to_receiver_(cast_environment_sender_),
273 video_bytes_encoded_(0),
274 audio_bytes_encoded_(0),
275 frames_sent_(0) {
276 testing_clock_.Advance(
277 base::TimeDelta::FromMilliseconds(kStartMillisecond));
278 }
279
280 void Configure(transport::VideoCodec video_codec,
281 transport::AudioCodec audio_codec,
282 int audio_sampling_frequency,
283 int max_number_of_video_buffers_used) {
284 audio_sender_config_.rtp_config.ssrc = 1;
285 audio_sender_config_.incoming_feedback_ssrc = 2;
286 audio_sender_config_.rtp_config.payload_type = 96;
287 audio_sender_config_.use_external_encoder = false;
288 audio_sender_config_.frequency = audio_sampling_frequency;
289 audio_sender_config_.channels = kAudioChannels;
290 audio_sender_config_.bitrate = kDefaultAudioEncoderBitrate;
291 audio_sender_config_.codec = audio_codec;
292 audio_sender_config_.rtp_config.max_delay_ms = kTargetDelay;
293
294 audio_receiver_config_.feedback_ssrc =
295 audio_sender_config_.incoming_feedback_ssrc;
296 audio_receiver_config_.incoming_ssrc = audio_sender_config_.rtp_config.ssrc;
297 audio_receiver_config_.rtp_payload_type =
298 audio_sender_config_.rtp_config.payload_type;
299 audio_receiver_config_.frequency = audio_sender_config_.frequency;
300 audio_receiver_config_.channels = kAudioChannels;
301 audio_receiver_config_.max_frame_rate = 100;
302 audio_receiver_config_.codec.audio = audio_sender_config_.codec;
303 audio_receiver_config_.rtp_max_delay_ms = kTargetDelay;
304
305 video_sender_config_.rtp_config.ssrc = 3;
306 video_sender_config_.incoming_feedback_ssrc = 4;
307 video_sender_config_.rtp_config.payload_type = 97;
308 video_sender_config_.use_external_encoder = false;
309 video_sender_config_.width = kVideoHdWidth;
310 video_sender_config_.height = kVideoHdHeight;
311 #if 0
312 video_sender_config_.max_bitrate = 10000000; // 10Mbit max
313 video_sender_config_.min_bitrate = 1000000; // 1Mbit min
314 video_sender_config_.start_bitrate = 1000000; // 1Mbit start
315 #else
316 video_sender_config_.max_bitrate = 4000000; // 4Mbit all the time
317 video_sender_config_.min_bitrate = 4000000;
318 video_sender_config_.start_bitrate = 4000000;
319 #endif
320 video_sender_config_.max_qp = 56;
321 video_sender_config_.min_qp = 4;
322 video_sender_config_.max_frame_rate = 30;
323 video_sender_config_.max_number_of_video_buffers_used =
324 max_number_of_video_buffers_used;
325 video_sender_config_.codec = video_codec;
326 video_sender_config_.rtp_config.max_delay_ms = kTargetDelay;
327
328 video_receiver_config_.feedback_ssrc =
329 video_sender_config_.incoming_feedback_ssrc;
330 video_receiver_config_.incoming_ssrc = video_sender_config_.rtp_config.ssrc;
331 video_receiver_config_.rtp_payload_type =
332 video_sender_config_.rtp_config.payload_type;
333 video_receiver_config_.codec.video = video_sender_config_.codec;
334 video_receiver_config_.frequency = kVideoFrequency;
335 video_receiver_config_.channels = 1;
336 video_receiver_config_.max_frame_rate = 100;
337 video_receiver_config_.rtp_max_delay_ms = kTargetDelay;
338 }
339
340 void SetSenderClockSkew(double skew, base::TimeDelta offset) {
341 testing_clock_sender_->SetSkew(skew, offset);
342 task_runner_sender_->SetSkew(1.0 / skew);
343 }
344
345 void SetReceiverClockSkew(double skew, base::TimeDelta offset) {
346 testing_clock_receiver_->SetSkew(skew, offset);
347 task_runner_receiver_->SetSkew(1.0 / skew);
348 }
349
350 void Create() {
351 cast_receiver_ = CastReceiver::Create(cast_environment_receiver_,
352 audio_receiver_config_,
353 video_receiver_config_,
354 &receiver_to_sender_);
355 net::IPEndPoint dummy_endpoint;
356 transport_sender_.Init(new transport::CastTransportSenderImpl(
357 NULL,
358 testing_clock_sender_,
359 dummy_endpoint,
360 base::Bind(&UpdateCastTransportStatus),
361 base::Bind(&IgnoreRawEvents),
362 base::TimeDelta::FromSeconds(1),
363 task_runner_sender_,
364 &sender_to_receiver_),
365 &video_bytes_encoded_,
366 &audio_bytes_encoded_);
367
368 cast_sender_ =
369 CastSender::Create(cast_environment_sender_, &transport_sender_);
370
371 // Initializing audio and video senders.
372 cast_sender_->InitializeAudio(audio_sender_config_,
373 base::Bind(&AudioInitializationStatus));
374 cast_sender_->InitializeVideo(video_sender_config_,
375 base::Bind(&VideoInitializationStatus),
376 CreateDefaultVideoEncodeAcceleratorCallback(),
377 CreateDefaultVideoEncodeMemoryCallback());
378
379 receiver_to_sender_.SetPacketReceiver(
380 cast_sender_->packet_receiver(), task_runner_, &testing_clock_);
381 sender_to_receiver_.SetPacketReceiver(
382 cast_receiver_->packet_receiver(), task_runner_, &testing_clock_);
383 }
384
385 virtual ~RunOneBenchmark() {
386 cast_sender_.reset();
387 cast_receiver_.reset();
388 task_runner_->RunTasks();
389 }
390
391 void SendFakeVideoFrame() {
392 frames_sent_++;
393 cast_sender_->video_frame_input()->InsertRawVideoFrame(
394 media::VideoFrame::CreateBlackFrame(gfx::Size(2, 2)),
395 testing_clock_sender_->NowTicks());
396 }
397
398 void RunTasks(int ms) {
399 task_runner_->Sleep(base::TimeDelta::FromMilliseconds(ms));
400 }
401
402 void BasicPlayerGotVideoFrame(
403 const scoped_refptr<media::VideoFrame>& video_frame,
404 const base::TimeTicks& render_time,
405 bool continuous) {
406 video_ticks_.push_back(
407 std::make_pair(testing_clock_receiver_->NowTicks(), render_time));
408 cast_receiver_->RequestDecodedVideoFrame(base::Bind(
409 &RunOneBenchmark::BasicPlayerGotVideoFrame, base::Unretained(this)));
410 }
411
412 void BasicPlayerGotAudioFrame(scoped_ptr<AudioBus> audio_bus,
413 const base::TimeTicks& playout_time,
414 bool is_continuous) {
415 audio_ticks_.push_back(
416 std::make_pair(testing_clock_receiver_->NowTicks(), playout_time));
417 cast_receiver_->RequestDecodedAudioFrame(base::Bind(
418 &RunOneBenchmark::BasicPlayerGotAudioFrame, base::Unretained(this)));
419 }
420
421 void StartBasicPlayer() {
422 cast_receiver_->RequestDecodedVideoFrame(base::Bind(
423 &RunOneBenchmark::BasicPlayerGotVideoFrame, base::Unretained(this)));
424 cast_receiver_->RequestDecodedAudioFrame(base::Bind(
425 &RunOneBenchmark::BasicPlayerGotAudioFrame, base::Unretained(this)));
426 }
427
428 scoped_ptr<test::PacketPipe> CreateSimplePipe(const MeasuringPoint& p) {
429 scoped_ptr<test::PacketPipe> pipe = test::NewBuffer(65536, p.bitrate);
430 pipe->AppendToPipe(
431 test::NewRandomDrop(p.percent_packet_drop / 100.0).Pass());
432 pipe->AppendToPipe(test::NewConstantDelay(p.latency / 1000.0));
433 return pipe.Pass();
434 }
435
436 void Run(const MeasuringPoint& p) {
437 available_bitrate_ = p.bitrate;
438 Configure(transport::kFakeSoftwareVideo, transport::kPcm16, 32000, 1);
439 receiver_to_sender_.SetPacketPipe(CreateSimplePipe(p).Pass());
440 sender_to_receiver_.SetPacketPipe(CreateSimplePipe(p).Pass());
441 Create();
442 StartBasicPlayer();
443
444 for (int frame = 0; frame < 1000; frame++) {
445 SendFakeVideoFrame();
446 RunTasks(kFrameTimerMs);
447 }
448 RunTasks(100 * kFrameTimerMs); // Empty the pipeline.
449 VLOG(1) << "=============INPUTS============";
450 VLOG(1) << "Bitrate: " << p.bitrate << " mbit/s";
451 VLOG(1) << "Latency: " << p.latency << " ms";
452 VLOG(1) << "Packet drop drop: " << p.percent_packet_drop << "%";
453 VLOG(1) << "=============OUTPUTS============";
454 VLOG(1) << "Frames lost: " << frames_lost();
455 VLOG(1) << "Late frames: " << late_frames();
456 VLOG(1) << "Playout margin: " << frame_playout_buffer().AsString();
457 VLOG(1) << "Video bandwidth used: " << video_bandwidth() << " mbit/s ("
458 << (video_bandwidth() * 100 / desired_video_bitrate()) << "%)";
459 VLOG(1) << "Good run: " << SimpleGood();
460 }
461
462 // Metrics
463 int frames_lost() const { return frames_sent_ - video_ticks_.size(); }
464
465 int late_frames() const {
466 int frames = 0;
467 // Ignore the first two seconds of video or so.
468 for (size_t i = 60; i < video_ticks_.size(); i++) {
469 if (video_ticks_[i].first > video_ticks_[i].second) {
470 frames++;
471 }
472 }
473 return frames;
474 }
475
476 test::MeanAndError frame_playout_buffer() const {
477 std::vector<double> values;
478 for (size_t i = 0; i < video_ticks_.size(); i++) {
479 values.push_back(
480 (video_ticks_[i].second - video_ticks_[i].first).InMillisecondsF());
481 }
482 return test::MeanAndError(values);
483 }
484
485 // Mbits per second
486 double video_bandwidth() const {
487 double seconds = (kFrameTimerMs * frames_sent_ / 1000.0);
488 double megabits = video_bytes_encoded_ * 8 / 1000000.0;
489 return megabits / seconds;
490 }
491
492 // Mbits per second
493 double audio_bandwidth() const {
494 double seconds = (kFrameTimerMs * frames_sent_ / 1000.0);
495 double megabits = audio_bytes_encoded_ * 8 / 1000000.0;
496 return megabits / seconds;
497 }
498
499 double desired_video_bitrate() {
500 return std::min<double>(available_bitrate_,
501 video_sender_config_.max_bitrate / 1000000.0);
502 }
503
504 bool SimpleGood() {
505 return frames_lost() <= 1 && late_frames() <= 1 &&
506 video_bandwidth() > desired_video_bitrate() * 0.8 &&
507 video_bandwidth() < desired_video_bitrate() * 1.2;
508 }
509
510 private:
511 FrameReceiverConfig audio_receiver_config_;
512 FrameReceiverConfig video_receiver_config_;
513 AudioSenderConfig audio_sender_config_;
514 VideoSenderConfig video_sender_config_;
515
516 base::TimeTicks start_time_;
517
518 // These run in "test time"
519 base::SimpleTestTickClock testing_clock_;
520 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_;
521
522 // These run on the sender timeline.
523 test::SkewedTickClock* testing_clock_sender_;
524 scoped_refptr<test::SkewedSingleThreadTaskRunner> task_runner_sender_;
525
526 // These run on the receiver timeline.
527 test::SkewedTickClock* testing_clock_receiver_;
528 scoped_refptr<test::SkewedSingleThreadTaskRunner> task_runner_receiver_;
529
530 scoped_refptr<CastEnvironment> cast_environment_sender_;
531 scoped_refptr<CastEnvironment> cast_environment_receiver_;
532
533 LoopBackTransport receiver_to_sender_;
534 LoopBackTransport sender_to_receiver_;
535 CastTransportSenderWrapper transport_sender_;
536 uint64 video_bytes_encoded_;
537 uint64 audio_bytes_encoded_;
538
539 scoped_ptr<CastReceiver> cast_receiver_;
540 scoped_ptr<CastSender> cast_sender_;
541
542 int frames_sent_;
543 double available_bitrate_;
544 std::vector<std::pair<base::TimeTicks, base::TimeTicks> > audio_ticks_;
545 std::vector<std::pair<base::TimeTicks, base::TimeTicks> > video_ticks_;
546 };
547
548 enum CacheResult { FOUND_TRUE, FOUND_FALSE, NOT_FOUND };
549
550 template <class T>
551 class BenchmarkCache {
552 public:
553 CacheResult Lookup(const T& x) {
554 base::AutoLock key(lock_);
555 for (size_t i = 0; i < results_.size(); i++) {
556 if (results_[i].second) {
557 if (x <= results_[i].first) {
558 VLOG(2) << "TRUE because: " << x.AsString()
559 << " <= " << results_[i].first.AsString();
560 return FOUND_TRUE;
561 }
562 } else {
563 if (x >= results_[i].first) {
564 VLOG(2) << "FALSE because: " << x.AsString()
565 << " >= " << results_[i].first.AsString();
566 return FOUND_FALSE;
567 }
568 }
569 }
570 return NOT_FOUND;
571 }
572
573 void Add(const T& x, bool result) {
574 base::AutoLock key(lock_);
575 VLOG(2) << "Cache Insert: " << x.AsString() << " = " << result;
576 results_.push_back(std::make_pair(x, result));
577 }
578
579 private:
580 base::Lock lock_;
581 std::vector<std::pair<T, bool> > results_;
582 };
583
584 struct SearchVariable {
585 SearchVariable() : base(0.0), grade(0.0) {}
586 SearchVariable(double b, double g) : base(b), grade(g) {}
587 SearchVariable blend(const SearchVariable& other, double factor) {
588 CHECK_GE(factor, 0);
589 CHECK_LE(factor, 1.0);
590 return SearchVariable(base * (1 - factor) + other.base * factor,
591 grade * (1 - factor) + other.grade * factor);
592 }
593 double value(double x) const { return base + grade * x; }
594 double base;
595 double grade;
596 };
597
598 struct SearchVector {
599 SearchVector blend(const SearchVector& other, double factor) {
600 SearchVector ret;
601 ret.bitrate = bitrate.blend(other.bitrate, factor);
602 ret.latency = latency.blend(other.latency, factor);
603 ret.packet_drop = packet_drop.blend(other.packet_drop, factor);
604 return ret;
605 }
606
607 SearchVector average(const SearchVector& other) {
608 return blend(other, 0.5);
609 }
610
611 MeasuringPoint GetMeasuringPoint(double v) const {
612 return MeasuringPoint(
613 bitrate.value(-v), latency.value(v), packet_drop.value(v));
614 }
615 std::string AsString(double v) { return GetMeasuringPoint(v).AsString(); }
616
617 SearchVariable bitrate;
618 SearchVariable latency;
619 SearchVariable packet_drop;
620 };
621
622 class CastBenchmark {
623 public:
624 bool RunOnePoint(const SearchVector& v, double multiplier) {
625 MeasuringPoint p = v.GetMeasuringPoint(multiplier);
626 VLOG(1) << "RUN: v = " << multiplier << " p = " << p.AsString();
627 if (p.bitrate <= 0) {
628 return false;
629 }
630 switch (cache_.Lookup(p)) {
631 case FOUND_TRUE:
632 return true;
633 case FOUND_FALSE:
634 return false;
635 case NOT_FOUND:
636 // Keep going
637 break;
638 }
639 bool result = true;
640 for (int tries = 0; tries < 3 && result; tries++) {
641 RunOneBenchmark benchmark;
642 benchmark.Run(p);
643 result &= benchmark.SimpleGood();
644 }
645 cache_.Add(p, result);
646 return result;
647 }
648
649 void BinarySearch(SearchVector v, double accuracy) {
650 double min = 0.0;
651 double max = 1.0;
652 while (RunOnePoint(v, max)) {
653 min = max;
654 max *= 2;
655 }
656
657 while (max - min > accuracy) {
658 double avg = (min + max) / 2;
659 if (RunOnePoint(v, avg)) {
660 min = avg;
661 } else {
662 max = avg;
663 }
664 }
665
666 // Print a data point to stdout.
667 base::AutoLock key(lock_);
668 MeasuringPoint p = v.GetMeasuringPoint(min);
669 fprintf(stdout, "%f %f %f\n", p.bitrate, p.latency, p.percent_packet_drop);
670 fflush(stdout);
671 }
672
673 void SpanningSearch(int max,
674 int x,
675 int y,
676 int skip,
677 SearchVector a,
678 SearchVector b,
679 SearchVector c,
680 double accuracy,
681 std::vector<linked_ptr<base::Thread> >* threads) {
682 static int thread_num = 0;
683 if (x > max) return;
684 if (skip > max) {
685 if (y > x) return;
686 SearchVector ab = a.blend(b, static_cast<double>(x) / max);
687 SearchVector ac = a.blend(c, static_cast<double>(x) / max);
688 SearchVector v = ab.blend(ac, x == y ? 1.0 : static_cast<double>(y) / x);
689 thread_num++;
690 (*threads)[thread_num % threads->size()]->message_loop()->PostTask(
691 FROM_HERE,
692 base::Bind(&CastBenchmark::BinarySearch,
693 base::Unretained(this),
694 v,
695 accuracy));
696 } else {
697 skip *= 2;
698 SpanningSearch(max, x, y, skip, a, b, c, accuracy, threads);
699 SpanningSearch(max, x + skip, y + skip, skip, a, b, c, accuracy, threads);
700 SpanningSearch(max, x + skip, y, skip, a, b, c, accuracy, threads);
701 SpanningSearch(max, x, y + skip, skip, a, b, c, accuracy, threads);
702 }
703 }
704
705 void Run() {
706 // Spanning search.
707
708 std::vector<linked_ptr<base::Thread> > threads;
709 for (int i = 0; i < 16; i++) {
710 threads.push_back(make_linked_ptr(new base::Thread(
711 base::StringPrintf("cast_bench_thread_%d", i))));
712 threads[i]->Start();
713 }
714
715 if (CommandLine::ForCurrentProcess()->HasSwitch("single-run")) {
716 SearchVector a;
717 a.bitrate.base = 100.0;
718 a.bitrate.grade = 1.0;
719 a.latency.grade = 1.0;
720 a.packet_drop.grade = 1.0;
721 threads[0]->message_loop()->PostTask(
722 FROM_HERE,
723 base::Bind(base::IgnoreResult(&CastBenchmark::RunOnePoint),
724 base::Unretained(this),
725 a,
726 1.0));
727 } else {
728 SearchVector a, b, c;
729 a.bitrate.base = b.bitrate.base = c.bitrate.base = 100.0;
730 a.bitrate.grade = 1.0;
731 b.latency.grade = 1.0;
732 c.packet_drop.grade = 1.0;
733
734 SpanningSearch(512,
735 0,
736 0,
737 1,
738 a,
739 b,
740 c,
741 0.01,
742 &threads);
743 }
744
745 for (size_t i = 0; i < threads.size(); i++) {
746 threads[i]->Stop();
747 }
748 }
749
750 private:
751 BenchmarkCache<MeasuringPoint> cache_;
752 base::Lock lock_;
753 };
754
755 } // namespace cast
756 } // namespace media
757
758 int main(int argc, char** argv) {
759 base::AtExitManager at_exit;
760 CommandLine::Init(argc, argv);
761 media::cast::CastBenchmark benchmark;
762 if (getenv("PROFILE_FILE")) {
763 std::string profile_file(getenv("PROFILE_FILE"));
764 base::debug::StartProfiling(profile_file);
765 benchmark.Run();
766 base::debug::StopProfiling();
767 } else {
768 benchmark.Run();
769 }
770 }
OLDNEW
« no previous file with comments | « media/cast/receiver/video_decoder.cc ('k') | media/cast/test/fake_single_thread_task_runner.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698