mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-01-19 04:21:17 +00:00
11323b6256
TM has been updated and restructured to be clearer, including splitting off assistant and creating a runtime parameter manager into their own files.
84 lines
3.3 KiB
Python
84 lines
3.3 KiB
Python
import re
|
|
from collections import namedtuple
|
|
|
|
from wa.framework.exception import ConfigError
|
|
from wa.framework.target.runtime_config import (SysfileValuesRuntimeConfig,
|
|
HotplugRuntimeConfig,
|
|
CpufreqRuntimeConfig,
|
|
CpuidleRuntimeConfig)
|
|
from wa.utils.types import obj_dict
|
|
|
|
|
|
class RuntimeParameterManager(object):
|
|
|
|
runtime_config_cls = [
|
|
# order matters
|
|
SysfileValuesRuntimeConfig,
|
|
HotplugRuntimeConfig,
|
|
CpufreqRuntimeConfig,
|
|
CpuidleRuntimeConfig,
|
|
]
|
|
|
|
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()
|
|
|
|
def get_config_for_name(self, name):
|
|
for rp_name, rp in self.runtime_params.iteritems():
|
|
if re.search(name, rp_name):
|
|
return rp.rt_config
|
|
return None
|
|
|
|
def get_cfg_point(self, name):
|
|
for rp_name, rp in self.runtime_params.iteritems():
|
|
if re.search(name, rp_name, re.IGNORECASE):
|
|
return rp.cfg_point
|
|
raise ConfigError('Unknown Runtime Parameter: {}'.format(name))
|