You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
3.3 KiB
116 lines
3.3 KiB
package com.sothr.imagetools.hash.util
|
|
|
|
import java.awt.image.{BufferedImage, ColorConvertOp, DataBufferByte}
|
|
|
|
import grizzled.slf4j.Logging
|
|
import net.coobird.thumbnailator.Thumbnails
|
|
|
|
object ImageUtil extends Logging {
|
|
|
|
/**
|
|
* Quickly convert an image to grayscale
|
|
*
|
|
* @param image image to convert to greyscale
|
|
* @return
|
|
*/
|
|
def convertToGray(image: BufferedImage): BufferedImage = {
|
|
val grayImage = new BufferedImage(image.getWidth, image.getHeight, BufferedImage.TYPE_BYTE_GRAY)
|
|
|
|
val op = new ColorConvertOp(
|
|
image.getColorModel.getColorSpace,
|
|
grayImage.getColorModel.getColorSpace,
|
|
null
|
|
)
|
|
|
|
op.filter(image, grayImage)
|
|
}
|
|
|
|
def resize(image: BufferedImage, size: Int, forced: Boolean = false): BufferedImage = {
|
|
//debug(s"Resizing an image to size: ${size}x${size} forced: $forced")
|
|
if (forced) {
|
|
Thumbnails.of(image).forceSize(size, size).asBufferedImage
|
|
} else {
|
|
Thumbnails.of(image).size(size, size).asBufferedImage
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get the raw data for an image
|
|
* Convert a buffered image into a 2d pixel data array
|
|
*
|
|
* @param image image to convert without using RGB
|
|
* @return
|
|
*/
|
|
def getImageData(image: BufferedImage): Array[Array[Int]] = {
|
|
|
|
val pixels = image.getRaster.getDataBuffer.asInstanceOf[DataBufferByte].getData
|
|
val numPixels = pixels.length
|
|
val width = image.getWidth
|
|
val height = image.getHeight
|
|
val isSingleChannel = if (numPixels == (width * height)) true else false
|
|
val hasAlphaChannel = image.getAlphaRaster != null
|
|
//debug(s"Converting image to 2d. width:$width height:$height")
|
|
|
|
val result = Array.ofDim[Int](height, width)
|
|
if (isSingleChannel) {
|
|
//debug(s"Processing Single Channel Image")
|
|
val pixelLength = 1
|
|
var row = 0
|
|
var col = 0
|
|
//debug(s"Processing pixels 0 until $numPixels by $pixelLength")
|
|
for (pixel <- 0 until numPixels by pixelLength) {
|
|
//debug(s"Processing pixel: $pixel/${numPixels - 1}")
|
|
val argb: Int = pixels(pixel).toInt //singleChannel
|
|
//debug(s"Pixel data: $argb")
|
|
result(row)(col) = argb
|
|
col += 1
|
|
if (col == width) {
|
|
col = 0
|
|
row += 1
|
|
}
|
|
}
|
|
}
|
|
else if (hasAlphaChannel) {
|
|
//debug(s"Processing Four Channel Image")
|
|
val pixelLength = 4
|
|
var row = 0
|
|
var col = 0
|
|
//debug(s"Processing pixels 0 until $numPixels by $pixelLength")
|
|
for (pixel <- 0 until numPixels by pixelLength) {
|
|
//debug(s"Processing pixel: $pixel/${numPixels - 1}")
|
|
var argb: Int = 0
|
|
argb += pixels(pixel).toInt << 24 //alpha
|
|
argb += pixels(pixel + 1).toInt //blue
|
|
argb += pixels(pixel + 2).toInt << 8 //green
|
|
argb += pixels(pixel + 3).toInt << 16 //red
|
|
result(row)(col) = argb
|
|
col += 1
|
|
if (col == width) {
|
|
col = 0
|
|
row += 1
|
|
}
|
|
}
|
|
} else {
|
|
//debug(s"Processing Three Channel Image")
|
|
val pixelLength = 3
|
|
var row = 0
|
|
var col = 0
|
|
//debug(s"Processing pixels 0 until $numPixels by $pixelLength")
|
|
for (pixel <- 0 until numPixels by pixelLength) {
|
|
//debug(s"Processing pixel: $pixel/${numPixels - 1}")
|
|
var argb: Int = 0
|
|
argb += -16777216; // 255 alpha
|
|
argb += pixels(pixel).toInt //blue
|
|
argb += pixels(pixel + 1).toInt << 8 //green
|
|
argb += pixels(pixel + 2).toInt << 16 //red
|
|
result(row)(col) = argb
|
|
col += 1
|
|
if (col == width) {
|
|
col = 0
|
|
row += 1
|
|
}
|
|
}
|
|
}
|
|
result
|
|
}
|
|
}
|