mirror of
https://github.com/ARM-software/workload-automation.git
synced 2024-10-06 10:51:13 +01:00
86 lines
3.4 KiB
Python
86 lines
3.4 KiB
Python
from collections import namedtuple
|
|
|
|
from wa.framework.exception import ConfigError
|
|
from wa.framework.target.runtime_config import (SysfileValuesRuntimeConfig,
|
|
HotplugRuntimeConfig,
|
|
CpufreqRuntimeConfig,
|
|
CpuidleRuntimeConfig,
|
|
AndroidRuntimeConfig)
|
|
from wa.utils.types import obj_dict, caseless_string
|
|
|
|
|
|
class RuntimeParameterManager(object):
|
|
|
|
runtime_config_cls = [
|
|
# order matters
|
|
SysfileValuesRuntimeConfig,
|
|
HotplugRuntimeConfig,
|
|
CpufreqRuntimeConfig,
|
|
CpuidleRuntimeConfig,
|
|
AndroidRuntimeConfig,
|
|
]
|
|
|
|
def __init__(self, target):
|
|
self.target = target
|
|
self.runtime_configs = [cls(self.target) for cls in self.runtime_config_cls]
|
|
self.runtime_params = {}
|
|
|
|
runtime_parameter = namedtuple('RuntimeParameter', 'cfg_point, rt_config')
|
|
for cfg in self.runtime_configs:
|
|
for param in cfg.supported_parameters:
|
|
if param.name in self.runtime_params:
|
|
msg = 'Duplicate runtime parameter name "{}": in both {} and {}'
|
|
raise RuntimeError(msg.format(param.name,
|
|
self.runtime_params[param.name].rt_config.name,
|
|
cfg.name))
|
|
self.runtime_params[param.name] = runtime_parameter(param, cfg)
|
|
|
|
# Uses corresponding config point to merge parameters
|
|
def merge_runtime_parameters(self, parameters):
|
|
merged_params = obj_dict()
|
|
for source in parameters:
|
|
for name, value in parameters[source].iteritems():
|
|
cp = self.get_cfg_point(name)
|
|
cp.set_value(merged_params, value)
|
|
return dict(merged_params)
|
|
|
|
# Validates runtime_parameters against each other
|
|
def validate_runtime_parameters(self, parameters):
|
|
self.clear_runtime_parameters()
|
|
self.set_runtime_parameters(parameters)
|
|
for cfg in self.runtime_configs:
|
|
cfg.validate_parameters()
|
|
|
|
# Writes the given parameters to the device.
|
|
def commit_runtime_parameters(self, parameters):
|
|
self.clear_runtime_parameters()
|
|
self.set_runtime_parameters(parameters)
|
|
for cfg in self.runtime_configs:
|
|
cfg.commit()
|
|
|
|
# Stores a set of parameters performing isolated validation when appropriate
|
|
def set_runtime_parameters(self, parameters):
|
|
for name, value in parameters.iteritems():
|
|
cfg = self.get_config_for_name(name)
|
|
if cfg is None:
|
|
msg = 'Unsupported runtime parameter: "{}"'
|
|
raise ConfigError(msg.format(name))
|
|
cfg.set_runtime_parameter(name, value)
|
|
|
|
def clear_runtime_parameters(self):
|
|
for cfg in self.runtime_configs:
|
|
cfg.clear()
|
|
cfg.set_defaults()
|
|
|
|
def get_config_for_name(self, name):
|
|
name = caseless_string(name)
|
|
if name in self.runtime_params:
|
|
return self.runtime_params[name].rt_config
|
|
return None
|
|
|
|
def get_cfg_point(self, name):
|
|
name = caseless_string(name)
|
|
if name in self.runtime_params:
|
|
return self.runtime_params[name].cfg_point
|
|
raise ConfigError('Unknown Runtime Parameter: {}'.format(name))
|