Updated eval_field_gradient to handle cases where x is an array

Author: max-models

psydac/fem/tensor.py

@@ -618,6 +618,16 @@ def eval_field_gradient(self, field, *eta, weights=None):
         bases_1 = []
         index   = []
 
+        # Check if `x` is iterable and loop over elements
+        if isinstance(eta[0], (list, np.ndarray)) and np.ndim(eta[0]) > 0:
+            for dim in range(1, self.ldim):
+                assert len(eta[0]) == len(eta[dim])
+            res_list = []
+            for i in range(len(eta[0])):
+                x = [eta[j][i] for j in range(self.ldim)]
+                res_list.append(self.eval_field_gradient(field, *x, weights=weights))
+            return np.array(res_list)
+
         for (x, xlim, space) in zip( eta, self.eta_lims, self.spaces ):
 
             knots   = space.knots