trick/trick_sims/SIM_rocket/scripts/Forces.py
Alex Lin 496de8c7a9
Disambiguate python use #1250 (#1251)
* Disambiguate python use #1250

Changed all calls to /usr/bin/python to /usr/bin/env python3.  Removed execute permissions on a lot of files that are not executable.

closes #1250
2022-03-22 15:15:40 -05:00

34 lines
1.1 KiB
Python
Executable File

#!/usr/bin/env python3
import matplotlib.pyplot as plt
import numpy as np
#
data = np.genfromtxt('../RUN_test/log_Rocket.csv',
delimiter=',',
skip_header=1,
skip_footer=1,
names=['t',
'vux', 'vuy',
'Fthrustx', 'Fthrusty',
'Fdragx', 'Fdragy',
'Fgravx','Fgravy',
'Ftotalx','Ftotaly',
'Mt',
'MassTotal', 'MassRate',
'Posx','Posy',
'velx','vely',
'Cd', 'MassEmpty' ],
dtype=(float, float)
)
curve1 = plt.plot(data['t'], data['Fthrusty'], 'C1-')
curve2 = plt.plot(data['t'], data['Fdragy'], 'C2-')
curve3 = plt.plot(data['t'], data['Fgravy'], 'C3-')
curve4 = plt.plot(data['t'], data['Ftotaly'], 'C4-')
plt.legend(('F_thrust', 'F_drag', 'F_grav', 'F_total'))
plt.title('Rockets Forces Over Time')
plt.xlabel('time (s)')
plt.ylabel('force (N)')
plt.grid(True)
plt.show()