summaryrefslogtreecommitdiff
path: root/zephyr/zmake/zmake/project.py
blob: f8079ce423bbb3f3b4e3161194e23b2ae5bb764e (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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
# Copyright 2020 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Module for project config wrapper object."""

import warnings
import yaml

# The version of jsonschema in the chroot has a bunch of
# DeprecationWarnings that fire when we import it.  Suppress these
# during the import to keep the noise down.
with warnings.catch_warnings():
    warnings.simplefilter('ignore')
    import jsonschema

import zmake.build_config as build_config
import zmake.modules as modules
import zmake.output_packers as packers
import zmake.util as util


def module_dts_overlay_name(modpath, board_name):
    """Given a board name, return the expected DTS overlay path.

    Args:
        modpath: the module path as a pathlib.Path object
        board_name: the name of the board

    Returns:
        A pathlib.Path object to the expected overlay path.
    """
    return modpath / 'zephyr' / 'dts' / 'board-overlays' / '{}.dts'.format(
        board_name)


class ProjectConfig:
    """An object wrapping zmake.yaml."""
    validator = jsonschema.Draft7Validator
    schema = {
        'type': 'object',
        'required': ['supported-zephyr-versions', 'board', 'output-type',
                     'toolchain'],
        'properties': {
            'supported-zephyr-versions': {
                'type': 'array',
                'items': {
                    'type': 'string',
                    'enum': ['v2.4', 'v2.5'],
                },
                'minItems': 1,
                'uniqueItems': True,
            },
            'board': {
                'type': 'string',
            },
            'modules': {
                'type': 'array',
                'items': {
                    'type': 'string',
                    'enum': list(modules.known_modules),
                },
            },
            'output-type': {
                'type': 'string',
                'enum': list(packers.packer_registry),
            },
            'toolchain': {
                'type': 'string',
            },
            'is-test': {
                'type': 'boolean',
            },
            'dts-overlays': {
                'type': 'array',
                'items': {
                    'type': 'string',
                },
            },
        },
    }

    def __init__(self, config_dict):
        self.validator.check_schema(self.schema)
        jsonschema.validate(config_dict, self.schema, cls=self.validator)
        self.config_dict = config_dict

    @property
    def supported_zephyr_versions(self):
        return [util.parse_zephyr_version(x)
                for x in self.config_dict['supported-zephyr-versions']]

    @property
    def board(self):
        return self.config_dict['board']

    @property
    def modules(self):
        return self.config_dict.get('modules', list(modules.known_modules))

    @property
    def output_packer(self):
        return packers.packer_registry[self.config_dict['output-type']]

    @property
    def toolchain(self):
        return self.config_dict['toolchain']

    @property
    def is_test(self):
        return self.config_dict.get('is-test', False)

    @property
    def dts_overlays(self):
        return self.config_dict.get('dts-overlays', [])


class Project:
    """An object encapsulating a project directory."""
    def __init__(self, project_dir, config_dict=None):
        self.project_dir = project_dir.resolve()
        if not config_dict:
            with open(self.project_dir / 'zmake.yaml') as f:
                config_dict = yaml.safe_load(f)
        self.config = ProjectConfig(config_dict)
        self.packer = self.config.output_packer(self)

    def iter_builds(self):
        """Iterate thru the build combinations provided by the project's packer.

        Yields:
            2-tuples of a build configuration name and a BuildConfig.
        """
        conf = build_config.BuildConfig(cmake_defs={'BOARD': self.config.board})
        if (self.project_dir / 'boards').is_dir():
            conf |= build_config.BuildConfig(
                cmake_defs={'BOARD_ROOT': str(self.project_dir)})
        prj_conf = self.project_dir / 'prj.conf'
        if prj_conf.is_file():
            conf |= build_config.BuildConfig(kconfig_files=[prj_conf])
        for build_name, packer_config in self.packer.configs():
            yield build_name, conf | packer_config

    def find_dts_overlays(self, modules):
        """Find appropriate dts overlays from registered modules.

        Args:
            modules: A dictionary of module names mapping to paths.

        Returns:
            A BuildConfig with relevant configurations to enable the
            found DTS overlay files.
        """
        overlays = []
        for module_path in modules.values():
            dts_path = module_dts_overlay_name(module_path, self.config.board)
            if dts_path.is_file():
                overlays.append(dts_path.resolve())

        overlays.extend(self.project_dir / f for f in self.config.dts_overlays)

        if overlays:
            return build_config.BuildConfig(
                cmake_defs={'DTC_OVERLAY_FILE': ';'.join(map(str, overlays))})
        else:
            return build_config.BuildConfig()

    def prune_modules(self, module_paths):
        """Reduce a modules dict to the ones required by this project.

        If this project does not define a modules list in the
        configuration, it is assumed that all known modules to Zmake
        are required.  This is typically inconsequential as Zephyr
        module design conventions require a Kconfig option to actually
        enable most modules.

        Args:
            module_paths: A dictionary mapping module names to their
                paths.  This dictionary is not modified.

        Returns:
            A new module_paths dictionary with only the modules
            required by this project.

        Raises:
            A KeyError, if a required module is unavailable.
        """
        result = {}
        for module in self.config.modules:
            try:
                result[module] = module_paths[module]
            except KeyError as e:
                raise KeyError(
                    'The {!r} module is required by the {} project, but is not '
                    'available.'.format(module, self.project_dir)) from e
        return result