Add Origin Power Map to Store Battery Auditing Data.

components/power/origin_power_map.cc

+// Copyright 2014 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 "components/power/origin_power_map.h"
+
+#include "base/logging.h"
+#include "url/gurl.h"
+
+// Hardcode to avoid dependency on //content.
+const char kChromeUIScheme[] = "chrome";
+
+namespace power {
+
+OriginPowerMap::OriginPowerMap() : total_consumed_() {
+}
+
+OriginPowerMap::~OriginPowerMap() {
+}
+
+int OriginPowerMap::GetPowerForOrigin(const GURL& url) {
+  GURL origin = url.GetOrigin();
+  if (origin_map_.find(origin) == origin_map_.end() || !total_consumed_)
+    return 0;
+  return origin_map_[origin] * 100 / total_consumed_;
+}
+
+void OriginPowerMap::AddPowerForOrigin(const GURL& url, double power) {
+  DCHECK_GE(power, 0);
+  GURL origin = url.GetOrigin();
+  if (!origin.is_valid() || origin.SchemeIs(kChromeUIScheme))
+    return;
+
+  origin_map_[origin] += power;
+  total_consumed_ += power;
+}
+
+scoped_ptr<OriginPowerMap::PercentOriginMap>
+OriginPowerMap::GetPercentOriginMap() {
+  scoped_ptr<OriginPowerMap::PercentOriginMap> percent_map(
+      new OriginPowerMap::PercentOriginMap);
+  for (OriginMap::iterator it = origin_map_.begin(); it != origin_map_.end();
+       ++it) {
+    (*percent_map)[it->first] = (it->second / total_consumed_ * 100);
+  }
+  return percent_map.Pass();
+}
+
+}  // namespace power