/*

 * Copyright (c) 2016, 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

 * @library /test/lib

 *

 * @bug     8145776

 * @author  a.stepanov

 * @summary A simple write-read test for the multi-page tiff.

 *          Create the file programmaticaly, then do some simple checks

 *          (number of pages, sizes, colors). Use -Dseed=X to set

 *          the random generator seed.

 *

 * @run     main MultiPageTest

 * @key     randomness

 */

import java.awt.*;

import java.awt.image.*;

import java.io.*;

import java.util.*;

import javax.imageio.*;

import javax.imageio.stream.*;

import jdk.test.lib.RandomFactory;

public class MultiPageTest {

    private final String fileName;

    private final int NUM_IMAGES = 51;

    private final static Random rnd = RandomFactory.getRandom();

    private final int w[], h[];

    private final Color colors[];

    private final int BLACK_SIZE = 100;

    private final int imageType;

    public MultiPageTest(int type, String tName) {

        imageType = type;

        fileName  = "test__" + tName + ".tif";

        w = new int[NUM_IMAGES + 4];

        h = new int[NUM_IMAGES + 4];

        for (int i = 2; i < NUM_IMAGES + 2; i++) {

            w[i] = 10 + rnd.nextInt(21);

            h[i] = 10 + rnd.nextInt(21);

        }

        w[0] = BLACK_SIZE;  h[0] = BLACK_SIZE;

        w[1] = BLACK_SIZE;  h[1] = BLACK_SIZE;

        w[NUM_IMAGES + 2] = BLACK_SIZE;  h[NUM_IMAGES + 2] = BLACK_SIZE;

        w[NUM_IMAGES + 3] = BLACK_SIZE;  h[NUM_IMAGES + 3] = BLACK_SIZE;

        colors = new Color[NUM_IMAGES + 4];

        for (int i = 2; i < NUM_IMAGES + 2; ++i) {

            colors[i] = new Color(

                rnd.nextInt(256), rnd.nextInt(256), rnd.nextInt(256));

        }

        colors[0] = Color.black;

        colors[1] = Color.black;

        colors[NUM_IMAGES + 2] = Color.black;

        colors[NUM_IMAGES + 3] = Color.black;

    }

    private ImageWriter getTIFFWriter() {

        Iterator<ImageWriter> writers = ImageIO.getImageWritersByFormatName("TIFF");

        if (!writers.hasNext()) {

            throw new RuntimeException("No writers available for " + fileName);

        }

        return writers.next();

    }

    private ImageReader getTIFFReader() {

        Iterator<ImageReader> readers = ImageIO.getImageReadersByFormatName("TIFF");

        if (!readers.hasNext()) {

            throw new RuntimeException("No readers available for " + fileName);

        }

        return readers.next();

    }

    private void createImage() throws Exception {

        OutputStream s = new BufferedOutputStream(new FileOutputStream(fileName));

        try (ImageOutputStream ios = ImageIO.createImageOutputStream(s)) {

            ImageWriter writer = getTIFFWriter();

            writer.setOutput(ios);

            Graphics g;

            BufferedImage blackImg =

                new BufferedImage(BLACK_SIZE, BLACK_SIZE, imageType);

            g = blackImg.getGraphics();

            g.setColor(Color.black);

            g.fillRect(0, 0, BLACK_SIZE, BLACK_SIZE);

            writer.prepareWriteSequence(null);

            for (int i = 2; i < NUM_IMAGES + 2; i++) {

                BufferedImage img = new BufferedImage(w[i], h[i], imageType);

                g = img.getGraphics();

                g.setColor(colors[i]);

                g.fillRect(0, 0, w[i], h[i]);

                writer.writeToSequence(new IIOImage(img, null, null), null);

            }

            writer.endWriteSequence();

            // check: insert to the beginning

            writer.writeInsert(0, new IIOImage(blackImg, null, null), null);

            // check: insert to non-zero position

            writer.writeInsert(1, new IIOImage(blackImg, null, null), null);

            // check: append to the end by index

            writer.writeInsert(NUM_IMAGES + 2,

                new IIOImage(blackImg, null, null), null);

            // check: append to the end using index "-1"

            writer.writeInsert(-1, new IIOImage(blackImg, null, null), null);

            ios.flush();

            writer.dispose();

        }

        s.close();

    }

    private void readAndCheckImage() throws Exception {

        ImageReader reader = getTIFFReader();

        ImageInputStream s = ImageIO.createImageInputStream(new File(fileName));

        reader.setInput(s);

        // check number of pages

        if ((NUM_IMAGES + 4) != reader.getNumImages(true)) {

            throw new RuntimeException("invalid number of images!");

        }

        // check colors / sizes

        for (int i = 0; i < NUM_IMAGES + 4; i++) {

            BufferedImage img = reader.read(i);

            int imw = w[i], imh = h[i];

            if ( (img.getWidth() != imw) || (img.getHeight() != imh) ) {

                throw new RuntimeException("NOK: size(" + i + ")");

            }

            Color

                    c1 = new Color(img.getRGB(0, 0)),

                    c2 = new Color(img.getRGB(imw / 2, imh / 2)),

                    c3 = new Color(img.getRGB(imw - 1, imh - 1));

            if (! (c1.equals(colors[i]) && c1.equals(c2) && c1.equals(c3) ) ) {

                throw new RuntimeException("NOK: color(" + i + ")");

            }

        }

        reader.dispose();

        s.close();

    }

    public void doTest() throws Exception {

        createImage();

        readAndCheckImage();

    }

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

        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

        };

        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"

        };

        for (int i = 0; i < types.length; i++) {

            System.out.println("image type: " + names[i]);

            (new MultiPageTest(types[i], names[i])).doTest();

        }

    }

}