Tools: add pretty diff size script

Tools/scripts/pretty_diff_size.py

@@ -0,0 +1,83 @@
+#!/usr/bin/env python
+"""This script intend to provide a pretty size diff between two binaries."""
+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)