print "____________________________________________________________________" print command subprocess.call(command, shell=True) figure_number = 1; directories = os.listdir(directory) for subdirectory in directories: fig = plt.figure(figure_number) figure_number = figure_number+1 data = numpy.loadtxt(directory+"/"+subdirectory+"/results_"+demo_labels[0]+".txt") if (len(demo_labels)==1): plotDynamicalSystem(data,[fig.add_subplot(1,2,1), fig.add_subplot(1,2,2)]) fig.canvas.set_window_title(subdirectory+" ("+demo_labels[0]+")") else: data_compare = numpy.loadtxt(directory+"/"+subdirectory+"/results_"+demo_labels[1]+".txt") axs = [fig.add_subplot(2,2,1), fig.add_subplot(2,2,2)] axs_diff = [fig.add_subplot(2,2,3), fig.add_subplot(2,2,4)] # Bit of a hack... We happen to know that SpringDamperSystem is only second order system if (subdirectory == "SpringDamperSystem"): axs = [fig.add_subplot(2,3,1), fig.add_subplot(2,3,2), fig.add_subplot(2,3,3)] axs_diff = [fig.add_subplot(2,3,4), fig.add_subplot(2,3,5), fig.add_subplot(2,3,6)] plotDynamicalSystemComparison(data,data_compare,demo_labels[0],demo_labels[1],axs,axs_diff) fig.canvas.set_window_title(subdirectory+" ("+demo_labels[0]+" vs "+demo_labels[1]+")") axs[1].legend() plt.show()
if (not os.path.isfile(executable)): print "" print "ERROR: Executable '"+executable+"' does not exist." print "Please call 'make install' in the build directory first." print "" sys.exit(-1); # Call the executable with the directory to which results should be written directory = "/tmp/demoExponentialSystem" subprocess.call([executable, directory]) fig = plt.figure(1) data_ana = numpy.loadtxt(directory+"/analytical.txt") plotDynamicalSystem(data_ana,[fig.add_subplot(1,2,1), fig.add_subplot(1,2,2)]) plt.title('analytical') fig = plt.figure(2) data_num = numpy.loadtxt(directory+"/numerical.txt") plotDynamicalSystem(data_num,[fig.add_subplot(1,2,1), fig.add_subplot(1,2,2)]) plt.title('numerical') fig = plt.figure(3) axs = [fig.add_subplot(2,2,1), fig.add_subplot(2,2,2)] axs_diff = [fig.add_subplot(2,2,3), fig.add_subplot(2,2,4)] plotDynamicalSystemComparison(data_ana,data_num,'analytical','numerical',axs,axs_diff) plt.show()
if (len(demo_labels) == 1): plotDynamicalSystem( data, [fig.add_subplot(1, 2, 1), fig.add_subplot(1, 2, 2)]) fig.canvas.set_window_title(subdirectory + " (" + demo_labels[0] + ")") else: data_compare = numpy.loadtxt(directory + "/" + subdirectory + "/results_" + demo_labels[1] + ".txt") axs = [fig.add_subplot(2, 2, 1), fig.add_subplot(2, 2, 2)] axs_diff = [fig.add_subplot(2, 2, 3), fig.add_subplot(2, 2, 4)] # Bit of a hack... We happen to know that SpringDamperSystem is only second order system if (subdirectory == "SpringDamperSystem"): axs = [ fig.add_subplot(2, 3, 1), fig.add_subplot(2, 3, 2), fig.add_subplot(2, 3, 3) ] axs_diff = [ fig.add_subplot(2, 3, 4), fig.add_subplot(2, 3, 5), fig.add_subplot(2, 3, 6) ] plotDynamicalSystemComparison(data, data_compare, demo_labels[0], demo_labels[1], axs, axs_diff) fig.canvas.set_window_title(subdirectory + " (" + demo_labels[0] + " vs " + demo_labels[1] + ")") axs[1].legend() plt.show()