/*
 * Copyright (c) 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
 * @bug 8162959
 * @summary Validate output of createScreenCapture new API which returns
 *          MultiResolutionImage on HiDPI Display and BufferedImage
 *          on Non-HiDPI Display.
 * @run main/othervm -Dsun.java2d.uiScale=1 ScreenCaptureTest 
 * @run main/othervm -Dsun.java2d.uiScale=2 ScreenCaptureTest 
 */

import java.awt.Dimension;
import java.awt.GraphicsEnvironment;
import java.awt.Image;
import java.awt.Rectangle;
import java.awt.Robot;
import java.awt.Toolkit;
import java.awt.geom.AffineTransform;
import java.awt.image.BufferedImage;
import java.awt.image.MultiResolutionImage;
import java.awt.BorderLayout;
import java.awt.Canvas;
import java.awt.Color;
import java.awt.Frame;
import java.awt.Graphics;
import java.awt.Panel;
import java.awt.Robot;

public class ScreenCaptureTest {

    private static Robot robot;
    private static Frame frame;
    private static boolean isHiDPI = true;
    private static final Color[] COLORS = {
        Color.GREEN, Color.BLUE, Color.ORANGE, Color.RED};

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

        frame = new Frame();
        frame.setBounds(0, 0, 400, 400);
        frame.setUndecorated(true);
	robot = new Robot();
        Panel panel = new Panel(new BorderLayout());
        Canvas canvas = new Canvas() {
            public void paint(Graphics g) {
                super.paint(g);
                int w = getWidth();
                int h = getHeight();
                g.setColor(COLORS[0]);
                g.fillRect(0, 0, w / 2, h / 2);
                g.setColor(COLORS[1]);
                g.fillRect(w / 2, 0, w / 2, h / 2);
                g.setColor(COLORS[2]);
                g.fillRect(0, h / 2, w / 2, h / 2);
                g.setColor(COLORS[3]);
                g.fillRect(w / 2, h / 2, w / 2, h / 2);
            }
        };

        panel.add(canvas);
        frame.add(panel);
        frame.setVisible(true);
        robot.delay(500);
	    robot.waitForIdle();

        int w = frame.getWidth();
        int h = frame.getHeight();

        // getPixelColor Test
        // Check pixel color in first quardant GREEN; x=100, y=100
        if (!robot.getPixelColor(w / 4, h / 4).equals(COLORS[0])) {
            throw new RuntimeException("Wrong Pixel Color! Expected GREEN");
        }
        // Check pixel color in second quardant; BLUE, x=300, y=100
        if (!robot.getPixelColor(3 * w / 4, h / 4).equals(COLORS[1])) {
            throw new RuntimeException("Wrong Pixel Color! Expected BLUE");
        }
        // Check pixel color in third quardant; ORANGE, x=100, y=300
        if (!robot.getPixelColor(w / 4, 3 * h / 4).equals(COLORS[2])) {
            throw new RuntimeException("Wrong Pixel Color! Expected ORANGE");
        }
        // Check pixel color in fourth quardant; RED, x=300, y=300
        if (!robot.getPixelColor(3 * w / 4, 3 * h / 4).equals(COLORS[3])) {
            throw new RuntimeException("Wrong Pixel Color! Expected RED");
        }

        // createScreenCaptureTest
        AffineTransform tx = GraphicsEnvironment.getLocalGraphicsEnvironment()
                .getDefaultScreenDevice().getDefaultConfiguration()
                .getDefaultTransform();

        if (tx.getScaleX() == 1 && tx.getScaleY() == 1) {
            isHiDPI = false;
        }

        if (!isHiDPI) {
            ValidateCreateScreenCapture(true);
            ValidateCreateScreenCapture(false);
        } else {
            ValidateCreateScreenCapture(true);
            ValidateCreateScreenCapture(false);
        }

        frame.dispose();
    }

    public static void ValidateCreateScreenCapture(boolean screenCaptureFlag) {
        Dimension size = Toolkit.getDefaultToolkit().getScreenSize();
        Image image = robot.createScreenCapture(new Rectangle(size),
                screenCaptureFlag);

        if (isHiDPI && screenCaptureFlag) {
            // HiDPI and Flag set to true
            if (!(image instanceof MultiResolutionImage)) {
                throw new RuntimeException("Expected MultiResoluitonImage. ");
            }
        } else if (isHiDPI && !screenCaptureFlag) {
            // HiDPI and Flag set to false
            if ((image instanceof MultiResolutionImage)) {
                throw new RuntimeException("Expected BufferedImage. ");
            }
        } else if (!isHiDPI) {
            // Non-HiDPI Display returns always BufferedImage, ignores Flag set.
            if (!(image instanceof BufferedImage)) {
                throw new RuntimeException("Expected BufferedImage. ");
            }
        }

    }
}