From d105264bad6c0485ef9c9d7aa8447ca7e6c31231 Mon Sep 17 00:00:00 2001
From: Ilya Lysenkov <no@email>
Date: Tue, 20 Sep 2011 13:35:21 +0000
Subject: [PATCH] Added tests of findCirclesGrid() with the asymmetric pattern.

---
 modules/calib3d/test/test_chesscorners.cpp | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/modules/calib3d/test/test_chesscorners.cpp b/modules/calib3d/test/test_chesscorners.cpp
index 0bca4e037e..9c53c0d2d8 100644
--- a/modules/calib3d/test/test_chesscorners.cpp
+++ b/modules/calib3d/test/test_chesscorners.cpp
@@ -74,23 +74,25 @@ void show_points( const Mat& gray, const Mat& u, const vector<Point2f>& v, Size
 }
 
 
-enum Pattern { CHESSBOARD, CIRCLES_GRID };
+enum Pattern { CHESSBOARD, CIRCLES_GRID, ASYMMETRIC_CIRCLES_GRID };
 
 class CV_ChessboardDetectorTest : public cvtest::BaseTest
 {
 public:
-    CV_ChessboardDetectorTest( Pattern pattern );
+    CV_ChessboardDetectorTest( Pattern pattern, int algorithmFlags = 0 );
 protected:
     void run(int);
     void run_batch(const string& filename);
     bool checkByGenerator();
 
     Pattern pattern;
+    int algorithmFlags;
 };
 
-CV_ChessboardDetectorTest::CV_ChessboardDetectorTest( Pattern _pattern )
+CV_ChessboardDetectorTest::CV_ChessboardDetectorTest( Pattern _pattern, int _algorithmFlags )
 {
     pattern = _pattern;
+    algorithmFlags = _algorithmFlags;
 }
 
 double calcError(const vector<Point2f>& v, const Mat& u)
@@ -147,6 +149,9 @@ void CV_ChessboardDetectorTest::run( int /*start_from */)
         case CIRCLES_GRID:
             run_batch("circles_list.dat");
             break;
+        case ASYMMETRIC_CIRCLES_GRID:
+            run_batch("acircles_list.dat");
+            break;
     }
 }
 
@@ -169,6 +174,9 @@ void CV_ChessboardDetectorTest::run_batch( const string& filename )
         case CIRCLES_GRID:
             folder = string(ts.get_data_path()) + "cameracalibration/circles/";
             break;
+        case ASYMMETRIC_CIRCLES_GRID:
+            folder = string(ts.get_data_path()) + "cameracalibration/asymmetric_circles/";
+            break;
     }
 
     FileStorage fs( folder + filename, FileStorage::READ );
@@ -229,6 +237,9 @@ void CV_ChessboardDetectorTest::run_batch( const string& filename )
             case CIRCLES_GRID:
                 result = findCirclesGrid(gray, pattern_size, v);
                 break;
+            case ASYMMETRIC_CIRCLES_GRID:
+                result = findCirclesGrid(gray, pattern_size, v, CALIB_CB_ASYMMETRIC_GRID | algorithmFlags);
+                break;
         }
         show_points( gray, Mat(), v, pattern_size, result );
         if( !result || v.size() != count_exp )
@@ -439,5 +450,7 @@ bool CV_ChessboardDetectorTest::checkByGenerator()
 
 TEST(Calib3d_ChessboardDetector, accuracy) {  CV_ChessboardDetectorTest test( CHESSBOARD ); test.safe_run(); }
 TEST(Calib3d_CirclesPatternDetector, accuracy) { CV_ChessboardDetectorTest test( CIRCLES_GRID ); test.safe_run(); }
+TEST(Calib3d_AsymmetricCirclesPatternDetector, accuracy) { CV_ChessboardDetectorTest test( ASYMMETRIC_CIRCLES_GRID ); test.safe_run(); }
+TEST(Calib3d_AsymmetricCirclesPatternDetectorWithClustering, accuracy) { CV_ChessboardDetectorTest test( ASYMMETRIC_CIRCLES_GRID, CALIB_CB_CLUSTERING ); test.safe_run(); }
 
 /* End of file. */