2 from matplotlib
import pyplot
as plt
7 from plotSetUp
import setSvgLatexMatplotlibSettings
8 sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)) +
"/..")
10 from helpFunctions
import LoadFile
11 from plotSetUp
import colors_diverging_short
20 use_shared_names=False
23 file_name =
"/friction_phase/" + output_type
25 file_name = output_type
26 parameters = loader.load_yaml_parameters(run_name)
27 phase_data = loader.load_simulation_output(
31 use_shared_names=use_shared_names
43 return ((amplitude * np.sin(period * 2*np.pi * data + phase_shift)) + vertical_shift)
53 sample_points = len(data)
56 x_range = np.linspace(0.0, sample_points*0.005, sample_points)
58 y_values = scipy.fftpack.fft(data)
61 x_values = np.linspace(0.0, 1.0/sample_spacing, sample_points/2)
63 y_values = 2.0/sample_points*np.abs(y_values[:sample_points//2])
65 plt.plot(x_values[1:], y_values[1:])
67 plt.suptitle(suptitle)
77 sample_spacing = 1/800.0
78 max_value = sample_points * sample_spacing
79 x_range = np.linspace(0, max_value, num=sample_points)
83 x_values = np.linspace(0.0, 1.0/(2.0*sample_spacing), sample_points/2)
84 plt.plot(x_range, sine_range + sine_range_2, color=
"#7fcdbb", linewidth=2.0)
86 plt.ylabel(
"\$g(x)=\sin(50*x)+0.5*\sin(80x)\$")
89 y_values = scipy.fftpack.fft(sine_range + sine_range_2)
91 y_values = 2.0/sample_points * np.abs(y_values[:sample_points//2])
93 plt.plot(x_values, y_values, color=
"#7fcdbb", linewidth=2.0)
101 output_name="pad_phase_"
103 for i
in range(100, 120):
104 loaded_data_pad =
load_csv_data(run_name, output_name + str(i))
105 position_data_pad = loaded_data_pad[0]
107 loaded_data_block =
load_csv_data(run_name,
"block_phase_" + str(i))
108 position_data_block_1 = loaded_data_block[0]
131 if __name__ ==
"__main__":
132 if(len(sys.argv) > 1):
133 print(
"Info: script call has one or more arguments,\ 134 trying to use the first as run name.")
135 run_name = sys.argv[1]
def make_sin_curve(data, amplitude=1, period=1, phase_shift=1, vertical_shift=0)
def load_csv_data(run_name, output_type, use_shared_names=False)
def calculate_fourier_spectrum(data, suptitle="")
def make_fourier_spectrum(run_name, output_name="pad_phase_")
def setSvgLatexMatplotlibSettings(pyplot)