/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.
*/
/* $Id: ImageIOJPEGImageWriter.java 750418 2009-03-05 11:03:54Z vhennebert $ */
package org.apache.xmlgraphics.image.writer.imageio;
import java.awt.image.RenderedImage;
import javax.imageio.ImageWriteParam;
import javax.imageio.ImageWriter;
import javax.imageio.metadata.IIOInvalidTreeException;
import javax.imageio.metadata.IIOMetadata;
import javax.imageio.metadata.IIOMetadataNode;
import javax.imageio.plugins.jpeg.JPEGImageWriteParam;
import org.apache.xmlgraphics.image.writer.ImageWriterParams;
/**
* ImageWriter that encodes JPEG images using Image I/O.
*
* @version $Id: ImageIOJPEGImageWriter.java 750418 2009-03-05 11:03:54Z vhennebert $
*/
public class ImageIOJPEGImageWriter extends ImageIOImageWriter {
private static final String JPEG_NATIVE_FORMAT = "javax_imageio_jpeg_image_1.0";
/**
* Main constructor.
*/
public ImageIOJPEGImageWriter() {
super("image/jpeg");
}
/**
* @see ImageIOImageWriter#updateMetadata(javax.imageio.metadata.IIOMetadata, ImageWriterParams)
*/
protected IIOMetadata updateMetadata(IIOMetadata meta, ImageWriterParams params) {
//ImageIODebugUtil.dumpMetadata(meta);
if (JPEG_NATIVE_FORMAT.equals(meta.getNativeMetadataFormatName())) {
meta = addAdobeTransform(meta);
IIOMetadataNode root = (IIOMetadataNode)meta.getAsTree(JPEG_NATIVE_FORMAT);
//IIOMetadataNode root = new IIOMetadataNode(jpegmeta);
IIOMetadataNode jv = getChildNode(root, "JPEGvariety");
if (jv == null) {
jv = new IIOMetadataNode("JPEGvariety");
root.appendChild(jv);
}
IIOMetadataNode child;
if (params.getResolution() != null) {
child = getChildNode(jv, "app0JFIF");
if (child == null) {
child = new IIOMetadataNode("app0JFIF");
jv.appendChild(child);
}
//JPEG gets special treatment because there seems to be a bug in
//the JPEG codec in ImageIO converting the pixel size incorrectly
//(or not at all) when using standard metadata format.
child.setAttribute("majorVersion", null);
child.setAttribute("minorVersion", null);
child.setAttribute("resUnits", "1"); //dots per inch
child.setAttribute("Xdensity", params.getResolution().toString());
child.setAttribute("Ydensity", params.getResolution().toString());
child.setAttribute("thumbWidth", null);
child.setAttribute("thumbHeight", null);
}
/*
IIOMetadataNode ms = getChildNode(root, "markerSequence");
if (ms == null) {
ms = new IIOMetadataNode("markerSequence");
root.appendChild(ms);
}*/
try {
meta.setFromTree(JPEG_NATIVE_FORMAT, root);
//meta.mergeTree(JPEG_NATIVE_FORMAT, root);
} catch (IIOInvalidTreeException e) {
throw new RuntimeException("Cannot update image metadata: "
+ e.getMessage(), e);
}
//ImageIODebugUtil.dumpMetadata(meta);
//meta = super.updateMetadata(meta, params);
//ImageIODebugUtil.dumpMetadata(meta);
}
return meta;
}
private static IIOMetadata addAdobeTransform(IIOMetadata meta) {
// add the adobe transformation (transform 1 -> to YCbCr)
IIOMetadataNode root = (IIOMetadataNode)meta.getAsTree(JPEG_NATIVE_FORMAT);
IIOMetadataNode markerSequence = getChildNode(root, "markerSequence");
if (markerSequence == null) {
throw new RuntimeException("Invalid metadata!");
}
IIOMetadataNode adobeTransform = getChildNode(markerSequence, "app14Adobe");
if (adobeTransform == null) {
adobeTransform = new IIOMetadataNode("app14Adobe");
adobeTransform.setAttribute("transform" , "1"); // convert RGB to YCbCr
adobeTransform.setAttribute("version", "101");
adobeTransform.setAttribute("flags0", "0");
adobeTransform.setAttribute("flags1", "0");
markerSequence.appendChild(adobeTransform);
} else {
adobeTransform.setAttribute("transform" , "1");
}
try {
meta.setFromTree(JPEG_NATIVE_FORMAT, root);
} catch (IIOInvalidTreeException e) {
throw new RuntimeException("Cannot update image metadata: "
+ e.getMessage(), e);
}
return meta;
}
/**
* @see ImageIOImageWriter#getDefaultWriteParam(javax.imageio.ImageWriter, java.awt.image.RenderedImage, ImageWriterParams)
*/
protected ImageWriteParam getDefaultWriteParam(
ImageWriter iiowriter, RenderedImage image,
ImageWriterParams params) {
JPEGImageWriteParam param = new JPEGImageWriteParam(iiowriter.getLocale());
//ImageTypeSpecifier type = ImageTypeSpecifier.createFromRenderedImage(image);
/*
ImageTypeSpecifier type = new ImageTypeSpecifier(
image.getColorModel(), image.getSampleModel());
*/
/* didn't work as expected...
ImageTypeSpecifier type = ImageTypeSpecifier.createFromBufferedImageType(
BufferedImage.TYPE_INT_RGB);
param.setDestinationType(type);
param.setSourceBands(new int[] {0, 1, 2});
*/
return param;
}
}