engeom 0.1.2__cp38-abi3-win_amd64.whl → 0.2.3__cp38-abi3-win_amd64.whl

engeom/pyvista.py

@@ -1,14 +1,19 @@
 """
-    This module contains helper functions for working with PyVista.
+This module contains helper functions for working with PyVista.
 """
+
 from __future__ import annotations
 
-from typing import List
+from typing import List, Any, Dict, Union, Iterable, Tuple
 
 import numpy
 from pyvista import ColorLike
 
-from .geom3 import Mesh, Curve3
+from .geom3 import Mesh, Curve3, Vector3, Point3
+from .metrology import Length3
+from .matplotlib import LabelPlace
+
+PlotCoords = Union[Point3, Vector3, Iterable[float]]
 
 try:
     import pyvista
@@ -16,48 +21,146 @@ except ImportError:
     pass
 else:
 
-    def mesh_polydata(mesh: Mesh) -> pyvista.PolyData:
-        """
-        Creates a PyVista PolyData object from a Mesh object.
-        :param mesh: a Mesh object
-        :return: a PyVista PolyData object
-        """
-
-        if pyvista is None:
-            raise ImportError("PyVista is not installed.")
-
-        vertices = mesh.clone_vertices()
-        faces = mesh.clone_triangles()
-        faces = numpy.hstack((numpy.ones((faces.shape[0], 1), dtype=faces.dtype) * 3, faces))
-        return pyvista.PolyData(vertices, faces)
-
-
-    def add_curves_to_plotter(
-            plotter: pyvista.Plotter,
-            curves: List[Curve3],
-            color: ColorLike = 'w',
-            width: float = 5.0,
-            label: str | None = None,
-            name: str | None = None,
-    ) -> List[pyvista.vtkActor]:
-        """
-        Adds curves to a PyVista plotter.
-        :param plotter:
-        :param curves:
-        :param color:
-        :param width:
-        :param label:
-        :param name:
-        :return:
-        """
-
-        if pyvista is None:
-            raise ImportError("PyVista is not installed.")
-
-        result_list = []
-        for curve in curves:
-            v = curve.clone_vertices()
-            added = plotter.add_lines(v, connected=True, color=color, width=width, label=label, name=name)
-            result_list.append(added)
-
-        return result_list
+    class PlotterHelper:
+        def __init__(self, plotter: pyvista.Plotter):
+            self.plotter = plotter
+
+        def add_curves(
+                self,
+                *curves: Curve3,
+                color: ColorLike = "w",
+                width: float = 5.0,
+                label: str | None = None,
+                name: str | None = None,
+        ) -> List[pyvista.vtkActor]:
+            """
+
+            :param curves:
+            :param color:
+            :param width:
+            :param label:
+            :param name:
+            :return:
+            """
+            result_list = []
+            for curve in curves:
+                added = self.plotter.add_lines(
+                    curve.points,
+                    connected=True,
+                    color=color,
+                    width=width,
+                    label=label,
+                    name=name,
+                )
+                result_list.append(added)
+
+            return result_list
+
+        def add_mesh(self, mesh: Mesh, **kwargs) -> pyvista.vtkActor:
+            """
+
+            :param mesh:
+            :return:
+            """
+            if "cmap" in kwargs:
+                cmap_extremes = _cmap_extremes(kwargs["cmap"])
+                kwargs.update(cmap_extremes)
+
+            prefix = numpy.ones((mesh.faces.shape[0], 1), dtype=mesh.faces.dtype)
+            faces = numpy.hstack((prefix * 3, mesh.faces))
+            data = pyvista.PolyData(mesh.vertices, faces)
+            return self.plotter.add_mesh(data, **kwargs)
+
+        def dimension(
+                self,
+                length: Length3,
+                template: str = "{value:.3f}",
+                label_place: LabelPlace = LabelPlace.Outside,
+                label_offset: float | None = None,
+                text_size: int = 16,
+                scale_value: float = 1.0,
+        ):
+            label_offset = label_offset or max(abs(length.value), 1.0) * 3
+
+            t_a = length.center.scalar_projection(length.a)
+            t_b = length.center.scalar_projection(length.b)
+
+            outside = length.center.at_distance(max(t_a, t_b))
+            inside = length.center.at_distance(min(t_a, t_b))
+
+            circles = []
+            builder = LineBuilder()
+
+            builder.add(inside - length.direction * label_offset * 0.25)
+            builder.add(inside)
+            circles.append(inside)
+            builder.skip()
+
+            circles.append(outside)
+            builder.add(outside)
+            builder.add(outside + length.direction * label_offset)
+
+            points = numpy.array([_tuplefy(p) for p in circles], dtype=numpy.float64)
+            self.plotter.add_points(points, color="black", point_size=4, render_points_as_spheres=True)
+
+            lines = builder.build()
+            self.plotter.add_lines(lines, color="black", width=1.5)
+
+            value = length.value * scale_value
+            label = pyvista.Label(text=template.format(value=value), position=lines[-1], size=text_size)
+            self.plotter.add_actor(label)
+
+        def arrow(self, start: PlotCoords, direction: PlotCoords,
+                  tip_length: float = 0.25,
+                  tip_radius: float = 0.1,
+                  shaft_radius: float = 0.05,
+                  **kwargs):
+            pd = pyvista.Arrow(_tuplefy(start), _tuplefy(direction), tip_length=tip_length, tip_radius=tip_radius,
+                               shaft_radius=shaft_radius)
+            self.plotter.add_mesh(pd, **kwargs, color="black")
+
+
+    def _cmap_extremes(item: Any) -> Dict[str, ColorLike]:
+        working = {}
+        try:
+            from matplotlib.colors import Colormap
+        except ImportError:
+            return working
+        else:
+            if isinstance(item, Colormap):
+                over = getattr(item, "_rgba_over", None)
+                under = getattr(item, "_rgba_under", None)
+                if over is not None:
+                    working["above_color"] = over
+                if under is not None:
+                    working["below_color"] = under
+            return working
+
+
+class LineBuilder:
+    def __init__(self):
+        self.vertices = []
+        self._skip = 1
+
+    def add(self, points: PlotCoords):
+        if self.vertices:
+            if self._skip > 0:
+                self._skip -= 1
+            else:
+                self.vertices.append(self.vertices[-1])
+
+        self.vertices.append(_tuplefy(points))
+
+    def skip(self):
+        self._skip = 2
+
+    def build(self) -> numpy.ndarray:
+        return numpy.array(self.vertices, dtype=numpy.float64)
+
+
+def _tuplefy(item: PlotCoords) -> Tuple[float, float, float]:
+    if isinstance(item, (Point3, Vector3)):
+        return item.x, item.y, item.z
+    else:
+        x, y, z, *_ = item
+        return x, y, z

{engeom-0.1.2.dist-info → engeom-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: engeom
-Version: 0.1.2
+Version: 0.2.3
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

engeom-0.2.3.dist-info/RECORD

@@ -0,0 +1,18 @@
+engeom-0.2.3.dist-info/METADATA,sha256=iGZ__VhZChSD4XCI0Wz0nfHMgMdD2GJEcaFqx6f1Jpo,339
+engeom-0.2.3.dist-info/WHEEL,sha256=_g1M2QM3kt1Ssm_sHOg_3TUY7GxNE2Ueyslb9ZDtPwk,94
+engeom/airfoil/__init__.py,sha256=G6m7JEvHVk3sM2JooJPOg8JNA3VuEp0EIqczSEbC_PY,180
+engeom/airfoil.pyi,sha256=0TVpXkolFUXbBqJp93FenA_XqvU7FD1DnbncAF0ubow,14654
+engeom/align/__init__.py,sha256=SEeMqeqLKqJC73Mg8GwPwd9NwWnl-dcCqJ4rPdh8yyc,196
+engeom/align.pyi,sha256=QCSKrTLkCoaIubcrPU9J-wDZe1lRP0GbPgWZmonXjo0,997
+engeom/engeom.pyi,sha256=Jia11rU8ZnMKdlXgfGeBPlSmsYPEfALM-_ufNwR0ibQ,254
+engeom/geom2/__init__.py,sha256=mRu8Zh6DE-EQyhxScoxszPqDjGVzGWVJEQO6RIAtS4A,174
+engeom/geom2.pyi,sha256=Jh0ES-Gvkl7sFQV7VG6KdwgDbCPhmiETG_YOObYazhU,22741
+engeom/geom3/__init__.py,sha256=DG5jt2xgS9WRNb58ZkkrcKQQO6bIG-irg-uV_BkHEj4,174
+engeom/geom3.pyi,sha256=hIIwBa-n08CaEcPR-dB_RhksPoR74HJrV4jT9CNiqhs,33556
+engeom/matplotlib.py,sha256=A0gdQshzE3G7joNHna4viYnioQtA8LVXfSuZ_X6AHeo,9001
+engeom/metrology/__init__.py,sha256=cpsB0-hJGitzW79Coxwf7r_mpNaeI6yG3myDEVdBJgk,186
+engeom/metrology.pyi,sha256=P_2pkoLUAOB0-RKppj0FN01XGY0jx1lGw9H1eKXrW8s,1144
+engeom/pyvista.py,sha256=lqqFxsl_YVXEh0_szuH-y9HWkqMXpuVxFrWJyn8mxKY,5382
+engeom/__init__.py,sha256=kYgFq3jq1quDfV013wEYQMlUBz4QNSpP6u8lFiuTHvc,115
+engeom/engeom.pyd,sha256=BfTGfLE8zCZdjb1HavQNjavko9FB5e6BYdknfbDSM70,2315776
+engeom-0.2.3.dist-info/RECORD,,

{engeom-0.1.2.dist-info → engeom-0.2.3.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.8.1)
+Generator: maturin (1.8.2)
 Root-Is-Purelib: false
 Tag: cp38-abi3-win_amd64

engeom-0.1.2.dist-info/RECORD

@@ -1,14 +0,0 @@
-engeom-0.1.2.dist-info/METADATA,sha256=ym_mnfPDlcTOzcYgeEZGMCFsgidl30syGoPDLY_B2eA,339
-engeom-0.1.2.dist-info/WHEEL,sha256=mYDPFrpmO1_UhonyEoMo2i0zHOwqc92hj_Q3CYuEBdM,94
-engeom/align/__init__.py,sha256=SEeMqeqLKqJC73Mg8GwPwd9NwWnl-dcCqJ4rPdh8yyc,196
-engeom/align.pyi,sha256=KBC0nwcyp4YMfY2hRN1gr3DqFah-unqAd_o1KYwJAqc,1022
-engeom/engeom.pyi,sha256=J44RFBaKfnr1z7WyXCeE-_wEJZvpRJBfHINfDPR93c4,204
-engeom/geom2/__init__.py,sha256=mRu8Zh6DE-EQyhxScoxszPqDjGVzGWVJEQO6RIAtS4A,174
-engeom/geom2.pyi,sha256=De3YQSGpS4UQ1zawtIsuVR4-7rcFjqmNvL9lbtzN_d4,17117
-engeom/geom3/__init__.py,sha256=DG5jt2xgS9WRNb58ZkkrcKQQO6bIG-irg-uV_BkHEj4,174
-engeom/geom3.pyi,sha256=fEFhdFmEJsDP3X0aE_N0heqMyyPnbYjpeB0hTR2xjOI,27295
-engeom/matplotlib.py,sha256=Y9JZqpEFHm5EVzbr20P-NdxndNZdDH32OhE_KjxG_S4,3025
-engeom/pyvista.py,sha256=Lyvm0K_rvhR2wZACtfwCPYwEuABqYSTdjvbBaQS9yI4,1721
-engeom/__init__.py,sha256=kYgFq3jq1quDfV013wEYQMlUBz4QNSpP6u8lFiuTHvc,115
-engeom/engeom.pyd,sha256=HaBm5BbeFpB1PMnziqm2ljTrSBgnU353rCX5H4bj3vM,1311232
-engeom-0.1.2.dist-info/RECORD,,