spectre.fourier
===============

.. py:module:: spectre.fourier






Module Contents
---------------

.. py:function:: check_equidistant(arr)

.. py:function:: shift_fftfreq(arr)

.. py:function:: spec_fft(tarr: numpy.array, zarr: numpy.array, freal: numpy.array, Mpol: int = None, Ntor: int = None)

   Fourier transform as in SPECTRE fortran source

   Args:
    - tarr: 1D numpy array of size nt. Poloidal angle coordinate, each point
            should be equidistant
    - zarr: 1D numpy array of size nz. Toroidal angle coordinate, each point
            should be equidistant
    - freal: 2D numpy array of size (nz,nt). Function, in real space, for which
            Fourier harmonics should be evaluated
    - Mpol: integer. Poloidal resolution. Default is nt/2.
    - Ntor: integer. Toroidal resolution. Default is nz/2.


.. py:data:: has_rocket_fft

.. py:data:: run_decorator_fft

.. py:function:: spec_fft_fast(tarr: numpy.array, zarr: numpy.array, freal_in: numpy.array, Mpol: int = None, Ntor: int = None)

   Fourier transform as in SPECTRE fortran source

   Args:
    - tarr: 1D numpy array of size nt. Poloidal angle coordinate, each point
                    should be equidistant
    - zarr: 1D numpy array of size nz. Toroidal angle coordinate, each point
                    should be equidistant
    - freal: 2D numpy array of size (nz,nt). Function, in real space, for which
                    Fourier harmonics should be evaluated
    - Mpol: integer. Poloidal resolution. Default is nt/2.
    - Ntor: integer. Toroidal resolution. Default is nz/2.