engagement_generation: engagement_generation/src/edu/wpi/hri/gen/policy/ref/DistractorSet.java Source File

00001 /*
00002  * Software License Agreement (BSD License)
00003  *
00004  *  Copyright (c) 2010, Worcester Polytechnic Institute
00005  *  All rights reserved.
00006  *
00007  *  Redistribution and use in source and binary forms, with or without
00008  *  modification, are permitted provided that the following conditions
00009  *  are met:
00010  *
00011  *   * Redistributions of source code must retain the above copyright
00012  *     notice, this list of conditions and the following disclaimer.
00013  *   * Redistributions in binary form must reproduce the above
00014  *     copyright notice, this list of conditions and the following
00015  *     disclaimer in the documentation and/or other materials provided
00016  *     with the distribution.
00017  *   * Neither the name of Worcester Polytechnic Institute. nor the names
00018  *     of its contributors may be used to endorse or promote products
00019  *     derived from this software without specific prior written permission.
00020  *
00021  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
00022  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
00023  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
00024  *  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
00025  *  COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
00026  *  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
00027  *  BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
00028  *  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
00029  *  CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
00030  *  LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
00031  *  ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
00032  *  POSSIBILITY OF SUCH DAMAGE.
00033  */
00034 package edu.wpi.hri.gen.policy.ref;
00035 
00036 import java.util.ArrayList;
00037 import java.util.List;
00038 
00039 import ros.pkg.bml_msgs.msg.Entity;
00040 
00050 public class DistractorSet {
00051 
00052         private final List<String> objects;
00053 
00061         public DistractorSet(Iterable<String> objs) {
00062                 this.objects = new ArrayList<String>();
00063                 for (String str : objs)
00064                         this.objects.add(str);
00065         }
00066 
00074         public DistractorSet(Entity[] objs) {
00075                 this.objects = new ArrayList<String>();
00076                 for (Entity ent : objs)
00077                         this.objects.add(ent.id);
00078         }
00079 
00085         public int size() {
00086                 return objects.size();
00087         }
00088 
00096         public boolean contains(String obj) {
00097                 return objects.contains(obj);
00098         }
00099 
00109         public DistractorSet intersection(DistractorSet other) {
00110                 List<String> inters = new ArrayList<String>();
00111                 for (String obj : objects)
00112                         if (other.objects.contains(obj))
00113                                 inters.add(obj);
00114 
00115                 return new DistractorSet(inters);
00116         }
00117 }