/*
 * Copyright (c) 2008, 2013, 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 %I% %E%
 * @bug 6683728
 * @summary Tests that a JApplet in a translucent JFrame works properly
 * @author Kenneth.Russell@sun.com: area=Graphics
 * @compile -XDignore.symbol.file=true TranslucentJAppletTest.java
 * @run main/manual/othervm TranslucentJAppletTest
 */

import java.awt.*;
import java.awt.image.*;

import javax.swing.*;

public class TranslucentJAppletTest {

    private static volatile GraphicsConfiguration graphicsConfig = null;
    private static JFrame frame;
    private static volatile boolean paintComponentCalled = false;

    private static void initAndShowGUI() {
        frame = new JFrame(graphicsConfig);
        JApplet applet = new JApplet();
        applet.setBackground(new Color(0, 0, 0, 0));
        JPanel panel = new JPanel() {
            protected void paintComponent(Graphics g) {
                paintComponentCalled = true;
                g.setColor(Color.RED);
                g.fillOval(0, 0, getWidth(), getHeight());
            }
        };
        panel.setDoubleBuffered(false);
        panel.setOpaque(false);
        applet.add(panel);
        frame.add(applet);
        frame.setBounds(100, 100, 200, 200);
        frame.setUndecorated(true);
        frame.setBackground(new Color(0, 0, 0, 0));
        frame.setVisible(true);
    }

    public static void main(String[] args)
        throws Exception
    {
        sun.awt.SunToolkit tk = (sun.awt.SunToolkit)Toolkit.getDefaultToolkit();

        final GraphicsEnvironment ge = GraphicsEnvironment.getLocalGraphicsEnvironment();
        for (GraphicsDevice gd : ge.getScreenDevices()) {
            if (gd.isWindowTranslucencySupported(
                        GraphicsDevice.WindowTranslucency.PERPIXEL_TRANSLUCENT))
            {
                for (GraphicsConfiguration gc : gd.getConfigurations()) {
                    if (gc.isTranslucencyCapable()) {
                        graphicsConfig = gc;
                        break;
                    }
                }
            }
            if (graphicsConfig != null) {
                break;
            }
        }
        if (graphicsConfig == null) {
            System.out.println("The system does not support translucency. Consider the test passed.");
            return;
        }

        Robot r = new Robot();
        Color color1 = r.getPixelColor(100, 100); // (0, 0) in frame coordinates

        SwingUtilities.invokeAndWait(new Runnable() {
            public void run() {
                initAndShowGUI();
            }
        });
        tk.realSync();

        if (!paintComponentCalled) {
            throw new RuntimeException("Test FAILED: panel's paintComponent() method is not called");
        }

        Color newColor1 = r.getPixelColor(100, 100);
        // unfortunately, robot.getPixelColor() doesn't work for some unknown reason
        // Color newColor2 = r.getPixelColor(200, 200);
        BufferedImage bim = r.createScreenCapture(new Rectangle(200, 200, 1, 1));
        Color newColor2 = new Color(bim.getRGB(0, 0));

        // Frame must be transparent at (100, 100) in screen coords
        if (!color1.equals(newColor1)) {
            System.err.println("color1 = " + color1);
            System.err.println("newColor1 = " + newColor1);
            throw new RuntimeException("Test FAILED: frame pixel at (0, 0) is not transparent");
        }

        // Frame must be RED at (200, 200) in screen coords
        if (!newColor2.equals(Color.RED)) {
            System.err.println("newColor2 = " + newColor2);
            throw new RuntimeException("Test FAILED: frame pixel at (100, 100) is not red (transparent?)");
        }

        System.out.println("Test PASSED");
    }
}
