98 lines
4.0 KiB
Java
98 lines
4.0 KiB
Java
package com.cringe_studios.christmastreescanning.scanning;
|
|
|
|
import java.awt.geom.Point2D;
|
|
import java.awt.image.BufferedImage;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
public class ImageScanner {
|
|
public static final byte GAMMA_DELTA_DEFAULT = 0;
|
|
|
|
static private Point2D getBrightestSpotGrayscale(BufferedImage grayScaleImage, byte gammaDelta) {
|
|
int[] imageColors = grayScaleImage.getRGB(0, 0, grayScaleImage.getWidth(), grayScaleImage.getHeight(), new int[grayScaleImage.getWidth() * grayScaleImage.getHeight()], 0, 0);
|
|
|
|
for (int i = 0; i < imageColors.length; i++) {
|
|
imageColors[i] &= 0x0000_00FF; // because the image is grayscale, only one component is needed
|
|
}
|
|
|
|
int brightestValue = 0;
|
|
|
|
for (int i = 0; i < imageColors.length; i++) {
|
|
if (brightestValue < imageColors[i]) {
|
|
brightestValue = imageColors[i];
|
|
}
|
|
}
|
|
|
|
List<Integer> brightSpots = new ArrayList<Integer>();
|
|
|
|
for (int i = 0; i < imageColors.length; i++) {
|
|
if (imageColors[i] >= brightestValue - gammaDelta) {
|
|
brightSpots.add(Integer.valueOf(i));
|
|
}
|
|
}
|
|
|
|
// The following line might be incorrect; TODO double check and write a test
|
|
Point2D brightPoints[] = brightSpots.stream()
|
|
.map(s -> new Point2D.Double(s.intValue() % grayScaleImage.getWidth(), s.intValue() / grayScaleImage.getWidth()))
|
|
.toArray(Point2D.Double[]::new);
|
|
|
|
Point2D brightestSpot = new PointInterpolator(brightPoints).getCenter();
|
|
|
|
return brightestSpot;
|
|
}
|
|
|
|
// gammaDelta is the difference between the brightest pixel and the darkest
|
|
// pixel included in the search set for the brightest spot
|
|
// The brightest point Operation is done per color dimension, so you could
|
|
// provide different images for red, green and blue brightest points
|
|
static public Point2D getBrightestSpot(BufferedImage redImage, BufferedImage greenImage, BufferedImage blueImage,
|
|
byte gammaDelta) {
|
|
Point2D brightestRedSpot = getBrightestSpotGrayscale(redImage, gammaDelta);
|
|
Point2D brightestGreenSpot = getBrightestSpotGrayscale(greenImage, gammaDelta);
|
|
Point2D brightestBlueSpot = getBrightestSpotGrayscale(blueImage, gammaDelta);
|
|
|
|
double RGdistance = brightestRedSpot.distanceSq(brightestGreenSpot);
|
|
double GBdistance = brightestGreenSpot.distanceSq(brightestBlueSpot);
|
|
double BRdistance = brightestBlueSpot.distanceSq(brightestRedSpot);
|
|
|
|
PointInterpolator brightestSpot;
|
|
|
|
// Get the center between the two closest points; a better implementation would
|
|
// work with a confidentnes-value
|
|
if (RGdistance < GBdistance && RGdistance < BRdistance) {
|
|
brightestSpot = new PointInterpolator(brightestRedSpot, brightestGreenSpot);
|
|
} else if (GBdistance < RGdistance && GBdistance < BRdistance) {
|
|
brightestSpot = new PointInterpolator(brightestGreenSpot, brightestBlueSpot);
|
|
} else {
|
|
brightestSpot = new PointInterpolator(brightestBlueSpot, brightestRedSpot);
|
|
}
|
|
|
|
return brightestSpot.getCenter();
|
|
}
|
|
|
|
static public Point2D getBrightestSpot(BufferedImage image, byte gammaDelta) {
|
|
BufferedImage redImage = new BufferedImage(image.getWidth(), image.getHeight(), BufferedImage.TYPE_BYTE_GRAY);
|
|
BufferedImage greenImage = new BufferedImage(image.getWidth(), image.getHeight(), BufferedImage.TYPE_BYTE_GRAY);
|
|
BufferedImage blueImage = new BufferedImage(image.getWidth(), image.getHeight(), BufferedImage.TYPE_BYTE_GRAY);
|
|
|
|
for (int i = 0; i < image.getWidth(); i++) {
|
|
for (int j = 0; j < image.getHeight(); j++) {
|
|
int currentRGB = image.getRGB(i, j);
|
|
int currentBlue = currentRGB & 0xFF;
|
|
int currentGreen = (currentRGB >> 8) & 0xFF;
|
|
int currentRed = (currentRGB >> 16) & 0xFF;
|
|
|
|
redImage.setRGB(i, j, (currentRed << 16) | (currentRed << 8) | currentRed);
|
|
greenImage.setRGB(i, j, (currentGreen << 16) | (currentGreen << 8) | currentGreen);
|
|
blueImage.setRGB(i, j, (currentBlue << 16) | (currentBlue << 8) | currentBlue);
|
|
}
|
|
}
|
|
|
|
return getBrightestSpot(redImage, greenImage, blueImage, gammaDelta);
|
|
}
|
|
|
|
static public Point2D getBrightestSpot(BufferedImage image) {
|
|
return getBrightestSpot(image, GAMMA_DELTA_DEFAULT);
|
|
}
|
|
}
|