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

Diff Legend

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