在3D体素空间中绘制通过3D线的所有体素

import numpy as np

def draw_line_in_voxel_space(start, end, voxel_size):
    x0, y0, z0 = start
    x1, y1, z1 = end
    
    dx = abs(x1 - x0)
    dy = abs(y1 - y0)
    dz = abs(z1 - z0)
    
    sx = 1 if x0 < x1 else -1
    sy = 1 if y0 < y1 else -1
    sz = 1 if z0 < z1 else -1
    
    err_x = dx / 2.0
    err_y = dy / 2.0
    err_z = dz / 2.0
    
    x, y, z = x0, y0, z0
    
    voxel_grid = np.zeros((100, 100, 100), dtype=bool)  # 假设体素空间大小为100x100x100
    
    while True:
        voxel_x = int(x / voxel_size)
        voxel_y = int(y / voxel_size)
        voxel_z = int(z / voxel_size)
        
        if 0 <= voxel_x < voxel_grid.shape[0] and 0 <= voxel_y < voxel_grid.shape[1] and 0 <= voxel_z < voxel_grid.shape[2]:
            voxel_grid[voxel_x, voxel_y, voxel_z] = True
        
        if x == x1 and y == y1 and z == z1:
            break
        
        err_x -= dy
        err_y -= dz
        err_z -= dx
        
        if err_x < 0:
            x += sx
            err_x += dx
        
        if err_y < 0:
            y += sy
            err_y += dy
        
        if err_z < 0:
            z += sz
            err_z += dz
    
    return voxel_grid

# 示例使用
start_point = (10, 20, 30)
end_point = (40, 50, 60)
voxel_size = 1.0

voxel_grid = draw_line_in_voxel_space(start_point, end_point, voxel_size)
print(voxel_grid)