mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-06-23 00:35:56 +01:00
TM: Update and Refactor
TM has been updated and restructured to be clearer, including splitting off assistant and creating a runtime parameter manager into their own files.
This commit is contained in:
wa
83
wa/framework/target/runtime_parameter_manager.py
Normal file
83
wa/framework/target/runtime_parameter_manager.py
Normal file
@ -0,0 +1,83 @@
|
||||
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))
|
Reference in New Issue
Block a user