1 from analysisSharedFunctions
import loadYaml, loadFile
2 from plotSetUp
import setSvgLatexMatplotlibSettings, \
3 colors_qualitative_short
as colors_short, \
5 createTimeStepValues, \
6 createSliderVelocityStepValues, \
7 choseColourFromDirection, \
10 from matplotlib
import pyplot
as plt
16 multiplot_figure = plt.figure()
17 axs = [multiplot_figure.add_subplot(pos)
for pos
in [221, 222, 223, 224]]
24 start = yaml_file[
"Parameters"][
"slider_speed"]
25 increment = yaml_file[
"Parameters"][
"increment"]
26 interval = yaml_file[
"Parameters"][
"interval"]
27 max_time = yaml_file[
"Parameters"][
"max_time"]
29 number_of_intervals = max_time // interval
30 for i
in range(0, number_of_intervals):
32 axs[ax_number].axvline(x=start + increment * i,
38 plt.axvline(x=start + increment * i,
56 step_as_continous=False
59 axs[ax_number].plot(time_steps, pad_data,
60 color=colors_short[color_index],
61 linestyle=linestyles[0],
67 if(step_as_continous):
70 axs[ax_number].set_xlim(x_lim[0], x_lim[1])
71 axs[ax_number].set_ylim(y_lim[0], y_lim[1])
73 axs[ax_number].set_ylabel(y_label)
74 axs[ax_number].set_xlabel(x_label)
86 step_as_continous=False
89 plt.plot(time_steps, pad_data,
90 color=colors_short[color_index],
91 linestyle=linestyles[0],
97 if(step_as_continous):
100 plt.ylim(y_lim[0], y_lim[1])
112 pad_position =
loadFile(run_name, yaml_file, file_name=
"pad_position")
120 x_label=
"\$\\nu\$", alpha=alpha,
121 label=label, step_as_continous=step_as_continous)
123 if(
"-m" in sys.argv):
124 x_lims = [[-0.01, 0.15], [0.29, 0.45], [0.65, 0.81], [1.07, 1.23]]
125 for i, x_lim
in enumerate(x_lims):
127 x_label=
"\$\\nu\$", ax_number=i, x_lim=x_lim, alpha=alpha,
128 step_as_continous=step_as_continous)
141 labels = [
"Increasing",
"Decreasing"]
143 for i, run_name
in enumerate(run_names):
144 plot_pad(run_name, alpha=alphas[i], label=labels[i])
151 if(len(sys.argv) == 2):
153 elif(len(sys.argv) > 2):
154 if(
"-m" in sys.argv
and "-s" in sys.argv):
156 elif(
"-m" in sys.argv
or "-s" in sys.argv):
162 if __name__ ==
"__main__":
def plot_pad(run_name, alpha=1.0, label="Pad")
def plot_multifigure(time_steps, pad_data, yaml_file, ax_number, y_label="\\, x_label="\t\", label="", alpha=1.0, x_lim=[-0.1, y_lim=[-1.6, step_as_continous=False)
def createSliderVelocityStepValues(block_data, yaml_file, step_as_continous=False)
def plotVerticalStepLines(yaml_file, ax_number=0)
def plot_single_pad(run_name)
def plot_compare_pads(run_names)
def checkStepAsContinous(yaml_file)
def loadFile(run_name, shared_names, selected='pad_position.csv')
def prettyPlotPad(time_steps, pad_data, yaml_file, y_label="\\, x_label="\t\", label="", alpha=1.0, y_lim=[-1.6, step_as_continous=False)
def choseColourFromDirection(yaml_file)
def setSvgLatexMatplotlibSettings(pyplot)