diff options
Diffstat (limited to 'geometry.py')
| -rw-r--r-- | geometry.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/geometry.py b/geometry.py index 48d19fc..cf92d53 100644 --- a/geometry.py +++ b/geometry.py @@ -33,7 +33,7 @@ def interleave(arr): raise Exception('shape mismatch') z = np.zeros(arr.shape[0], dtype=np.uint64) - for i in range(arr[0][0].nbytes*8): + for i in range(arr.dtype.itemsize*8): z |= (arr[:,2] & 1 << i) << (2*i) | (arr[:,1] & 1 << i) << (2*i+1) | (arr[:,0] & 1 << i) << (2*i+2) return z @@ -48,7 +48,7 @@ def morton_order(mesh, bits=3): mean_positions = quantize(np.mean(mesh, axis=1)) - return interleave_arr(mean_positions) + return interleave(mean_positions) class Solid(object): def __init__(self, mesh, inside, outside): |