You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
3.2 KiB
89 lines
3.2 KiB
#!/usr/bin/env python |
|
|
|
''' |
|
This script intend to provide a pretty size diff between two binaries. |
|
|
|
AP_FLAKE8_CLEAN |
|
''' |
|
|
|
import os |
|
from argparse import ArgumentParser |
|
from tabulate import tabulate |
|
|
|
parser = ArgumentParser(description="Print binary size difference with master.") |
|
parser.add_argument("-m", "--master", dest='master', type=str, help="Master Binaries Path", required=True) |
|
parser.add_argument("-s", "--second", dest='second', type=str, help="Second Binaries Path", required=True) |
|
|
|
args = parser.parse_args() |
|
|
|
|
|
def sizes_for_file(filepath): |
|
"""Get binary size information with size.""" |
|
print("Get binary size of %s" % filepath) |
|
cmd = "size %s" % (filepath,) |
|
stuff = os.popen(cmd).read() |
|
lines = stuff.splitlines()[1:] |
|
size_list = [] |
|
for line in lines: |
|
row = line.strip().split() |
|
size_list.append(dict( |
|
text=int(row[0]), |
|
data=int(row[1]), |
|
bss=int(row[2]), |
|
total=int(row[3]), |
|
)) |
|
return size_list |
|
|
|
|
|
def print_table(summary_data_list_second, summary_data_list_master): |
|
"""Print the binaries size diff on a table.""" |
|
print_data = [] |
|
print("") |
|
# print(summary_data_list_second) |
|
# print(summary_data_list_master) |
|
for name in summary_data_list_second[0]: |
|
for master_name in summary_data_list_master[0]: |
|
if name == master_name: |
|
col_data = [name] |
|
for key in ["text", "data", "bss", "total"]: |
|
bvalue = summary_data_list_second[0][name].get(key) |
|
mvalue = summary_data_list_master[0][name].get(key) |
|
if key == "total" and mvalue is None: |
|
mvalue = summary_data_list_master[0][name].get("text") + summary_data_list_master[0][name].get( |
|
"data") + summary_data_list_master[0][name].get("bss") |
|
diff = (bvalue - mvalue) * 100.0 / mvalue |
|
signum = "+" if diff > 0.0 else "" |
|
print_diff = str(bvalue - mvalue) |
|
print_diff += " (" + signum + "%0.4f%%" % diff + ")" |
|
col_data.append(print_diff) |
|
print_data.append(col_data) |
|
print(tabulate(print_data, headers=["Binary", "text", "data", "bss", "total"])) |
|
|
|
|
|
def extract_binaries_size(path): |
|
"""Seach and extract binary size for each binary in the given path.""" |
|
print("Extracting binaries size on %s" % path) |
|
binaries_list = [] |
|
for file in os.listdir(args.master): |
|
fileNoExt = os.path.splitext(file)[0] |
|
binaries_list.append(fileNoExt) |
|
binaries_list = list(dict.fromkeys(binaries_list)) |
|
if len(binaries_list) == 0: |
|
print("Failed to get binaries") |
|
|
|
size_dict = None |
|
for binaries in binaries_list: |
|
binary_path = os.path.join(path, binaries) |
|
parsed = sizes_for_file(binary_path) |
|
if size_dict is None: |
|
size_dict = [{binaries.lower(): parsed[0]}] |
|
else: |
|
size_dict[0].update({binaries.lower(): parsed[0]}) |
|
print("Success !") |
|
return size_dict |
|
|
|
|
|
master_dict = extract_binaries_size(args.master) |
|
second_dict = extract_binaries_size(args.second) |
|
|
|
print_table(second_dict, master_dict)
|
|
|