boofcv.abst.sfm.d2.PlToGrayMotion2D 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.abst.sfm.d2;

import boofcv.abst.sfm.AccessPointTracks;
import boofcv.core.image.GConvertImage;
import boofcv.core.image.GeneralizedImageOps;
import boofcv.struct.image.ImageGray;
import boofcv.struct.image.Planar;
import georegression.struct.InvertibleTransform;
import georegression.struct.point.Point2D_F64;

import java.util.List;

/**
 * Wrapper which converts a planar image into a gray scale image before computing its image motion.
 *
 * @author Peter Abeles
 */
public class PlToGrayMotion2D,IT extends InvertibleTransform>
	implements ImageMotion2D,IT>, AccessPointTracks
{
	// motion estimation algorithm for a single band image
	ImageMotion2D motion;
	// if supposed, provides access to track points
	AccessPointTracks access;
	// storage for gray scale image
	T gray;

	public PlToGrayMotion2D(ImageMotion2D motion , Class imageType ) {
		this.motion = motion;
		gray = GeneralizedImageOps.createSingleBand(imageType,1,1);
		if( motion instanceof AccessPointTracks ) {
			access = (AccessPointTracks)motion;
		}
	}

	@Override
	public boolean process(Planar input) {
		gray.reshape(input.width,input.height);
		GConvertImage.average(input, gray);
		return motion.process(gray);
	}

	@Override
	public void reset() {
		motion.reset();
	}

	@Override
	public void setToFirst() {
		motion.setToFirst();
	}

	@Override
	public IT getFirstToCurrent() {
		return motion.getFirstToCurrent();
	}

	@Override
	public Class getTransformType() {
		return motion.getTransformType();
	}

	@Override
	public long getTrackId(int index) {
		return access.getTrackId(index);
	}

	@Override
	public List getAllTracks() {
		return access.getAllTracks();
	}

	@Override
	public boolean isInlier(int index) {
		return access.isInlier(index);
	}

	@Override
	public boolean isNew(int index) {
		return access.isNew(index);
	}
}