avrc :3

anno3/avrc/assignments/coff/dzcnapy_plotlib.py

@@ -0,0 +1,57 @@
+"""
+Library for plotting graphs in DZCNAPY
+"""
+import matplotlib
+import matplotlib.pyplot as plt
+matplotlib.rc("font", family="Arial")
+matplotlib.style.use("grayscale")
+
+attrs = {
+    "edge_color" : "gray",
+    "font_family" : "Liberation Sans Narrow",
+    "font_size" : 15,
+    "font_weight" : "bold",
+    "node_color" : "pink",
+    "node_size" : 700,
+    "width" : 2,
+}
+thick_attrs = attrs.copy()
+thick_attrs["alpha"] = 0.5
+thick_attrs["width"] = 15
+
+small_attrs = attrs.copy()
+small_attrs["node_size"] = 50
+small_attrs["font_size"] = 10
+
+medium_attrs = small_attrs.copy()
+medium_attrs["node_size"] = 250
+
+def set_extent(positions, axes, title=None):
+    """
+    Given node coordinates pos and the subplot,
+    calculate and set its extent.
+    """
+    axes.tick_params(labelbottom="off")
+    axes.tick_params(labelleft="off")
+    if title:
+        axes.set_title(title)
+
+    x_values, y_values = zip(*positions.values())
+    x_max = max(x_values)
+    y_max = max(y_values)
+    x_min = min(x_values)
+    y_min = min(y_values)
+    x_margin = (x_max - x_min) * 0.1
+    y_margin = (y_max - y_min) * 0.1
+    try:
+        axes.set_xlim(x_min - x_margin, x_max + x_margin)
+        axes.set_ylim(y_min - y_margin, y_max + y_margin)
+    except AttributeError:
+        axes.xlim(x_min - x_margin, x_max + x_margin)
+        axes.ylim(y_min - y_margin, y_max + y_margin)
+
+def plot(fname, save = False):
+    plt.tight_layout()
+    if save:
+        plt.savefig("plots/{}.pdf".format(fname), dpi=600)
+    plt.show()