def wrapper(*args, **kwargs):
prof = cProfile.Profile()
re = prof.runcall(func, *args, **kwargs)
path = os.path.join(dir_name, func.__name__ + ".prof")
SL.generate_path(path)
prof.dump_stats(path)
return re
def plot_list(value_dict, ordered_param_range_dict, path, to_html=True, to_csv=False):
"""Plot the values in the dictionary value_dict in a html list and/or a csv file.
:param value_dict: Dictionary (parameters tuple -> value) containing the values to be ploted.
:type value_dict: dict(tuple -> int)
:param ordered_param_range_dict: Ordered dictionary (parameter name -> range of the parameter).
:type ordered_param_range_dict: Shared.OrderedDict(str -> range)
:param path: Path to the plot file without suffix.
:type path: path
:param to_html: Option to plot a html list (Default: False).
:type to_html: bool
:param to_csv: Option to build a csv file (Default: False).
:type to_csv: bool
:param x_plots: Number of plots on the x-axis (Default: 2)
:type x_plots: int
"""
StoreLoad.generate_path(path)
data_list = []
for (key, value) in value_dict.items():
if value is None:
value = ' '
elif value == '*':
value = Parameters.zero_v_symbol
data_list.append(list(key) + [value])
data_list.sort()
columns = list(ordered_param_range_dict.keys())+['dimension']
data_frame = pandas.DataFrame(data=data_list, columns=columns)
if to_html:
html_path = path + '.html'
data_frame.to_html(html_path)
if to_csv:
csv_path = path + '.csv'
data_frame.to_csv(csv_path)
def plot_all_graphs_to_file(self, skip_existing=False):
""" Plots all graphs in the basis of this vector space and stores then into their repsective file paths.
The respective filename is the index of the graph in the basis (plus .png).
"""
ba = self.get_basis()
for (j, G) in enumerate(ba):
path = os.path.join(self.get_plot_path(), '%d.png' % (j))
if (not skip_existing) or (not os.path.isfile(path)):
StoreLoad.generate_path(path)
P = self.plot_graph(G)
P.save(path)
def plot_graph_to_file(self, G):
""" Plots a single graph and stores it into the standard file path.
Does not canonize the graph before plot.
The filename is the graphs g6 code.
:param G: The graph to be drawn. Must belong to this vector space.
:type G: Graph.
"""
g6 = self.graph_to_ascii(G)
path = os.path.join(self.get_plot_path(), g6 + '.png')
StoreLoad.generate_path(path)
P = self.plot_graph(G)
P.save(path)
def plot_info(data_list, header_list, path, to_html=True, to_csv=False):
"""Write a data list to a html or csv file.
:param data_list:
:type data_list: list(list)
:param header_list: List with the parameter and property names.
:type header_list: list(str)
:param path: Path to the file without suffix.
:type path: path
:param to_html: Option to write data to a html file (Default: True).
:type to_html: bool
:param to_csv: Option to write data to a csv file (Default: False).
:type to_csv: bool
:return:
"""
StoreLoad.generate_path(path)
data_frame = pandas.DataFrame(data=data_list, columns=header_list)
if to_html:
html_path = path + '.html'
data_frame.to_html(html_path)
if to_csv:
csv_path = path + '.csv'
data_frame.to_csv(csv_path)
def plot_2d_array(value_dict, ordered_param_range_dict, path, parameter_order=(0, 1)):
"""Plot a 2 dimensional array given by the value_dict.
:param value_dict: Dictionary (parameters tuple -> value) containing the values to be ploted.
:type value_dict: dict(tuple -> int)
:param ordered_param_range_dict: Ordered dictionary (parameter name -> range of the parameter).
:type ordered_param_range_dict: Shared.OrderedDict(str -> range)
:param path: Path to the plot file without suffix.
:type path: path
:param parameter_order: Permutation of the parameter indices, to specify the order of
the parameters (Default: None/given ordering). Example: (1, 0) to plot the second parameter on the x-axis
and the first on the y-axis.
:type parameter_order: list(int)
"""
path += '.png'
if parameter_order in {(0, 1), (1, 0)}:
(x_idx, y_idx) = parameter_order
else:
raise ValueError('invalid parameter order')
inverse_order = tuple(Shared.Perm(list(parameter_order)).inverse())
(x_label, x_range) = list(ordered_param_range_dict.items())[x_idx]
(y_label, y_range) = list(ordered_param_range_dict.items())[y_idx]
if len(list(x_range)) == 0 or len(list(y_range)) == 0:
logging.warn('empty parameter range: nothing to plot')
return
x_min = min(x_range)
x_max = max(x_range)
x_size = (x_max + 1 - x_min) * Parameters.x_width
y_min = min(y_range)
y_max = max(y_range)
y_size = (y_max + 1 - y_min) * Parameters.y_width
fig, ax = plt.subplots(figsize=(x_size, y_size))
plt.xlabel(x_label)
plt.ylabel(y_label)
plt.xlim(x_min, x_max)
plt.ylim(y_min, y_max)
for coordinates in itertools.product(x_range, y_range):
old_coordinates = tuple(coordinates[i] for i in inverse_order)
v = value_dict.get(old_coordinates)
if v is not None:
if v == '*':
v = Parameters.zero_v_symbol
elif v == 0:
v = Parameters.zero_symbol
(x, y) = coordinates
ax.text(x, y, str(v), va='center', ha='center')
x_ticks_grid = np.arange(x_min - 0.5, x_max + 1, 1)
y_ticks_grid = np.arange(y_min - 0.5, y_max + 1, 1)
x_ticks = np.arange(x_min, x_max + 1, 1)
y_ticks = np.arange(y_min, y_max + 1, 1)
ax.set_xticks(x_ticks)
ax.set_yticks(y_ticks)
ax.set_xticks(x_ticks_grid, minor=True)
ax.set_yticks(y_ticks_grid, minor=True)
ax.grid(which='minor')
plt.tight_layout()
StoreLoad.generate_path(path)
plt.savefig(path)
def plot_3d_array(value_dict, ordered_param_range_dict, path, parameter_order=(0, 1, 2), x_plots=2):
"""Plot a 3 dimensional array given by the value_dict as a list of 2 dimensional plots with
x_plots per line.
:param value_dict: Dictionary (parameters tuple -> value) containing the values to be ploted.
:type value_dict: dict(tuple -> int)
:param ordered_param_range_dict: Ordered dictionary (parameter name -> range of the parameter).
:type ordered_param_range_dict: Shared.OrderedDict(str -> range)
:param path: Path to the plot file without suffix.
:type path: path
:param x_plots: Number of plots on the x-axis (Default: 2)
:type x_plots: int
:param parameter_order: tuple(non-negative int), optional: Permutation of the parameter indices, to specify the order of
the parameters (Default: None/given ordering). Only for plots. Example: (1, 2, 0) to plot the second parameter
on the x-axis, the third on the y-axis and the first on the z-axis.
:type parameter_order: list(int)
"""
path += '.png'
if parameter_order in {(0, 1, 2), (0, 2, 1), (1, 0, 2), (1, 2, 0), (2, 0, 1), (2, 1, 0)}:
(x_idx, y_idx, z_idx) = parameter_order
else:
raise ValueError('invalid parameter order')
inverse_order = tuple(Shared.Perm(list(parameter_order)).inverse())
(x_label, x_range) = ordered_param_range_dict.items()[x_idx]
(y_label, y_range) = ordered_param_range_dict.items()[y_idx]
(z_label, z_range) = ordered_param_range_dict.items()[z_idx]
if len(list(x_range)) == 0 or len(list(y_range)) == 0 or len(list(z_range)) == 0:
logging.warn('empty parameter range: nothing to plot')
return
x_min = min(x_range)
x_max = max(x_range)
x_size = (x_max + 1 - x_min) * Parameters.x_width
y_min = min(y_range)
y_max = max(y_range)
y_size = (y_max + 1 - y_min) * Parameters.y_width
z_min = min(z_range)
z_max = max(z_range)
z_size = z_max + 1 - z_min
if z_size % x_plots:
y_plots = int(round(float(z_size)/x_plots + 0.5))
else:
y_plots = z_size / x_plots
fig, axarr = plt.subplots(
y_plots, x_plots, figsize=(x_plots*x_size, y_plots*y_size))
for z in z_range:
ax = _get_ax(axarr, x_plots, y_plots, z, z_min)
ax.set_xlabel(x_label)
ax.set_ylabel(y_label)
ax.set_title(str(z)+' '+z_label)
ax.set_xlim(x_min, x_max)
ax.set_ylim(y_min, y_max)
for coordinates in itertools.product(x_range, y_range, z_range):
old_coordinates = tuple(coordinates[i] for i in inverse_order)
v = value_dict.get(old_coordinates)
if v is not None:
if v == '*':
v = Parameters.zero_v_symbol
elif v == 0:
v = Parameters.zero_symbol
(x, y, z) = coordinates
_get_ax(axarr, x_plots, y_plots, z, z_min).text(
x, y, str(v), va='center', ha='center')
x_ticks_grid = np.arange(x_min - 0.5, x_max + 1, 1)
y_ticks_grid = np.arange(y_min - 0.5, y_max + 1, 1)
x_ticks = np.arange(x_min, x_max + 1, 1)
y_ticks = np.arange(y_min, y_max + 1, 1)
for z in z_range:
ax = _get_ax(axarr, x_plots, y_plots, z, z_min)
ax.set_xticks(x_ticks)
ax.set_yticks(y_ticks)
ax.set_xticks(x_ticks_grid, minor=True)
ax.set_yticks(y_ticks_grid, minor=True)
ax.grid(which='minor')
plt.tight_layout()
if z_size % x_plots:
for z in range(z_max + 1, z_min + x_plots*y_plots):
ax = _get_ax(axarr, x_plots, y_plots, z, z_min)
fig.delaxes(ax)
StoreLoad.generate_path(path)
plt.savefig(path)
def set_log_file(log_file):
log_path = os.path.join(Parameters.log_dir, log_file)
SL.generate_path(log_path)
logger.addHandler(logging.FileHandler(log_path))