def _params_mixture(n):
return (
ParamVectorDesc('r', n), # polar coord (radius)
ParamVectorDesc('t', n), # polar coord (angle)
ParamVectorDesc('a', n), # amplitude
ParamVectorDesc('s', n), # size
ParamVectorDesc('b', n), # shape
ParamVectorDesc('q', n), # axis ratio (b/a)
ParamVectorDesc('p', n)) # position angle relative to t
def params_rnw(self, nnodes):
return (ParamVectorDesc('a', nnodes), ParamVectorDesc('p', nnodes),
ParamVectorDesc('s', nnodes))
def _params_pw_harmonic(order, nnodes):
params = []
params += [ParamVectorDesc('a', nnodes)]
params += [ParamVectorDesc('p', nnodes)] * (order > 0)
return tuple(params)
def params_rnw(self, nnodes):
return (ParamVectorDesc('vl', nnodes), )
def params_rnw(self, nrnodes):
if self._kwargs['rnodes']:
return tuple([ParamVectorDesc(p, nrnodes) for p in self._params])
else:
return tuple()
def _make_param_descs(key, nnodes, nw):
return {key: ParamVectorDesc(key, nnodes) if nw else ParamScalarDesc(key)}