edges in numerical data

Author: tomvanmele

CHANGELOG.md

@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+* Added `compas_fd.types.IntNx2`.
+
 ### Changed
 
 ### Removed

src/compas_fd/solvers/fd_numerical_data.py

@@ -14,6 +14,7 @@
 from compas_fd.types import FloatNx1
 from compas_fd.types import FloatNx3
 from compas_fd.types import FloatNxM
+from compas_fd.types import IntNx2
 from compas_fd.types import IntNxM
 
 from .result import Result
@@ -26,6 +27,7 @@ class FDNumericalData:
     free: int
     fixed: int
     xyz: FloatNx3
+    edges: IntNx2
     C: IntNxM
     q: FloatNx1
     Q: FloatNxM
@@ -77,7 +79,7 @@ def from_params(
         A = C.T.dot(Q).dot(C)
         Ai = Ci.T.dot(Q).dot(Ci)
         Af = Ci.T.dot(Q).dot(Cf)
-        return cls(free, fixed, xyz, C, q, Q, p, A, Ai, Af)
+        return cls(free, fixed, xyz, edges, C, q, Q, p, A, Ai, Af)
 
     @classmethod
     def from_mesh(cls, mesh: Mesh) -> "FDNumericalData":

src/compas_fd/types.py

@@ -18,6 +18,12 @@
 ]
 """An array-like object, with each item in the array containing one (1) float."""
 
+IntNx2 = Union[
+    Sequence[Annotated[Sequence[int], 2]],
+    Annotated[npt.NDArray[np.int32], Literal["*, 2"]],
+]
+"""An array-like object, with each item in the array containing two (2) integers."""
+
 IntNxM = Annotated[npt.NDArray[np.int32], Literal["*, *"]]
 
 FloatNxM = Annotated[npt.NDArray[np.float64], Literal["*, *"]]