From 986386c280b528fd71a5a2ba8b4e2291492bce8b Mon Sep 17 00:00:00 2001
From: dsjohns2 <dsjohns2@illinois.edu>
Date: Sat, 21 Oct 2017 12:06:15 -0500
Subject: [PATCH] Add end to plots
---
POWER/plot.py | 22 ++++++++++++++++------
1 file changed, 16 insertions(+), 6 deletions(-)
diff --git a/POWER/plot.py b/POWER/plot.py
index 16ec403..1e8ffe9 100755
--- a/POWER/plot.py
+++ b/POWER/plot.py
@@ -40,6 +40,7 @@ import numpy as np
import matplotlib
import matplotlib.pyplot as plt
import sys
+import os
run_name = sys.argv[1]
@@ -51,34 +52,43 @@ py_hplus = python_strain[:, 1]
py_phase = python_phase[:, 1]
py_amp = python_amp[:, 1]
+cur_max_time = python_strain[0][0]
+cur_max_amp = abs(pow(python_strain[0][1], 2)) + abs(pow(python_strain[0][2], 2))
+for i in python_strain[:]:
+ cur_time = i[0]
+ cur_amp = abs(pow(i[1], 2)) + abs(pow(i[2], 2))
+ if(cur_amp>cur_max_amp):
+ cur_max_amp = cur_amp
+ cur_max_time = cur_time
+
run_name_without_resolution = run_name.split("_")[0]
matplotlib.rcParams.update({'font.size': 12})
matplotlib.rcParams.update({'font.family': 'serif'})
plt.title(run_name_without_resolution)
plt.plot(py_t, py_hplus)
-plt.xlim((0,None))
+plt.xlim((0,cur_max_time+200))
plt.ylabel('${\mathfrak{R}} [h(t)]$')
plt.xlabel('Time [M]')
-plt.savefig("./Extrapolated_Strain/"+run_name+"/strain_plot_l2_m2.png")
+plt.savefig("./Extrapolated_Strain/"+run_name+"/"+run_name+"_strain_plot_l2_m2.png")
plt.close()
matplotlib.rcParams.update({'font.size': 12})
matplotlib.rcParams.update({'font.family': 'serif'})
plt.title(run_name_without_resolution)
plt.plot(py_t, py_phase)
-plt.xlim((0,None))
+plt.xlim((0,cur_max_time+200))
plt.ylabel('Phase')
plt.xlabel('Time [M]')
-plt.savefig("./Extrapolated_Strain/"+run_name+"/phase_plot_l2_m2.png")
+plt.savefig("./Extrapolated_Strain/"+run_name+"/"+run_name+"_phase_plot_l2_m2.png")
plt.close()
matplotlib.rcParams.update({'font.size': 12})
matplotlib.rcParams.update({'font.family': 'serif'})
plt.title(run_name_without_resolution)
plt.plot(py_t, py_amp)
-plt.xlim((0,None))
+plt.xlim((0,cur_max_time+200))
plt.ylabel('Amplitude')
plt.xlabel('Time [M]')
-plt.savefig("./Extrapolated_Strain/"+run_name+"/amplitude_plot_l2_m2.png")
+plt.savefig("./Extrapolated_Strain/"+run_name+"/"+run_name+"_amplitude_plot_l2_m2.png")
plt.close()
--
GitLab