xref: /openbmc/u-boot/tools/buildman/builder.py (revision 40f11fce7c686f2b51ae109c4085fb5988c5631c)
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,
1775971ab5cSSimon Glass                 gnu_make='make', checkout=True, show_unknown=True, step=1,
178d2ce658dSSimon Glass                 no_subdirs=False, full_path=False, verbose_build=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
192bb1501f2SSimon Glass            no_subdirs: Don't create subdirectories when building current
193bb1501f2SSimon Glass                source for a single board
194bb1501f2SSimon Glass            full_path: Return the full path in CROSS_COMPILE and don't set
195bb1501f2SSimon Glass                PATH
196d2ce658dSSimon Glass            verbose_build: Run build with V=1 and don't use 'make -s'
197fc3fe1c2SSimon Glass        """
198fc3fe1c2SSimon Glass        self.toolchains = toolchains
199fc3fe1c2SSimon Glass        self.base_dir = base_dir
200fc3fe1c2SSimon Glass        self._working_dir = os.path.join(base_dir, '.bm-work')
201fc3fe1c2SSimon Glass        self.threads = []
202fc3fe1c2SSimon Glass        self.active = True
203fc3fe1c2SSimon Glass        self.do_make = self.Make
20499796923SMasahiro Yamada        self.gnu_make = gnu_make
205fc3fe1c2SSimon Glass        self.checkout = checkout
206fc3fe1c2SSimon Glass        self.num_threads = num_threads
207fc3fe1c2SSimon Glass        self.num_jobs = num_jobs
208fc3fe1c2SSimon Glass        self.already_done = 0
209fc3fe1c2SSimon Glass        self.force_build = False
210fc3fe1c2SSimon Glass        self.git_dir = git_dir
211fc3fe1c2SSimon Glass        self._show_unknown = show_unknown
212fc3fe1c2SSimon Glass        self._timestamp_count = 10
213fc3fe1c2SSimon Glass        self._build_period_us = None
214fc3fe1c2SSimon Glass        self._complete_delay = None
215fc3fe1c2SSimon Glass        self._next_delay_update = datetime.now()
216fc3fe1c2SSimon Glass        self.force_config_on_failure = True
2174266dc28SSimon Glass        self.force_build_failures = False
21897e91526SSimon Glass        self.force_reconfig = False
219fc3fe1c2SSimon Glass        self._step = step
220189a4968SSimon Glass        self.in_tree = False
22128370c1bSSimon Glass        self._error_lines = 0
2225971ab5cSSimon Glass        self.no_subdirs = no_subdirs
223bb1501f2SSimon Glass        self.full_path = full_path
224d2ce658dSSimon Glass        self.verbose_build = verbose_build
225fc3fe1c2SSimon Glass
226fc3fe1c2SSimon Glass        self.col = terminal.Color()
227fc3fe1c2SSimon Glass
228e30965dbSSimon Glass        self._re_function = re.compile('(.*): In function.*')
229e30965dbSSimon Glass        self._re_files = re.compile('In file included from.*')
230e30965dbSSimon Glass        self._re_warning = re.compile('(.*):(\d*):(\d*): warning: .*')
231e30965dbSSimon Glass        self._re_note = re.compile('(.*):(\d*):(\d*): note: this is the location of the previous.*')
232e30965dbSSimon Glass
233fc3fe1c2SSimon Glass        self.queue = Queue.Queue()
234fc3fe1c2SSimon Glass        self.out_queue = Queue.Queue()
235fc3fe1c2SSimon Glass        for i in range(self.num_threads):
236190064b4SSimon Glass            t = builderthread.BuilderThread(self, i)
237fc3fe1c2SSimon Glass            t.setDaemon(True)
238fc3fe1c2SSimon Glass            t.start()
239fc3fe1c2SSimon Glass            self.threads.append(t)
240fc3fe1c2SSimon Glass
241fc3fe1c2SSimon Glass        self.last_line_len = 0
242190064b4SSimon Glass        t = builderthread.ResultThread(self)
243fc3fe1c2SSimon Glass        t.setDaemon(True)
244fc3fe1c2SSimon Glass        t.start()
245fc3fe1c2SSimon Glass        self.threads.append(t)
246fc3fe1c2SSimon Glass
247fc3fe1c2SSimon Glass        ignore_lines = ['(make.*Waiting for unfinished)', '(Segmentation fault)']
248fc3fe1c2SSimon Glass        self.re_make_err = re.compile('|'.join(ignore_lines))
249fc3fe1c2SSimon Glass
250fc3fe1c2SSimon Glass    def __del__(self):
251fc3fe1c2SSimon Glass        """Get rid of all threads created by the builder"""
252fc3fe1c2SSimon Glass        for t in self.threads:
253fc3fe1c2SSimon Glass            del t
254fc3fe1c2SSimon Glass
255b2ea7ab2SSimon Glass    def SetDisplayOptions(self, show_errors=False, show_sizes=False,
256ed966657SSimon Glass                          show_detail=False, show_bloat=False,
257ed966657SSimon Glass                          list_error_boards=False):
258b2ea7ab2SSimon Glass        """Setup display options for the builder.
259b2ea7ab2SSimon Glass
260b2ea7ab2SSimon Glass        show_errors: True to show summarised error/warning info
261b2ea7ab2SSimon Glass        show_sizes: Show size deltas
262b2ea7ab2SSimon Glass        show_detail: Show detail for each board
263b2ea7ab2SSimon Glass        show_bloat: Show detail for each function
264ed966657SSimon Glass        list_error_boards: Show the boards which caused each error/warning
265b2ea7ab2SSimon Glass        """
266b2ea7ab2SSimon Glass        self._show_errors = show_errors
267b2ea7ab2SSimon Glass        self._show_sizes = show_sizes
268b2ea7ab2SSimon Glass        self._show_detail = show_detail
269b2ea7ab2SSimon Glass        self._show_bloat = show_bloat
270ed966657SSimon Glass        self._list_error_boards = list_error_boards
271b2ea7ab2SSimon Glass
272fc3fe1c2SSimon Glass    def _AddTimestamp(self):
273fc3fe1c2SSimon Glass        """Add a new timestamp to the list and record the build period.
274fc3fe1c2SSimon Glass
275fc3fe1c2SSimon Glass        The build period is the length of time taken to perform a single
276fc3fe1c2SSimon Glass        build (one board, one commit).
277fc3fe1c2SSimon Glass        """
278fc3fe1c2SSimon Glass        now = datetime.now()
279fc3fe1c2SSimon Glass        self._timestamps.append(now)
280fc3fe1c2SSimon Glass        count = len(self._timestamps)
281fc3fe1c2SSimon Glass        delta = self._timestamps[-1] - self._timestamps[0]
282fc3fe1c2SSimon Glass        seconds = delta.total_seconds()
283fc3fe1c2SSimon Glass
284fc3fe1c2SSimon Glass        # If we have enough data, estimate build period (time taken for a
285fc3fe1c2SSimon Glass        # single build) and therefore completion time.
286fc3fe1c2SSimon Glass        if count > 1 and self._next_delay_update < now:
287fc3fe1c2SSimon Glass            self._next_delay_update = now + timedelta(seconds=2)
288fc3fe1c2SSimon Glass            if seconds > 0:
289fc3fe1c2SSimon Glass                self._build_period = float(seconds) / count
290fc3fe1c2SSimon Glass                todo = self.count - self.upto
291fc3fe1c2SSimon Glass                self._complete_delay = timedelta(microseconds=
292fc3fe1c2SSimon Glass                        self._build_period * todo * 1000000)
293fc3fe1c2SSimon Glass                # Round it
294fc3fe1c2SSimon Glass                self._complete_delay -= timedelta(
295fc3fe1c2SSimon Glass                        microseconds=self._complete_delay.microseconds)
296fc3fe1c2SSimon Glass
297fc3fe1c2SSimon Glass        if seconds > 60:
298fc3fe1c2SSimon Glass            self._timestamps.popleft()
299fc3fe1c2SSimon Glass            count -= 1
300fc3fe1c2SSimon Glass
301fc3fe1c2SSimon Glass    def ClearLine(self, length):
302fc3fe1c2SSimon Glass        """Clear any characters on the current line
303fc3fe1c2SSimon Glass
304fc3fe1c2SSimon Glass        Make way for a new line of length 'length', by outputting enough
305fc3fe1c2SSimon Glass        spaces to clear out the old line. Then remember the new length for
306fc3fe1c2SSimon Glass        next time.
307fc3fe1c2SSimon Glass
308fc3fe1c2SSimon Glass        Args:
309fc3fe1c2SSimon Glass            length: Length of new line, in characters
310fc3fe1c2SSimon Glass        """
311fc3fe1c2SSimon Glass        if length < self.last_line_len:
3124653a882SSimon Glass            Print(' ' * (self.last_line_len - length), newline=False)
3134653a882SSimon Glass            Print('\r', newline=False)
314fc3fe1c2SSimon Glass        self.last_line_len = length
315fc3fe1c2SSimon Glass        sys.stdout.flush()
316fc3fe1c2SSimon Glass
317fc3fe1c2SSimon Glass    def SelectCommit(self, commit, checkout=True):
318fc3fe1c2SSimon Glass        """Checkout the selected commit for this build
319fc3fe1c2SSimon Glass        """
320fc3fe1c2SSimon Glass        self.commit = commit
321fc3fe1c2SSimon Glass        if checkout and self.checkout:
322fc3fe1c2SSimon Glass            gitutil.Checkout(commit.hash)
323fc3fe1c2SSimon Glass
324fc3fe1c2SSimon Glass    def Make(self, commit, brd, stage, cwd, *args, **kwargs):
325fc3fe1c2SSimon Glass        """Run make
326fc3fe1c2SSimon Glass
327fc3fe1c2SSimon Glass        Args:
328fc3fe1c2SSimon Glass            commit: Commit object that is being built
329fc3fe1c2SSimon Glass            brd: Board object that is being built
330fd18a89eSRoger Meier            stage: Stage that we are at (mrproper, config, build)
331fc3fe1c2SSimon Glass            cwd: Directory where make should be run
332fc3fe1c2SSimon Glass            args: Arguments to pass to make
333fc3fe1c2SSimon Glass            kwargs: Arguments to pass to command.RunPipe()
334fc3fe1c2SSimon Glass        """
33599796923SMasahiro Yamada        cmd = [self.gnu_make] + list(args)
336fc3fe1c2SSimon Glass        result = command.RunPipe([cmd], capture=True, capture_stderr=True,
337fc3fe1c2SSimon Glass                cwd=cwd, raise_on_error=False, **kwargs)
338*40f11fceSSimon Glass        if self.verbose_build:
339*40f11fceSSimon Glass            result.stdout = '%s\n' % (' '.join(cmd)) + result.stdout
340*40f11fceSSimon Glass            result.combined = '%s\n' % (' '.join(cmd)) + result.combined
341fc3fe1c2SSimon Glass        return result
342fc3fe1c2SSimon Glass
343fc3fe1c2SSimon Glass    def ProcessResult(self, result):
344fc3fe1c2SSimon Glass        """Process the result of a build, showing progress information
345fc3fe1c2SSimon Glass
346fc3fe1c2SSimon Glass        Args:
347e5a0e5d8SSimon Glass            result: A CommandResult object, which indicates the result for
348e5a0e5d8SSimon Glass                    a single build
349fc3fe1c2SSimon Glass        """
350fc3fe1c2SSimon Glass        col = terminal.Color()
351fc3fe1c2SSimon Glass        if result:
352fc3fe1c2SSimon Glass            target = result.brd.target
353fc3fe1c2SSimon Glass
354fc3fe1c2SSimon Glass            if result.return_code < 0:
355fc3fe1c2SSimon Glass                self.active = False
356fc3fe1c2SSimon Glass                command.StopAll()
357fc3fe1c2SSimon Glass                return
358fc3fe1c2SSimon Glass
359fc3fe1c2SSimon Glass            self.upto += 1
360fc3fe1c2SSimon Glass            if result.return_code != 0:
361fc3fe1c2SSimon Glass                self.fail += 1
362fc3fe1c2SSimon Glass            elif result.stderr:
363fc3fe1c2SSimon Glass                self.warned += 1
364fc3fe1c2SSimon Glass            if result.already_done:
365fc3fe1c2SSimon Glass                self.already_done += 1
366e5a0e5d8SSimon Glass            if self._verbose:
3674653a882SSimon Glass                Print('\r', newline=False)
368e5a0e5d8SSimon Glass                self.ClearLine(0)
369e5a0e5d8SSimon Glass                boards_selected = {target : result.brd}
370e5a0e5d8SSimon Glass                self.ResetResultSummary(boards_selected)
371e5a0e5d8SSimon Glass                self.ProduceResultSummary(result.commit_upto, self.commits,
372e5a0e5d8SSimon Glass                                          boards_selected)
373fc3fe1c2SSimon Glass        else:
374fc3fe1c2SSimon Glass            target = '(starting)'
375fc3fe1c2SSimon Glass
376fc3fe1c2SSimon Glass        # Display separate counts for ok, warned and fail
377fc3fe1c2SSimon Glass        ok = self.upto - self.warned - self.fail
378fc3fe1c2SSimon Glass        line = '\r' + self.col.Color(self.col.GREEN, '%5d' % ok)
379fc3fe1c2SSimon Glass        line += self.col.Color(self.col.YELLOW, '%5d' % self.warned)
380fc3fe1c2SSimon Glass        line += self.col.Color(self.col.RED, '%5d' % self.fail)
381fc3fe1c2SSimon Glass
382fc3fe1c2SSimon Glass        name = ' /%-5d  ' % self.count
383fc3fe1c2SSimon Glass
384fc3fe1c2SSimon Glass        # Add our current completion time estimate
385fc3fe1c2SSimon Glass        self._AddTimestamp()
386fc3fe1c2SSimon Glass        if self._complete_delay:
387fc3fe1c2SSimon Glass            name += '%s  : ' % self._complete_delay
388fc3fe1c2SSimon Glass        # When building all boards for a commit, we can print a commit
389fc3fe1c2SSimon Glass        # progress message.
390fc3fe1c2SSimon Glass        if result and result.commit_upto is None:
391fc3fe1c2SSimon Glass            name += 'commit %2d/%-3d' % (self.commit_upto + 1,
392fc3fe1c2SSimon Glass                    self.commit_count)
393fc3fe1c2SSimon Glass
394fc3fe1c2SSimon Glass        name += target
3954653a882SSimon Glass        Print(line + name, newline=False)
396e5a0e5d8SSimon Glass        length = 14 + len(name)
397fc3fe1c2SSimon Glass        self.ClearLine(length)
398fc3fe1c2SSimon Glass
399fc3fe1c2SSimon Glass    def _GetOutputDir(self, commit_upto):
400fc3fe1c2SSimon Glass        """Get the name of the output directory for a commit number
401fc3fe1c2SSimon Glass
402fc3fe1c2SSimon Glass        The output directory is typically .../<branch>/<commit>.
403fc3fe1c2SSimon Glass
404fc3fe1c2SSimon Glass        Args:
405fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
406fc3fe1c2SSimon Glass        """
4075971ab5cSSimon Glass        commit_dir = None
408fea5858eSSimon Glass        if self.commits:
409fc3fe1c2SSimon Glass            commit = self.commits[commit_upto]
410fc3fe1c2SSimon Glass            subject = commit.subject.translate(trans_valid_chars)
411fc3fe1c2SSimon Glass            commit_dir = ('%02d_of_%02d_g%s_%s' % (commit_upto + 1,
412fc3fe1c2SSimon Glass                    self.commit_count, commit.hash, subject[:20]))
4135971ab5cSSimon Glass        elif not self.no_subdirs:
414fea5858eSSimon Glass            commit_dir = 'current'
4155971ab5cSSimon Glass        if not commit_dir:
4165971ab5cSSimon Glass            return self.base_dir
4175971ab5cSSimon Glass        return os.path.join(self.base_dir, commit_dir)
418fc3fe1c2SSimon Glass
419fc3fe1c2SSimon Glass    def GetBuildDir(self, commit_upto, target):
420fc3fe1c2SSimon Glass        """Get the name of the build directory for a commit number
421fc3fe1c2SSimon Glass
422fc3fe1c2SSimon Glass        The build directory is typically .../<branch>/<commit>/<target>.
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        output_dir = self._GetOutputDir(commit_upto)
429fc3fe1c2SSimon Glass        return os.path.join(output_dir, target)
430fc3fe1c2SSimon Glass
431fc3fe1c2SSimon Glass    def GetDoneFile(self, commit_upto, target):
432fc3fe1c2SSimon Glass        """Get the name of the done file for a commit number
433fc3fe1c2SSimon Glass
434fc3fe1c2SSimon Glass        Args:
435fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
436fc3fe1c2SSimon Glass            target: Target name
437fc3fe1c2SSimon Glass        """
438fc3fe1c2SSimon Glass        return os.path.join(self.GetBuildDir(commit_upto, target), 'done')
439fc3fe1c2SSimon Glass
440fc3fe1c2SSimon Glass    def GetSizesFile(self, commit_upto, target):
441fc3fe1c2SSimon Glass        """Get the name of the sizes file for a commit number
442fc3fe1c2SSimon Glass
443fc3fe1c2SSimon Glass        Args:
444fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
445fc3fe1c2SSimon Glass            target: Target name
446fc3fe1c2SSimon Glass        """
447fc3fe1c2SSimon Glass        return os.path.join(self.GetBuildDir(commit_upto, target), 'sizes')
448fc3fe1c2SSimon Glass
449fc3fe1c2SSimon Glass    def GetFuncSizesFile(self, commit_upto, target, elf_fname):
450fc3fe1c2SSimon Glass        """Get the name of the funcsizes file for a commit number and ELF file
451fc3fe1c2SSimon Glass
452fc3fe1c2SSimon Glass        Args:
453fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
454fc3fe1c2SSimon Glass            target: Target name
455fc3fe1c2SSimon Glass            elf_fname: Filename of elf image
456fc3fe1c2SSimon Glass        """
457fc3fe1c2SSimon Glass        return os.path.join(self.GetBuildDir(commit_upto, target),
458fc3fe1c2SSimon Glass                            '%s.sizes' % elf_fname.replace('/', '-'))
459fc3fe1c2SSimon Glass
460fc3fe1c2SSimon Glass    def GetObjdumpFile(self, commit_upto, target, elf_fname):
461fc3fe1c2SSimon Glass        """Get the name of the objdump file for a commit number and ELF file
462fc3fe1c2SSimon Glass
463fc3fe1c2SSimon Glass        Args:
464fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
465fc3fe1c2SSimon Glass            target: Target name
466fc3fe1c2SSimon Glass            elf_fname: Filename of elf image
467fc3fe1c2SSimon Glass        """
468fc3fe1c2SSimon Glass        return os.path.join(self.GetBuildDir(commit_upto, target),
469fc3fe1c2SSimon Glass                            '%s.objdump' % elf_fname.replace('/', '-'))
470fc3fe1c2SSimon Glass
471fc3fe1c2SSimon Glass    def GetErrFile(self, commit_upto, target):
472fc3fe1c2SSimon Glass        """Get the name of the err file for a commit number
473fc3fe1c2SSimon Glass
474fc3fe1c2SSimon Glass        Args:
475fc3fe1c2SSimon Glass            commit_upto: Commit number to use (0..self.count-1)
476fc3fe1c2SSimon Glass            target: Target name
477fc3fe1c2SSimon Glass        """
478fc3fe1c2SSimon Glass        output_dir = self.GetBuildDir(commit_upto, target)
479fc3fe1c2SSimon Glass        return os.path.join(output_dir, 'err')
480fc3fe1c2SSimon Glass
481fc3fe1c2SSimon Glass    def FilterErrors(self, lines):
482fc3fe1c2SSimon Glass        """Filter out errors in which we have no interest
483fc3fe1c2SSimon Glass
484fc3fe1c2SSimon Glass        We should probably use map().
485fc3fe1c2SSimon Glass
486fc3fe1c2SSimon Glass        Args:
487fc3fe1c2SSimon Glass            lines: List of error lines, each a string
488fc3fe1c2SSimon Glass        Returns:
489fc3fe1c2SSimon Glass            New list with only interesting lines included
490fc3fe1c2SSimon Glass        """
491fc3fe1c2SSimon Glass        out_lines = []
492fc3fe1c2SSimon Glass        for line in lines:
493fc3fe1c2SSimon Glass            if not self.re_make_err.search(line):
494fc3fe1c2SSimon Glass                out_lines.append(line)
495fc3fe1c2SSimon Glass        return out_lines
496fc3fe1c2SSimon Glass
497fc3fe1c2SSimon Glass    def ReadFuncSizes(self, fname, fd):
498fc3fe1c2SSimon Glass        """Read function sizes from the output of 'nm'
499fc3fe1c2SSimon Glass
500fc3fe1c2SSimon Glass        Args:
501fc3fe1c2SSimon Glass            fd: File containing data to read
502fc3fe1c2SSimon Glass            fname: Filename we are reading from (just for errors)
503fc3fe1c2SSimon Glass
504fc3fe1c2SSimon Glass        Returns:
505fc3fe1c2SSimon Glass            Dictionary containing size of each function in bytes, indexed by
506fc3fe1c2SSimon Glass            function name.
507fc3fe1c2SSimon Glass        """
508fc3fe1c2SSimon Glass        sym = {}
509fc3fe1c2SSimon Glass        for line in fd.readlines():
510fc3fe1c2SSimon Glass            try:
511fc3fe1c2SSimon Glass                size, type, name = line[:-1].split()
512fc3fe1c2SSimon Glass            except:
5134653a882SSimon Glass                Print("Invalid line in file '%s': '%s'" % (fname, line[:-1]))
514fc3fe1c2SSimon Glass                continue
515fc3fe1c2SSimon Glass            if type in 'tTdDbB':
516fc3fe1c2SSimon Glass                # function names begin with '.' on 64-bit powerpc
517fc3fe1c2SSimon Glass                if '.' in name[1:]:
518fc3fe1c2SSimon Glass                    name = 'static.' + name.split('.')[0]
519fc3fe1c2SSimon Glass                sym[name] = sym.get(name, 0) + int(size, 16)
520fc3fe1c2SSimon Glass        return sym
521fc3fe1c2SSimon Glass
522fc3fe1c2SSimon Glass    def GetBuildOutcome(self, commit_upto, target, read_func_sizes):
523fc3fe1c2SSimon Glass        """Work out the outcome of a build.
524fc3fe1c2SSimon Glass
525fc3fe1c2SSimon Glass        Args:
526fc3fe1c2SSimon Glass            commit_upto: Commit number to check (0..n-1)
527fc3fe1c2SSimon Glass            target: Target board to check
528fc3fe1c2SSimon Glass            read_func_sizes: True to read function size information
529fc3fe1c2SSimon Glass
530fc3fe1c2SSimon Glass        Returns:
531fc3fe1c2SSimon Glass            Outcome object
532fc3fe1c2SSimon Glass        """
533fc3fe1c2SSimon Glass        done_file = self.GetDoneFile(commit_upto, target)
534fc3fe1c2SSimon Glass        sizes_file = self.GetSizesFile(commit_upto, target)
535fc3fe1c2SSimon Glass        sizes = {}
536fc3fe1c2SSimon Glass        func_sizes = {}
537fc3fe1c2SSimon Glass        if os.path.exists(done_file):
538fc3fe1c2SSimon Glass            with open(done_file, 'r') as fd:
539fc3fe1c2SSimon Glass                return_code = int(fd.readline())
540fc3fe1c2SSimon Glass                err_lines = []
541fc3fe1c2SSimon Glass                err_file = self.GetErrFile(commit_upto, target)
542fc3fe1c2SSimon Glass                if os.path.exists(err_file):
543fc3fe1c2SSimon Glass                    with open(err_file, 'r') as fd:
544fc3fe1c2SSimon Glass                        err_lines = self.FilterErrors(fd.readlines())
545fc3fe1c2SSimon Glass
546fc3fe1c2SSimon Glass                # Decide whether the build was ok, failed or created warnings
547fc3fe1c2SSimon Glass                if return_code:
548fc3fe1c2SSimon Glass                    rc = OUTCOME_ERROR
549fc3fe1c2SSimon Glass                elif len(err_lines):
550fc3fe1c2SSimon Glass                    rc = OUTCOME_WARNING
551fc3fe1c2SSimon Glass                else:
552fc3fe1c2SSimon Glass                    rc = OUTCOME_OK
553fc3fe1c2SSimon Glass
554fc3fe1c2SSimon Glass                # Convert size information to our simple format
555fc3fe1c2SSimon Glass                if os.path.exists(sizes_file):
556fc3fe1c2SSimon Glass                    with open(sizes_file, 'r') as fd:
557fc3fe1c2SSimon Glass                        for line in fd.readlines():
558fc3fe1c2SSimon Glass                            values = line.split()
559fc3fe1c2SSimon Glass                            rodata = 0
560fc3fe1c2SSimon Glass                            if len(values) > 6:
561fc3fe1c2SSimon Glass                                rodata = int(values[6], 16)
562fc3fe1c2SSimon Glass                            size_dict = {
563fc3fe1c2SSimon Glass                                'all' : int(values[0]) + int(values[1]) +
564fc3fe1c2SSimon Glass                                        int(values[2]),
565fc3fe1c2SSimon Glass                                'text' : int(values[0]) - rodata,
566fc3fe1c2SSimon Glass                                'data' : int(values[1]),
567fc3fe1c2SSimon Glass                                'bss' : int(values[2]),
568fc3fe1c2SSimon Glass                                'rodata' : rodata,
569fc3fe1c2SSimon Glass                            }
570fc3fe1c2SSimon Glass                            sizes[values[5]] = size_dict
571fc3fe1c2SSimon Glass
572fc3fe1c2SSimon Glass            if read_func_sizes:
573fc3fe1c2SSimon Glass                pattern = self.GetFuncSizesFile(commit_upto, target, '*')
574fc3fe1c2SSimon Glass                for fname in glob.glob(pattern):
575fc3fe1c2SSimon Glass                    with open(fname, 'r') as fd:
576fc3fe1c2SSimon Glass                        dict_name = os.path.basename(fname).replace('.sizes',
577fc3fe1c2SSimon Glass                                                                    '')
578fc3fe1c2SSimon Glass                        func_sizes[dict_name] = self.ReadFuncSizes(fname, fd)
579fc3fe1c2SSimon Glass
580fc3fe1c2SSimon Glass            return Builder.Outcome(rc, err_lines, sizes, func_sizes)
581fc3fe1c2SSimon Glass
582fc3fe1c2SSimon Glass        return Builder.Outcome(OUTCOME_UNKNOWN, [], {}, {})
583fc3fe1c2SSimon Glass
584fc3fe1c2SSimon Glass    def GetResultSummary(self, boards_selected, commit_upto, read_func_sizes):
585fc3fe1c2SSimon Glass        """Calculate a summary of the results of building a commit.
586fc3fe1c2SSimon Glass
587fc3fe1c2SSimon Glass        Args:
588fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise
589fc3fe1c2SSimon Glass            commit_upto: Commit number to summarize (0..self.count-1)
590fc3fe1c2SSimon Glass            read_func_sizes: True to read function size information
591fc3fe1c2SSimon Glass
592fc3fe1c2SSimon Glass        Returns:
593fc3fe1c2SSimon Glass            Tuple:
594fc3fe1c2SSimon Glass                Dict containing boards which passed building this commit.
595fc3fe1c2SSimon Glass                    keyed by board.target
596e30965dbSSimon Glass                List containing a summary of error lines
597ed966657SSimon Glass                Dict keyed by error line, containing a list of the Board
598ed966657SSimon Glass                    objects with that error
599e30965dbSSimon Glass                List containing a summary of warning lines
600e30965dbSSimon Glass                Dict keyed by error line, containing a list of the Board
601e30965dbSSimon Glass                    objects with that warning
602fc3fe1c2SSimon Glass        """
603e30965dbSSimon Glass        def AddLine(lines_summary, lines_boards, line, board):
604e30965dbSSimon Glass            line = line.rstrip()
605e30965dbSSimon Glass            if line in lines_boards:
606e30965dbSSimon Glass                lines_boards[line].append(board)
607e30965dbSSimon Glass            else:
608e30965dbSSimon Glass                lines_boards[line] = [board]
609e30965dbSSimon Glass                lines_summary.append(line)
610e30965dbSSimon Glass
611fc3fe1c2SSimon Glass        board_dict = {}
612fc3fe1c2SSimon Glass        err_lines_summary = []
613ed966657SSimon Glass        err_lines_boards = {}
614e30965dbSSimon Glass        warn_lines_summary = []
615e30965dbSSimon Glass        warn_lines_boards = {}
616fc3fe1c2SSimon Glass
617fc3fe1c2SSimon Glass        for board in boards_selected.itervalues():
618fc3fe1c2SSimon Glass            outcome = self.GetBuildOutcome(commit_upto, board.target,
619fc3fe1c2SSimon Glass                                           read_func_sizes)
620fc3fe1c2SSimon Glass            board_dict[board.target] = outcome
621e30965dbSSimon Glass            last_func = None
622e30965dbSSimon Glass            last_was_warning = False
623e30965dbSSimon Glass            for line in outcome.err_lines:
624e30965dbSSimon Glass                if line:
625e30965dbSSimon Glass                    if (self._re_function.match(line) or
626e30965dbSSimon Glass                            self._re_files.match(line)):
627e30965dbSSimon Glass                        last_func = line
628ed966657SSimon Glass                    else:
629e30965dbSSimon Glass                        is_warning = self._re_warning.match(line)
630e30965dbSSimon Glass                        is_note = self._re_note.match(line)
631e30965dbSSimon Glass                        if is_warning or (last_was_warning and is_note):
632e30965dbSSimon Glass                            if last_func:
633e30965dbSSimon Glass                                AddLine(warn_lines_summary, warn_lines_boards,
634e30965dbSSimon Glass                                        last_func, board)
635e30965dbSSimon Glass                            AddLine(warn_lines_summary, warn_lines_boards,
636e30965dbSSimon Glass                                    line, board)
637e30965dbSSimon Glass                        else:
638e30965dbSSimon Glass                            if last_func:
639e30965dbSSimon Glass                                AddLine(err_lines_summary, err_lines_boards,
640e30965dbSSimon Glass                                        last_func, board)
641e30965dbSSimon Glass                            AddLine(err_lines_summary, err_lines_boards,
642e30965dbSSimon Glass                                    line, board)
643e30965dbSSimon Glass                        last_was_warning = is_warning
644e30965dbSSimon Glass                        last_func = None
645e30965dbSSimon Glass        return (board_dict, err_lines_summary, err_lines_boards,
646e30965dbSSimon Glass                warn_lines_summary, warn_lines_boards)
647fc3fe1c2SSimon Glass
648fc3fe1c2SSimon Glass    def AddOutcome(self, board_dict, arch_list, changes, char, color):
649fc3fe1c2SSimon Glass        """Add an output to our list of outcomes for each architecture
650fc3fe1c2SSimon Glass
651fc3fe1c2SSimon Glass        This simple function adds failing boards (changes) to the
652fc3fe1c2SSimon Glass        relevant architecture string, so we can print the results out
653fc3fe1c2SSimon Glass        sorted by architecture.
654fc3fe1c2SSimon Glass
655fc3fe1c2SSimon Glass        Args:
656fc3fe1c2SSimon Glass             board_dict: Dict containing all boards
657fc3fe1c2SSimon Glass             arch_list: Dict keyed by arch name. Value is a string containing
658fc3fe1c2SSimon Glass                    a list of board names which failed for that arch.
659fc3fe1c2SSimon Glass             changes: List of boards to add to arch_list
660fc3fe1c2SSimon Glass             color: terminal.Colour object
661fc3fe1c2SSimon Glass        """
662fc3fe1c2SSimon Glass        done_arch = {}
663fc3fe1c2SSimon Glass        for target in changes:
664fc3fe1c2SSimon Glass            if target in board_dict:
665fc3fe1c2SSimon Glass                arch = board_dict[target].arch
666fc3fe1c2SSimon Glass            else:
667fc3fe1c2SSimon Glass                arch = 'unknown'
668fc3fe1c2SSimon Glass            str = self.col.Color(color, ' ' + target)
669fc3fe1c2SSimon Glass            if not arch in done_arch:
67063c619eeSSimon Glass                str = ' %s  %s' % (self.col.Color(color, char), str)
671fc3fe1c2SSimon Glass                done_arch[arch] = True
672fc3fe1c2SSimon Glass            if not arch in arch_list:
673fc3fe1c2SSimon Glass                arch_list[arch] = str
674fc3fe1c2SSimon Glass            else:
675fc3fe1c2SSimon Glass                arch_list[arch] += str
676fc3fe1c2SSimon Glass
677fc3fe1c2SSimon Glass
678fc3fe1c2SSimon Glass    def ColourNum(self, num):
679fc3fe1c2SSimon Glass        color = self.col.RED if num > 0 else self.col.GREEN
680fc3fe1c2SSimon Glass        if num == 0:
681fc3fe1c2SSimon Glass            return '0'
682fc3fe1c2SSimon Glass        return self.col.Color(color, str(num))
683fc3fe1c2SSimon Glass
684fc3fe1c2SSimon Glass    def ResetResultSummary(self, board_selected):
685fc3fe1c2SSimon Glass        """Reset the results summary ready for use.
686fc3fe1c2SSimon Glass
687fc3fe1c2SSimon Glass        Set up the base board list to be all those selected, and set the
688fc3fe1c2SSimon Glass        error lines to empty.
689fc3fe1c2SSimon Glass
690fc3fe1c2SSimon Glass        Following this, calls to PrintResultSummary() will use this
691fc3fe1c2SSimon Glass        information to work out what has changed.
692fc3fe1c2SSimon Glass
693fc3fe1c2SSimon Glass        Args:
694fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise, keyed by
695fc3fe1c2SSimon Glass                board.target
696fc3fe1c2SSimon Glass        """
697fc3fe1c2SSimon Glass        self._base_board_dict = {}
698fc3fe1c2SSimon Glass        for board in board_selected:
699fc3fe1c2SSimon Glass            self._base_board_dict[board] = Builder.Outcome(0, [], [], {})
700fc3fe1c2SSimon Glass        self._base_err_lines = []
701e30965dbSSimon Glass        self._base_warn_lines = []
702e30965dbSSimon Glass        self._base_err_line_boards = {}
703e30965dbSSimon Glass        self._base_warn_line_boards = {}
704fc3fe1c2SSimon Glass
705fc3fe1c2SSimon Glass    def PrintFuncSizeDetail(self, fname, old, new):
706fc3fe1c2SSimon Glass        grow, shrink, add, remove, up, down = 0, 0, 0, 0, 0, 0
707fc3fe1c2SSimon Glass        delta, common = [], {}
708fc3fe1c2SSimon Glass
709fc3fe1c2SSimon Glass        for a in old:
710fc3fe1c2SSimon Glass            if a in new:
711fc3fe1c2SSimon Glass                common[a] = 1
712fc3fe1c2SSimon Glass
713fc3fe1c2SSimon Glass        for name in old:
714fc3fe1c2SSimon Glass            if name not in common:
715fc3fe1c2SSimon Glass                remove += 1
716fc3fe1c2SSimon Glass                down += old[name]
717fc3fe1c2SSimon Glass                delta.append([-old[name], name])
718fc3fe1c2SSimon Glass
719fc3fe1c2SSimon Glass        for name in new:
720fc3fe1c2SSimon Glass            if name not in common:
721fc3fe1c2SSimon Glass                add += 1
722fc3fe1c2SSimon Glass                up += new[name]
723fc3fe1c2SSimon Glass                delta.append([new[name], name])
724fc3fe1c2SSimon Glass
725fc3fe1c2SSimon Glass        for name in common:
726fc3fe1c2SSimon Glass                diff = new.get(name, 0) - old.get(name, 0)
727fc3fe1c2SSimon Glass                if diff > 0:
728fc3fe1c2SSimon Glass                    grow, up = grow + 1, up + diff
729fc3fe1c2SSimon Glass                elif diff < 0:
730fc3fe1c2SSimon Glass                    shrink, down = shrink + 1, down - diff
731fc3fe1c2SSimon Glass                delta.append([diff, name])
732fc3fe1c2SSimon Glass
733fc3fe1c2SSimon Glass        delta.sort()
734fc3fe1c2SSimon Glass        delta.reverse()
735fc3fe1c2SSimon Glass
736fc3fe1c2SSimon Glass        args = [add, -remove, grow, -shrink, up, -down, up - down]
737fc3fe1c2SSimon Glass        if max(args) == 0:
738fc3fe1c2SSimon Glass            return
739fc3fe1c2SSimon Glass        args = [self.ColourNum(x) for x in args]
740fc3fe1c2SSimon Glass        indent = ' ' * 15
7414653a882SSimon Glass        Print('%s%s: add: %s/%s, grow: %s/%s bytes: %s/%s (%s)' %
742fc3fe1c2SSimon Glass              tuple([indent, self.col.Color(self.col.YELLOW, fname)] + args))
7434653a882SSimon Glass        Print('%s  %-38s %7s %7s %+7s' % (indent, 'function', 'old', 'new',
7444653a882SSimon Glass                                         'delta'))
745fc3fe1c2SSimon Glass        for diff, name in delta:
746fc3fe1c2SSimon Glass            if diff:
747fc3fe1c2SSimon Glass                color = self.col.RED if diff > 0 else self.col.GREEN
748fc3fe1c2SSimon Glass                msg = '%s  %-38s %7s %7s %+7d' % (indent, name,
749fc3fe1c2SSimon Glass                        old.get(name, '-'), new.get(name,'-'), diff)
7504653a882SSimon Glass                Print(msg, colour=color)
751fc3fe1c2SSimon Glass
752fc3fe1c2SSimon Glass
753fc3fe1c2SSimon Glass    def PrintSizeDetail(self, target_list, show_bloat):
754fc3fe1c2SSimon Glass        """Show details size information for each board
755fc3fe1c2SSimon Glass
756fc3fe1c2SSimon Glass        Args:
757fc3fe1c2SSimon Glass            target_list: List of targets, each a dict containing:
758fc3fe1c2SSimon Glass                    'target': Target name
759fc3fe1c2SSimon Glass                    'total_diff': Total difference in bytes across all areas
760fc3fe1c2SSimon Glass                    <part_name>: Difference for that part
761fc3fe1c2SSimon Glass            show_bloat: Show detail for each function
762fc3fe1c2SSimon Glass        """
763fc3fe1c2SSimon Glass        targets_by_diff = sorted(target_list, reverse=True,
764fc3fe1c2SSimon Glass        key=lambda x: x['_total_diff'])
765fc3fe1c2SSimon Glass        for result in targets_by_diff:
766fc3fe1c2SSimon Glass            printed_target = False
767fc3fe1c2SSimon Glass            for name in sorted(result):
768fc3fe1c2SSimon Glass                diff = result[name]
769fc3fe1c2SSimon Glass                if name.startswith('_'):
770fc3fe1c2SSimon Glass                    continue
771fc3fe1c2SSimon Glass                if diff != 0:
772fc3fe1c2SSimon Glass                    color = self.col.RED if diff > 0 else self.col.GREEN
773fc3fe1c2SSimon Glass                msg = ' %s %+d' % (name, diff)
774fc3fe1c2SSimon Glass                if not printed_target:
7754653a882SSimon Glass                    Print('%10s  %-15s:' % ('', result['_target']),
7764653a882SSimon Glass                          newline=False)
777fc3fe1c2SSimon Glass                    printed_target = True
7784653a882SSimon Glass                Print(msg, colour=color, newline=False)
779fc3fe1c2SSimon Glass            if printed_target:
7804653a882SSimon Glass                Print()
781fc3fe1c2SSimon Glass                if show_bloat:
782fc3fe1c2SSimon Glass                    target = result['_target']
783fc3fe1c2SSimon Glass                    outcome = result['_outcome']
784fc3fe1c2SSimon Glass                    base_outcome = self._base_board_dict[target]
785fc3fe1c2SSimon Glass                    for fname in outcome.func_sizes:
786fc3fe1c2SSimon Glass                        self.PrintFuncSizeDetail(fname,
787fc3fe1c2SSimon Glass                                                 base_outcome.func_sizes[fname],
788fc3fe1c2SSimon Glass                                                 outcome.func_sizes[fname])
789fc3fe1c2SSimon Glass
790fc3fe1c2SSimon Glass
791fc3fe1c2SSimon Glass    def PrintSizeSummary(self, board_selected, board_dict, show_detail,
792fc3fe1c2SSimon Glass                         show_bloat):
793fc3fe1c2SSimon Glass        """Print a summary of image sizes broken down by section.
794fc3fe1c2SSimon Glass
795fc3fe1c2SSimon Glass        The summary takes the form of one line per architecture. The
796fc3fe1c2SSimon Glass        line contains deltas for each of the sections (+ means the section
797fc3fe1c2SSimon Glass        got bigger, - means smaller). The nunmbers are the average number
798fc3fe1c2SSimon Glass        of bytes that a board in this section increased by.
799fc3fe1c2SSimon Glass
800fc3fe1c2SSimon Glass        For example:
801fc3fe1c2SSimon Glass           powerpc: (622 boards)   text -0.0
802fc3fe1c2SSimon Glass          arm: (285 boards)   text -0.0
803fc3fe1c2SSimon Glass          nds32: (3 boards)   text -8.0
804fc3fe1c2SSimon Glass
805fc3fe1c2SSimon Glass        Args:
806fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise, keyed by
807fc3fe1c2SSimon Glass                board.target
808fc3fe1c2SSimon Glass            board_dict: Dict containing boards for which we built this
809fc3fe1c2SSimon Glass                commit, keyed by board.target. The value is an Outcome object.
810fc3fe1c2SSimon Glass            show_detail: Show detail for each board
811fc3fe1c2SSimon Glass            show_bloat: Show detail for each function
812fc3fe1c2SSimon Glass        """
813fc3fe1c2SSimon Glass        arch_list = {}
814fc3fe1c2SSimon Glass        arch_count = {}
815fc3fe1c2SSimon Glass
816fc3fe1c2SSimon Glass        # Calculate changes in size for different image parts
817fc3fe1c2SSimon Glass        # The previous sizes are in Board.sizes, for each board
818fc3fe1c2SSimon Glass        for target in board_dict:
819fc3fe1c2SSimon Glass            if target not in board_selected:
820fc3fe1c2SSimon Glass                continue
821fc3fe1c2SSimon Glass            base_sizes = self._base_board_dict[target].sizes
822fc3fe1c2SSimon Glass            outcome = board_dict[target]
823fc3fe1c2SSimon Glass            sizes = outcome.sizes
824fc3fe1c2SSimon Glass
825fc3fe1c2SSimon Glass            # Loop through the list of images, creating a dict of size
826fc3fe1c2SSimon Glass            # changes for each image/part. We end up with something like
827fc3fe1c2SSimon Glass            # {'target' : 'snapper9g45, 'data' : 5, 'u-boot-spl:text' : -4}
828fc3fe1c2SSimon Glass            # which means that U-Boot data increased by 5 bytes and SPL
829fc3fe1c2SSimon Glass            # text decreased by 4.
830fc3fe1c2SSimon Glass            err = {'_target' : target}
831fc3fe1c2SSimon Glass            for image in sizes:
832fc3fe1c2SSimon Glass                if image in base_sizes:
833fc3fe1c2SSimon Glass                    base_image = base_sizes[image]
834fc3fe1c2SSimon Glass                    # Loop through the text, data, bss parts
835fc3fe1c2SSimon Glass                    for part in sorted(sizes[image]):
836fc3fe1c2SSimon Glass                        diff = sizes[image][part] - base_image[part]
837fc3fe1c2SSimon Glass                        col = None
838fc3fe1c2SSimon Glass                        if diff:
839fc3fe1c2SSimon Glass                            if image == 'u-boot':
840fc3fe1c2SSimon Glass                                name = part
841fc3fe1c2SSimon Glass                            else:
842fc3fe1c2SSimon Glass                                name = image + ':' + part
843fc3fe1c2SSimon Glass                            err[name] = diff
844fc3fe1c2SSimon Glass            arch = board_selected[target].arch
845fc3fe1c2SSimon Glass            if not arch in arch_count:
846fc3fe1c2SSimon Glass                arch_count[arch] = 1
847fc3fe1c2SSimon Glass            else:
848fc3fe1c2SSimon Glass                arch_count[arch] += 1
849fc3fe1c2SSimon Glass            if not sizes:
850fc3fe1c2SSimon Glass                pass    # Only add to our list when we have some stats
851fc3fe1c2SSimon Glass            elif not arch in arch_list:
852fc3fe1c2SSimon Glass                arch_list[arch] = [err]
853fc3fe1c2SSimon Glass            else:
854fc3fe1c2SSimon Glass                arch_list[arch].append(err)
855fc3fe1c2SSimon Glass
856fc3fe1c2SSimon Glass        # We now have a list of image size changes sorted by arch
857fc3fe1c2SSimon Glass        # Print out a summary of these
858fc3fe1c2SSimon Glass        for arch, target_list in arch_list.iteritems():
859fc3fe1c2SSimon Glass            # Get total difference for each type
860fc3fe1c2SSimon Glass            totals = {}
861fc3fe1c2SSimon Glass            for result in target_list:
862fc3fe1c2SSimon Glass                total = 0
863fc3fe1c2SSimon Glass                for name, diff in result.iteritems():
864fc3fe1c2SSimon Glass                    if name.startswith('_'):
865fc3fe1c2SSimon Glass                        continue
866fc3fe1c2SSimon Glass                    total += diff
867fc3fe1c2SSimon Glass                    if name in totals:
868fc3fe1c2SSimon Glass                        totals[name] += diff
869fc3fe1c2SSimon Glass                    else:
870fc3fe1c2SSimon Glass                        totals[name] = diff
871fc3fe1c2SSimon Glass                result['_total_diff'] = total
872fc3fe1c2SSimon Glass                result['_outcome'] = board_dict[result['_target']]
873fc3fe1c2SSimon Glass
874fc3fe1c2SSimon Glass            count = len(target_list)
875fc3fe1c2SSimon Glass            printed_arch = False
876fc3fe1c2SSimon Glass            for name in sorted(totals):
877fc3fe1c2SSimon Glass                diff = totals[name]
878fc3fe1c2SSimon Glass                if diff:
879fc3fe1c2SSimon Glass                    # Display the average difference in this name for this
880fc3fe1c2SSimon Glass                    # architecture
881fc3fe1c2SSimon Glass                    avg_diff = float(diff) / count
882fc3fe1c2SSimon Glass                    color = self.col.RED if avg_diff > 0 else self.col.GREEN
883fc3fe1c2SSimon Glass                    msg = ' %s %+1.1f' % (name, avg_diff)
884fc3fe1c2SSimon Glass                    if not printed_arch:
8854653a882SSimon Glass                        Print('%10s: (for %d/%d boards)' % (arch, count,
8864653a882SSimon Glass                              arch_count[arch]), newline=False)
887fc3fe1c2SSimon Glass                        printed_arch = True
8884653a882SSimon Glass                    Print(msg, colour=color, newline=False)
889fc3fe1c2SSimon Glass
890fc3fe1c2SSimon Glass            if printed_arch:
8914653a882SSimon Glass                Print()
892fc3fe1c2SSimon Glass                if show_detail:
893fc3fe1c2SSimon Glass                    self.PrintSizeDetail(target_list, show_bloat)
894fc3fe1c2SSimon Glass
895fc3fe1c2SSimon Glass
896fc3fe1c2SSimon Glass    def PrintResultSummary(self, board_selected, board_dict, err_lines,
897e30965dbSSimon Glass                           err_line_boards, warn_lines, warn_line_boards,
898e30965dbSSimon Glass                           show_sizes, show_detail, show_bloat):
899fc3fe1c2SSimon Glass        """Compare results with the base results and display delta.
900fc3fe1c2SSimon Glass
901fc3fe1c2SSimon Glass        Only boards mentioned in board_selected will be considered. This
902fc3fe1c2SSimon Glass        function is intended to be called repeatedly with the results of
903fc3fe1c2SSimon Glass        each commit. It therefore shows a 'diff' between what it saw in
904fc3fe1c2SSimon Glass        the last call and what it sees now.
905fc3fe1c2SSimon Glass
906fc3fe1c2SSimon Glass        Args:
907fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise, keyed by
908fc3fe1c2SSimon Glass                board.target
909fc3fe1c2SSimon Glass            board_dict: Dict containing boards for which we built this
910fc3fe1c2SSimon Glass                commit, keyed by board.target. The value is an Outcome object.
911fc3fe1c2SSimon Glass            err_lines: A list of errors for this commit, or [] if there is
912fc3fe1c2SSimon Glass                none, or we don't want to print errors
913ed966657SSimon Glass            err_line_boards: Dict keyed by error line, containing a list of
914ed966657SSimon Glass                the Board objects with that error
915e30965dbSSimon Glass            warn_lines: A list of warnings for this commit, or [] if there is
916e30965dbSSimon Glass                none, or we don't want to print errors
917e30965dbSSimon Glass            warn_line_boards: Dict keyed by warning line, containing a list of
918e30965dbSSimon Glass                the Board objects with that warning
919fc3fe1c2SSimon Glass            show_sizes: Show image size deltas
920fc3fe1c2SSimon Glass            show_detail: Show detail for each board
921fc3fe1c2SSimon Glass            show_bloat: Show detail for each function
922fc3fe1c2SSimon Glass        """
923e30965dbSSimon Glass        def _BoardList(line, line_boards):
924ed966657SSimon Glass            """Helper function to get a line of boards containing a line
925ed966657SSimon Glass
926ed966657SSimon Glass            Args:
927ed966657SSimon Glass                line: Error line to search for
928ed966657SSimon Glass            Return:
929ed966657SSimon Glass                String containing a list of boards with that error line, or
930ed966657SSimon Glass                '' if the user has not requested such a list
931ed966657SSimon Glass            """
932ed966657SSimon Glass            if self._list_error_boards:
933ed966657SSimon Glass                names = []
934e30965dbSSimon Glass                for board in line_boards[line]:
935f66153beSSimon Glass                    if not board.target in names:
936ed966657SSimon Glass                        names.append(board.target)
937ed966657SSimon Glass                names_str = '(%s) ' % ','.join(names)
938ed966657SSimon Glass            else:
939ed966657SSimon Glass                names_str = ''
940ed966657SSimon Glass            return names_str
941ed966657SSimon Glass
942e30965dbSSimon Glass        def _CalcErrorDelta(base_lines, base_line_boards, lines, line_boards,
943e30965dbSSimon Glass                            char):
944e30965dbSSimon Glass            better_lines = []
945e30965dbSSimon Glass            worse_lines = []
946e30965dbSSimon Glass            for line in lines:
947e30965dbSSimon Glass                if line not in base_lines:
948e30965dbSSimon Glass                    worse_lines.append(char + '+' +
949e30965dbSSimon Glass                            _BoardList(line, line_boards) + line)
950e30965dbSSimon Glass            for line in base_lines:
951e30965dbSSimon Glass                if line not in lines:
952e30965dbSSimon Glass                    better_lines.append(char + '-' +
953e30965dbSSimon Glass                            _BoardList(line, base_line_boards) + line)
954e30965dbSSimon Glass            return better_lines, worse_lines
955e30965dbSSimon Glass
956fc3fe1c2SSimon Glass        better = []     # List of boards fixed since last commit
957fc3fe1c2SSimon Glass        worse = []      # List of new broken boards since last commit
958fc3fe1c2SSimon Glass        new = []        # List of boards that didn't exist last time
959fc3fe1c2SSimon Glass        unknown = []    # List of boards that were not built
960fc3fe1c2SSimon Glass
961fc3fe1c2SSimon Glass        for target in board_dict:
962fc3fe1c2SSimon Glass            if target not in board_selected:
963fc3fe1c2SSimon Glass                continue
964fc3fe1c2SSimon Glass
965fc3fe1c2SSimon Glass            # If the board was built last time, add its outcome to a list
966fc3fe1c2SSimon Glass            if target in self._base_board_dict:
967fc3fe1c2SSimon Glass                base_outcome = self._base_board_dict[target].rc
968fc3fe1c2SSimon Glass                outcome = board_dict[target]
969fc3fe1c2SSimon Glass                if outcome.rc == OUTCOME_UNKNOWN:
970fc3fe1c2SSimon Glass                    unknown.append(target)
971fc3fe1c2SSimon Glass                elif outcome.rc < base_outcome:
972fc3fe1c2SSimon Glass                    better.append(target)
973fc3fe1c2SSimon Glass                elif outcome.rc > base_outcome:
974fc3fe1c2SSimon Glass                    worse.append(target)
975fc3fe1c2SSimon Glass            else:
976fc3fe1c2SSimon Glass                new.append(target)
977fc3fe1c2SSimon Glass
978fc3fe1c2SSimon Glass        # Get a list of errors that have appeared, and disappeared
979e30965dbSSimon Glass        better_err, worse_err = _CalcErrorDelta(self._base_err_lines,
980e30965dbSSimon Glass                self._base_err_line_boards, err_lines, err_line_boards, '')
981e30965dbSSimon Glass        better_warn, worse_warn = _CalcErrorDelta(self._base_warn_lines,
982e30965dbSSimon Glass                self._base_warn_line_boards, warn_lines, warn_line_boards, 'w')
983fc3fe1c2SSimon Glass
984fc3fe1c2SSimon Glass        # Display results by arch
985e30965dbSSimon Glass        if (better or worse or unknown or new or worse_err or better_err
986e30965dbSSimon Glass                or worse_warn or better_warn):
987fc3fe1c2SSimon Glass            arch_list = {}
988fc3fe1c2SSimon Glass            self.AddOutcome(board_selected, arch_list, better, '',
989fc3fe1c2SSimon Glass                    self.col.GREEN)
990fc3fe1c2SSimon Glass            self.AddOutcome(board_selected, arch_list, worse, '+',
991fc3fe1c2SSimon Glass                    self.col.RED)
992fc3fe1c2SSimon Glass            self.AddOutcome(board_selected, arch_list, new, '*', self.col.BLUE)
993fc3fe1c2SSimon Glass            if self._show_unknown:
994fc3fe1c2SSimon Glass                self.AddOutcome(board_selected, arch_list, unknown, '?',
995fc3fe1c2SSimon Glass                        self.col.MAGENTA)
996fc3fe1c2SSimon Glass            for arch, target_list in arch_list.iteritems():
9974653a882SSimon Glass                Print('%10s: %s' % (arch, target_list))
99828370c1bSSimon Glass                self._error_lines += 1
999fc3fe1c2SSimon Glass            if better_err:
10004653a882SSimon Glass                Print('\n'.join(better_err), colour=self.col.GREEN)
100128370c1bSSimon Glass                self._error_lines += 1
1002fc3fe1c2SSimon Glass            if worse_err:
10034653a882SSimon Glass                Print('\n'.join(worse_err), colour=self.col.RED)
100428370c1bSSimon Glass                self._error_lines += 1
1005e30965dbSSimon Glass            if better_warn:
10064653a882SSimon Glass                Print('\n'.join(better_warn), colour=self.col.CYAN)
1007e30965dbSSimon Glass                self._error_lines += 1
1008e30965dbSSimon Glass            if worse_warn:
10094653a882SSimon Glass                Print('\n'.join(worse_warn), colour=self.col.MAGENTA)
1010e30965dbSSimon Glass                self._error_lines += 1
1011fc3fe1c2SSimon Glass
1012fc3fe1c2SSimon Glass        if show_sizes:
1013fc3fe1c2SSimon Glass            self.PrintSizeSummary(board_selected, board_dict, show_detail,
1014fc3fe1c2SSimon Glass                                  show_bloat)
1015fc3fe1c2SSimon Glass
1016fc3fe1c2SSimon Glass        # Save our updated information for the next call to this function
1017fc3fe1c2SSimon Glass        self._base_board_dict = board_dict
1018fc3fe1c2SSimon Glass        self._base_err_lines = err_lines
1019e30965dbSSimon Glass        self._base_warn_lines = warn_lines
1020e30965dbSSimon Glass        self._base_err_line_boards = err_line_boards
1021e30965dbSSimon Glass        self._base_warn_line_boards = warn_line_boards
1022fc3fe1c2SSimon Glass
1023fc3fe1c2SSimon Glass        # Get a list of boards that did not get built, if needed
1024fc3fe1c2SSimon Glass        not_built = []
1025fc3fe1c2SSimon Glass        for board in board_selected:
1026fc3fe1c2SSimon Glass            if not board in board_dict:
1027fc3fe1c2SSimon Glass                not_built.append(board)
1028fc3fe1c2SSimon Glass        if not_built:
10294653a882SSimon Glass            Print("Boards not built (%d): %s" % (len(not_built),
10304653a882SSimon Glass                  ', '.join(not_built)))
1031fc3fe1c2SSimon Glass
1032b2ea7ab2SSimon Glass    def ProduceResultSummary(self, commit_upto, commits, board_selected):
1033e30965dbSSimon Glass            (board_dict, err_lines, err_line_boards, warn_lines,
1034e30965dbSSimon Glass                    warn_line_boards) = self.GetResultSummary(
1035ed966657SSimon Glass                    board_selected, commit_upto,
1036ed966657SSimon Glass                    read_func_sizes=self._show_bloat)
1037b2ea7ab2SSimon Glass            if commits:
1038b2ea7ab2SSimon Glass                msg = '%02d: %s' % (commit_upto + 1,
1039b2ea7ab2SSimon Glass                        commits[commit_upto].subject)
10404653a882SSimon Glass                Print(msg, colour=self.col.BLUE)
1041b2ea7ab2SSimon Glass            self.PrintResultSummary(board_selected, board_dict,
1042ed966657SSimon Glass                    err_lines if self._show_errors else [], err_line_boards,
1043e30965dbSSimon Glass                    warn_lines if self._show_errors else [], warn_line_boards,
1044b2ea7ab2SSimon Glass                    self._show_sizes, self._show_detail, self._show_bloat)
1045fc3fe1c2SSimon Glass
1046b2ea7ab2SSimon Glass    def ShowSummary(self, commits, board_selected):
1047fc3fe1c2SSimon Glass        """Show a build summary for U-Boot for a given board list.
1048fc3fe1c2SSimon Glass
1049fc3fe1c2SSimon Glass        Reset the result summary, then repeatedly call GetResultSummary on
1050fc3fe1c2SSimon Glass        each commit's results, then display the differences we see.
1051fc3fe1c2SSimon Glass
1052fc3fe1c2SSimon Glass        Args:
1053fc3fe1c2SSimon Glass            commit: Commit objects to summarise
1054fc3fe1c2SSimon Glass            board_selected: Dict containing boards to summarise
1055fc3fe1c2SSimon Glass        """
1056fea5858eSSimon Glass        self.commit_count = len(commits) if commits else 1
1057fc3fe1c2SSimon Glass        self.commits = commits
1058fc3fe1c2SSimon Glass        self.ResetResultSummary(board_selected)
105928370c1bSSimon Glass        self._error_lines = 0
1060fc3fe1c2SSimon Glass
1061fc3fe1c2SSimon Glass        for commit_upto in range(0, self.commit_count, self._step):
1062b2ea7ab2SSimon Glass            self.ProduceResultSummary(commit_upto, commits, board_selected)
106328370c1bSSimon Glass        if not self._error_lines:
10644653a882SSimon Glass            Print('(no errors to report)', colour=self.col.GREEN)
1065fc3fe1c2SSimon Glass
1066fc3fe1c2SSimon Glass
1067fc3fe1c2SSimon Glass    def SetupBuild(self, board_selected, commits):
1068fc3fe1c2SSimon Glass        """Set up ready to start a build.
1069fc3fe1c2SSimon Glass
1070fc3fe1c2SSimon Glass        Args:
1071fc3fe1c2SSimon Glass            board_selected: Selected boards to build
1072fc3fe1c2SSimon Glass            commits: Selected commits to build
1073fc3fe1c2SSimon Glass        """
1074fc3fe1c2SSimon Glass        # First work out how many commits we will build
1075fea5858eSSimon Glass        count = (self.commit_count + self._step - 1) / self._step
1076fc3fe1c2SSimon Glass        self.count = len(board_selected) * count
1077fc3fe1c2SSimon Glass        self.upto = self.warned = self.fail = 0
1078fc3fe1c2SSimon Glass        self._timestamps = collections.deque()
1079fc3fe1c2SSimon Glass
1080fc3fe1c2SSimon Glass    def GetThreadDir(self, thread_num):
1081fc3fe1c2SSimon Glass        """Get the directory path to the working dir for a thread.
1082fc3fe1c2SSimon Glass
1083fc3fe1c2SSimon Glass        Args:
1084fc3fe1c2SSimon Glass            thread_num: Number of thread to check.
1085fc3fe1c2SSimon Glass        """
1086fc3fe1c2SSimon Glass        return os.path.join(self._working_dir, '%02d' % thread_num)
1087fc3fe1c2SSimon Glass
1088fea5858eSSimon Glass    def _PrepareThread(self, thread_num, setup_git):
1089fc3fe1c2SSimon Glass        """Prepare the working directory for a thread.
1090fc3fe1c2SSimon Glass
1091fc3fe1c2SSimon Glass        This clones or fetches the repo into the thread's work directory.
1092fc3fe1c2SSimon Glass
1093fc3fe1c2SSimon Glass        Args:
1094fc3fe1c2SSimon Glass            thread_num: Thread number (0, 1, ...)
1095fea5858eSSimon Glass            setup_git: True to set up a git repo clone
1096fc3fe1c2SSimon Glass        """
1097fc3fe1c2SSimon Glass        thread_dir = self.GetThreadDir(thread_num)
1098190064b4SSimon Glass        builderthread.Mkdir(thread_dir)
1099fc3fe1c2SSimon Glass        git_dir = os.path.join(thread_dir, '.git')
1100fc3fe1c2SSimon Glass
1101fc3fe1c2SSimon Glass        # Clone the repo if it doesn't already exist
1102fc3fe1c2SSimon Glass        # TODO(sjg@chromium): Perhaps some git hackery to symlink instead, so
1103fc3fe1c2SSimon Glass        # we have a private index but uses the origin repo's contents?
1104fea5858eSSimon Glass        if setup_git and self.git_dir:
1105fc3fe1c2SSimon Glass            src_dir = os.path.abspath(self.git_dir)
1106fc3fe1c2SSimon Glass            if os.path.exists(git_dir):
1107fc3fe1c2SSimon Glass                gitutil.Fetch(git_dir, thread_dir)
1108fc3fe1c2SSimon Glass            else:
11094653a882SSimon Glass                Print('Cloning repo for thread %d' % thread_num)
1110fc3fe1c2SSimon Glass                gitutil.Clone(src_dir, thread_dir)
1111fc3fe1c2SSimon Glass
1112fea5858eSSimon Glass    def _PrepareWorkingSpace(self, max_threads, setup_git):
1113fc3fe1c2SSimon Glass        """Prepare the working directory for use.
1114fc3fe1c2SSimon Glass
1115fc3fe1c2SSimon Glass        Set up the git repo for each thread.
1116fc3fe1c2SSimon Glass
1117fc3fe1c2SSimon Glass        Args:
1118fc3fe1c2SSimon Glass            max_threads: Maximum number of threads we expect to need.
1119fea5858eSSimon Glass            setup_git: True to set up a git repo clone
1120fc3fe1c2SSimon Glass        """
1121190064b4SSimon Glass        builderthread.Mkdir(self._working_dir)
1122fc3fe1c2SSimon Glass        for thread in range(max_threads):
1123fea5858eSSimon Glass            self._PrepareThread(thread, setup_git)
1124fc3fe1c2SSimon Glass
1125fc3fe1c2SSimon Glass    def _PrepareOutputSpace(self):
1126fc3fe1c2SSimon Glass        """Get the output directories ready to receive files.
1127fc3fe1c2SSimon Glass
1128fc3fe1c2SSimon Glass        We delete any output directories which look like ones we need to
1129fc3fe1c2SSimon Glass        create. Having left over directories is confusing when the user wants
1130fc3fe1c2SSimon Glass        to check the output manually.
1131fc3fe1c2SSimon Glass        """
11321a915675SSimon Glass        if not self.commits:
11331a915675SSimon Glass            return
1134fc3fe1c2SSimon Glass        dir_list = []
1135fc3fe1c2SSimon Glass        for commit_upto in range(self.commit_count):
1136fc3fe1c2SSimon Glass            dir_list.append(self._GetOutputDir(commit_upto))
1137fc3fe1c2SSimon Glass
1138fc3fe1c2SSimon Glass        for dirname in glob.glob(os.path.join(self.base_dir, '*')):
1139fc3fe1c2SSimon Glass            if dirname not in dir_list:
1140fc3fe1c2SSimon Glass                shutil.rmtree(dirname)
1141fc3fe1c2SSimon Glass
1142e5a0e5d8SSimon Glass    def BuildBoards(self, commits, board_selected, keep_outputs, verbose):
1143fc3fe1c2SSimon Glass        """Build all commits for a list of boards
1144fc3fe1c2SSimon Glass
1145fc3fe1c2SSimon Glass        Args:
1146fc3fe1c2SSimon Glass            commits: List of commits to be build, each a Commit object
1147fc3fe1c2SSimon Glass            boards_selected: Dict of selected boards, key is target name,
1148fc3fe1c2SSimon Glass                    value is Board object
1149fc3fe1c2SSimon Glass            keep_outputs: True to save build output files
1150e5a0e5d8SSimon Glass            verbose: Display build results as they are completed
11512c3deb97SSimon Glass        Returns:
11522c3deb97SSimon Glass            Tuple containing:
11532c3deb97SSimon Glass                - number of boards that failed to build
11542c3deb97SSimon Glass                - number of boards that issued warnings
1155fc3fe1c2SSimon Glass        """
1156fea5858eSSimon Glass        self.commit_count = len(commits) if commits else 1
1157fc3fe1c2SSimon Glass        self.commits = commits
1158e5a0e5d8SSimon Glass        self._verbose = verbose
1159fc3fe1c2SSimon Glass
1160fc3fe1c2SSimon Glass        self.ResetResultSummary(board_selected)
1161f3d015cbSThierry Reding        builderthread.Mkdir(self.base_dir, parents = True)
1162fea5858eSSimon Glass        self._PrepareWorkingSpace(min(self.num_threads, len(board_selected)),
1163fea5858eSSimon Glass                commits is not None)
1164fc3fe1c2SSimon Glass        self._PrepareOutputSpace()
1165fc3fe1c2SSimon Glass        self.SetupBuild(board_selected, commits)
1166fc3fe1c2SSimon Glass        self.ProcessResult(None)
1167fc3fe1c2SSimon Glass
1168fc3fe1c2SSimon Glass        # Create jobs to build all commits for each board
1169fc3fe1c2SSimon Glass        for brd in board_selected.itervalues():
1170190064b4SSimon Glass            job = builderthread.BuilderJob()
1171fc3fe1c2SSimon Glass            job.board = brd
1172fc3fe1c2SSimon Glass            job.commits = commits
1173fc3fe1c2SSimon Glass            job.keep_outputs = keep_outputs
1174fc3fe1c2SSimon Glass            job.step = self._step
1175fc3fe1c2SSimon Glass            self.queue.put(job)
1176fc3fe1c2SSimon Glass
1177fc3fe1c2SSimon Glass        # Wait until all jobs are started
1178fc3fe1c2SSimon Glass        self.queue.join()
1179fc3fe1c2SSimon Glass
1180fc3fe1c2SSimon Glass        # Wait until we have processed all output
1181fc3fe1c2SSimon Glass        self.out_queue.join()
11824653a882SSimon Glass        Print()
1183fc3fe1c2SSimon Glass        self.ClearLine(0)
11842c3deb97SSimon Glass        return (self.fail, self.warned)
1185