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 }