LoopStructural 1.0.1__zip

Miniconda/envs/loop/Lib/site-packages/LoopStructural/interpolators/surfe_wrapper.py

@@ -0,0 +1,119 @@
+import sys, os
+
+from LoopStructural.utils.helper import get_vectors
+from .geological_interpolator import GeologicalInterpolator
+
+import logging
+import numpy as np
+
+logger = logging.getLogger(__name__)
+
+if 'SURFE' in os.environ:
+    sys.path.append(os.environ['SURFE'])
+if 'SURFE' not in os.environ:
+    logger.error("Please add SURFE to your environment variables to specify the location \n "
+                 "of the SURFE binaries")
+import surfepy
+
+
+
+class SurfeRBFInterpolator(GeologicalInterpolator):
+    """
+
+    """
+    def __init__(self, method='single_surface'):
+        GeologicalInterpolator.__init__(self)
+        self.surfe = None
+        if method == 'single_surface':
+            logger.info("Using single surface interpolator")
+            self.surfe = surfepy.Surfe_API(1)
+        if method == 'Laujaunie':
+            logger.info("Using Laujaunie method")
+            self.surfe = surfepy.Surfe_API(2)
+        if method == 'horizons':
+            logger.info("Using surfe horizon")
+            self.surfe = surfepy.Surfe_API(4)
+
+    def add_gradient_ctr_pts(self):
+        points = self.get_gradient_constraints()
+        if points.shape[0] > 0:
+            logger.info("Adding ")
+            strike_vector, dip_vector = get_vectors(points[:, 3:6])
+
+            strike_vector = np.hstack([points[:, :3], strike_vector.T])
+            dip_vector = np.hstack([points[:, :3], dip_vector.T])
+            self.surfe.SetTangentConstraints(strike_vector)
+            self.surfe.SetTangentConstraints(dip_vector)
+
+
+    def add_norm_ctr_pts(self):
+        points = self.get_norm_constraints()
+        if points.shape[0]>0:
+            self.surfe.SetPlanarConstraints(points[:, :6])
+
+    def add_ctr_pts(self):
+
+        points = self.get_value_constraints()
+        if points.shape[0]> 0:
+            # self.surfe.SetInterfaceConstraints(points[:,:4])
+            for i in range(points.shape[0]):
+            
+                self.surfe.AddInterfaceConstraint(points[i, 0], points[i, 1], points[i, 2], points[i, 3], )
+
+    def add_tangent_ctr_pts(self):
+        pass
+
+    def _solve(self, **kwargs):
+        self.surfe.ComputeInterpolant()
+
+    def _setup_interpolator(self, **kwargs):
+        self.add_gradient_ctr_pts()
+        self.add_norm_ctr_pts()
+        self.add_ctr_pts()
+        self.add_tangent_ctr_pts()
+
+        kernel = kwargs.get("kernel", 'r3')
+        logger.info("Setting surfe RBF kernel to %s" % kernel)
+        regression = kwargs.get("regression_smoothing", 0.)
+        if regression > 0:
+            logger.info("Using regression smoothing %f" % regression)
+
+        greedy = kwargs.get("greedy", (0, 0))
+
+        if greedy[0] > 0 or greedy[1] > 0:
+            logger.info("Using greedy algorithm: inferface %f and angular %f" %
+                        (greedy[0], greedy[1]))
+            self.surfe.SetRegressionSmoothin(True, greedy[0], greedy[1])
+        poly_order = kwargs.get('poly_order',None)
+        if poly_order:
+            logger.info("Setting poly order to %i"%poly_order)
+            self.surfe.SetPolynomialOrder(poly_order)
+        global_anisotropy = kwargs.get("anisotropy", False)
+        if global_anisotropy:
+            logger.info("Using global anisotropy")
+            self.surfe.SetGlobalAnisotropy(global_anisotropy)
+        radius = kwargs.get("radius",False)
+        if radius:
+            logger.info("Setting RBF radius to %f"%radius)
+            self.surfe.SetRBFShapeParameter(radius)
+
+    def update(self):
+        return self.surfe.InterpolantComputed()
+
+    def evaluate_value(self, evaluation_points):
+        evaluation_points = np.array(evaluation_points)
+        evaluated = np.zeros(evaluation_points.shape[0])
+        mask = np.any(evaluation_points == np.nan, axis=1)
+
+        if evaluation_points[~mask, :].shape[0] > 0:
+            evaluated[~mask] = self.surfe.EvaluateInterpolantAtPoints(
+                evaluation_points[~mask])
+        return evaluated
+    def evaluate_gradient(self, evaluation_points):
+        evaluation_points = np.array(evaluation_points)
+        evaluated = np.zeros(evaluation_points.shape)
+        mask = np.any(evaluation_points == np.nan, axis=1)
+        if evaluation_points[~mask, :].shape[0] > 0:
+            evaluated[~mask,:] = self.surfe.EvaluateVectorInterpolantAtPoints(
+                evaluation_points[~mask])
+        return evaluated

Miniconda/envs/loop/Lib/site-packages/LoopStructural/modelling/__init__.py

@@ -0,0 +1,46 @@
+"""
+Geological modelling classes and functions
+
+Main API for creating a GeologicalModel
+=======================================
+.. autosummary::
+    :toctree: _autosummary
+
+    core.geological_model
+
+Geological Features
+===================
+.. autosummary::
+    :toctree: _autosummary
+
+    features.geological_feature
+    features.geological_feature_builder
+    features.region_feature
+    features.unconformity_feature
+    features.structural_frame
+    features.structural_frame_builder
+
+Folding classes
+===============
+.. autosummary::
+    :toctree: _autosummary
+
+    fold.fold
+    fold.fold_rotation_angle
+    fold.fold_rotation_angle_feature
+    fold.foldframe
+    fold.svariogram
+
+
+Fault classes
+===============
+
+.. autosummary::
+    :toctree: _autosummary
+
+    fault.fault_segment
+    fault.fault_function
+    fault.fault_function_feature
+
+"""
+