source_tag=src_tags[0],
                   wavelet=wl,
                   absorption_zone=absorption_zone,
                   lumping=True)

    locEW = Locator(domain, rcvEW_locations)
    tracerEW = SimpleSEGYWriter(receiver_group=rgEW,
                                source=src_loc_2D,
                                sampling_interval=sampling_interval)
    if DIM == 3:
        locNS = Locator(domain, rcvNS_locations)
        tracerNS = SimpleSEGYWriter(receiver_group=rgNS,
                                    source=src_loc_2D,
                                    sampling_interval=sampling_interval)

    if not tracerEW.obspy_available():
        print(
            "\nWARNING: obspy not available, SEGY files will not be written\n")
    elif getMPISizeWorld() > 1:
        print(
            "\nWARNING: SEGY files cannot be written with multiple processes\n"
        )

    t = 0.
    mkDir('tmp')
    n = 0
    while t < t_end:
        t, p = sw.update(t + sampling_interval)
        tracerEW.addRecord(locEW(p))
        if DIM == 3: tracerNS.addRecord(locNS(p))
        print(t, locEW(p)[:4], wl.getValue(t))
Example #2
0
            sampling_interval=sampling_interval,
            text="x-displacement - north-south line",
        )
        tracerNS_y = SimpleSEGYWriter(
            receiver_group=rgNS,
            source=src_loc_2D,
            sampling_interval=sampling_interval,
            text="y-displacement - north-south line",
        )
        tracerNS_z = SimpleSEGYWriter(
            receiver_group=rgNS,
            source=src_loc_2D,
            sampling_interval=sampling_interval,
            text="z-displacement - north-south line",
        )
    if not tracerEW_x.obspy_available():
        print("\nWARNING: obspy not available, SEGY files will not be written\n")
    elif getMPISizeWorld() > 1:
        print("\nWARNING: SEGY files cannot be written with multiple processes\n")

    t = 0.0
    mkDir("tmp")
    n = 0
    while t < t_end:
        t, u = sw.update(t + sampling_interval)
        tracerEW_x.addRecord(locEW(u[0]))
        tracerEW_z.addRecord(locEW(u[DIM - 1]))
        if DIM == 3:
            tracerEW_y.addRecord(locEW(u[1]))
            tracerNS_x.addRecord(locNS(u[0]))
            tracerNS_y.addRecord(locNS(u[1]))