tree_d_3d.kernels.quadtree_moments.quadtree_moments_numpy

The NumPy version of the quadtree_moments kernel, for machines without Pyccel.

Vectorized over the particles of each node; the same arguments and results as quadtree_moments_pyccel.py.

Functions

Name	Description
quadtree_moments	Signed moments of every node about its centre; node_s is xx, xy, yy.

quadtree_momentsfunction#

View source

def quadtree_moments(pos: np.ndarray, charge: np.ndarray, perm: np.ndarray, node_count: int, node_center: np.ndarray, node_start: np.ndarray, node_end: np.ndarray, node_q: np.ndarray, node_qabs: np.ndarray, node_p: np.ndarray, node_s: np.ndarray) -> None

Signed moments of every node about its centre; node_s is xx, xy, yy.