/*
 * 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;
    }

}