ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/gvpe/src/iom.h
(Generate patch)

Comparing gvpe/src/iom.h (file contents):
Revision 1.9 by pcg, Wed Apr 2 05:15:00 2003 UTC vs.
Revision 1.18 by pcg, Sun Mar 21 12:50:05 2004 UTC

1/* 1/*
2 iom.h -- I/O multiplexor 2 iom.h -- generic I/O multiplexor
3 Copyright (C) 2003, 2004 Marc Lehmann <pcg@goof.com>
3 4
4 This program is free software; you can redistribute it and/or modify 5 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by 6 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 2 of the License, or 7 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version. 8 (at your option) any later version.
… …
14 You should have received a copy of the GNU General Public License 15 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software 16 along with this program; if not, write to the Free Software
16 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 17 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17*/ 18*/
18 19
19#ifndef VPE_IOM_H__ 20#ifndef IOM_H__
20#define VPE_IOM_H__ 21#define IOM_H__
21 22
22#include <vector> 23// required:
23 24// - a vector template like simplevec or stl's vector
24#include <sys/poll.h> 25// - defines for all watcher types required in your app
26// edit iom_conf.h as appropriate.
27#include "iom_conf.h"
25 28
26#include "callback.h" 29#include "callback.h"
27#include "slog.h" 30
31#ifndef IOM_IO
32# define IOM_IO 0
33#endif
34#ifndef IOM_TIME
35# define IOM_TIME 0
36#endif
37#ifndef IOM_CHECK
38# define IOM_CHECK 0
39#endif
40#ifndef IOM_IDLE
41# define IOM_IDLE 0
42#endif
28 43
29typedef double tstamp; 44typedef double tstamp;
30
31extern tstamp NOW; 45extern tstamp NOW;
32 46
47struct watcher;
48#if IOM_IO
33struct io_watcher; 49struct io_watcher;
50#endif
51#if IOM_TIME
34struct time_watcher; 52struct time_watcher;
53#endif
54#if IOM_CHECK
55struct check_watcher;
56#endif
57#if IOM_IDLE
58struct idle_watcher;
59#endif
60
61template<class watcher>
62struct io_manager_vec : vector<watcher *> {
63#if IOM_CHECK
64 bool activity;
65#endif
66
67 void erase_unordered (unsigned int pos)
68 {
69 watcher *w = (*this)[size () - 1];
70 pop_back ();
71
72 if (size ())
73 if ((*this)[pos] = w)
74 w->active = pos + 1;
75 }
76};
35 77
36class io_manager { 78class io_manager {
37 vector<pollfd> pfs; 79#if IOM_IO
38 vector<const io_watcher *> iow; 80 io_manager_vec<io_watcher> iow;
39 vector<time_watcher *> tw; // actually a heap 81#endif
82#if IOM_CHECK
83 io_manager_vec<check_watcher> cw;
84#endif
85#if IOM_TIME
86 io_manager_vec<time_watcher> tw;
87#endif
88#if IOM_IDLE
89 io_manager_vec<idle_watcher> iw;
90#endif
40 91
41 void idle_cb (tstamp &ts); time_watcher *idle; 92 template<class watcher>
93 void reg (watcher *w, io_manager_vec<watcher> &queue);
94
95 template<class watcher>
96 void unreg (watcher *w, io_manager_vec<watcher> &queue);
97
42public: 98public:
43
44 void reschedule_time_watchers ();
45
46 // register a watcher 99 // register a watcher
47 void reg (int fd, short events, io_watcher *w); 100#if IOM_IO
48 void unreg (io_watcher *w); 101 void reg (io_watcher *w); void unreg (io_watcher *w);
49 void reg (time_watcher *w); 102#endif
50 void unreg (time_watcher *w); 103#if IOM_TIME
104 void reg (time_watcher *w); void unreg (time_watcher *w);
105#endif
106#if IOM_CHECK
107 void reg (check_watcher *w); void unreg (check_watcher *w);
108#endif
109#if IOM_IDLE
110 void reg (idle_watcher *w); void unreg (idle_watcher *w);
111#endif
51 112
52 void loop (); 113 void loop ();
53 114
54 io_manager (); 115 io_manager ();
55 ~io_manager (); 116 ~io_manager ();
56}; 117};
57 118
58extern io_manager iom; 119extern io_manager iom; // a singleton, together with it's construction/destruction problems.
59 120
60struct io_watcher : callback2<void, int, short> { 121struct watcher {
61 template<class O1, class O2> 122 int active; /* 0 == inactive, else index into respective vector */
62 io_watcher (O1 *object, void (O2::*method)(int fd, short revents))
63 : callback2<void, int, short>(object,method)
64 { }
65 123
66 ~io_watcher () 124 watcher() : active(0) { }
67 {
68 iom.unreg (this);
69 }
70
71 void start (int fd, short events)
72 {
73 iom.reg (fd, events, this);
74 }
75
76 void stop ()
77 {
78 iom.unreg (this);
79 }
80}; 125};
81 126
82#define TSTAMP_CANCEL -1. 127#if IOM_IO
128enum { EVENT_READ = 1, EVENT_WRITE = 2 };
83 129
130struct io_watcher : watcher, callback2<void, io_watcher &, short> {
131 int fd;
132 short events;
133
134 void set (int fd_, short events_) { fd = fd_; events = events_; }
135
136 void set (short events_) { set (fd, events_); }
137 void start () { iom.reg (this); }
138 void start (int fd_, short events_) { set (fd_, events_); iom.reg (this); }
139 void stop () { iom.unreg (this); }
140
141 template<class O1, class O2>
142 io_watcher (O1 *object, void (O2::*method)(io_watcher &, short))
143 : callback2<void, io_watcher &, short>(object,method)
144 { }
145 ~io_watcher () { stop (); }
146};
147#endif
148
149#if IOM_TIME
84struct time_watcher : callback1<void, tstamp &> { 150struct time_watcher : watcher, callback1<void, time_watcher &> {
85 bool registered; // already registered?
86 tstamp at; 151 tstamp at;
87 152
88 template<class O1, class O2>
89 time_watcher (O1 *object, void (O2::*method)(tstamp &))
90 : callback1<void, tstamp &>(object,method)
91 , registered(false)
92 { }
93
94 ~time_watcher ()
95 {
96 iom.unreg (this);
97 }
98
99 void set (tstamp when);
100 void trigger (); 153 void trigger ();
101 154
102 void operator ()() 155 void set (tstamp when) { at = when; }
103 { 156 void operator ()() { trigger (); }
104 trigger (); 157 void start () { iom.reg (this); }
105 } 158 void start (tstamp when) { set (when); iom.reg (this); }
159 void stop () { iom.unreg (this); }
106 160
107 void start (); 161 template<class O1, class O2>
108 void start (tstamp when) 162 time_watcher (O1 *object, void (O2::*method)(time_watcher &))
109 { 163 : callback1<void, time_watcher &>(object,method), at(0)
110 set (when);
111 } 164 { }
165 ~time_watcher () { stop (); }
166};
167#endif
112 168
113 void stop () 169#if IOM_CHECK
114 { 170// run before checking for new events
115 iom.unreg (this); 171struct check_watcher : watcher, callback1<void, check_watcher &> {
116 } 172 void start () { iom.reg (this); }
173 void stop () { iom.unreg (this); }
117 174
118 void reset (tstamp when = TSTAMP_CANCEL) 175 template<class O1, class O2>
119 { 176 check_watcher (O1 *object, void (O2::*method)(check_watcher &))
120 stop (); 177 : callback1<void, check_watcher &>(object,method)
121 at = when;
122 } 178 { }
179 ~check_watcher () { stop (); }
123}; 180};
181#endif
182
183#if IOM_IDLE
184// run after checking for any i/o, but before waiting
185struct idle_watcher : watcher, callback1<void, idle_watcher &> {
186 void start () { iom.reg (this); }
187 void stop () { iom.unreg (this); }
188
189 template<class O1, class O2>
190 idle_watcher (O1 *object, void (O2::*method)(idle_watcher &))
191 : callback1<void, idle_watcher &>(object,method)
192 { }
193 ~idle_watcher () { stop (); }
194};
195#endif
124 196
125#endif 197#endif
126 198

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines