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