1 files changed, 151 insertions, 0 deletions
diff --git a/util/dram_sweep_plot.py b/util/dram_sweep_plot.py
new file mode 100755
index 000000000..a57de3d1c
--- /dev/null
+++ b/util/dram_sweep_plot.py
@@ -0,0 +1,151 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2014 ARM Limited
+# All rights reserved
+#
+# The license below extends only to copyright in the software and shall
+# not be construed as granting a license to any other intellectual
+# property including but not limited to intellectual property relating
+# to a hardware implementation of the functionality of the software
+# licensed hereunder.  You may use the software subject to the license
+# terms below provided that you ensure that this notice is replicated
+# unmodified and in its entirety in all distributions of the software,
+# modified or unmodified, in source code or in binary form.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met: redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer;
+# redistributions in binary form must reproduce the above copyright
+# notice, this list of conditions and the following disclaimer in the
+# documentation and/or other materials provided with the distribution;
+# neither the name of the copyright holders nor the names of its
+# contributors may be used to endorse or promote products derived from
+# this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+# Authors: Andreas Hansson
+
+try:
+
+    from mpl_toolkits.mplot3d import Axes3D
+    from matplotlib import cm
+    from matplotlib.ticker import LinearLocator, FormatStrFormatter
+    import matplotlib.pyplot as plt
+    import numpy as np
+except ImportError:
+    print "Failed to import matplotlib and numpy"
+    exit(-1)
+
+import sys
+import re
+
+# Determine the parameters of the sweep from the simout output, and
+# then parse the stats and plot the 3D surface corresponding to the
+# different combinations of parallel banks, and stride size, as
+# generated by the config/dram/sweep.py script
+def main():
+
+    if len(sys.argv) != 2:
+        print "Usage: ", sys.argv[0], " <simout directory>"
+        exit(-1)
+
+    try:
+        stats = open(sys.argv[1] + '/stats.txt', 'r')
+    except IOError:
+        print "Failed to open ", sys.argv[1] + '/stats.txt', " for reading"
+        exit(-1)
+
+    try:
+        simout = open(sys.argv[1] + '/simout', 'r')
+    except IOError:
+        print "Failed to open ", sys.argv[1] + '/simout', " for reading"
+        exit(-1)
+
+    # Get the burst size, number of banks and the maximum stride from
+    # the simulation output
+    got_sweep = False
+
+    for line in simout:
+        match = re.match("DRAM sweep with "
+                         "burst: (\d+), banks: (\d+), max stride: (\d+)", line)
+        if match:
+            burst_size = int(match.groups(0)[0])
+            banks = int(match.groups(0)[1])
+            max_size = int(match.groups(0)[2])
+            got_sweep = True
+
+    simout.close()
+
+    if not got_sweep:
+        print "Failed to establish sweep details, ensure simout is up-to-date"
+        exit(-1)
+
+
+    # Collect the bus utilisation as our Z-axis, we do this in a 2D
+    # grid corresponding to each iteration over the various stride
+    # sizes.
+    z = []
+    zs = []
+    i = 0
+
+    # Now parse the stats
+    for line in stats:
+        match = re.match(".*busUtil\s+(\d+\.\d+)\s+#.*", line)
+        if match:
+            bus_util = float(match.groups(0)[0])
+            z.append(bus_util)
+            i += 1
+            # If we have completed a sweep over the stride sizes,
+            # start anew
+            if i == max_size / burst_size:
+                zs.append(z)
+                z = []
+                i = 0
+
+    stats.close()
+
+    # We should have a 2D grid with as many columns as banks
+    if len(zs) != banks:
+        print "Unexpected number of data points in stats output"
+        exit(-1)
+
+    fig = plt.figure()
+    ax = fig.gca(projection='3d')
+    X = np.arange(burst_size, max_size + 1, burst_size)
+    Y = np.arange(1, banks + 1, 1)
+    X, Y = np.meshgrid(X, Y)
+
+    # the values in the util are banks major, so we see groups for each
+    # stride size in order
+    Z = np.array(zs)
+
+    surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
+                           linewidth=0, antialiased=False)
+
+    # Change the tick frequency to 64
+    start, end = ax.get_xlim()
+    ax.xaxis.set_ticks(np.arange(start, end + 1, 64))
+
+    ax.set_xlabel('Bytes per activate')
+    ax.set_ylabel('Banks')
+    ax.set_zlabel('Efficiency (%)')
+
+    # Add a colorbar
+    fig.colorbar(surf, shrink=0.5, pad=.1, aspect=10)
+
+    plt.show()
+
+if __name__ == "__main__":
+    main()