def visualize_diff(lines,extras=[]):
points=dict()
for line,col in lines:
for point in list(line.slow_all_vertices()):
points.setdefault(point,[]).append(col)
sqs=[]
for p,collist in points.items():
R,G,B=0,0,0
for r,g,b in collist:
R+=r
G+=g
B+=b
R/=len(collist)
G/=len(collist)
B/=len(collist)
sqs.append(Square(p.get_x()+0.1,p.get_y()+0.1,p.get_x()+0.9,p.get_y()+0.9,(R,G,B)))
for line,col in lines:
print "Drawing Line:",line.get_x1_inexact(),\
line.get_y1_inexact(),\
line.get_x2_inexact(),\
line.get_y2_inexact(),line.taxilen()
if line.taxilen()==0:
sqs.append(Square(
line.get_v1().get_x()+0.4,
line.get_v1().get_y()+0.4,
line.get_v2().get_x()+0.6,
line.get_v2().get_y()+0.6,
tuple(tx/2 for tx in col)))
continue
if (abs(line.get_x1_inexact()-
line.get_x2_inexact())+
abs(line.get_y1_inexact()-
line.get_y2_inexact()))<=0.01:
sqs.append(visualize.Line(
line.get_v1().get_x()+0.5,
line.get_v1().get_y()+0.5,
line.get_v2().get_x()+0.5,
line.get_v2().get_y()+0.5,
tuple(tx/2 for tx in col)))
#Square(line.get_x1_inexact()-0.1,
# line.get_y1_inexact()-0.1,
# line.get_x2_inexact()+0.1,
# line.get_y2_inexact()+0.1,
# tuple(tx/2 for tx in col)))
else:
sqs.append(visualize.Line(
line.get_x1_inexact(),
line.get_y1_inexact(),
line.get_x2_inexact(),
line.get_y2_inexact(),
tuple(tx/2 for tx in col)
))
sqs.extend(extras)
draw_things(sqs)
def dump_cells(bo):
for cell in list(bo.dbg_step5_get_cells()):
sqs=[]
for edge in list(cell.dbg_get_edges()):
line=Line(edge.get_v1(),edge.get_v2())
sqs.append(visualize.Line(
line.get_v1().get_x(),
line.get_v1().get_y(),
line.get_v2().get_x(),
line.get_v2().get_y(),
(255,0,0)))
#print "Cell cover:",list(cell.get_shapes())
#print "Cell type:",cell.get_classification()
draw_things(sqs)
def merge_shapes(shape_a,shape_b,vis=False):
bo=BooleanOp()
print "Shape_a,shape_b: ",shape_a,shape_b
bo.step1_add_lines(shape_a,shape_b)
bo.step2_intersect_lines()
bo.step3_create_edges()
bo.step4_eliminate_deadends()
bo.step5_create_cells()
print "Calling determine hierarchy"
bo.step5b_determine_cell_hierarchy()
#dump_cells(bo)
bo.step6_determine_cell_cover()
bas=BooleanOrStrategy()
bo.step7_classify_cells(bas)
bo.step8_merge_cells();
bo.step9_calc_result();
#bo.step10_eliminate_enclosed_cells();
#bo.step8_merge_cells()
#bo.step9_calc_result()
shape=bo.step11_get_result()
polys=list(shape.get_polys())
if vis:
plines=set()
for poly in polys:
#print "Poly: %s %s"%(poly.get_kind_str(),list(poly.get_lines()))
plines=plines.union(frozenset(poly.get_lines()))
sqs=[]
for line in plines:
sqs.append(visualize.Line(
line.get_v1().get_x(),
line.get_v1().get_y(),
line.get_v2().get_x(),
line.get_v2().get_y(),
(255,0,0)))
draw_things(sqs)
return shape
def vistest_add_lines():
#print "Press any key when debugger ready"
#raw_input()
bo=BooleanOp()
poly_a=Polygon(vvector([
Vertex(0,0),Vertex(2,0),
Vertex(2,2),Vertex(0,2)]))
poly_b=Polygon(vvector([
Vertex(1,1),Vertex(3,1),
Vertex(3,3),Vertex(1,3)]))
shape_a=Shape("shape_a",poly_a)
shape_b=Shape("shape_b",poly_b)
#print "Shape_a,shape_b: ",shape_a,shape_b
bo.step1_add_lines(shape_a,shape_b)
bo.step2_intersect_lines()
lines=list(bo.dbg_step2_get_split_lines())
#print "The found, split lines:"
#print "\n".join(str(x) for x in lines)
assert len(lines)==12
assert Line(Vertex(0,0),Vertex(2,0)) in lines
assert Line(Vertex(2,0),Vertex(2,1)) in lines
assert Line(Vertex(2,1),Vertex(2,2)) in lines
assert Line(Vertex(2,2),Vertex(1,2)) in lines
assert Line(Vertex(1,2),Vertex(0,2)) in lines
assert Line(Vertex(0,2),Vertex(0,0)) in lines
assert Line(Vertex(1,1),Vertex(2,1)) in lines
assert Line(Vertex(2,1),Vertex(3,1)) in lines
assert Line(Vertex(3,1),Vertex(3,3)) in lines
assert Line(Vertex(3,3),Vertex(1,3)) in lines
assert Line(Vertex(1,3),Vertex(1,2)) in lines
assert Line(Vertex(1,2),Vertex(1,1)) in lines
bo.step3_create_edges()
edges=bo.dbg_step3_and_4_get_edges()
edgelines=dict((Line(edge.get_v1(),edge.get_v2()),edge) for edge in edges)
assert set(edgelines.keys())==set(lines)
for key,edge in edgelines.items():
if key.get_v1().get_y()==key.get_v2().get_y():
assert edge.get_k_approx()==0.0
assert edge.get_is_vertical()==False
else:
#vertical
assert edge.get_k_approx()==0.0 #also for vertical but:
assert edge.get_is_vertical()==True
bo.step4_eliminate_deadends()
edges=bo.dbg_step3_and_4_get_edges()
edgelines=dict((Line(edge.get_v1(),edge.get_v2()),edge) for edge in edges)
assert set(edgelines.keys())==set(lines)
for key,edge in edgelines.items():
if key.get_v1().get_y()==key.get_v2().get_y():
assert edge.get_k_approx()==0.0
assert edge.get_is_vertical()==False
else:
#vertical
assert edge.get_k_approx()==0.0 #also for vertical but:
assert edge.get_is_vertical()==True
bo.step5_create_cells()
cells=list(bo.dbg_step5_get_cells())
#print "Cells:\n","\n".join(str(x) for x in cells)
def lineset(*vertices):
s=set()
for v1,v2 in zip(vertices,vertices[1:]+vertices[:1]):
s.add(NondirLine(v1,v2))
return frozenset(s)
def lineset2(*vertices):
s=set()
for v1,v2 in zip(vertices,vertices[1:]+vertices[:1]):
s.add(Line(v1,v2))
return frozenset(s)
facit_cells={
lineset(Vertex(0,0),Vertex(2,0),Vertex(2,1),Vertex(3,1),
Vertex(3,3),Vertex(1,3),Vertex(1,2),Vertex(0,2)):"outline",
lineset(Vertex(1,1),Vertex(2,1),Vertex(2,2),Vertex(1,2)):"center",
lineset(Vertex(2,1),Vertex(3,1),Vertex(3,3),Vertex(1,3),
Vertex(1,2),Vertex(2,2)):"upper_right",
lineset(Vertex(0,0),Vertex(2,0),Vertex(2,1),Vertex(1,1),
Vertex(1,2),Vertex(0,2)):"lower_left"
}
#print "Facit cells","\n".join(str(x) for x in facit_cells.keys())
assert len(cells)==4
for cell in cells:
edges=cell.dbg_get_edges()
edgelines=set(NondirLine(edge.get_v1(),edge.get_v2()) for edge in edges)
if not (edgelines in facit_cells.keys()):
sqs=[]
for line in edgelines:
sqs.append(visualize.Line(
line.get_v1().get_x(),
line.get_v1().get_y(),
line.get_v2().get_x(),
line.get_v2().get_y(),
(255,0,0)))
draw_things(sqs)
bo.step6_determine_cell_cover()
cells=list(bo.dbg_step5_get_cells())
for cell in cells:
edges=cell.dbg_get_edges()
edgelines=frozenset(NondirLine(edge.get_v1(),edge.get_v2()) for edge in edges)
shapes=list(cell.get_shapes())
realcellname=facit_cells[edgelines]
facit_cell_shapes={
'lower_left':('shape_a',),
'center':('shape_a','shape_b'),
'upper_right':('shape_b',),
'outline':()
}
#print "Facit:",facit_cell_shapes[realcellname]
#print "Actual:",tuple(shapes)
assert facit_cell_shapes[realcellname]==tuple(shapes)
#print "Cell: %s, shapes: %s"%(realcellname,", ".join(shapes))
bas=BooleanOrStrategy()
bo.step7_classify_cells(bas)
cells=list(bo.dbg_step5_get_cells())
for cell in cells:
edges=cell.dbg_get_edges()
edgelines=frozenset(NondirLine(edge.get_v1(),edge.get_v2()) for edge in edges)
shapes=list(cell.get_shapes())
realcellname=facit_cells[edgelines]
facit_cell_kind={
'lower_left':'SOLID',
'center':'SOLID',
'upper_right':'SOLID',
'outline':'VOID'
}
#print "Cell %s classification: %s"%(realcellname,cell.get_classification())
assert cell.get_classification()==facit_cell_kind[realcellname]
bo.step8_merge_cells()
cells=list(bo.dbg_step5_get_cells())
def lookup_cellname(cell):
edges=cell.dbg_get_edges()
edgelines=frozenset(NondirLine(edge.get_v1(),edge.get_v2()) for edge in edges)
shapes=list(cell.get_shapes())
realcellname=facit_cells[edgelines]
return realcellname
for cell in cells:
facit_cell_merged={
'lower_left':0,
'center':0,
'upper_right':0,
'outline':1
}
realcellname=lookup_cellname(cell)
#print "Cell %s merged_poly: %s, class: %s"%(realcellname,cell.get_merged_poly(),cell.get_classification())
#print "%s: Neighbors:"%realcellname," ".join(lookup_cellname(ce) for ce in cell.get_neighbors())
assert cell.get_merged_poly()==facit_cell_merged[realcellname]
bo.step9_calc_result()
shape=bo.step9_get_result()
polys=list(shape.get_polys())
assert len(polys)==1
poly,=polys
"Poly: %s %s"%(poly.get_kind_str(),list(poly.get_lines()))
assert poly.get_kind_str()=="SOLID"
plines=frozenset(poly.get_lines())
should=lineset2(Vertex(0,0),Vertex(2,0),Vertex(2,1),Vertex(3,1),
Vertex(3,3),Vertex(1,3),Vertex(1,2),Vertex(0,2))
#print "Is: ",plines
#print "Should: ",should
assert plines==should
sqs=[]
for line in plines:
sqs.append(visualize.Line(
line.get_v1().get_x(),
line.get_v1().get_y(),
line.get_v2().get_x(),
line.get_v2().get_y(),
(255,0,0)))
draw_things(sqs)
def vistest_merge_shapes():
for x in xrange(10):
x=5
poly_a=Polygon(vvector([
Vertex(0+x,0),Vertex(2+x,0),
Vertex(2+x,2),Vertex(0+x,2)]))
poly_b=Polygon(vvector([
Vertex(1,1),Vertex(3,1),
Vertex(3,4),Vertex(1,4)]))
shape_a=Shape("shape_a",poly_a)
shape_b=Shape("shape_b",poly_b)
# OOOO
# OOOO
# OOOOOO
# OOOO
# OOOO
bo=BooleanOp()
#print "Shape_a,shape_b: ",shape_a,shape_b
bo.step1_add_lines(shape_a,shape_b)
bo.step2_intersect_lines()
splitset=set([NondirLine(x.get_v1(),x.get_v2()) for x in bo.dbg_step2_get_split_lines()])
#print "Split result: %s"%("\n".join(str(l) for l in splitset),)
assert not NondirLine(Vertex(0,2),Vertex(2,2)) in splitset
bo.step3_create_edges()
edges=list(bo.dbg_step3_and_4_get_edges())
nondiredges=list(NondirLine(x.get_v1(),x.get_v2()) for x in bo.dbg_step3_and_4_get_edges())
sqs=[]
cnt=0
#print "Edges: %s"%(edges,)
for edge in edges:
line=Line(edge.get_v1(),edge.get_v2())
r=(25*cnt)%128+128
g=(128*cnt)%128+128
b=(64*cnt)%128+128
sqs.append(visualize.Line(
line.get_v1().get_x(),
line.get_v1().get_y(),
line.get_v2().get_x(),
line.get_v2().get_y(),
(r,g,b)))
sqs.append(visualize.Square(
line.get_v1().get_x()-0.1,
line.get_v1().get_y()-0.1,
line.get_v1().get_x()+0.1,
line.get_v1().get_y()+0.1,
(r,g,b)))
cnt+=1
draw_things(sqs)
assert not (NondirLine(Vertex(0,2),Vertex(2,2)) in nondiredges)
bo.step4_eliminate_deadends()
bo.step5_create_cells()
bo.step6_determine_cell_cover()
bas=BooleanOrStrategy()
bo.step7_classify_cells(bas)
for cell in list(bo.dbg_step5_get_cells()):
sqs=[]
for edge in list(cell.dbg_get_edges()):
line=Line(edge.get_v1(),edge.get_v2())
sqs.append(visualize.Line(
line.get_v1().get_x(),
line.get_v1().get_y(),
line.get_v2().get_x(),
line.get_v2().get_y(),
(255,0,0)))
#print "Cell cover:",list(cell.get_shapes())
#print "Cell type:",cell.get_classification()
draw_things(sqs)
bo.step8_merge_cells()
bo.step9_calc_result()
shape=bo.step9_get_result()
polys=list(shape.get_polys())
plines=set()
for poly in polys:
#print "Poly: %s %s"%(poly.get_kind_str(),list(poly.get_lines()))
plines=plines.union(frozenset(poly.get_lines()))
sqs=[]
edges=set(bo.dbg_step3_and_4_get_edges())
for line in plines:
sqs.append(visualize.Line(
line.get_v1().get_x(),
line.get_v1().get_y(),
line.get_v2().get_x(),
line.get_v2().get_y(),
(0,200,0)))
draw_things(sqs)
