def test_segment_contours_in_region(filepath):
path2d = pathengine.suPath2D()
offset = -14
line_width = 1 #int(abs(offset)/2)
pe = pathengine.pathEngine()
pe.generate_contours_from_img(filepath, True)
pe.im = cv2.cvtColor(pe.im, cv2.COLOR_GRAY2BGR)
contour_tree = pe.convert_hiearchy_to_PyPolyTree()
path2d.group_boundary = pe.get_contours_from_each_connected_region(
contour_tree, '0')
color = (255, 0, 0)
## Build a init graph from boundaries
# distance threshold between two adjacent layers
dist_th = abs(offset) * 1.2
iB = 0
for boundary in path2d.group_boundary.values():
msg = "Region {}: has {} boundry contours.".format(iB, len(boundary))
print(msg)
iso_contours = pe.fill_closed_region_with_iso_contours(
boundary, offset)
# init contour graph for each region
num_contours = 0
iso_contours_2D = []
for i in range(len(iso_contours)):
for j in range(len(iso_contours[i])):
# resample and convert to np.array
iso_contours[i][j] = pe.resample_curve_by_equal_dist(
iso_contours[i][j], abs(offset / 2))
iso_contours_2D.append(iso_contours[i][j])
num_contours += 1
# @R is the relationship matrix
R = np.zeros((num_contours, num_contours)).astype(int)
# @input: iso_contours c_{i,j}
i = 0
for cs in iso_contours[:
-1]: # for each group contour[i], where i*offset reprents the distance from boundaries
j1 = 0
for c1 in cs:
c1_id = path2d.get_contour_id(i, j1, iso_contours)
pathengine.suPath2D.draw_line(np.vstack([c1, c1[0]]), pe.im,
color, 1, 2)
pathengine.suPath2D.draw_text(str(c1_id + 1), c1[0], pe.im,
(0, 0, 255))
j2 = 0
for c2 in iso_contours[i + 1]:
dist = scid.cdist(c1, c2, 'euclidean')
min_dist = np.min(dist)
#print(dist)
if (min_dist < dist_th):
c2_id = path2d.get_contour_id(i + 1, j2, iso_contours)
R[c1_id][c2_id] = 1
#debug: get indexes of two closest points
gId = np.argmin(dist)
pid_c1 = int(gId / dist.shape[1])
pid_c2 = gId - dist.shape[1] * pid_c1
pathengine.suPath2D.draw_line(
np.asarray([c1[pid_c1], c2[pid_c2]]), pe.im,
(0, 0, 255), 1, 0)
j2 += 1
j1 += 1
i += 1
#visualize
graph = suGraph.suGraph()
#graph.init_from_matrix(R)
pockets = graph.classify_nodes_by_type(R)
N = len(pockets)
colors = path2d.generate_RGB_list(N)
p_id = 0
for p in pockets:
print(np.array(p) + 1)
for idx in p:
pathengine.suPath2D.draw_line(
np.vstack([iso_contours_2D[idx], iso_contours_2D[idx][0]]),
pe.im, colors[p_id], 2, 4)
p_id += 1
path2d.group_isocontours.append(iso_contours)
path2d.group_isocontours_2D.append(iso_contours_2D)
path2d.group_relationship_matrix.append(R)
iB += 1
graph.to_Mathematica("")
gray = cv2.cvtColor(pe.im, cv2.COLOR_BGR2GRAY)
#ret, mask = cv2.threshold(gray, 1, 255,cv2.THRESH_BINARY)
pe.im[np.where((pe.im == [0, 0, 0]).all(axis=2))] = [255, 255, 255]
cv2.imwrite("d:/tmp.png", pe.im)
cv2.imshow("Art", pe.im)
cv2.waitKey(0)
def test_pocket_spiral(filepath, offset=-14, reverseImage=True):
path2d = pathengine.suPath2D()
line_width = 1 #int(abs(offset)/2)
pe = pathengine.pathEngine()
pe.generate_contours_from_img(filepath, reverseImage)
pe.im = cv2.cvtColor(pe.im, cv2.COLOR_GRAY2BGR)
contour_tree = pe.convert_hiearchy_to_PyPolyTree()
path2d.group_boundary = pe.get_contours_from_each_connected_region(
contour_tree, '0')
## Build a init graph from boundaries
# contour distance threshold between adjacent layers
dist_th = abs(offset) * 1.05
iB = 0
for boundary in path2d.group_boundary.values():
msg = "Region {}: has {} boundry contours.".format(iB, len(boundary))
print(msg)
iso_contours = pe.fill_closed_region_with_iso_contours(
boundary, offset)
# init contour graph for each region
num_contours = 0
iso_contours_2D = []
map_ij = []
for i in range(len(iso_contours)):
for j in range(len(iso_contours[i])):
# resample and convert to np.array
iso_contours[i][j] = pe.resample_curve_by_equal_dist(
iso_contours[i][j],
abs(offset) / 4)
if (i == 0):
iso_contours_2D.append(np.flip(iso_contours[i][j], 0))
else:
iso_contours_2D.append(iso_contours[i][j])
map_ij.append([i, j])
num_contours += 1
# @R is the relationship matrix
R = np.zeros((num_contours, num_contours)).astype(int)
i = 0
for cs in iso_contours[:
-1]: # for each group contour[i], where i*offset reprents the distance from boundaries
j1 = 0
for c1 in cs:
c1_id = path2d.get_contour_id(i, j1, iso_contours)
j2 = 0
for c2 in iso_contours[i + 1]:
dist = scid.cdist(c1, c2, 'euclidean')
min_dist = np.min(dist)
#print(dist)
if (min_dist < dist_th):
c2_id = path2d.get_contour_id(i + 1, j2, iso_contours)
R[c1_id][c2_id] = 1
j2 += 1
j1 += 1
i += 1
#visualize
graph = suGraph.suGraph()
#graph.init_from_matrix(R)
#graph.simplify(map_ij)
pockets = graph.classify_nodes_by_type(R, map_ij)
#graph.to_Mathematica("")
print(pockets)
# gen spiral for each pocket
spirals = []
for p in pockets:
cs = []
for c_id in p:
cs.append(iso_contours_2D[c_id])
if (c_id == 37):
pe.path2d.draw_text(str(c_id + 1),
iso_contours_2D[c_id][0], pe.im)
if (len(cs) != 0):
spiral = pe.build_spiral_for_pocket(cs)
spirals.append(spiral)
for p_id in range(len(pockets)):
#node = graph.get_node(where_pocket_id = p_id)
#if node.is_typeII():
#connect_spiral(node_pre.pocket_id, p_id)
#connect_spiral(node_next.pocket_id, p_id)
if len(pockets[p_id]) == 1:
color = [0, 0, 255]
else:
color = [255, 0, 0]
path2d.draw_line(spirals[p_id], pe.im, color, 1)
graph.to_Mathematica("")
path2d.group_isocontours.append(iso_contours)
path2d.group_isocontours_2D.append(iso_contours_2D)
path2d.group_relationship_matrix.append(R)
iB += 1
graph.connect_node_by_spiral(spirals)
gray = cv2.cvtColor(pe.im, cv2.COLOR_BGR2GRAY)
pe.im[np.where((pe.im == [0, 0, 0]).all(axis=2))] = [255, 255, 255]
cv2.imshow("Art", pe.im)
cv2.waitKey(0)
def test_filling_with_continues_spiral(filepath,
offset=-14,
reverseImage=True):
path2d = pathengine.suPath2D()
line_width = 1 #int(abs(offset)/2)
pe = pathengine.pathEngine()
pe.generate_contours_from_img(filepath, reverseImage)
pe.im = cv2.cvtColor(pe.im, cv2.COLOR_GRAY2BGR)
contour_tree = pe.convert_hiearchy_to_PyPolyTree()
path2d.group_boundary = pe.get_contours_from_each_connected_region(
contour_tree, '0')
iB = 0
for boundary in path2d.group_boundary.values():
msg = "Region {}: has {} boundry contours.".format(iB, len(boundary))
print(msg)
#pathengine.suPath2D.convto_cw(boundary)
iso_contours = pe.fill_closed_region_with_iso_contours(
boundary, offset)
# init contour graph for iso contour by a distance matrix
num_contours = 0
iso_contours_2D, graph = pe.init_isocontour_graph(iso_contours)
graph.to_Mathematica("")
if not graph.is_connected():
print("not connected")
ret = pe.reconnect_from_leaf_node(graph, iso_contours,
abs(offset * 1.2))
if (ret):
print("re-connect...")
graph.to_Mathematica("")
# generate a minimum-weight spanning tree
graph.to_reverse_delete_MST()
graph.to_Mathematica("")
# generate a minimum-weight spanning tree
pocket_graph = graph.gen_pockets_graph()
pocket_graph.to_Mathematica("")
# generate spiral for each pockets
# deep first search
spirals = {}
def dfs_connect_path_from_bottom(i, nodes, iso_contours_2D, spirals,
offset):
node = nodes[i]
msg = '{} make spiral {}'.format(i + 1, np.asarray(node.data) + 1)
print(msg)
cs = []
for ii in node.data:
cs.append(iso_contours_2D[ii])
spirals[i] = pe.build_spiral_for_pocket(cs)
#path2d.draw_line(spirals[i], pe.im, [255,255,0],1)
if (len(node.next) > 0):
for ic in node.next:
dfs_connect_path_from_bottom(ic, nodes, iso_contours_2D,
spirals, offset)
if (ic == 22 and i == 18):
print("debug")
if (len(spirals[ic]) / len(spirals[i]) > 2):
spirals[i] = pe.test_connect_two_pockets(
spirals[ic], spirals[i], abs(offset))
msg = '{} insert {}'.format(ic + 1, i + 1)
print(msg)
else:
spirals[i] = pe.test_connect_two_pockets(
spirals[i], spirals[ic], abs(offset))
msg = '{} insert {}'.format(i + 1, ic + 1)
print(msg)
if (i == 6 and ic == 7):
msg = "{}: {}, {}: {}".format(i, len(spirals[i]), ic,
len(spirals[ic]))
print(msg)
return
dfs_connect_path_from_bottom(0, pocket_graph.nodes, iso_contours_2D,
spirals, offset)
#cs = []
#for i in node.data:
#cs.append(iso_contours_2D[i])
#if(len(cs) !=0):
#spiral = pe.build_spiral_for_pocket(cs)
#test
verts = []
cs = []
ll = [35, 37, 39]
#k = 0
#for jj in ll:
##path2d.draw_line(iso_contours_2D[jj], pe.im, colors[k],1)
#cs.append(iso_contours_2D[jj-1])
##k += 1
#for p in iso_contours_2D[jj-1]:
#verts.append(p)
#verts = np.array(verts).reshape([len(verts),2])
#draw_pca_for_pocket(spirals[9])
#cv2.circle(pe.im, tuple(spirals[0][-1].astype(int)), 5, (0,0,255))
#il = 0
#for l in ll:
#path2d.draw_line(spirals.get(l-1), pe.im, colors[il],1)
#msg = "{}: {}".format(l-1, len(spirals[l-1]))
#print(msg)
#il+=1
for i in range(len(iso_contours_2D)):
c = ()
if (pathengine.suPath2D.ccw(iso_contours_2D[i])):
c = (255, 0, 0) #ccw
else:
c = (0, 0, 255)
#pathengine.suPath2D.draw_line(iso_contours_2D[i], pe.im, c ,1)
##cv2.circle(pe.im, tuple(iso_contours_2D[i][0].astype(int)), 5, (255,255,0), -1)
##cv2.circle(pe.im, tuple(iso_contours_2D[i][5].astype(int)), 5, (0,0,255), -1)
#pathengine.suPath2D.draw_text(str(i + 1), iso_contours_2D[i][0], pe.im)
spirals[0] = pe.smooth_curve_by_savgol(spirals[0], 3, 1)
pathengine.suPath2D.draw_line(spirals[0], pe.im, [100, 255, 100], 1)
#spiral id
#for i in range(len(spirals)):
#pathengine.suPath2D.draw_text(str(i + 1), spirals[i][0], pe.im)
kappa, smooth = css.compute_curve_css(spirals[22], 2)
css_idx = css.find_css_point(kappa)
for i in css_idx:
cv2.circle(pe.im, tuple(spirals[22][i].astype(int)), 2,
(255, 255, 0), -1)
id_sp = 0
kappa, smooth = css.compute_curve_css(spirals[id_sp], 4)
css_idx = css.find_css_point(kappa)
for i in css_idx:
cv2.circle(pe.im, tuple(spirals[id_sp][i].astype(int)), 2,
(255, 255, 0), -1)
pathengine.suPath2D.draw_line(spirals[22], pe.im, [0, 0, 255], 1)
#pathengine.suPath2D.draw_line(iso_contours_2D[36], pe.im, [0,0,255] ,1)
#path2d.draw_line(spirals.get(3), pe.im, [0,0,255],1)
#path2d.draw_line(spirals.get(1), pe.im, [255,0,20],1)
#path2d.draw_line(spirals.get(2), pe.im, [0,255,33],1)
#colors = pathengine.suPath2D.generate_RGB_list(len(spirals))
#for i in range(len(spirals)):
#path2d.draw_line(spirals[i], pe.im, colors[i],1)
#sptmp = pe.build_spiral_for_pocket(cs)
#path2d.draw_line(sptmp, pe.im, [0,0,255],1)
#path2d.draw_line(spirals.get(9), pe.im, [0,0,255],1)
#path2d.draw_line(spirals.get(5), pe.im, [0,255,0],1)
#path2d.draw_line(spirals[7], pe.im, [255,255,0],1)
iB += 1
#graph.connect_node_by_spiral(spirals)
gray = cv2.cvtColor(pe.im, cv2.COLOR_BGR2GRAY)
pe.im[np.where((pe.im == [0, 0, 0]).all(axis=2))] = [255, 255, 255]
cv2.imwrite("d:/tmp.png", pe.im)
cv2.imshow("Art", pe.im)
cv2.waitKey(0)