pfc-geometry 0.2.17__py3-none-any.whl → 0.2.20__py3-none-any.whl

geometry/__init__.py

@@ -11,6 +11,7 @@ this program. If not, see <http://www.gnu.org/licenses/>.
 """
 from .base import Base
 from .time import Time
+from . import angles as angles
 from .point import *
 from .quaternion import *
 from .gps import GPS
@@ -27,13 +28,3 @@ def Euldeg(*args, **kwargs) -> Quaternion:
     return Quaternion.from_euler(Point(*args, **kwargs).radians())
 
 
-def angle_diff(a, b): 
-    d1 = a - b
-    d2 = d1 - 2 * np.pi
-    bd=np.abs(d2) < np.abs(d1)
-    d1[bd] = d2[bd]
-    d3 = d1 + 2 * np.pi
-    bd=np.abs(d3) < np.abs(d1)
-    d1[bd] = d3[bd]
-
-    return d1

geometry/angles.py

@@ -0,0 +1,19 @@
+import numpy as np
+import numpy.typing as npt
+
+def unwind(angles: npt.NDArray, center: float = 0.0):
+    """given an angle, unwind it to the range [-pi, pi] around a center point."""
+    turns = np.round((angles - center) / (2 * np.pi)) 
+    return angles - turns * 2 * np.pi
+
+
+def difference(a, b): 
+    d1 = a - b
+    d2 = d1 - 2 * np.pi
+    bd=np.abs(d2) < np.abs(d1)
+    d1[bd] = d2[bd]
+    d3 = d1 + 2 * np.pi
+    bd=np.abs(d3) < np.abs(d1)
+    d1[bd] = d3[bd]
+
+    return d1

geometry/base.py

@@ -252,6 +252,9 @@ class Base:
     def __neg__(self) -> Self:
         return self.__class__(-self.data)
 
+    def __pow__(self, power: Number) -> Self:
+        return self.__class__(self.data ** power)
+
     @dprep
     def dot(self, other: Self) -> Self:
         return np.einsum("ij,ij->i", self.data, other)

geometry/coordinate_frame.py

@@ -87,3 +87,36 @@ class Coord(Base):
 
     def axes(self):
         return Point.concatenate([self.x_axis, self.y_axis, self.z_axis])
+
+    def plot(self, fig=None, scale=1, label: str = None):
+        import plotly.graph_objects as go
+        if fig is None:
+            fig = go.Figure(layout=dict(scene=dict(aspectmode="data")))
+
+        if len(self) > 1:
+            for c in self:
+                fig = c.plot(fig)
+            return fig
+        fig.add_trace(
+            go.Scatter3d(
+                x=self.origin.x,
+                y=self.origin.y,
+                z=self.origin.z,
+                mode="markers",
+                name="Origin",
+                marker=dict(size=5, color="black"),
+            )
+        )
+        colors = ["red", "green", "blue"]
+        for i, axis in enumerate([self.x_axis, self.y_axis, self.z_axis]):
+            fig.add_trace(
+                go.Scatter3d(
+                    x=[self.origin.x[0], (self.origin.x + axis.x * scale)[0]],
+                    y=[self.origin.y[0], (self.origin.y + axis.y * scale)[0]],
+                    z=[self.origin.z[0], (self.origin.z + axis.z * scale)[0]],
+                    mode="lines",
+                    name=f"{label or 'Axis'} {Point.cols[i]}",
+                    line=dict(width=2, color=colors.pop(0))
+                )
+            )
+        return fig

geometry/point.py

@@ -150,18 +150,45 @@ class Point(Base):
     def zeros(count=1):
         return Point(np.zeros((count, 3)))
 
+    @staticmethod
+    def circle_xy(radius: float, n: int) -> Point:
+        """
+        Generate points on a circle in the specified plane.
+        
+        :param radius: Radius of the circle.
+        :param n: Number of points to generate.
+        :return: Points on the circle as a Point object.
+        """
+        
+        angles = np.linspace(0, 2 * np.pi, n, endpoint=False)
+        return Point(radius * np.cos(angles), radius * np.sin(angles), np.zeros(n))
+
+    @staticmethod
+    def ellipse_xy(a: float, b: float, n: int) -> Point:
+        """
+        Generate points on an ellipse in the specified plane.
+        
+        :param a: Semi-major axis length.
+        :param b: Semi-minor axis length.
+        :param n: Number of points to generate.
+        :return: Points on the ellipse as a Point object.
+        """
+        
+        angles = np.linspace(0, 2 * np.pi, n, endpoint=False)
+        return Point(a * np.cos(angles), b * np.sin(angles), np.zeros(n))
+
     def bearing(self):
         return np.arctan2(self.y, self.x)
 
-    def plot3d(self, **kwargs):
+    def plot3d(self, fig=None, **kwargs):
         import plotly.graph_objects as go
-        fig = go.Figure()
 
-        fig.add_trace(go.Scatter3d(x=self.x, y=self.y, z=self.z, **kwargs))
-        fig.update_layout(
-            scene=dict(aspectmode="data"),
-        )
-        return fig
+        _fig = go.Figure() if fig is None else fig
+
+        _fig.add_trace(go.Scatter3d(x=self.x, y=self.y, z=self.z, **kwargs))
+        if fig is None:
+            _fig.update_layout(scene=dict(aspectmode="data"))
+        return _fig
 
     def plotxy(self):
         import plotly.express as px
@@ -183,6 +210,18 @@ class Point(Base):
         return px.line(self.df, x="z", y="x").update_layout(
             yaxis=dict(scaleanchor="x", scaleratio=1, title="x"), xaxis=dict(title="z")
         )
+    def plotxz(self):
+        import plotly.express as px
+
+        return px.line(self.df, x="x", y="z").update_layout(
+            yaxis=dict(scaleanchor="x", scaleratio=1, title="x"), xaxis=dict(title="z")
+        )
+    def arbitrary_perpendicular(self) -> Point:
+        min_axes = np.argmin(np.abs(self.data), axis=1)
+        cvecs = Point.concatenate(
+            [Point(*[1 if axis == i else 0 for i in np.arange(3)]) for axis in min_axes]
+        )
+        return cross(self, cvecs)
 
 
 def Points(*args, **kwargs):
@@ -262,10 +301,6 @@ def min_angle_between(p1: Point, p2: Point):
     return np.minimum(angle, np.pi - angle)
 
 
-def arbitrary_perpendicular(v: Point) -> Point:
-    return Point(-v.y, v.x, 0).unit()
-
-
 def vector_norm(point: Point):
     return abs(point)
 

geometry/quaternion.py

@@ -12,7 +12,7 @@ this program. If not, see <http://www.gnu.org/licenses/>.
 from __future__ import annotations
 from .point import Point
 from .base import Base
-from geometry import PZ
+from geometry.point import PZ
 import numpy as np
 import numpy.typing as npt
 import pandas as pd

geometry/transformation.py

@@ -40,9 +40,6 @@ class Transformation(Base):
     @property
     def q(self):    
         return Quaternion(self.data[:,3:])
-    
-    def offset(self, p: Point):
-        return Transformation(self.p + p, self.q)
 
     def __getattr__(self, name):
         if name in list("xyz"):
@@ -94,7 +91,6 @@ class Transformation(Base):
             coord_b.origin - coord_a.origin,
             -q1 * q2
         )
-
     
     def apply(self, oin: Point | Quaternion | Self | Coord):
         if isinstance(oin, Point):
@@ -115,6 +111,14 @@ class Transformation(Base):
         else:
             raise TypeError(f"expected a Point or a Quaternion, got a {oin.__class__.__name__}")
 
+    def offset(self, p: Point | Self):
+        if isinstance(p, Point):
+            return Transformation(self.p + p, self.q)
+        elif isinstance(p, self.__class__):
+            return Transformation(self.p + p.p, self.q * p.q)
+        else:
+            raise TypeError(f"expected a Point or a Transformation, got a {p.__class__.__name__}")
+
     def translate(self, point: Point):
         return point + self.p
 
@@ -134,11 +138,12 @@ class Transformation(Base):
         return outarr
         
 
-    def plot_3d(self, size: float=3, vis:Literal["coord", "plane"]="coord"):
+    def plot(self, fig=None, size: float=3, vis:Literal["coord", "plane"]="coord"):
         import plotly.graph_objects as go
         from plotting.traces import axestrace, meshes
-        import plotting.templates
-        fig = go.Figure(layout=dict(template="generic3d+clean_paper"))
+        if fig is None:
+            import plotting.templates
+            fig = go.Figure(layout=dict(template="generic3d+clean_paper"))
         if vis=="coord":
             fig.add_traces(axestrace(self, length=size))
         elif vis=="plane":

{pfc_geometry-0.2.17.dist-info → pfc_geometry-0.2.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pfc-geometry
-Version: 0.2.17
+Version: 0.2.20
 Summary: A library for working with 3D geometry.
 License-File: LICENSE
 Requires-Python: >=3.12

pfc_geometry-0.2.20.dist-info/RECORD

@@ -0,0 +1,17 @@
+geometry/__init__.py,sha256=Qipz0oxyBg-C_-sFLVI8LKW9Yq8_keR-CbyuseAvTGg,1082
+geometry/angles.py,sha256=Gw4PXr2kQMvvKECsugmi5kNiqOnK8T3PZSo3IKUpxPY,480
+geometry/base.py,sha256=6bM4J7VGhstel6qu-C4xU39vU4dZ71S9FZMY9Y7xDZY,14966
+geometry/checks.py,sha256=o8yMBAdU5Vy0EspBYaof4fPGgRSFZhRDhzBjRPsLd0M,375
+geometry/coordinate_frame.py,sha256=6KzRu0sSPzmHj5vLIuX5Efi0OWdEF6gQtKPf63wsaRk,4358
+geometry/gps.py,sha256=EsokABt40ZoltpAQfKrRc4kA-Lc2ScP_ltJNF7pvAWc,3654
+geometry/mass.py,sha256=BUWBSITwpdRfpJR5-oJTd16BI7FLZt8rhxdzr0cx1HY,1675
+geometry/point.py,sha256=wjAau7PiTpoIHMMvDDexXq8PisE5RdAQqT7Ze3NTm6A,8477
+geometry/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+geometry/quaternion.py,sha256=dZwpViiVPwHUhRk7DhOwNab3pLhIWXyHk3DVh9mukSk,12143
+geometry/time.py,sha256=VTfMHLxhcws8YESYvxxP8W_vSePvl4lwKRXHxFWhJeA,2695
+geometry/transformation.py,sha256=wnxoDN9-IeapVlrS_Da4hXyxtPCyoa_UcCsB8QVgxuo,5539
+geometry/utils.py,sha256=q7-aaxDzRDwl78-3XzdpcJuh5iL7I8lN5agk2WNWjSY,3443
+pfc_geometry-0.2.20.dist-info/METADATA,sha256=XXAr-l0CNGbvC2GTXtUj0os-mDRYN_3OK7LwAft9LoU,1629
+pfc_geometry-0.2.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pfc_geometry-0.2.20.dist-info/licenses/LICENSE,sha256=z72U6pv-bQgJ_Svr4uCXnMjemsp38aSerhHEdEAOMJ4,7632
+pfc_geometry-0.2.20.dist-info/RECORD,,

pfc_geometry-0.2.17.dist-info/RECORD

@@ -1,16 +0,0 @@
-geometry/__init__.py,sha256=HNhMyemIJzDq1nDjrr09eX5PS7q9ULscSbYsXss3JRM,1253
-geometry/base.py,sha256=aYHsY8WdXFN17EYcUTjjn7Rh9Z8LAZpvn-jsMm_niRU,14869
-geometry/checks.py,sha256=o8yMBAdU5Vy0EspBYaof4fPGgRSFZhRDhzBjRPsLd0M,375
-geometry/coordinate_frame.py,sha256=YTCAtCUuIq5LAsO-P9FwFs55f4qpWP9pUP6mf-Nhk54,3145
-geometry/gps.py,sha256=EsokABt40ZoltpAQfKrRc4kA-Lc2ScP_ltJNF7pvAWc,3654
-geometry/mass.py,sha256=BUWBSITwpdRfpJR5-oJTd16BI7FLZt8rhxdzr0cx1HY,1675
-geometry/point.py,sha256=FR2TBAV-YyKei0VA4ECDGA7ah5KRUhSKa2u_pVfAx-s,7057
-geometry/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-geometry/quaternion.py,sha256=J7Yk7qAUsH8SfjhcqQAF7Lw5hk_LGfpotVaVRyJnmeQ,12137
-geometry/time.py,sha256=VTfMHLxhcws8YESYvxxP8W_vSePvl4lwKRXHxFWhJeA,2695
-geometry/transformation.py,sha256=pEJXS7JYxu-f05ELTUUU4w7swcQPAgAxQNI6vPp5VCg,5242
-geometry/utils.py,sha256=q7-aaxDzRDwl78-3XzdpcJuh5iL7I8lN5agk2WNWjSY,3443
-pfc_geometry-0.2.17.dist-info/METADATA,sha256=9vxRWWyYmXUu7wmjaqTtMp015ePjd4BPAOJnktIfgI4,1629
-pfc_geometry-0.2.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pfc_geometry-0.2.17.dist-info/licenses/LICENSE,sha256=z72U6pv-bQgJ_Svr4uCXnMjemsp38aSerhHEdEAOMJ4,7632
-pfc_geometry-0.2.17.dist-info/RECORD,,