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 "content/public/common/url_constants.h"
+#include "url/gurl.h"
+
+namespace power {
+
+OriginPowerMap::OriginPowerMap() : total_consumed_(0.0) {
+}
+
+OriginPowerMap::~OriginPowerMap() {
+}
+
+int OriginPowerMap::GetPowerForOrigin(const GURL& url) {
+  if (!total_consumed_)
+    return 0;
+
+  OriginMap::const_iterator it = origin_map_.find(url.GetOrigin());
+  return it == origin_map_.end() ? 0 :
+      static_cast<int>(it->second * 100 / total_consumed_ + 0.5);
+}
+
+void OriginPowerMap::AddPowerForOrigin(const GURL& url, double power) {
+  DCHECK_GE(power, 0);
+  GURL origin = url.GetOrigin();
+  if (!origin.is_valid() || origin.SchemeIs(content::kChromeUIScheme))
+    return;
+
+  origin_map_[origin] += power;
+  total_consumed_ += power;
+}
+
+OriginPowerMap::PercentOriginMap OriginPowerMap::GetPercentOriginMap() {
+  OriginPowerMap::PercentOriginMap percent_map;
+
+  if (!total_consumed_)
+    return percent_map;
+
+  for (OriginMap::iterator it = origin_map_.begin(); it != origin_map_.end();
+       ++it) {
+    percent_map[it->first] =
+        static_cast<int>(it->second * 100 / total_consumed_ + 0.5);
+  }
+  return percent_map;
+}
+
+}  // namespace power