Add Origin Power Map to Store Battery Auditing Data.

chrome/browser/power/origin_power_map.cc

Index: chrome/browser/power/origin_power_map.cc
diff --git a/chrome/browser/power/origin_power_map.cc b/chrome/browser/power/origin_power_map.cc
new file mode 100644
index 0000000000000000000000000000000000000000..ca65807dbf81cbf15962577f706e70a37dfaec4c
--- /dev/null
+++ b/chrome/browser/power/origin_power_map.cc
@@ -0,0 +1,40 @@
+// 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 "chrome/browser/power/origin_power_map.h"
+
+#include "base/logging.h"
+#include "content/public/common/url_constants.h"
+#include "url/gurl.h"
+
+OriginPowerMap::OriginPowerMap() {
+}
+
+OriginPowerMap::~OriginPowerMap() {
+}
+
+int OriginPowerMap::GetPowerForOrigin(const GURL& origin) {
+  if (origin_map_.find(origin) == origin_map_.end() || !total_consumed_)
+    return 0;
+  return origin_map_[origin] * 100 / total_consumed_;
+}
+
+void OriginPowerMap::AddPowerForOrigin(const GURL& origin, double power) {
+  if (!origin.is_valid() || origin.SchemeIs(content::kChromeUIScheme))
+    return;
+
+  origin_map_[origin] += power;

[sky, 2014/08/08 23:15:28]

I was actually saying something differently. I think you should do
origin.GetOrigin() here and let people submit whatever URLs they want. Same
thing in the GetPowerForOrigin() above.

[Daniel Nishi, 2014/08/08 23:38:07]

Oh, I see. I misunderstood.

I've allowed arbitrary URLs to be passed in and the origin is now used.

+  total_consumed_ += power;
+}
+
+scoped_ptr<const OriginPowerMap::PercentOriginMap>
+OriginPowerMap::GetPercentOriginMap() {
+  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 scoped_ptr<const OriginPowerMap::PercentOriginMap>(percent_map);
+}