boofcv.abst.calib.PlanarDetectorChessboard Maven / Gradle / Ivy

/*
 * Copyright (c) 2011-2015, Peter Abeles. All Rights Reserved.
 *
 * This file is part of BoofCV (http://boofcv.org).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package boofcv.abst.calib;

import boofcv.alg.feature.detect.chess.DetectChessboardFiducial;
import boofcv.alg.shapes.polygon.BinaryPolygonConvexDetector;
import boofcv.factory.shape.FactoryShapeDetector;
import boofcv.struct.image.ImageFloat32;
import georegression.struct.point.Point2D_F64;

import java.util.ArrayList;
import java.util.List;

/**
 * Wrapper around {@link DetectChessboardFiducial} for {@link PlanarCalibrationDetector}
 * 
 * @author Peter Abeles
 */
public class PlanarDetectorChessboard implements PlanarCalibrationDetector {

	DetectChessboardFiducial alg;

	List layoutPoints;
	List detected;

	public PlanarDetectorChessboard(ConfigChessboard config) {

		BinaryPolygonConvexDetector detectorSquare =
				FactoryShapeDetector.polygon(config.square, ImageFloat32.class);

		alg = new DetectChessboardFiducial(
				config.numCols,config.numRows,detectorSquare,ImageFloat32.class);
		alg.setUserBinaryThreshold(config.binaryGlobalThreshold);
		alg.setUserAdaptiveRadius(config.binaryAdaptiveRadius);
		alg.setUserAdaptiveBias(config.binaryAdaptiveBias);

		layoutPoints = gridChess(config.numCols,config.numRows,config.squareWidth);
	}

	@Override
	public boolean process(ImageFloat32 input) {
		if( alg.process(input) )  {
			detected = new ArrayList();
			List found = alg.getCalibrationPoints();
			for (int i = 0; i < found.size(); i++) {
				detected.add( found.get(i).copy() );
			}
			return true;
		} else {
			return false;
		}
	}

	@Override
	public List getDetectedPoints() {
		return detected;
	}

	@Override
	public List getLayout() {
		return layoutPoints;
	}

	public DetectChessboardFiducial getAlg() {
		return alg;
	}

	/**
	 * This target is composed of a checkered chess board like squares.  Each corner of an interior square
	 * touches an adjacent square, but the sides are separated.  Only interior square corners provide
	 * calibration points.
	 *
	 * @param numCols Number of grid columns in the calibration target
	 * @param numRows Number of grid rows in the calibration target
	 * @param squareWidth How wide each square is.  Units are target dependent.
	 * @return Target description
	 */
	public static List gridChess( int numCols , int numRows , double squareWidth )
	{
		List all = new ArrayList();

		// convert it into the number of calibration points
		numCols = numCols - 1;
		numRows = numRows - 1;

		// center the grid around the origin. length of a size divided by two
		double startX = -((numCols-1)*squareWidth)/2.0;
		double startY = -((numRows-1)*squareWidth)/2.0;

		for( int i = numRows-1; i >= 0; i-- ) {
			double y = startY+i*squareWidth;
			for( int j = 0; j < numCols; j++ ) {
				double x = startX+j*squareWidth;
				all.add( new Point2D_F64(x,y));
			}
		}

		return all;
	}
}