def print_fig(self, out_path='', method=''):
"""
print figure of chosen method
as .png to out_path(default='images/')
"""
out_filename = '{0}-dx_{2:.3f}-dt_{3:.3f}-CFL_{4:.2f}-Ne_{5:.2f}-PE_{1:.2f}-steps_{6}' \
.format(self.signal_shape, self.PE, self.dx, self.dt, self.CFL, self.NE, self.steps)
#if (method in self.figures) and (out_path != ''):
my_fig = self.get_fig()
create_png(out_path, out_filename, my_fig)
print("plotted {!s} to {!s}".format(out_filename + '.png', out_path))
def main(dx_vec=config_file.dx_vec,
dt_vec=config_file.dt_vec,
c_vec=config_file.c_vec,
v_vec=config_file.v_vec,
step_vec=config_file.step_vec,
signal_vec=config_file.signal_vec,
method_vec=config_file.method_vec,
output_folder=config_file.folder_output,
):
t_0 = t.time()
# check_vec ... will store information about the stability of each method
# for each CFL-NE-combination
check_vec = {}
# iterate over all combinations!
for par in product(dx_vec, dt_vec, c_vec, v_vec, step_vec, signal_vec):
print "\n", par
# ProjectData handles input data, figures, etc.
pd = ProjectData(dx=par[0], dt=par[1], c=par[2],
v=par[3], steps=par[4], signal=par[5])
try:
# iterate over all methods in method_vec
for method in method_vec:
if not method in check_vec:
check_vec[method] = []
pd.methods[method] = MethodData(pd, method)
t_1 = t.time()
# calculate using chosen method
pd.calc(method)
t_2 = t.time()
print("\n method: %s: area = %.2f" % (method, pd.methods[method].get_area()))
print ("time - %s: %.2f ms" % (method, (t_2 - t_1) * 1000))
print ("time per step - %s: %.2f ms" % (method, (t_2 - t_1) / par[4] * 1000))
check_vec[method].append([
pd.CFL,
round(pd.NE, 2),
round(pd.PE, 2),
pd.methods[method].is_not_neg(),
pd.methods[method].is_nearly_zero(),
pd.methods[method].is_not_huge()
])
# export results
if config_file.flag_print_simulation_plot:
pd.print_fig(out_path=output_folder + 'images/')
if config_file.flag_write_simulation_as_csv:
pd.write_as_csv(out_path=output_folder + 'csv/')
pd.del_fig()
finally:
# Delete ProjectData instance and its figures now.
# Otherwise figures may be wrong
del pd
for method in check_vec.keys():
if config_file.flag_print_stability_data_plot:
export_stability_check_results(output_folder + config_file.subfolder_plot_stability,
config_file.stability_file_signature,
method,
check_vec[method])
if config_file.flag_write_stability_data_as_csv:
fig_stability = draw_stability_plot(check_vec[method])
create_png(output_folder + config_file.subfolder_plot_stability,
'stability_' + method, fig_stability)
t_n = t.time()
print ("runtime main: %.3f s" % (t_n - t_0))
