TARANG.libp.global_fns.spectral_setup

Attributes

comm
rank
nprocs
kx
ky
kz
kfactor
Ngrid
Ngrid
Ngrid
dealiasing_index
k_max
min_radius_outside
k_min
k_array

Functions

getSplit(lendim, P)	Function to return the split indices, and block sizes when a dimension of length lendim is
redistribute(APre, APost, axisPre, axisPost, comm)	Function to redistribute numpy array (almost) equally
xderiv(Ak)
yderiv(Ak)
zderiv(Ak)
forward_transform(A, Ak)
inverse_transform(Ak, A, temp)
dealias(Ak)
reality_cond(Ak, temp_xy)
generate_spectrum2D(Ak, ek, Tk, nlin, temp)
generate_spectrum3D(Ak, ek, Tk, nlin, temp)
boundary_sin_cond(Ak)
boundary_cos_cond(Ak)
craya_to_cartesian(u1, Vkx, Vkz, k_mag, i, j)
compute_phi(i, j)

Module Contents

TARANG.libp.global_fns.spectral_setup.comm

TARANG.libp.global_fns.spectral_setup.rank

TARANG.libp.global_fns.spectral_setup.nprocs

TARANG.libp.global_fns.spectral_setup.getSplit(lendim, P)

Function to return the split indices, and block sizes when a dimension of length lendim is divided into P blocks. (P need not divide lendim) – :param lendim: Length of the dimension to be split (integer) :param P: Number of parts for the dimension to be broken into

Returns:

a list of length P containing (start index, block size, end index) of all blocks

Return type:

split

Examples

• getSplit(4, 4) returns [(0, 1, 1), (1, 1, 2), (2, 1, 3), (3, 1, 4)]

• getSplit(6, 4) returns [(0, 2, 2), (2, 2, 4), (4, 1, 5), (5, 1, 6)]

Note: Not tested when P > lendim

TARANG.libp.global_fns.spectral_setup.redistribute(APre, APost, axisPre, axisPost, comm)

Function to redistribute numpy array (almost) equally among all nodes. Array is initally distributed along axisPre and this function returns the array redistributed along axisPost. – :param APre: The array to redistribute (initially distributed along axisPre) :param axisPre: The axis along which array is distributed (default 0) :param axisPost: The axis along which array is to be redistributed (default 1). :param comm: MPI Communicator object (default MPI.COMM_WORLD)

Returns:

Redistributed array (distributed along axisPost)

Return type:

APost

Examples

• if axisPre is 0, axisPost = 1, comm.Get_size() is 3,

  and initially arrays have shapes [2, 8, 6], [2, 8, 6], [1, 8, 6], then redistributed arrays will have shapes [5, 3, 6], [5, 3, 6], [5, 2, 6]

Note: Not tested when comm.Get_size() > Apre.shape[axisPost]

TARANG.libp.global_fns.spectral_setup.kx

TARANG.libp.global_fns.spectral_setup.ky

TARANG.libp.global_fns.spectral_setup.kz

TARANG.libp.global_fns.spectral_setup.kfactor

TARANG.libp.global_fns.spectral_setup.Ngrid

TARANG.libp.global_fns.spectral_setup.Ngrid

TARANG.libp.global_fns.spectral_setup.Ngrid

TARANG.libp.global_fns.spectral_setup.dealiasing_index

TARANG.libp.global_fns.spectral_setup.k_max

TARANG.libp.global_fns.spectral_setup.min_radius_outside

TARANG.libp.global_fns.spectral_setup.k_min

TARANG.libp.global_fns.spectral_setup.k_array

TARANG.libp.global_fns.spectral_setup.xderiv(Ak)

TARANG.libp.global_fns.spectral_setup.yderiv(Ak)

TARANG.libp.global_fns.spectral_setup.zderiv(Ak)

TARANG.libp.global_fns.spectral_setup.forward_transform(A, Ak)

TARANG.libp.global_fns.spectral_setup.inverse_transform(Ak, A, temp)

TARANG.libp.global_fns.spectral_setup.dealias(Ak)

TARANG.libp.global_fns.spectral_setup.reality_cond(Ak, temp_xy)

TARANG.libp.global_fns.spectral_setup.generate_spectrum2D(Ak, ek, Tk, nlin, temp)

TARANG.libp.global_fns.spectral_setup.generate_spectrum3D(Ak, ek, Tk, nlin, temp)

TARANG.libp.global_fns.spectral_setup.boundary_sin_cond(Ak)

TARANG.libp.global_fns.spectral_setup.boundary_cos_cond(Ak)

TARANG.libp.global_fns.spectral_setup.craya_to_cartesian(u1, Vkx, Vkz, k_mag, i, j)

TARANG.libp.global_fns.spectral_setup.compute_phi(i, j)