LoopStructural 1.0.3__zip → 1.0.71.dev0__zip

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

@@ -1,3 +1,6 @@
+"""
+Wrapper for using surfepy
+"""
 import sys, os
 
 from LoopStructural.utils.helper import get_vectors
@@ -6,13 +9,9 @@ from .geological_interpolator import GeologicalInterpolator
 import logging
 import numpy as np
 
-logger = logging.getLogger(__name__)
+from LoopStructural.utils import getLogger
+logger = 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
 
 
@@ -77,13 +76,13 @@ class SurfeRBFInterpolator(GeologicalInterpolator):
         regression = kwargs.get("regression_smoothing", 0.)
         if regression > 0:
             logger.info("Using regression smoothing %f" % regression)
-
+            self.surfe.SetRegressionSmoothing(True,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])
+            self.surfe.SetGreedyAlgorithm(True, greedy[0], greedy[1])
         poly_order = kwargs.get('poly_order',None)
         if poly_order:
             logger.info("Setting poly order to %i"%poly_order)