Added polar conversions to Coordinate

Author: MrClock8163

src/geocompy/data.py

@@ -913,4 +913,51 @@ class Coordinate(Vector):
 
     """
 
-    # SIM = 0x04000 # TPSSim
+    @classmethod
+    def from_polar(
+        cls,
+        hz: Angle,
+        v: Angle,
+        dist: float
+    ) -> Self:
+        """
+        Constructs 3D cartesian coordinate from polar coordinate.
+
+        Parameters
+        ----------
+        hz : Angle
+            Horizontal angle.
+        v : Angle
+            Vertical angle.
+        dist : float
+            Slope distance.
+
+        Returns
+        -------
+        Coordinate
+        """
+        dist2d = math.sin(v) * dist
+        x = math.sin(hz) * dist2d
+        y = math.cos(hz) * dist2d
+        z = math.cos(v) * dist
+
+        return cls(x, y, z)
+
+    def to_polar(self) -> tuple[Angle, Angle, float]:
+        """
+        Converts 3D cartesian coordinates to polar coordinates.
+
+        Returns
+        -------
+        tuple
+            Horizontal and vertical angles and slope distance.
+        """
+        dist2d = math.sqrt(self.x**2 + self.y**2)
+        dist = math.sqrt(dist2d**2 + self.z**2)
+        hz = math.atan2(self.x, self.y)
+        v = math.atan2(dist2d, self.z)
+        return (
+            Angle(hz, normalize=True, positive=True),
+            Angle(v, normalize=True, positive=True),
+            dist
+        )

tests/test_data.py

@@ -1,3 +1,4 @@
+import math
 from enum import Enum
 
 import pytest
@@ -156,3 +157,17 @@ def test_arithmetic(self):
         assert type(+c1) is Coordinate
         c3 = +c1
         assert c3 is not c1
+
+    def test_polar(self):
+        c1 = Coordinate(-1, -1, -1)
+        p1 = c1.to_polar()
+
+        assert float(p1[0]) == approx(math.radians(225))
+        assert float(p1[1]) == approx(math.radians(125.2643897))
+        assert p1[2] == approx(math.sqrt(3))
+
+        c2 = Coordinate.from_polar(*p1)
+
+        assert c1.x == approx(c2.x)
+        assert c1.y == approx(c2.y)
+        assert c1.z == approx(c2.z)