tree_d_3d.kernels.quadtree_scatter.quadtree_scatter_numpy

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

Vectorized over the particles; the same arguments and results as quadtree_scatter_pyccel.py.

Functions

Name	Description
quadtree_scatter	One level of the tree build: move every particle of a split node into its child.

quadtree_scatterfunction#

View source

def quadtree_scatter(perm: np.ndarray, slot_node: np.ndarray, quadrant: np.ndarray, below: np.ndarray, node_start: np.ndarray, node_child: np.ndarray, new_perm: np.ndarray, new_slot_node: np.ndarray) -> None

One level of the tree build: move every particle of a split node into its child.