blob: efb46c63b902593ad8332264e0342c7b311f4bae [file] [log] [blame]
// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "cc/benchmarks/unittest_only_benchmark.h"
#include "base/bind.h"
#include "base/memory/ptr_util.h"
#include "base/single_thread_task_runner.h"
#include "base/values.h"
#include "cc/benchmarks/unittest_only_benchmark_impl.h"
namespace cc {
UnittestOnlyBenchmark::UnittestOnlyBenchmark(std::unique_ptr<base::Value> value,
DoneCallback callback)
: MicroBenchmark(std::move(callback)), create_impl_benchmark_(false) {
if (!value)
return;
base::DictionaryValue* settings = nullptr;
value->GetAsDictionary(&settings);
if (!settings)
return;
if (settings->HasKey("run_benchmark_impl"))
settings->GetBoolean("run_benchmark_impl", &create_impl_benchmark_);
}
UnittestOnlyBenchmark::~UnittestOnlyBenchmark() {
weak_ptr_factory_.InvalidateWeakPtrs();
}
void UnittestOnlyBenchmark::DidUpdateLayers(LayerTreeHost* layer_tree_host) {
NotifyDone(nullptr);
}
bool UnittestOnlyBenchmark::ProcessMessage(std::unique_ptr<base::Value> value) {
base::DictionaryValue* message = nullptr;
value->GetAsDictionary(&message);
bool can_handle;
if (message->HasKey("can_handle")) {
message->GetBoolean("can_handle", &can_handle);
if (can_handle)
return true;
}
return false;
}
void UnittestOnlyBenchmark::RecordImplResults(
std::unique_ptr<base::Value> results) {
NotifyDone(std::move(results));
}
std::unique_ptr<MicroBenchmarkImpl> UnittestOnlyBenchmark::CreateBenchmarkImpl(
scoped_refptr<base::SingleThreadTaskRunner> origin_task_runner) {
if (!create_impl_benchmark_)
return base::WrapUnique<MicroBenchmarkImpl>(nullptr);
return base::WrapUnique(new UnittestOnlyBenchmarkImpl(
origin_task_runner, nullptr,
base::BindOnce(&UnittestOnlyBenchmark::RecordImplResults,
weak_ptr_factory_.GetWeakPtr())));
}
} // namespace cc