pyoculus.solvers.qfm.rfft2D

pyoculus.solvers.qfm.rfft2D(f, mpol=None, ntor=None)

Perform 2D Fourier transform from real space to cosine and sine.

Parameters:
  • f (np.array) – The data in real space. If f is 2D, then the last axis will be the axis along which FFT is computed.

  • mpol (int, optional) – The number of theta points on output. Defaults to f.shape[-2] // 4.

  • ntor (int, optional) – The number of phi points on output. Defaults to f.shape[-1] // 4.

Returns:

Cosine components \(n \in \{0, ..., m_{pol}\}\) np.array: Sine components \(n \in \{-n_{tor}, ..., n_{tor}\}\)

Return type:

np.array