6 #include <msp/core/systemerror.h>
7 #include <msp/strings/format.h>
8 #include <msp/time/units.h>
17 using namespace Msp::IO;
19 inline int sys_poll_event(PollEvent event)
23 if(event&~(P_INPUT|P_PRIO|P_OUTPUT))
24 throw invalid_argument("sys_poll_event");
38 inline PollEvent poll_event_from_sys(int event)
40 PollEvent result = P_NONE;
46 result = result|P_INPUT;
48 result = result|P_PRIO;
50 result = result|P_OUTPUT;
52 result = result|P_ERROR;
58 inline PollEvent do_poll(Base &obj, PollEvent pe, int timeout)
64 DWORD ret = WaitForSingleObject(obj.get_event_handle(), timeout);
65 if(ret==WAIT_OBJECT_0)
67 else if(ret==WAIT_FAILED)
68 throw system_error("WaitForSingleObject");
72 pollfd pfd = { obj.get_event_handle(), sys_poll_event(pe), 0 };
74 int ret = ::poll(&pfd, 1, timeout);
80 throw system_error("poll");
83 return poll_event_from_sys(pfd.revents);
93 struct Poller::Private
96 vector<HANDLE> handles;
108 void Poller::set_object(Base &obj, PollEvent ev)
110 // Verify that the object has an event handle
112 obj.get_event_handle();
114 EventMap::iterator i = objects.find(&obj);
127 if(objects.size()>=MAXIMUM_WAIT_OBJECTS)
128 throw logic_error("Maximum number of wait objects reached");
130 objects.insert(EventMap::value_type(&obj, ev));
141 int Poller::poll(const Time::TimeDelta &timeout)
143 if(timeout<Time::zero)
144 throw invalid_argument("Poller::poll");
146 return do_poll(static_cast<int>(timeout/Time::msec));
149 void Poller::rebuild_array()
152 priv->handles.clear();
154 for(EventMap::iterator i=objects.begin(); i!=objects.end(); ++i)
155 priv->handles.push_back(i->first->get_event_handle());
159 for(EventMap::iterator i=objects.begin(); i!=objects.end(); ++i)
162 p.fd = i->first->get_event_handle();
163 p.events = sys_poll_event(i->second);
164 priv->pfd.push_back(p);
168 objs_changed = false;
171 int Poller::do_poll(int timeout)
182 DWORD ret = WaitForMultipleObjects(priv->handles.size(), &priv->handles.front(), false, timeout);
183 if(/*ret>=WAIT_OBJECT_0 &&*/ ret<WAIT_OBJECT_0+priv->handles.size())
185 EventMap::iterator i = objects.begin();
186 advance(i, ret-WAIT_OBJECT_0);
187 poll_result.push_back(Slot(i->first, i->second));
191 else if(ret==WAIT_FAILED)
192 throw system_error("WaitForMultipleObjects");
196 int ret = ::poll(&priv->pfd.front(), priv->pfd.size(), timeout);
202 throw system_error("poll");
206 EventMap::iterator j = objects.begin();
207 for(vector<pollfd>::iterator i=priv->pfd.begin(); (i!=priv->pfd.end() && n>0); ++i, ++j)
210 poll_result.push_back(Slot(j->first, poll_event_from_sys(i->revents)));
219 PollEvent poll(Base &obj, PollEvent pe)
221 return do_poll(obj, pe, -1);
224 PollEvent poll(Base &obj, PollEvent pe, const Time::TimeDelta &timeout)
226 if(timeout<Time::zero)
227 throw invalid_argument("poll");
229 return do_poll(obj, pe, static_cast<int>(timeout/Time::msec));