fix: adding vector/value points

Author: lachlangrose

LoopStructural/datatypes/__init__.py

@@ -1,2 +1,3 @@
 from ._surface import Surface
 from ._bounding_box import BoundingBox
+from ._point import ValuePoints, VectorPoints

LoopStructural/datatypes/_point.py

@@ -1,5 +1,47 @@
-# from dataclasses import dataclass
+from dataclasses import dataclass
+import numpy as np
 
 
-# @dataclass
-# class Value:
+@dataclass
+class ValuePoints:
+    locations: np.ndarray
+    values: np.ndarray
+    name: str
+
+    def to_dict(self):
+        return {
+            "locations": self.locations,
+            "values": self.values,
+            "name": self.name,
+        }
+
+    def vtk(self):
+        import pyvista as pv
+
+        points = pv.PolyData(self.locations)
+        points["values"] = self.values
+        return points
+
+
+@dataclass
+class VectorPoints:
+    locations: np.ndarray
+    vectors: np.ndarray
+    name: str
+
+    def to_dict(self):
+        return {
+            "locations": self.locations,
+            "vectors": self.vectors,
+            "name": self.name,
+        }
+
+    def vtk(self):
+        import pyvista as pv
+
+        points = pv.PolyData(self.locations)
+        points.point_data.set_vectors(self.vectors, 'vectors')
+        geom = pv.Arrow()
+
+        # Perform the glyph
+        return points.glyph(orient="vectors", geom=geom)