summaryrefslogtreecommitdiff
path: root/utils/tuning/libtuning/parsers/raspberrypi_parser.py
blob: f1da4592471082e84b59606d08c4208ef8f90486 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
# SPDX-License-Identifier: GPL-2.0-or-later
#
# Copyright (C) 2022, Paul Elder <paul.elder@ideasonboard.com>
#
# Parser for Raspberry Pi config file format

from .parser import Parser

import json
import numbers

import libtuning.utils as utils


class RaspberryPiParser(Parser):
    def __init__(self):
        super().__init__()

    # The string in the 'disable' and 'plot' lists are formatted as
    # 'rpi.{module_name}'.
    # @brief Enumerate, as a module, @a listt if its value exists in @a dictt
    #        and it is the name of a valid module in @a modules
    def _enumerate_rpi_modules(self, listt, dictt, modules):
        for x in listt:
            name = x.replace('rpi.', '')
            if name not in dictt:
                continue
            module = utils.get_module_by_typename(modules, name)
            if module is not None:
                yield module

    def _valid_macbeth_option(self, value):
        if not isinstance(value, dict):
            return False

        if list(value.keys()) != ['small', 'show']:
            return False

        for val in value.values():
            if not isinstance(val, numbers.Number):
                return False

        return True

    def parse(self, config_file: str, modules: list) -> (dict, list):
        with open(config_file, 'r') as config_json:
            config = json.load(config_json)

        disable = []
        for module in self._enumerate_rpi_modules(config['disable'], config, modules):
            disable.append(module)
            # Remove the disabled module's config too
            config.pop(module.name)
        config.pop('disable')

        # The raspberrypi config format has 'plot' map to a list of module
        # names which should be plotted. libtuning has each module contain the
        # plot information in itself so do this conversion.

        for module in self._enumerate_rpi_modules(config['plot'], config, modules):
            # It's fine to set the value of a potentially disabled module, as
            # the object still exists at this point
            module.appendValue('debug', 'plot')
        config.pop('plot')

        # Convert the keys from module name to module instance

        new_config = {}

        for module_name in config:
            module = utils.get_module_by_type_name(modules, module_name)
            if module is not None:
                new_config[module] = config[module_name]

        new_config['general'] = {}

        if 'blacklevel' in config:
            if not isinstance(config['blacklevel'], numbers.Number):
                raise TypeError('Config "blacklevel" must be a number')
            # Raspberry Pi's ctt config has magic blacklevel value -1 to mean
            # "get it from the image metadata". Since we already do that in
            # Image, don't save it to the config here.
            if config['blacklevel'] >= 0:
                new_config['general']['blacklevel'] = config['blacklevel']

        if 'macbeth' in config:
            if not self._valid_macbeth_option(config['macbeth']):
                raise TypeError('Config "macbeth" must be a dict: {"small": number, "show": number}')
            new_config['general']['macbeth'] = config['macbeth']
        else:
            new_config['general']['macbeth'] = {'small': 0, 'show': 0}

        return new_config, disable