mirror of
https://github.com/opencv/opencv.git
synced 2024-12-06 02:29:14 +08:00
244 lines
6.1 KiB
C
244 lines
6.1 KiB
C
#include <opencv2/imgproc/imgproc_c.h>
|
|
#include "opencv2/highgui/highgui.hpp"
|
|
|
|
#include <stdio.h>
|
|
void help()
|
|
{
|
|
printf("\nThis program demostrates iterative construction of\n"
|
|
"delaunay triangulation and voronoi tesselation.\n"
|
|
"It draws a random set of points in an image and then delaunay triangulates them.\n"
|
|
"Usage: \n"
|
|
"./delaunay \n"
|
|
"\nThis program builds the traingulation interactively, you may stop this process by\n"
|
|
"hitting any key.\n");
|
|
}
|
|
|
|
CvSubdiv2D* init_delaunay( CvMemStorage* storage,
|
|
CvRect rect )
|
|
{
|
|
CvSubdiv2D* subdiv;
|
|
|
|
subdiv = cvCreateSubdiv2D( CV_SEQ_KIND_SUBDIV2D, sizeof(*subdiv),
|
|
sizeof(CvSubdiv2DPoint),
|
|
sizeof(CvQuadEdge2D),
|
|
storage );
|
|
cvInitSubdivDelaunay2D( subdiv, rect );
|
|
|
|
return subdiv;
|
|
}
|
|
|
|
|
|
void draw_subdiv_point( IplImage* img, CvPoint2D32f fp, CvScalar color )
|
|
{
|
|
cvCircle( img, cvPoint(cvRound(fp.x), cvRound(fp.y)), 3, color, CV_FILLED, 8, 0 );
|
|
}
|
|
|
|
|
|
void draw_subdiv_edge( IplImage* img, CvSubdiv2DEdge edge, CvScalar color )
|
|
{
|
|
CvSubdiv2DPoint* org_pt;
|
|
CvSubdiv2DPoint* dst_pt;
|
|
CvPoint2D32f org;
|
|
CvPoint2D32f dst;
|
|
CvPoint iorg, idst;
|
|
|
|
org_pt = cvSubdiv2DEdgeOrg(edge);
|
|
dst_pt = cvSubdiv2DEdgeDst(edge);
|
|
|
|
if( org_pt && 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 );
|
|
}
|
|
}
|
|
|
|
|
|
void draw_subdiv( IplImage* img, CvSubdiv2D* subdiv,
|
|
CvScalar delaunay_color, CvScalar voronoi_color )
|
|
{
|
|
CvSeqReader reader;
|
|
int i, total = subdiv->edges->total;
|
|
int elem_size = subdiv->edges->elem_size;
|
|
|
|
cvStartReadSeq( (CvSeq*)(subdiv->edges), &reader, 0 );
|
|
|
|
for( i = 0; i < total; i++ )
|
|
{
|
|
CvQuadEdge2D* edge = (CvQuadEdge2D*)(reader.ptr);
|
|
|
|
if( CV_IS_SET_ELEM( edge ))
|
|
{
|
|
draw_subdiv_edge( img, (CvSubdiv2DEdge)edge + 1, voronoi_color );
|
|
draw_subdiv_edge( img, (CvSubdiv2DEdge)edge, delaunay_color );
|
|
}
|
|
|
|
CV_NEXT_SEQ_ELEM( elem_size, reader );
|
|
}
|
|
}
|
|
|
|
|
|
void locate_point( CvSubdiv2D* subdiv, CvPoint2D32f fp, IplImage* img,
|
|
CvScalar active_color )
|
|
{
|
|
CvSubdiv2DEdge e;
|
|
CvSubdiv2DEdge e0 = 0;
|
|
CvSubdiv2DPoint* p = 0;
|
|
|
|
cvSubdiv2DLocate( subdiv, fp, &e0, &p );
|
|
|
|
if( e0 )
|
|
{
|
|
e = e0;
|
|
do
|
|
{
|
|
draw_subdiv_edge( img, e, active_color );
|
|
e = cvSubdiv2DGetEdge(e,CV_NEXT_AROUND_LEFT);
|
|
}
|
|
while( e != e0 );
|
|
}
|
|
|
|
draw_subdiv_point( img, fp, active_color );
|
|
}
|
|
|
|
|
|
void draw_subdiv_facet( IplImage* img, CvSubdiv2DEdge edge )
|
|
{
|
|
CvSubdiv2DEdge t = edge;
|
|
int i, count = 0;
|
|
CvPoint* buf = 0;
|
|
|
|
// count number of edges in facet
|
|
do
|
|
{
|
|
count++;
|
|
t = cvSubdiv2DGetEdge( t, CV_NEXT_AROUND_LEFT );
|
|
} while (t != edge );
|
|
|
|
buf = (CvPoint*)malloc( count * sizeof(buf[0]));
|
|
|
|
// gather points
|
|
t = edge;
|
|
for( i = 0; i < count; i++ )
|
|
{
|
|
CvSubdiv2DPoint* pt = cvSubdiv2DEdgeOrg( t );
|
|
if( !pt ) break;
|
|
buf[i] = cvPoint( cvRound(pt->pt.x), cvRound(pt->pt.y));
|
|
t = cvSubdiv2DGetEdge( t, CV_NEXT_AROUND_LEFT );
|
|
}
|
|
|
|
if( i == count )
|
|
{
|
|
CvSubdiv2DPoint* pt = cvSubdiv2DEdgeDst( cvSubdiv2DRotateEdge( edge, 1 ));
|
|
cvFillConvexPoly( img, buf, count, CV_RGB(rand()&255,rand()&255,rand()&255), CV_AA, 0 );
|
|
cvPolyLine( img, &buf, &count, 1, 1, CV_RGB(0,0,0), 1, CV_AA, 0);
|
|
draw_subdiv_point( img, pt->pt, CV_RGB(0,0,0));
|
|
}
|
|
free( buf );
|
|
}
|
|
|
|
void paint_voronoi( CvSubdiv2D* subdiv, IplImage* img )
|
|
{
|
|
CvSeqReader reader;
|
|
int i, total = subdiv->edges->total;
|
|
int elem_size = subdiv->edges->elem_size;
|
|
|
|
cvCalcSubdivVoronoi2D( subdiv );
|
|
|
|
cvStartReadSeq( (CvSeq*)(subdiv->edges), &reader, 0 );
|
|
|
|
for( i = 0; i < total; i++ )
|
|
{
|
|
CvQuadEdge2D* edge = (CvQuadEdge2D*)(reader.ptr);
|
|
|
|
if( CV_IS_SET_ELEM( edge ))
|
|
{
|
|
CvSubdiv2DEdge e = (CvSubdiv2DEdge)edge;
|
|
// left
|
|
draw_subdiv_facet( img, cvSubdiv2DRotateEdge( e, 1 ));
|
|
|
|
// right
|
|
draw_subdiv_facet( img, cvSubdiv2DRotateEdge( e, 3 ));
|
|
}
|
|
|
|
CV_NEXT_SEQ_ELEM( elem_size, reader );
|
|
}
|
|
}
|
|
|
|
|
|
void run(void)
|
|
{
|
|
char win[] = "source";
|
|
int i;
|
|
CvRect rect = { 0, 0, 600, 600 };
|
|
CvMemStorage* storage;
|
|
CvSubdiv2D* subdiv;
|
|
IplImage* img;
|
|
CvScalar active_facet_color, delaunay_color, voronoi_color, bkgnd_color;
|
|
|
|
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, 0 );
|
|
|
|
cvNamedWindow( win, 1 );
|
|
|
|
storage = cvCreateMemStorage(0);
|
|
subdiv = init_delaunay( storage, rect );
|
|
|
|
printf("Delaunay triangulation will be build now interactively.\n"
|
|
"To stop the process, press any key\n\n");
|
|
|
|
for( i = 0; i < 200; i++ )
|
|
{
|
|
CvPoint2D32f fp = cvPoint2D32f( (float)(rand()%(rect.width-10)+5),
|
|
(float)(rand()%(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, 0 );
|
|
draw_subdiv( img, subdiv, delaunay_color, voronoi_color );
|
|
cvShowImage( win, img );
|
|
|
|
if( cvWaitKey( 100 ) >= 0 )
|
|
break;
|
|
}
|
|
|
|
cvSet( img, bkgnd_color, 0 );
|
|
paint_voronoi( subdiv, img );
|
|
cvShowImage( win, img );
|
|
|
|
cvWaitKey(0);
|
|
|
|
cvReleaseMemStorage( &storage );
|
|
cvReleaseImage(&img);
|
|
cvDestroyWindow( win );
|
|
}
|
|
|
|
int main( int argc, char** argv )
|
|
{
|
|
#ifdef _MSC_VER
|
|
argc, argv;
|
|
#endif
|
|
help();
|
|
run();
|
|
return 0;
|
|
}
|
|
|
|
#ifdef _EiC
|
|
main( 1, "delaunay.c" );
|
|
#endif
|