Skip to content
Snippets Groups Projects

Enable flat (marker) evaluation

Merged Stefan Possanner requested to merge enable-flat-profile-evaluation into master
9 files
+ 824
783
Compare changes
  • Side-by-side
  • Inline
Files
9
@@ -61,7 +61,7 @@ class SplineFourierBases:
'''Index range of sine modes in `mn` list.'''
return self._range_sin
def eval_stz(self, s, th, ze, coef, der=None):
def eval_stz(self, s, th, ze, coef, der=None, flat_eval=False):
"""Evaluate a B-spline x Fourier basis, or its first or second derivatives, at `s`, `theta`, `zeta`.
Parameters
@@ -81,13 +81,16 @@ class SplineFourierBases:
der : str
Which derivative to evaluate: None, "s", "th", "ze", "ss", "thth", "zeze", "sth", "sze" or "thze".
flat_eval : bool
Whether to do flat (marker) evaluation.
Returns
-------
float or meshgrid numpy.ndarray
Evaluated coordinate.
"""
s, th, ze = GVEC_domain.prepare_args(s, th, ze)
s, th, ze = GVEC_domain.prepare_args(s, th, ze, flat_eval=flat_eval)
if isinstance(s, np.ndarray):
vals = np.zeros((s + th + ze).shape)
else:
Loading