boofcv.gui.RenderCalibrationTargetsGraphics2D Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of boofcv-swing 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) 2021, 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.gui;

import boofcv.abst.fiducial.calib.RenderCalibrationTargets;
import boofcv.io.image.ConvertBufferedImage;
import boofcv.struct.image.GrayF32;
import boofcv.struct.image.GrayU8;
import georegression.struct.point.Point2D_F64;

import java.awt.*;
import java.awt.geom.Ellipse2D;
import java.awt.geom.Rectangle2D;
import java.awt.image.BufferedImage;

/**
 * Renders calibration targets using {@link Graphics2D}.
 *
 * @author Peter Abeles
 */
@SuppressWarnings({"NullAway.Init"})
public class RenderCalibrationTargetsGraphics2D extends RenderCalibrationTargets {
	int padding;
	double unitsToPixels;

	BufferedImage buffered;
	Graphics2D g2;

	double paperWidth, paperHeight;

	int offsetX, offsetY;

	public RenderCalibrationTargetsGraphics2D( int padding, double unitsToPixels ) {
		this.padding = padding;
		this.unitsToPixels = unitsToPixels;
	}

	public void setPaperSize( double widthUnits, double heightUnits ) {
		this.paperWidth = widthUnits;
		this.paperHeight = heightUnits;
	}

	@Override
	public void specifySize( double width, double height ) {

		int w = (int)(unitsToPixels*width + 0.5);
		int h = (int)(unitsToPixels*height + 0.5);

		if (paperWidth <= 0 || paperHeight <= 0) {
			offsetX = offsetY = padding;
		} else {
			offsetX = ((int)(unitsToPixels*paperWidth + 0.5) - w)/2;
			offsetY = ((int)(unitsToPixels*paperHeight + 0.5) - h)/2;
		}

		if (offsetX <= 0 || offsetY <= 0)
			buffered = new BufferedImage(w, h, BufferedImage.TYPE_INT_RGB);
		else
			buffered = new BufferedImage(w + 2*offsetX, h + 2*offsetY, BufferedImage.TYPE_INT_RGB);

		g2 = buffered.createGraphics();
		g2.setColor(Color.WHITE);
		g2.fillRect(0, 0, buffered.getWidth(), buffered.getHeight());
		g2.setColor(Color.BLACK);
		BoofSwingUtil.antialiasing(g2);
	}

	@Override
	public void markerToTarget( double x, double y, Point2D_F64 p ) {
		p.x = offsetX + x*unitsToPixels;
		p.y = offsetY + y*unitsToPixels;
	}

	@Override
	public void drawSquare( double x, double y, double width ) {
		Rectangle2D.Double r = new Rectangle2D.Double();
		r.x = offsetX + x*unitsToPixels;
		r.y = offsetY + y*unitsToPixels;
		r.width = width*unitsToPixels;
		r.height = r.width;

		g2.fill(r);
	}

	@Override
	public void drawCircle( double cx, double cy, double diameter ) {
		Ellipse2D.Double ellipse = new Ellipse2D.Double();
		ellipse.x = offsetX + (cx - diameter/2)*unitsToPixels;
		ellipse.y = offsetY + (cy - diameter/2)*unitsToPixels;
		ellipse.width = diameter*unitsToPixels;
		ellipse.height = diameter*unitsToPixels;
		g2.fill(ellipse);
	}

	public double getWidthWorld() {
		return buffered.getWidth()/unitsToPixels;
	}

	public double getHeightWorld() {
		return buffered.getHeight()/unitsToPixels;
	}

	public int getOffsetX() {
		return offsetX;
	}

	public int getOffsetY() {
		return offsetY;
	}

	public GrayU8 getGrayU8() {
		GrayU8 gray = new GrayU8(buffered.getWidth(), buffered.getHeight());
		ConvertBufferedImage.convertFrom(buffered, gray);
		return gray;
	}

	public GrayF32 getGrayF32() {
		GrayF32 gray = new GrayF32(buffered.getWidth(), buffered.getHeight());
		ConvertBufferedImage.convertFrom(buffered, gray);
		return gray;
	}

	public BufferedImage getBuffered() {
		return buffered;
	}
}