LoopStructural 1.0.4__zip

Miniconda/envs/loop/Lib/site-packages/LoopStructural/visualisation/rotation_angle_plotter.py

@@ -0,0 +1,82 @@
+import logging
+
+import matplotlib.pyplot as plt
+import numpy as np
+
+logger = logging.getLogger(__name__)
+
+
+class RotationAnglePlotter:
+    def __init__(self, feature, axis=True):
+        """
+
+        """
+        self.fig, self.ax = plt.subplots(2, 2, figsize=(30, 15))
+        self.ax[0][0].set_ylim(-90, 90)
+        self.ax[1][0].set_ylim(-90, 90)
+        self.feature = feature
+
+    def plot(self, x, y, ix, iy, symb,**kwargs):
+        """
+
+        Parameters
+        ----------
+        x : np.array
+            vector of x
+        y
+        ix
+        iy
+        symb
+
+        Returns
+        -------
+
+        """
+        return self.ax[iy][ix].plot(x, y, symb,**kwargs)
+    def default_titles(self):
+        self.ax[0][0].set_title('Fold Axis S-Plot')
+        self.ax[0][1].set_title('Fold Axis S-Variogram')
+        self.ax[1][0].set_title('Fold Limb S-Plot')
+        self.ax[1][1].set_title('Fold Limb S-Variogram')
+
+        self.ax[1][1].set_xlabel('Variogram Steps')
+        self.ax[1][1].set_ylabel('Fold Limb S-Variogram')
+        self.ax[1][0].set_ylabel('Fold Limb Rotation Angle')
+        self.ax[1][0].set_xlabel('Fold Frame Axial Surface Field')
+
+        self.ax[0][1].set_xlabel('Variogram Steps')
+        self.ax[0][1].set_ylabel('Fold Axis S-Variogram')
+        self.ax[0][0].set_ylabel('Fold Axis Rotation Angle')
+        self.ax[0][0].set_xlabel('Fold Frame Axis Direction Field')
+    def add_fold_limb_data(self, symb="bo",**kwargs):
+        fold_frame = self.feature.fold.fold_limb_rotation.fold_frame_coordinate
+        rotation = self.feature.fold.fold_limb_rotation.rotation_angle
+        return self.plot(fold_frame, rotation, 0, 1, symb,**kwargs)
+        
+
+    def add_fold_limb_curve(self, symb='r-',**kwargs):
+        x = np.linspace(self.feature.fold.foldframe[0].min(),self.feature.fold.foldframe[0].max(),100)
+        return self.plot(x,self.feature.fold.fold_limb_rotation(x), 0, 1, symb,**kwargs)
+
+    def add_axis_svariogram(self, symb='bo',**kwargs):
+        svariogram = self.feature.fold.fold_axis_rotation.svario
+        if svariogram:
+            svariogram.calc_semivariogram()
+            return self.plot(svariogram.lags, svariogram.variogram, 1, 0, symb,**kwargs)
+
+    def add_limb_svariogram(self, symb='bo', **kwargs):
+        svariogram = self.feature.fold.fold_limb_rotation.svario
+        if svariogram:
+            svariogram.calc_semivariogram()
+            return self.plot(svariogram.lags, svariogram.variogram, 1, 1, symb,**kwargs)
+
+    def add_fold_axis_data(self, symb='bo',**kwargs):
+        fold_frame = self.feature.fold.fold_axis_rotation.fold_frame_coordinate
+        rotation = self.feature.fold.fold_axis_rotation.rotation_angle
+        return self.plot(fold_frame, rotation, 0, 0, symb,**kwargs)
+
+    def add_fold_axis_curve(self, symb='r-',**kwargs):
+        x = np.linspace(self.feature.fold.foldframe[1].min(),self.feature.fold.foldframe[1].max(),100)
+        return self.plot(x,self.feature.fold.fold_axis_rotation(x), 0, 0, symb,**kwargs)
+
+

Miniconda/envs/loop/Lib/site-packages/LoopStructural/visualisation/sphinx_scraper.py

@@ -0,0 +1,34 @@
+from LoopStructural.visualisation import model_visualisation as model_visualisation
+def _get_loop_visualisation_scraper():
+    return Scraper()
+
+class Scraper:
+    """
+    Save ``pyvista.Plotter`` objects.
+
+    Used by sphinx-gallery to generate the plots from the code in the examples.
+
+    Pass an instance of this class to ``sphinx_gallery_conf`` in your
+    ``conf.py`` as the ``"image_scrapers"`` argument.
+    """
+
+    def __call__(self, block, block_vars, gallery_conf):
+        """Save the figures generated after running example code.
+
+        Called by sphinx-gallery.
+
+        """
+        try:
+            from sphinx_gallery.scrapers import figure_rst
+        except ImportError:
+            raise ImportError('You must install `sphinx_gallery`')
+        image_names = list()
+        image_path_iterator = block_vars["image_path_iterator"]
+        figures = model_visualisation._OPEN_VIEWERS
+        for address, plotter in figures.items():
+            plotter.lv['background'] = 'white'
+            fname = next(image_path_iterator)
+            plotter.save(fname)
+            image_names.append(fname)
+        model_visualisation.close_all() # close and clear all plotters
+        return figure_rst(image_names, gallery_conf["src_dir"])

Miniconda/envs/loop/Lib/site-packages/LoopStructural-1.0.4-py3.7.egg-info/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 1.0
+Name: LoopStructural
+Version: 1.0.4
+Summary: UNKNOWN
+Home-page: UNKNOWN
+Author: UNKNOWN
+Author-email: UNKNOWN
+License: UNKNOWN
+Description: UNKNOWN
+Platform: UNKNOWN

Miniconda/envs/loop/Lib/site-packages/LoopStructural-1.0.4-py3.7.egg-info/SOURCES.txt

@@ -0,0 +1,60 @@
+README.md
+setup.py
+LoopStructural/__init__.py
+LoopStructural.egg-info/PKG-INFO
+LoopStructural.egg-info/SOURCES.txt
+LoopStructural.egg-info/dependency_links.txt
+LoopStructural.egg-info/requires.txt
+LoopStructural.egg-info/top_level.txt
+LoopStructural/interpolators/cython/dsi_helper.c
+LoopStructural/datasets/__init__.py
+LoopStructural/datasets/_base.py
+LoopStructural/interpolators/__init__.py
+LoopStructural/interpolators/discrete_fold_interpolator.py
+LoopStructural/interpolators/discrete_interpolator.py
+LoopStructural/interpolators/finite_difference_interpolator.py
+LoopStructural/interpolators/geological_interpolator.py
+LoopStructural/interpolators/operator.py
+LoopStructural/interpolators/piecewiselinear_interpolator.py
+LoopStructural/interpolators/structured_grid.py
+LoopStructural/interpolators/structured_tetra.py
+LoopStructural/interpolators/surfe_wrapper.py
+LoopStructural/interpolators/cython/__init__.py
+LoopStructural/interpolators/cython/dsi_helper.c
+LoopStructural/modelling/__init__.py
+LoopStructural/modelling/core/__init__.py
+LoopStructural/modelling/core/geological_model.py
+LoopStructural/modelling/fault/__init__.py
+LoopStructural/modelling/fault/fault_function.py
+LoopStructural/modelling/fault/fault_function_feature.py
+LoopStructural/modelling/fault/fault_segment.py
+LoopStructural/modelling/features/__init__.py
+LoopStructural/modelling/features/cross_product_geological_feature.py
+LoopStructural/modelling/features/geological_feature.py
+LoopStructural/modelling/features/geological_feature_builder.py
+LoopStructural/modelling/features/region_feature.py
+LoopStructural/modelling/features/structural_frame.py
+LoopStructural/modelling/features/structural_frame_builder.py
+LoopStructural/modelling/features/unconformity_feature.py
+LoopStructural/modelling/fold/__init__.py
+LoopStructural/modelling/fold/fold.py
+LoopStructural/modelling/fold/fold_rotation_angle.py
+LoopStructural/modelling/fold/fold_rotation_angle_feature.py
+LoopStructural/modelling/fold/foldframe.py
+LoopStructural/modelling/fold/svariogram.py
+LoopStructural/utils/__init__.py
+LoopStructural/utils/exceptions.py
+LoopStructural/utils/helper.py
+LoopStructural/utils/map2loop.py
+LoopStructural/utils/utils.py
+LoopStructural/visualisation/__init__.py
+LoopStructural/visualisation/map_viewer.py
+LoopStructural/visualisation/model_plotter.py
+LoopStructural/visualisation/model_visualisation.py
+LoopStructural/visualisation/rotation_angle_plotter.py
+LoopStructural/visualisation/sphinx_scraper.py
+tests/__init__.py
+tests/test_faults.py
+tests/test_fold.py
+tests/test_interpolator.py
+tests/test_refolded.py

Miniconda/envs/loop/Lib/site-packages/LoopStructural-1.0.4-py3.7.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

Miniconda/envs/loop/Lib/site-packages/LoopStructural-1.0.4-py3.7.egg-info/requires.txt

@@ -0,0 +1,8 @@
+Cython
+numpy
+pandas
+scipy
+matplotlib
+lavavu
+scikit-image
+scikit-learn

Miniconda/envs/loop/Lib/site-packages/LoopStructural-1.0.4-py3.7.egg-info/top_level.txt

@@ -0,0 +1,2 @@
+LoopStructural
+tests

Miniconda/envs/loop/Lib/site-packages/tests/test_faults.py

@@ -0,0 +1,17 @@
+from LoopStructural import GeologicalModel
+from LoopStructural.datasets import load_intrusion
+data, bb = load_intrusion()
+model = GeologicalModel(bb[0,:],bb[1,:])
+model.set_model_data(data)
+fault = model.create_and_add_fault('fault',
+                                   500,
+                                   nelements=2000,
+                                   steps=4,
+                                   interpolatortype='PLI',
+                                   buffer=0.2)
+strati = model.create_and_add_foliation('strati',
+                                        nelements=30000,
+                                        interpolatortype='PLI',
+                                        cgw=0.1,
+                                        buffer=0.1
+                                        )

Miniconda/envs/loop/Lib/site-packages/tests/test_fold.py

@@ -0,0 +1,57 @@
+from LoopStructural import GeologicalModel
+from LoopStructural.datasets import load_noddy_single_fold
+
+import pandas as pd
+
+data, boundary_points = load_noddy_single_fold()
+data.head()
+def test_average_fold_axis():
+    mdata = pd.concat([data[:100],data[data['feature_name']=='s1']])
+    model = GeologicalModel(boundary_points[0,:],boundary_points[1,:])
+    model.set_model_data(mdata)
+    fold_frame = model.create_and_add_fold_frame('s1',nelements=10000)
+    stratigraphy = model.create_and_add_folded_foliation('s0',
+                                                   fold_frame,
+                                                    nelements=10000,
+                                                    av_fold_axis=True
+                                                   # fold_axis=[-6.51626577e-06, -5.00013645e-01, -8.66017526e-01],
+                                                   # limb_wl=1
+                                                         )
+def test_fixed_fold_axis():
+    mdata = pd.concat([data[:100],data[data['feature_name']=='s1']])
+    model = GeologicalModel(boundary_points[0,:],boundary_points[1,:])
+    model.set_model_data(mdata)
+    fold_frame = model.create_and_add_fold_frame('s1',nelements=10000)
+    stratigraphy = model.create_and_add_folded_foliation('s0',
+                                                   fold_frame,
+                                                    nelements=10000,
+                                                    # av_fold_axis=True
+                                                   fold_axis=[-6.51626577e-06, -5.00013645e-01, -8.66017526e-01],
+                                                   # limb_wl=1
+                                                         )
+
+def test_fixed_wavelength():
+    mdata = pd.concat([data[:100],data[data['feature_name']=='s1']])
+    model = GeologicalModel(boundary_points[0,:],boundary_points[1,:])
+    model.set_model_data(mdata)
+    fold_frame = model.create_and_add_fold_frame('s1',nelements=10000)
+    stratigraphy = model.create_and_add_folded_foliation('s0',
+                                                   fold_frame,
+                                                    nelements=10000,
+                                                    # av_fold_axis=True
+                                                   fold_axis=[-6.51626577e-06, -5.00013645e-01, -8.66017526e-01],
+                                                   limb_wl=1
+                                                         )
+
+def test_no_fold_frame():
+    mdata = pd.concat([data[:100],data[data['feature_name']=='s1']])
+    model = GeologicalModel(boundary_points[0,:],boundary_points[1,:])
+    model.set_model_data(mdata)
+    fold_frame = model.create_and_add_fold_frame('s1',nelements=10000)
+    stratigraphy = model.create_and_add_folded_foliation('s0',
+                                                   # fold_frame,
+                                                    nelements=10000,
+                                                    # av_fold_axis=True
+                                                   fold_axis=[-6.51626577e-06, -5.00013645e-01, -8.66017526e-01],
+                                                   limb_wl=1
+                                                         )

Miniconda/envs/loop/Lib/site-packages/tests/test_interpolator.py

@@ -0,0 +1,88 @@
+from LoopStructural import GeologicalModel
+from LoopStructural.datasets import load_claudius
+
+def test_create_model():
+    data, bb = load_claudius()
+    model = GeologicalModel(bb[0,:],bb[1,:])
+
+def test_add_data():
+    data, bb = load_claudius()
+    model = GeologicalModel(bb[0,:],bb[1,:])
+    model.set_model_data(data)
+
+def test_create_stratigraphy_FDI_cg():
+    data, bb = load_claudius()
+    model = GeologicalModel(bb[0, :], bb[1, :])
+    model.set_model_data(data)
+    s0 = model.create_and_add_foliation('s0',
+                                        interpolatortype='FDI',
+                                        nelements=1000,
+                                        solver='cg',
+                                        damp=False)
+
+
+def test_remove_constraints_PLI():
+    data, bb = load_claudius()
+    model = GeologicalModel(bb[0, :], bb[1, :])
+    model.set_model_data(data)
+    s0 = model.create_and_add_foliation('s0',
+                                        interpolatortype='FDI',
+                                        nelements=1000,
+                                        solver='cg',
+                                        damp=False)
+
+def test_create_stratigraphy_FDI_lu():
+    data, bb = load_claudius()
+    model = GeologicalModel(bb[0, :], bb[1, :])
+    model.set_model_data(data)
+    s0 = model.create_and_add_foliation('s0',
+                                        interpolatortype='FDI',
+                                        nelements=1000,
+                                        solver='lu',
+                                        damp=True)
+
+
+def test_create_stratigraphy_FDI_pyamg():
+    data, bb = load_claudius()
+    model = GeologicalModel(bb[0, :], bb[1, :])
+    model.set_model_data(data)
+    s0 = model.create_and_add_foliation('s0',
+                                        interpolatortype='FDI',
+                                        nelements=1000,
+                                        solver='pyamg',
+                                        damp=True)
+
+def test_create_stratigraphy_PLI_cg():
+    data, bb = load_claudius()
+    model = GeologicalModel(bb[0, :], bb[1, :])
+    model.set_model_data(data)
+    s0 = model.create_and_add_foliation('s0',
+                                        interpolatortype='PLI',
+                                        nelements=1000,
+                                        solver='cg',
+                                        damp=False)
+
+def test_create_stratigraphy_PLI_lu():
+    data, bb = load_claudius()
+    model = GeologicalModel(bb[0, :], bb[1, :])
+    model.set_model_data(data)
+    s0 = model.create_and_add_foliation('s0',
+                                        interpolatortype='PLI',
+                                        nelements=1000,
+                                        solver='lu',
+                                        damp=True)
+
+
+def test_create_stratigraphy_PLI_pyamg():
+    data, bb = load_claudius()
+    model = GeologicalModel(bb[0, :], bb[1, :])
+    model.set_model_data(data)
+    s0 = model.create_and_add_foliation('s0',
+                                        interpolatortype='PLI',
+                                        nelements=1000,
+                                        solver='pyamg',
+                                        damp=True)
+
+def test_model_with_data_outside_of_bounding_box():
+    pass
+

Miniconda/envs/loop/Lib/site-packages/tests/test_refolded.py

@@ -0,0 +1,22 @@
+from LoopStructural import GeologicalModel
+from LoopStructural.datasets import load_laurent2016
+
+def average_axis():
+    data, bb = load_laurent2016()
+
+    model = GeologicalModel(bb[0,:],bb[1,:])
+    model.set_model_data(data)
+    s2 = model.create_and_add_fold_frame('s2',
+                                         nelements=10000)
+
+    s1 = model.create_and_add_folded_fold_frame('s1',
+                                                limb_wl=.4,
+                                                av_fold_axis=True,
+                                                nelements=50000
+                                               )
+
+    s0 = model.create_and_add_folded_fold_frame('s0',
+                                                limb_wl=1.,
+                                                av_fold_axis=True,
+                                                nelements=50000
+                                               )