#============================================================================== # 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) },