My Project
plotCalculatedFrictionAmplitude.py
Go to the documentation of this file.
1 from frictionImports import *
2 
3 python_path = os.path.dirname(os.path.realpath(__file__))
4 
5 def getFileWithPaths(path):
6  paths_to_amplitude_files = np.loadtxt(python_path + path, delimiter = "\n", dtype = bytes).astype(str)
7  print(paths_to_amplitude_files)
8  return paths_to_amplitude_files
9 
10 def loadAmplitudeFiles(paths):
11  if("-notMultiple" in sys.argv):
12  print(python_path + "/../../../results/" + str(paths))
13  amplitude_files = [np.loadtxt(python_path + "/../../../results/" + str(paths), delimiter = ",")]
14  else:
15  amplitude_files = np.array([np.loadtxt(python_path + "/../../../results/" + path, delimiter = ",") for path in paths])
16  print(amplitude_files)
17 
18  if('--direction' in sys.argv):
19  colors = ['#a6611a','#dfc27d','#f5f5f5','#80cdc1','#018571']
20  legends = ['dec: pad', 'dec: no pad', 'inc: pad', 'inc: no pad', 'inc: inf. weight']
21  elif('--inf' in sys.argv):
22  colors = ['#a6611a','#dfc27d','#80cdc1','#018571']
23  legends = ['inc: no pad', 'inc: inf. weight', 'inc: pad', 'dec: pad', 'dec: no pad']
24  else:
25  colors = ['black', 'green', 'orange', 'pink']
26  legends = ['pad', 'no pad', 'no neighbor', 'half mass']
27 
28  for i in range(0,len(amplitude_files)):
29  print(len(amplitude_files[i]))
30  if(len(amplitude_files[i]) > 3):
31  plt.plot(amplitude_files[i][0], -amplitude_files[i][3], color = colors[i])
32  plt.errorbar(amplitude_files[i][0], amplitude_files[i][1], yerr=amplitude_files[i][2], ecolor='red', color = colors[i], label = legends[i])
33  if("-log" in sys.argv):
34  plt.xscale('log')
35  plt.ylabel("\$A\$")
36  plt.xlabel("\$\\nu\$")
37  plt.legend()
38  plt.show()
39 
40 def run():
41  # path_1 = "/FrictionAmplitudeResultPaths/test_fusing_amplitude_files.csv"
42  # path_2 = "/FrictionAmplitudeResultPaths/test_long_amplitude_run.csv"
43  #path_3 = "/FrictionAmplitudeResultPaths/test_bk_bkp_amplitude_files_3.csv"
44  #path_4 = "/FrictionAmplitudeResultPaths/test_bk_bkp_amplitude_files_2.csv"
45  path_5 = "/FrictionAmplitudeResultPaths/test_bk_bkp_amplitude_files_2.csv"
46  # paths_1 = getFileWithPaths(path_1)
47  # paths_2 = getFileWithPaths(path_2)
48  #paths_3 = getFileWithPaths(path_3)
49  #paths_4 = getFileWithPaths(path_4)
50  paths_5 = getFileWithPaths(path_5)
51  # loadAmplitudeFiles(paths_1)
52  # loadAmplitudeFiles(paths_2)
53  #loadAmplitudeFiles(paths_3)
54  #loadAmplitudeFiles(paths_4)
55  loadAmplitudeFiles(paths_5)
56 
57 if __name__ == "__main__":
58  run()