xref: /aosp_15_r20/build/make/tools/perf/pretty.py (revision 9e94795a3d4ef5c1d47486f9a02bb378756cea8a)
1*9e94795aSAndroid Build Coastguard Worker# Copyright (C) 2023 The Android Open Source Project
2*9e94795aSAndroid Build Coastguard Worker#
3*9e94795aSAndroid Build Coastguard Worker# Licensed under the Apache License, Version 2.0 (the "License");
4*9e94795aSAndroid Build Coastguard Worker# you may not use this file except in compliance with the License.
5*9e94795aSAndroid Build Coastguard Worker# You may obtain a copy of the License at
6*9e94795aSAndroid Build Coastguard Worker#
7*9e94795aSAndroid Build Coastguard Worker#      http://www.apache.org/licenses/LICENSE-2.0
8*9e94795aSAndroid Build Coastguard Worker#
9*9e94795aSAndroid Build Coastguard Worker# Unless required by applicable law or agreed to in writing, software
10*9e94795aSAndroid Build Coastguard Worker# distributed under the License is distributed on an "AS IS" BASIS,
11*9e94795aSAndroid Build Coastguard Worker# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12*9e94795aSAndroid Build Coastguard Worker# See the License for the specific language governing permissions and
13*9e94795aSAndroid Build Coastguard Worker# limitations under the License.
14*9e94795aSAndroid Build Coastguard Worker
15*9e94795aSAndroid Build Coastguard Worker# Formatting utilities
16*9e94795aSAndroid Build Coastguard Worker
17*9e94795aSAndroid Build Coastguard Workerclass Sentinel():
18*9e94795aSAndroid Build Coastguard Worker    pass
19*9e94795aSAndroid Build Coastguard Worker
20*9e94795aSAndroid Build Coastguard WorkerSEPARATOR = Sentinel()
21*9e94795aSAndroid Build Coastguard Worker
22*9e94795aSAndroid Build Coastguard Workerdef FormatTable(data, prefix="", alignments=[]):
23*9e94795aSAndroid Build Coastguard Worker    """Pretty print a table.
24*9e94795aSAndroid Build Coastguard Worker
25*9e94795aSAndroid Build Coastguard Worker    Prefixes each row with `prefix`.
26*9e94795aSAndroid Build Coastguard Worker    """
27*9e94795aSAndroid Build Coastguard Worker    if not data:
28*9e94795aSAndroid Build Coastguard Worker        return ""
29*9e94795aSAndroid Build Coastguard Worker    widths = [max([len(x) if x else 0 for x in col]) for col
30*9e94795aSAndroid Build Coastguard Worker              in zip(*[d for d in data if not isinstance(d, Sentinel)])]
31*9e94795aSAndroid Build Coastguard Worker    result = ""
32*9e94795aSAndroid Build Coastguard Worker    colsep = "  "
33*9e94795aSAndroid Build Coastguard Worker    for row in data:
34*9e94795aSAndroid Build Coastguard Worker        result += prefix
35*9e94795aSAndroid Build Coastguard Worker        if row == SEPARATOR:
36*9e94795aSAndroid Build Coastguard Worker            for w in widths:
37*9e94795aSAndroid Build Coastguard Worker                result += "-" * w
38*9e94795aSAndroid Build Coastguard Worker                result += colsep
39*9e94795aSAndroid Build Coastguard Worker            result += "\n"
40*9e94795aSAndroid Build Coastguard Worker        else:
41*9e94795aSAndroid Build Coastguard Worker            for i in range(len(row)):
42*9e94795aSAndroid Build Coastguard Worker                cell = row[i] if row[i] else ""
43*9e94795aSAndroid Build Coastguard Worker                if i >= len(alignments) or alignments[i] == "R":
44*9e94795aSAndroid Build Coastguard Worker                    result += " " * (widths[i] - len(cell))
45*9e94795aSAndroid Build Coastguard Worker                result += cell
46*9e94795aSAndroid Build Coastguard Worker                if i < len(alignments) and alignments[i] == "L":
47*9e94795aSAndroid Build Coastguard Worker                    result += " " * (widths[i] - len(cell))
48*9e94795aSAndroid Build Coastguard Worker                result += colsep
49*9e94795aSAndroid Build Coastguard Worker            result += "\n"
50*9e94795aSAndroid Build Coastguard Worker    return result
51*9e94795aSAndroid Build Coastguard Worker
52*9e94795aSAndroid Build Coastguard Worker
53