exported spatial hashing

Author: imMadsen

Chipmunk2D

@@ -554,6 +554,11 @@ def use_spatial_hash(self, dim: float, count: int) -> None:
         """
         cp.cpSpaceUseSpatialHash(self._space, dim, count)
 
+    def get_space_hash_objects_per_cell(self) -> List:
+        cell_counts = ffi.new(f'int[{cp.cpSpaceGetSpatialHashTotalCells(self._space)}]')
+        cp.cpSpaceGetSpatialHashObjectCounts(self._space, cell_counts)
+        return cell_counts
+
     def step(self, dt: float) -> None:
         """Update the space for the given time step.
 

pymunk/space.py

@@ -1233,6 +1233,12 @@ void cpSpaceReindexShapesForBody(cpSpace *space, cpBody *body);
 /// Switch the space to use a spatial has as it's spatial index.
 void cpSpaceUseSpatialHash(cpSpace *space, cpFloat dim, int count);
 
+// MARK: P7 (Crowd Crushing)
+
+int cpSpaceGetSpatialHashTotalCells(cpSpace * space);
+
+void cpSpaceGetSpatialHashObjectCounts(cpSpace * space, int *cell_counts);
+
 // MARK: Time Stepping
 
 /// Step the space forward in time by @c dt.