quTARANG.univ package

Submodules

quTARANG.univ.fns module

quTARANG.univ.fns.gradient(params, arr, G)

Calculates the gradient using spectral method along different axes depending on dimensionality.

The result is stored in the Vx, Vy, Vz arrays inside U. U is an attrubite of the GPE object.

Parameters

arr : Array G : GPE object

quTARANG.univ.fns.integralk(params, arrk, grid: Grid)

Integrates an array in real space (Trapezoid method).

Parameters

arr : Array params : Grid

Returns

complex

quTARANG.univ.fns.integralr(params, arr, grid: Grid)

Integrates an array in real space (Trapezoid method).

Parameters

arr : Array grid : Grid

Returns

complex

quTARANG.univ.grid module

class quTARANG.univ.grid.Grid(params: Params)

Bases: object

setupX(params: Params)

Setup grid along x.

setupY(params: Params)

Setup grid along y.

setupZ(params: Params)

Setup grid along x.

setup_grid(params: Params)

setup_meshgrid(params)

Generates the mesh for the simulation.

quTARANG.univ.my_fft module

quTARANG.univ.my_fft.forward_transform(params, arr)

Forward Fourier Transform

Parameters

arr : Array

Returns

Forward Fourier transform of the input array

quTARANG.univ.my_fft.inverse_transform(params, arr)

Inverse Fourier Transform

Parameters

arr : Array

Returns

Inverse Fourier tranform of the input array

quTARANG.univ.params module

class quTARANG.univ.params.Params(N: list = [128, 1, 1], L: float = [32, 1, 1], g: float = 0, dt: float = 0.001, tmax: float = 1, scheme: str = 'TSSP', imgtime: bool = False, rms: list = [False, 0, 100], ektk: list = [False, 0, 100], energy: list = [False, 0, 100], wfc: list = [False, 0, 100], gpu: bool = False, gpu_rank: int = 0, path: str = '/home/docs/checkouts/readthedocs.org/user_builds/qutarang/checkouts/latest/docs/source/output', **kwargs)

Bases: object

Class to store all necessary parameters of the simulation.

Module contents