--- /dev/null 2018-04-17 14:16:19.391357000 +0530 +++ new/test/jdk/javax/imageio/plugins/png/VerifyRGBValuesFromBKGDChunk.java 2018-04-17 15:19:00.865702729 +0530 @@ -0,0 +1,171 @@ +/* + * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. + * + * This code is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + */ + +/* + * @test + * @bug 6574555 5109146 + * @summary Test verifies that PNGImageWriter encodes the R, G, B + * values of bKGD chunk properly. + * @run main VerifyRGBValuesFromBKGDChunk + */ + +import java.awt.image.BufferedImage; +import java.io.IOException; +import java.util.Iterator; +import javax.imageio.ImageTypeSpecifier; +import javax.imageio.ImageWriter; +import javax.imageio.ImageReader; +import javax.imageio.ImageIO; +import javax.imageio.ImageWriteParam; +import javax.imageio.metadata.IIOInvalidTreeException; +import javax.imageio.metadata.IIOMetadata; +import javax.imageio.metadata.IIOMetadataNode; +import javax.imageio.IIOImage; +import javax.imageio.stream.ImageInputStream; +import javax.imageio.stream.ImageOutputStream; +import java.io.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; +import java.io.InputStream; + +public class VerifyRGBValuesFromBKGDChunk { + + private static IIOMetadata encodeMetadata; + private static ImageWriter writer; + private static BufferedImage img; + private static ImageWriteParam param; + private static boolean nativeBKGDFail, standardBKGDFail; + private static IIOMetadataNode bKGD_RGBNode; + private static final String BKGDRED = "100"; + private static final String BKGDGREEN = "150"; + private static final String BKGDBLUE = "200"; + + private static void mergeStandardMetadata() throws IIOInvalidTreeException { + IIOMetadataNode background_rgb = new IIOMetadataNode("BackgroundColor"); + + background_rgb.setAttribute("red", BKGDRED); + background_rgb.setAttribute("green", BKGDGREEN); + background_rgb.setAttribute("blue", BKGDBLUE); + + IIOMetadataNode chroma = new IIOMetadataNode("Chroma"); + chroma.appendChild(background_rgb); + IIOMetadataNode encodeRoot = new IIOMetadataNode("javax_imageio_1.0"); + encodeRoot.appendChild(chroma); + + encodeMetadata.mergeTree("javax_imageio_1.0", encodeRoot); + } + + private static void mergeNativeMetadata() throws IIOInvalidTreeException { + IIOMetadataNode bKGD_rgb = new IIOMetadataNode("bKGD_RGB"); + + bKGD_rgb.setAttribute("red", BKGDRED); + bKGD_rgb.setAttribute("green", BKGDGREEN); + bKGD_rgb.setAttribute("blue", BKGDBLUE); + + IIOMetadataNode bKGD = new IIOMetadataNode("bKGD"); + bKGD.appendChild(bKGD_rgb); + IIOMetadataNode encodeRoot = + new IIOMetadataNode("javax_imageio_png_1.0"); + encodeRoot.appendChild(bKGD); + + encodeMetadata.mergeTree("javax_imageio_png_1.0", encodeRoot); + } + + private static void writeAndReadMetadata() throws IOException { + ByteArrayOutputStream baos = new ByteArrayOutputStream(); + ImageOutputStream ios = ImageIO.createImageOutputStream(baos); + writer.setOutput(ios); + + writer.write(encodeMetadata, + new IIOImage(img, null, encodeMetadata), param); + writer.dispose(); + + baos.flush(); + byte[] imageByteArray = baos.toByteArray(); + baos.close(); + + // Get bKGD chunk from image and verify the values + InputStream input= new ByteArrayInputStream(imageByteArray); + ImageInputStream iis = ImageIO.createImageInputStream(input); + Iterator iterReader = + ImageIO.getImageReadersBySuffix("png"); + ImageReader reader = iterReader.next(); + reader.setInput(iis, false, false); + + IIOMetadata decodeMetadata = reader.getImageMetadata(0); + IIOMetadataNode decodeRoot = + (IIOMetadataNode) decodeMetadata. + getAsTree("javax_imageio_png_1.0"); + bKGD_RGBNode = (IIOMetadataNode) + decodeRoot.getElementsByTagName("bKGD_RGB").item(0); + reader.dispose(); + } + + private static boolean verifyRGBValues() { + return (!(BKGDRED.equals(bKGD_RGBNode.getAttribute("red")) && + BKGDGREEN.equals(bKGD_RGBNode.getAttribute("green")) && + BKGDBLUE.equals(bKGD_RGBNode.getAttribute("blue")))); + } + + private static void VerifyNativeRGBValuesFromBKGDChunk() + throws IOException { + + mergeNativeMetadata(); + writeAndReadMetadata(); + nativeBKGDFail = verifyRGBValues(); + } + + private static void VerifyStandardRGBValuesFromBKGDChunk() + throws IOException { + + mergeStandardMetadata(); + writeAndReadMetadata(); + standardBKGDFail = verifyRGBValues(); + } + + public static void main(String[] args) throws IOException { + int width = 1; + int height = 1; + img = new BufferedImage(width, height, BufferedImage.TYPE_INT_BGR); + Iterator iterWriter = + ImageIO.getImageWritersBySuffix("png"); + writer = iterWriter.next(); + + param = writer.getDefaultWriteParam(); + ImageTypeSpecifier specifier = + ImageTypeSpecifier. + createFromBufferedImageType(BufferedImage.TYPE_INT_BGR); + encodeMetadata = writer.getDefaultImageMetadata(specifier, param); + + // Verify bKGD RGB values after merging metadata using native tree + VerifyNativeRGBValuesFromBKGDChunk(); + + // Verify bKGD RGB values after merging metadata using standard tree + VerifyStandardRGBValuesFromBKGDChunk(); + + if (nativeBKGDFail || standardBKGDFail) { + throw new RuntimeException("bKGD RGB values are not stored" + + " properly"); + } + } +} +