get_waveforms.py

"""
Load waveform data for processing


# Copyright (c) 2018 by SCCER-SoE and SED at ETHZ

Version 0.0, 23.10.2018, Joseph Doetsch (doetschj)
             23.10.2018, Semih Demir

"""

from obspy import Stream
import pyasdf


def get_waveforms(param, load_file, trig_time):
    aparam = param['Acquisition']
    gparam = param['General']
    asdf_folder = param['Trigger']['asdf_folder']
    #print(gparam['stats']['network'] + '_001')
    start_time = trig_time - param['Trigger']['starttime']
    end_time = trig_time + param['Trigger']['endtime']

    ch_in = range(1, gparam['sensor_count' ] +1)
    r = range(len(ch_in))
    if len(load_file) == 1:  # load 2 snippets if event in overlap otherwise load 1
        wf_stream = Stream()
        ds = pyasdf.ASDFDataSet(asdf_folder + '/' + load_file[0], mode='r')
        for k in r:
            wf_stream += ds.get_waveforms(network=gparam['stats']['network'], station=str(int(ch_in[k])).zfill(3), location='00',
                                          channel='001',
                                          starttime=start_time,
                                          endtime=end_time,
                                          tag="raw_recording")

    else:
        ds1 = pyasdf.ASDFDataSet(asdf_folder + '/' + load_file[0], mode='r')
        ds2 = pyasdf.ASDFDataSet(asdf_folder + '/' + load_file[1], mode='r') #GR_001

        if end_time > ds2.waveforms.aparam['stats']['network']+'_001'.raw_recording.traces[0].stats["starttime"]:
            sta_p1 = Stream()
            sta_p2 = Stream()
            for k in r:
                sta_p1 += ds1.get_waveforms(network=gparam['stats']['network'], station=str(int(ch_in[k])).zfill(3), location='00',
                                            channel='001',
                                            starttime=start_time,
                                            endtime=ds1.waveforms.gparam['stats']['network']+'_'+'001'.raw_recording.traces[0].stats["endtime"],
                                            tag="raw_recording")
                sta_p2 += ds2.get_waveforms(network=gparam['stats']['network'], station=str(int(ch_in[k])).zfill(3), location='00',
                                            channel='001',
                                            starttime=ds2.waveforms.print(gparam['stats']['network'] + '_' +'001').raw_recording.traces[0].stats["starttime"],
                                            endtime=end_time,
                                            tag="raw_recording")
                wf_stream = sta_p1 + sta_p2
                wf_stream.merge(method=1, interpolation_samples=0)

        else:
            wf_stream = Stream()
            for k in r:
                wf_stream += ds1.get_waveforms(network=aparam['stats']['network'], station=str(int(ch_in[k])).zfill(3), location='00',
                                               channel='001',
                                               starttime=start_time,
                                               endtime=end_time,
                                               tag="raw_recording")
    return wf_stream