From ed207450f109b3c939a491d3ac7db6c484d3b286 Mon Sep 17 00:00:00 2001
From: dsjohns2 <dsjohns2@illinois.edu>
Date: Tue, 17 Oct 2017 22:21:28 -0500
Subject: [PATCH] Add code to shift waveform peak to t=0
---
POWER/shift_peak_to_zero.py | 23 +++++++++++++++++++++++
1 file changed, 23 insertions(+)
create mode 100755 POWER/shift_peak_to_zero.py
diff --git a/POWER/shift_peak_to_zero.py b/POWER/shift_peak_to_zero.py
new file mode 100755
index 0000000..d49ad49
--- /dev/null
+++ b/POWER/shift_peak_to_zero.py
@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+
+import numpy
+import sys
+import os
+
+for path in sys.argv[1:]:
+ main_dir = path
+ sim = path.split("/")[-2]
+
+ strain_data = numpy.loadtxt(path)
+ cur_max_time = strain_data[0][0]
+ cur_max_amp = abs(pow(strain_data[0][1], 2)) + abs(pow(strain_data[0][2], 2))
+ for i in strain_data[:]:
+ 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
+ for i in strain_data[:]:
+ i[0] -= cur_max_time
+ numpy.savetxt("./Extrapolated_Strain/"+sim+"/"+sim+"_shifted_radially_extrapolated_strain.dat", strain_data)
+ print(cur_max_time)
--
GitLab