00001 /* 00002 * Copyright (c) 2008, Willow Garage, Inc. 00003 * All rights reserved. 00004 * 00005 * Redistribution and use in source and binary forms, with or without 00006 * modification, are permitted provided that the following conditions are met: 00007 * 00008 * * Redistributions of source code must retain the above copyright 00009 * notice, this list of conditions and the following disclaimer. 00010 * * Redistributions in binary form must reproduce the above copyright 00011 * notice, this list of conditions and the following disclaimer in the 00012 * documentation and/or other materials provided with the distribution. 00013 * * Neither the name of the Willow Garage, Inc. nor the names of its 00014 * contributors may be used to endorse or promote products derived from 00015 * this software without specific prior written permission. 00016 * 00017 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 00018 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 00019 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 00020 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE 00021 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 00022 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 00023 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 00024 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 00025 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 00026 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 00027 * POSSIBILITY OF SUCH DAMAGE. 00028 * 00029 */ 00030 00039 #ifndef GRAPH_MAPPING_UTILS_GENERAL_H 00040 #define GRAPH_MAPPING_UTILS_GENERAL_H 00041 00042 #include <ros/assert.h> 00043 #include <cstdlib> 00044 #include <boost/foreach.hpp> 00045 #include <map> 00046 #include <set> 00047 #include <stdexcept> 00048 00049 namespace graph_mapping_utils 00050 { 00051 00052 // Check if a container contains a key 00053 template <class Key, class Container> 00054 bool contains (const Container& container, const Key& key) 00055 { 00056 return container.find(key)!=container.end(); 00057 } 00058 00059 00060 // Get value corresponding to a key in a map, while asserting if it's not there 00061 template <class K, class V, class C, class A> 00062 const V& keyValue (const std::map<K, V, C, A>& m, const K& key) 00063 { 00064 typename std::map<K, V, C, A>::const_iterator pos = m.find(key); 00065 if (pos==m.end()) 00066 throw std::out_of_range("Map did not contain key it was expected to"); 00067 return pos->second; 00068 } 00069 00073 template <class T> 00074 std::set<T> sampleSubset (const std::set<T>& s, const int n) 00075 { 00076 if (n>(int)s.size() || n < 0) 00077 return s; 00078 else { 00079 std::set<T> s2; 00080 unsigned elements_left_to_sample = s.size(); 00081 unsigned elements_needed = n; 00082 BOOST_FOREACH (const T& x, s) { 00083 const double p = (double) elements_needed/elements_left_to_sample; 00084 const double q = (double)rand()/(double)RAND_MAX; 00085 if (q<p) { 00086 s2.insert(x); 00087 elements_needed--; 00088 } 00089 elements_left_to_sample--; 00090 } 00091 return s2; 00092 } 00093 } 00094 00095 } // namespace 00096 00097 #endif // include guard