vcs.maemo.org Git - opencv/blob - samples/octave/houghlines.m

   1 #! /usr/bin/env octave
   2 ## This is a standalone program. Pass an image name as a first parameter of the program.
   3
   4 cv;
   5 highgui;
   6
   7 ## toggle between CV_HOUGH_STANDARD and CV_HOUGH_PROBILISTIC
   8 USE_STANDARD=0;
   9
  10 filename = "../../docs/ref/pics/building.jpg"
  11 if (size(argv, 1)>=1)
  12   filename = argv(){1};
  13 endif
  14
  15 src=cvLoadImage(filename, 0);
  16 if (!swig_this(src))
  17   printf("Error opening image %s\n",filename);
  18   exit(-1);
  19 endif
  20
  21 dst = cvCreateImage( cvGetSize(src), 8, 1 );
  22 color_dst = cvCreateImage( cvGetSize(src), 8, 3 );
  23 storage = cvCreateMemStorage(0);
  24 lines = 0;
  25 cvCanny( src, dst, 50, 200, 3 );
  26 cvCvtColor( dst, color_dst, CV_GRAY2BGR );
  27
  28 if (USE_STANDARD)
  29   lines = cvHoughLines2( dst, storage, CV_HOUGH_STANDARD, 1, CV_PI/180, 100, 0, 0 );
  30
  31   for i=0:min(lines.total, 100)-1,
  32     line = lines{i};
  33     rho = line{0};
  34     theta = line{1};
  35     pt1 = CvPoint();
  36     pt2 = CvPoint();
  37     a = cos(theta);
  38     b = sin(theta);
  39     x0 = a*rho;
  40     y0 = b*rho;
  41     pt1.x = cvRound(x0 + 1000*(-b));
  42     pt1.y = cvRound(y0 + 1000*(a));
  43     pt2.x = cvRound(x0 - 1000*(-b));
  44     pt2.y = cvRound(y0 - 1000*(a));
  45     cvLine( color_dst, pt1, pt2, CV_RGB(255,0,0), 3, 8 );
  46   endfor
  47
  48 else
  49   lines = cvHoughLines2( dst, storage, CV_HOUGH_PROBABILISTIC, 1, CV_PI/180, 50, 50, 10 );
  50   for line = CvSeq_map(lines),
  51     line = line{1};
  52     cvLine( color_dst, line{0}, line{1}, CV_RGB(255,0,0), 3, 8 );
  53   endfor
  54 endif
  55
  56 cvNamedWindow( "Source", 1 );
  57 cvShowImage( "Source", src );
  58
  59 cvNamedWindow( "Hough", 1 );
  60 cvShowImage( "Hough", color_dst );
  61
  62 cvWaitKey(0);