Added an option to normalize traces

Author: geojunky

seismic/receiver_fn/rf_3dmigrate.py

@@ -29,6 +29,8 @@
 @click.option('--relax-sanity-checks', is_flag=True, default=False, show_default=True,
               help='RF traces with amplitudes > 1.0 or troughs around onset time are dropped by default. '
                    'This option allows RF traces with amplitudes > 1.0 to pass through')
+@click.option('--normalize', is_flag=True, default=False, show_default=True,
+              help='Normalizes each trace by its maximum absolute value')
 @click.option('--dz', type=float, default=0.1, show_default=True,
               help='Depth-step (km)')
 @click.option('--max-depth', type=click.FloatRange(0, 750), default=150, show_default=True,
@@ -51,7 +53,7 @@
               default=100, show_default=True,
               help='Maximum depth (km) up to which velocities from the ANT model are to be extracted. '
                    'Has no impact if --ant-model is not speficied.')
-def main(rf_h5_file, output_h5_file, relax_sanity_checks, dz, max_depth, fmin, fmax,
+def main(rf_h5_file, output_h5_file, relax_sanity_checks, normalize, dz, max_depth, fmin, fmax,
          min_slope_ratio, ant_model, ant_model_max_depth):
     """Perform 3D migration of RFs
     RF_H5_FILE : Path to RFs in H5 format
@@ -71,7 +73,8 @@ def main(rf_h5_file, output_h5_file, relax_sanity_checks, dz, max_depth, fmin, f
     m = Migrator(rf_filename=rf_h5_file, dz=dz, max_depth=max_depth,
                  min_slope_ratio=min_slope_ratio, ant_model=am,
                  logger=log)
-    m.process_streams(output_h5_file, relax_sanity_checks=relax_sanity_checks, fmin=fmin, fmax=fmax)
+    m.process_streams(output_h5_file, relax_sanity_checks=relax_sanity_checks,
+                      normalize=normalize, fmin=fmin, fmax=fmax)
 # end
 
 if __name__ == "__main__":

seismic/receiver_fn/rf_ccp_util.py

@@ -209,7 +209,7 @@ def __init__(self, rf_filename, dz, max_depth, min_slope_ratio=-1,
     # end func
 
     def process_streams(self, output_file, relax_sanity_checks=False,
-                        fmin=None, fmax=None, model='iasp91'):
+                        normalize=False, fmin=None, fmax=None, model='iasp91'):
         proc_hkeys = None
         if(self._rank == 0):
             hkeys = get_obspyh5_index(self._rf_filename, seeds_only=True)
@@ -279,6 +279,14 @@ def process_streams(self, output_file, relax_sanity_checks=False,
                 continue
             # end if
 
+            # normalize traces
+            if(normalize):
+                for tr in p_traces:
+                    max_val = np.max(np.fabs(tr.data))
+                    if(max_val > 0): tr.data /= max_val
+                # end for
+            # end if
+
             has_reverberations = rf_corrections.has_reverberations(p_traces)
             if(has_reverberations):
                 if (self._logger):