Index: components/power/origin_power_map.cc |
diff --git a/components/power/origin_power_map.cc b/components/power/origin_power_map.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..e356173373a284a7dbf7159e6533fa48d58c4bb3 |
--- /dev/null |
+++ b/components/power/origin_power_map.cc |
@@ -0,0 +1,49 @@ |
+// 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 |