diff options
| -rwxr-xr-x | scripts/histogram.py | 70 |
1 files changed, 35 insertions, 35 deletions
diff --git a/scripts/histogram.py b/scripts/histogram.py index f1f8464..1ae5505 100755 --- a/scripts/histogram.py +++ b/scripts/histogram.py @@ -20,20 +20,49 @@ import argparse import importlib -import inspect import json import os import sys import matplotlib.pyplot as plt -currentdir = os.path.dirname( - os.path.abspath(inspect.getfile(inspect.currentframe()))) -parentdir = os.path.dirname(currentdir) -sys.path.insert(0, parentdir) + +def plot_interactive(hist): + """Plot 4 histograms for different sized allocations using matplotlib""" + sizes = [] + sizes_smaller_4k = [] + sizes_smaller_32k = [] + sizes_bigger_32k = [] + for size, amount in hist.items(): + size = int(size) + sizes.append(size * amount) + if size < 4096: + sizes_smaller_4k.append(size * amount) + if size < 32000: + sizes_smaller_32k.append(size * amount) + else: + sizes_bigger_32k.append(size * amount) + + plt.figure(0) + plt.hist(sizes_smaller_4k, 200) + plt.title("Sizes smaller than 4K") + + plt.figure(1) + plt.hist(sizes_smaller_32k, 200) + plt.title("Sizes smaller than 32K") + + plt.figure(2) + plt.hist(sizes_bigger_32k, 200) + plt.title("Sizes bigger than 32K") + + plt.figure(3) + plt.hist(sizes, 200) + plt.title("All sizes") + plt.show() def main(): + """Plot an interactive histogram from malt or chattymalloc output file""" parser = argparse.ArgumentParser( description="Plot histograms using a malt or chattymalloc output file") parser.add_argument("input_file", @@ -84,36 +113,7 @@ def main(): chattyparser.plot_hist_ascii(f"{fpath}.hist.txt", hist, calls) if args.interactive: - sizes = [] - sizes_smaller_4K = [] - sizes_smaller_32K = [] - sizes_bigger_32K = [] - for size, amount in hist.items(): - size = int(size) - sizes.append(size * amount) - if size < 4096: - sizes_smaller_4K.append(size * amount) - if size < 32000: - sizes_smaller_32K.append(size * amount) - else: - sizes_bigger_32K.append(size * amount) - - plt.figure(0) - plt.hist(sizes_smaller_4K, 200) - plt.title("Sizes smaller than 4K") - - plt.figure(1) - plt.hist(sizes_smaller_32K, 200) - plt.title("Sizes smaller than 32K") - - plt.figure(2) - plt.hist(sizes_bigger_32K, 200) - plt.title("Sizes bigger than 32K") - - plt.figure(3) - plt.hist(sizes, 200) - plt.title("All sizes") - plt.show() + plot_interactive(hist) if __name__ == "__main__": |