-
Notifications
You must be signed in to change notification settings - Fork 0
/
Power Spectrum.py
92 lines (80 loc) · 2.69 KB
/
Power Spectrum.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#importing functions for psd
from Functions import *
from MainConfigurationFile import roughness, experiment, Step
from MainConfigurationFile import StartTimes, EndTimes, ReservoirHeight, EntryPressure, ExitPressure, types
# Retrieve Data from Experiment
actualpath = getExperimentPath(roughness, experiment, experiment, types)
print(actualpath)
pressure = tdms_df(actualpath)
#Retrieve Data from laser
laserpath1 = getLaserPath(roughness, experiment,'1', types)
laser1 = laser_df(laserpath1)
laserpath2 = getLaserPath(roughness, experiment, '2', types)
laser2 = laser_df(laserpath2)
#slices
First= StartTimes[Step-1]
Last = EndTimes[Step-1]
First = First
Last = Last
#pressure drop
reservoir = ReservoirHeight[Step-1]
before = EntryPressure[Step-1]
after = ExitPressure[Step-1]
Start = ((First+Last)/2)-50
End = Start+100
#creating psd for pressure data
smoothp = pressure_df_smooth(pressure)
roughp = pressure_df_rough(pressure)
pressureslice = pressure_slice_df(smoothp, Start, End)
pressureslice1 = pressure_slice_df(roughp, Start, End)
ppsd = smooth_pressure_psd(pressureslice, Start, End)
ppsd1 = rough_pressure_psd(pressureslice1, Start, End)
#creating psd for laser data
ldv1 = laser_snr_filter(laser1, 2.0)
ldv1 = laser_df_for_graph(ldv1)
ldv1 = laser_interpolate(ldv1)
ldv2 = laser_snr_filter(laser2, 2.0)
ldv2 = laser_df_for_graph(ldv2)
ldv2 = laser_interpolate(ldv2)
slicer = laser_slice_df(ldv1, Start, End)
slicer2 = laser_slice_df(ldv2, Start, End)
lpsd = laser_psd(slicer, Start, End)
lpsd1 = laser_psd(slicer2, Start, End)
#plotting the psd for pressure data
fig, ax = plt.subplots(3,1)
ax[0].plot(roughp)
ax[0].plot(smoothp)
ax[0].legend(["After", "Before"], loc='upper right')
ax[0].set_ylabel('Pressure (mBar/m)')
ax[0].grid(True, which = 'both')
ax[1].plot(pressureslice1)
ax[1].plot(pressureslice)
ax[1].legend(["After", "Before"], loc='upper right')
ax[1].set_ylabel('Pressure (mBar/m)')
ax[1].grid(True, which = 'both')
ax[2].plot(ppsd1)
ax[2].plot(ppsd)
ax[2].legend(["After", "Before"], loc='upper right')
ax[2].set_ylabel('Energy (%)')
ax[2].grid(True, which = 'both')
ax[0].set_title('Pressure Power Spectrum')
#plotting the psd for laser data
fig, ax = plt.subplots(3,1)
ax[0].plot(ldv2)
ax[0].plot(ldv1)
ax[0].set_ylim(-.5,2.5)
ax[0].legend(["After", "Before"], loc='upper right')
ax[0].set_ylabel('Speed (m/sec)')
ax[0].grid(True, which = 'both')
ax[1].plot(slicer2)
ax[1].plot(slicer)
ax[1].legend(["After", "Before"], loc='upper right')
ax[1].set_ylabel('Speed (m/sec)')
ax[1].grid(True, which = 'both')
ax[2].plot(lpsd1)
ax[2].plot(lpsd)
ax[2].legend(["After", "Before"], loc='upper right')
ax[2].set_ylabel('Energy (%)')
ax[2].grid(True, which = 'both')
ax[0].set_title('Laser Power Spectrum')
plt.show()