blob: c8753fa3864341bcf6b630aab5f71a481655c510 [file] [log] [blame]
#! /usr/bin/env python
# Copyright 2016 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.
"""A script to manipulate device CPU frequency."""
from __future__ import print_function
import argparse
import os
import pprint
import sys
if __name__ == '__main__':
sys.path.append(
os.path.abspath(
os.path.join(os.path.dirname(__file__), '..', '..', '..')))
from devil.android import device_utils
from devil.android.perf import perf_control
from devil.android.tools import script_common
from devil.utils import logging_common
def SetScalingGovernor(device, args):
p = perf_control.PerfControl(device)
p.SetScalingGovernor(args.governor)
def GetScalingGovernor(device, _args):
p = perf_control.PerfControl(device)
for cpu, governor in p.GetScalingGovernor():
print('%s %s: %s' % (str(device), cpu, governor))
def ListAvailableGovernors(device, _args):
p = perf_control.PerfControl(device)
for cpu, governors in p.ListAvailableGovernors():
print('%s %s: %s' % (str(device), cpu, pprint.pformat(governors)))
def main(raw_args):
parser = argparse.ArgumentParser()
logging_common.AddLoggingArguments(parser)
script_common.AddEnvironmentArguments(parser)
parser.add_argument(
'--device',
dest='devices',
action='append',
default=[],
help='Devices for which the governor should be set. Defaults to all.')
subparsers = parser.add_subparsers()
set_governor = subparsers.add_parser('set-governor')
set_governor.add_argument('governor', help='Desired CPU governor.')
set_governor.set_defaults(func=SetScalingGovernor)
get_governor = subparsers.add_parser('get-governor')
get_governor.set_defaults(func=GetScalingGovernor)
list_governors = subparsers.add_parser('list-governors')
list_governors.set_defaults(func=ListAvailableGovernors)
args = parser.parse_args(raw_args)
logging_common.InitializeLogging(args)
script_common.InitializeEnvironment(args)
devices = device_utils.DeviceUtils.HealthyDevices(device_arg=args.devices)
parallel_devices = device_utils.DeviceUtils.parallel(devices)
parallel_devices.pMap(args.func, args)
return 0
if __name__ == '__main__':
sys.exit(main(sys.argv[1:]))