Exemplo n.º 1
0
def convert_from_asdf(asdf_fn,
                      outputdir,
                      tag=None,
                      filetype="sac",
                      output_staxml=True,
                      output_quakeml=True,
                      _verbose=True):
    """
    Convert the waveform in asdf to different types of file
    """
    filetype = filetype.upper()
    if filetype not in ["SAC", "MSEED"]:
        raise ValueError("Supported filetype: 1) sac; 2) mseed")

    if not os.path.exists(asdf_fn):
        raise ValueError("No asdf file: %s" % asdf_fn)
    if not os.path.exists(outputdir):
        os.makedirs(outputdir)

    if isinstance(tag, str):
        tag_list = [tag]

    print("Input ASDF: %s" % asdf_fn)
    print("Output dir: %s" % outputdir)
    print("Output StationXML and Quakeml: [%s, %s]" %
          (output_staxml, output_quakeml))

    ds = ASDFDataSet(asdf_fn, mode='r')

    if output_quakeml:
        if len(ds.events) >= 1:
            filename = os.path.join(outputdir, "Quakeml.xml")
            if _verbose:
                print("Quakeml file: %s" % filename)
            ds.events.write(filename, format="QUAKEML")

    sta_list = ds.waveforms.list()

    for station_name in sta_list:
        if _verbose:
            print("Convert station: %s" % station_name)
        station_name2 = station_name.replace(".", "_")
        station = getattr(ds.waveforms, station_name2)
        default_tag_list = station.get_waveform_tags()
        if tag is None:
            tag_list = default_tag_list
        for _tag in tag_list:
            if _tag not in default_tag_list:
                print("Tag(%s) not in Station(%s) taglist(%s)" %
                      (_tag, station_name2, default_tag_list))
                continue
            try:
                stream, inv = ds.get_data_for_tag(station_name2, _tag)
            except Exception:
                print("Error for station:", station_name2)
            if filetype == "SAC":
                write_stream_to_sac(stream, outputdir, _tag)
            elif filetype == "MSEED":
                filename = os.path.join(outputdir,
                                        "%s.%s.mseed" % (station_name, _tag))
                stream.write(filename, format="MSEED")
            if output_staxml:
                filename = os.path.join(outputdir,
                                        "%s.%s.xml" % (station_name, _tag))
                try:
                    inv.write(filename, format="STATIONXML")
                except Exception:
                    print("Error creating STATIONXML: %f" % filename)

    del ds
Exemplo n.º 2
0
def convert_from_asdf(asdf_fn, outputdir, tag=None, filetype="sac",
                      output_staxml=True, output_quakeml=True,
                      _verbose=True):
    """
    Convert the waveform in asdf to different types of file
    """
    filetype = filetype.upper()
    if filetype not in ["SAC", "MSEED"]:
        raise ValueError("Supported filetype: 1) sac; 2) mseed")

    if not os.path.exists(asdf_fn):
        raise ValueError("No asdf file: %s" % asdf_fn)
    if not os.path.exists(outputdir):
        os.makedirs(outputdir)

    if isinstance(tag, str):
        tag_list = [tag]

    print("Input ASDF: %s" % asdf_fn)
    print("Output dir: %s" % outputdir)
    print("Output StationXML and Quakeml: [%s, %s]" % (output_staxml,
                                                       output_quakeml))

    ds = ASDFDataSet(asdf_fn, mode='r')

    if output_quakeml:
        if len(ds.events) >= 1:
            filename = os.path.join(outputdir, "Quakeml.xml")
            if _verbose:
                print("Quakeml file: %s" % filename)
            ds.events.write(filename, format="QUAKEML")

    sta_list = ds.waveforms.list()

    for station_name in sta_list:
        if _verbose:
            print("Convert station: %s" % station_name)
        station_name2 = station_name.replace(".", "_")
        station = getattr(ds.waveforms, station_name2)
        default_tag_list = station.get_waveform_tags()
        if tag is None:
            tag_list = default_tag_list
        for _tag in tag_list:
            if _tag not in default_tag_list:
                print("Tag(%s) not in Station(%s) taglist(%s)" %
                      (_tag, station_name2, default_tag_list))
                continue
            try:
                stream, inv = ds.get_data_for_tag(station_name2, _tag)
            except:
                print("Error for station:", station_name2)
            if filetype == "SAC":
                write_stream_to_sac(stream, outputdir, _tag)
            elif filetype == "MSEED":
                filename = os.path.join(outputdir, "%s.%s.mseed"
                                        % (station_name, _tag))
                stream.write(filename, format="MSEED")
            if output_staxml:
                filename = os.path.join(outputdir, "%s.%s.xml"
                                        % (station_name, _tag))
                try:
                    inv.write(filename, format="STATIONXML")
                except:
                    print("Error creating STATIONXML: %f" % filename)

    del ds