python多叉树_python中高效的四叉树实现

对于我正在做的一个项目，我试图写一些代码来检测二维空间中非点粒子之间的碰撞。我的目标是尝试检测几千个粒子的碰撞，每个时间步至少检测几次，我知道这对python来说是一个很高的要求。我遵循了这个实现四叉树的blog post来显著减少我需要进行的成对检查的数量。所以我认为我遇到的问题是这个功能：def get_index(self, particle):

index = -1

bounds = particle.aabb

v_midpoint = self.bounds.x + self.bounds.width/2

h_midpoint = self.bounds.y + self.bounds.height/2

top_quad = bounds.y < h_midpoint and bounds.y + bounds.height < h_midpoint

bot_quad = bounds.y > h_midpoint

if bounds.x < v_midpoint and bounds.x + bounds.width < v_midpoint:

if top_quad:

index = 1

elif bot_quad:

index = 2

elif bounds.x > v_midpoint:

if top_quad:

index = 0

elif bot_quad:

index = 3

return index

这个函数在我最初的分析中是一个瓶颈，我需要它快速膨胀，因为它的高调用计数。最初，我只是提供一个对象轴对齐的边界框，它几乎以我需要的速度工作，然后意识到我无法确定哪些粒子可能实际上正在碰撞。所以现在我将一个粒子列表传递给我的四叉树构造函数，并使用类属性aabb来获取边界。在

有没有什么方法可以把类似的东西传递给一个对象指针而不是整个对象？另外，还有其他的建议来优化上面的代码吗？在