1// Copyright 2013 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#include "cc/debug/unittest_only_benchmark.h"
6
7#include "base/bind.h"
8#include "base/message_loop/message_loop_proxy.h"
9#include "base/values.h"
10#include "cc/debug/unittest_only_benchmark_impl.h"
11
12namespace cc {
13
14UnittestOnlyBenchmark::UnittestOnlyBenchmark(scoped_ptr<base::Value> value,
15                                             const DoneCallback& callback)
16    : MicroBenchmark(callback),
17      create_impl_benchmark_(false),
18      weak_ptr_factory_(this) {
19  if (!value)
20    return;
21
22  base::DictionaryValue* settings = NULL;
23  value->GetAsDictionary(&settings);
24  if (!settings)
25    return;
26
27  if (settings->HasKey("run_benchmark_impl"))
28    settings->GetBoolean("run_benchmark_impl", &create_impl_benchmark_);
29}
30
31UnittestOnlyBenchmark::~UnittestOnlyBenchmark() {
32  weak_ptr_factory_.InvalidateWeakPtrs();
33}
34
35void UnittestOnlyBenchmark::DidUpdateLayers(LayerTreeHost* host) {
36  NotifyDone(scoped_ptr<base::Value>());
37}
38
39bool UnittestOnlyBenchmark::ProcessMessage(scoped_ptr<base::Value> value) {
40  base::DictionaryValue* message = NULL;
41  value->GetAsDictionary(&message);
42  bool can_handle;
43  if (message->HasKey("can_handle")) {
44    message->GetBoolean("can_handle", &can_handle);
45    if (can_handle)
46      return true;
47  }
48  return false;
49}
50
51void UnittestOnlyBenchmark::RecordImplResults(scoped_ptr<base::Value> results) {
52  NotifyDone(results.Pass());
53}
54
55scoped_ptr<MicroBenchmarkImpl> UnittestOnlyBenchmark::CreateBenchmarkImpl(
56    scoped_refptr<base::MessageLoopProxy> origin_loop) {
57  if (!create_impl_benchmark_)
58    return make_scoped_ptr<MicroBenchmarkImpl>(NULL);
59
60  return scoped_ptr<MicroBenchmarkImpl>(new UnittestOnlyBenchmarkImpl(
61      origin_loop,
62      NULL,
63      base::Bind(&UnittestOnlyBenchmark::RecordImplResults,
64                 weak_ptr_factory_.GetWeakPtr())));
65}
66
67}  // namespace cc
68