add type annotations for adaptive/learner/triangulation.py

Author: basnijholt

adaptive/learner/triangulation.py

@@ -3,12 +3,14 @@
 from collections.abc import Iterable, Sized
 from itertools import chain, combinations
 from math import factorial
+from typing import Any, Iterator, List, Optional, Set, Tuple, Union
 
 import numpy as np
 import scipy.spatial
+from numpy import bool_, float64, int32, ndarray
 
 
-def fast_norm(v):
+def fast_norm(v: Union[Tuple[float64, float64, float64], ndarray]) -> float:
     # notice this method can be even more optimised
     if len(v) == 2:
         return math.sqrt(v[0] * v[0] + v[1] * v[1])
@@ -17,7 +19,15 @@ def fast_norm(v):
     return math.sqrt(np.dot(v, v))
 
 
-def fast_2d_point_in_simplex(point, simplex, eps=1e-8):
+def fast_2d_point_in_simplex(
+    point: Union[Tuple[int, int], Tuple[float, float], Tuple[float64, float64]],
+    simplex: Union[
+        List[Union[Tuple[int, int], Tuple[float64, float64]]],
+        List[Tuple[float64, float64]],
+        ndarray,
+    ],
+    eps: float = 1e-8,
+) -> Union[bool, bool_]:
     (p0x, p0y), (p1x, p1y), (p2x, p2y) = simplex
     px, py = point
 
@@ -31,7 +41,7 @@ def fast_2d_point_in_simplex(point, simplex, eps=1e-8):
     return (t >= -eps) and (s + t <= 1 + eps)
 
 
-def point_in_simplex(point, simplex, eps=1e-8):
+def point_in_simplex(point: Any, simplex: Any, eps: float = 1e-8) -> Union[bool, bool_]:
     if len(point) == 2:
         return fast_2d_point_in_simplex(point, simplex, eps)
 
@@ -42,7 +52,7 @@ def point_in_simplex(point, simplex, eps=1e-8):
     return all(alpha > -eps) and sum(alpha) < 1 + eps
 
 
-def fast_2d_circumcircle(points):
+def fast_2d_circumcircle(points: ndarray) -> Tuple[Tuple[float64, float64], float]:
     """Compute the center and radius of the circumscribed circle of a triangle
 
     Parameters
@@ -78,7 +88,9 @@ def fast_2d_circumcircle(points):
     return (x + points[0][0], y + points[0][1]), radius
 
 
-def fast_3d_circumcircle(points):
+def fast_3d_circumcircle(
+    points: ndarray,
+) -> Tuple[Tuple[float64, float64, float64], float]:
     """Compute the center and radius of the circumscribed shpere of a simplex.
 
     Parameters
@@ -118,7 +130,7 @@ def fast_3d_circumcircle(points):
     return center, radius
 
 
-def fast_det(matrix):
+def fast_det(matrix: ndarray) -> float64:
     matrix = np.asarray(matrix, dtype=float)
     if matrix.shape == (2, 2):
         return matrix[0][0] * matrix[1][1] - matrix[1][0] * matrix[0][1]
@@ -129,7 +141,13 @@ def fast_det(matrix):
         return np.linalg.det(matrix)
 
 
-def circumsphere(pts):
+def circumsphere(
+    pts: ndarray,
+) -> Union[
+    Tuple[Tuple[float64, float64, float64, float64], float],
+    Tuple[Tuple[float64, float64], float],
+    Tuple[Tuple[float64, float64, float64], float],
+]:
     dim = len(pts) - 1
     if dim == 2:
         return fast_2d_circumcircle(pts)
@@ -155,7 +173,7 @@ def circumsphere(pts):
     return tuple(center), radius
 
 
-def orientation(face, origin):
+def orientation(face: Any, origin: Any) -> Union[int, float64]:
     """Compute the orientation of the face with respect to a point, origin.
 
     Parameters
@@ -181,11 +199,18 @@ def orientation(face, origin):
     return sign
 
 
-def is_iterable_and_sized(obj):
+def is_iterable_and_sized(obj: Any) -> bool:
     return isinstance(obj, Iterable) and isinstance(obj, Sized)
 
 
-def simplex_volume_in_embedding(vertices) -> float:
+def simplex_volume_in_embedding(
+    vertices: Union[
+        List[Tuple[float64, float64, float64]],
+        List[Tuple[float, float64, float64]],
+        List[Tuple[float64, float64, float64, float64]],
+        List[Tuple[float64, float64, float64, float64, float64]],
+    ]
+) -> float:
     """Calculate the volume of a simplex in a higher dimensional embedding.
     That is: dim > len(vertices) - 1. For example if you would like to know the
     surface area of a triangle in a 3d space.
@@ -266,7 +291,7 @@ class Triangulation:
         or more simplices in the
     """
 
-    def __init__(self, coords):
+    def __init__(self, coords: Any) -> None:
         if not is_iterable_and_sized(coords):
             raise TypeError("Please provide a 2-dimensional list of points")
         coords = list(coords)
@@ -305,27 +330,27 @@ def __init__(self, coords):
         for simplex in initial_tri.simplices:
             self.add_simplex(simplex)
 
-    def delete_simplex(self, simplex):
+    def delete_simplex(self, simplex: Any) -> None:
         simplex = tuple(sorted(simplex))
         self.simplices.remove(simplex)
         for vertex in simplex:
             self.vertex_to_simplices[vertex].remove(simplex)
 
-    def add_simplex(self, simplex):
+    def add_simplex(self, simplex: Any) -> None:
         simplex = tuple(sorted(simplex))
         self.simplices.add(simplex)
         for vertex in simplex:
             self.vertex_to_simplices[vertex].add(simplex)
 
-    def get_vertices(self, indices):
+    def get_vertices(self, indices: Any) -> Any:
         return [self.get_vertex(i) for i in indices]
 
-    def get_vertex(self, index):
+    def get_vertex(self, index: Optional[Union[int32, int]]) -> Any:
         if index is None:
             return None
         return self.vertices[index]
 
-    def get_reduced_simplex(self, point, simplex, eps=1e-8) -> list:
+    def get_reduced_simplex(self, point: Any, simplex: Any, eps: float = 1e-8) -> list:
         """Check whether vertex lies within a simplex.
 
         Returns
@@ -350,11 +375,13 @@ def get_reduced_simplex(self, point, simplex, eps=1e-8) -> list:
 
         return [simplex[i] for i in result]
 
-    def point_in_simplex(self, point, simplex, eps=1e-8):
+    def point_in_simplex(
+        self, point: Any, simplex: Any, eps: float = 1e-8
+    ) -> Union[bool, bool_]:
         vertices = self.get_vertices(simplex)
         return point_in_simplex(point, vertices, eps)
 
-    def locate_point(self, point):
+    def locate_point(self, point: Any) -> Any:
         """Find to which simplex the point belongs.
 
         Return indices of the simplex containing the point.
@@ -366,10 +393,12 @@ def locate_point(self, point):
         return ()
 
     @property
-    def dim(self):
+    def dim(self) -> int:
         return len(self.vertices[0])
 
-    def faces(self, dim=None, simplices=None, vertices=None):
+    def faces(
+        self, dim: None = None, simplices: Optional[Any] = None, vertices: None = None
+    ) -> Iterator[Any]:
         """Iterator over faces of a simplex or vertex sequence."""
         if dim is None:
             dim = self.dim
@@ -394,7 +423,7 @@ def containing(self, face):
         """Simplices containing a face."""
         return set.intersection(*(self.vertex_to_simplices[i] for i in face))
 
-    def _extend_hull(self, new_vertex, eps=1e-8):
+    def _extend_hull(self, new_vertex: Any, eps: float = 1e-8) -> Any:
         # count multiplicities in order to get all hull faces
         multiplicities = Counter(face for face in self.faces())
         hull_faces = [face for face, count in multiplicities.items() if count == 1]
@@ -434,7 +463,13 @@ def _extend_hull(self, new_vertex, eps=1e-8):
 
         return new_simplices
 
-    def circumscribed_circle(self, simplex, transform):
+    def circumscribed_circle(
+        self, simplex: Any, transform: ndarray
+    ) -> Union[
+        Tuple[Tuple[float64, float64, float64, float64], float],
+        Tuple[Tuple[float64, float64], float],
+        Tuple[Tuple[float64, float64, float64], float],
+    ]:
         """Compute the center and radius of the circumscribed circle of a simplex.
 
         Parameters
@@ -450,7 +485,9 @@ def circumscribed_circle(self, simplex, transform):
         pts = np.dot(self.get_vertices(simplex), transform)
         return circumsphere(pts)
 
-    def point_in_cicumcircle(self, pt_index, simplex, transform):
+    def point_in_cicumcircle(
+        self, pt_index: int, simplex: Any, transform: ndarray
+    ) -> bool_:
         # return self.fast_point_in_circumcircle(pt_index, simplex, transform)
         eps = 1e-8
 
@@ -460,10 +497,15 @@ def point_in_cicumcircle(self, pt_index, simplex, transform):
         return np.linalg.norm(center - pt) < (radius * (1 + eps))
 
     @property
-    def default_transform(self):
+    def default_transform(self) -> ndarray:
         return np.eye(self.dim)
 
-    def bowyer_watson(self, pt_index, containing_simplex=None, transform=None):
+    def bowyer_watson(
+        self,
+        pt_index: int,
+        containing_simplex: Optional[Any] = None,
+        transform: Optional[ndarray] = None,
+    ) -> Any:
         """Modified Bowyer-Watson point adding algorithm.
 
         Create a hole in the triangulation around the new point,
@@ -523,10 +565,10 @@ def bowyer_watson(self, pt_index, containing_simplex=None, transform=None):
         new_triangles = self.vertex_to_simplices[pt_index]
         return bad_triangles - new_triangles, new_triangles - bad_triangles
 
-    def _simplex_is_almost_flat(self, simplex):
+    def _simplex_is_almost_flat(self, simplex: Any) -> bool_:
         return self._relative_volume(simplex) < 1e-8
 
-    def _relative_volume(self, simplex):
+    def _relative_volume(self, simplex: Any) -> float64:
         """Compute the volume of a simplex divided by the average (Manhattan)
         distance of its vertices. The advantage of this is that the relative
         volume is only dependent on the shape of the simplex and not on the
@@ -537,7 +579,12 @@ def _relative_volume(self, simplex):
         average_edge_length = np.mean(np.abs(vectors))
         return self.volume(simplex) / (average_edge_length ** self.dim)
 
-    def add_point(self, point, simplex=None, transform=None):
+    def add_point(
+        self,
+        point: Any,
+        simplex: Optional[Any] = None,
+        transform: Optional[ndarray] = None,
+    ) -> Any:
         """Add a new vertex and create simplices as appropriate.
 
         Parameters
@@ -586,13 +633,13 @@ def add_point(self, point, simplex=None, transform=None):
             self.vertices.append(point)
             return self.bowyer_watson(pt_index, actual_simplex, transform)
 
-    def volume(self, simplex):
+    def volume(self, simplex: Any) -> float:
         prefactor = np.math.factorial(self.dim)
         vertices = np.array(self.get_vertices(simplex))
         vectors = vertices[1:] - vertices[0]
         return float(abs(fast_det(vectors)) / prefactor)
 
-    def volumes(self):
+    def volumes(self) -> List[float]:
         return [self.volume(sim) for sim in self.simplices]
 
     def reference_invariant(self):
@@ -609,21 +656,29 @@ def vertex_invariant(self, vertex):
         """Simplices originating from a vertex don't overlap."""
         raise NotImplementedError
 
-    def get_neighbors_from_vertices(self, simplex):
+    def get_neighbors_from_vertices(self, simplex: Any) -> Any:
         return set.union(*[self.vertex_to_simplices[p] for p in simplex])
 
-    def get_face_sharing_neighbors(self, neighbors, simplex):
+    def get_face_sharing_neighbors(self, neighbors: Any, simplex: Any) -> Any:
         """Keep only the simplices sharing a whole face with simplex."""
         return {
             simpl for simpl in neighbors if len(set(simpl) & set(simplex)) == self.dim
         }  # they share a face
 
-    def get_simplices_attached_to_points(self, indices):
+    def get_simplices_attached_to_points(self, indices: Any) -> Any:
         # Get all simplices that share at least a point with the simplex
         neighbors = self.get_neighbors_from_vertices(indices)
         return self.get_face_sharing_neighbors(neighbors, indices)
 
-    def get_opposing_vertices(self, simplex):
+    def get_opposing_vertices(
+        self,
+        simplex: Union[
+            Tuple[int32, int, int],
+            Tuple[int32, int32, int],
+            Tuple[int32, int32, int32],
+            Tuple[int, int, int],
+        ],
+    ) -> Any:
         if simplex not in self.simplices:
             raise ValueError("Provided simplex is not part of the triangulation")
         neighbors = self.get_simplices_attached_to_points(simplex)
@@ -641,7 +696,7 @@ def find_opposing_vertex(vertex):
         return result
 
     @property
-    def hull(self):
+    def hull(self) -> Union[Set[int32], Set[int], Set[Union[int32, int]]]:
         """Compute hull from triangulation.
 
         Parameters