Per-rel vs30 perturbation and some minor edits

ucgmsim · Sep 9, 2024 · f8afadb · f8afadb
1 parent bc28747
commit f8afadb
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 14 deletions.
diff --git a/srf_generation/source_parameter_generation/uncertainties/versions/sjn_realisations_vO2syn.py b/srf_generation/source_parameter_generation/uncertainties/versions/sjn_realisations_vO2syn.py
@@ -138,16 +138,16 @@ def generate_source_params(
     #measurement_uncertainty = 0.1
     #site_model_uncertainty = 0.3
     #total_vs30_uncertainty = vs30_data["sigma"].values
-    total_vs30_uncertainty = 0.2
-
-    if vs30_data is not None:
-        print("Got vs30")
-        realisation["vs30"] = vs30_data.copy(deep=True)
-        #print(vs30_data["median"].values)
-        #realisation["vs30"]["vs30"] = distributions.truncated_log_normal(vs30_data["median"].values, total_vs30_uncertainty, 4)
-        realisation["vs30"]["vs30"] = vs30_data["median"].values
-    else:
-        print("Didn't get vs30")
+    #total_vs30_uncertainty = 0.2
+
+    #if vs30_data is not None:
+    #    print("Got vs30")
+    #    realisation["vs30"] = vs30_data.copy(deep=True)
+    #    #print(vs30_data["median"].values)
+    #    #realisation["vs30"]["vs30"] = distributions.truncated_log_normal(vs30_data["median"].values, total_vs30_uncertainty, 4)
+    #    realisation["vs30"]["vs30"] = vs30_data["median"].values
+    #else:
+    #    print("Didn't get vs30")
 
     ### End of custom code area
 

diff --git a/srf_generation/source_parameter_generation/uncertainties/vs30_perturb.py b/srf_generation/source_parameter_generation/uncertainties/vs30_perturb.py
@@ -11,9 +11,8 @@
 
 
 def perturb_vs30(vs30_data, measurement_uncertainty=0.06, site_model_uncertainty=0.3):
-    total_vs30_uncertainty = np.sqrt(
-        (vs30_data["sigma"].values) ** 2 + measurement_uncertainty**2
-    )
+    total_vs30_uncertainty = 0.5
+    #total_vs30_uncertainty = np.sqrt((vs30_data["sigma"].values) ** 2 + measurement_uncertainty**2)
 
     assert vs30_data is not None
     perturbed_vs30 = vs30_data.copy(deep=True)
@@ -49,4 +48,8 @@ def perturb_vs30(vs30_data, measurement_uncertainty=0.06, site_model_uncertainty
     vs30_median_name=args.vs30_median.stem
     for i in range(args.n_rels):
         perturbed_vs30 = perturb_vs30(vs30_df)
-        perturbed_vs30.to_csv(args.outdir/f"{vs30_median_name}_REL{i+1:02d}.vs30", columns=["vs30"], sep=" ", index=True, header=False)
+        outfile=args.outdir/f"{vs30_median_name}_REL{i+1:02d}.vs30"
+        perturbed_vs30.to_csv(outfile, columns=["vs30"], sep=" ", index=True, header=False)
+        if outfile.exists():
+            print(f"{outfile} is written")
+