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.10 by pcg, Wed Apr 2 21:02:25 2003 UTC vs.
Revision 1.19 by pcg, Fri Apr 2 14:42:45 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
43#ifndef IOM_SIG
44# define IOM_SIG 0
45#endif
28 46
29typedef double tstamp; 47typedef double tstamp;
30
31extern tstamp NOW; 48extern tstamp NOW;
32 49
50struct watcher;
51#if IOM_IO
33struct io_watcher; 52struct io_watcher;
53#endif
54#if IOM_TIME
34struct time_watcher; 55struct time_watcher;
56#endif
57#if IOM_CHECK
58struct check_watcher;
59#endif
60#if IOM_IDLE
61struct idle_watcher;
62#endif
63#if IOM_SIG
64struct sig_watcher;
65#endif
66
67template<class watcher>
68struct io_manager_vec : vector<watcher *> {
69#if IOM_CHECK
70 bool activity;
71#endif
72
73 void erase_unordered (unsigned int pos)
74 {
75 watcher *w = (*this)[size () - 1];
76 pop_back ();
77
78 if (size ())
79 if ((*this)[pos] = w)
80 w->active = pos + 1;
81 }
82};
35 83
36class io_manager { 84class io_manager {
37 vector<pollfd> pfs; 85#if IOM_IO
38 vector<io_watcher *> iow; 86 io_manager_vec<io_watcher> iow;
39 vector<time_watcher *> tw; // actually a heap 87#endif
88#if IOM_CHECK
89 io_manager_vec<check_watcher> cw;
90#endif
91#if IOM_TIME
92 io_manager_vec<time_watcher> tw;
93#endif
94#if IOM_IDLE
95 io_manager_vec<idle_watcher> iw;
96#endif
97#if IOM_SIG
98 typedef io_manager_vec<sig_watcher> sig_vec;
99 vector<sig_vec *> sw;
100 static void sighandler (int signum);
101#endif
40 102
41 void idle_cb (time_watcher &w); time_watcher *idle; 103 template<class watcher>
104 void reg (watcher *w, io_manager_vec<watcher> &queue);
105
106 template<class watcher>
107 void unreg (watcher *w, io_manager_vec<watcher> &queue);
108
42public: 109public:
43
44 void reschedule_time_watchers ();
45
46 // register a watcher 110 // register a watcher
47 void reg (io_watcher *w); 111#if IOM_IO
48 void unreg (io_watcher *w); 112 void reg (io_watcher *w); void unreg (io_watcher *w);
49 void reg (time_watcher *w); 113#endif
50 void unreg (time_watcher *w); 114#if IOM_TIME
115 void reg (time_watcher *w); void unreg (time_watcher *w);
116#endif
117#if IOM_CHECK
118 void reg (check_watcher *w); void unreg (check_watcher *w);
119#endif
120#if IOM_IDLE
121 void reg (idle_watcher *w); void unreg (idle_watcher *w);
122#endif
123#if IOM_SIG
124 void reg (sig_watcher *w); void unreg (sig_watcher *w);
125#endif
51 126
52 void loop (); 127 void loop ();
53 128
54 io_manager (); 129 io_manager ();
55 ~io_manager (); 130 ~io_manager ();
56}; 131};
57 132
58extern io_manager iom; 133extern io_manager iom; // a singleton, together with it's construction/destruction problems.
59 134
135struct watcher {
136 int active; /* 0 == inactive, else index into respective vector */
137
138 watcher () : active (0) { }
139};
140
141#if IOM_IO
142enum { EVENT_READ = 1, EVENT_WRITE = 2 };
143
60struct io_watcher : callback2<void, io_watcher &, short> { 144struct io_watcher : watcher, callback2<void, io_watcher &, short> {
61 int fd; 145 int fd;
62 short events; 146 short events;
63 147
148 void set (int fd_, short events_) { fd = fd_; events = events_; }
149
150 void set (short events_) { set (fd, events_); }
151 void start () { iom.reg (this); }
152 void start (int fd_, short events_) { set (fd_, events_); iom.reg (this); }
153 void stop () { iom.unreg (this); }
154
64 template<class O1, class O2> 155 template<class O1, class O2>
65 io_watcher (O1 *object, void (O2::*method)(io_watcher &, short)) 156 io_watcher (O1 *object, void (O2::*method) (io_watcher &, short))
66 : callback2<void, io_watcher &, short>(object,method) 157 : callback2<void, io_watcher &, short> (object,method)
158 { }
159 ~io_watcher () { stop (); }
160};
161#endif
162
163#if IOM_TIME
164struct time_watcher : watcher, callback1<void, time_watcher &> {
165 tstamp at;
166
167 void trigger ();
168
169 void set (tstamp when) { at = when; }
170 void operator () () { trigger (); }
171 void start () { iom.reg (this); }
172 void start (tstamp when) { set (when); iom.reg (this); }
173 void stop () { iom.unreg (this); }
174
175 template<class O1, class O2>
176 time_watcher (O1 *object, void (O2::*method) (time_watcher &))
177 : callback1<void, time_watcher &> (object,method), at (0)
178 { }
179 ~time_watcher () { stop (); }
180};
181#endif
182
183#if IOM_CHECK
184// run before checking for new events
185struct check_watcher : watcher, callback1<void, check_watcher &> {
186 void start () { iom.reg (this); }
187 void stop () { iom.unreg (this); }
188
189 template<class O1, class O2>
190 check_watcher (O1 *object, void (O2::*method) (check_watcher &))
191 : callback1<void, check_watcher &> (object,method)
192 { }
193 ~check_watcher () { stop (); }
194};
195#endif
196
197#if IOM_IDLE
198// run after checking for any i/o, but before waiting
199struct idle_watcher : watcher, callback1<void, idle_watcher &> {
200 void start () { iom.reg (this); }
201 void stop () { iom.unreg (this); }
202
203 template<class O1, class O2>
204 idle_watcher (O1 *object, void (O2::*method) (idle_watcher &))
205 : callback1<void, idle_watcher &> (object,method)
67 { } 206 { }
68 207 ~idle_watcher () { stop (); }
69 ~io_watcher ()
70 {
71 iom.unreg (this);
72 }
73
74 void start (int fd_, short events_)
75 {
76 fd = fd_;
77 events = events_;
78 iom.reg (this);
79 }
80
81 void stop ()
82 {
83 iom.unreg (this);
84 }
85}; 208};
209#endif
86 210
87#define TSTAMP_CANCEL -1. 211#if IOM_SIG
88
89struct time_watcher : callback1<void, time_watcher &> { 212struct sig_watcher : watcher, callback1<void, sig_watcher &> {
90 bool registered; // already registered? 213 int signum;
91 tstamp at;
92 214
215 void start (int signum);
216 void stop () { iom.unreg (this); }
217
93 template<class O1, class O2> 218 template<class O1, class O2>
94 time_watcher (O1 *object, void (O2::*method)(time_watcher &)) 219 sig_watcher (O1 *object, void (O2::*method) (sig_watcher &))
95 : callback1<void, time_watcher &>(object,method) 220 : callback1<void, sig_watcher &> (object,method), signum (-1)
96 , registered(false)
97 { } 221 { }
98 222 ~sig_watcher () { stop (); }
99 ~time_watcher ()
100 {
101 iom.unreg (this);
102 }
103
104 void set (tstamp when);
105 void trigger ();
106
107 void operator ()()
108 {
109 trigger ();
110 }
111
112 void start ();
113 void start (tstamp when)
114 {
115 set (when);
116 }
117
118 void stop ()
119 {
120 iom.unreg (this);
121 }
122
123 void reset (tstamp when = TSTAMP_CANCEL)
124 {
125 stop ();
126 at = when;
127 }
128}; 223};
129
130#endif 224#endif
131 225
226#endif
227

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines