jdk/test/sanity/client/lib/jemmy/src/org/netbeans/jemmy/image/ImageTool.java
changeset 36744 a00905527ec2
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/sanity/client/lib/jemmy/src/org/netbeans/jemmy/image/ImageTool.java	Wed Mar 30 19:05:58 2016 -0700
@@ -0,0 +1,199 @@
+/*
+ * Copyright (c) 1997, 2016, 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.
+ */
+package org.netbeans.jemmy.image;
+
+import java.awt.AWTException;
+import java.awt.Color;
+import java.awt.Component;
+import java.awt.Rectangle;
+import java.awt.Robot;
+import java.awt.Toolkit;
+import java.awt.image.BufferedImage;
+
+import org.netbeans.jemmy.JemmyException;
+
+/**
+ * Contains util methods to work with images.
+ *
+ * @author Alexandre Iline (alexandre.iline@oracle.com)
+ */
+public class ImageTool {
+
+    /**
+     * Gets an image from a rectange on screen.
+     *
+     * @param rect a rectangle on screen in absolute screen coordinates.
+     * @return a captured image.
+     */
+    public static BufferedImage getImage(Rectangle rect) {
+        try {
+            return new Robot().createScreenCapture(rect);
+        } catch (AWTException e) {
+            throw (new JemmyException("Exception during screen capturing", e));
+        }
+    }
+
+    /**
+     * Gets an image from a component.
+     *
+     * @param comp a visible component.
+     * @return a captured image.
+     */
+    public static BufferedImage getImage(Component comp) {
+        return (getImage(new Rectangle(comp.getLocationOnScreen(),
+                comp.getSize())));
+    }
+
+    /**
+     * Gets the whole screen image.
+     *
+     * @return a captured image.
+     */
+    public static BufferedImage getImage() {
+        return getImage(new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()));
+    }
+
+    /**
+     * Increases image.
+     *
+     * @param image an image to enlarge.
+     * @param zoom A scale.
+     * @return a result image.
+     */
+    public static BufferedImage enlargeImage(BufferedImage image, int zoom) {
+        int wight = image.getWidth();
+        int height = image.getHeight();
+        BufferedImage result = new BufferedImage(wight * zoom,
+                height * zoom,
+                image.getType());
+        int rgb;
+        for (int x = 0; x < wight; x++) {
+            for (int y = 0; y < height; y++) {
+                rgb = image.getRGB(x, y);
+                for (int i = 0; i < zoom; i++) {
+                    for (int j = 0; j < zoom; j++) {
+                        result.setRGB(x * zoom + i,
+                                y * zoom + j,
+                                rgb);
+                    }
+                }
+            }
+        }
+        return result;
+    }
+
+    /**
+     * @deprecated Use subtractImage(BufferedImage, BufferedImage) instead.
+     * @param minuend an image to subtract from.
+     * @param deduction an image to subtract.
+     * @return a result image.
+     */
+    @Deprecated
+    public static BufferedImage substractImage(BufferedImage minuend, BufferedImage deduction) {
+        return subtractImage(minuend, deduction);
+    }
+
+    /**
+     * Subtracts second image from first one. Could be used to save file
+     * difference for future analysis.
+     *
+     * @param minuend an image to subtract from.
+     * @param deduction an image to subtract.
+     * @return a result image.
+     */
+    public static BufferedImage subtractImage(BufferedImage minuend, BufferedImage deduction) {
+        return subtractImage(minuend, deduction, 0, 0);
+    }
+
+    /**
+     * @deprecated Use subtractImage(BufferedImage, BufferedImage, int, int)
+     * instead.
+     * @param minuend an image to subtract from.
+     * @param deduction an image to subtract.
+     * @return a result image.
+     */
+    @Deprecated
+    public static BufferedImage substractImage(BufferedImage minuend, BufferedImage deduction, int relativeX, int relativeY) {
+        return subtractImage(minuend, deduction, relativeX, relativeY);
+    }
+
+    /**
+     * Subtracts subimage from image. Could be used to save file difference for
+     * future analysis.
+     *
+     * @param minuend an image to subtract from.
+     * @param deduction an image to subtract.
+     * @param relativeX - deduction-in-minuend X coordinate
+     * @param relativeY - deduction-in-minuend Y coordinate
+     * @return a result image.
+     */
+    public static BufferedImage subtractImage(BufferedImage minuend, BufferedImage deduction, int relativeX, int relativeY) {
+        int mWidth = minuend.getWidth();
+        int mHeight = minuend.getHeight();
+        int dWidth = deduction.getWidth();
+        int dHeight = deduction.getHeight();
+
+        int maxWidth = (mWidth > relativeX + dWidth) ? mWidth : (relativeX + dWidth);
+        int maxHeight = (mHeight > relativeY + dHeight) ? mHeight : (relativeY + dHeight);
+
+        BufferedImage result = new BufferedImage(maxWidth, maxHeight, BufferedImage.TYPE_INT_RGB);
+        int mColor, dColor;
+        for (int x = 0; x < maxWidth; x++) {
+            for (int y = 0; y < maxHeight; y++) {
+                if (x >= mWidth
+                        || y >= mHeight) {
+                    mColor = 0;
+                } else {
+                    mColor = minuend.getRGB(x, y);
+                }
+                if (x >= dWidth + relativeX
+                        || y >= dHeight + relativeY
+                        || x < relativeX
+                        || y < relativeY) {
+                    dColor = 0;
+                } else {
+                    dColor = deduction.getRGB(x - relativeX, y - relativeY);
+                }
+                result.setRGB(x, y, subtractColors(mColor, dColor));
+            }
+        }
+        return result;
+    }
+
+    private static int subtractColors(int mRGB, int dRGB) {
+        Color mColor = new Color(mRGB);
+        Color dColor = new Color(dRGB);
+        int red = subtractColor(mColor.getRed(), dColor.getRed());
+        int green = subtractColor(mColor.getGreen(), dColor.getGreen());
+        int blue = subtractColor(mColor.getBlue(), dColor.getBlue());
+        return new Color(red, green, blue).getRGB();
+    }
+
+    private static int subtractColor(int mColor, int dColor) {
+        if (mColor >= dColor) {
+            return mColor - dColor;
+        } else {
+            return mColor - dColor + 0Xff;
+        }
+    }
+}