/*

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

 */

import java.awt.AWTException;

import java.awt.Color;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.Frame;

import java.awt.Graphics;

import java.awt.Point;

import java.awt.Rectangle;

import java.awt.Robot;

import java.awt.Toolkit;

import java.awt.image.BufferedImage;

import java.awt.image.VolatileImage;

/*

 *

 *

 * Tests that the use of shared memory pixmaps isn't broken:

 * create a VolatileImage, fill it with red color, copy it to the screen

 * make sure the pixels on the screen are red.

 *

 * Note that we force the use of shared memory pixmaps in the shell script.

 *

 * @author Dmitri.Trembovetski

 */

public class SharedMemoryPixmapsTest {

    static final int IMAGE_SIZE = 100;

    static boolean show = false;

    final Frame testFrame;

    /** Creates a new instance of SharedMemoryPixmapsTest */

    public SharedMemoryPixmapsTest() {

        testFrame = new Frame("SharedMemoryPixmapsTest");

        testFrame.add(new TestComponent());

        testFrame.pack();

        testFrame.setVisible(true);

    }

    public static void main(String[] args) {

        for (String s : args) {

            if ("-show".equals(s)) {

                show = true;

            } else {

                System.err.println("Usage: SharedMemoryPixmapsTest [-show]");

            }

        }

        new SharedMemoryPixmapsTest();

    }

    private class TestComponent extends Component {

        VolatileImage vi = null;

        boolean tested = false;

        void initVI() {

            int res;

            if (vi == null) {

                res = VolatileImage.IMAGE_INCOMPATIBLE;

            } else {

                res = vi.validate(getGraphicsConfiguration());

            }

            if (res == VolatileImage.IMAGE_INCOMPATIBLE) {

                if (vi != null) vi.flush();

                vi = createVolatileImage(IMAGE_SIZE, IMAGE_SIZE);

                vi.validate(getGraphicsConfiguration());

                res = VolatileImage.IMAGE_RESTORED;

            }

            if (res == VolatileImage.IMAGE_RESTORED) {

                Graphics vig = vi.getGraphics();

                vig.setColor(Color.red);

                vig.fillRect(0, 0, vi.getWidth(), vi.getHeight());

                vig.dispose();

            }

        }

        @Override

        public synchronized void paint(Graphics g) {

            do {

                g.setColor(Color.green);

                g.fillRect(0, 0, getWidth(), getHeight());

                initVI();

                g.drawImage(vi, 0, 0, null);

            } while (vi.contentsLost());

            Toolkit.getDefaultToolkit().sync();

            if (!tested) {

                if (testRendering()) {

                    System.err.println("Test Passed");

                } else {

                    System.err.println("Test Failed");

                }

                tested = true;

            }

            if (!show) {

                testFrame.setVisible(false);

                testFrame.dispose();

            }

        }

        private boolean testRendering() throws RuntimeException {

            try {

                Thread.sleep(2000);

            } catch (InterruptedException ex) {}

            Robot r = null;

            try {

                r = new Robot();

            } catch (AWTException ex) {

                ex.printStackTrace();

                throw new RuntimeException("Can't create Robot");

            }

            Point p = getLocationOnScreen();

            BufferedImage b =

                r.createScreenCapture(new Rectangle(p, getPreferredSize()));

            for (int y = 0; y < b.getHeight(); y++) {

                for (int x = 0; x < b.getWidth(); x++) {

                    if (b.getRGB(x, y) != Color.red.getRGB()) {

                        System.err.println("Incorrect pixel" + " at "

                            + x + "x" + y + " : " +

                            Integer.toHexString(b.getRGB(x, y)));

                        if (show) {

                            return false;

                        }

                        System.err.println("Test Failed");

                        System.exit(1);

                    }

                }

            }

            return true;

        }

        @Override

        public Dimension getPreferredSize() {

            return new Dimension(IMAGE_SIZE, IMAGE_SIZE);

        }

    }

}