Fixing initial cell radius.

gouarin · Dec 1, 2017 · b24e51b · b24e51b
1 parent dd32e62
commit b24e51b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nbody/numba/nbody/barnes_hut_array/quadTree.py b/nbody/numba/nbody/barnes_hut_array/quadTree.py
@@ -14,7 +14,7 @@ def __init__(self, bmin, bmax, size):
         self.cell_center = np.zeros((2*size+1, 2))
         self.cell_radius = np.zeros(2*size+1)
         self.cell_center[0] = self.center
-        self.cell_radius[0] = self.box_size
+        self.cell_radius[0] = 0.5*self.box_size
 
     def buildTree(self, particles):
         self.ncell = numba_functions.buildTree(self.center, self.box_size, self.child, self.cell_center, self.cell_radius, particles)

diff --git a/nbody/python/nbody/barnes_hut_array/quadTree.py b/nbody/python/nbody/barnes_hut_array/quadTree.py
@@ -13,7 +13,7 @@ def __init__(self, bmin, bmax, size):
         self.cell_center = np.zeros((2*size+1, 2))
         self.cell_radius = np.zeros(2*size+1)
         self.cell_center[0] = self.center
-        self.cell_radius[0] = self.box_size
+        self.cell_radius[0] = 0.5*self.box_size
 
     def buildTree(self, particles):
         for ip, p in enumerate(particles):