fix: add featureId when parsing fault_orientations (#177)

Loop3D · Jan 13, 2025 · 924c2cf · 924c2cf
1 parent 9ca260f
commit 924c2cf
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/map2loop/mapdata.py b/map2loop/mapdata.py
@@ -763,6 +763,14 @@ def parse_fault_orientations(self) -> tuple:
         else:
             fault_orientations["ID"] = numpy.arange(len(fault_orientations))
         self.data[Datatype.FAULT_ORIENTATION] = fault_orientations
+
+        if config["featureid_column"] in self.raw_data[Datatype.FAULT_ORIENTATION]:
+            fault_orientations["featureId"] = self.raw_data[Datatype.FAULT_ORIENTATION][
+                config["featureid_column"]
+            ]
+        else:
+            fault_orientations["featureId"] = numpy.arange(len(fault_orientations))
+
         return (False, "")
 
     @beartype.beartype