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.4 by pcg, Fri Mar 21 23:17:01 2003 UTC vs.
Revision 1.21 by pcg, Thu Sep 2 07:50:43 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:
24// - a vector template like simplevec or stl's vector
25// - defines for all watcher types required in your app
26// edit iom_conf.h as appropriate.
27#include "iom_conf.h"
23 28
24#include <sys/poll.h> 29#include "callback.h"
25 30
26#include "slog.h" 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
27 46
28typedef double tstamp; 47typedef double tstamp;
29
30extern tstamp NOW; 48extern tstamp NOW;
31 49
32template<class R, class A> class callback; 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
35 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)[this->size () - 1];
76 this->pop_back ();
77
78 if (this->size ())
79 if ((*this)[pos] = w)
80 w->active = pos + 1;
81 }
82};
83
84// only used as a namespace, and for initialisation purposes
36class io_manager { 85class io_manager {
37 vector<pollfd> pfs; 86 template<class watcher>
38 vector<io_watcher *> iow; 87 static void reg (watcher &w, io_manager_vec<watcher> &queue);
39 vector<time_watcher *> tw; // actually a heap
40 88
41 void idle_cb (tstamp &ts); time_watcher *idle; 89 template<class watcher>
90 static void unreg (watcher &w, io_manager_vec<watcher> &queue);
91
42public: 92public:
43
44 // register a watcher 93 // register a watcher
45 void reg (int fd, short events, io_watcher *w); 94#if IOM_IO
46 void unreg (io_watcher *w); 95 static void reg (io_watcher &w); static void unreg (io_watcher &w);
47 void reg (time_watcher *w); 96#endif
48 void unreg (time_watcher *w); 97#if IOM_TIME
98 static void reg (time_watcher &w); static void unreg (time_watcher &w);
99#endif
100#if IOM_CHECK
101 static void reg (check_watcher &w); static void unreg (check_watcher &w);
102#endif
103#if IOM_IDLE
104 static void reg (idle_watcher &w); static void unreg (idle_watcher &w);
105#endif
106#if IOM_SIG
107 static void reg (sig_watcher &w); static void unreg (sig_watcher &w);
108#endif
49 109
50 void loop (); 110 static void loop ();
51
52 io_manager ();
53 ~io_manager ();
54}; 111};
55 112
56extern io_manager iom; 113struct watcher {
114 int active; /* 0 == inactive, else index into respective vector */
57 115
116 watcher () : active (0) { }
117};
118
119#if IOM_IO
120enum { EVENT_READ = 1, EVENT_WRITE = 2 };
121
122struct io_watcher : watcher, callback2<void, io_watcher &, short> {
123 int fd;
124 short events;
125
126 void set (int fd_, short events_) { fd = fd_; events = events_; }
127
128 void set (short events_) { set (fd, events_); }
129 void start () { io_manager::reg (*this); }
130 void start (int fd_, short events_) { set (fd_, events_); io_manager::reg (*this); }
131 void stop () { io_manager::unreg (*this); }
132
58template<class R, class A> 133 template<class O1, class O2>
59class callback {
60 struct object { };
61
62 void *obj;
63 R (object::*meth)(A arg);
64
65 // a proxy is a kind of recipe on how to call a specific class method
66 struct proxy_base {
67 virtual R call (void *obj, void (object::*meth)(A), A arg) = 0;
68 };
69 template<class O1, class O2>
70 struct proxy : proxy_base {
71 virtual R call (void *obj, void (object::*meth)(A), A arg)
72 {
73 ((reinterpret_cast<O1 *>(obj)) ->* (reinterpret_cast<void (O2::*)(A)>(meth)))
74 (arg);
75 }
76 };
77
78 proxy_base *prxy;
79
80public:
81 template<class O1, class O2>
82 callback (O1 *object, void (O2::*method)(A))
83 {
84 static proxy<O1,O2> p;
85 obj = reinterpret_cast<void *>(object);
86 meth = reinterpret_cast<void (object::*)(A)>(method);
87 prxy = &p;
88 }
89
90 R call(A arg)
91 {
92 return prxy->call (obj, meth, arg);
93 }
94
95 R operator ()(A arg)
96 {
97 return call (arg);
98 }
99};
100
101struct io_watcher : callback<void, short> {
102 template<class O1, class O2>
103 io_watcher (O1 *object, void (O2::*method)(short revents)) 134 io_watcher (O1 *object, void (O2::*method) (io_watcher &, short))
135 : callback2<void, io_watcher &, short> (object,method)
136 { }
137 ~io_watcher () { stop (); }
138};
139#endif
140
141#if IOM_TIME
142struct time_watcher : watcher, callback1<void, time_watcher &> {
143 tstamp at;
144
145 void trigger ();
146
147 void set (tstamp when) { at = when; }
148 void operator () () { trigger (); }
149 void start () { io_manager::reg (*this); }
150 void start (tstamp when) { set (when); io_manager::reg (*this); }
151 void stop () { io_manager::unreg (*this); }
152
153 template<class O1, class O2>
154 time_watcher (O1 *object, void (O2::*method) (time_watcher &))
155 : callback1<void, time_watcher &> (object,method), at (0)
156 { }
157 ~time_watcher () { stop (); }
158};
159#endif
160
161#if IOM_CHECK
162// run before checking for new events
163struct check_watcher : watcher, callback1<void, check_watcher &> {
164 void start () { io_manager::reg (*this); }
165 void stop () { io_manager::unreg (*this); }
166
167 template<class O1, class O2>
168 check_watcher (O1 *object, void (O2::*method) (check_watcher &))
169 : callback1<void, check_watcher &> (object,method)
170 { }
171 ~check_watcher () { stop (); }
172};
173#endif
174
175#if IOM_IDLE
176// run after checking for any i/o, but before waiting
177struct idle_watcher : watcher, callback1<void, idle_watcher &> {
178 void start () { io_manager::reg (*this); }
179 void stop () { io_manager::unreg (*this); }
180
181 template<class O1, class O2>
182 idle_watcher (O1 *object, void (O2::*method) (idle_watcher &))
104 : callback<void, short>(object,method) 183 : callback1<void, idle_watcher &> (object,method)
105 { } 184 { }
106 185 ~idle_watcher () { stop (); }
107 void start (int fd, short events)
108 {
109 iom.reg (fd, events, this);
110 }
111
112 void stop ()
113 {
114 iom.unreg (this);
115 }
116}; 186};
187#endif
117 188
118#define TSTAMP_CANCEL -1. 189#if IOM_SIG
190struct sig_watcher : watcher, callback1<void, sig_watcher &> {
191 int signum;
119 192
120struct time_watcher : callback<void, tstamp &> { 193 void start (int signum);
121 tstamp at; 194 void stop () { io_manager::unreg (*this); }
122 195
123 template<class O1, class O2> 196 template<class O1, class O2>
124 time_watcher (O1 *object, void (O2::*method)(tstamp &)) 197 sig_watcher (O1 *object, void (O2::*method) (sig_watcher &))
125 : callback<void, tstamp &>(object,method) 198 : callback1<void, sig_watcher &> (object,method), signum (-1)
126 { } 199 { }
127 200 ~sig_watcher () { stop (); }
128 void set (tstamp when);
129 void trigger ();
130
131 void operator ()()
132 {
133 trigger ();
134 }
135
136 void start ();
137 void start (tstamp when)
138 {
139 set (when);
140 }
141
142 void stop ()
143 {
144 iom.unreg (this);
145 }
146
147 void reset (tstamp when = TSTAMP_CANCEL)
148 {
149 stop ();
150 at = when;
151 }
152}; 201};
153
154#endif 202#endif
155 203
204#endif
205

Diff Legend

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