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 java.io.PrintStream;
  26 import java.util.List;
  27 import org.jemmy.control.Wrap;
  28 import org.jemmy.control.Wrapper;
  29 import org.jemmy.env.Environment;
  30 
  31 /**
  32  *
  33  * @param <CONTROL>
  34  * @author shura
  35  */
  36 public class HierarchyLookup<CONTROL> extends AbstractLookup<CONTROL> {
  37 
  38     ControlHierarchy hierarchy;
  39 
  40     /**
  41      *
  42      * @param env
  43      * @param hierarchy
  44      * @param wrapper
  45      * @param controlClass
  46      * @param criteria
  47      */
  48     public HierarchyLookup(Environment env, ControlHierarchy hierarchy, Wrapper wrapper, Class<CONTROL> controlClass, LookupCriteria<CONTROL> criteria) {
  49         super(env, controlClass, criteria, wrapper);
  50         this.hierarchy = hierarchy;
  51     }
  52 
  53     /**
  54      *
  55      * @param env
  56      * @param hierarchy
  57      * @param controlClass
  58      * @param criteria
  59      */
  60     public HierarchyLookup(Environment env, ControlHierarchy hierarchy, Class<CONTROL> controlClass, LookupCriteria<CONTROL> criteria) {
  61         this(env, hierarchy, Wrap.getWrapper(), controlClass, criteria);
  62     }
  63 
  64     @Override
  65     List getChildren(Object subParent) {
  66         if(subParent != null) {
  67             return hierarchy.getChildren(subParent);
  68         } else {
  69             return hierarchy.getControls();
  70         }
  71     }
  72 
  73     /**
  74      * 
  75      * @param out
  76      * @param lookup
  77      */
  78     @Override
  79     protected void dump(PrintStream out, Lookup<? extends CONTROL> lookup) {
  80         int size = lookup.size();
  81         for (int i = 0; i < size; i++) {
  82             CONTROL object = lookup.get(i);
  83             dumpOne(out, object, calcPrefix(object));
  84         }
  85     }
  86 
  87     private String calcPrefix(CONTROL child) {
  88         String res = "";
  89         Object sp = child;
  90         while((sp = hierarchy.getParent(sp)) != null) {
  91             res += PREFIX_DELTA;
  92         }
  93         return res;
  94     }
  95 }