diff options
Diffstat (limited to 'scripts/plot.py')
| -rwxr-xr-x | scripts/plot.py | 101 |
1 files changed, 59 insertions, 42 deletions
diff --git a/scripts/plot.py b/scripts/plot.py index 4bffa86..9056df4 100755 --- a/scripts/plot.py +++ b/scripts/plot.py @@ -11,18 +11,24 @@ import matplotlib from matplotlib import pyplot -matplotlib.rc('text', **{ - 'usetex': True}) -matplotlib.rc('font', **{ - 'family': 'serif', - 'serif': 'Times', - 'sans-serif': 'Helvetica', - 'monospace': 'Courier'}) +matplotlib.rc("text", **{"usetex": True}) + +matplotlib.rc( + "font", + **{ + "family": "serif", + "serif": "Times", + "sans-serif": "Helvetica", + "monospace": "Courier", + } +) pyplot.subplots_adjust(wspace=0.2, hspace=0.2) + class SensibleScalarFormatter(matplotlib.ticker.ScalarFormatter): "ScalarFormatter which rounds order of magnitude to a multiple of 3" + def __init__(self): matplotlib.ticker.ScalarFormatter.__init__(self) self.set_powerlimits([-6, 6]) @@ -30,46 +36,56 @@ class SensibleScalarFormatter(matplotlib.ticker.ScalarFormatter): def _set_orderOfMagnitude(self, value_range): # Calculate "best" order in the usual way - matplotlib.ticker.ScalarFormatter._set_orderOfMagnitude(self, value_range) + matplotlib.ticker.ScalarFormatter._set_orderOfMagnitude( + self, value_range + ) # Round down to sensible (millions, billions, etc) order - self.orderOfMagnitude = self.orderOfMagnitude - (self.orderOfMagnitude % 3) + self.orderOfMagnitude = self.orderOfMagnitude - ( + self.orderOfMagnitude % 3 + ) self.set_scientific(True) + file_prefix = os.path.commonprefix(sys.argv[1:]) -n_plots = len(sys.argv) - 2 +n_plots = len(sys.argv) - 2 for i in range(n_plots): - filename = sys.argv[i+2] - file = open(filename, 'r') + filename = sys.argv[i + 2] + file = open(filename, "r") - # pyplot.clf() - - ax = pyplot.subplot(math.ceil(math.sqrt(n_plots)), - math.ceil(math.sqrt(n_plots)), - i + 1) + ax = pyplot.subplot( + math.ceil(math.sqrt(n_plots)), math.ceil(math.sqrt(n_plots)), i + 1 + ) ax.xaxis.set_major_formatter(SensibleScalarFormatter()) ax.yaxis.set_major_formatter(SensibleScalarFormatter()) - for a in ['x', 'y']: - ax.grid(which='major', axis=a, zorder=1, - linewidth=0.5, linestyle=':', color='0', dashes=[0.5, 8.0]) - - header = file.readline() + for a in ["x", "y"]: + ax.grid( + which="major", + axis=a, + zorder=1, + linewidth=0.5, + linestyle=":", + color="0", + dashes=[0.5, 8.0], + ) + + header = file.readline() columns = header[1:].split() - pyplot.xlabel('Elements') - pyplot.ylabel('Time (s)') + pyplot.xlabel("Elements") + pyplot.ylabel("Time (s)") times = [] for i in columns: times.append([]) for line in file: - if line[0] == '#': + if line[0] == "#": continue fields = line.split() - num = 0 + num = 0 for i in fields: times[num].append([float(i)]) num += 1 @@ -77,21 +93,22 @@ for i in range(n_plots): file.close() for i in range(len(times) - 1): - matplotlib.pyplot.plot(times[0], times[i + 1], '-o', label=columns[i + 1]) - - pyplot.legend(loc='upper left', - handletextpad=0.15, borderpad=0.20, borderaxespad=0, - labelspacing=0.10, columnspacing=0, - framealpha=0.90) - - pyplot.title(os.path.splitext(filename[len(file_prefix):])[0].title()) - - # out = filename.replace('.dat', '.png') - # print('Writing %s' % out) - # matplotlib.pyplot.savefig(out, bbox_inches='tight', pad_inches=0.025) - -print('Writing %s' % sys.argv[1]) + matplotlib.pyplot.plot( + times[0], times[i + 1], "-o", label=columns[i + 1] + ) + + pyplot.legend( + loc="upper left", + handletextpad=0.15, + borderpad=0.20, + borderaxespad=0, + labelspacing=0.10, + columnspacing=0, + framealpha=0.90, + ) + + pyplot.title(os.path.splitext(filename[len(file_prefix) :])[0].title()) + +print("Writing %s" % sys.argv[1]) matplotlib.pyplot.tight_layout() matplotlib.pyplot.savefig(sys.argv[1]) - -#matplotlib.pyplot.show() |