pfc-geometry 0.2.10__py3-none-any.whl → 0.2.11__py3-none-any.whl

geometry/base.py

@@ -353,3 +353,9 @@ class Base:
                 self.data,
             )).ffill().bfill().to_numpy()
         )
+
+    def ffill(self):
+        return self.__class__(pd.DataFrame(self.data).ffill().to_numpy())
+
+    def bfill(self):
+        return self.__class__(pd.DataFrame(self.data).bfill().to_numpy())

geometry/coordinate_frame.py

@@ -9,7 +9,7 @@ FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
 You should have received a copy of the GNU General Public License along with
 this program. If not, see <http://www.gnu.org/licenses/>.
 """
-
+from __future__ import annotations
 from geometry import Point, Quaternion, PX, PY, PZ, P0
 from typing import List
 import numpy as np
@@ -33,7 +33,7 @@ class Coord(Base):
         self.z_axis=Point(self.data[:,9:12])
     
     @staticmethod
-    def from_axes(o:Point, x:Point, y:Point, z:Point):
+    def from_axes(o:Point, x:Point, y:Point, z:Point) -> Coord:
         assert len(o) == len(x) == len(y) == len(z)
         return Coord(np.concatenate([
             o.data,
@@ -43,25 +43,25 @@ class Coord(Base):
         ],axis=1))
 
     @staticmethod
-    def zero(count=1):
+    def zero(count=1) -> Coord:
         return Coord.from_nothing(count)
 
     @staticmethod
-    def from_nothing(count=1):
+    def from_nothing(count=1) -> Coord:
         return Coord.from_axes(P0(count), PX(1,count), PY(1,count), PZ(1,count))
 
     @staticmethod
-    def from_xy(origin: Point, x_axis: Point, y_axis: Point):
+    def from_xy(origin: Point, x_axis: Point, y_axis: Point) -> Coord:
         z_axis = x_axis.cross(y_axis)
         return Coord.from_axes(origin, x_axis, z_axis.cross(x_axis), z_axis)
 
     @staticmethod
-    def from_yz(origin: Point, y_axis: Point, z_axis: Point):
+    def from_yz(origin: Point, y_axis: Point, z_axis: Point) -> Coord:
         x_axis = y_axis.cross(z_axis)
         return Coord.from_axes(origin, x_axis, y_axis, x_axis.cross(y_axis))
 
     @staticmethod
-    def from_zx(origin: Point, z_axis: Point, x_axis: Point):
+    def from_zx(origin: Point, z_axis: Point, x_axis: Point) -> Coord:
         y_axis = z_axis.cross(x_axis)
         return Coord.from_axes(origin, y_axis.cross(z_axis), y_axis, z_axis)
 
@@ -71,7 +71,7 @@ class Coord(Base):
     def inverse_rotation_matrix(self):
         return Quaternion.from_rotation_matrix(self.rotation_matrix()).inverse().to_rotation_matrix()
 
-    def rotate(self, rotation=Quaternion):
+    def rotate(self, rotation=Quaternion) -> Coord:
         return Coord.from_axes(
             self.origin,
             rotation.transform_point(self.x_axis),
@@ -82,7 +82,7 @@ class Coord(Base):
     def __eq__(self, other):
         return self.data == other.data
 
-    def translate(self, point):
+    def translate(self, point) -> Coord:
         return Coord.from_axes(self.origin + point, self.x_axis, self.y_axis, self.z_axis)
 
     def axes(self):

geometry/point.py

@@ -14,6 +14,8 @@ from .base import Base
 import numpy as np
 import pandas as pd
 from warnings import warn
+from numbers import Number
+import numpy.typing as npt
 
 
 class Point(Base):
@@ -62,16 +64,16 @@ class Point(Base):
         return abs(Point.angles(self, p2))
     
     @staticmethod
-    def X(value=1, count=1):
-        return Point(np.tile([value,0,0], (count, 1)))
+    def X(value: Number | npt.NDArray=1, count=1):
+        return np.tile(value, count) * Point(1,0,0)
 
     @staticmethod
     def Y(value=1, count=1):
-        return Point(np.tile([0,value,0], (count, 1)))
+        return np.tile(value, count) * Point(0,1,0)
 
     @staticmethod
     def Z(value=1, count=1):
-        return Point(np.tile([0,0,value], (count, 1)))
+        return np.tile(value, count) * Point(0,0,1)
 
     def rotate(self, rmat=np.ndarray):
         if len(rmat.shape) == 3:
@@ -108,6 +110,9 @@ class Point(Base):
     def zeros(count=1):
         return Point(np.zeros((count,3)))
 
+    def bearing(self):
+        return np.arctan2(self.y, self.x)
+
 def Points(*args, **kwargs):
     warn("Points is deprecated, you can now just use Point", DeprecationWarning)
     return Point(*args, **kwargs)
@@ -136,7 +141,7 @@ def ppmeth(func):
 
 
 @ppmeth
-def cross(a, b) -> Point:
+def cross(a: Point, b: Point) -> Point:
     return Point(np.cross(a.data, b.data))
  
 

geometry/quaternion.py

@@ -253,7 +253,10 @@ class Quaternion(Base):
         # does the rotation reverse the Z axis?
         return np.sign(self.transform_point(PZ()).z) > 0
 
-    
+    def bearing(self, p: Point=None):
+        if p is None:
+            p = Point.X()
+        return self.transform_point(p).bearing()
     
         
 

{pfc_geometry-0.2.10.dist-info → pfc_geometry-0.2.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pfc-geometry
-Version: 0.2.10
+Version: 0.2.11
 Summary: A package for handling 3D geometry with a nice interface
 Author-email: Thomas David <thomasdavid0@gmail.com>
 License: GNU GPL v3

pfc_geometry-0.2.11.dist-info/RECORD

@@ -0,0 +1,15 @@
+geometry/__init__.py,sha256=HNhMyemIJzDq1nDjrr09eX5PS7q9ULscSbYsXss3JRM,1253
+geometry/base.py,sha256=yNiO7Fe-S5c4wtvgNmQ2jrX7Yt4tB3eBekYHK0_psYc,11926
+geometry/coordinate_frame.py,sha256=YTCAtCUuIq5LAsO-P9FwFs55f4qpWP9pUP6mf-Nhk54,3145
+geometry/gps.py,sha256=Fs3hakSQ754HUqRsA7NWg_MSEdYxNqyiu4gu6EDrFqI,3381
+geometry/mass.py,sha256=BUWBSITwpdRfpJR5-oJTd16BI7FLZt8rhxdzr0cx1HY,1675
+geometry/point.py,sha256=XF7sdCaCD-si9UAumpyvxzMOTX3utVogJ2tt_7vt7b0,5185
+geometry/quaternion.py,sha256=nggmRu_8xCQNYlf8GcilC4zJZLsac3dwUQKu5pxL_38,9522
+geometry/testing.py,sha256=o8yMBAdU5Vy0EspBYaof4fPGgRSFZhRDhzBjRPsLd0M,375
+geometry/time.py,sha256=ZWO6yThXesHkNsROJfUruF-G4Lpcx149x0H__0HPLvw,1206
+geometry/transformation.py,sha256=eunEC924zPBLhaVSEWE-IClrRfitvUKjSdOTaa2tdDQ,4705
+pfc_geometry-0.2.11.dist-info/LICENSE,sha256=z72U6pv-bQgJ_Svr4uCXnMjemsp38aSerhHEdEAOMJ4,7632
+pfc_geometry-0.2.11.dist-info/METADATA,sha256=lEO6DesuEkg7KVSx98UNygwFmmH4eyYTTA6q80cTcK0,1979
+pfc_geometry-0.2.11.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+pfc_geometry-0.2.11.dist-info/top_level.txt,sha256=RWbWhWYclEM-OFtXLCB4eLv-jxNnlJB-NPC2YM587Fo,9
+pfc_geometry-0.2.11.dist-info/RECORD,,

{pfc_geometry-0.2.10.dist-info → pfc_geometry-0.2.11.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (73.0.1)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

pfc_geometry-0.2.10.dist-info/RECORD

@@ -1,15 +0,0 @@
-geometry/__init__.py,sha256=HNhMyemIJzDq1nDjrr09eX5PS7q9ULscSbYsXss3JRM,1253
-geometry/base.py,sha256=rJSDViQlW6zHwP6n2VT7uulfargKIDEmUCV5sPhghOM,11735
-geometry/coordinate_frame.py,sha256=qO3jqUWF4PW4BOkw5rBMTXfOzBjlA1HYwGYmPtu2M8M,3039
-geometry/gps.py,sha256=Fs3hakSQ754HUqRsA7NWg_MSEdYxNqyiu4gu6EDrFqI,3381
-geometry/mass.py,sha256=BUWBSITwpdRfpJR5-oJTd16BI7FLZt8rhxdzr0cx1HY,1675
-geometry/point.py,sha256=7lWj1NyQsKRJZnNeyDrssAtfZLfWl4vTK9EDOscLjxw,5038
-geometry/quaternion.py,sha256=PH7YKAVnQ8vZ8RZdtnrIaqx3FmHHwQ2aTEEdofrng_U,9392
-geometry/testing.py,sha256=o8yMBAdU5Vy0EspBYaof4fPGgRSFZhRDhzBjRPsLd0M,375
-geometry/time.py,sha256=ZWO6yThXesHkNsROJfUruF-G4Lpcx149x0H__0HPLvw,1206
-geometry/transformation.py,sha256=eunEC924zPBLhaVSEWE-IClrRfitvUKjSdOTaa2tdDQ,4705
-pfc_geometry-0.2.10.dist-info/LICENSE,sha256=z72U6pv-bQgJ_Svr4uCXnMjemsp38aSerhHEdEAOMJ4,7632
-pfc_geometry-0.2.10.dist-info/METADATA,sha256=aIrRdNxRV07oKLdlKXzsjcano6Y_q36vR2ONr-XRrew,1979
-pfc_geometry-0.2.10.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
-pfc_geometry-0.2.10.dist-info/top_level.txt,sha256=RWbWhWYclEM-OFtXLCB4eLv-jxNnlJB-NPC2YM587Fo,9
-pfc_geometry-0.2.10.dist-info/RECORD,,