150 lines
3.9 KiB
Python
Executable File
150 lines
3.9 KiB
Python
Executable File
#!/usr/bin/python
|
|
"""
|
|
the script demostrates iterative construction of
|
|
delaunay triangulation and voronoi tesselation
|
|
|
|
Original Author (C version): ?
|
|
Converted to Python by: Roman Stanchak
|
|
"""
|
|
from opencv.cv import *
|
|
from opencv.highgui import *
|
|
from random import random,randint
|
|
|
|
def draw_subdiv_point( img, fp, color ):
|
|
cvCircle( img, cvPoint(cvRound(fp.x), cvRound(fp.y)), 3, color, CV_FILLED, 8, 0 );
|
|
|
|
|
|
|
|
def draw_subdiv_edge( img, edge, color ):
|
|
org_pt = cvSubdiv2DEdgeOrg(edge);
|
|
dst_pt = cvSubdiv2DEdgeDst(edge);
|
|
|
|
if org_pt and dst_pt :
|
|
|
|
org = org_pt.pt;
|
|
dst = dst_pt.pt;
|
|
|
|
iorg = cvPoint( cvRound( org.x ), cvRound( org.y ));
|
|
idst = cvPoint( cvRound( dst.x ), cvRound( dst.y ));
|
|
|
|
cvLine( img, iorg, idst, color, 1, CV_AA, 0 );
|
|
|
|
|
|
def draw_subdiv( img, subdiv, delaunay_color, voronoi_color ):
|
|
|
|
total = subdiv.edges.total;
|
|
elem_size = subdiv.edges.elem_size;
|
|
|
|
for edge in subdiv.edges:
|
|
edge_rot = cvSubdiv2DRotateEdge( edge, 1 )
|
|
|
|
if( CV_IS_SET_ELEM( edge )):
|
|
draw_subdiv_edge( img, edge_rot, voronoi_color );
|
|
draw_subdiv_edge( img, edge, delaunay_color );
|
|
|
|
|
|
def locate_point( subdiv, fp, img, active_color ):
|
|
|
|
[res, e0, p] = cvSubdiv2DLocate( subdiv, fp );
|
|
|
|
if e0:
|
|
e = e0
|
|
while True:
|
|
draw_subdiv_edge( img, e, active_color );
|
|
e = cvSubdiv2DGetEdge(e,CV_NEXT_AROUND_LEFT);
|
|
if e == e0:
|
|
break
|
|
|
|
draw_subdiv_point( img, fp, active_color );
|
|
|
|
|
|
def draw_subdiv_facet( img, edge ):
|
|
|
|
t = edge;
|
|
count = 0;
|
|
|
|
# count number of edges in facet
|
|
while count == 0 or t != edge:
|
|
count+=1
|
|
t = cvSubdiv2DGetEdge( t, CV_NEXT_AROUND_LEFT );
|
|
|
|
buf = []
|
|
|
|
# gather points
|
|
t = edge;
|
|
for i in range(count):
|
|
assert t>4
|
|
pt = cvSubdiv2DEdgeOrg( t );
|
|
if not pt:
|
|
break;
|
|
buf.append( cvPoint( cvRound(pt.pt.x), cvRound(pt.pt.y) ) );
|
|
t = cvSubdiv2DGetEdge( t, CV_NEXT_AROUND_LEFT );
|
|
|
|
if( len(buf)==count ):
|
|
pt = cvSubdiv2DEdgeDst( cvSubdiv2DRotateEdge( edge, 1 ));
|
|
cvFillConvexPoly( img, buf, CV_RGB(randint(0,255),randint(0,255),randint(0,255)), CV_AA, 0 );
|
|
cvPolyLine( img, [buf], 1, CV_RGB(0,0,0), 1, CV_AA, 0);
|
|
draw_subdiv_point( img, pt.pt, CV_RGB(0,0,0));
|
|
|
|
def paint_voronoi( subdiv, img ):
|
|
total = subdiv.edges.total;
|
|
elem_size = subdiv.edges.elem_size;
|
|
|
|
cvCalcSubdivVoronoi2D( subdiv );
|
|
|
|
for edge in subdiv.edges:
|
|
|
|
if( CV_IS_SET_ELEM( edge )):
|
|
# left
|
|
draw_subdiv_facet( img, cvSubdiv2DRotateEdge( edge, 1 ));
|
|
|
|
# right
|
|
draw_subdiv_facet( img, cvSubdiv2DRotateEdge( edge, 3 ));
|
|
|
|
if __name__ == '__main__':
|
|
win = "source";
|
|
rect = cvRect( 0, 0, 600, 600 );
|
|
|
|
active_facet_color = CV_RGB( 255, 0, 0 );
|
|
delaunay_color = CV_RGB( 0,0,0);
|
|
voronoi_color = CV_RGB(0, 180, 0);
|
|
bkgnd_color = CV_RGB(255,255,255);
|
|
|
|
img = cvCreateImage( cvSize(rect.width,rect.height), 8, 3 );
|
|
cvSet( img, bkgnd_color );
|
|
|
|
cvNamedWindow( win, 1 );
|
|
|
|
storage = cvCreateMemStorage(0);
|
|
subdiv = cvCreateSubdivDelaunay2D( rect, storage );
|
|
|
|
print "Delaunay triangulation will be build now interactively."
|
|
print "To stop the process, press any key\n";
|
|
|
|
for i in range(200):
|
|
fp = cvPoint2D32f( random()*(rect.width-10)+5, random()*(rect.height-10)+5 )
|
|
|
|
locate_point( subdiv, fp, img, active_facet_color );
|
|
cvShowImage( win, img );
|
|
|
|
if( cvWaitKey( 100 ) >= 0 ):
|
|
break;
|
|
|
|
cvSubdivDelaunay2DInsert( subdiv, fp );
|
|
cvCalcSubdivVoronoi2D( subdiv );
|
|
cvSet( img, bkgnd_color );
|
|
draw_subdiv( img, subdiv, delaunay_color, voronoi_color );
|
|
cvShowImage( win, img );
|
|
|
|
if( cvWaitKey( 100 ) >= 0 ):
|
|
break;
|
|
|
|
|
|
cvSet( img, bkgnd_color );
|
|
paint_voronoi( subdiv, img );
|
|
cvShowImage( win, img );
|
|
|
|
cvWaitKey(0);
|
|
|
|
cvDestroyWindow( win );
|