57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
"""
|
|
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()
|