add alternative labels in json output

tobyli · Aug 1, 2016 · 1fecd5f · 1fecd5f
1 parent 776a17b
commit 1fecd5f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/app/src/main/java/edu/cmu/hcii/sugilite/communication/json/SugiliteFilterJSON.java b/app/src/main/java/edu/cmu/hcii/sugilite/communication/json/SugiliteFilterJSON.java
@@ -2,6 +2,10 @@
 
 import android.graphics.Rect;
 
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
+
 import edu.cmu.hcii.sugilite.model.block.UIElementMatchingFilter;
 
 /**
@@ -21,6 +25,8 @@ public SugiliteFilterJSON(UIElementMatchingFilter filter){
                 this.parentFilter = new SugiliteFilterJSON(filter.getParentFilter());
             if(filter.getChildFilter() != null)
                 this.childFilter = new SugiliteFilterJSON(filter.getChildFilter());
+            if(filter.alternativeLabels != null && filter.alternativeLabels.size() > 0)
+                alternativeLabels = new HashSet<>(filter.alternativeLabels);
         }
     }
     public UIElementMatchingFilter toUIElementMatchingFilter(){
@@ -38,8 +44,11 @@ public UIElementMatchingFilter toUIElementMatchingFilter(){
             filter.setParentFilter(parentFilter.toUIElementMatchingFilter());
         if(childFilter != null)
             filter.setChildFilter(childFilter.toUIElementMatchingFilter());
+        if(alternativeLabels != null)
+            filter.alternativeLabels = new HashSet<>(alternativeLabels);
         return filter;
     }
     public String text, contentDescription, viewId, packageName, className, boundsInScreen, boundsInParent;
     public SugiliteFilterJSON parentFilter, childFilter;
+    public Set<Map.Entry<String, String>> alternativeLabels;
 }