00001 /* 00002 * Software License Agreement (BSD License) 00003 * 00004 * Copyright (c) 2008, Willow Garage, Inc. 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 Willow Garage, Inc. nor the names of its 00018 * contributors may be used to endorse or promote products derived 00019 * 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 00035 #ifndef ROSCPP_POLL_SET_H 00036 #define ROSCPP_POLL_SET_H 00037 00038 #include <vector> 00039 00040 #include <poll.h> 00041 00042 #include <boost/shared_ptr.hpp> 00043 #include <boost/function.hpp> 00044 #include <boost/thread/mutex.hpp> 00045 00046 namespace ros 00047 { 00048 00049 class Transport; 00050 typedef boost::shared_ptr<Transport> TransportPtr; 00051 00058 class PollSet 00059 { 00060 public: 00061 PollSet(); 00062 ~PollSet(); 00063 00064 typedef boost::function<void(int)> SocketUpdateFunc; 00074 bool addSocket(int sock, const SocketUpdateFunc& update_func, const TransportPtr& transport = TransportPtr()); 00081 bool delSocket(int sock); 00082 00090 bool addEvents(int sock, int events); 00098 bool delEvents(int sock, int events); 00099 00112 void update(int poll_timeout); 00113 00118 void signal(); 00119 00120 private: 00124 void createNativePollset(); 00125 00129 void onLocalPipeEvents(int events); 00130 00131 struct SocketInfo 00132 { 00133 TransportPtr transport_; 00134 SocketUpdateFunc func_; 00135 int fd_; 00136 int events_; 00137 }; 00138 typedef std::map<int, SocketInfo> M_SocketInfo; 00139 M_SocketInfo socket_info_; 00140 boost::mutex socket_info_mutex_; 00141 bool sockets_changed_; 00142 00143 boost::mutex just_deleted_mutex_; 00144 typedef std::vector<int> V_int; 00145 V_int just_deleted_; 00146 00147 std::vector<struct pollfd> ufds_; 00148 00149 boost::mutex signal_mutex_; 00150 int signal_pipe_[2]; 00151 }; 00152 00153 } 00154 00155 #endif // ROSCPP_POLL_SET_H