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;
}

scoped_ptr<OriginPowerMap::Subscription>
OriginPowerMap::AddPowerConsumptionUpdatedCallback(
    const base::Closure& callback) {
  return callback_list_.Add(callback);
}

void OriginPowerMap::OnAllOriginsUpdated() {
  callback_list_.Notify();
}

void OriginPowerMap::ClearOriginMap() {
  origin_map_.clear();
  total_consumed_ = 0;
}

}  // namespace power