/*

 * Copyright (c) 2007, 2010, 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 6664068 6666931

 * @summary Tests that resizing a window to which a tight loop is rendering

 * doesn't produce artifacts or crashes

 * @author Dmitri.Trembovetski@sun.com: area=Graphics

 * @run main/othervm OnScreenRenderingResizeTest

 * @run main/othervm -Dsun.java2d.d3d=false OnScreenRenderingResizeTest

 */

import java.awt.AWTException;

import java.awt.Color;

import java.awt.EventQueue;

import java.awt.Frame;

import java.awt.Graphics;

import java.awt.Graphics2D;

import java.awt.GraphicsConfiguration;

import java.awt.Insets;

import java.awt.Point;

import java.awt.Rectangle;

import java.awt.Robot;

import java.awt.event.WindowAdapter;

import java.awt.event.WindowEvent;

import java.awt.image.BufferedImage;

import java.awt.image.VolatileImage;

import java.io.File;

import java.io.IOException;

import javax.imageio.ImageIO;

public class OnScreenRenderingResizeTest {

    private static volatile boolean done = false;

    private static volatile boolean nocheck = false;

    private static final int FRAME_W = 256;

    private static final int FRAME_H = 256;

    private static final int IMAGE_W = 128;

    private static final int IMAGE_H = 128;

    private static long RUN_TIME = 1000*20;

    private static final Color renderColor = Color.green;

    private static final Color bgColor = Color.white;

    public static void main(String[] args) {

        for (String arg : args) {

            if ("-inf".equals(arg)) {

                System.err.println("Test will run indefinitely");

                RUN_TIME = Long.MAX_VALUE;

            } else  if ("-nocheck".equals(arg)) {

                System.err.println("Test will not check rendering results");

                nocheck = true;

            } else {

                System.err.println("Usage: OnScreenRenderingResizeTest [-inf][-nocheck]");

            }

        }

        BufferedImage output =

            new BufferedImage(IMAGE_W, IMAGE_H, BufferedImage.TYPE_INT_RGB);

        output.setAccelerationPriority(0.0f);

        Graphics g = output.getGraphics();

        g.setColor(renderColor);

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

        final Frame frame = new Frame("OnScreenRenderingResizeTest") {

            public void paint(Graphics g) {}

            public void update(Graphics g) {}

        };

        frame.setBackground(bgColor);

        frame.pack();

        frame.setSize(FRAME_W, FRAME_H);

        frame.addWindowListener(new WindowAdapter() {

            public void windowClosing(WindowEvent e) {

                done = true;

            }

        });

        try {

            EventQueue.invokeAndWait(new Runnable() {

                public void run() {

                    frame.setVisible(true);

                }

            });

            // wait for Vista's effects to complete

            Thread.sleep(2000);

        } catch (Exception ex) {

            ex.printStackTrace();

        }

        GraphicsConfiguration gc = frame.getGraphicsConfiguration();

        int maxW = gc.getBounds().width /2;

        int maxH = gc.getBounds().height/2;

        int minW = frame.getWidth();

        int minH = frame.getHeight();

        int incW = 10, incH = 10, cnt = 0;

        Robot robot = null;

        if (!nocheck && gc.getColorModel().getPixelSize() > 8) {

            try {

                robot = new Robot();

            } catch (AWTException ex) {

                System.err.println("Robot creation failed, continuing.");

            }

        } else {

            System.err.println("No screen rendering checks.");

        }

        VolatileImage vi = gc.createCompatibleVolatileImage(512, 512);

        vi.validate(gc);

        long timeStarted = System.currentTimeMillis();

        while (!done && (System.currentTimeMillis() - timeStarted) < RUN_TIME) {

            if (++cnt > 100) {

                int w = frame.getWidth() + incW;

                int h = frame.getHeight() + incH;

                if (w < minW || w > maxW ) {

                    incW = -incW;

                }

                if (h < minH || h > maxH ) {

                    incH = -incH;

                }

                frame.setSize(w, h);

                cnt = 0;

            }

            // try to put the device into non-default state, for example,

            // this operation below will set the transform

            vi.validate(gc);

            Graphics2D vig = (Graphics2D)vi.getGraphics();

            vig.rotate(30.0f, vi.getWidth()/2, vi.getHeight()/2);

            vig.drawImage(output, 0, 0,

                          vi.getWidth(), vi.getHeight(), null);

            Insets in = frame.getInsets();

            frame.getGraphics().drawImage(output, in.left, in.top, null);

            if (cnt == 90 && robot != null) {

                // area where we blitted to should be either white or green

                Point p = frame.getLocationOnScreen();

                p.translate(in.left+10, in.top+10);

                BufferedImage bi =

                    robot.createScreenCapture(

                        new Rectangle(p.x, p.y, IMAGE_W/2, IMAGE_H/2));

                int accepted1[] = { Color.white.getRGB(), Color.green.getRGB()};

                checkBI(bi, accepted1);

                // the are where we didn't render should stay white

                p = frame.getLocationOnScreen();

                p.translate(in.left, in.top+IMAGE_H+5);

                bi = robot.createScreenCapture(

                    new Rectangle(p.x, p.y,

                                  frame.getWidth()-in.left-in.right,

                                  frame.getHeight()-in.top-in.bottom-5-IMAGE_H));

                int accepted2[] = { Color.white.getRGB() };

                checkBI(bi, accepted1);

            }

            Thread.yield();

        }

        frame.dispose();

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

    }

    private static void checkBI(BufferedImage bi, int accepted[]) {

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

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

                int pix = bi.getRGB(x, y);

                boolean found = false;

                for (int acc : accepted) {

                    if (pix == acc) {

                        found = true;

                        break;

                    }

                }

                if (!found) {

                    try {

                        String name = "OnScreenRenderingResizeTest.png";

                        ImageIO.write(bi, "png", new File(name));

                        System.out.println("Screen shot file: " + name);

                    } catch (IOException ex) {}

                    throw new

                        RuntimeException("Test failed at " + x + "-" + y +

                                         " rgb=0x" + Integer.toHexString(pix));

                }

            }

        }

    }

}