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.control;
  24 
  25 
  26 import java.util.HashMap;
  27 import org.jemmy.env.Environment;
  28 
  29 
  30 /**
  31  * This is an implementation of {@code Wrapper} which keeps a map between 
  32  * control classes and wrap classes.
  33  * @author shura
  34  */
  35 public class WrapperImpl extends AbstractWrapper {
  36 
  37     private HashMap<Class, Class<Wrap>> theWrappers;
  38 
  39     /**
  40      *
  41      * @param env
  42      */
  43     @SuppressWarnings("unchecked")
  44     public WrapperImpl(Environment env) {
  45         super(env);
  46         theWrappers = new HashMap<Class, Class<Wrap>>();
  47     }
  48 
  49     /**
  50      *
  51      * @param <P>
  52      * @param controlClass
  53      * @param wrapperClass
  54      */
  55     public <P> void add(Class controlClass, Class<Wrap> wrapperClass) {
  56         theWrappers.put(controlClass, wrapperClass);
  57         // TODO: Improve output
  58 //        getEnvironment().getOutput().println("Added \"" + wrapperClass.getName() + "\"" +
  59 //                " wrapper for \"" + controlClass.getName() + "\" control type");
  60     }
  61 
  62     @Override
  63     protected Class<Wrap> getWrapClass(Class controlClass) {
  64         return theWrappers.get(controlClass);
  65     }
  66 
  67 }