added file saving in SAC format

Author: paudetseis

obstools/atacr/classes.py

@@ -32,7 +32,7 @@
 from pkg_resources import resource_filename
 from pathlib import Path
 np.seterr(all='ignore')
-np.set_printoptions(threshold=sys.maxsize)
+#np.set_printoptions(threshold=sys.maxsize)
 
 
 class Power(object):

obstools/scripts/atacr_correct_event.py

@@ -442,6 +442,21 @@ def main(args=None):
                                 file = correctpath / eventfile.stem
                                 eventstream.save(str(file) + '.day.pkl')
 
+                                # Now save as SAC files
+                                for key, value in tfaverage.tf_list.items():
+                                    if value and eventstream.ev_list[key]:
+                                        nameZ = '.sta.' + key + '.' 
+                                        nameZ += sta.channel+'Z.SAC'
+                                        fileZ = correctpath / (eventfile.stem + nameZ)
+                                        trZ = eventstream.sth.select(component='Z')[0].copy()
+                                        trZ.data = eventstream.correct[key]
+                                        trZ = utils.update_stats(trZ, 
+                                            sta.latitude, sta.longitude, 
+                                            sta.elevation, 'Z')
+                                        trZ.write(str(fileZ), format='SAC')
+
+
+
                 # This case refers to the "daily" spectral averages
                 else:
                     if not args.skip_daily:
@@ -483,6 +498,20 @@ def main(args=None):
                                 file = correctpath / eventfile.stem
                                 eventstream.save(str(file) + '.sta.pkl')
 
+                                # Now save as SAC files
+                                for key, value in tfaverage.tf_list.items():
+                                    if value and eventstream.ev_list[key]:
+                                        nameZ = '.day.' + key + '.' 
+                                        nameZ += sta.channel+'Z.SAC'
+                                        fileZ = correctpath / (eventfile.stem + nameZ)
+                                        trZ = eventstream.sth.select(component='Z')[0].copy()
+                                        trZ.data = eventstream.correct[key]
+                                        trZ = utils.update_stats(trZ, 
+                                            sta.latitude, sta.longitude, 
+                                            sta.elevation, 'Z')
+                                        trZ.write(str(fileZ), format='SAC')
+
+
 if __name__ == "__main__":
 
     # Run main program

obstools/scripts/atacr_download_event.py

@@ -515,7 +515,7 @@ def main(args=None):
                     '.'+str(time.minute).zfill(2)
 
                 # Define file names (to check if files already exist)
-                filename = eventpath / (tstamp+'.event.pkl')
+                filename = eventpath / (tstamp+'.pkl')
                 # Horizontal 1 channel
                 file1 = eventpath / (tstamp+'.'+sta.channel+'1.SAC')
                 # Horizontal 2 channel