| /* |
| * 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 compression modes of BMP plugin |
| */ |
| |
| import java.awt.Color; |
| import java.awt.Graphics2D; |
| import java.awt.Rectangle; |
| 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.IOException; |
| import java.util.Iterator; |
| |
| 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; |
| import javax.imageio.stream.ImageOutputStream; |
| |
| public class BMPWriteParamTest { |
| |
| static final String format = "BMP"; |
| |
| public static void main(String[] args) { |
| |
| ImageWriter iw = null; |
| Iterator writers = ImageIO.getImageWritersByFormatName(format); |
| if (!writers.hasNext()) { |
| throw new RuntimeException("No available Image writer for "+format); |
| } |
| iw = (ImageWriter)writers.next(); |
| |
| try { |
| BufferedImage img = createTestImage(); |
| |
| BufferedImage bmp_res = getWriteResult(img, "BMP"); |
| BufferedImage png_res = getWriteResult(img, "PNG"); |
| |
| compare(bmp_res, png_res); |
| } catch (Exception ex) { |
| ex.printStackTrace(); |
| throw new RuntimeException("Unexpected exception: " + ex); |
| } |
| } |
| |
| private static BufferedImage getWriteResult(BufferedImage img, |
| String format |
| ) throws IOException { |
| ImageWriter iw = null; |
| Iterator writers = ImageIO.getImageWritersByFormatName(format); |
| while (writers.hasNext()) { |
| iw = (ImageWriter)writers.next(); |
| System.out.println(format + " -> " + iw.toString()); |
| } |
| if (iw==null) { |
| throw new RuntimeException("No available Image writer for "+format); |
| } |
| ImageWriteParam param = iw.getDefaultWriteParam(); |
| |
| param.setSourceRegion(new Rectangle(10, 10, 31, 31)); |
| param.setSourceSubsampling(3, 3, 0, 0); |
| |
| IIOMetadata meta = iw.getDefaultImageMetadata(new ImageTypeSpecifier(img), param); |
| |
| IIOImage iio_img = new IIOImage(img, null, meta); |
| |
| ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
| ImageOutputStream ios = ImageIO.createImageOutputStream(baos); |
| iw.setOutput(ios); |
| iw.write(meta, iio_img, param); |
| ios.flush(); |
| |
| byte[] ba_image = baos.toByteArray(); |
| |
| ByteArrayInputStream bais = new ByteArrayInputStream(ba_image); |
| |
| ImageReader ir = null; |
| |
| Iterator readers = ImageIO.getImageReadersByFormatName(format); |
| while (readers.hasNext()) { |
| ir = (ImageReader)readers.next(); |
| System.out.println(format + " -> " + ir.toString()); |
| } |
| if (ir==null) { |
| throw new RuntimeException("No available Image reader for "+format); |
| } |
| |
| ir.setInput(ImageIO.createImageInputStream(bais)); |
| |
| BufferedImage res = ir.read(0); |
| return res; |
| } |
| |
| private static BufferedImage createTestImage() |
| throws IOException { |
| |
| int w = 50; |
| int h = 50; |
| BufferedImage b = new BufferedImage(w, h, BufferedImage.TYPE_INT_RGB); |
| Graphics2D g = b.createGraphics(); |
| g.setColor(Color.red); |
| g.fillRect(0,0, w, h); |
| g.setColor(Color.white); |
| for (int i=10; i<=10+30; i+= 3) { |
| g.drawLine(i, 10, i, 40); |
| g.drawLine(10, i, 40, i); |
| } |
| return b; |
| } |
| |
| private static boolean compare(final BufferedImage in, |
| final BufferedImage out) |
| { |
| final 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) { |
| // showDiff(in, out); |
| throw new RuntimeException("Pixels differ at " + i + |
| ", " + j + " new = " + Integer.toHexString(newrgb) + " old = " + Integer.toHexString(oldrgb)); |
| } |
| } |
| } |
| return true; |
| } |
| } |