| /* |
| * Copyright (c) 2003, 2017, 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 4641872 |
| * @summary Tests writing and reading abilities of WBMP plugin |
| */ |
| |
| import java.awt.Color; |
| import java.awt.Graphics2D; |
| import java.awt.image.BufferedImage; |
| import java.awt.image.ColorModel; |
| import java.awt.image.Raster; |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.FileOutputStream; |
| import java.io.IOException; |
| import java.util.Iterator; |
| |
| import javax.imageio.IIOException; |
| import javax.imageio.IIOImage; |
| import javax.imageio.ImageIO; |
| import javax.imageio.ImageReader; |
| import javax.imageio.ImageTypeSpecifier; |
| import javax.imageio.ImageWriteParam; |
| import javax.imageio.ImageWriter; |
| import javax.imageio.metadata.IIOMetadata; |
| |
| public class WBMPPluginTest { |
| |
| private static final int[] types = { |
| BufferedImage.TYPE_INT_RGB, // = 1; |
| BufferedImage.TYPE_INT_ARGB, // = 2; |
| BufferedImage.TYPE_INT_ARGB_PRE, // = 3; |
| BufferedImage.TYPE_INT_BGR, // = 4; |
| BufferedImage.TYPE_3BYTE_BGR, // = 5; |
| BufferedImage.TYPE_4BYTE_ABGR, // = 6; |
| BufferedImage.TYPE_4BYTE_ABGR_PRE, // 7 |
| BufferedImage.TYPE_USHORT_565_RGB, // 8 |
| BufferedImage.TYPE_USHORT_555_RGB, // 9 |
| BufferedImage.TYPE_BYTE_GRAY, // 10 |
| BufferedImage.TYPE_USHORT_GRAY, //11 |
| BufferedImage.TYPE_BYTE_BINARY, //12 |
| BufferedImage.TYPE_BYTE_INDEXED //13 |
| }; |
| |
| private static String format = "WBMP"; |
| |
| private static ImageReader ir = null; |
| private static ImageWriter iw = null; |
| private BufferedImage img; |
| private ImageWriteParam param; |
| private ByteArrayOutputStream baos; |
| |
| private static void init() { |
| |
| Iterator i = ImageIO.getImageWritersByFormatName(format); |
| if (!i.hasNext()) { |
| throw new RuntimeException("No available ImageWrites for "+format+" format!"); |
| } |
| iw = (ImageWriter)i.next(); |
| |
| i = ImageIO.getImageReadersByFormatName(format); |
| if (!i.hasNext()) { |
| throw new RuntimeException("No available ImageReaders for " +format+" format!"); |
| } |
| |
| ir = (ImageReader)i.next(); |
| } |
| |
| public static void main(String[] args) { |
| if (args.length > 0) { |
| format = args[0]; |
| System.out.println("Test format " + format); |
| } |
| |
| init(); |
| ImageIO.setUseCache(false); |
| |
| for (int i=0; i<types.length; i++) { |
| boolean bPassed = true; |
| Object reason = null; |
| |
| try { |
| |
| BufferedImage image = createTestImage(types[i]); |
| |
| ImageWriteParam param = iw.getDefaultWriteParam(); |
| |
| WBMPPluginTest t = new WBMPPluginTest(image, param); |
| boolean res = false; |
| res = t.test(); |
| if (!res) { |
| bPassed = false; |
| reason = new String("Null result"); |
| } |
| } catch (IllegalArgumentException ex) { |
| System.out.println("Expected exception type was caught: " + ex); |
| |
| } catch (Throwable ex ) { |
| System.out.println("FAILED"); |
| ex.printStackTrace(); |
| bPassed = false; |
| reason = ex; |
| throw new RuntimeException("Test for type " + types[i] + " FAILED due to exception"); |
| } |
| /* |
| System.out.println("Type " + types[i] + " result: " + |
| (bPassed ? "PASSED" : "FAILED") + |
| ((reason != null) ? (" Reason: " + reason) : "")); |
| */ |
| System.out.println("Test for type " + types[i] + " PASSED"); |
| } |
| |
| System.out.println("END OF TEST"); |
| } |
| |
| public WBMPPluginTest(BufferedImage img, ImageWriteParam param) { |
| |
| this.img = img; |
| this.param = param; |
| baos = new ByteArrayOutputStream(); |
| } |
| |
| public boolean test() throws IIOException, IOException { |
| |
| ir.reset(); |
| iw.reset(); |
| |
| String[] suffixes = iw.getOriginatingProvider().getFileSuffixes(); |
| |
| IIOMetadata md = iw.getDefaultImageMetadata(new ImageTypeSpecifier(img), param); |
| IIOImage iio_img = new IIOImage(img, null, md); |
| |
| System.out.println("Image type " + img.getType()); |
| |
| String fname = "test"+img.getType()+"."+suffixes[0]; |
| |
| iw.setOutput(ImageIO.createImageOutputStream(new FileOutputStream(new File(fname)))); |
| System.out.print("write image ... "); |
| iw.write(iio_img); |
| System.out.println("OK"); |
| System.out.print("read image ... "); |
| |
| byte[] ba_image = baos.toByteArray(); |
| |
| ByteArrayInputStream bais = new ByteArrayInputStream(ba_image); |
| |
| ir.setInput(ImageIO.createImageInputStream(new FileInputStream(new File(fname)))); |
| |
| BufferedImage res = ir.read(0); |
| System.out.println("OK"); |
| |
| System.out.print("compare images ... "); |
| boolean r = compare(img,res); |
| System.out.println(r?"OK":"FAILED"); |
| return r; |
| } |
| |
| private boolean compare(BufferedImage in, BufferedImage out) { |
| int width = in.getWidth(); |
| int height = in.getHeight(); |
| if (out.getWidth() != width || out.getHeight() != height) { |
| throw new RuntimeException("Dimensions changed!"); |
| } |
| |
| Raster oldras = in.getRaster(); |
| ColorModel oldcm = in.getColorModel(); |
| Raster newras = out.getRaster(); |
| ColorModel newcm = out.getColorModel(); |
| |
| for (int j = 0; j < height; j++) { |
| for (int i = 0; i < width; i++) { |
| Object oldpixel = oldras.getDataElements(i, j, null); |
| int oldrgb = oldcm.getRGB(oldpixel); |
| int oldalpha = oldcm.getAlpha(oldpixel); |
| |
| Object newpixel = newras.getDataElements(i, j, null); |
| int newrgb = newcm.getRGB(newpixel); |
| int newalpha = newcm.getAlpha(newpixel); |
| |
| if (newrgb != oldrgb || |
| newalpha != oldalpha) { |
| throw new RuntimeException("Pixels differ at " + i + |
| ", " + j); |
| } |
| } |
| } |
| return true; |
| } |
| |
| |
| private static BufferedImage createTestImage(int type) throws IOException { |
| |
| int w = 200; |
| int h = 200; |
| BufferedImage b = new BufferedImage(w, h, type); |
| Graphics2D g = b.createGraphics(); |
| g.setColor(Color.white); |
| g.fillRect(0,0, w, h); |
| g.setColor(Color.black); |
| g.fillOval(10, 10, w -20, h-20); |
| |
| return b; |
| } |
| |
| } |