def test_edge_sort():
b=BooleanOp()
v=vorig(0,0)
for x in xrange(100):
if x==0:
ang=0
else:
ang=x/6.0
def Vertex(a,b):
x=a*cos(ang)-b*sin(ang)
y=b*cos(ang)+a*sin(ang)
return vorig(int(x),int(y))
for side in [0,1]:
l=Line(Vertex(-10,0),Vertex(0,0))
tosort=lvector([
Line(Vertex(0,0),Vertex(-10,10)),
Line(Vertex(0,0),Vertex(0,10)),
Line(Vertex(0,0),Vertex(10,10))])
out=list(b.dbg_step5_sort_edges(v,l,tosort,side))
if side==0:
assert out[0]==Line(Vertex(0,0),Vertex(-10,10))
assert out[1]==Line(Vertex(0,0),Vertex(0,10))
assert out[2]==Line(Vertex(0,0),Vertex(10,10))
else:
assert out[2]==Line(Vertex(0,0),Vertex(-10,10))
assert out[1]==Line(Vertex(0,0),Vertex(0,10))
assert out[0]==Line(Vertex(0,0),Vertex(10,10))
tosort=lvector([
Line(Vertex(0,0),Vertex(-10,10)),
Line(Vertex(0,10),Vertex(0,0)),
Line(Vertex(0,0),Vertex(10,10))])
out=list(b.dbg_step5_sort_edges(v,l,tosort,side))
if side==0:
assert out[0]==Line(Vertex(0,0),Vertex(-10,10))
assert out[1]==Line(Vertex(0,10),Vertex(0,0))
assert out[2]==Line(Vertex(0,0),Vertex(10,10))
else:
assert out[2]==Line(Vertex(0,0),Vertex(-10,10))
assert out[1]==Line(Vertex(0,10),Vertex(0,0))
assert out[0]==Line(Vertex(0,0),Vertex(10,10))
tosort=lvector([
Line(Vertex(-10,10),Vertex(0,0)),
Line(Vertex(0,10),Vertex(0,0)),
Line(Vertex(0,0),Vertex(10,10))])
out=list(b.dbg_step5_sort_edges(v,l,tosort,side))
if side==0:
assert out[0]==Line(Vertex(-10,10),Vertex(0,0))
assert out[1]==Line(Vertex(0,10),Vertex(0,0))
assert out[2]==Line(Vertex(0,0),Vertex(10,10))
else:
assert out[2]==Line(Vertex(-10,10),Vertex(0,0))
assert out[1]==Line(Vertex(0,10),Vertex(0,0))
assert out[0]==Line(Vertex(0,0),Vertex(10,10))
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)
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