/*
 * Copyright (c) 2006, 2025, 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
 * @key headful
 * @bug 8020443 6899304 4737732
 * @summary Tests that Toolkit.getScreenInsets() returns correct insets
 * @library /test/lib
 * @build jdk.test.lib.Platform
 * @run main ScreenInsetsTest
 */

import java.awt.Frame;
import java.awt.GraphicsConfiguration;
import java.awt.GraphicsDevice;
import java.awt.GraphicsEnvironment;
import java.awt.Insets;
import java.awt.Rectangle;
import java.awt.Toolkit;

import jdk.test.lib.Platform;

public class ScreenInsetsTest {
    private static final int SIZE = 100;
    // Allow a margin tolerance of 1 pixel due to scaling
    private static final int MARGIN_TOLERANCE = 1;

    public static void main(String[] args) throws InterruptedException {
        GraphicsEnvironment ge = GraphicsEnvironment.getLocalGraphicsEnvironment();
        GraphicsDevice[] gds = ge.getScreenDevices();

        for (int screen = 0; screen < gds.length; ++screen) {
            GraphicsDevice gd = gds[screen];
            GraphicsConfiguration gc = gd.getDefaultConfiguration();
            Rectangle bounds = gc.getBounds();
            Insets insets = Toolkit.getDefaultToolkit().getScreenInsets(gc);
            int workAreaWidth = bounds.width - insets.left - insets.right;
            int workAreaHeight = bounds.height - insets.top - insets.bottom;
            System.out.println("Screen #" + screen);
            System.out.println("Screen Bounds: " + bounds);
            System.out.println("Insets: " + insets);

            Frame frame = new Frame(gc);
            frame.setLocation(bounds.x + (bounds.width - SIZE) / 2,
                    bounds.y + (bounds.height - SIZE) / 2);
            frame.setSize(SIZE, SIZE);

            /*
             * On Windows, undecorated maximized frames are placed over the taskbar.
             * Use a decorated frame instead.
             */
            if (Platform.isWindows()) {
                frame.setUndecorated(false);
            } else {
                frame.setUndecorated(true);
            }

            frame.setVisible(true);

            // Maximize Frame to reach the struts
            frame.setExtendedState(java.awt.Frame.MAXIMIZED_BOTH);
            Thread.sleep(2000);

            Rectangle frameBounds = frame.getBounds();
            System.out.println("Frame bounds: " + frameBounds);

            frame.dispose();

            /*
             * On Windows, the top-left corner of an undecorated maximized frame
             * may have negative coordinates (x, y).
             * Adjust the frame bounds accordingly.
             */
            if (frameBounds.x < bounds.x) {
                frameBounds.width -= (bounds.x - frameBounds.x) * 2;
                frameBounds.x = bounds.x;
            }
            if (frameBounds.y < bounds.y) {
                frameBounds.height -= (bounds.y - frameBounds.y) * 2;
                frameBounds.y = bounds.y;
            }
            System.out.println("Adjusted Frame bounds: " + frameBounds);

            if (bounds.x + insets.left != frameBounds.x
                    || bounds.y + insets.top != frameBounds.y
                    || Math.abs(workAreaWidth - frameBounds.width) > MARGIN_TOLERANCE
                    || Math.abs(workAreaHeight - frameBounds.height) > MARGIN_TOLERANCE) {
                throw new RuntimeException("Test FAILED! Wrong screen #" +
                        screen + " insets: " + insets);
            }
        }
        System.out.println("Test PASSED!");
    }
}
