set extra header markers using seisFile methods

crotwell · Oct 24, 2024 · d3a3852 · d3a3852
1 parent 9c13f28
commit d3a3852
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 8 deletions.
diff --git a/src/main/java/edu/sc/seis/TauP/cmdline/TauP_SetMSeed3.java b/src/main/java/edu/sc/seis/TauP/cmdline/TauP_SetMSeed3.java
@@ -15,6 +15,7 @@
 
 import java.io.*;
 
+import edu.sc.seis.seisFile.mseed3.ehbag.Marker;
 import org.json.JSONArray;
 import org.json.JSONObject;
 import picocli.CommandLine;
@@ -192,6 +193,11 @@ public void processRecord(MSeed3Record dr3) throws TauPException {
                 if (evTime == null) {
                     System.err.println("Unable to extract event origin time, skipping record");
                 }
+
+                List<Marker> mList = new ArrayList<>();
+                for (Arrival arrival : arrivals) {
+                    mList.add(createEHMarker(arrival, evTime));
+                }
                 insertMarkers(bag, arrivals, evTime);
             }
         }
@@ -208,12 +214,10 @@ public static void insertMarkers(JSONObject bag, List<Arrival> arrivals, Instant
         bag.put("mark", markers);
     }
 
-    public static JSONObject createEHMarker(Arrival arrival, Instant evTime) {
-        JSONObject mark = new JSONObject();
-        mark.put("n", arrival.getName());
-        mark.put("tm", TimeUtils.toISOString(evTime.plusMillis(Math.round(arrival.getTime()*1000))));
-        mark.put("mtype", "md");
-        return mark;
+    public static Marker createEHMarker(Arrival arrival, Instant evTime) {
+        Marker m = new Marker(arrival.getName(), evTime.plusMillis(Math.round(arrival.getTime()*1000)).atZone(TimeUtils.TZ_UTC));
+        m.setType(MSeed3EHKeys.MARKER_MODELED);
+        return m;
     }
 
     Event findQuakeInTime(Instant time, Duration tol) {
@@ -244,7 +248,10 @@ public String getEhKey() {
     }
 
     @CommandLine.Option(names = "--taupeh",
-            description = "key to store full TauP JSON output within extra headers within, otherwise use abbreviated 'bag' style markers.")
+            arity = "0..1",
+            fallbackValue = "taup",
+            description = "key to store full TauP JSON output within extra headers within, otherwise use abbreviated 'bag' style markers."+
+                    "If specified without parameter, extra header key of ${FALLBACK-VALUE} will be used.")
     public void setEhKey(String ehKey) {
         this.ehKey = ehKey;
     }

diff --git a/src/main/java/edu/sc/seis/TauP/cmdline/TauP_Spikes.java b/src/main/java/edu/sc/seis/TauP/cmdline/TauP_Spikes.java
@@ -33,6 +33,7 @@
 import edu.sc.seis.seisFile.fdsnws.quakeml.*;
 import edu.sc.seis.seisFile.mseed3.FDSNSourceId;
 import edu.sc.seis.seisFile.mseed3.MSeed3EH;
+import edu.sc.seis.seisFile.mseed3.MSeed3EHKeys;
 import edu.sc.seis.seisFile.mseed3.MSeed3Record;
 import edu.sc.seis.seisFile.mseed3.ehbag.Marker;
 import edu.sc.seis.seisFile.mseed3.ehbag.Path;
@@ -96,6 +97,11 @@ public void validateArguments() throws TauPException {
         }
     }
 
+    @Override
+    public void init() throws TauPException {
+        super.init();
+    }
+
     @Override
     public void start() throws IOException, TauPException {
         try {
@@ -220,7 +226,7 @@ public MSeed3EH createEH(double degrees, ZonedDateTime startDateTime, List<Arriv
         for (Arrival a : allArrivals) {
             ZonedDateTime arrTime = startDateTime.plusNanos(Math.round(a.getTime() * 1e9));
             String desc = "";
-            Marker m = new Marker(a.getName(), arrTime, Marker.MARKER_PREDIC_MODEL, desc);
+            Marker m = new Marker(a.getName(), arrTime, MSeed3EHKeys.MARKER_MODELED, desc);
             eh.addToBag(m);
         }
         return eh;