/*

 * 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 8151269

 * @author a.stepanov

 * @summary Multiresolution image: check that the base resolution variant

 *          source is passed to the corresponding ImageConsumer

 * @run main MultiresolutionSourceTest

 */

import java.awt.*;

import java.awt.image.*;

public class MultiresolutionSourceTest {

    private static class Checker implements ImageConsumer {

        private final int refW, refH, refType;

        private final boolean refHasAlpha;

        private final Color refColor;

        public Checker(int     w,

                       int     h,

                       Color   c,

                       boolean hasAlpha,

                       int     transferType) {

            refW = w;

            refH = h;

            refColor = c;

            refHasAlpha = hasAlpha;

            refType = transferType;

        }

        @Override

        public void imageComplete(int status) {}

        @Override

        public void setColorModel(ColorModel model) {

            boolean a = model.hasAlpha();

            if (a != refHasAlpha) {

                throw new RuntimeException("invalid hasAlpha: " + a);

            }

            int tt = model.getTransferType();

            if (tt != refType) {

                throw new RuntimeException("invalid transfer type: " + tt);

            }

        }

        @Override

        public void setDimensions(int w, int h) {

            if (w != refW) { throw new RuntimeException("invalid width: " + w +

                ", expected: " + refW); }

            if (h != refH) { throw new RuntimeException("invalid height: " + h +

                ", expected: " + refH); }

        }

        @Override

        public void setHints(int flags) {}

        @Override

        public void setPixels(int x, int y, int w, int h, ColorModel model,

                              byte pixels[], int offset, int scansize) {

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

                int p = pixels[i];

                // just in case...

                Color c = model.hasAlpha() ?

                    new Color(model.getRed  (p),

                              model.getGreen(p),

                              model.getBlue (p),

                              model.getAlpha(p)) :

                    new Color(model.getRGB(p));

                if (!c.equals(refColor)) {

                    throw new RuntimeException("invalid color: " + c +

                        ", expected: " + refColor);

                }

            }

        }

        @Override

        public void setPixels(int x, int y, int w, int h, ColorModel model,

                              int pixels[], int offset, int scansize) {

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

                int p = pixels[i];

                Color c = model.hasAlpha() ?

                    new Color(model.getRed  (p),

                              model.getGreen(p),

                              model.getBlue (p),

                              model.getAlpha(p)) :

                    new Color(model.getRGB(p));

                if (!c.equals(refColor)) {

                    throw new RuntimeException("invalid color: " + c +

                        ", expected: " + refColor);

                }

            }

        }

        @Override

        public void setProperties(java.util.Hashtable props) {}

    }

    private static BufferedImage generateImage(int w, int h, Color c, int type) {

        BufferedImage img = new BufferedImage(w, h, type);

        Graphics g = img.getGraphics();

        g.setColor(c);

        g.fillRect(0, 0, w, h);

        return img;

    }

    public static void main(String[] args) {

        final int w1 = 20, w2 = 100, h1 = 30, h2 = 50;

        final Color

            c1 = new Color(255, 0, 0, 100), c2 = Color.BLACK, gray = Color.GRAY;

        BufferedImage img1 =

            generateImage(w1, h1, c1, BufferedImage.TYPE_INT_ARGB);

        BufferedImage dummy =

            generateImage(w1 + 5, h1 + 5, gray, BufferedImage.TYPE_BYTE_GRAY);

        BufferedImage img2 =

            generateImage(w2, h2, c2, BufferedImage.TYPE_BYTE_BINARY);

        BufferedImage vars[] = new BufferedImage[] {img1, dummy, img2};

        // default base image index (zero)

        BaseMultiResolutionImage mri1 = new BaseMultiResolutionImage(vars);

        // base image index = 2

        BaseMultiResolutionImage mri2 = new BaseMultiResolutionImage(2, vars);

        // do checks

        mri1.getSource().startProduction(

                new Checker(w1, h1, c1, true, DataBuffer.TYPE_INT));

        mri2.getSource().startProduction(

                new Checker(w2, h2, c2, false, DataBuffer.TYPE_BYTE));

    }

}