Added flag to remove negative timestamps in LF 3D seismograms

bbp/comps/lf_seismograms.py

@@ -75,12 +75,14 @@ class LFSeismograms(object):
     tmpdir directory
     """
 
-    def __init__(self, i_seis_dir, i_r_stations, sim_id=0):
+    def __init__(self, i_seis_dir, i_r_stations,
+                 allow_negative_timestamps=False, sim_id=0):
         """
         Initialize class variables
         """
         self.seis_dir = i_seis_dir
         self.r_stations = i_r_stations
+        self.allow_negative_timestamps = allow_negative_timestamps
         self.sim_id = sim_id
 
     def run(self):
@@ -108,4 +110,28 @@ def run(self):
                                          "%d.%s-lf.bbp" % (int(self.sim_id),
                                                                station_name))
             print("Copying LF file for station %s..." % (station_name))
-            shutil.copy2(a_input_file, a_output_file)
+
+            if self.allow_negative_timestamps:
+                shutil.copy2(a_input_file, a_output_file)
+            else:
+                # Copy LF seismograms but remove points with t<0
+                fp_in = open(a_input_file, 'r')
+                fp_out = open(a_output_file, 'w')
+                for line in fp_in:
+                    pieces = line.split()
+                    try:
+                        if pieces[0] == '#' or pieces[0] == '%':
+                            fp_out.write(line)
+                        elif float(pieces[0]) < -0.0001:
+                            continue
+                        elif float(pieces[0]) < 0.0001:
+                            fp_out.write("0.0\t%s\t%s\t%s\n" % (pieces[1],
+                                                                pieces[2],
+                                                                pieces[3]))
+                        else:
+                            fp_out.write(line)
+                    except ValueError:
+                        fp_out.write(line)
+                fp_in.close()
+                fp_out.flush()
+                fp_out.close()