fix: adding polarity to LS datastructure

Loop3D · Dec 3, 2021 · 61842f0 · 61842f0
1 parent 2e185b5
commit 61842f0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/LoopStructural/modelling/core/geological_model.py b/LoopStructural/modelling/core/geological_model.py
@@ -509,13 +509,17 @@ def data(self, data):
                     self._data[h] = 1.0
                 if h == "coord":
                     self._data[h] = 0
+                if h == "polarity":
+                    self._data[h] = 1.
+        # LS wants polarity as -1 or 1, change 0 to -1
+        self._data.loc[self._data["polarity"]==0, "polarity"] = -1.
         self.data.loc[np.isnan(self.data["w"]), "w"] = 1.0
         if "strike" in self._data and "dip" in self._data:
             logger.info("Converting strike and dip to vectors")
             mask = np.all(~np.isnan(self._data.loc[:, ["strike", "dip"]]), axis=1)
             self._data.loc[mask, gradient_vec_names()] = strike_dip_vector(
                 self._data.loc[mask, "strike"], self._data.loc[mask, "dip"]
-            )
+            )*self._data.loc[mask, "polarity"][:,None]
             self._data.drop(["strike", "dip"], axis=1, inplace=True)
 
     def set_model_data(self, data):

diff --git a/LoopStructural/utils/helper.py b/LoopStructural/utils/helper.py
@@ -400,6 +400,8 @@ def interface_name():
 def feature_name():
     return ["feature_name"]
 
+def polarity_name():
+    return ["polarity"]
 
 def all_heading():
     return (
@@ -412,6 +414,7 @@ def all_heading():
         + coord_name()
         + feature_name()
         + interface_name()
+        + polarity_name()
     )