def makedata3d(cpara, velmod3d, log=setuplog(), raytracing3d=True):
mod3d = Mod3DPerturbation(velmod3d, cpara.depth_axis, velmod=cpara.velmod)
sta_info = Station(cpara.stalist)
if cpara.rayp_lib is not None:
srayp = np.load(cpara.rayp_lib)
else:
srayp = None
RFdepth = []
for i in range(sta_info.stla.shape[0]):
rfdep = {}
evt_lst = join(cpara.rfpath, sta_info.station[i],
sta_info.station[i] + 'finallist.dat')
stadatar = RFStation(evt_lst, only_r=True)
stadatar.stel = sta_info.stel[i]
stadatar.stla = sta_info.stla[i]
stadatar.stlo = sta_info.stlo[i]
if stadatar.prime_phase == 'P':
sphere = True
else:
sphere = False
log.RF2depthlog.info('the {}th/{} station with {} events'.format(
i + 1, sta_info.stla.shape[0], stadatar.ev_num))
if raytracing3d:
pplat_s, pplon_s, pplat_p, pplon_p, newtpds = psrf_3D_raytracing(
stadatar, cpara.depth_axis, mod3d, srayp=srayp, sphere=sphere)
else:
pplat_s, pplon_s, pplat_p, pplon_p, raylength_s, raylength_p, tps = psrf_1D_raytracing(
stadatar,
cpara.depth_axis,
srayp=srayp,
sphere=sphere,
phase=cpara.phase)
newtpds = psrf_3D_migration(pplat_s, pplon_s, pplat_p, pplon_p,
raylength_s, raylength_p, tps,
cpara.depth_axis, mod3d)
amp3d, end_index = time2depth(stadatar, cpara.depth_axis, newtpds)
rfdep['station'] = sta_info.station[i]
rfdep['stalat'] = sta_info.stla[i]
rfdep['stalon'] = sta_info.stlo[i]
rfdep['depthrange'] = cpara.depth_axis
# rfdep['events'] = _convert_str_mat(stadatar.event)
rfdep['bazi'] = stadatar.bazi
rfdep['rayp'] = stadatar.rayp
# rfdep['phases'] = _convert_str_mat(stadatar.phase)
rfdep['moveout_correct'] = amp3d
rfdep['piercelat'] = pplat_s
rfdep['piercelon'] = pplon_s
rfdep['stopindex'] = end_index
RFdepth.append(rfdep)
np.save(cpara.depthdat, RFdepth)
def makedata(cpara, velmod3d=None, modfolder1d=None, log=setuplog()):
ismod1d = False
if velmod3d is not None:
if isinstance(velmod3d, str):
velmod = velmod3d
else:
raise ValueError('Path to 3d velocity model should be in str')
elif modfolder1d is not None:
if isinstance(modfolder1d, str):
if exists(modfolder1d):
ismod1d = True
else:
raise FileNotFoundError(
'No such folder of {}'.format(modfolder1d))
else:
ValueError('Path to 1d velocity model files should be in str')
else:
ismod1d = True
# cpara = ccppara(cfg_file)
sta_info = Station(cpara.stalist)
RFdepth = []
for i in range(sta_info.stla.shape[0]):
rfdep = {}
evt_lst = join(cpara.rfpath, sta_info.station[i],
sta_info.station[i] + 'finallist.dat')
stadatar = RFStation(evt_lst, only_r=True)
stadatar.stel = sta_info.stel[i]
stadatar.stla = sta_info.stla[i]
stadatar.stlo = sta_info.stlo[i]
log.RF2depthlog.info('the {}th/{} station with {} events'.format(
i + 1, sta_info.stla.shape[0], stadatar.ev_num))
piercelat = np.zeros([stadatar.ev_num, cpara.depth_axis.shape[0]])
piercelon = np.zeros([stadatar.ev_num, cpara.depth_axis.shape[0]])
if stadatar.prime_phase == 'P':
sphere = True
else:
sphere = False
if ismod1d:
if modfolder1d is not None:
velmod = _load_mod(modfolder1d, sta_info.station[i])
else:
velmod = cpara.velmod
PS_RFdepth, end_index, x_s, _ = psrf2depth(stadatar,
cpara.depth_axis,
velmod=velmod,
srayp=cpara.rayp_lib,
sphere=sphere,
phase=cpara.phase)
for j in range(stadatar.ev_num):
piercelat[j], piercelon[j] = latlon_from(sta_info.stla[i],
sta_info.stlo[i],
stadatar.bazi[j],
rad2deg(x_s[j]))
rfdep['station'] = sta_info.station[i]
rfdep['stalat'] = sta_info.stla[i]
rfdep['stalon'] = sta_info.stlo[i]
rfdep['depthrange'] = cpara.depth_axis
# rfdep['events'] = _convert_str_mat(stadatar.event)
rfdep['bazi'] = stadatar.bazi
rfdep['rayp'] = stadatar.rayp
# rfdep['phases'] = stadatar.phase[i]
rfdep['moveout_correct'] = PS_RFdepth
rfdep['piercelat'] = piercelat
rfdep['piercelon'] = piercelon
rfdep['stopindex'] = end_index
RFdepth.append(rfdep)
# savemat(cpara.depthdat, {'RFdepth': RFdepth})
np.save(cpara.depthdat, RFdepth)