4 from matplotlib
import pyplot
as plt
6 import matplotlib
as mpl
8 mpl.rcParams[
'svg.fonttype'] =
"none" 9 mpl.rcParams[
'font.size'] = 16
12 mpl.rc(
'font', **{
'family':
'serif',
'serif': [
'Computer Modern']})
13 mpl.rc(
'text', usetex=
True)
18 from yaml
import CLoader
as Loader, CDumper
as Dumper
20 from yaml
import Loader, Dumper
24 def __init__(self, run_name = "", numeric_method = "_midpoint_"):
30 self.
results_path = os.path.dirname(os.path.realpath(__file__)) +
"/../../../results/" 32 results_folder =
"/results/" 34 stream = open(self.
results_path + run_name +
"/doc/yaml/parameters.yaml",
'r') 39 self.
pad_position = np.loadtxt(self.
results_path + run_name + results_folder + shared_names + numeric_method +
"pad_position.csv", delimiter =
',')
40 if(
not only_position):
41 self.
pad_velocity = np.loadtxt(self.
results_path + run_name + results_folder + shared_names + numeric_method +
"pad_velocity.csv", delimiter =
',')
42 self.
pad_friction = np.loadtxt(self.
results_path + run_name + results_folder + shared_names + numeric_method +
"pad_friction.csv", delimiter =
',')
56 save_interval = self.
run_parameters[
"Parameters"][
"save_interval_dt"]
58 print(
"Look like this run does not contain a \"save_interval_dt\" parameter.")
62 self.
x = np.linspace(start = start, stop = stop, num = number_of_points)
98 n_segments = int(sys.argv[-1])
99 print(
"Number of intervals: ", n_segments)
101 print(
"Please enter the number of segments as the last argument when using \'-s\'")
103 n_data_points_in_segment = int(len(self.
x)/n_segments)
105 for i
in range(n_segments):
106 start_index = i * n_data_points_in_segment
107 end_index = (i + 1) * n_data_points_in_segment
109 if(i == n_segments - 1):
112 plt.plot(self.
x[start_index:end_index], self.
pad_position[start_index:end_index], label =
"Segment " + str(i))
121 if(len(sys.argv) > 2):
122 plot_pad =
PlotPad(sys.argv[1], sys.argv[2])
123 elif(len(sys.argv) == 2):
124 plot_pad =
PlotPad(sys.argv[1])
126 print(
"This will probably not work, have you remembered to give input to the command?")
129 plot_pad.plot_pad_position()
130 plot_pad.plot_segments()
131 if(
not only_position):
132 plot_pad.plot_pad_velocity()
133 plot_pad.plot_pad_friction()
134 plot_pad.plot_pad_position_vers_velocity()
137 if __name__ ==
"__main__":
def __init__(self, run_name="", numeric_method="_midpoint_")
def plot_pad_friction(self)
def plot_pad_velocity(self)
def plot_pad_position_vers_velocity(self)
def plot_pad_position(self)
def loadFiles(self, run_name, numeric_method)