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