Merge pull request #283 from bclayton-usgs/gmm-dist

Ground Motion Vs. Distance Calculations

src/gov/usgs/earthquake/nshmp/GroundMotions.java

@@ -0,0 +1,109 @@
+package gov.usgs.earthquake.nshmp;
+
+//....................................... Import ..........................................
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.Maps;
+import com.google.common.primitives.Doubles;
+
+import gov.usgs.earthquake.nshmp.data.Data;
+import gov.usgs.earthquake.nshmp.gmm.Gmm;
+import gov.usgs.earthquake.nshmp.gmm.GmmInput;
+import gov.usgs.earthquake.nshmp.gmm.Imt;
+import gov.usgs.earthquake.nshmp.gmm.ScalarGroundMotion;
+import gov.usgs.earthquake.nshmp.util.Maths;
+//--------------------------------------- End Import --------------------------------------
+
+
+
+public class GroundMotions {
+
+
+  //................................ Method: distanceGroundMotion .........................
+  public static DistanceResult distanceGroundMotions(
+      Set<Gmm> gmms,
+      GmmInput inputModel,
+      Imt imt,
+      double rMax) {
+
+
+    //.............................. Set Distance and Gmm Inputs ..........................
+    int round = 5;
+  		double rJB;
+    double rX;
+    double rRup;
+    double rMin = 0.001;
+    double rPoints = 100.0;
+    double rStep = (Math.log10(rMax/rMin))/(rPoints-1);
+    double[] distance = Data.round(round, Data.pow10(Data.buildSequence(
+        Math.log10(rMin), Math.log10(rMax), rStep, true)));
+
+    List<GmmInput> gmmInputs = new ArrayList<>(); 
+
+    for(double r : distance) {
+      rJB = r;
+      rX = r;
+      rRup = Maths.hypot(r, inputModel.zTop);
+      GmmInput.Builder gmmBuilder = GmmInput.builder().fromModel(inputModel);
+      gmmInputs.add(gmmBuilder.distances(rJB, rRup, rX).build());
+    }
+    //-------------------------------------------------------------------------------------
+
+
+    //........................... Calculate Ground Motion .................................
+    Map<Gmm, List<Double>> distanceMap = Maps.newEnumMap(Gmm.class);
+    Map<Gmm, List<Double>> meanMap = Maps.newEnumMap(Gmm.class);
+    Map<Gmm, List<Double>> sigmaMap = Maps.newEnumMap(Gmm.class);
+
+    for(Gmm gmm : gmms) {
+      ImmutableList.Builder<Double> means = ImmutableList.builder();
+      ImmutableList.Builder<Double> sigmas = ImmutableList.builder();
+
+      for(GmmInput gmmInput : gmmInputs) {
+        ScalarGroundMotion gm = gmm.instance(imt).calc(gmmInput);
+        means.add(gm.mean());
+        sigmas.add(gm.sigma()); 
+      }
+
+      meanMap.put(gmm, means.build());
+      sigmaMap.put(gmm, sigmas.build());
+      distanceMap.put(gmm, Doubles.asList(distance));
+    }
+    //-------------------------------------------------------------------------------------
+
+    return new DistanceResult(
+        Maps.immutableEnumMap(distanceMap),
+        Maps.immutableEnumMap(meanMap),
+        Maps.immutableEnumMap(sigmaMap)
+    );
+  }
+  //------------------------------ End Method: distanceGroundMotion -----------------------
+
+
+
+  //............................. Class: DistanceResult ...................................
+  public static class DistanceResult {
+
+    public final Map<Gmm, List<Double>> means;
+    public final Map<Gmm, List<Double>> distance;
+    public final Map<Gmm, List<Double>> sigmas;
+
+    DistanceResult(
+        Map<Gmm, List<Double>> distance,
+        Map<Gmm, List<Double>> means,
+        Map<Gmm, List<Double>> sigmas){
+      this.distance = distance;    
+      this.means = means;
+      this.sigmas = sigmas;
+    }
+  }
+  //------------------------------ End Class: DistanceResult ------------------------------
+
+
+
+}
+//-------------------------------- End Class: GroundMotions -------------------------------

src/gov/usgs/earthquake/nshmp/gmm/GmmInput.java

@@ -194,10 +194,11 @@ public Builder fromModel(GmmInput model) {
       b.vsInf = model.vsInf;
       b.z1p0 = model.z1p0;
       b.z2p5 = model.z2p5;
-      flags.set(0, SIZE);
+      b.flags.set(0, SIZE);
       return b;
     }
 
+
     /**
      * Return a {@code Builder} prepopulated with default values. Builder has
      * the following presets: