Spaces:
Runtime error
Runtime error
File size: 1,940 Bytes
2cdd41c 1615d09 2cdd41c 1615d09 2cdd41c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import numpy as np
cimport cython
cimport numpy as np
from libc.stdlib cimport free, malloc
ctypedef struct qnode:
int row
int col
int layer
int orig_row
int orig_col
@cython.infer_types(True)
@cython.boundscheck(False)
@cython.wraparound(False)
@cython.nonecheck(False)
def get_dist_maps(np.ndarray[np.float32_t, ndim=2, mode="c"] points,
int height, int width, float norm_delimeter):
cdef np.ndarray[np.float32_t, ndim=3, mode="c"] dist_maps = \
np.full((2, height, width), 1e6, dtype=np.float32, order="C")
cdef int *dxy = [-1, 0, 0, -1, 0, 1, 1, 0]
cdef int i, j, x, y, dx, dy
cdef qnode v
cdef qnode *q = <qnode *> malloc((4 * height * width + 1) * sizeof(qnode))
cdef int qhead = 0, qtail = -1
cdef float ndist
for i in range(points.shape[0]):
x, y = round(points[i, 0]), round(points[i, 1])
if x >= 0:
qtail += 1
q[qtail].row = x
q[qtail].col = y
q[qtail].orig_row = x
q[qtail].orig_col = y
if i >= points.shape[0] / 2:
q[qtail].layer = 1
else:
q[qtail].layer = 0
dist_maps[q[qtail].layer, x, y] = 0
while qtail - qhead + 1 > 0:
v = q[qhead]
qhead += 1
for k in range(4):
x = v.row + dxy[2 * k]
y = v.col + dxy[2 * k + 1]
ndist = ((x - v.orig_row)/norm_delimeter) ** 2 + ((y - v.orig_col)/norm_delimeter) ** 2
if (x >= 0 and y >= 0 and x < height and y < width and
dist_maps[v.layer, x, y] > ndist):
qtail += 1
q[qtail].orig_col = v.orig_col
q[qtail].orig_row = v.orig_row
q[qtail].layer = v.layer
q[qtail].row = x
q[qtail].col = y
dist_maps[v.layer, x, y] = ndist
free(q)
return dist_maps
|