1 from frictionImports
import *
3 python_path = os.path.dirname(os.path.realpath(__file__))
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
11 if(
"-notMultiple" in sys.argv):
12 print(python_path +
"/../../../results/" + str(paths))
13 amplitude_files = [np.loadtxt(python_path +
"/../../../results/" + str(paths), delimiter =
",")]
15 amplitude_files = np.array([np.loadtxt(python_path +
"/../../../results/" + path, delimiter =
",")
for path
in paths])
16 print(amplitude_files)
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']
25 colors = [
'black',
'green',
'orange',
'pink']
26 legends = [
'pad',
'no pad',
'no neighbor',
'half mass']
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):
36 plt.xlabel(
"\$\\nu\$")
45 path_5 =
"/FrictionAmplitudeResultPaths/test_bk_bkp_amplitude_files_2.csv" 57 if __name__ ==
"__main__":
def loadAmplitudeFiles(paths)
def getFileWithPaths(path)