Added ssp_clear_state function.

Clear state from possible previous run at beginning of ssp_run function.

Author: krisvanneste

sourcespec2/source_spec.py

@@ -17,6 +17,45 @@
 """
 
 
+def ssp_clear_state(clear_config=False, clear_options=True):
+    """
+    Clear state from a previous run.
+    Some submodules contain global variables caching information
+    about the current run, which need to be cleared before
+    launching a new run.
+
+    :param clear_config: whether to clear global config as well
+    :type clear_config: bool
+    :param clear_options: whether to clear options in global config
+    :type clear_options: bool
+    """
+    from .setup import logging
+    logging.OLDLOGFILE = None
+    # Not sure if we should reset LOGGER?
+    #logging.LOGGER = None
+
+    from . import ssp_wave_arrival
+    ssp_wave_arrival.add_arrival_to_trace.pick_cache = dict()
+    ssp_wave_arrival.add_arrival_to_trace.travel_time_cache = dict()
+    ssp_wave_arrival.add_arrival_to_trace.angle_cache = dict()
+
+    from . import ssp_plot_traces
+    ssp_plot_traces.SAVED_FIGURE_CODES = []
+    ssp_plot_traces.BBOX = None
+
+    from . import ssp_plot_spectra
+    ssp_plot_spectra.SAVED_FIGURE_CODES = []
+    ssp_plot_spectra.BBOX = None
+
+    from .setup import config
+    if clear_config:
+        # This clears the entire config, which is not what we want
+        #config.clear()
+        config.__init__()
+    elif clear_options:
+        config.options.clear()
+
+
 def ssp_run(st, inventory, ssp_event, picks, allow_exit=False):
     """
     Run source_spec as function with collected traces, station inventory,
@@ -50,6 +89,9 @@ def ssp_run(st, inventory, ssp_event, picks, allow_exit=False):
     if not allow_exit:
         ssp_exit.SSP_EXIT_CALLED = True
 
+    # Clear state from possible previous run
+    ssp_clear_state(clear_config=False, clear_options=False)
+
     # Create output folder if required, save config and setup logging
     from .setup import get_outdir_path, save_config, setup_logging
     if getattr(config.options, 'outdir', None):