Update to 2.0.0 tree from current Fremantle build
[opencv] / samples / octave / houghlines.m
diff --git a/samples/octave/houghlines.m b/samples/octave/houghlines.m
new file mode 100644 (file)
index 0000000..d8abf8b
--- /dev/null
@@ -0,0 +1,62 @@
+#! /usr/bin/env octave
+## This is a standalone program. Pass an image name as a first parameter of the program.
+
+cv;
+highgui;
+
+## toggle between CV_HOUGH_STANDARD and CV_HOUGH_PROBILISTIC
+USE_STANDARD=0;
+
+filename = "../../docs/ref/pics/building.jpg"
+if (size(argv, 1)>=1)
+  filename = argv(){1};
+endif
+
+src=cvLoadImage(filename, 0);
+if (!swig_this(src))
+  printf("Error opening image %s\n",filename);
+  exit(-1);
+endif
+
+dst = cvCreateImage( cvGetSize(src), 8, 1 );
+color_dst = cvCreateImage( cvGetSize(src), 8, 3 );
+storage = cvCreateMemStorage(0);
+lines = 0;
+cvCanny( src, dst, 50, 200, 3 );
+cvCvtColor( dst, color_dst, CV_GRAY2BGR );
+
+if (USE_STANDARD)
+  lines = cvHoughLines2( dst, storage, CV_HOUGH_STANDARD, 1, CV_PI/180, 100, 0, 0 );
+
+  for i=0:min(lines.total, 100)-1,
+    line = lines{i};
+    rho = line{0};
+    theta = line{1};
+    pt1 = CvPoint();
+    pt2 = CvPoint();
+    a = cos(theta);
+    b = sin(theta);
+    x0 = a*rho;
+    y0 = b*rho;
+    pt1.x = cvRound(x0 + 1000*(-b));
+    pt1.y = cvRound(y0 + 1000*(a));
+    pt2.x = cvRound(x0 - 1000*(-b));
+    pt2.y = cvRound(y0 - 1000*(a));
+    cvLine( color_dst, pt1, pt2, CV_RGB(255,0,0), 3, 8 );
+  endfor
+
+else
+  lines = cvHoughLines2( dst, storage, CV_HOUGH_PROBABILISTIC, 1, CV_PI/180, 50, 50, 10 );
+  for line = CvSeq_map(lines),
+    line = line{1};
+    cvLine( color_dst, line{0}, line{1}, CV_RGB(255,0,0), 3, 8 );
+  endfor
+endif
+
+cvNamedWindow( "Source", 1 );
+cvShowImage( "Source", src );
+
+cvNamedWindow( "Hough", 1 );
+cvShowImage( "Hough", color_dst );
+
+cvWaitKey(0);