#==============================================================================
# Process data
#==============================================================================
b_param_dict = {
'ilev': ilev,
'c2threshb': .35,
'c2threshe': .35,
'c2thresh1': .35,
'ainuin': .9999,
'ainlin': .00001,
'nar': 9,
'tbw': 2.,
'thetae': [0, 90, 0]
}
zp_obj = zp.Z3D2EDI(station_z3d_dir=local_station_path,
rr_station_z3d_dir=rr_local_station_path) #,
# station_ts_dir=local_station_path.joinpath('TS'))
zp_obj.birrp_exe = birrp_path
zp_obj.calibration_path = coil_calibration_path
# zp_obj.survey_config_fn = Path(zp_obj.station_ts_dir).joinpath('{0}.cfg'.format(station))
# zp_obj.survey_config.read_survey_config_file(zp_obj.survey_config_fn,
# station)
zp_obj._tol_dict[4]['s_diff'] = 6 * 24 * 4 * 3600
#zp_obj._tol_dict[256]['s_diff'] = .5 * 256 * 3600
kw_dict = {
'df_fn': dfn,
'df_list': use_df_list,
'notch_dict': {
4096: None,
256: None,
4: None
copy_path = Path(r"/mnt/hgfs/MT_Data/GV2020/EDI_Files_birrp")
# copy = True
copy = False
index = {4: -1, 256: -1, 4096: -1}
# copy = True
if not copy:
plot = True
else:
plot = False
edi_list = []
for df in [4096, 256, 4]:
bf_path = Path.joinpath(station_path, "BF", str(df))
edi_list.append(list(bf_path.glob("*.edi"))[index[df]])
zp_obj = zp.Z3D2EDI()
zp_obj.station_ts_dir = station_path.as_posix()
c_edi = zp_obj.combine_edi_files(edi_list,
sr_dict={
4096: (1000.0, 12.5),
256: (12.49, 0.12),
4: (0.119, 0.00001)
})
if plot:
rp = zp_obj.plot_responses(edi_list + [c_edi])
if copy:
shutil.copy(c_edi, Path.joinpath(copy_path,
station_path.parts[-2] + ".edi"))
import mtpy.imaging.plotnresponses as pmr
import mtpy.imaging.plotresponse as pr
station = 'gz330'
station_path = Path(r"/mnt/hgfs/MT_Data/GZ2021/{0}".format(station))
# station_path = Path(r"c:\MT\MT_Data\GZ2021\{0}".format(station))
s_dict = {"4096": 0, "256": 0, "4": 0}
edi_fn_list = []
for sr, index in s_dict.items():
if index == 0:
fn = f"{station}.edi"
else:
fn = f"{station}_{index}.edi"
edi_fn_list.append(station_path.joinpath("TS", "BF", sr, fn))
p1 = pmr.PlotMultipleResponses(fn_list=edi_fn_list, plot_style="compare")
k = zp.Z3D2EDI()
k.station_ts_dir = station_path.joinpath("TS")
nedi = k.combine_edi_files(edi_fn_list,
sr_dict={4096:(1000., 50),
256:(49.9, .01),
4:(.01, .00001)})
p2 = pr.PlotResponse(fn=nedi, plot_tipper="yri", fig_num=3)
os.path.join(local_path, rr))
else:
rr_local_station_path = os.path.join(
local_path, rr_station)
b_param_dict["ilev"] = 1
else:
rr_local_station_path = None
b_param_dict["ilev"] = 0
# ==========================================================================
# Process data
# ==========================================================================
zp_obj = zp.Z3D2EDI(
station_z3d_dir=local_station_path,
rr_station_z3d_dir=rr_local_station_path,
station_ts_dir=os.path.join(local_station_path, "TS"),
)
zp_obj.birrp_exe = birrp_path
zp_obj.calibration_path = coil_calibration_path
zp_obj.get_calibrations(zp_obj.calibration_path)
try:
plot_obj, comb_edi_fn, sdf = zp_obj.process_data(
df_list=[4],
notch_dict={4: None},
max_blocks=4,
sr_dict={
4096: (1000.0, 25),
256: (24.999, 0.1),
4: (0.1, 0.00001)
},
