1 /* 2 * Copyright (c) 2007, 2017 Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 package org.jemmy.lookup; 26 27 import org.jemmy.Rectangle; 28 import org.jemmy.control.Wrap; 29 import org.jemmy.control.Wrapper; 30 import org.jemmy.image.Image; 31 32 /** 33 * 34 * @author shura 35 */ 36 public class ImageLookup<T> implements LookupCriteria<T> { 37 private Wrapper wrapper; 38 private Image image; 39 private Class<T> type; 40 private Rectangle subArea; 41 private Image lastDiff = null; 42 private Image lastImage = null; 43 44 public ImageLookup(Wrapper wrapper, Class<T> type, Image image, Rectangle subArea) { 45 this.wrapper = wrapper; 46 this.type = type; 47 this.image = image; 48 this.subArea = subArea; 49 } 50 51 public ImageLookup(Wrapper wrapper, Class<T> type, Image image) { 52 this(wrapper, type, image, null); 53 } 54 55 public boolean check(T control) { 56 Wrap<?> wrap = wrapper.wrap(type, control); 57 if(subArea == null) { 58 lastImage = wrap.getScreenImage(); 59 } else { 60 lastImage = wrap.getScreenImage(subArea); 61 } 62 return (lastDiff = image.compareTo(lastImage)) == null; 63 } 64 65 public Image getLastDiff() { 66 return lastDiff; 67 } 68 69 public Image getLastImage() { 70 return lastImage; 71 } 72 }