--- gvpe/src/iom.h 2003/04/02 05:15:00 1.9 +++ gvpe/src/iom.h 2003/04/04 05:26:45 1.11 @@ -21,6 +21,8 @@ #include +#include + #include #include "callback.h" @@ -35,16 +37,16 @@ class io_manager { vector pfs; - vector iow; + vector iow; vector tw; // actually a heap - void idle_cb (tstamp &ts); time_watcher *idle; + void idle_cb (time_watcher &w); time_watcher *idle; public: void reschedule_time_watchers (); // register a watcher - void reg (int fd, short events, io_watcher *w); + void reg (io_watcher *w); void unreg (io_watcher *w); void reg (time_watcher *w); void unreg (time_watcher *w); @@ -55,22 +57,37 @@ ~io_manager (); }; -extern io_manager iom; +extern io_manager iom; // a singleton, together with it's construction/destruction problems. + +struct io_watcher : callback2 { + pollfd *p; -struct io_watcher : callback2 { template - io_watcher (O1 *object, void (O2::*method)(int fd, short revents)) - : callback2(object,method) + io_watcher (O1 *object, void (O2::*method)(io_watcher &, short)) + : callback2(object,method) { } - ~io_watcher () + ~io_watcher (); + + void set(int fd, short events) { - iom.unreg (this); + assert (p); + p->fd = fd; + p->events = events; + } + + void set(short events) + { + assert (p); + p->events = events; } void start (int fd, short events) { - iom.reg (fd, events, this); + iom.reg (this); // make sure pfd is set + + p->fd = fd; + p->events = events; } void stop () @@ -81,20 +98,17 @@ #define TSTAMP_CANCEL -1. -struct time_watcher : callback1 { +struct time_watcher : callback1 { bool registered; // already registered? tstamp at; template - time_watcher (O1 *object, void (O2::*method)(tstamp &)) - : callback1(object,method) + time_watcher (O1 *object, void (O2::*method)(time_watcher &)) + : callback1(object,method) , registered(false) { } - ~time_watcher () - { - iom.unreg (this); - } + ~time_watcher (); void set (tstamp when); void trigger (); @@ -118,6 +132,7 @@ void reset (tstamp when = TSTAMP_CANCEL) { stop (); + at = when; } };