mirror of
https://github.com/dromara/tianai-captcha.git
synced 2026-05-09 15:13:04 +08:00
初始提交，滑块验证码功能已实现
This commit is contained in:
liushaofeng
+39
@@ -0,0 +1,39 @@
|
||||
package cloud.tianai.captcha.tianaicaptcha.template.slider;
|
||||
|
||||
import lombok.AllArgsConstructor;
|
||||
import lombok.Data;
|
||||
|
||||
/**
|
||||
* @Author: 天爱有情
|
||||
* @Date 2020/5/29 8:04
|
||||
* @Description 滑块验证码
|
||||
*/
|
||||
@Data
|
||||
@AllArgsConstructor
|
||||
public class SliderCaptchaInfo {
|
||||
|
||||
/**
|
||||
* x轴
|
||||
*/
|
||||
private Integer x;
|
||||
/**
|
||||
* x轴百分比
|
||||
*/
|
||||
private Float xPercent;
|
||||
/**
|
||||
* y轴
|
||||
*/
|
||||
private Integer y;
|
||||
/**
|
||||
* 背景图
|
||||
*/
|
||||
private String backgroundImage;
|
||||
/**
|
||||
* 移动图
|
||||
*/
|
||||
private String sliderImage;
|
||||
|
||||
public static SliderCaptchaInfo of(Integer x, Float xPercent, Integer y, String backgroundImage, String sliderImage) {
|
||||
return new SliderCaptchaInfo(x, xPercent, y, backgroundImage, sliderImage);
|
||||
}
|
||||
}
|
||||
+384
@@ -0,0 +1,384 @@
|
||||
package cloud.tianai.captcha.tianaicaptcha.template.slider;
|
||||
|
||||
import lombok.SneakyThrows;
|
||||
|
||||
import javax.imageio.ImageIO;
|
||||
import java.awt.*;
|
||||
import java.awt.geom.Area;
|
||||
import java.awt.image.BufferedImage;
|
||||
import java.awt.image.PixelGrabber;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.File;
|
||||
import java.math.BigDecimal;
|
||||
import java.net.URL;
|
||||
import java.util.List;
|
||||
import java.util.*;
|
||||
import java.util.concurrent.atomic.AtomicBoolean;
|
||||
|
||||
/**
|
||||
* @Author: 天爱有情
|
||||
* @Date 2020/5/29 8:06
|
||||
* @Description 滑块验证码模板
|
||||
*/
|
||||
public class SliderCaptchaTemplate {
|
||||
|
||||
/** 默认的resource资源文件路径.*/
|
||||
public static final String DEFAULT_SLIDER_IMAGE_RESOURCE_PATH = "META-INF/cut-image/resource";
|
||||
/** 默认的template资源文件路径.*/
|
||||
public static final String DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH = "META-INF/cut-image/template";
|
||||
|
||||
public static final String ACTIVE_IMAGE_NAME = "active.png";
|
||||
public static final String CUT_IMAGE_NAME = "cut.png";
|
||||
public static final String FIXED_IMAGE_NAME = "fixed.png";
|
||||
public static final String MATRIX_IMAGE_NAME = "matrix.png";
|
||||
|
||||
/** resource图片.*/
|
||||
private static List<URL> resourceImageFiles = new ArrayList<>(20);
|
||||
/** 模板图片.*/
|
||||
private static List<Map<String, URL>> templateImageFiles = new ArrayList<>(2);
|
||||
|
||||
static {
|
||||
// 添加一些系统的资源文件
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/1.jpg")));
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/2.jpg")));
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/3.jpg")));
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/4.jpg")));
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/5.jpg")));
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/6.jpg")));
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/7.jpg")));
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/8.jpg")));
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/9.jpg")));
|
||||
addResource(getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_RESOURCE_PATH.concat("/10.jpg")));
|
||||
|
||||
// 添加一些系统的 模板文件
|
||||
Map<String, URL> template1 = new HashMap<>(4);
|
||||
template1.put(ACTIVE_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/1/active.png")));
|
||||
template1.put(CUT_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/1/cut.png")));
|
||||
template1.put(FIXED_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/1/fixed.png")));
|
||||
template1.put(MATRIX_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/1/matrix.png")));
|
||||
addTemplate(template1);
|
||||
|
||||
|
||||
Map<String, URL> template2 = new HashMap<>(4);
|
||||
template2.put(ACTIVE_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/2/active.png")));
|
||||
template2.put(CUT_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/2/cut.png")));
|
||||
template2.put(FIXED_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/2/fixed.png")));
|
||||
template2.put(MATRIX_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/2/matrix.png")));
|
||||
addTemplate(template2);
|
||||
|
||||
Map<String, URL> template3 = new HashMap<>(4);
|
||||
template3.put(ACTIVE_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/3/active.png")));
|
||||
template3.put(CUT_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/3/cut.png")));
|
||||
template3.put(FIXED_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/3/fixed.png")));
|
||||
template3.put(MATRIX_IMAGE_NAME, getClassLoader().getResource(DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH.concat("/3/matrix.png")));
|
||||
addTemplate(template3);
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
private final AtomicBoolean loadResources = new AtomicBoolean(false);
|
||||
|
||||
private String sliderImageResourcePath = DEFAULT_SLIDER_IMAGE_RESOURCE_PATH;
|
||||
private String sliderImageTemplatePath = DEFAULT_SLIDER_IMAGE_TEMPLATE_PATH;
|
||||
|
||||
public SliderCaptchaTemplate() {
|
||||
// 加载系统资源文件
|
||||
}
|
||||
|
||||
|
||||
public SliderCaptchaTemplate(String sliderImageResourcePath, String sliderImageTemplatePath) {
|
||||
this.sliderImageResourcePath = sliderImageResourcePath;
|
||||
this.sliderImageTemplatePath = sliderImageTemplatePath;
|
||||
// 加载系统资源文件
|
||||
}
|
||||
|
||||
public SliderCaptchaTemplate(List<URL> r, List<Map<String, URL>> t) {
|
||||
resourceImageFiles = r;
|
||||
templateImageFiles = t;
|
||||
}
|
||||
|
||||
public static void addResource(URL url) {
|
||||
resourceImageFiles.remove(url);
|
||||
resourceImageFiles.add(url);
|
||||
}
|
||||
|
||||
public static void setResource(List<URL> resources) {
|
||||
resourceImageFiles = resources;
|
||||
}
|
||||
|
||||
public static void setTemplates(List<Map<String, URL>> imageTemplates) {
|
||||
templateImageFiles = imageTemplates;
|
||||
}
|
||||
|
||||
public static void deleteResource(URL resource) {
|
||||
resourceImageFiles.remove(resource);
|
||||
}
|
||||
|
||||
public static void deleteTemplate(Map<String, URL> template) {
|
||||
templateImageFiles.remove(template);
|
||||
}
|
||||
|
||||
public static void addTemplate(Map<String, URL> template) {
|
||||
templateImageFiles.remove(template);
|
||||
templateImageFiles.add(template);
|
||||
}
|
||||
|
||||
private static ClassLoader getClassLoader() {
|
||||
ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
|
||||
if (classLoader == null) {
|
||||
classLoader = SliderCaptchaTemplate.getClassLoader();
|
||||
}
|
||||
if (classLoader == null) {
|
||||
classLoader = ClassLoader.getSystemClassLoader();
|
||||
}
|
||||
return classLoader;
|
||||
}
|
||||
|
||||
public SliderCaptchaInfo getSlideImageInfo(){
|
||||
return getSlideImageInfo("jpg", "png");
|
||||
}
|
||||
|
||||
public SliderCaptchaInfo getSlideImageInfoForWebp(){
|
||||
return getSlideImageInfo("webp", "webp");
|
||||
}
|
||||
|
||||
|
||||
@SneakyThrows
|
||||
public SliderCaptchaInfo getSlideImageInfo(String targetFormatName, String matrixFormatName) {
|
||||
URL resourceImage = getRandomResourceImage();
|
||||
Map<String, URL> templateImages = getRandomTemplateImages();
|
||||
|
||||
BufferedImage cutBackground = warpFile2BufferedImage(resourceImage);
|
||||
BufferedImage targetBackground = warpFile2BufferedImage(resourceImage);
|
||||
|
||||
BufferedImage fixedTemplate = warpFile2BufferedImage(getTemplateFile(templateImages, FIXED_IMAGE_NAME));
|
||||
BufferedImage activeTemplate = warpFile2BufferedImage(getTemplateFile(templateImages, ACTIVE_IMAGE_NAME));
|
||||
BufferedImage matrixTemplate = warpFile2BufferedImage(getTemplateFile(templateImages, MATRIX_IMAGE_NAME));
|
||||
BufferedImage cutTemplate = warpFile2BufferedImage(getTemplateFile(templateImages, CUT_IMAGE_NAME));
|
||||
|
||||
// 获取随机的 x 和 y 轴
|
||||
Random random = new Random();
|
||||
int randomX = random.nextInt(targetBackground.getWidth() - fixedTemplate.getWidth() * 2) + fixedTemplate.getWidth();
|
||||
int randomY = random.nextInt(targetBackground.getHeight() - fixedTemplate.getHeight());
|
||||
|
||||
coverImage(targetBackground, fixedTemplate, randomX, randomY);
|
||||
BufferedImage cutImage = cutImage(cutBackground, cutTemplate, randomX, randomY);
|
||||
coverImage(cutImage, activeTemplate, 0, 0);
|
||||
coverImage(matrixTemplate, cutImage, 0, randomY);
|
||||
// 计算滑块百分比
|
||||
Float xPercent = (float) randomX / targetBackground.getWidth();
|
||||
|
||||
String backGroundImageBase64 = transformBase64(targetBackground, targetFormatName);
|
||||
String sliderImageBase64 = transformBase64(matrixTemplate, matrixFormatName);
|
||||
|
||||
return SliderCaptchaInfo.of(randomX, xPercent, randomY, backGroundImageBase64, sliderImageBase64);
|
||||
}
|
||||
|
||||
/**
|
||||
* 百分比对比
|
||||
*
|
||||
* @param newPercentage 用户百分比
|
||||
* @param oriPercentage 原百分比
|
||||
* @return true 成功 false 失败
|
||||
*/
|
||||
public boolean percentageContrast(Float newPercentage, Float oriPercentage) {
|
||||
boolean falg = false;
|
||||
BigDecimal num = BigDecimal.valueOf(0.05d).setScale(2, BigDecimal.ROUND_HALF_UP);
|
||||
BigDecimal newPercentageBig = new BigDecimal(newPercentage).setScale(2, BigDecimal.ROUND_HALF_UP);
|
||||
BigDecimal oriPercentageBig = new BigDecimal(oriPercentage).setScale(2, BigDecimal.ROUND_HALF_UP);
|
||||
//最小百分比
|
||||
BigDecimal minOriPercentage = oriPercentageBig.subtract(num).setScale(2, BigDecimal.ROUND_HALF_UP);
|
||||
//最大百分比
|
||||
BigDecimal maxOriPercentage = oriPercentageBig.add(num).setScale(2, BigDecimal.ROUND_HALF_UP);
|
||||
if (newPercentageBig.compareTo(minOriPercentage) > 0 && maxOriPercentage.compareTo(newPercentageBig) > 0) {
|
||||
falg = true;
|
||||
}
|
||||
return falg;
|
||||
}
|
||||
|
||||
|
||||
private String transformBase64(BufferedImage bufferedImage, String formatName) {
|
||||
byte[] data = null;
|
||||
try (ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream()) {
|
||||
ImageIO.write(bufferedImage, formatName, byteArrayOutputStream);
|
||||
//转换成字节码
|
||||
data = byteArrayOutputStream.toByteArray();
|
||||
} catch (Exception e) {
|
||||
System.out.println(e.getMessage());
|
||||
}
|
||||
String base64 = Base64.getEncoder().encodeToString(data);
|
||||
return "data:image/"+formatName+";base64,".concat(base64);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 通过模板图片抠图（不透明部分）
|
||||
*
|
||||
* @param origin 源图片
|
||||
* @param template 模板图片
|
||||
* @param x 坐标轴x
|
||||
* @param y 坐标轴y
|
||||
* @return BufferedImage
|
||||
*/
|
||||
@SneakyThrows
|
||||
private static BufferedImage cutImage(BufferedImage origin, BufferedImage template, int x, int y) {
|
||||
int bw = template.getWidth(null);
|
||||
int bh = template.getHeight(null);
|
||||
int lw = origin.getWidth(null);
|
||||
int lh = origin.getHeight(null);
|
||||
//得到透明的区域(人物轮廓)
|
||||
Shape imageShape = getImageShape(template, false);
|
||||
//合成后的图片
|
||||
BufferedImage image = new BufferedImage(bw, bh, BufferedImage.TYPE_INT_ARGB);
|
||||
Graphics2D graphics = image.createGraphics();
|
||||
//设置画布为透明
|
||||
image = graphics.getDeviceConfiguration().createCompatibleImage(bw, bh, Transparency.TRANSLUCENT);
|
||||
graphics.dispose();
|
||||
Graphics2D graphics2 = image.createGraphics();
|
||||
//取交集(限制可以画的范围为shape的范围)
|
||||
graphics2.clip(imageShape);
|
||||
//抗锯齿
|
||||
graphics2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
|
||||
graphics2.setStroke(new BasicStroke(5, BasicStroke.CAP_BUTT, BasicStroke.JOIN_BEVEL));
|
||||
graphics2.drawImage(origin, -x, -y, lw, lh, null);
|
||||
graphics2.dispose();
|
||||
return image;
|
||||
}
|
||||
|
||||
/**
|
||||
* 将Image图像中的透明/不透明部分转换为Shape图形
|
||||
*
|
||||
* @param img 图片信息
|
||||
* @param transparent 是否透明
|
||||
* @return Shape
|
||||
* @throws InterruptedException 异常
|
||||
*/
|
||||
public static Shape getImageShape(Image img, boolean transparent) throws InterruptedException {
|
||||
ArrayList<Integer> x = new ArrayList<>();
|
||||
ArrayList<Integer> y = new ArrayList<>();
|
||||
int width = img.getWidth(null);
|
||||
int height = img.getHeight(null);
|
||||
|
||||
// 首先获取图像所有的像素信息
|
||||
PixelGrabber pgr = new PixelGrabber(img, 0, 0, -1, -1, true);
|
||||
pgr.grabPixels();
|
||||
int[] pixels = (int[]) pgr.getPixels();
|
||||
|
||||
// 循环像素
|
||||
for (int i = 0; i < pixels.length; i++) {
|
||||
// 筛选，将不透明的像素的坐标加入到坐标ArrayList x和y中
|
||||
int alpha = (pixels[i] >> 24) & 0xff;
|
||||
if (alpha != 0) {
|
||||
x.add(i % width > 0 ? i % width - 1 : 0);
|
||||
y.add(i % width == 0 ? (i == 0 ? 0 : i / width - 1) : i / width);
|
||||
}
|
||||
}
|
||||
|
||||
// 建立图像矩阵并初始化(0为透明,1为不透明)
|
||||
int[][] matrix = new int[height][width];
|
||||
for (int i = 0; i < height; i++) {
|
||||
for (int j = 0; j < width; j++) {
|
||||
matrix[i][j] = 0;
|
||||
}
|
||||
}
|
||||
|
||||
// 导入坐标ArrayList中的不透明坐标信息
|
||||
for (int c = 0; c < x.size(); c++) {
|
||||
matrix[y.get(c)][x.get(c)] = 1;
|
||||
}
|
||||
|
||||
/*
|
||||
* 逐一水平"扫描"图像矩阵的每一行，将透明（这里也可以取不透明的）的像素生成为Rectangle，
|
||||
* 再将每一行的Rectangle通过Area类的rec对象进行合并， 最后形成一个完整的Shape图形
|
||||
*/
|
||||
Area rec = new Area();
|
||||
int temp = 0;
|
||||
//生成Shape时是1取透明区域还是取非透明区域的flag
|
||||
int flag = transparent ? 0 : 1;
|
||||
|
||||
for (int i = 0; i < height; i++) {
|
||||
for (int j = 0; j < width; j++) {
|
||||
if (matrix[i][j] == flag) {
|
||||
if (temp == 0) {
|
||||
temp = j;
|
||||
}
|
||||
if (j == width) {
|
||||
if (temp == 0) {
|
||||
rec.add(new Area(new Rectangle(j, i, 1, 1)));
|
||||
} else {
|
||||
rec.add(new Area(new Rectangle(temp, i, j - temp, 1)));
|
||||
temp = 0;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if (temp != 0) {
|
||||
rec.add(new Area(new Rectangle(temp, i, j - temp, 1)));
|
||||
temp = 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
temp = 0;
|
||||
}
|
||||
return rec;
|
||||
}
|
||||
|
||||
/**
|
||||
* 图片覆盖（覆盖图压缩到width*height大小，覆盖到底图上）
|
||||
*
|
||||
* @param baseBufferedImage 底图
|
||||
* @param coverBufferedImage 覆盖图
|
||||
* @param x 起始x轴
|
||||
* @param y 起始y轴
|
||||
*/
|
||||
private static void coverImage(BufferedImage baseBufferedImage, BufferedImage coverBufferedImage,
|
||||
int x, int y) {
|
||||
// 创建Graphics2D对象，用在底图对象上绘图
|
||||
Graphics2D g2d = baseBufferedImage.createGraphics();
|
||||
// 绘制
|
||||
g2d.drawImage(coverBufferedImage, x, y, coverBufferedImage.getWidth(), coverBufferedImage.getHeight(), null);
|
||||
// 释放图形上下文使用的系统资源
|
||||
g2d.dispose();
|
||||
}
|
||||
|
||||
private URL getTemplateFile(Map<String, URL> templateImages, String imageName) {
|
||||
URL url = templateImages.get(imageName);
|
||||
if (url == null) {
|
||||
throw new IllegalArgumentException("查找模板异常， 该模板下未找到 ");
|
||||
}
|
||||
return url;
|
||||
}
|
||||
|
||||
private Map<String, URL> getRandomTemplateImages() {
|
||||
if (templateImageFiles.size() == 1) {
|
||||
return templateImageFiles.get(0);
|
||||
}
|
||||
int templateNo = new Random().nextInt(templateImageFiles.size());
|
||||
return templateImageFiles.get(templateNo);
|
||||
}
|
||||
|
||||
@SneakyThrows
|
||||
private static BufferedImage warpFile2BufferedImage(URL resourceImage) {
|
||||
if (resourceImage == null) {
|
||||
throw new IllegalArgumentException("包装文件到 BufferedImage 失败， file不能为空");
|
||||
}
|
||||
return ImageIO.read(resourceImage);
|
||||
}
|
||||
|
||||
private URL getRandomResourceImage() {
|
||||
int targetNo = new Random().nextInt(resourceImageFiles.size());
|
||||
return resourceImageFiles.get(targetNo);
|
||||
}
|
||||
|
||||
public static void main(String[] args) {
|
||||
SliderCaptchaTemplate sliderCaptchaTemplate = new SliderCaptchaTemplate();
|
||||
// 生成滑块图片
|
||||
SliderCaptchaInfo slideImageInfo = sliderCaptchaTemplate.getSlideImageInfo();
|
||||
// 获取背景图片的base64
|
||||
String backgroundImage = slideImageInfo.getBackgroundImage();
|
||||
// 获取滑块图片
|
||||
slideImageInfo.getSliderImage();
|
||||
// 获取滑块被背景图片的百分比， (校验图片使用)
|
||||
Float xPercent = slideImageInfo.getXPercent();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,38 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
enum VP8StatusCode {
|
||||
VP8_STATUS_OK,
|
||||
VP8_STATUS_OUT_OF_MEMORY,
|
||||
VP8_STATUS_INVALID_PARAM,
|
||||
VP8_STATUS_BITSTREAM_ERROR,
|
||||
VP8_STATUS_UNSUPPORTED_FEATURE,
|
||||
VP8_STATUS_SUSPENDED,
|
||||
VP8_STATUS_USER_ABORT,
|
||||
VP8_STATUS_NOT_ENOUGH_DATA,;
|
||||
|
||||
private static VP8StatusCode[] VALUES = values();
|
||||
|
||||
public static VP8StatusCode getStatusCode( int aValue ) {
|
||||
if ( aValue >= 0 && aValue < VALUES.length ) {
|
||||
return VALUES[ aValue ];
|
||||
}
|
||||
else {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,99 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.scijava.nativelib.DefaultJniExtractor;
|
||||
import org.scijava.nativelib.NativeLibraryUtil;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.nio.ByteOrder;
|
||||
|
||||
@Slf4j
|
||||
final class WebP {
|
||||
private static boolean NATIVE_LIBRARY_LOADED = false;
|
||||
|
||||
static synchronized void loadNativeLibrary() {
|
||||
if (!NATIVE_LIBRARY_LOADED) {
|
||||
NATIVE_LIBRARY_LOADED = true;
|
||||
try {
|
||||
NativeLibraryUtil.loadNativeLibrary(new DefaultJniExtractor(WebP.class), "webp-imageio");
|
||||
} catch (IOException e) {
|
||||
log.debug("IOException creating DefaultJniExtractor", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static {
|
||||
loadNativeLibrary();
|
||||
}
|
||||
|
||||
private WebP() {
|
||||
}
|
||||
|
||||
public static int[] decode(WebPDecoderOptions aOptions, byte[] aData, int aOffset, int aLength, int[] aOut) throws IOException {
|
||||
if (aOptions == null) {
|
||||
throw new NullPointerException("Decoder options may not be null");
|
||||
}
|
||||
|
||||
if (aData == null) {
|
||||
throw new NullPointerException("Input data may not be null");
|
||||
}
|
||||
|
||||
if (aOffset + aLength > aData.length) {
|
||||
throw new IllegalArgumentException("Offset/length exceeds array size");
|
||||
}
|
||||
|
||||
int[] pixels = decode(aOptions.fPointer, aData, aOffset, aLength, aOut, ByteOrder.nativeOrder().equals(ByteOrder.BIG_ENDIAN));
|
||||
VP8StatusCode status = VP8StatusCode.getStatusCode(aOut[0]);
|
||||
switch (status) {
|
||||
case VP8_STATUS_OK:
|
||||
break;
|
||||
case VP8_STATUS_OUT_OF_MEMORY:
|
||||
throw new OutOfMemoryError();
|
||||
default:
|
||||
throw new IOException("Decode returned code " + status);
|
||||
}
|
||||
|
||||
return pixels;
|
||||
}
|
||||
|
||||
private static native int[] decode(long aDecoderOptionsPointer, byte[] aData, int aOffset, int aLength, int[] aFlags, boolean aBigEndian);
|
||||
|
||||
public static int[] getInfo(byte[] aData, int aOffset, int aLength) throws IOException {
|
||||
int[] out = new int[2];
|
||||
int result = getInfo(aData, aOffset, aLength, out);
|
||||
if (result == 0) {
|
||||
throw new IOException("Invalid WebP data");
|
||||
}
|
||||
|
||||
return out;
|
||||
}
|
||||
|
||||
private static native int getInfo(byte[] aData, int aOffset, int aLength, int[] aOut);
|
||||
|
||||
public static byte[] encodeRGBA(WebPEncoderOptions aOptions, byte[] aRgbaData, int aWidth, int aHeight, int aStride) {
|
||||
return encodeRGBA(aOptions.fPointer, aRgbaData, aWidth, aHeight, aStride);
|
||||
}
|
||||
|
||||
private static native byte[] encodeRGBA(long aConfig, byte[] aRgbaData, int aWidth, int aHeight, int aStride);
|
||||
|
||||
public static byte[] encodeRGB(WebPEncoderOptions aOptions, byte[] aRgbaData, int aWidth, int aHeight, int aStride) {
|
||||
return encodeRGB(aOptions.fPointer, aRgbaData, aWidth, aHeight, aStride);
|
||||
}
|
||||
|
||||
private static native byte[] encodeRGB(long aConfig, byte[] aRgbaData, int aWidth, int aHeight, int aStride);
|
||||
}
|
||||
@@ -0,0 +1,182 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
public final class WebPDecoderOptions {
|
||||
static {
|
||||
WebP.loadNativeLibrary();
|
||||
}
|
||||
|
||||
long fPointer;
|
||||
|
||||
public WebPDecoderOptions() {
|
||||
fPointer = createDecoderOptions();
|
||||
if ( fPointer == 0 ) {
|
||||
throw new OutOfMemoryError();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void finalize() throws Throwable {
|
||||
super.finalize();
|
||||
deleteDecoderOptions( fPointer );
|
||||
fPointer = 0L;
|
||||
}
|
||||
|
||||
public int getCropHeight() {
|
||||
return getCropHeight( fPointer );
|
||||
}
|
||||
|
||||
public void setCropHeight( int aCropHeight ) {
|
||||
setCropHeight( fPointer, aCropHeight );
|
||||
}
|
||||
|
||||
public int getCropLeft() {
|
||||
return getCropLeft( fPointer );
|
||||
}
|
||||
|
||||
public void setCropLeft( int aCropLeft ) {
|
||||
setCropLeft( fPointer, aCropLeft );
|
||||
}
|
||||
|
||||
public int getCropTop() {
|
||||
return getCropTop( fPointer );
|
||||
}
|
||||
|
||||
public void setCropTop( int aCropTop ) {
|
||||
setCropTop( fPointer, aCropTop );
|
||||
}
|
||||
|
||||
public int getCropWidth() {
|
||||
return getCropWidth( fPointer );
|
||||
}
|
||||
|
||||
public void setCropWidth( int aCropWidth ) {
|
||||
setCropWidth( fPointer, aCropWidth );
|
||||
}
|
||||
|
||||
public boolean isFancyUpsampling() {
|
||||
return !isNoFancyUpsampling( fPointer );
|
||||
}
|
||||
|
||||
public void setFancyUpsampling( boolean aFancyUpsampling ) {
|
||||
setNoFancyUpsampling( fPointer, !aFancyUpsampling );
|
||||
}
|
||||
|
||||
public int getScaledHeight() {
|
||||
return getScaledHeight( fPointer );
|
||||
}
|
||||
|
||||
public void setScaledHeight( int aScaledHeight ) {
|
||||
setScaledHeight( fPointer, aScaledHeight );
|
||||
}
|
||||
|
||||
public int getScaledWidth() {
|
||||
return getScaledWidth( fPointer );
|
||||
}
|
||||
|
||||
public void setScaledWidth( int aScaledWidth ) {
|
||||
setScaledWidth( fPointer, aScaledWidth );
|
||||
}
|
||||
|
||||
public boolean isUseCropping() {
|
||||
return isUseCropping( fPointer );
|
||||
}
|
||||
|
||||
public void setUseCropping( boolean aUseCropping ) {
|
||||
setUseCropping( fPointer, aUseCropping );
|
||||
}
|
||||
|
||||
public boolean isUseScaling() {
|
||||
return isUseScaling( fPointer );
|
||||
}
|
||||
|
||||
public void setUseScaling( boolean aUseScaling ) {
|
||||
setUseScaling( fPointer, aUseScaling );
|
||||
}
|
||||
|
||||
public boolean isUseThreads() {
|
||||
return isUseThreads( fPointer );
|
||||
}
|
||||
|
||||
public void setUseThreads( boolean aUseThreads ) {
|
||||
setUseThreads( fPointer, aUseThreads );
|
||||
}
|
||||
|
||||
public boolean isBypassFiltering() {
|
||||
return isBypassFiltering( fPointer );
|
||||
}
|
||||
|
||||
public void setBypassFiltering( boolean aBypassFiltering ) {
|
||||
setBypassFiltering( fPointer, aBypassFiltering );
|
||||
}
|
||||
|
||||
private static native long createDecoderOptions();
|
||||
|
||||
private static native void deleteDecoderOptions( long aPointer );
|
||||
|
||||
private static native int getCropHeight( long aPointer );
|
||||
|
||||
private static native void setCropHeight( long aPointer, int aCropHeight );
|
||||
|
||||
private static native int getCropLeft( long aPointer );
|
||||
|
||||
private static native void setCropLeft( long aPointer, int aCropLeft );
|
||||
|
||||
private static native int getCropTop( long aPointer );
|
||||
|
||||
private static native void setCropTop( long aPointer, int aCropTop );
|
||||
|
||||
private static native int getCropWidth( long aPointer );
|
||||
|
||||
private static native void setCropWidth( long aPointer, int aCropWidth );
|
||||
|
||||
private static native boolean isForceRotation( long aPointer );
|
||||
|
||||
private static native void setForceRotation( long aPointer, boolean aForceRotation );
|
||||
|
||||
private static native boolean isNoEnhancement( long aPointer );
|
||||
|
||||
private static native void setNoEnhancement( long aPointer, boolean aNoEnhancement );
|
||||
|
||||
private static native boolean isNoFancyUpsampling( long aPointer );
|
||||
|
||||
private static native void setNoFancyUpsampling( long aPointer, boolean aFancyUpsampling );
|
||||
|
||||
private static native int getScaledHeight( long aPointer );
|
||||
|
||||
private static native void setScaledHeight( long aPointer, int aScaledHeight );
|
||||
|
||||
private static native int getScaledWidth( long aPointer );
|
||||
|
||||
private static native void setScaledWidth( long aPointer, int aScaledWidth );
|
||||
|
||||
private static native boolean isUseCropping( long aPointer );
|
||||
|
||||
private static native void setUseCropping( long aPointer, boolean aUseCropping );
|
||||
|
||||
private static native boolean isUseScaling( long aPointer );
|
||||
|
||||
private static native void setUseScaling( long aPointer, boolean aUseScaling );
|
||||
|
||||
private static native boolean isUseThreads( long aPointer );
|
||||
|
||||
private static native void setUseThreads( long aPointer, boolean aUseThreads );
|
||||
|
||||
private static native boolean isBypassFiltering( long aPointer );
|
||||
|
||||
private static native void setBypassFiltering( long aPointer, boolean aBypassFiltering );
|
||||
}
|
||||
@@ -0,0 +1,310 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
public class WebPEncoderOptions {
|
||||
static {
|
||||
WebP.loadNativeLibrary();
|
||||
}
|
||||
|
||||
long fPointer;
|
||||
|
||||
public WebPEncoderOptions() {
|
||||
fPointer = createConfig();
|
||||
if ( fPointer == 0 ) {
|
||||
throw new OutOfMemoryError();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void finalize() throws Throwable {
|
||||
super.finalize();
|
||||
deleteConfig( fPointer );
|
||||
fPointer = 0L;
|
||||
}
|
||||
|
||||
private static native long createConfig();
|
||||
|
||||
private static native void deleteConfig( long aPointer );
|
||||
|
||||
long getPointer() {
|
||||
return fPointer;
|
||||
}
|
||||
|
||||
public float getCompressionQuality() {
|
||||
return getQuality(fPointer);
|
||||
}
|
||||
|
||||
public void setCompressionQuality( float quality ) {
|
||||
setQuality( fPointer, quality );
|
||||
}
|
||||
|
||||
public boolean isLossless() {
|
||||
return getLossless( fPointer ) != 0;
|
||||
}
|
||||
|
||||
public void setLossless( boolean aLossless ) {
|
||||
setLossless(fPointer, aLossless ? 1 : 0);
|
||||
}
|
||||
|
||||
public int getTargetSize() {
|
||||
return getTargetSize( fPointer );
|
||||
}
|
||||
|
||||
public void setTargetSize( int aTargetSize ) {
|
||||
setTargetSize( fPointer, aTargetSize );
|
||||
}
|
||||
|
||||
public float getTargetPSNR() {
|
||||
return getTargetPSNR( fPointer );
|
||||
}
|
||||
|
||||
public void setTargetPSNR( float aTargetPSNR ) {
|
||||
setTargetPSNR( fPointer, aTargetPSNR );
|
||||
}
|
||||
|
||||
public int getMethod() {
|
||||
return getMethod( fPointer );
|
||||
}
|
||||
|
||||
public void setMethod( int aMethod ) {
|
||||
setMethod( fPointer, aMethod );
|
||||
}
|
||||
|
||||
public int getSegments() {
|
||||
return getSegments( fPointer );
|
||||
}
|
||||
|
||||
public void setSegments( int aSegments ) {
|
||||
setSegments( fPointer, aSegments );
|
||||
}
|
||||
|
||||
public int getSnsStrength() {
|
||||
return getSnsStrength( fPointer );
|
||||
}
|
||||
|
||||
public void setSnsStrength( int aSnsStrength ) {
|
||||
setSnsStrength( fPointer, aSnsStrength );
|
||||
}
|
||||
|
||||
public int getFilterStrength() {
|
||||
return getFilterStrength( fPointer );
|
||||
}
|
||||
|
||||
public void setFilterStrength( int aFilterStrength ) {
|
||||
setFilterStrength( fPointer, aFilterStrength );
|
||||
}
|
||||
|
||||
public int getFilterSharpness() {
|
||||
return getFilterSharpness( fPointer );
|
||||
}
|
||||
|
||||
public void setFilterSharpness( int aFilterSharpness ) {
|
||||
setFilterSharpness( fPointer, aFilterSharpness );
|
||||
}
|
||||
|
||||
public int getFilterType() {
|
||||
return getFilterType( fPointer );
|
||||
}
|
||||
|
||||
public void setFilterType( int aFilterType ) {
|
||||
setFilterType( fPointer, aFilterType );
|
||||
}
|
||||
|
||||
public boolean isAutoAdjustFilterStrength() {
|
||||
return getAutofilter( fPointer ) != 0;
|
||||
}
|
||||
|
||||
public void setAutoAdjustFilterStrength( boolean aAutofilter ) {
|
||||
setAutofilter( fPointer, aAutofilter ? 1 : 0 );
|
||||
}
|
||||
|
||||
public int getEntropyAnalysisPassCount() {
|
||||
return getPass( fPointer );
|
||||
}
|
||||
|
||||
public void setEntropyAnalysisPassCount( int aPass ) {
|
||||
setPass( fPointer, aPass );
|
||||
}
|
||||
|
||||
public boolean isShowCompressed() {
|
||||
return getShowCompressed( fPointer ) != 0;
|
||||
}
|
||||
|
||||
public void setShowCompressed( boolean aShowCompressed ) {
|
||||
setShowCompressed( fPointer, aShowCompressed ? 1 : 0 );
|
||||
}
|
||||
|
||||
public int getPreprocessing() {
|
||||
return getPreprocessing( fPointer );
|
||||
}
|
||||
|
||||
public void setPreprocessing( int aPreprocessing ) {
|
||||
setPreprocessing( fPointer, aPreprocessing );
|
||||
}
|
||||
|
||||
public int getPartitions() {
|
||||
return getPartitions( fPointer );
|
||||
}
|
||||
|
||||
public void setPartitions( int aPartitions ) {
|
||||
setPartitions( fPointer, aPartitions );
|
||||
}
|
||||
|
||||
public int getPartitionLimit() {
|
||||
return getPartitionLimit( fPointer );
|
||||
}
|
||||
|
||||
public void setPartitionLimit( int aPartitionLimit ) {
|
||||
setPartitionLimit( fPointer, aPartitionLimit );
|
||||
}
|
||||
|
||||
public int getAlphaCompression() {
|
||||
return getAlphaCompression( fPointer );
|
||||
}
|
||||
|
||||
public void setAlphaCompression( int aAlphaCompression ) {
|
||||
setAlphaCompression( fPointer, aAlphaCompression );
|
||||
}
|
||||
|
||||
public int getAlphaFiltering() {
|
||||
return getAlphaFiltering( fPointer );
|
||||
}
|
||||
|
||||
public void setAlphaFiltering( int aAlphaFiltering ) {
|
||||
setAlphaFiltering( fPointer, aAlphaFiltering );
|
||||
}
|
||||
|
||||
public int getAlphaQuality() {
|
||||
return getAlphaQuality( fPointer );
|
||||
}
|
||||
|
||||
public void setAlphaQuality( int aAlphaQuality ) {
|
||||
setAlphaQuality( fPointer, aAlphaQuality );
|
||||
}
|
||||
|
||||
public boolean isEmulateJpegSize() {
|
||||
return getEmulateJpegSize( fPointer ) != 0;
|
||||
}
|
||||
|
||||
public void setEmulateJpegSize( boolean aEmulateJpegSize ) {
|
||||
setEmulateJpegSize( fPointer, aEmulateJpegSize ? 1 : 0 );
|
||||
}
|
||||
|
||||
public int getThreadLevel() {
|
||||
return getThreadLevel( fPointer );
|
||||
}
|
||||
|
||||
public void setThreadLevel( int aThreadLevel ) {
|
||||
setThreadLevel( fPointer, aThreadLevel );
|
||||
}
|
||||
|
||||
public boolean isReduceMemoryUsage() {
|
||||
return getLowMemory( fPointer ) != 0;
|
||||
}
|
||||
|
||||
public void setReduceMemoryUsage( boolean aLowMemory ) {
|
||||
setLowMemory( fPointer, aLowMemory ? 1 : 0 );
|
||||
}
|
||||
|
||||
private static native float getQuality( long aPointer );
|
||||
|
||||
private static native void setQuality( long aPointer, float aQuality );
|
||||
|
||||
private static native int getTargetSize( long aPointer );
|
||||
|
||||
private static native void setTargetSize( long aPointer, int aTargetSize );
|
||||
|
||||
private static native float getTargetPSNR( long aPointer );
|
||||
|
||||
private static native void setTargetPSNR( long aPointer, float aTargetPSNR );
|
||||
|
||||
private static native int getMethod( long aPointer );
|
||||
|
||||
private static native void setMethod( long aPointer, int aMethod );
|
||||
|
||||
private static native int getSegments( long aPointer );
|
||||
|
||||
private static native void setSegments( long aPointer, int aSegments );
|
||||
|
||||
private static native int getSnsStrength( long aPointer );
|
||||
|
||||
private static native void setSnsStrength( long aPointer, int aSnsStrength );
|
||||
|
||||
private static native int getFilterStrength( long aPointer );
|
||||
|
||||
private static native void setFilterStrength( long aPointer, int aFilterStrength );
|
||||
|
||||
private static native int getFilterSharpness( long aPointer );
|
||||
|
||||
private static native void setFilterSharpness( long aPointer, int aFilterSharpness );
|
||||
|
||||
private static native int getFilterType( long aPointer );
|
||||
|
||||
private static native void setFilterType( long aPointer, int aFilterType );
|
||||
|
||||
private static native int getAutofilter( long aPointer );
|
||||
|
||||
private static native void setAutofilter( long aPointer, int aAutofilter );
|
||||
|
||||
private static native int getPass( long aPointer );
|
||||
|
||||
private static native void setPass( long aPointer, int aPass );
|
||||
|
||||
private static native int getShowCompressed( long aPointer );
|
||||
|
||||
private static native void setShowCompressed( long aPointer, int aShowCompressed );
|
||||
|
||||
private static native int getPreprocessing( long aPointer );
|
||||
|
||||
private static native void setPreprocessing( long aPointer, int aPreprocessing );
|
||||
|
||||
private static native int getPartitions( long aPointer );
|
||||
|
||||
private static native void setPartitions( long aPointer, int aPartitions );
|
||||
|
||||
private static native int getPartitionLimit( long aPointer );
|
||||
|
||||
private static native void setPartitionLimit( long aPointer, int aPartitionLimit );
|
||||
|
||||
private static native int getAlphaCompression( long aPointer );
|
||||
|
||||
private static native void setAlphaCompression( long aPointer, int aAlphaCompression );
|
||||
|
||||
private static native int getAlphaFiltering( long aPointer );
|
||||
|
||||
private static native void setAlphaFiltering( long aPointer, int aAlphaFiltering );
|
||||
|
||||
private static native int getAlphaQuality( long aPointer );
|
||||
|
||||
private static native void setAlphaQuality( long aPointer, int aAlphaQuality );
|
||||
|
||||
private static native int getLossless( long aPointer );
|
||||
|
||||
private static native void setLossless( long aPointer, int aLossless );
|
||||
|
||||
private static native int getEmulateJpegSize( long aPointer );
|
||||
|
||||
private static native void setEmulateJpegSize( long aPointer, int aEmulateJpegSize );
|
||||
|
||||
private static native int getThreadLevel( long aPointer );
|
||||
|
||||
private static native void setThreadLevel( long aPointer, int aThreadLevel );
|
||||
|
||||
private static native int getLowMemory( long aPointer );
|
||||
|
||||
private static native void setLowMemory( long aPointer, int aLowMemory );
|
||||
}
|
||||
@@ -0,0 +1,107 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
import javax.imageio.ImageReader;
|
||||
import javax.imageio.spi.ImageReaderSpi;
|
||||
import javax.imageio.stream.ImageInputStream;
|
||||
import java.io.IOException;
|
||||
import java.nio.ByteOrder;
|
||||
import java.util.Arrays;
|
||||
import java.util.Locale;
|
||||
|
||||
/**
|
||||
*
|
||||
*/
|
||||
public class WebPImageReaderSpi extends ImageReaderSpi {
|
||||
private static final byte[] RIFF = new byte[]{ 'R', 'I', 'F', 'F' };
|
||||
private static final byte[] WEBP = new byte[]{ 'W', 'E', 'B', 'P' };
|
||||
private static final byte[] VP8_ = new byte[]{ 'V', 'P', '8', ' ' };
|
||||
private static final byte[] VP8L = new byte[]{ 'V', 'P', '8', 'L' };
|
||||
private static final byte[] VP8X = new byte[]{ 'V', 'P', '8', 'X' };
|
||||
|
||||
public WebPImageReaderSpi() {
|
||||
super(
|
||||
"Luciad",
|
||||
"1.0",
|
||||
new String[]{ "WebP", "webp" },
|
||||
new String[]{ "webp" },
|
||||
new String[]{ "image/webp" },
|
||||
WebPReader.class.getName(),
|
||||
new Class[] { ImageInputStream.class },
|
||||
new String[]{ WebPImageWriterSpi.class.getName() },
|
||||
false,
|
||||
null,
|
||||
null,
|
||||
null,
|
||||
null,
|
||||
false,
|
||||
null,
|
||||
null,
|
||||
null,
|
||||
null
|
||||
);
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageReader createReaderInstance( Object extension ) throws IOException {
|
||||
return new WebPReader( this );
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean canDecodeInput( Object source ) throws IOException {
|
||||
if ( !( source instanceof ImageInputStream ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
ImageInputStream stream = ( ImageInputStream ) source;
|
||||
byte[] b = new byte[ 4 ];
|
||||
ByteOrder oldByteOrder = stream.getByteOrder();
|
||||
stream.mark();
|
||||
stream.setByteOrder( ByteOrder.LITTLE_ENDIAN );
|
||||
|
||||
try {
|
||||
stream.readFully( b );
|
||||
if ( !Arrays.equals( b, RIFF ) ) {
|
||||
return false;
|
||||
}
|
||||
long chunkLength = stream.readUnsignedInt();
|
||||
long streamLength = stream.length();
|
||||
if ( streamLength != -1 && streamLength != chunkLength + 8 ) {
|
||||
return false;
|
||||
}
|
||||
stream.readFully( b );
|
||||
if ( !Arrays.equals( b, WEBP ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
stream.readFully( b );
|
||||
if ( !Arrays.equals( b, VP8_ ) && !Arrays.equals( b, VP8L ) && !Arrays.equals( b, VP8X ) ) {
|
||||
return false;
|
||||
}
|
||||
} finally {
|
||||
stream.setByteOrder( oldByteOrder );
|
||||
stream.reset();
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getDescription( Locale locale ) {
|
||||
return "WebP Reader";
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,110 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
import javax.imageio.ImageTypeSpecifier;
|
||||
import javax.imageio.ImageWriter;
|
||||
import javax.imageio.spi.ImageWriterSpi;
|
||||
import javax.imageio.stream.ImageOutputStream;
|
||||
import java.awt.color.ColorSpace;
|
||||
import java.awt.image.ColorModel;
|
||||
import java.awt.image.ComponentColorModel;
|
||||
import java.awt.image.ComponentSampleModel;
|
||||
import java.awt.image.DataBuffer;
|
||||
import java.awt.image.DirectColorModel;
|
||||
import java.awt.image.SampleModel;
|
||||
import java.awt.image.SinglePixelPackedSampleModel;
|
||||
import java.io.IOException;
|
||||
import java.util.Locale;
|
||||
|
||||
/**
|
||||
*
|
||||
*/
|
||||
public class WebPImageWriterSpi extends ImageWriterSpi {
|
||||
public WebPImageWriterSpi() {
|
||||
super(
|
||||
"Luciad",
|
||||
"1.0",
|
||||
new String[]{ "WebP", "webp" },
|
||||
new String[]{ "webp" },
|
||||
new String[]{ "image/webp" },
|
||||
WebPReader.class.getName(),
|
||||
new Class[]{ ImageOutputStream.class },
|
||||
new String[]{ WebPImageReaderSpi.class.getName() },
|
||||
false,
|
||||
null,
|
||||
null,
|
||||
null,
|
||||
null,
|
||||
false,
|
||||
null,
|
||||
null,
|
||||
null,
|
||||
null
|
||||
);
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean canEncodeImage( ImageTypeSpecifier type ) {
|
||||
ColorModel colorModel = type.getColorModel();
|
||||
SampleModel sampleModel = type.getSampleModel();
|
||||
int transferType = sampleModel.getTransferType();
|
||||
|
||||
if ( colorModel instanceof ComponentColorModel ) {
|
||||
if ( !( sampleModel instanceof ComponentSampleModel ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if ( transferType != DataBuffer.TYPE_BYTE && transferType != DataBuffer.TYPE_INT ) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else if ( colorModel instanceof DirectColorModel ) {
|
||||
if ( !( sampleModel instanceof SinglePixelPackedSampleModel ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if ( transferType != DataBuffer.TYPE_INT ) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
ColorSpace colorSpace = colorModel.getColorSpace();
|
||||
if ( !( colorSpace.isCS_sRGB() ) ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
int[] sampleSize = sampleModel.getSampleSize();
|
||||
for ( int i = 0; i < sampleSize.length; i++ ) {
|
||||
if ( sampleSize[ i ] > 8 ) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageWriter createWriterInstance( Object extension ) throws IOException {
|
||||
return new WebPWriter( this );
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getDescription( Locale locale ) {
|
||||
return "WebP Writer";
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,118 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
import javax.imageio.ImageReadParam;
|
||||
|
||||
public final class WebPReadParam extends ImageReadParam {
|
||||
private WebPDecoderOptions fOptions;
|
||||
|
||||
public WebPReadParam() {
|
||||
fOptions = new WebPDecoderOptions();
|
||||
}
|
||||
|
||||
public void setScaledHeight(int aScaledHeight) {
|
||||
fOptions.setScaledHeight(aScaledHeight);
|
||||
}
|
||||
|
||||
public void setUseScaling(boolean aUseScaling) {
|
||||
fOptions.setUseScaling(aUseScaling);
|
||||
}
|
||||
|
||||
public void setUseThreads(boolean aUseThreads) {
|
||||
fOptions.setUseThreads(aUseThreads);
|
||||
}
|
||||
|
||||
public int getCropHeight() {
|
||||
return fOptions.getCropHeight();
|
||||
}
|
||||
|
||||
public int getScaledWidth() {
|
||||
return fOptions.getScaledWidth();
|
||||
}
|
||||
|
||||
public boolean isUseCropping() {
|
||||
return fOptions.isUseCropping();
|
||||
}
|
||||
|
||||
public void setCropWidth(int aCropWidth) {
|
||||
fOptions.setCropWidth(aCropWidth);
|
||||
}
|
||||
|
||||
public boolean isBypassFiltering() {
|
||||
return fOptions.isBypassFiltering();
|
||||
}
|
||||
|
||||
public int getCropLeft() {
|
||||
return fOptions.getCropLeft();
|
||||
}
|
||||
|
||||
public int getCropWidth() {
|
||||
return fOptions.getCropWidth();
|
||||
}
|
||||
|
||||
public int getScaledHeight() {
|
||||
return fOptions.getScaledHeight();
|
||||
}
|
||||
|
||||
public void setBypassFiltering(boolean aBypassFiltering) {
|
||||
fOptions.setBypassFiltering(aBypassFiltering);
|
||||
}
|
||||
|
||||
public void setUseCropping(boolean aUseCropping) {
|
||||
fOptions.setUseCropping(aUseCropping);
|
||||
}
|
||||
|
||||
public void setCropHeight(int aCropHeight) {
|
||||
fOptions.setCropHeight(aCropHeight);
|
||||
}
|
||||
|
||||
public void setFancyUpsampling(boolean aFancyUpsampling) {
|
||||
fOptions.setFancyUpsampling(aFancyUpsampling);
|
||||
}
|
||||
|
||||
public boolean isUseThreads() {
|
||||
return fOptions.isUseThreads();
|
||||
}
|
||||
|
||||
public boolean isFancyUpsampling() {
|
||||
return fOptions.isFancyUpsampling();
|
||||
}
|
||||
|
||||
public boolean isUseScaling() {
|
||||
return fOptions.isUseScaling();
|
||||
}
|
||||
|
||||
public void setCropLeft(int aCropLeft) {
|
||||
fOptions.setCropLeft(aCropLeft);
|
||||
}
|
||||
|
||||
public int getCropTop() {
|
||||
return fOptions.getCropTop();
|
||||
}
|
||||
|
||||
public void setScaledWidth(int aScaledWidth) {
|
||||
fOptions.setScaledWidth(aScaledWidth);
|
||||
}
|
||||
|
||||
public void setCropTop(int aCropTop) {
|
||||
fOptions.setCropTop(aCropTop);
|
||||
}
|
||||
|
||||
WebPDecoderOptions getDecoderOptions() {
|
||||
return fOptions;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,177 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
import javax.imageio.ImageReadParam;
|
||||
import javax.imageio.ImageReader;
|
||||
import javax.imageio.ImageTypeSpecifier;
|
||||
import javax.imageio.metadata.IIOMetadata;
|
||||
import javax.imageio.spi.ImageReaderSpi;
|
||||
import javax.imageio.stream.ImageInputStream;
|
||||
import java.awt.image.BufferedImage;
|
||||
import java.awt.image.ColorModel;
|
||||
import java.awt.image.DataBufferInt;
|
||||
import java.awt.image.DirectColorModel;
|
||||
import java.awt.image.SampleModel;
|
||||
import java.awt.image.WritableRaster;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.nio.ByteOrder;
|
||||
import java.util.Collections;
|
||||
import java.util.Hashtable;
|
||||
import java.util.Iterator;
|
||||
|
||||
class WebPReader extends ImageReader {
|
||||
private byte[] fData;
|
||||
private int fWidth;
|
||||
private int fHeight;
|
||||
|
||||
WebPReader( ImageReaderSpi originatingProvider ) {
|
||||
super( originatingProvider );
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setInput( Object input, boolean seekForwardOnly, boolean ignoreMetadata ) {
|
||||
super.setInput( input, seekForwardOnly, ignoreMetadata );
|
||||
fData = null;
|
||||
fWidth = -1;
|
||||
fHeight = -1;
|
||||
}
|
||||
|
||||
@Override
|
||||
public int getNumImages( boolean allowSearch ) throws IOException {
|
||||
return 1;
|
||||
}
|
||||
|
||||
private void readHeader() throws IOException {
|
||||
if ( fWidth != -1 && fHeight != -1 ) {
|
||||
return;
|
||||
}
|
||||
|
||||
readData();
|
||||
int[] info = WebP.getInfo( fData, 0, fData.length );
|
||||
fWidth = info[ 0 ];
|
||||
fHeight = info[ 1 ];
|
||||
}
|
||||
|
||||
private void readData() throws IOException {
|
||||
if ( fData != null ) {
|
||||
return;
|
||||
}
|
||||
|
||||
ImageInputStream input = ( ImageInputStream ) getInput();
|
||||
long length = input.length();
|
||||
if ( length > Integer.MAX_VALUE ) {
|
||||
throw new IOException( "Cannot read image of size " + length );
|
||||
}
|
||||
|
||||
if ( input.getStreamPosition() != 0L ) {
|
||||
if ( isSeekForwardOnly() ) {
|
||||
throw new IOException();
|
||||
}
|
||||
else {
|
||||
input.seek( 0 );
|
||||
}
|
||||
}
|
||||
|
||||
byte[] data;
|
||||
if ( length > 0 ) {
|
||||
data = new byte[ ( int ) length ];
|
||||
input.readFully( data );
|
||||
}
|
||||
else {
|
||||
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
||||
byte[] buffer = new byte[ 4096 ];
|
||||
int bytesRead;
|
||||
while ( ( bytesRead = input.read( buffer ) ) != -1 ) {
|
||||
out.write( buffer, 0, bytesRead );
|
||||
}
|
||||
out.close();
|
||||
data = out.toByteArray();
|
||||
}
|
||||
fData = data;
|
||||
}
|
||||
|
||||
private void checkIndex( int imageIndex ) {
|
||||
if ( imageIndex != 0 ) {
|
||||
throw new IndexOutOfBoundsException( "Invalid image index: " + imageIndex );
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public int getWidth( int imageIndex ) throws IOException {
|
||||
checkIndex( imageIndex );
|
||||
readHeader();
|
||||
return fWidth;
|
||||
}
|
||||
|
||||
@Override
|
||||
public int getHeight( int imageIndex ) throws IOException {
|
||||
checkIndex( imageIndex );
|
||||
readHeader();
|
||||
return fHeight;
|
||||
}
|
||||
|
||||
@Override
|
||||
public IIOMetadata getStreamMetadata() throws IOException {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public IIOMetadata getImageMetadata( int imageIndex ) throws IOException {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Iterator<ImageTypeSpecifier> getImageTypes( int imageIndex ) throws IOException {
|
||||
return Collections.singletonList(
|
||||
ImageTypeSpecifier.createFromBufferedImageType( BufferedImage.TYPE_INT_ARGB )
|
||||
).iterator();
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageReadParam getDefaultReadParam() {
|
||||
return new WebPReadParam();
|
||||
}
|
||||
|
||||
@Override
|
||||
public BufferedImage read( int imageIndex, ImageReadParam param ) throws IOException {
|
||||
checkIndex( imageIndex );
|
||||
readData();
|
||||
readHeader();
|
||||
WebPReadParam readParam = param != null ? (WebPReadParam) param : new WebPReadParam();
|
||||
|
||||
int[] outParams = new int[4];
|
||||
int[] pixels = WebP.decode(readParam.getDecoderOptions(), fData, 0, fData.length, outParams);
|
||||
|
||||
int width = outParams[1];
|
||||
int height = outParams[2];
|
||||
boolean alpha = outParams[3] != 0;
|
||||
|
||||
ColorModel colorModel;
|
||||
if ( alpha ) {
|
||||
colorModel = new DirectColorModel( 32, 0x00ff0000, 0x0000ff00, 0x000000ff, 0xff000000 );
|
||||
} else {
|
||||
colorModel = new DirectColorModel( 24, 0x00ff0000, 0x0000ff00, 0x000000ff, 0x00000000 );
|
||||
}
|
||||
|
||||
SampleModel sampleModel = colorModel.createCompatibleSampleModel( width, height );
|
||||
DataBufferInt db = new DataBufferInt( pixels, width * height );
|
||||
WritableRaster raster = WritableRaster.createWritableRaster(sampleModel, db, null);
|
||||
|
||||
return new BufferedImage( colorModel, raster, false, new Hashtable<Object, Object>() );
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,234 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
import javax.imageio.ImageWriteParam;
|
||||
import java.util.Locale;
|
||||
|
||||
public class WebPWriteParam extends ImageWriteParam {
|
||||
public static final int LOSSY_COMPRESSION = 0;
|
||||
public static final int LOSSLESS_COMPRESSION = 1;
|
||||
|
||||
private final boolean fDefaultLossless;
|
||||
private WebPEncoderOptions fOptions;
|
||||
|
||||
public WebPWriteParam( Locale aLocale ) {
|
||||
super( aLocale );
|
||||
fOptions = new WebPEncoderOptions();
|
||||
fDefaultLossless = fOptions.isLossless();
|
||||
canWriteCompressed = true;
|
||||
compressionTypes = new String[]{
|
||||
"Lossy",
|
||||
"Lossless"
|
||||
};
|
||||
compressionType = compressionTypes[fDefaultLossless ? LOSSLESS_COMPRESSION : LOSSY_COMPRESSION];
|
||||
compressionQuality = fOptions.getCompressionQuality() / 100f;
|
||||
compressionMode = MODE_EXPLICIT;
|
||||
}
|
||||
|
||||
@Override
|
||||
public float getCompressionQuality() {
|
||||
return super.getCompressionQuality();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setCompressionQuality( float quality ) {
|
||||
super.setCompressionQuality( quality );
|
||||
fOptions.setCompressionQuality( quality * 100f );
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setCompressionType( String compressionType ) {
|
||||
super.setCompressionType( compressionType );
|
||||
for ( int i = 0; i < compressionTypes.length; i++ ) {
|
||||
if ( compressionTypes[i].equals( compressionType ) ) {
|
||||
fOptions.setLossless( i == LOSSLESS_COMPRESSION );
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void unsetCompression() {
|
||||
super.unsetCompression();
|
||||
fOptions.setLossless( fDefaultLossless );
|
||||
}
|
||||
|
||||
public void setSnsStrength(int aSnsStrength) {
|
||||
fOptions.setSnsStrength(aSnsStrength);
|
||||
}
|
||||
|
||||
public void setAlphaQuality(int aAlphaQuality) {
|
||||
fOptions.setAlphaQuality(aAlphaQuality);
|
||||
}
|
||||
|
||||
public int getSegments() {
|
||||
return fOptions.getSegments();
|
||||
}
|
||||
|
||||
public int getPreprocessing() {
|
||||
return fOptions.getPreprocessing();
|
||||
}
|
||||
|
||||
public int getFilterStrength() {
|
||||
return fOptions.getFilterStrength();
|
||||
}
|
||||
|
||||
public void setEmulateJpegSize(boolean aEmulateJpegSize) {
|
||||
fOptions.setEmulateJpegSize(aEmulateJpegSize);
|
||||
}
|
||||
|
||||
public int getPartitions() {
|
||||
return fOptions.getPartitions();
|
||||
}
|
||||
|
||||
public void setTargetPSNR(float aTargetPSNR) {
|
||||
fOptions.setTargetPSNR(aTargetPSNR);
|
||||
}
|
||||
|
||||
public int getEntropyAnalysisPassCount() {
|
||||
return fOptions.getEntropyAnalysisPassCount();
|
||||
}
|
||||
|
||||
public int getPartitionLimit() {
|
||||
return fOptions.getPartitionLimit();
|
||||
}
|
||||
|
||||
public int getFilterType() {
|
||||
return fOptions.getFilterType();
|
||||
}
|
||||
|
||||
public int getFilterSharpness() {
|
||||
return fOptions.getFilterSharpness();
|
||||
}
|
||||
|
||||
public int getAlphaQuality() {
|
||||
return fOptions.getAlphaQuality();
|
||||
}
|
||||
|
||||
public boolean isShowCompressed() {
|
||||
return fOptions.isShowCompressed();
|
||||
}
|
||||
|
||||
public boolean isReduceMemoryUsage() {
|
||||
return fOptions.isReduceMemoryUsage();
|
||||
}
|
||||
|
||||
public void setThreadLevel(int aThreadLevel) {
|
||||
fOptions.setThreadLevel(aThreadLevel);
|
||||
}
|
||||
|
||||
public boolean isAutoAdjustFilterStrength() {
|
||||
return fOptions.isAutoAdjustFilterStrength();
|
||||
}
|
||||
|
||||
public void setReduceMemoryUsage(boolean aLowMemory) {
|
||||
fOptions.setReduceMemoryUsage(aLowMemory);
|
||||
}
|
||||
|
||||
public void setFilterStrength(int aFilterStrength) {
|
||||
fOptions.setFilterStrength(aFilterStrength);
|
||||
}
|
||||
|
||||
public int getTargetSize() {
|
||||
return fOptions.getTargetSize();
|
||||
}
|
||||
|
||||
public void setEntropyAnalysisPassCount(int aPass) {
|
||||
fOptions.setEntropyAnalysisPassCount(aPass);
|
||||
}
|
||||
|
||||
public void setFilterSharpness(int aFilterSharpness) {
|
||||
fOptions.setFilterSharpness(aFilterSharpness);
|
||||
}
|
||||
|
||||
public int getAlphaFiltering() {
|
||||
return fOptions.getAlphaFiltering();
|
||||
}
|
||||
|
||||
public int getSnsStrength() {
|
||||
return fOptions.getSnsStrength();
|
||||
}
|
||||
|
||||
public void setPartitionLimit(int aPartitionLimit) {
|
||||
fOptions.setPartitionLimit(aPartitionLimit);
|
||||
}
|
||||
|
||||
public void setMethod(int aMethod) {
|
||||
fOptions.setMethod(aMethod);
|
||||
}
|
||||
|
||||
public void setAlphaFiltering(int aAlphaFiltering) {
|
||||
fOptions.setAlphaFiltering(aAlphaFiltering);
|
||||
}
|
||||
|
||||
public int getMethod() {
|
||||
return fOptions.getMethod();
|
||||
}
|
||||
|
||||
public void setFilterType(int aFilterType) {
|
||||
fOptions.setFilterType(aFilterType);
|
||||
}
|
||||
|
||||
public void setPartitions(int aPartitions) {
|
||||
fOptions.setPartitions(aPartitions);
|
||||
}
|
||||
|
||||
public void setAutoAdjustFilterStrength(boolean aAutofilter) {
|
||||
fOptions.setAutoAdjustFilterStrength(aAutofilter);
|
||||
}
|
||||
|
||||
public boolean isEmulateJpegSize() {
|
||||
return fOptions.isEmulateJpegSize();
|
||||
}
|
||||
|
||||
public int getAlphaCompression() {
|
||||
return fOptions.getAlphaCompression();
|
||||
}
|
||||
|
||||
public void setShowCompressed(boolean aShowCompressed) {
|
||||
fOptions.setShowCompressed(aShowCompressed);
|
||||
}
|
||||
|
||||
public void setSegments(int aSegments) {
|
||||
fOptions.setSegments(aSegments);
|
||||
}
|
||||
|
||||
public float getTargetPSNR() {
|
||||
return fOptions.getTargetPSNR();
|
||||
}
|
||||
|
||||
public int getThreadLevel() {
|
||||
return fOptions.getThreadLevel();
|
||||
}
|
||||
|
||||
public void setTargetSize(int aTargetSize) {
|
||||
fOptions.setTargetSize(aTargetSize);
|
||||
}
|
||||
|
||||
public void setAlphaCompression(int aAlphaCompression) {
|
||||
fOptions.setAlphaCompression(aAlphaCompression);
|
||||
}
|
||||
|
||||
public void setPreprocessing(int aPreprocessing) {
|
||||
fOptions.setPreprocessing(aPreprocessing);
|
||||
}
|
||||
|
||||
WebPEncoderOptions getEncoderOptions() {
|
||||
return fOptions;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,372 @@
|
||||
/*
|
||||
* Copyright 2013 Luciad (http://www.luciad.com)
|
||||
*
|
||||
* 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 com.luciad.imageio.webp;
|
||||
|
||||
import javax.imageio.IIOImage;
|
||||
import javax.imageio.ImageTypeSpecifier;
|
||||
import javax.imageio.ImageWriteParam;
|
||||
import javax.imageio.ImageWriter;
|
||||
import javax.imageio.metadata.IIOMetadata;
|
||||
import javax.imageio.spi.ImageWriterSpi;
|
||||
import javax.imageio.stream.ImageOutputStream;
|
||||
|
||||
import java.awt.*;
|
||||
import java.awt.geom.AffineTransform;
|
||||
import java.awt.image.*;
|
||||
import java.io.IOException;
|
||||
|
||||
class WebPWriter extends ImageWriter {
|
||||
WebPWriter(ImageWriterSpi originatingProvider) {
|
||||
super(originatingProvider);
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageWriteParam getDefaultWriteParam() {
|
||||
return new WebPWriteParam(getLocale());
|
||||
}
|
||||
|
||||
@Override
|
||||
public IIOMetadata convertImageMetadata(IIOMetadata inData, ImageTypeSpecifier imageType, ImageWriteParam param) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public IIOMetadata convertStreamMetadata(IIOMetadata inData, ImageWriteParam param) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public IIOMetadata getDefaultImageMetadata(ImageTypeSpecifier imageType, ImageWriteParam param) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public IIOMetadata getDefaultStreamMetadata(ImageWriteParam param) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void write(IIOMetadata streamMetadata, IIOImage image, ImageWriteParam param) throws IOException {
|
||||
if (param == null) {
|
||||
param = getDefaultWriteParam();
|
||||
}
|
||||
|
||||
WebPWriteParam writeParam = (WebPWriteParam) param;
|
||||
|
||||
ImageOutputStream output = (ImageOutputStream) getOutput();
|
||||
RenderedImage ri = image.getRenderedImage();
|
||||
|
||||
byte[] encodedData = encode(writeParam.getEncoderOptions(), ri);
|
||||
output.write(encodedData);
|
||||
}
|
||||
|
||||
private static byte[] encode(WebPEncoderOptions aOptions, RenderedImage aImage) throws IOException
|
||||
{
|
||||
if (aOptions == null) {
|
||||
throw new NullPointerException("Encoder options may not be null");
|
||||
}
|
||||
|
||||
if (aImage == null) {
|
||||
throw new NullPointerException("Image may not be null");
|
||||
}
|
||||
|
||||
boolean encodeAlpha = hasTranslucency(aImage);
|
||||
if (encodeAlpha) {
|
||||
byte[] rgbaData = getRGBA(aImage);
|
||||
return WebP.encodeRGBA(aOptions, rgbaData, aImage.getWidth(), aImage.getHeight(), aImage.getWidth() * 4);
|
||||
} else {
|
||||
byte[] rgbData = getRGB(aImage);
|
||||
return WebP.encodeRGB(aOptions, rgbData, aImage.getWidth(), aImage.getHeight(), aImage.getWidth() * 3);
|
||||
}
|
||||
}
|
||||
|
||||
private static boolean hasTranslucency(RenderedImage aRi) {
|
||||
return aRi.getColorModel().hasAlpha();
|
||||
}
|
||||
|
||||
private static int getShift(int aMask) {
|
||||
int shift = 0;
|
||||
while (((aMask >> shift) & 0x1) == 0) {
|
||||
shift++;
|
||||
}
|
||||
return shift;
|
||||
}
|
||||
|
||||
private static byte[] getRGB(RenderedImage aRi) throws IOException {
|
||||
int width = aRi.getWidth();
|
||||
int height = aRi.getHeight();
|
||||
|
||||
ColorModel colorModel = aRi.getColorModel();
|
||||
if (colorModel instanceof ComponentColorModel) {
|
||||
ComponentSampleModel sampleModel = (ComponentSampleModel) aRi.getSampleModel();
|
||||
int type = sampleModel.getTransferType();
|
||||
if (type == DataBuffer.TYPE_BYTE) {
|
||||
return extractComponentRGBByte(width, height, sampleModel, ((DataBufferByte) aRi.getData().getDataBuffer()));
|
||||
} else if (type == DataBuffer.TYPE_INT) {
|
||||
return extractComponentRGBInt(width, height, sampleModel, ((DataBufferInt) aRi.getData().getDataBuffer()));
|
||||
} else {
|
||||
throw new IOException("Incompatible image: " + aRi);
|
||||
}
|
||||
} else if (colorModel instanceof DirectColorModel) {
|
||||
SinglePixelPackedSampleModel sampleModel = (SinglePixelPackedSampleModel) aRi.getSampleModel();
|
||||
int type = sampleModel.getTransferType();
|
||||
if (type == DataBuffer.TYPE_INT) {
|
||||
return extractDirectRGBInt(width, height, (DirectColorModel) colorModel, sampleModel, ((DataBufferInt) aRi.getData().getDataBuffer()));
|
||||
} else {
|
||||
throw new IOException("Incompatible image: " + aRi);
|
||||
}
|
||||
} else {
|
||||
BufferedImage i = new BufferedImage(aRi.getWidth(), aRi.getHeight(), BufferedImage.TYPE_INT_RGB);
|
||||
Graphics2D g = i.createGraphics();
|
||||
g.drawRenderedImage(aRi, new AffineTransform());
|
||||
g.dispose();
|
||||
return getRGB(i);
|
||||
}
|
||||
}
|
||||
|
||||
private static byte[] extractDirectRGBInt(int aWidth, int aHeight, DirectColorModel aColorModel, SinglePixelPackedSampleModel aSampleModel, DataBufferInt aDataBuffer) {
|
||||
byte[] out = new byte[aWidth * aHeight * 3];
|
||||
|
||||
int rMask = aColorModel.getRedMask();
|
||||
int gMask = aColorModel.getGreenMask();
|
||||
int bMask = aColorModel.getBlueMask();
|
||||
int rShift = getShift(rMask);
|
||||
int gShift = getShift(gMask);
|
||||
int bShift = getShift(bMask);
|
||||
int[] bank = aDataBuffer.getBankData()[0];
|
||||
int scanlineStride = aSampleModel.getScanlineStride();
|
||||
int scanIx = 0;
|
||||
for (int b = 0, y = 0; y < aHeight; y++) {
|
||||
int pixIx = scanIx;
|
||||
for (int x = 0; x < aWidth; x++, b += 3) {
|
||||
int pixel = bank[pixIx++];
|
||||
out[b] = (byte) ((pixel & rMask) >>> rShift);
|
||||
out[b + 1] = (byte) ((pixel & gMask) >>> gShift);
|
||||
out[b + 2] = (byte) ((pixel & bMask) >>> bShift);
|
||||
}
|
||||
scanIx += scanlineStride;
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
private static byte[] extractComponentRGBInt(int aWidth, int aHeight, ComponentSampleModel aSampleModel, DataBufferInt aDataBuffer) {
|
||||
byte[] out = new byte[aWidth * aHeight * 3];
|
||||
|
||||
int[] bankIndices = aSampleModel.getBankIndices();
|
||||
int[] rBank = aDataBuffer.getBankData()[bankIndices[0]];
|
||||
int[] gBank = aDataBuffer.getBankData()[bankIndices[1]];
|
||||
int[] bBank = aDataBuffer.getBankData()[bankIndices[2]];
|
||||
|
||||
int[] bankOffsets = aSampleModel.getBandOffsets();
|
||||
int rScanIx = bankOffsets[0];
|
||||
int gScanIx = bankOffsets[1];
|
||||
int bScanIx = bankOffsets[2];
|
||||
|
||||
int pixelStride = aSampleModel.getPixelStride();
|
||||
int scanlineStride = aSampleModel.getScanlineStride();
|
||||
for (int b = 0, y = 0; y < aHeight; y++) {
|
||||
int rPixIx = rScanIx;
|
||||
int gPixIx = gScanIx;
|
||||
int bPixIx = bScanIx;
|
||||
for (int x = 0; x < aWidth; x++, b += 3) {
|
||||
out[b] = (byte) rBank[rPixIx];
|
||||
rPixIx += pixelStride;
|
||||
out[b + 1] = (byte) gBank[gPixIx];
|
||||
gPixIx += pixelStride;
|
||||
out[b + 2] = (byte) bBank[bPixIx];
|
||||
bPixIx += pixelStride;
|
||||
}
|
||||
rScanIx += scanlineStride;
|
||||
gScanIx += scanlineStride;
|
||||
bScanIx += scanlineStride;
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
private static byte[] extractComponentRGBByte(int aWidth, int aHeight, ComponentSampleModel aSampleModel, DataBufferByte aDataBuffer) {
|
||||
byte[] out = new byte[aWidth * aHeight * 3];
|
||||
|
||||
int[] bankIndices = aSampleModel.getBankIndices();
|
||||
byte[] rBank = aDataBuffer.getBankData()[bankIndices[0]];
|
||||
byte[] gBank = aDataBuffer.getBankData()[bankIndices[1]];
|
||||
byte[] bBank = aDataBuffer.getBankData()[bankIndices[2]];
|
||||
|
||||
int[] bankOffsets = aSampleModel.getBandOffsets();
|
||||
int rScanIx = bankOffsets[0];
|
||||
int gScanIx = bankOffsets[1];
|
||||
int bScanIx = bankOffsets[2];
|
||||
|
||||
int pixelStride = aSampleModel.getPixelStride();
|
||||
int scanlineStride = aSampleModel.getScanlineStride();
|
||||
for (int b = 0, y = 0; y < aHeight; y++) {
|
||||
int rPixIx = rScanIx;
|
||||
int gPixIx = gScanIx;
|
||||
int bPixIx = bScanIx;
|
||||
for (int x = 0; x < aWidth; x++, b += 3) {
|
||||
out[b] = rBank[rPixIx];
|
||||
rPixIx += pixelStride;
|
||||
out[b + 1] = gBank[gPixIx];
|
||||
gPixIx += pixelStride;
|
||||
out[b + 2] = bBank[bPixIx];
|
||||
bPixIx += pixelStride;
|
||||
}
|
||||
rScanIx += scanlineStride;
|
||||
gScanIx += scanlineStride;
|
||||
bScanIx += scanlineStride;
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
private static byte[] getRGBA(RenderedImage aRi) throws IOException {
|
||||
int width = aRi.getWidth();
|
||||
int height = aRi.getHeight();
|
||||
|
||||
ColorModel colorModel = aRi.getColorModel();
|
||||
if (colorModel instanceof ComponentColorModel) {
|
||||
ComponentSampleModel sampleModel = (ComponentSampleModel) aRi.getSampleModel();
|
||||
int type = sampleModel.getTransferType();
|
||||
if (type == DataBuffer.TYPE_BYTE) {
|
||||
return extractComponentRGBAByte(width, height, sampleModel, ((DataBufferByte) aRi.getData().getDataBuffer()));
|
||||
} else if (type == DataBuffer.TYPE_INT) {
|
||||
return extractComponentRGBAInt(width, height, sampleModel, ((DataBufferInt) aRi.getData().getDataBuffer()));
|
||||
} else {
|
||||
throw new IOException("Incompatible image: " + aRi);
|
||||
}
|
||||
} else if (colorModel instanceof DirectColorModel) {
|
||||
SinglePixelPackedSampleModel sampleModel = (SinglePixelPackedSampleModel) aRi.getSampleModel();
|
||||
int type = sampleModel.getTransferType();
|
||||
if (type == DataBuffer.TYPE_INT) {
|
||||
return extractDirectRGBAInt(width, height, (DirectColorModel) colorModel, sampleModel, ((DataBufferInt) aRi.getData().getDataBuffer()));
|
||||
} else {
|
||||
throw new IOException("Incompatible image: " + aRi);
|
||||
}
|
||||
} else {
|
||||
BufferedImage i = new BufferedImage(aRi.getWidth(), aRi.getHeight(), BufferedImage.TYPE_INT_ARGB);
|
||||
Graphics2D g = i.createGraphics();
|
||||
g.drawRenderedImage(aRi, new AffineTransform());
|
||||
g.dispose();
|
||||
return getRGBA(i);
|
||||
}
|
||||
}
|
||||
|
||||
private static byte[] extractDirectRGBAInt(int aWidth, int aHeight, DirectColorModel aColorModel, SinglePixelPackedSampleModel aSampleModel, DataBufferInt aDataBuffer) {
|
||||
byte[] out = new byte[aWidth * aHeight * 4];
|
||||
|
||||
int rMask = aColorModel.getRedMask();
|
||||
int gMask = aColorModel.getGreenMask();
|
||||
int bMask = aColorModel.getBlueMask();
|
||||
int aMask = aColorModel.getAlphaMask();
|
||||
int rShift = getShift(rMask);
|
||||
int gShift = getShift(gMask);
|
||||
int bShift = getShift(bMask);
|
||||
int aShift = getShift(aMask);
|
||||
int[] bank = aDataBuffer.getBankData()[0];
|
||||
int scanlineStride = aSampleModel.getScanlineStride();
|
||||
int scanIx = 0;
|
||||
for (int b = 0, y = 0; y < aHeight; y++) {
|
||||
int pixIx = scanIx;
|
||||
for (int x = 0; x < aWidth; x++, b += 4) {
|
||||
int pixel = bank[pixIx++];
|
||||
out[b] = (byte) ((pixel & rMask) >>> rShift);
|
||||
out[b + 1] = (byte) ((pixel & gMask) >>> gShift);
|
||||
out[b + 2] = (byte) ((pixel & bMask) >>> bShift);
|
||||
out[b + 3] = (byte) ((pixel & aMask) >>> aShift);
|
||||
}
|
||||
scanIx += scanlineStride;
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
private static byte[] extractComponentRGBAInt(int aWidth, int aHeight, ComponentSampleModel aSampleModel, DataBufferInt aDataBuffer) {
|
||||
byte[] out = new byte[aWidth * aHeight * 4];
|
||||
|
||||
int[] bankIndices = aSampleModel.getBankIndices();
|
||||
int[] rBank = aDataBuffer.getBankData()[bankIndices[0]];
|
||||
int[] gBank = aDataBuffer.getBankData()[bankIndices[1]];
|
||||
int[] bBank = aDataBuffer.getBankData()[bankIndices[2]];
|
||||
int[] aBank = aDataBuffer.getBankData()[bankIndices[3]];
|
||||
|
||||
int[] bankOffsets = aSampleModel.getBandOffsets();
|
||||
int rScanIx = bankOffsets[0];
|
||||
int gScanIx = bankOffsets[1];
|
||||
int bScanIx = bankOffsets[2];
|
||||
int aScanIx = bankOffsets[3];
|
||||
|
||||
int pixelStride = aSampleModel.getPixelStride();
|
||||
int scanlineStride = aSampleModel.getScanlineStride();
|
||||
for (int b = 0, y = 0; y < aHeight; y++) {
|
||||
int rPixIx = rScanIx;
|
||||
int gPixIx = gScanIx;
|
||||
int bPixIx = bScanIx;
|
||||
int aPixIx = aScanIx;
|
||||
for (int x = 0; x < aWidth; x++, b += 4) {
|
||||
out[b] = (byte) rBank[rPixIx];
|
||||
rPixIx += pixelStride;
|
||||
out[b + 1] = (byte) gBank[gPixIx];
|
||||
gPixIx += pixelStride;
|
||||
out[b + 2] = (byte) bBank[bPixIx];
|
||||
bPixIx += pixelStride;
|
||||
out[b + 3] = (byte) aBank[aPixIx];
|
||||
aPixIx += pixelStride;
|
||||
}
|
||||
rScanIx += scanlineStride;
|
||||
gScanIx += scanlineStride;
|
||||
bScanIx += scanlineStride;
|
||||
aScanIx += scanlineStride;
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
private static byte[] extractComponentRGBAByte(int aWidth, int aHeight, ComponentSampleModel aSampleModel, DataBufferByte aDataBuffer) {
|
||||
byte[] out = new byte[aWidth * aHeight * 4];
|
||||
|
||||
int[] bankIndices = aSampleModel.getBankIndices();
|
||||
byte[] rBank = aDataBuffer.getBankData()[bankIndices[0]];
|
||||
byte[] gBank = aDataBuffer.getBankData()[bankIndices[1]];
|
||||
byte[] bBank = aDataBuffer.getBankData()[bankIndices[2]];
|
||||
byte[] aBank = aDataBuffer.getBankData()[bankIndices[3]];
|
||||
|
||||
int[] bankOffsets = aSampleModel.getBandOffsets();
|
||||
int rScanIx = bankOffsets[0];
|
||||
int gScanIx = bankOffsets[1];
|
||||
int bScanIx = bankOffsets[2];
|
||||
int aScanIx = bankOffsets[3];
|
||||
|
||||
int pixelStride = aSampleModel.getPixelStride();
|
||||
int scanlineStride = aSampleModel.getScanlineStride();
|
||||
for (int b = 0, y = 0; y < aHeight; y++) {
|
||||
int rPixIx = rScanIx;
|
||||
int gPixIx = gScanIx;
|
||||
int bPixIx = bScanIx;
|
||||
int aPixIx = aScanIx;
|
||||
for (int x = 0; x < aWidth; x++, b += 4) {
|
||||
out[b] = rBank[rPixIx];
|
||||
rPixIx += pixelStride;
|
||||
out[b + 1] = gBank[gPixIx];
|
||||
gPixIx += pixelStride;
|
||||
out[b + 2] = bBank[bPixIx];
|
||||
bPixIx += pixelStride;
|
||||
out[b + 3] = aBank[aPixIx];
|
||||
aPixIx += pixelStride;
|
||||
}
|
||||
rScanIx += scanlineStride;
|
||||
gScanIx += scanlineStride;
|
||||
bScanIx += scanlineStride;
|
||||
aScanIx += scanlineStride;
|
||||
}
|
||||
return out;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,37 @@
|
||||
package example;
|
||||
|
||||
import com.luciad.imageio.webp.WebPReadParam;
|
||||
|
||||
import javax.imageio.ImageIO;
|
||||
import javax.imageio.ImageReader;
|
||||
import javax.imageio.stream.FileImageInputStream;
|
||||
import java.awt.image.BufferedImage;
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
|
||||
public class DecodeTest {
|
||||
public static void main(String args[]) throws IOException {
|
||||
String inputWebpPath = "test_pic/test.webp";
|
||||
String outputJpgPath = "test_pic/test_.jpg";
|
||||
String outputJpegPath = "test_pic/test_.jpeg";
|
||||
String outputPngPath = "test_pic/test_.png";
|
||||
|
||||
// Obtain a WebP ImageReader instance
|
||||
ImageReader reader = ImageIO.getImageReadersByMIMEType("image/webp").next();
|
||||
|
||||
// Configure decoding parameters
|
||||
WebPReadParam readParam = new WebPReadParam();
|
||||
readParam.setBypassFiltering(true);
|
||||
|
||||
// Configure the input on the ImageReader
|
||||
reader.setInput(new FileImageInputStream(new File(inputWebpPath)));
|
||||
|
||||
// Decode the image
|
||||
BufferedImage image = reader.read(0, readParam);
|
||||
|
||||
ImageIO.write(image, "png", new File(outputPngPath));
|
||||
ImageIO.write(image, "jpg", new File(outputJpgPath));
|
||||
ImageIO.write(image, "jpeg", new File(outputJpegPath));
|
||||
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,37 @@
|
||||
package example;
|
||||
|
||||
import com.luciad.imageio.webp.WebPWriteParam;
|
||||
|
||||
import javax.imageio.IIOImage;
|
||||
import javax.imageio.ImageIO;
|
||||
import javax.imageio.ImageWriter;
|
||||
import javax.imageio.stream.FileImageOutputStream;
|
||||
import java.awt.image.BufferedImage;
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
|
||||
public class EncodeTest {
|
||||
public static void main(String args[]) throws IOException {
|
||||
String inputPngPath = "test_pic/test.png";
|
||||
String inputJpgPath = "test_pic/test.jpg";
|
||||
String outputWebpPath = "test_pic/test_.webp";
|
||||
|
||||
// Obtain an image to encode from somewhere
|
||||
BufferedImage image = ImageIO.read(new File(inputJpgPath));
|
||||
|
||||
// Obtain a WebP ImageWriter instance
|
||||
ImageWriter writer = ImageIO.getImageWritersByMIMEType("image/webp").next();
|
||||
|
||||
// Configure encoding parameters
|
||||
WebPWriteParam writeParam = new WebPWriteParam(writer.getLocale());
|
||||
writeParam.setCompressionMode(WebPWriteParam.MODE_DEFAULT);
|
||||
|
||||
// Configure the output on the ImageWriter
|
||||
writer.setOutput(new FileImageOutputStream(new File(outputWebpPath)));
|
||||
|
||||
// Encode
|
||||
long st = System.currentTimeMillis();
|
||||
writer.write(null, new IIOImage(image, null, null), writeParam);
|
||||
System.out.println("cost: " + (System.currentTimeMillis() - st));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user