def test_weighting_two(tmpdir, cmtsource):
dcon_two = construct_dcon_two()
weight_config = DefaultWeightConfig(normalize_by_energy=False,
normalize_by_category=False,
comp_weight={
"Z": 1.0,
"R": 1.0,
"T": 1.0
},
love_dist_weight=1.0,
pnl_dist_weight=1.0,
rayleigh_dist_weight=1.0,
azi_exp_idx=0.5)
config = Config(6,
dlocation=0.5,
ddepth=0.5,
dmoment=1.0e22,
zero_trace=True,
weight_data=True,
station_correction=True,
weight_config=weight_config)
srcinv = Cmt3D(cmtsource, dcon_two, config)
srcinv.source_inversion()
srcinv.plot_new_synt_seismograms(str(tmpdir))
def setup_inversion(cmt):
dcon_two = construct_dcon_two()
weight_config = DefaultWeightConfig(normalize_by_energy=False,
normalize_by_category=False,
comp_weight={
"Z": 1.0,
"R": 1.0,
"T": 1.0
},
love_dist_weight=1.0,
pnl_dist_weight=1.0,
rayleigh_dist_weight=1.0,
azi_exp_idx=0.5)
config = Config(6,
dlocation=0.5,
ddepth=0.5,
dmoment=1.0e22,
zero_trace=True,
weight_data=True,
station_correction=True,
weight_config=weight_config)
srcinv = Cmt3D(cmt, dcon_two, config)
srcinv.source_inversion()
return srcinv
def test_cmt_bootstrap(cmtsource, tmpdir):
dcon_two = construct_dcon_two()
weight_config = DefaultWeightConfig(normalize_by_energy=False,
normalize_by_category=False,
comp_weight={
"Z": 1.0,
"R": 1.0,
"T": 1.0
},
love_dist_weight=1.0,
pnl_dist_weight=1.0,
rayleigh_dist_weight=1.0,
azi_exp_idx=0.5)
config = Config(9,
dlocation=0.5,
ddepth=0.5,
dmoment=1.0e22,
zero_trace=True,
weight_data=True,
station_correction=True,
weight_config=weight_config,
bootstrap=True,
bootstrap_repeat=20,
bootstrap_subset_ratio=0.4)
srcinv = Cmt3D(cmtsource, dcon_two, config)
srcinv.source_inversion()
srcinv.plot_summary(str(tmpdir))
def default_config():
return DefaultWeightConfig(normalize_by_energy=False,
normalize_by_category=False,
comp_weight={
"Z": 2.0,
"R": 1.0,
"T": 2.0
},
love_dist_weight=0.78,
pnl_dist_weight=1.15,
rayleigh_dist_weight=0.55,
azi_exp_idx=0.5)
def main(cmt_filename):
# Define parameter directory
param_path = os.path.join(
os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "params")
# Load Database Parameters
databaseparam_path = os.path.join(param_path,
"Database/DatabaseParameters.yml")
DB_params = smart_read_yaml(databaseparam_path, mpi_mode=is_mpi_env())
# Inversion Params
inversionparam_path = os.path.join(param_path,
"CMTInversion/InversionParams.yml")
INV_params = smart_read_yaml(inversionparam_path, mpi_mode=is_mpi_env())
# Get processing path from cmt_filename in database
cmt_dir = os.path.dirname(os.path.abspath(cmt_filename))
# Create cmt source:
cmtsource = CMTSource.from_CMTSOLUTION_file(cmt_filename)
# Window directory
window_dir = os.path.join(cmt_dir, "window_data")
# Inversion dictionary directory
inv_dict_dir = os.path.join(cmt_dir, "inversion", "inversion_dicts")
# Inversion dictionaries
inv_dicts = glob.glob(os.path.join(inv_dict_dir, "*"))
# Inversion output directory
inv_out_dir = os.path.join(cmt_dir, "inversion", "inversion_output")
if DB_params["verbose"]:
print("\n#######################################################")
print("# #")
print("# Starting inversion ... #")
print("# #")
print("#######################################################\n")
# Creating Data container
dcon = DataContainer(parlist=PARLIST[:DB_params["npar"]])
for _i, inv_dict in enumerate(inv_dicts):
# Get processing band
bandstring = str(os.path.basename(inv_dict)).split(".")[1]
band = [float(x) for x in bandstring.split("_")]
if DB_params["verbose"]:
print("\n")
print(" " + 54 * "*")
print(" Getting data for inversion from period band:")
print(" Low: %d s || High: %d s" % tuple(band))
print(" " + 54 * "*" + "\n")
# Load inversion file dictionary
asdf_dict = smart_read_yaml(inv_dict, mpi_mode=is_mpi_env())
window_files = glob.glob(
os.path.join(window_dir,
"windows." + bandstring + "*[!stats].json"))
# Adding measurements
for _j, window_file in enumerate(window_files):
# Print Inversion parameters:
if DB_params["verbose"]:
print(" Adding measurements to data container:")
print(
" _____________________________________________________\n"
)
# Add measurements from ASDF file and windowfile
# if _j == 0:
if DB_params["verbose"]:
print(" Window file:\n", " ", window_file)
print("\n ASDF files:")
for key, value in asdf_dict.items():
print(" ", key + ":", value)
dcon.add_measurements_from_asdf(window_file, asdf_dict)
# else:
# if DB_params["verbose"]:
# print(" Window file:\n", " ", window_file)
# dcon.add_measurements_from_sac(window_file)
if DB_params["verbose"]:
print(
" _____________________________________________________\n"
)
print(" ... \n\n")
if DB_params["verbose"]:
print(" Inverting for a new moment tensor .... ")
print(" " + 54 * "*" + "\n\n")
# Setting up weight config
weight_config = DefaultWeightConfig(normalize_by_energy=False,
normalize_by_category=False,
comp_weight={
"Z": 1.0,
"R": 1.0,
"T": 1.0
},
love_dist_weight=1.0,
pnl_dist_weight=1.0,
rayleigh_dist_weight=1.0,
azi_exp_idx=0.5)
# Setting up general inversion config
config = Config(DB_params["npar"],
dlocation=float(INV_params["config"]["dlocation"]),
ddepth=float(INV_params["config"]["ddepth"]),
dmoment=float(INV_params["config"]["dmoment"]),
weight_data=True,
station_correction=True,
zero_trace=True,
double_couple=False,
bootstrap=True,
bootstrap_repeat=100,
weight_config=weight_config)
srcinv = Cmt3D(cmtsource, dcon, config)
srcinv.source_inversion()
# plot result
srcinv.plot_summary(inv_out_dir, figure_format="pdf")
if DB_params["verbose"]:
print(" DONE inversion for period band: %d - %d s.\n" % tuple(band))
if DB_params["verbose"]:
print("\n#######################################################")
print("# #")
print("# Inversion DONE. #")
print("# #")
print("#######################################################\n")