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