engagement_recognition: engagement_recognition/src/recognition/objects/engagement

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 // @author Brett Ponsler (bponsler (at) wpi (dot) edu)
00035 #ifndef _ENGAGEMENT_OBJECTS_H
00036 #define _ENGAGEMENT_OBJECTS_H
00037 
00038 #include <vector>
00039 #include <string>
00040 #include "../events/event_sink.h"
00041 #include <bml_msgs/Entity.h>
00042 #include <engagement_msgs/Performance.h>
00043 
00044 namespace objects
00045 {
00046 
00053 bool areObjects(std::vector<bml_msgs::Entity> vec);
00054 
00062 bool isFace(std::vector<bml_msgs::Entity> vec, std::string id);
00063 
00070 bml_msgs::Entity objectToEntity(std::string object);
00071 
00078 std::vector<bml_msgs::Entity> getEntitiesFromObjects(std::vector<std::string> vec);
00079 
00086 std::vector<std::string> getObjectsFromEntities(std::vector<bml_msgs::Entity> vec);
00087 
00094 std::string toString(std::vector<std::string> vec);
00095 
00103 bool same(std::vector<std::string> vec1, std::vector<std::string> vec2);
00104 
00113 bool contains(std::vector<std::string> vec1, std::vector<std::string> vec2);
00114 
00122 bool intersect(std::vector<std::string> vec1, std::vector<std::string> vec2);
00123 
00132 std::vector<std::string> addObjects(std::vector<std::string> vec1,
00133                                     std::vector<std::string> vec2);
00134 
00143 std::vector<std::string> removeObjects(std::vector<std::string> vec1,
00144                                        std::vector<std::string> vec2);
00145 
00153 std::vector<std::string> highestProbability(std::vector<std::string> objects,
00154                                             std::vector<float> probs);
00155 
00161 void keepHighest(engagement_msgs::Performance &msg);
00162 
00170 bool pertains(EventSink *sink, std::vector<std::string> objects);
00171 
00172 } // objects namespace
00173 
00174 #endif //_ENGAGEMENT_OBJECTS_H
engagement_recognition/src/recognition/objects/engagement_objects.h
