1 // Copyright 2014 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "components/power/origin_power_map.h" 6 7 #include "base/logging.h" 8 #include "content/public/common/url_constants.h" 9 #include "url/gurl.h" 10 11 namespace power { 12 13 OriginPowerMap::OriginPowerMap() : total_consumed_(0.0) { 14 } 15 16 OriginPowerMap::~OriginPowerMap() { 17 } 18 19 int OriginPowerMap::GetPowerForOrigin(const GURL& url) { 20 if (!total_consumed_) 21 return 0; 22 23 OriginMap::const_iterator it = origin_map_.find(url.GetOrigin()); 24 return it == origin_map_.end() ? 0 : 25 static_cast<int>(it->second * 100 / total_consumed_ + 0.5); 26 } 27 28 void OriginPowerMap::AddPowerForOrigin(const GURL& url, double power) { 29 DCHECK_GE(power, 0); 30 GURL origin = url.GetOrigin(); 31 if (!origin.is_valid() || origin.SchemeIs(content::kChromeUIScheme)) 32 return; 33 34 origin_map_[origin] += power; 35 total_consumed_ += power; 36 } 37 38 OriginPowerMap::PercentOriginMap OriginPowerMap::GetPercentOriginMap() { 39 OriginPowerMap::PercentOriginMap percent_map; 40 41 if (!total_consumed_) 42 return percent_map; 43 44 for (OriginMap::iterator it = origin_map_.begin(); it != origin_map_.end(); 45 ++it) { 46 percent_map[it->first] = 47 static_cast<int>(it->second * 100 / total_consumed_ + 0.5); 48 } 49 return percent_map; 50 } 51 52 scoped_ptr<OriginPowerMap::Subscription> 53 OriginPowerMap::AddPowerConsumptionUpdatedCallback( 54 const base::Closure& callback) { 55 return callback_list_.Add(callback); 56 } 57 58 void OriginPowerMap::OnAllOriginsUpdated() { 59 callback_list_.Notify(); 60 } 61 62 void OriginPowerMap::ClearOriginMap() { 63 origin_map_.clear(); 64 total_consumed_ = 0; 65 } 66 67 } // namespace power 68
