vector field: more convinient api method.

nortikin · Dec 24, 2022 · fe9ea68 · fe9ea68
1 parent 0315d04
commit fe9ea68
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/utils/field/vector.py b/utils/field/vector.py
@@ -38,6 +38,12 @@ def evaluate(self, point):
     def evaluate_grid(self, xs, ys, zs):
         raise Exception("not implemented")
 
+    def evaluate_array(self, points):
+        xs = points[:,0]
+        ys = points[:,1]
+        zs = points[:,2]
+        return self.evaluate_grid(xs, ys, zs)
+
 class SvMatrixVectorField(SvVectorField):
 
     def __init__(self, matrix):