xref: /openbmc/u-boot/tools/buildman/builder.py (revision 5971ab5c44cb0c32c88fcdc90a3a9b6430463c4c)
1fc3fe1c2SSimon Glass# Copyright (c) 2013 The Chromium OS Authors.
2fc3fe1c2SSimon Glass#
3fc3fe1c2SSimon Glass# Bloat-o-meter code used here Copyright 2004 Matt Mackall <mpm@selenic.com>
4fc3fe1c2SSimon Glass#
51a459660SWolfgang Denk# SPDX-License-Identifier:	GPL-2.0+
6fc3fe1c2SSimon Glass#
7fc3fe1c2SSimon Glass
8fc3fe1c2SSimon Glassimport collections
9fc3fe1c2SSimon Glassfrom datetime import datetime, timedelta
10fc3fe1c2SSimon Glassimport glob
11fc3fe1c2SSimon Glassimport os
12fc3fe1c2SSimon Glassimport re
13fc3fe1c2SSimon Glassimport Queue
14fc3fe1c2SSimon Glassimport shutil
15fc3fe1c2SSimon Glassimport string
16fc3fe1c2SSimon Glassimport sys
17fc3fe1c2SSimon Glassimport time
18fc3fe1c2SSimon Glass
19190064b4SSimon Glassimport builderthread
20fc3fe1c2SSimon Glassimport command
21fc3fe1c2SSimon Glassimport gitutil
22fc3fe1c2SSimon Glassimport terminal
234653a882SSimon Glassfrom terminal import Print
24fc3fe1c2SSimon Glassimport toolchain
25fc3fe1c2SSimon Glass
26fc3fe1c2SSimon Glass
27fc3fe1c2SSimon Glass"""
28fc3fe1c2SSimon GlassTheory of Operation
29fc3fe1c2SSimon Glass
30fc3fe1c2SSimon GlassPlease see README for user documentation, and you should be familiar with
31fc3fe1c2SSimon Glassthat before trying to make sense of this.
32fc3fe1c2SSimon Glass
33fc3fe1c2SSimon GlassBuildman works by keeping the machine as busy as possible, building different
34fc3fe1c2SSimon Glasscommits for different boards on multiple CPUs at once.
35fc3fe1c2SSimon Glass
36fc3fe1c2SSimon GlassThe source repo (self.git_dir) contains all the commits to be built. Each
37fc3fe1c2SSimon Glassthread works on a single board at a time. It checks out the first commit,
38fc3fe1c2SSimon Glassconfigures it for that board, then builds it. Then it checks out the next
39fc3fe1c2SSimon Glasscommit and builds it (typically without re-configuring). When it runs out
40fc3fe1c2SSimon Glassof commits, it gets another job from the builder and starts again with that
41fc3fe1c2SSimon Glassboard.
42fc3fe1c2SSimon Glass
43fc3fe1c2SSimon GlassClearly the builder threads could work either way - they could check out a
44fc3fe1c2SSimon Glasscommit and then built it for all boards. Using separate directories for each
45fc3fe1c2SSimon Glasscommit/board pair they could leave their build product around afterwards
46fc3fe1c2SSimon Glassalso.
47fc3fe1c2SSimon Glass
48fc3fe1c2SSimon GlassThe intent behind building a single board for multiple commits, is to make
49fc3fe1c2SSimon Glassuse of incremental builds. Since each commit is built incrementally from
50fc3fe1c2SSimon Glassthe previous one, builds are faster. Reconfiguring for a different board
51fc3fe1c2SSimon Glassremoves all intermediate object files.
52fc3fe1c2SSimon Glass
53fc3fe1c2SSimon GlassMany threads can be working at once, but each has its own working directory.
54fc3fe1c2SSimon GlassWhen a thread finishes a build, it puts the output files into a result
55fc3fe1c2SSimon Glassdirectory.
56fc3fe1c2SSimon Glass
57fc3fe1c2SSimon GlassThe base directory used by buildman is normally '../<branch>', i.e.
58fc3fe1c2SSimon Glassa directory higher than the source repository and named after the branch
59fc3fe1c2SSimon Glassbeing built.
60fc3fe1c2SSimon Glass
61fc3fe1c2SSimon GlassWithin the base directory, we have one subdirectory for each commit. Within
62fc3fe1c2SSimon Glassthat is one subdirectory for each board. Within that is the build output for
63fc3fe1c2SSimon Glassthat commit/board combination.
64fc3fe1c2SSimon Glass
65fc3fe1c2SSimon GlassBuildman also create working directories for each thread, in a .bm-work/
66fc3fe1c2SSimon Glasssubdirectory in the base dir.
67fc3fe1c2SSimon Glass
68fc3fe1c2SSimon GlassAs an example, say we are building branch 'us-net' for boards 'sandbox' and
69fc3fe1c2SSimon Glass'seaboard', and say that us-net has two commits. We will have directories
70fc3fe1c2SSimon Glasslike this:
71fc3fe1c2SSimon Glass
72fc3fe1c2SSimon Glassus-net/             base directory
73fc3fe1c2SSimon Glass    01_of_02_g4ed4ebc_net--Add-tftp-speed-/
74fc3fe1c2SSimon Glass        sandbox/
75fc3fe1c2SSimon Glass            u-boot.bin
76fc3fe1c2SSimon Glass        seaboard/
77fc3fe1c2SSimon Glass            u-boot.bin
78fc3fe1c2SSimon Glass    02_of_02_g4ed4ebc_net--Check-tftp-comp/
79fc3fe1c2SSimon Glass        sandbox/
80fc3fe1c2SSimon Glass            u-boot.bin
81fc3fe1c2SSimon Glass        seaboard/
82fc3fe1c2SSimon Glass            u-boot.bin
83fc3fe1c2SSimon Glass    .bm-work/
84fc3fe1c2SSimon Glass        00/         working directory for thread 0 (contains source checkout)
85fc3fe1c2SSimon Glass            build/  build output
86fc3fe1c2SSimon Glass        01/         working directory for thread 1
87fc3fe1c2SSimon Glass            build/  build output
88fc3fe1c2SSimon Glass        ...
89fc3fe1c2SSimon Glassu-boot/             source directory
90fc3fe1c2SSimon Glass    .git/           repository
91fc3fe1c2SSimon Glass"""
92fc3fe1c2SSimon Glass
93fc3fe1c2SSimon Glass# Possible build outcomes
94fc3fe1c2SSimon GlassOUTCOME_OK, OUTCOME_WARNING, OUTCOME_ERROR, OUTCOME_UNKNOWN = range(4)
95fc3fe1c2SSimon Glass
96fc3fe1c2SSimon Glass# Translate a commit subject into a valid filename
97fc3fe1c2SSimon Glasstrans_valid_chars = string.maketrans("/: ", "---")
98fc3fe1c2SSimon Glass
99fc3fe1c2SSimon Glass
100fc3fe1c2SSimon Glassclass Builder:
101fc3fe1c2SSimon Glass    """Class for building U-Boot for a particular commit.
102fc3fe1c2SSimon Glass
103fc3fe1c2SSimon Glass    Public members: (many should ->private)
104fc3fe1c2SSimon Glass        active: True if the builder is active and has not been stopped
105fc3fe1c2SSimon Glass        already_done: Number of builds already completed
106fc3fe1c2SSimon Glass        base_dir: Base directory to use for builder
107fc3fe1c2SSimon Glass        checkout: True to check out source, False to skip that step.
108fc3fe1c2SSimon Glass            This is used for testing.
109fc3fe1c2SSimon Glass        col: terminal.Color() object
110fc3fe1c2SSimon Glass        count: Number of commits to build
111fc3fe1c2SSimon Glass        do_make: Method to call to invoke Make
112fc3fe1c2SSimon Glass        fail: Number of builds that failed due to error
113fc3fe1c2SSimon Glass        force_build: Force building even if a build already exists
114fc3fe1c2SSimon Glass        force_config_on_failure: If a commit fails for a board, disable
115fc3fe1c2SSimon Glass            incremental building for the next commit we build for that
116fc3fe1c2SSimon Glass            board, so that we will see all warnings/errors again.
1174266dc28SSimon Glass        force_build_failures: If a previously-built build (i.e. built on
1184266dc28SSimon Glass            a previous run of buildman) is marked as failed, rebuild it.
119fc3fe1c2SSimon Glass        git_dir: Git directory containing source repository
120fc3fe1c2SSimon Glass        last_line_len: Length of the last line we printed (used for erasing
121fc3fe1c2SSimon Glass            it with new progress information)
122fc3fe1c2SSimon Glass        num_jobs: Number of jobs to run at once (passed to make as -j)
123fc3fe1c2SSimon Glass        num_threads: Number of builder threads to run
124fc3fe1c2SSimon Glass        out_queue: Queue of results to process
125fc3fe1c2SSimon Glass        re_make_err: Compiled regular expression for ignore_lines
126fc3fe1c2SSimon Glass        queue: Queue of jobs to run
127fc3fe1c2SSimon Glass        threads: List of active threads
128fc3fe1c2SSimon Glass        toolchains: Toolchains object to use for building
129fc3fe1c2SSimon Glass        upto: Current commit number we are building (0.count-1)
130fc3fe1c2SSimon Glass        warned: Number of builds that produced at least one warning
13197e91526SSimon Glass        force_reconfig: Reconfigure U-Boot on each comiit. This disables
13297e91526SSimon Glass            incremental building, where buildman reconfigures on the first
13397e91526SSimon Glass            commit for a baord, and then just does an incremental build for
13497e91526SSimon Glass            the following commits. In fact buildman will reconfigure and
13597e91526SSimon Glass            retry for any failing commits, so generally the only effect of
13697e91526SSimon Glass            this option is to slow things down.
137189a4968SSimon Glass        in_tree: Build U-Boot in-tree instead of specifying an output
138189a4968SSimon Glass            directory separate from the source code. This option is really
139189a4968SSimon Glass            only useful for testing in-tree builds.
140fc3fe1c2SSimon Glass
141fc3fe1c2SSimon Glass    Private members:
142fc3fe1c2SSimon Glass        _base_board_dict: Last-summarised Dict of boards
143fc3fe1c2SSimon Glass        _base_err_lines: Last-summarised list of errors
144e30965dbSSimon Glass        _base_warn_lines: Last-summarised list of warnings
145fc3fe1c2SSimon Glass        _build_period_us: Time taken for a single build (float object).
146fc3fe1c2SSimon Glass        _complete_delay: Expected delay until completion (timedelta)
147fc3fe1c2SSimon Glass        _next_delay_update: Next time we plan to display a progress update
148fc3fe1c2SSimon Glass                (datatime)
149fc3fe1c2SSimon Glass        _show_unknown: Show unknown boards (those not built) in summary
150fc3fe1c2SSimon Glass        _timestamps: List of timestamps for the completion of the last
151fc3fe1c2SSimon Glass            last _timestamp_count builds. Each is a datetime object.
152fc3fe1c2SSimon Glass        _timestamp_count: Number of timestamps to keep in our list.
153fc3fe1c2SSimon Glass        _working_dir: Base working directory containing all threads
154fc3fe1c2SSimon Glass    """
155fc3fe1c2SSimon Glass    class Outcome:
156fc3fe1c2SSimon Glass        """Records a build outcome for a single make invocation
157fc3fe1c2SSimon Glass
158fc3fe1c2SSimon Glass        Public Members:
159fc3fe1c2SSimon Glass            rc: Outcome value (OUTCOME_...)
160fc3fe1c2SSimon Glass            err_lines: List of error lines or [] if none
161fc3fe1c2SSimon Glass            sizes: Dictionary of image size information, keyed by filename
162fc3fe1c2SSimon Glass                - Each value is itself a dictionary containing
163fc3fe1c2SSimon Glass                    values for 'text', 'data' and 'bss', being the integer
164fc3fe1c2SSimon Glass                    size in bytes of each section.
165fc3fe1c2SSimon Glass            func_sizes: Dictionary keyed by filename - e.g. 'u-boot'. Each
166fc3fe1c2SSimon Glass                    value is itself a dictionary:
167fc3fe1c2SSimon Glass                        key: function name
168fc3fe1c2SSimon Glass                        value: Size of function in bytes
169fc3fe1c2SSimon Glass        """
170fc3fe1c2SSimon Glass        def __init__(self, rc, err_lines, sizes, func_sizes):
171fc3fe1c2SSimon Glass            self.rc = rc
172fc3fe1c2SSimon Glass            self.err_lines = err_lines
173fc3fe1c2SSimon Glass            self.sizes = sizes
174fc3fe1c2SSimon Glass            self.func_sizes = func_sizes
175fc3fe1c2SSimon Glass
176fc3fe1c2SSimon Glass    def __init__(self, toolchains, base_dir, git_dir, num_threads, num_jobs,
177*5971ab5cSSimon Glass                 gnu_make='make', checkout=True, show_unknown=True, step=1,
178*5971ab5cSSimon Glass                 no_subdirs=False):
179fc3fe1c2SSimon Glass        """Create a new Builder object
180fc3fe1c2SSimon Glass
181fc3fe1c2SSimon Glass        Args:
182fc3fe1c2SSimon Glass            toolchains: Toolchains object to use for building
183fc3fe1c2SSimon Glass            base_dir: Base directory to use for builder
184fc3fe1c2SSimon Glass            git_dir: Git directory containing source repository
185fc3fe1c2SSimon Glass            num_threads: Number of builder threads to run
186fc3fe1c2SSimon Glass            num_jobs: Number of jobs to run at once (passed to make as -j)
18799796923SMasahiro Yamada            gnu_make: the command name of GNU Make.
188fc3fe1c2SSimon Glass            checkout: True to check out source, False to skip that step.
189fc3fe1c2SSimon Glass                This is used for testing.
190fc3fe1c2SSimon Glass            show_unknown: Show unknown boards (those not built) in summary
191fc3fe1c2SSimon Glass            step: 1 to process every commit, n to process every nth commit
192fc3fe1c2SSimon Glass        """
193fc3fe1c2SSimon Glass        self.toolchains = toolchains
194fc3fe1c2SSimon Glass        self.base_dir = base_dir
195fc3fe1c2SSimon Glass        self._working_dir = os.path.join(base_dir, '.bm-work')
196fc3fe1c2SSimon Glass        self.threads = []
197fc3fe1c2SSimon Glass        self.active = True
198fc3fe1c2SSimon Glass        self.do_make = self.Make
19999796923SMasahiro Yamada        self.gnu_make = gnu_make
200fc3fe1c2SSimon Glass        self.checkout = checkout
201fc3fe1c2SSimon Glass        self.num_threads = num_threads
202fc3fe1c2SSimon Glass        self.num_jobs = num_jobs
203fc3fe1c2SSimon Glass        self.already_done = 0
204fc3fe1c2SSimon Glass        self.force_build = False
205fc3fe1c2SSimon Glass        self.git_dir = git_dir
206fc3fe1c2SSimon Glass        self._show_unknown = show_unknown
207fc3fe1c2SSimon Glass        self._timestamp_count = 10
208fc3fe1c2SSimon Glass        self._build_period_us = None
209fc3fe1c2SSimon Glass        self._complete_delay = None
210fc3fe1c2SSimon Glass        self._next_delay_update = datetime.now()
211fc3fe1c2SSimon Glass        self.force_config_on_failure = True
2124266dc28SSimon Glass        self.force_build_failures = False
21397e91526SSimon Glass        self.force_reconfig = False
214fc3fe1c2SSimon Glass        self._step = step
215189a4968SSimon Glass        self.in_tree = False
21628370c1bSSimon Glass        self._error_lines = 0
217*5971ab5cSSimon Glass        self.no_subdirs = no_subdirs
218fc3fe1c2SSimon Glass
219fc3fe1c2SSimon Glass        self.col = terminal.Color()
220fc3fe1c2SSimon Glass
221e30965dbSSimon Glass        self._re_function = re.compile('(.*): In function.*')
222e30965dbSSimon Glass        self._re_files = re.compile('In file included from.*')
223e30965dbSSimon Glass        self._re_warning = re.compile('(.*):(\d*):(\d*): warning: .*')
224e30965dbSSimon Glass        self._re_note = re.compile('(.*):(\d*):(\d*): note: this is the location of the previous.*')
225e30965dbSSimon Glass
226fc3fe1c2SSimon Glass        self.queue = Queue.Queue()
227fc3fe1c2SSimon Glass        self.out_queue = Queue.Queue()
228fc3fe1c2SSimon Glass        for i in range(self.num_threads):
229190064b4SSimon Glass            t = builderthread.BuilderThread(self, i)
230fc3fe1c2SSimon Glass            t.setDaemon(True)
231fc3fe1c2SSimon Glass            t.start()
232fc3fe1c2SSimon Glass            self.threads.append(t)
233fc3fe1c2SSimon Glass
234fc3fe1c2SSimon Glass        self.last_line_len = 0
235190064b4SSimon Glass        t = builderthread.ResultThread(self)
236fc3fe1c2SSimon Glass        t.setDaemon(True)
237fc3fe1c2SSimon Glass        t.start()
238fc3fe1c2SSimon Glass        self.threads.append(t)
239fc3fe1c2SSimon Glass
240fc3fe1c2SSimon Glass        ignore_lines = ['(make.*Waiting for unfinished)', '(Segmentation fault)']
241fc3fe1c2SSimon Glass        self.re_make_err = re.compile('|'.join(ignore_lines))
242fc3fe1c2SSimon Glass
243fc3fe1c2SSimon Glass    def __del__(self):
244fc3fe1c2SSimon Glass        """Get rid of all threads created by the builder"""
245fc3fe1c2SSimon Glass        for t in self.threads:
246fc3fe1c2SSimon Glass            del t
247fc3fe1c2SSimon Glass
248b2ea7ab2SSimon Glass    def SetDisplayOptions(self, show_errors=False, show_sizes=False,
249ed966657SSimon Glass                          show_detail=False, show_bloat=False,
250ed966657SSimon Glass                          list_error_boards=False):
251b2ea7ab2SSimon Glass        """Setup display options for the builder.
252b2ea7ab2SSimon Glass
253b2ea7ab2SSimon Glass        show_errors: True to show summarised error/warning info
254b2ea7ab2SSimon Glass        show_sizes: Show size deltas
255b2ea7ab2SSimon Glass        show_detail: Show detail for each board
256b2ea7ab2SSimon Glass        show_bloat: Show detail for each function
257ed966657SSimon Glass        list_error_boards: Show the boards which caused each error/warning
258b2ea7ab2SSimon Glass        """
259b2ea7ab2SSimon Glass        self._show_errors = show_errors
260b2ea7ab2SSimon Glass        self._show_sizes = show_sizes
261b2ea7ab2SSimon Glass        self._show_detail = show_detail
262b2ea7ab2SSimon Glass        self._show_bloat = show_bloat
263ed966657SSimon Glass        self._list_error_boards = list_error_boards
264b2ea7ab2SSimon Glass
265fc3fe1c2SSimon Glass    def _AddTimestamp(self):
266fc3fe1c2SSimon Glass        """Add a new timestamp to the list and record the build period.
267fc3fe1c2SSimon Glass
268fc3fe1c2SSimon Glass        The build period is the length of time taken to perform a single
269fc3fe1c2SSimon Glass        build (one board, one commit).
270fc3fe1c2SSimon Glass        """
271fc3fe1c2SSimon Glass        now = datetime.now()
272fc3fe1c2SSimon Glass        self._timestamps.append(now)
273fc3fe1c2SSimon Glass        count = len(self._timestamps)
274fc3fe1c2SSimon Glass        delta = self._timestamps[-1] - self._timestamps[0]
275fc3fe1c2SSimon Glass        seconds = delta.total_seconds()
276fc3fe1c2SSimon Glass
277fc3fe1c2SSimon Glass        # If we have enough data, estimate build period (time taken for a
278fc3fe1c2SSimon Glass        # single build) and therefore completion time.
279fc3fe1c2SSimon Glass        if count > 1 and self._next_delay_update < now:
280fc3fe1c2SSimon Glass            self._next_delay_update = now + timedelta(seconds=2)
281fc3fe1c2SSimon Glass            if seconds > 0:
282fc3fe1c2SSimon Glass                self._build_period = float(seconds) / count
283fc3fe1c2SSimon Glass                todo = self.count - self.upto
284fc3fe1c2SSimon Glass                self._complete_delay = timedelta(microseconds=
285fc3fe1c2SSimon Glass                        self._build_period * todo * 1000000)
286fc3fe1c2SSimon Glass                # Round it
287fc3fe1c2SSimon Glass                self._complete_delay -= timedelta(
288fc3fe1c2SSimon Glass                        microseconds=self._complete_delay.microseconds)
289fc3fe1c2SSimon Glass
290fc3fe1c2SSimon Glass        if seconds > 60:
291fc3fe1c2SSimon Glass            self._timestamps.popleft()
292fc3fe1c2SSimon Glass            count -= 1
293fc3fe1c2SSimon Glass
294fc3fe1c2SSimon Glass    def ClearLine(self, length):
295fc3fe1c2SSimon Glass        """Clear any characters on the current line
296fc3fe1c2SSimon Glass
297fc3fe1c2SSimon Glass        Make way for a new line of length 'length', by outputting enough
298fc3fe1c2SSimon Glass        spaces to clear out the old line. Then remember the new length for
299fc3fe1c2SSimon Glass        next time.
300fc3fe1c2SSimon Glass
301fc3fe1c2SSimon Glass        Args:
302fc3fe1c2SSimon Glass            length: Length of new line, in characters
303fc3fe1c2SSimon Glass        """
304fc3fe1c2SSimon Glass        if length < self.last_line_len:
3054653a882SSimon Glass            Print(' ' * (self.last_line_len - length), newline=False)
3064653a882SSimon Glass            Print('\r', newline=False)
307fc3fe1c2SSimon Glass        self.last_line_len = length
308fc3fe1c2SSimon Glass        sys.stdout.flush()
309fc3fe1c2SSimon Glass
310fc3fe1c2SSimon Glass    def SelectCommit(self, commit, checkout=True):
311fc3fe1c2SSimon Glass        """Checkout the selected commit for this build
312fc3fe1c2SSimon Glass        """
313fc3fe1c2SSimon Glass        self.commit = commit
314fc3fe1c2SSimon Glass        if checkout and self.checkout:
315fc3fe1c2SSimon Glass            gitutil.Checkout(commit.hash)
316fc3fe1c2SSimon Glass
317fc3fe1c2SSimon Glass    def Make(self, commit, brd, stage, cwd, *args, **kwargs):
318fc3fe1c2SSimon Glass        """Run make
319fc3fe1c2SSimon Glass
320fc3fe1c2SSimon Glass        Args:
321fc3fe1c2SSimon Glass            commit: Commit object that is being built
322fc3fe1c2SSimon Glass            brd: Board object that is being built
323fd18a89eSRoger Meier            stage: Stage that we are at (mrproper, config, build)
324fc3fe1c2SSimon Glass            cwd: Directory where make should be run
325fc3fe1c2SSimon Glass            args: Arguments to pass to make
326fc3fe1c2SSimon Glass            kwargs: Arguments to pass to command.RunPipe()
327fc3fe1c2SSimon Glass        """
32899796923SMasahiro Yamada        cmd = [self.gnu_make] + list(args)
329fc3fe1c2SSimon Glass        result = command.RunPipe([cmd], capture=True, capture_stderr=True,
330fc3fe1c2SSimon Glass                cwd=cwd, raise_on_error=False, **kwargs)
331fc3fe1c2SSimon Glass        return result
332fc3fe1c2SSimon Glass
333fc3fe1c2SSimon Glass    def ProcessResult(self, result):
334fc3fe1c2SSimon Glass        """Process the result of a build, showing progress information
335fc3fe1c2SSimon Glass
336fc3fe1c2SSimon Glass        Args:
337e5a0e5d8SSimon Glass            result: A CommandResult object, which indicates the result for
338e5a0e5d8SSimon Glass                    a single build
339fc3fe1c2SSimon Glass        """
340fc3fe1c2SSimon Glass        col = terminal.Color()
341fc3fe1c2SSimon Glass        if result:
342fc3fe1c2SSimon Glass            target = result.brd.target
343fc3fe1c2SSimon Glass
344fc3fe1c2SSimon Glass            if result.return_code < 0:
345fc3fe1c2SSimon Glass                self.active = False
346fc3fe1c2SSimon Glass                command.StopAll()
347fc3fe1c2SSimon Glass                return
348fc3fe1c2SSimon Glass
349fc3fe1c2SSimon Glass            self.upto += 1
350fc3fe1c2SSimon Glass            if result.return_code != 0:
351fc3fe1c2SSimon Glass                self.fail += 1
352fc3fe1c2SSimon Glass            elif result.stderr:
353fc3fe1c2SSimon Glass                self.warned += 1
354fc3fe1c2SSimon Glass            if result.already_done:
355fc3fe1c2SSimon Glass                self.already_done += 1
356e5a0e5d8SSimon Glass            if self._verbose:
3574653a882SSimon Glass                Print('\r', newline=False)
358e5a0e5d8SSimon Glass                self.ClearLine(0)
359e5a0e5d8SSimon Glass                boards_selected = {target : result.brd}
360e5a0e5d8SSimon Glass                self.ResetResultSummary(boards_selected)
361e5a0e5d8SSimon Glass                self.ProduceResultSummary(result.commit_upto, self.commits,
362e5a0e5d8SSimon Glass                                          boards_selected)
363fc3fe1c2SSimon Glass        else:
364fc3fe1c2SSimon Glass            target = '(starting)'
365fc3fe1c2SSimon Glass
366fc3fe1c2SSimon Glass        # Display separate counts for ok, warned and fail
367fc3fe1c2SSimon Glass        ok = self.upto - self.warned - self.fail
368fc3fe1c2SSimon Glass        line = '\r' + self.col.Color(self.col.GREEN, '%5d' % ok)
369fc3fe1c2SSimon Glass        line += self.col.Color(self.col.YELLOW, '%5d' % self.warned)
370fc3fe1c2SSimon Glass        line += self.col.Color(self.col.RED, '%5d' % self.fail)
371fc3fe1c2SSimon Glass
372fc3fe1c2SSimon Glass        name = ' /%-5d  ' % self.count
373fc3fe1c2SSimon Glass
374fc3fe1c2SSimon Glass        # Add our current completion time estimate
375fc3fe1c2SSimon Glass        self._AddTimestamp()
376fc3fe1c2SSimon Glass        if self._complete_delay:
377fc3fe1c2SSimon Glass            name += '%s  : ' % self._complete_delay
378fc3fe1c2SSimon Glass        # When building all boards for a commit, we can print a commit
379fc3fe1c2SSimon Glass        # progress message.
380fc3fe1c2SSimon Glass        if result and result.commit_upto is None:
381fc3fe1c2SSimon Glass            name += 'commit %2d/%-3d' % (self.commit_upto + 1,
382fc3fe1c2SSimon Glass                    self.commit_count)
383fc3fe1c2SSimon Glass
384fc3fe1c2SSimon Glass        name += target
3854653a882SSimon Glass        Print(line + name, newline=False)
386e5a0e5d8SSimon Glass        length = 14 + len(name)
387fc3fe1c2SSimon Glass        self.ClearLine(length)
388fc3fe1c2SSimon Glass
389fc3fe1c2SSimon Glass    def _GetOutputDir(self, commit_upto):
390fc3fe1c2SSimon Glass        """Get the name of the output directory for a commit number
391fc3fe1c2SSimon Glass
392fc3fe1c2SSimon Glass        The output directory is typically .../<branch>/<commit>.
393fc3fe1c2SSimon Glass
394fc3fe1c2SSimon Glass        Args:
395fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
396fc3fe1c2SSimon Glass        """
397*5971ab5cSSimon Glass        commit_dir = None
398fea5858eSSimon Glass        if self.commits:
399fc3fe1c2SSimon Glass            commit = self.commits[commit_upto]
400fc3fe1c2SSimon Glass            subject = commit.subject.translate(trans_valid_chars)
401fc3fe1c2SSimon Glass            commit_dir = ('%02d_of_%02d_g%s_%s' % (commit_upto + 1,
402fc3fe1c2SSimon Glass                    self.commit_count, commit.hash, subject[:20]))
403*5971ab5cSSimon Glass        elif not self.no_subdirs:
404fea5858eSSimon Glass            commit_dir = 'current'
405*5971ab5cSSimon Glass        if not commit_dir:
406*5971ab5cSSimon Glass            return self.base_dir
407*5971ab5cSSimon Glass        return os.path.join(self.base_dir, commit_dir)
408fc3fe1c2SSimon Glass
409fc3fe1c2SSimon Glass    def GetBuildDir(self, commit_upto, target):
410fc3fe1c2SSimon Glass        """Get the name of the build directory for a commit number
411fc3fe1c2SSimon Glass
412fc3fe1c2SSimon Glass        The build directory is typically .../<branch>/<commit>/<target>.
413fc3fe1c2SSimon Glass
414fc3fe1c2SSimon Glass        Args:
415fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
416fc3fe1c2SSimon Glass            target: Target name
417fc3fe1c2SSimon Glass        """
418fc3fe1c2SSimon Glass        output_dir = self._GetOutputDir(commit_upto)
419fc3fe1c2SSimon Glass        return os.path.join(output_dir, target)
420fc3fe1c2SSimon Glass
421fc3fe1c2SSimon Glass    def GetDoneFile(self, commit_upto, target):
422fc3fe1c2SSimon Glass        """Get the name of the done file for a commit number
423fc3fe1c2SSimon Glass
424fc3fe1c2SSimon Glass        Args:
425fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
426fc3fe1c2SSimon Glass            target: Target name
427fc3fe1c2SSimon Glass        """
428fc3fe1c2SSimon Glass        return os.path.join(self.GetBuildDir(commit_upto, target), 'done')
429fc3fe1c2SSimon Glass
430fc3fe1c2SSimon Glass    def GetSizesFile(self, commit_upto, target):
431fc3fe1c2SSimon Glass        """Get the name of the sizes file for a commit number
432fc3fe1c2SSimon Glass
433fc3fe1c2SSimon Glass        Args:
434fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
435fc3fe1c2SSimon Glass            target: Target name
436fc3fe1c2SSimon Glass        """
437fc3fe1c2SSimon Glass        return os.path.join(self.GetBuildDir(commit_upto, target), 'sizes')
438fc3fe1c2SSimon Glass
439fc3fe1c2SSimon Glass    def GetFuncSizesFile(self, commit_upto, target, elf_fname):
440fc3fe1c2SSimon Glass        """Get the name of the funcsizes file for a commit number and ELF file
441fc3fe1c2SSimon Glass
442fc3fe1c2SSimon Glass        Args:
443fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
444fc3fe1c2SSimon Glass            target: Target name
445fc3fe1c2SSimon Glass            elf_fname: Filename of elf image
446fc3fe1c2SSimon Glass        """
447fc3fe1c2SSimon Glass        return os.path.join(self.GetBuildDir(commit_upto, target),
448fc3fe1c2SSimon Glass                            '%s.sizes' % elf_fname.replace('/', '-'))
449fc3fe1c2SSimon Glass
450fc3fe1c2SSimon Glass    def GetObjdumpFile(self, commit_upto, target, elf_fname):
451fc3fe1c2SSimon Glass        """Get the name of the objdump file for a commit number and ELF file
452fc3fe1c2SSimon Glass
453fc3fe1c2SSimon Glass        Args:
454fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
455fc3fe1c2SSimon Glass            target: Target name
456fc3fe1c2SSimon Glass            elf_fname: Filename of elf image
457fc3fe1c2SSimon Glass        """
458fc3fe1c2SSimon Glass        return os.path.join(self.GetBuildDir(commit_upto, target),
459fc3fe1c2SSimon Glass                            '%s.objdump' % elf_fname.replace('/', '-'))
460fc3fe1c2SSimon Glass
461fc3fe1c2SSimon Glass    def GetErrFile(self, commit_upto, target):
462fc3fe1c2SSimon Glass        """Get the name of the err file for a commit number
463fc3fe1c2SSimon Glass
464fc3fe1c2SSimon Glass        Args:
465fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
466fc3fe1c2SSimon Glass            target: Target name
467fc3fe1c2SSimon Glass        """
468fc3fe1c2SSimon Glass        output_dir = self.GetBuildDir(commit_upto, target)
469fc3fe1c2SSimon Glass        return os.path.join(output_dir, 'err')
470fc3fe1c2SSimon Glass
471fc3fe1c2SSimon Glass    def FilterErrors(self, lines):
472fc3fe1c2SSimon Glass        """Filter out errors in which we have no interest
473fc3fe1c2SSimon Glass
474fc3fe1c2SSimon Glass        We should probably use map().
475fc3fe1c2SSimon Glass
476fc3fe1c2SSimon Glass        Args:
477fc3fe1c2SSimon Glass            lines: List of error lines, each a string
478fc3fe1c2SSimon Glass        Returns:
479fc3fe1c2SSimon Glass            New list with only interesting lines included
480fc3fe1c2SSimon Glass        """
481fc3fe1c2SSimon Glass        out_lines = []
482fc3fe1c2SSimon Glass        for line in lines:
483fc3fe1c2SSimon Glass            if not self.re_make_err.search(line):
484fc3fe1c2SSimon Glass                out_lines.append(line)
485fc3fe1c2SSimon Glass        return out_lines
486fc3fe1c2SSimon Glass
487fc3fe1c2SSimon Glass    def ReadFuncSizes(self, fname, fd):
488fc3fe1c2SSimon Glass        """Read function sizes from the output of 'nm'
489fc3fe1c2SSimon Glass
490fc3fe1c2SSimon Glass        Args:
491fc3fe1c2SSimon Glass            fd: File containing data to read
492fc3fe1c2SSimon Glass            fname: Filename we are reading from (just for errors)
493fc3fe1c2SSimon Glass
494fc3fe1c2SSimon Glass        Returns:
495fc3fe1c2SSimon Glass            Dictionary containing size of each function in bytes, indexed by
496fc3fe1c2SSimon Glass            function name.
497fc3fe1c2SSimon Glass        """
498fc3fe1c2SSimon Glass        sym = {}
499fc3fe1c2SSimon Glass        for line in fd.readlines():
500fc3fe1c2SSimon Glass            try:
501fc3fe1c2SSimon Glass                size, type, name = line[:-1].split()
502fc3fe1c2SSimon Glass            except:
5034653a882SSimon Glass                Print("Invalid line in file '%s': '%s'" % (fname, line[:-1]))
504fc3fe1c2SSimon Glass                continue
505fc3fe1c2SSimon Glass            if type in 'tTdDbB':
506fc3fe1c2SSimon Glass                # function names begin with '.' on 64-bit powerpc
507fc3fe1c2SSimon Glass                if '.' in name[1:]:
508fc3fe1c2SSimon Glass                    name = 'static.' + name.split('.')[0]
509fc3fe1c2SSimon Glass                sym[name] = sym.get(name, 0) + int(size, 16)
510fc3fe1c2SSimon Glass        return sym
511fc3fe1c2SSimon Glass
512fc3fe1c2SSimon Glass    def GetBuildOutcome(self, commit_upto, target, read_func_sizes):
513fc3fe1c2SSimon Glass        """Work out the outcome of a build.
514fc3fe1c2SSimon Glass
515fc3fe1c2SSimon Glass        Args:
516fc3fe1c2SSimon Glass            commit_upto: Commit number to check (0..n-1)
517fc3fe1c2SSimon Glass            target: Target board to check
518fc3fe1c2SSimon Glass            read_func_sizes: True to read function size information
519fc3fe1c2SSimon Glass
520fc3fe1c2SSimon Glass        Returns:
521fc3fe1c2SSimon Glass            Outcome object
522fc3fe1c2SSimon Glass        """
523fc3fe1c2SSimon Glass        done_file = self.GetDoneFile(commit_upto, target)
524fc3fe1c2SSimon Glass        sizes_file = self.GetSizesFile(commit_upto, target)
525fc3fe1c2SSimon Glass        sizes = {}
526fc3fe1c2SSimon Glass        func_sizes = {}
527fc3fe1c2SSimon Glass        if os.path.exists(done_file):
528fc3fe1c2SSimon Glass            with open(done_file, 'r') as fd:
529fc3fe1c2SSimon Glass                return_code = int(fd.readline())
530fc3fe1c2SSimon Glass                err_lines = []
531fc3fe1c2SSimon Glass                err_file = self.GetErrFile(commit_upto, target)
532fc3fe1c2SSimon Glass                if os.path.exists(err_file):
533fc3fe1c2SSimon Glass                    with open(err_file, 'r') as fd:
534fc3fe1c2SSimon Glass                        err_lines = self.FilterErrors(fd.readlines())
535fc3fe1c2SSimon Glass
536fc3fe1c2SSimon Glass                # Decide whether the build was ok, failed or created warnings
537fc3fe1c2SSimon Glass                if return_code:
538fc3fe1c2SSimon Glass                    rc = OUTCOME_ERROR
539fc3fe1c2SSimon Glass                elif len(err_lines):
540fc3fe1c2SSimon Glass                    rc = OUTCOME_WARNING
541fc3fe1c2SSimon Glass                else:
542fc3fe1c2SSimon Glass                    rc = OUTCOME_OK
543fc3fe1c2SSimon Glass
544fc3fe1c2SSimon Glass                # Convert size information to our simple format
545fc3fe1c2SSimon Glass                if os.path.exists(sizes_file):
546fc3fe1c2SSimon Glass                    with open(sizes_file, 'r') as fd:
547fc3fe1c2SSimon Glass                        for line in fd.readlines():
548fc3fe1c2SSimon Glass                            values = line.split()
549fc3fe1c2SSimon Glass                            rodata = 0
550fc3fe1c2SSimon Glass                            if len(values) > 6:
551fc3fe1c2SSimon Glass                                rodata = int(values[6], 16)
552fc3fe1c2SSimon Glass                            size_dict = {
553fc3fe1c2SSimon Glass                                'all' : int(values[0]) + int(values[1]) +
554fc3fe1c2SSimon Glass                                        int(values[2]),
555fc3fe1c2SSimon Glass                                'text' : int(values[0]) - rodata,
556fc3fe1c2SSimon Glass                                'data' : int(values[1]),
557fc3fe1c2SSimon Glass                                'bss' : int(values[2]),
558fc3fe1c2SSimon Glass                                'rodata' : rodata,
559fc3fe1c2SSimon Glass                            }
560fc3fe1c2SSimon Glass                            sizes[values[5]] = size_dict
561fc3fe1c2SSimon Glass
562fc3fe1c2SSimon Glass            if read_func_sizes:
563fc3fe1c2SSimon Glass                pattern = self.GetFuncSizesFile(commit_upto, target, '*')
564fc3fe1c2SSimon Glass                for fname in glob.glob(pattern):
565fc3fe1c2SSimon Glass                    with open(fname, 'r') as fd:
566fc3fe1c2SSimon Glass                        dict_name = os.path.basename(fname).replace('.sizes',
567fc3fe1c2SSimon Glass                                                                    '')
568fc3fe1c2SSimon Glass                        func_sizes[dict_name] = self.ReadFuncSizes(fname, fd)
569fc3fe1c2SSimon Glass
570fc3fe1c2SSimon Glass            return Builder.Outcome(rc, err_lines, sizes, func_sizes)
571fc3fe1c2SSimon Glass
572fc3fe1c2SSimon Glass        return Builder.Outcome(OUTCOME_UNKNOWN, [], {}, {})
573fc3fe1c2SSimon Glass
574fc3fe1c2SSimon Glass    def GetResultSummary(self, boards_selected, commit_upto, read_func_sizes):
575fc3fe1c2SSimon Glass        """Calculate a summary of the results of building a commit.
576fc3fe1c2SSimon Glass
577fc3fe1c2SSimon Glass        Args:
578fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise
579fc3fe1c2SSimon Glass            commit_upto: Commit number to summarize (0..self.count-1)
580fc3fe1c2SSimon Glass            read_func_sizes: True to read function size information
581fc3fe1c2SSimon Glass
582fc3fe1c2SSimon Glass        Returns:
583fc3fe1c2SSimon Glass            Tuple:
584fc3fe1c2SSimon Glass                Dict containing boards which passed building this commit.
585fc3fe1c2SSimon Glass                    keyed by board.target
586e30965dbSSimon Glass                List containing a summary of error lines
587ed966657SSimon Glass                Dict keyed by error line, containing a list of the Board
588ed966657SSimon Glass                    objects with that error
589e30965dbSSimon Glass                List containing a summary of warning lines
590e30965dbSSimon Glass                Dict keyed by error line, containing a list of the Board
591e30965dbSSimon Glass                    objects with that warning
592fc3fe1c2SSimon Glass        """
593e30965dbSSimon Glass        def AddLine(lines_summary, lines_boards, line, board):
594e30965dbSSimon Glass            line = line.rstrip()
595e30965dbSSimon Glass            if line in lines_boards:
596e30965dbSSimon Glass                lines_boards[line].append(board)
597e30965dbSSimon Glass            else:
598e30965dbSSimon Glass                lines_boards[line] = [board]
599e30965dbSSimon Glass                lines_summary.append(line)
600e30965dbSSimon Glass
601fc3fe1c2SSimon Glass        board_dict = {}
602fc3fe1c2SSimon Glass        err_lines_summary = []
603ed966657SSimon Glass        err_lines_boards = {}
604e30965dbSSimon Glass        warn_lines_summary = []
605e30965dbSSimon Glass        warn_lines_boards = {}
606fc3fe1c2SSimon Glass
607fc3fe1c2SSimon Glass        for board in boards_selected.itervalues():
608fc3fe1c2SSimon Glass            outcome = self.GetBuildOutcome(commit_upto, board.target,
609fc3fe1c2SSimon Glass                                           read_func_sizes)
610fc3fe1c2SSimon Glass            board_dict[board.target] = outcome
611e30965dbSSimon Glass            last_func = None
612e30965dbSSimon Glass            last_was_warning = False
613e30965dbSSimon Glass            for line in outcome.err_lines:
614e30965dbSSimon Glass                if line:
615e30965dbSSimon Glass                    if (self._re_function.match(line) or
616e30965dbSSimon Glass                            self._re_files.match(line)):
617e30965dbSSimon Glass                        last_func = line
618ed966657SSimon Glass                    else:
619e30965dbSSimon Glass                        is_warning = self._re_warning.match(line)
620e30965dbSSimon Glass                        is_note = self._re_note.match(line)
621e30965dbSSimon Glass                        if is_warning or (last_was_warning and is_note):
622e30965dbSSimon Glass                            if last_func:
623e30965dbSSimon Glass                                AddLine(warn_lines_summary, warn_lines_boards,
624e30965dbSSimon Glass                                        last_func, board)
625e30965dbSSimon Glass                            AddLine(warn_lines_summary, warn_lines_boards,
626e30965dbSSimon Glass                                    line, board)
627e30965dbSSimon Glass                        else:
628e30965dbSSimon Glass                            if last_func:
629e30965dbSSimon Glass                                AddLine(err_lines_summary, err_lines_boards,
630e30965dbSSimon Glass                                        last_func, board)
631e30965dbSSimon Glass                            AddLine(err_lines_summary, err_lines_boards,
632e30965dbSSimon Glass                                    line, board)
633e30965dbSSimon Glass                        last_was_warning = is_warning
634e30965dbSSimon Glass                        last_func = None
635e30965dbSSimon Glass        return (board_dict, err_lines_summary, err_lines_boards,
636e30965dbSSimon Glass                warn_lines_summary, warn_lines_boards)
637fc3fe1c2SSimon Glass
638fc3fe1c2SSimon Glass    def AddOutcome(self, board_dict, arch_list, changes, char, color):
639fc3fe1c2SSimon Glass        """Add an output to our list of outcomes for each architecture
640fc3fe1c2SSimon Glass
641fc3fe1c2SSimon Glass        This simple function adds failing boards (changes) to the
642fc3fe1c2SSimon Glass        relevant architecture string, so we can print the results out
643fc3fe1c2SSimon Glass        sorted by architecture.
644fc3fe1c2SSimon Glass
645fc3fe1c2SSimon Glass        Args:
646fc3fe1c2SSimon Glass             board_dict: Dict containing all boards
647fc3fe1c2SSimon Glass             arch_list: Dict keyed by arch name. Value is a string containing
648fc3fe1c2SSimon Glass                    a list of board names which failed for that arch.
649fc3fe1c2SSimon Glass             changes: List of boards to add to arch_list
650fc3fe1c2SSimon Glass             color: terminal.Colour object
651fc3fe1c2SSimon Glass        """
652fc3fe1c2SSimon Glass        done_arch = {}
653fc3fe1c2SSimon Glass        for target in changes:
654fc3fe1c2SSimon Glass            if target in board_dict:
655fc3fe1c2SSimon Glass                arch = board_dict[target].arch
656fc3fe1c2SSimon Glass            else:
657fc3fe1c2SSimon Glass                arch = 'unknown'
658fc3fe1c2SSimon Glass            str = self.col.Color(color, ' ' + target)
659fc3fe1c2SSimon Glass            if not arch in done_arch:
660fc3fe1c2SSimon Glass                str = self.col.Color(color, char) + '  ' + str
661fc3fe1c2SSimon Glass                done_arch[arch] = True
662fc3fe1c2SSimon Glass            if not arch in arch_list:
663fc3fe1c2SSimon Glass                arch_list[arch] = str
664fc3fe1c2SSimon Glass            else:
665fc3fe1c2SSimon Glass                arch_list[arch] += str
666fc3fe1c2SSimon Glass
667fc3fe1c2SSimon Glass
668fc3fe1c2SSimon Glass    def ColourNum(self, num):
669fc3fe1c2SSimon Glass        color = self.col.RED if num > 0 else self.col.GREEN
670fc3fe1c2SSimon Glass        if num == 0:
671fc3fe1c2SSimon Glass            return '0'
672fc3fe1c2SSimon Glass        return self.col.Color(color, str(num))
673fc3fe1c2SSimon Glass
674fc3fe1c2SSimon Glass    def ResetResultSummary(self, board_selected):
675fc3fe1c2SSimon Glass        """Reset the results summary ready for use.
676fc3fe1c2SSimon Glass
677fc3fe1c2SSimon Glass        Set up the base board list to be all those selected, and set the
678fc3fe1c2SSimon Glass        error lines to empty.
679fc3fe1c2SSimon Glass
680fc3fe1c2SSimon Glass        Following this, calls to PrintResultSummary() will use this
681fc3fe1c2SSimon Glass        information to work out what has changed.
682fc3fe1c2SSimon Glass
683fc3fe1c2SSimon Glass        Args:
684fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise, keyed by
685fc3fe1c2SSimon Glass                board.target
686fc3fe1c2SSimon Glass        """
687fc3fe1c2SSimon Glass        self._base_board_dict = {}
688fc3fe1c2SSimon Glass        for board in board_selected:
689fc3fe1c2SSimon Glass            self._base_board_dict[board] = Builder.Outcome(0, [], [], {})
690fc3fe1c2SSimon Glass        self._base_err_lines = []
691e30965dbSSimon Glass        self._base_warn_lines = []
692e30965dbSSimon Glass        self._base_err_line_boards = {}
693e30965dbSSimon Glass        self._base_warn_line_boards = {}
694fc3fe1c2SSimon Glass
695fc3fe1c2SSimon Glass    def PrintFuncSizeDetail(self, fname, old, new):
696fc3fe1c2SSimon Glass        grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
697fc3fe1c2SSimon Glass        delta, common = [], {}
698fc3fe1c2SSimon Glass
699fc3fe1c2SSimon Glass        for a in old:
700fc3fe1c2SSimon Glass            if a in new:
701fc3fe1c2SSimon Glass                common[a] = 1
702fc3fe1c2SSimon Glass
703fc3fe1c2SSimon Glass        for name in old:
704fc3fe1c2SSimon Glass            if name not in common:
705fc3fe1c2SSimon Glass                remove += 1
706fc3fe1c2SSimon Glass                down += old[name]
707fc3fe1c2SSimon Glass                delta.append([-old[name], name])
708fc3fe1c2SSimon Glass
709fc3fe1c2SSimon Glass        for name in new:
710fc3fe1c2SSimon Glass            if name not in common:
711fc3fe1c2SSimon Glass                add += 1
712fc3fe1c2SSimon Glass                up += new[name]
713fc3fe1c2SSimon Glass                delta.append([new[name], name])
714fc3fe1c2SSimon Glass
715fc3fe1c2SSimon Glass        for name in common:
716fc3fe1c2SSimon Glass                diff = new.get(name, 0) - old.get(name, 0)
717fc3fe1c2SSimon Glass                if diff > 0:
718fc3fe1c2SSimon Glass                    grow, up = grow + 1, up + diff
719fc3fe1c2SSimon Glass                elif diff < 0:
720fc3fe1c2SSimon Glass                    shrink, down = shrink + 1, down - diff
721fc3fe1c2SSimon Glass                delta.append([diff, name])
722fc3fe1c2SSimon Glass
723fc3fe1c2SSimon Glass        delta.sort()
724fc3fe1c2SSimon Glass        delta.reverse()
725fc3fe1c2SSimon Glass
726fc3fe1c2SSimon Glass        args = [add, -remove, grow, -shrink, up, -down, up - down]
727fc3fe1c2SSimon Glass        if max(args) == 0:
728fc3fe1c2SSimon Glass            return
729fc3fe1c2SSimon Glass        args = [self.ColourNum(x) for x in args]
730fc3fe1c2SSimon Glass        indent = ' ' * 15
7314653a882SSimon Glass        Print('%s%s: add: %s/%s, grow: %s/%s bytes: %s/%s (%s)' %
732fc3fe1c2SSimon Glass              tuple([indent, self.col.Color(self.col.YELLOW, fname)] + args))
7334653a882SSimon Glass        Print('%s  %-38s %7s %7s %+7s' % (indent, 'function', 'old', 'new',
7344653a882SSimon Glass                                         'delta'))
735fc3fe1c2SSimon Glass        for diff, name in delta:
736fc3fe1c2SSimon Glass            if diff:
737fc3fe1c2SSimon Glass                color = self.col.RED if diff > 0 else self.col.GREEN
738fc3fe1c2SSimon Glass                msg = '%s  %-38s %7s %7s %+7d' % (indent, name,
739fc3fe1c2SSimon Glass                        old.get(name, '-'), new.get(name,'-'), diff)
7404653a882SSimon Glass                Print(msg, colour=color)
741fc3fe1c2SSimon Glass
742fc3fe1c2SSimon Glass
743fc3fe1c2SSimon Glass    def PrintSizeDetail(self, target_list, show_bloat):
744fc3fe1c2SSimon Glass        """Show details size information for each board
745fc3fe1c2SSimon Glass
746fc3fe1c2SSimon Glass        Args:
747fc3fe1c2SSimon Glass            target_list: List of targets, each a dict containing:
748fc3fe1c2SSimon Glass                    'target': Target name
749fc3fe1c2SSimon Glass                    'total_diff': Total difference in bytes across all areas
750fc3fe1c2SSimon Glass                    <part_name>: Difference for that part
751fc3fe1c2SSimon Glass            show_bloat: Show detail for each function
752fc3fe1c2SSimon Glass        """
753fc3fe1c2SSimon Glass        targets_by_diff = sorted(target_list, reverse=True,
754fc3fe1c2SSimon Glass        key=lambda x: x['_total_diff'])
755fc3fe1c2SSimon Glass        for result in targets_by_diff:
756fc3fe1c2SSimon Glass            printed_target = False
757fc3fe1c2SSimon Glass            for name in sorted(result):
758fc3fe1c2SSimon Glass                diff = result[name]
759fc3fe1c2SSimon Glass                if name.startswith('_'):
760fc3fe1c2SSimon Glass                    continue
761fc3fe1c2SSimon Glass                if diff != 0:
762fc3fe1c2SSimon Glass                    color = self.col.RED if diff > 0 else self.col.GREEN
763fc3fe1c2SSimon Glass                msg = ' %s %+d' % (name, diff)
764fc3fe1c2SSimon Glass                if not printed_target:
7654653a882SSimon Glass                    Print('%10s  %-15s:' % ('', result['_target']),
7664653a882SSimon Glass                          newline=False)
767fc3fe1c2SSimon Glass                    printed_target = True
7684653a882SSimon Glass                Print(msg, colour=color, newline=False)
769fc3fe1c2SSimon Glass            if printed_target:
7704653a882SSimon Glass                Print()
771fc3fe1c2SSimon Glass                if show_bloat:
772fc3fe1c2SSimon Glass                    target = result['_target']
773fc3fe1c2SSimon Glass                    outcome = result['_outcome']
774fc3fe1c2SSimon Glass                    base_outcome = self._base_board_dict[target]
775fc3fe1c2SSimon Glass                    for fname in outcome.func_sizes:
776fc3fe1c2SSimon Glass                        self.PrintFuncSizeDetail(fname,
777fc3fe1c2SSimon Glass                                                 base_outcome.func_sizes[fname],
778fc3fe1c2SSimon Glass                                                 outcome.func_sizes[fname])
779fc3fe1c2SSimon Glass
780fc3fe1c2SSimon Glass
781fc3fe1c2SSimon Glass    def PrintSizeSummary(self, board_selected, board_dict, show_detail,
782fc3fe1c2SSimon Glass                         show_bloat):
783fc3fe1c2SSimon Glass        """Print a summary of image sizes broken down by section.
784fc3fe1c2SSimon Glass
785fc3fe1c2SSimon Glass        The summary takes the form of one line per architecture. The
786fc3fe1c2SSimon Glass        line contains deltas for each of the sections (+ means the section
787fc3fe1c2SSimon Glass        got bigger, - means smaller). The nunmbers are the average number
788fc3fe1c2SSimon Glass        of bytes that a board in this section increased by.
789fc3fe1c2SSimon Glass
790fc3fe1c2SSimon Glass        For example:
791fc3fe1c2SSimon Glass           powerpc: (622 boards)   text -0.0
792fc3fe1c2SSimon Glass          arm: (285 boards)   text -0.0
793fc3fe1c2SSimon Glass          nds32: (3 boards)   text -8.0
794fc3fe1c2SSimon Glass
795fc3fe1c2SSimon Glass        Args:
796fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise, keyed by
797fc3fe1c2SSimon Glass                board.target
798fc3fe1c2SSimon Glass            board_dict: Dict containing boards for which we built this
799fc3fe1c2SSimon Glass                commit, keyed by board.target. The value is an Outcome object.
800fc3fe1c2SSimon Glass            show_detail: Show detail for each board
801fc3fe1c2SSimon Glass            show_bloat: Show detail for each function
802fc3fe1c2SSimon Glass        """
803fc3fe1c2SSimon Glass        arch_list = {}
804fc3fe1c2SSimon Glass        arch_count = {}
805fc3fe1c2SSimon Glass
806fc3fe1c2SSimon Glass        # Calculate changes in size for different image parts
807fc3fe1c2SSimon Glass        # The previous sizes are in Board.sizes, for each board
808fc3fe1c2SSimon Glass        for target in board_dict:
809fc3fe1c2SSimon Glass            if target not in board_selected:
810fc3fe1c2SSimon Glass                continue
811fc3fe1c2SSimon Glass            base_sizes = self._base_board_dict[target].sizes
812fc3fe1c2SSimon Glass            outcome = board_dict[target]
813fc3fe1c2SSimon Glass            sizes = outcome.sizes
814fc3fe1c2SSimon Glass
815fc3fe1c2SSimon Glass            # Loop through the list of images, creating a dict of size
816fc3fe1c2SSimon Glass            # changes for each image/part. We end up with something like
817fc3fe1c2SSimon Glass            # {'target' : 'snapper9g45, 'data' : 5, 'u-boot-spl:text' : -4}
818fc3fe1c2SSimon Glass            # which means that U-Boot data increased by 5 bytes and SPL
819fc3fe1c2SSimon Glass            # text decreased by 4.
820fc3fe1c2SSimon Glass            err = {'_target' : target}
821fc3fe1c2SSimon Glass            for image in sizes:
822fc3fe1c2SSimon Glass                if image in base_sizes:
823fc3fe1c2SSimon Glass                    base_image = base_sizes[image]
824fc3fe1c2SSimon Glass                    # Loop through the text, data, bss parts
825fc3fe1c2SSimon Glass                    for part in sorted(sizes[image]):
826fc3fe1c2SSimon Glass                        diff = sizes[image][part] - base_image[part]
827fc3fe1c2SSimon Glass                        col = None
828fc3fe1c2SSimon Glass                        if diff:
829fc3fe1c2SSimon Glass                            if image == 'u-boot':
830fc3fe1c2SSimon Glass                                name = part
831fc3fe1c2SSimon Glass                            else:
832fc3fe1c2SSimon Glass                                name = image + ':' + part
833fc3fe1c2SSimon Glass                            err[name] = diff
834fc3fe1c2SSimon Glass            arch = board_selected[target].arch
835fc3fe1c2SSimon Glass            if not arch in arch_count:
836fc3fe1c2SSimon Glass                arch_count[arch] = 1
837fc3fe1c2SSimon Glass            else:
838fc3fe1c2SSimon Glass                arch_count[arch] += 1
839fc3fe1c2SSimon Glass            if not sizes:
840fc3fe1c2SSimon Glass                pass    # Only add to our list when we have some stats
841fc3fe1c2SSimon Glass            elif not arch in arch_list:
842fc3fe1c2SSimon Glass                arch_list[arch] = [err]
843fc3fe1c2SSimon Glass            else:
844fc3fe1c2SSimon Glass                arch_list[arch].append(err)
845fc3fe1c2SSimon Glass
846fc3fe1c2SSimon Glass        # We now have a list of image size changes sorted by arch
847fc3fe1c2SSimon Glass        # Print out a summary of these
848fc3fe1c2SSimon Glass        for arch, target_list in arch_list.iteritems():
849fc3fe1c2SSimon Glass            # Get total difference for each type
850fc3fe1c2SSimon Glass            totals = {}
851fc3fe1c2SSimon Glass            for result in target_list:
852fc3fe1c2SSimon Glass                total = 0
853fc3fe1c2SSimon Glass                for name, diff in result.iteritems():
854fc3fe1c2SSimon Glass                    if name.startswith('_'):
855fc3fe1c2SSimon Glass                        continue
856fc3fe1c2SSimon Glass                    total += diff
857fc3fe1c2SSimon Glass                    if name in totals:
858fc3fe1c2SSimon Glass                        totals[name] += diff
859fc3fe1c2SSimon Glass                    else:
860fc3fe1c2SSimon Glass                        totals[name] = diff
861fc3fe1c2SSimon Glass                result['_total_diff'] = total
862fc3fe1c2SSimon Glass                result['_outcome'] = board_dict[result['_target']]
863fc3fe1c2SSimon Glass
864fc3fe1c2SSimon Glass            count = len(target_list)
865fc3fe1c2SSimon Glass            printed_arch = False
866fc3fe1c2SSimon Glass            for name in sorted(totals):
867fc3fe1c2SSimon Glass                diff = totals[name]
868fc3fe1c2SSimon Glass                if diff:
869fc3fe1c2SSimon Glass                    # Display the average difference in this name for this
870fc3fe1c2SSimon Glass                    # architecture
871fc3fe1c2SSimon Glass                    avg_diff = float(diff) / count
872fc3fe1c2SSimon Glass                    color = self.col.RED if avg_diff > 0 else self.col.GREEN
873fc3fe1c2SSimon Glass                    msg = ' %s %+1.1f' % (name, avg_diff)
874fc3fe1c2SSimon Glass                    if not printed_arch:
8754653a882SSimon Glass                        Print('%10s: (for %d/%d boards)' % (arch, count,
8764653a882SSimon Glass                              arch_count[arch]), newline=False)
877fc3fe1c2SSimon Glass                        printed_arch = True
8784653a882SSimon Glass                    Print(msg, colour=color, newline=False)
879fc3fe1c2SSimon Glass
880fc3fe1c2SSimon Glass            if printed_arch:
8814653a882SSimon Glass                Print()
882fc3fe1c2SSimon Glass                if show_detail:
883fc3fe1c2SSimon Glass                    self.PrintSizeDetail(target_list, show_bloat)
884fc3fe1c2SSimon Glass
885fc3fe1c2SSimon Glass
886fc3fe1c2SSimon Glass    def PrintResultSummary(self, board_selected, board_dict, err_lines,
887e30965dbSSimon Glass                           err_line_boards, warn_lines, warn_line_boards,
888e30965dbSSimon Glass                           show_sizes, show_detail, show_bloat):
889fc3fe1c2SSimon Glass        """Compare results with the base results and display delta.
890fc3fe1c2SSimon Glass
891fc3fe1c2SSimon Glass        Only boards mentioned in board_selected will be considered. This
892fc3fe1c2SSimon Glass        function is intended to be called repeatedly with the results of
893fc3fe1c2SSimon Glass        each commit. It therefore shows a 'diff' between what it saw in
894fc3fe1c2SSimon Glass        the last call and what it sees now.
895fc3fe1c2SSimon Glass
896fc3fe1c2SSimon Glass        Args:
897fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise, keyed by
898fc3fe1c2SSimon Glass                board.target
899fc3fe1c2SSimon Glass            board_dict: Dict containing boards for which we built this
900fc3fe1c2SSimon Glass                commit, keyed by board.target. The value is an Outcome object.
901fc3fe1c2SSimon Glass            err_lines: A list of errors for this commit, or [] if there is
902fc3fe1c2SSimon Glass                none, or we don't want to print errors
903ed966657SSimon Glass            err_line_boards: Dict keyed by error line, containing a list of
904ed966657SSimon Glass                the Board objects with that error
905e30965dbSSimon Glass            warn_lines: A list of warnings for this commit, or [] if there is
906e30965dbSSimon Glass                none, or we don't want to print errors
907e30965dbSSimon Glass            warn_line_boards: Dict keyed by warning line, containing a list of
908e30965dbSSimon Glass                the Board objects with that warning
909fc3fe1c2SSimon Glass            show_sizes: Show image size deltas
910fc3fe1c2SSimon Glass            show_detail: Show detail for each board
911fc3fe1c2SSimon Glass            show_bloat: Show detail for each function
912fc3fe1c2SSimon Glass        """
913e30965dbSSimon Glass        def _BoardList(line, line_boards):
914ed966657SSimon Glass            """Helper function to get a line of boards containing a line
915ed966657SSimon Glass
916ed966657SSimon Glass            Args:
917ed966657SSimon Glass                line: Error line to search for
918ed966657SSimon Glass            Return:
919ed966657SSimon Glass                String containing a list of boards with that error line, or
920ed966657SSimon Glass                '' if the user has not requested such a list
921ed966657SSimon Glass            """
922ed966657SSimon Glass            if self._list_error_boards:
923ed966657SSimon Glass                names = []
924e30965dbSSimon Glass                for board in line_boards[line]:
925f66153beSSimon Glass                    if not board.target in names:
926ed966657SSimon Glass                        names.append(board.target)
927ed966657SSimon Glass                names_str = '(%s) ' % ','.join(names)
928ed966657SSimon Glass            else:
929ed966657SSimon Glass                names_str = ''
930ed966657SSimon Glass            return names_str
931ed966657SSimon Glass
932e30965dbSSimon Glass        def _CalcErrorDelta(base_lines, base_line_boards, lines, line_boards,
933e30965dbSSimon Glass                            char):
934e30965dbSSimon Glass            better_lines = []
935e30965dbSSimon Glass            worse_lines = []
936e30965dbSSimon Glass            for line in lines:
937e30965dbSSimon Glass                if line not in base_lines:
938e30965dbSSimon Glass                    worse_lines.append(char + '+' +
939e30965dbSSimon Glass                            _BoardList(line, line_boards) + line)
940e30965dbSSimon Glass            for line in base_lines:
941e30965dbSSimon Glass                if line not in lines:
942e30965dbSSimon Glass                    better_lines.append(char + '-' +
943e30965dbSSimon Glass                            _BoardList(line, base_line_boards) + line)
944e30965dbSSimon Glass            return better_lines, worse_lines
945e30965dbSSimon Glass
946fc3fe1c2SSimon Glass        better = []     # List of boards fixed since last commit
947fc3fe1c2SSimon Glass        worse = []      # List of new broken boards since last commit
948fc3fe1c2SSimon Glass        new = []        # List of boards that didn't exist last time
949fc3fe1c2SSimon Glass        unknown = []    # List of boards that were not built
950fc3fe1c2SSimon Glass
951fc3fe1c2SSimon Glass        for target in board_dict:
952fc3fe1c2SSimon Glass            if target not in board_selected:
953fc3fe1c2SSimon Glass                continue
954fc3fe1c2SSimon Glass
955fc3fe1c2SSimon Glass            # If the board was built last time, add its outcome to a list
956fc3fe1c2SSimon Glass            if target in self._base_board_dict:
957fc3fe1c2SSimon Glass                base_outcome = self._base_board_dict[target].rc
958fc3fe1c2SSimon Glass                outcome = board_dict[target]
959fc3fe1c2SSimon Glass                if outcome.rc == OUTCOME_UNKNOWN:
960fc3fe1c2SSimon Glass                    unknown.append(target)
961fc3fe1c2SSimon Glass                elif outcome.rc < base_outcome:
962fc3fe1c2SSimon Glass                    better.append(target)
963fc3fe1c2SSimon Glass                elif outcome.rc > base_outcome:
964fc3fe1c2SSimon Glass                    worse.append(target)
965fc3fe1c2SSimon Glass            else:
966fc3fe1c2SSimon Glass                new.append(target)
967fc3fe1c2SSimon Glass
968fc3fe1c2SSimon Glass        # Get a list of errors that have appeared, and disappeared
969e30965dbSSimon Glass        better_err, worse_err = _CalcErrorDelta(self._base_err_lines,
970e30965dbSSimon Glass                self._base_err_line_boards, err_lines, err_line_boards, '')
971e30965dbSSimon Glass        better_warn, worse_warn = _CalcErrorDelta(self._base_warn_lines,
972e30965dbSSimon Glass                self._base_warn_line_boards, warn_lines, warn_line_boards, 'w')
973fc3fe1c2SSimon Glass
974fc3fe1c2SSimon Glass        # Display results by arch
975e30965dbSSimon Glass        if (better or worse or unknown or new or worse_err or better_err
976e30965dbSSimon Glass                or worse_warn or better_warn):
977fc3fe1c2SSimon Glass            arch_list = {}
978fc3fe1c2SSimon Glass            self.AddOutcome(board_selected, arch_list, better, '',
979fc3fe1c2SSimon Glass                    self.col.GREEN)
980fc3fe1c2SSimon Glass            self.AddOutcome(board_selected, arch_list, worse, '+',
981fc3fe1c2SSimon Glass                    self.col.RED)
982fc3fe1c2SSimon Glass            self.AddOutcome(board_selected, arch_list, new, '*', self.col.BLUE)
983fc3fe1c2SSimon Glass            if self._show_unknown:
984fc3fe1c2SSimon Glass                self.AddOutcome(board_selected, arch_list, unknown, '?',
985fc3fe1c2SSimon Glass                        self.col.MAGENTA)
986fc3fe1c2SSimon Glass            for arch, target_list in arch_list.iteritems():
9874653a882SSimon Glass                Print('%10s: %s' % (arch, target_list))
98828370c1bSSimon Glass                self._error_lines += 1
989fc3fe1c2SSimon Glass            if better_err:
9904653a882SSimon Glass                Print('\n'.join(better_err), colour=self.col.GREEN)
99128370c1bSSimon Glass                self._error_lines += 1
992fc3fe1c2SSimon Glass            if worse_err:
9934653a882SSimon Glass                Print('\n'.join(worse_err), colour=self.col.RED)
99428370c1bSSimon Glass                self._error_lines += 1
995e30965dbSSimon Glass            if better_warn:
9964653a882SSimon Glass                Print('\n'.join(better_warn), colour=self.col.CYAN)
997e30965dbSSimon Glass                self._error_lines += 1
998e30965dbSSimon Glass            if worse_warn:
9994653a882SSimon Glass                Print('\n'.join(worse_warn), colour=self.col.MAGENTA)
1000e30965dbSSimon Glass                self._error_lines += 1
1001fc3fe1c2SSimon Glass
1002fc3fe1c2SSimon Glass        if show_sizes:
1003fc3fe1c2SSimon Glass            self.PrintSizeSummary(board_selected, board_dict, show_detail,
1004fc3fe1c2SSimon Glass                                  show_bloat)
1005fc3fe1c2SSimon Glass
1006fc3fe1c2SSimon Glass        # Save our updated information for the next call to this function
1007fc3fe1c2SSimon Glass        self._base_board_dict = board_dict
1008fc3fe1c2SSimon Glass        self._base_err_lines = err_lines
1009e30965dbSSimon Glass        self._base_warn_lines = warn_lines
1010e30965dbSSimon Glass        self._base_err_line_boards = err_line_boards
1011e30965dbSSimon Glass        self._base_warn_line_boards = warn_line_boards
1012fc3fe1c2SSimon Glass
1013fc3fe1c2SSimon Glass        # Get a list of boards that did not get built, if needed
1014fc3fe1c2SSimon Glass        not_built = []
1015fc3fe1c2SSimon Glass        for board in board_selected:
1016fc3fe1c2SSimon Glass            if not board in board_dict:
1017fc3fe1c2SSimon Glass                not_built.append(board)
1018fc3fe1c2SSimon Glass        if not_built:
10194653a882SSimon Glass            Print("Boards not built (%d): %s" % (len(not_built),
10204653a882SSimon Glass                  ', '.join(not_built)))
1021fc3fe1c2SSimon Glass
1022b2ea7ab2SSimon Glass    def ProduceResultSummary(self, commit_upto, commits, board_selected):
1023e30965dbSSimon Glass            (board_dict, err_lines, err_line_boards, warn_lines,
1024e30965dbSSimon Glass                    warn_line_boards) = self.GetResultSummary(
1025ed966657SSimon Glass                    board_selected, commit_upto,
1026ed966657SSimon Glass                    read_func_sizes=self._show_bloat)
1027b2ea7ab2SSimon Glass            if commits:
1028b2ea7ab2SSimon Glass                msg = '%02d: %s' % (commit_upto + 1,
1029b2ea7ab2SSimon Glass                        commits[commit_upto].subject)
10304653a882SSimon Glass                Print(msg, colour=self.col.BLUE)
1031b2ea7ab2SSimon Glass            self.PrintResultSummary(board_selected, board_dict,
1032ed966657SSimon Glass                    err_lines if self._show_errors else [], err_line_boards,
1033e30965dbSSimon Glass                    warn_lines if self._show_errors else [], warn_line_boards,
1034b2ea7ab2SSimon Glass                    self._show_sizes, self._show_detail, self._show_bloat)
1035fc3fe1c2SSimon Glass
1036b2ea7ab2SSimon Glass    def ShowSummary(self, commits, board_selected):
1037fc3fe1c2SSimon Glass        """Show a build summary for U-Boot for a given board list.
1038fc3fe1c2SSimon Glass
1039fc3fe1c2SSimon Glass        Reset the result summary, then repeatedly call GetResultSummary on
1040fc3fe1c2SSimon Glass        each commit's results, then display the differences we see.
1041fc3fe1c2SSimon Glass
1042fc3fe1c2SSimon Glass        Args:
1043fc3fe1c2SSimon Glass            commit: Commit objects to summarise
1044fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise
1045fc3fe1c2SSimon Glass        """
1046fea5858eSSimon Glass        self.commit_count = len(commits) if commits else 1
1047fc3fe1c2SSimon Glass        self.commits = commits
1048fc3fe1c2SSimon Glass        self.ResetResultSummary(board_selected)
104928370c1bSSimon Glass        self._error_lines = 0
1050fc3fe1c2SSimon Glass
1051fc3fe1c2SSimon Glass        for commit_upto in range(0, self.commit_count, self._step):
1052b2ea7ab2SSimon Glass            self.ProduceResultSummary(commit_upto, commits, board_selected)
105328370c1bSSimon Glass        if not self._error_lines:
10544653a882SSimon Glass            Print('(no errors to report)', colour=self.col.GREEN)
1055fc3fe1c2SSimon Glass
1056fc3fe1c2SSimon Glass
1057fc3fe1c2SSimon Glass    def SetupBuild(self, board_selected, commits):
1058fc3fe1c2SSimon Glass        """Set up ready to start a build.
1059fc3fe1c2SSimon Glass
1060fc3fe1c2SSimon Glass        Args:
1061fc3fe1c2SSimon Glass            board_selected: Selected boards to build
1062fc3fe1c2SSimon Glass            commits: Selected commits to build
1063fc3fe1c2SSimon Glass        """
1064fc3fe1c2SSimon Glass        # First work out how many commits we will build
1065fea5858eSSimon Glass        count = (self.commit_count + self._step - 1) / self._step
1066fc3fe1c2SSimon Glass        self.count = len(board_selected) * count
1067fc3fe1c2SSimon Glass        self.upto = self.warned = self.fail = 0
1068fc3fe1c2SSimon Glass        self._timestamps = collections.deque()
1069fc3fe1c2SSimon Glass
1070fc3fe1c2SSimon Glass    def GetThreadDir(self, thread_num):
1071fc3fe1c2SSimon Glass        """Get the directory path to the working dir for a thread.
1072fc3fe1c2SSimon Glass
1073fc3fe1c2SSimon Glass        Args:
1074fc3fe1c2SSimon Glass            thread_num: Number of thread to check.
1075fc3fe1c2SSimon Glass        """
1076fc3fe1c2SSimon Glass        return os.path.join(self._working_dir, '%02d' % thread_num)
1077fc3fe1c2SSimon Glass
1078fea5858eSSimon Glass    def _PrepareThread(self, thread_num, setup_git):
1079fc3fe1c2SSimon Glass        """Prepare the working directory for a thread.
1080fc3fe1c2SSimon Glass
1081fc3fe1c2SSimon Glass        This clones or fetches the repo into the thread's work directory.
1082fc3fe1c2SSimon Glass
1083fc3fe1c2SSimon Glass        Args:
1084fc3fe1c2SSimon Glass            thread_num: Thread number (0, 1, ...)
1085fea5858eSSimon Glass            setup_git: True to set up a git repo clone
1086fc3fe1c2SSimon Glass        """
1087fc3fe1c2SSimon Glass        thread_dir = self.GetThreadDir(thread_num)
1088190064b4SSimon Glass        builderthread.Mkdir(thread_dir)
1089fc3fe1c2SSimon Glass        git_dir = os.path.join(thread_dir, '.git')
1090fc3fe1c2SSimon Glass
1091fc3fe1c2SSimon Glass        # Clone the repo if it doesn't already exist
1092fc3fe1c2SSimon Glass        # TODO(sjg@chromium): Perhaps some git hackery to symlink instead, so
1093fc3fe1c2SSimon Glass        # we have a private index but uses the origin repo's contents?
1094fea5858eSSimon Glass        if setup_git and self.git_dir:
1095fc3fe1c2SSimon Glass            src_dir = os.path.abspath(self.git_dir)
1096fc3fe1c2SSimon Glass            if os.path.exists(git_dir):
1097fc3fe1c2SSimon Glass                gitutil.Fetch(git_dir, thread_dir)
1098fc3fe1c2SSimon Glass            else:
10994653a882SSimon Glass                Print('Cloning repo for thread %d' % thread_num)
1100fc3fe1c2SSimon Glass                gitutil.Clone(src_dir, thread_dir)
1101fc3fe1c2SSimon Glass
1102fea5858eSSimon Glass    def _PrepareWorkingSpace(self, max_threads, setup_git):
1103fc3fe1c2SSimon Glass        """Prepare the working directory for use.
1104fc3fe1c2SSimon Glass
1105fc3fe1c2SSimon Glass        Set up the git repo for each thread.
1106fc3fe1c2SSimon Glass
1107fc3fe1c2SSimon Glass        Args:
1108fc3fe1c2SSimon Glass            max_threads: Maximum number of threads we expect to need.
1109fea5858eSSimon Glass            setup_git: True to set up a git repo clone
1110fc3fe1c2SSimon Glass        """
1111190064b4SSimon Glass        builderthread.Mkdir(self._working_dir)
1112fc3fe1c2SSimon Glass        for thread in range(max_threads):
1113fea5858eSSimon Glass            self._PrepareThread(thread, setup_git)
1114fc3fe1c2SSimon Glass
1115fc3fe1c2SSimon Glass    def _PrepareOutputSpace(self):
1116fc3fe1c2SSimon Glass        """Get the output directories ready to receive files.
1117fc3fe1c2SSimon Glass
1118fc3fe1c2SSimon Glass        We delete any output directories which look like ones we need to
1119fc3fe1c2SSimon Glass        create. Having left over directories is confusing when the user wants
1120fc3fe1c2SSimon Glass        to check the output manually.
1121fc3fe1c2SSimon Glass        """
11221a915675SSimon Glass        if not self.commits:
11231a915675SSimon Glass            return
1124fc3fe1c2SSimon Glass        dir_list = []
1125fc3fe1c2SSimon Glass        for commit_upto in range(self.commit_count):
1126fc3fe1c2SSimon Glass            dir_list.append(self._GetOutputDir(commit_upto))
1127fc3fe1c2SSimon Glass
1128fc3fe1c2SSimon Glass        for dirname in glob.glob(os.path.join(self.base_dir, '*')):
1129fc3fe1c2SSimon Glass            if dirname not in dir_list:
1130fc3fe1c2SSimon Glass                shutil.rmtree(dirname)
1131fc3fe1c2SSimon Glass
1132e5a0e5d8SSimon Glass    def BuildBoards(self, commits, board_selected, keep_outputs, verbose):
1133fc3fe1c2SSimon Glass        """Build all commits for a list of boards
1134fc3fe1c2SSimon Glass
1135fc3fe1c2SSimon Glass        Args:
1136fc3fe1c2SSimon Glass            commits: List of commits to be build, each a Commit object
1137fc3fe1c2SSimon Glass            boards_selected: Dict of selected boards, key is target name,
1138fc3fe1c2SSimon Glass                    value is Board object
1139fc3fe1c2SSimon Glass            keep_outputs: True to save build output files
1140e5a0e5d8SSimon Glass            verbose: Display build results as they are completed
11412c3deb97SSimon Glass        Returns:
11422c3deb97SSimon Glass            Tuple containing:
11432c3deb97SSimon Glass                - number of boards that failed to build
11442c3deb97SSimon Glass                - number of boards that issued warnings
1145fc3fe1c2SSimon Glass        """
1146fea5858eSSimon Glass        self.commit_count = len(commits) if commits else 1
1147fc3fe1c2SSimon Glass        self.commits = commits
1148e5a0e5d8SSimon Glass        self._verbose = verbose
1149fc3fe1c2SSimon Glass
1150fc3fe1c2SSimon Glass        self.ResetResultSummary(board_selected)
1151f3d015cbSThierry Reding        builderthread.Mkdir(self.base_dir, parents = True)
1152fea5858eSSimon Glass        self._PrepareWorkingSpace(min(self.num_threads, len(board_selected)),
1153fea5858eSSimon Glass                commits is not None)
1154fc3fe1c2SSimon Glass        self._PrepareOutputSpace()
1155fc3fe1c2SSimon Glass        self.SetupBuild(board_selected, commits)
1156fc3fe1c2SSimon Glass        self.ProcessResult(None)
1157fc3fe1c2SSimon Glass
1158fc3fe1c2SSimon Glass        # Create jobs to build all commits for each board
1159fc3fe1c2SSimon Glass        for brd in board_selected.itervalues():
1160190064b4SSimon Glass            job = builderthread.BuilderJob()
1161fc3fe1c2SSimon Glass            job.board = brd
1162fc3fe1c2SSimon Glass            job.commits = commits
1163fc3fe1c2SSimon Glass            job.keep_outputs = keep_outputs
1164fc3fe1c2SSimon Glass            job.step = self._step
1165fc3fe1c2SSimon Glass            self.queue.put(job)
1166fc3fe1c2SSimon Glass
1167fc3fe1c2SSimon Glass        # Wait until all jobs are started
1168fc3fe1c2SSimon Glass        self.queue.join()
1169fc3fe1c2SSimon Glass
1170fc3fe1c2SSimon Glass        # Wait until we have processed all output
1171fc3fe1c2SSimon Glass        self.out_queue.join()
11724653a882SSimon Glass        Print()
1173fc3fe1c2SSimon Glass        self.ClearLine(0)
11742c3deb97SSimon Glass        return (self.fail, self.warned)
1175