Обобщение 3D выборки дисков

Я хочу создать выборку диска Пуассона для трехмерной сетки. Я использовал реализацию https://github.com/emulbreh/bridson, чтобы обобщить ее на 3D. Однако кажется, что я что-то упускаю, и я не могу найти проблему. Эта реализация следует алгоритму Роберта Бридсона (https://www.cs.ubc.ca/~rbridson/docs/bridson-siggraph07-poissondisk.pdf)

Что я сделал, так это превратил каждый 2D-объект в 3D и создал новую точку со сферическими координатами.

from random import random
from math import cos, sin, floor, sqrt, pi, ceil


def euclidean_distance(a, b):
    dx = a[0] - b[0]
    dy = a[1] - b[1]
    dz = a[2] - b[2]

    return sqrt(dx * dx + dy * dy + dz * dz)


def poisson_disc_samples(width, height,thick, r, k=5, distance=euclidean_distance, random=random):
    tau = 2 * pi
    cellsize = r / sqrt(2)

    grid_width = int(ceil(width / cellsize))
    grid_height = int(ceil(height / cellsize))
    grid_thick = int(ceil(thick / cellsize))

    grid = [None] * (grid_width * grid_height * grid_thick)
    def grid_coords(p):
        return int(floor(p[0] / cellsize)), int(floor(p[1] / cellsize)), int(floor(p[2] / cellsize))

    def fits(p, gx, gy, gz):
        yrange = list(range(max(gy - 2, 0), min(gy + 3, grid_height)))
        zrange = list(range(max(gz - 2, 0), min(gz + 3, grid_thick)))
        for x in range(max(gx - 2, 0), min(gx + 3, grid_width)):
            for y in yrange:
                for z in zrange:
                    g = grid[x + y + z  * grid_width]
                    if g is None:
                        continue
                    if distance(p, g) <= r:
                        return False
        return True
    p = width * random(), height * random(), thick * random()
    queue = [p]
    grid_x, grid_y, grid_z = grid_coords(p)
    grid[grid_x + grid_y + grid_z * grid_width] = p

    while queue:
        qi = int(random() * len(queue))
        qx, qy, qz = queue[qi]
        queue[qi] = queue[-1]
        queue.pop()
        for _ in range(k):
            alpha = tau * random()
            theta = tau * random()
            d = r * sqrt(3 * random() + 1)
            px = qx + d * cos(alpha) * sin(theta)
            py = qy + d * sin(alpha) * sin(theta)
            pz = qz + d * cos(theta)
            if not (0 <= px < width and 0 <= py < height):
                continue
            p = (px, py, pz)
            grid_x, grid_y, grid_z = grid_coords(p)
            if not fits(p, grid_x, grid_y,grid_z):
                continue
            queue.append(p)
            grid[grid_x + grid_y + grid_z * grid_width] = p
    return [p for p in grid if p is not None]

r = 10
samples = poisson_disc_samples(100, 100, 100, r=10, random=random)

print(samples)

Результаты, которые я ищу, - это просто печать выбранных 3D-точек, однако программа не печатает никаких ошибок, и я чувствую это, потому что она не находит образцы, и после этого компьютер дает сбой.

Я думаю, что это может быть функция fits (она отвечает за проверку того, пусты ли сетки или нет), которая вызывает проблему.