boofcv.alg.sfm.d2.FactoryStitchingTransform Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of boofcv-sfm Show documentation
BoofCV is an open source Java library for real-time computer vision and robotics applications.
There is a newer version: 1.1.7
/*
 * Copyright (c) 2011-2017, 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.alg.sfm.d2;

import boofcv.alg.distort.PixelTransformAffine_F32;
import boofcv.alg.distort.PixelTransformHomography_F32;
import boofcv.struct.distort.PixelTransform2_F32;
import georegression.struct.ConvertFloatType;
import georegression.struct.affine.Affine2D_F32;
import georegression.struct.affine.Affine2D_F64;
import georegression.struct.homography.Homography2D_F32;
import georegression.struct.homography.Homography2D_F64;

/**
 * Factory for creating {@link StitchingTransform} of different motion models.
 *
 * @author Peter Abeles
 */
public class FactoryStitchingTransform {

	public static StitchingTransform createAffine_F64() {

		return new StitchingTransform() {
			Affine2D_F32 input_F32 = new Affine2D_F32();

			@Override
			public PixelTransform2_F32 convertPixel(Affine2D_F64 input, PixelTransform2_F32 output) {

				ConvertFloatType.convert(input,input_F32);

				if( output != null ) {
					((PixelTransformAffine_F32)output).set(input_F32);
				} else {
					PixelTransformAffine_F32 a = new PixelTransformAffine_F32();
					a.set(input_F32);
					output = a;

				}

				return output;
			}

			@Override
			public Homography2D_F64 convertH(Affine2D_F64 input, Homography2D_F64 output) {
				if( output == null )
					output = new Homography2D_F64();
				output.set(
						input.a11,input.a12,input.tx,
						input.a21,input.a22,input.ty,
						0,0,1);
				return output;
			}
		};
	}

	public static StitchingTransform createAffine_F32() {

		return new StitchingTransform() {
			@Override
			public PixelTransform2_F32 convertPixel(Affine2D_F32 input, PixelTransform2_F32 output) {
				if( output != null ) {
					((PixelTransformAffine_F32)output).set(input);
				} else {
					output = new PixelTransformAffine_F32(input);
				}

				return output;
			}

			@Override
			public Homography2D_F64 convertH(Affine2D_F32 input, Homography2D_F64 output) {
				if( output == null )
					output = new Homography2D_F64();
				output.set(
						input.a11,input.a12,input.tx,
						input.a21,input.a22,input.ty,
						0,0,1);
				return output;
			}
		};
	}

	public static StitchingTransform createHomography_F32() {
		return new StitchingTransform() {
			@Override
			public PixelTransform2_F32 convertPixel(Homography2D_F32 input, PixelTransform2_F32 output) {
				if( output != null ) {
					((PixelTransformHomography_F32)output).set(input);
				} else {
					output = new PixelTransformHomography_F32(input);
				}

				return output;
			}

			@Override
			public Homography2D_F64 convertH(Homography2D_F32 input, Homography2D_F64 output) {
				if( output == null )
					output = new Homography2D_F64();
				output.set(
						input.a11,input.a12,input.a13,
						input.a21,input.a22,input.a23,
						input.a31,input.a32,input.a33);
				return output;
			}
		};
	}

	public static StitchingTransform createHomography_F64() {
		return new StitchingTransform() {
			@Override
			public PixelTransform2_F32 convertPixel(Homography2D_F64 input, PixelTransform2_F32 output) {
				if( output != null ) {
					((PixelTransformHomography_F32)output).set(input);
				} else {
					output = new PixelTransformHomography_F32(input);
				}

				return output;
			}

			@Override
			public Homography2D_F64 convertH(Homography2D_F64 input, Homography2D_F64 output) {
				if( output == null )
					output = new Homography2D_F64();
				output.set(input);
				return output;
			}
		};
	}
}