/*
 * Copyright (c) 2016, 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     8144991 8150154
 * @author  a.stepanov
 * @summary Check if repeating image writing doesn't fail
 *          (particularly, no AIOOB occurs)
 *
 * @run     main RepeatingWriteTest
 */


import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.*;
import javax.imageio.*;
import javax.imageio.stream.*;
import java.util.Iterator;
import java.util.Locale;
import javax.imageio.plugins.bmp.BMPImageWriteParam;
import javax.imageio.plugins.jpeg.JPEGImageWriteParam;

public class RepeatingWriteTest {

    private static final int TYPES[] = new int[]{
       BufferedImage.TYPE_INT_RGB,        BufferedImage.TYPE_INT_ARGB,
       BufferedImage.TYPE_INT_ARGB_PRE,   BufferedImage.TYPE_INT_BGR,
       BufferedImage.TYPE_3BYTE_BGR,      BufferedImage.TYPE_4BYTE_ABGR,
       BufferedImage.TYPE_4BYTE_ABGR_PRE, BufferedImage.TYPE_BYTE_GRAY,
       BufferedImage.TYPE_USHORT_GRAY,    BufferedImage.TYPE_BYTE_BINARY,
       BufferedImage.TYPE_BYTE_INDEXED,   BufferedImage.TYPE_USHORT_565_RGB,
       BufferedImage.TYPE_USHORT_555_RGB};

    private static final String NAMES[] = new String[]{
       "TYPE_INT_RGB",        "TYPE_INT_ARGB",
       "TYPE_INT_ARGB_PRE",   "TYPE_INT_BGR",
       "TYPE_3BYTE_BGR",      "TYPE_4BYTE_ABGR",
       "TYPE_4BYTE_ABGR_PRE", "TYPE_BYTE_GRAY",
       "TYPE_USHORT_GRAY",    "TYPE_BYTE_BINARY",
       "TYPE_BYTE_INDEXED",   "TYPE_USHORT_565_RGB",
       "TYPE_USHORT_555_RGB"};

    private static final int SZ1 = 200, SZ2 = 100;
    private static final Color C1 = Color.BLACK, C2 = Color.WHITE;

    private static ImageWriter getWriter(String format) {
        Iterator<ImageWriter> writers =
            ImageIO.getImageWritersByFormatName(format);
        if (!writers.hasNext()) {
            throw new RuntimeException("no writers available for " + format);
        }
        return writers.next();
    }

    private static ImageReader getReader(String format) {
        Iterator<ImageReader> readers =
            ImageIO.getImageReadersByFormatName(format);
        if (!readers.hasNext()) {
            throw new RuntimeException("no readers available for " + format);
        }
        return readers.next();
    }

    private static class ImageCheckException extends Exception {
        public ImageCheckException(String msg) { super(msg); }
    }

    private final String format;

    public RepeatingWriteTest(String format) { this.format = format; }

    private void checkImage(String fileName, boolean is1st) throws Exception {

        Color cRef = is1st ? C1  : C2;
        int  szRef = is1st ? SZ1 : SZ2;

        ImageReader reader = getReader(format);
        ImageInputStream iis = ImageIO.createImageInputStream(new File(fileName));
        reader.setInput(iis);
        BufferedImage img = reader.read(0);
        Color c = new Color(img.getRGB(szRef / 2, szRef / 2));
        int w = img.getWidth(), h = img.getHeight();

        if (w != szRef || h != szRef) {
            throw new ImageCheckException(fileName +
                ": invalid image size " + w + " x " + h +
                " expected " + szRef + " x " + szRef);
        }

        if (!c.equals(cRef)) {
            throw new ImageCheckException(fileName +
                ": invalid image color " + c +
                " expected " + cRef);
        }
    }

    private void doTest(int i, int j) throws Exception {

        String pair = NAMES[i] + " " + NAMES[j];

        // some type checks: avoid IO exceptions
        if ((format.equals("jpeg") || format.equals("bmp")) &&
            (pair.contains("USHORT_GRAY") ||
             pair.contains("ARGB") || pair.contains("ABGR"))) {
            return;
        }

        // If JDK-8163323 is fixed this if block should be removed.
        if (format.equals("tiff")
            && (NAMES[i].equals("TYPE_USHORT_555_RGB")
                 || NAMES[i].equals("TYPE_USHORT_565_RGB"))
                && (NAMES[j].equals("TYPE_USHORT_555_RGB")
                    || NAMES[j].equals("TYPE_USHORT_565_RGB"))) {
            return;
        }

        String f1 = "test-1-" + NAMES[i] + "." + format;
        String f2 = "test-2-" + NAMES[j] + "." + format;

        ImageWriter writer = getWriter(format);

        // --- write 1st image ---
        OutputStream s = new BufferedOutputStream(new FileOutputStream(f1));
        ImageOutputStream ios = ImageIO.createImageOutputStream(s);
        writer.setOutput(ios);

        BufferedImage img = new BufferedImage(SZ1, SZ1, TYPES[i]);
        Graphics g = img.getGraphics();
        g.setColor(C1);
        g.fillRect(0, 0, SZ1, SZ1);
        g.dispose();

        if (format.equals("jpeg")) {
            writer.write(null, new IIOImage(img, null, null),
                new JPEGImageWriteParam(Locale.getDefault()));
        } if (format.equals("bmp")) {
            writer.write(null, new IIOImage(img, null, null),
                new BMPImageWriteParam());
        } else {
            writer.write(img);
        }
        ios.flush();
        s.close();

        // --- write 2nd image ---
        s = new BufferedOutputStream(new FileOutputStream(f2));
        ios = ImageIO.createImageOutputStream(s);
        writer.setOutput(ios);

        img = new BufferedImage(SZ2, SZ2, TYPES[j]);
        g = img.getGraphics();
        g.setColor(C2);
        g.fillRect(0, 0, SZ2, SZ2);
        g.dispose();

        if (format.equals("jpeg")) {
            writer.write(null, new IIOImage(img, null, null),
                new JPEGImageWriteParam(Locale.getDefault()));
        } if (format.equals("bmp")) {
            writer.write(null, new IIOImage(img, null, null),
                new BMPImageWriteParam());
        } else {
            writer.write(img);
        }
        ios.flush();
        s.close();

        // --- check files ---
        checkImage(f1, true);
        checkImage(f2, false);
    }

    public static void main(String args[]) throws Exception {


        int n = TYPES.length;
        int nAIOOB = 0, nChecksFailed = 0;

        String formats[] = {"bmp", "jpeg", "gif", "png", "tiff"};

        for (String f: formats) {
            System.out.println("\nformat: " + f);
            RepeatingWriteTest test = new RepeatingWriteTest(f);
            for (int i = 0; i < n; ++i) {
                for (int j = 0; j < n; ++j) {
                    try {
                        test.doTest(i, j);
                    } catch (ArrayIndexOutOfBoundsException e) {
                        System.err.println(f + ": AIOOB for pair " +
                            NAMES[i] + ", " + NAMES[j] + ": " + e.getMessage());
                        nAIOOB++;
                    } catch (ImageCheckException e) {
                        System.err.println(f +
                            ": image check failed for " + e.getMessage());
                        nChecksFailed++;
                    }
                }
            }
        }

        if (nAIOOB > 0 || nChecksFailed > 0) {
            throw new RuntimeException("test failed: " + nAIOOB + " AIOOBs, " +
                nChecksFailed + " image check failures");
        }
    }
}