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.8 by pcg, Wed Apr 2 03:06:22 2003 UTC vs.
Revision 1.26 by pcg, Wed Dec 15 02:59:55 2004 UTC

1/* 1/*
2 iom.h -- I/O multiplexor 2 iom.h -- generic I/O multiplexer
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
35 66
67template<class watcher>
68struct io_manager_vec : vector<watcher *> {
69 void erase_unordered (unsigned int pos)
70 {
71 watcher *w = (*this)[this->size () - 1];
72 this->pop_back ();
73
74 if (!this->empty ())
75 if (((*this)[pos] = w)) // '=' is correct!
76 w->active = pos + 1;
77 }
78};
79
80// only used as a namespace, and for initialisation purposes
36class io_manager { 81class io_manager {
37 vector<pollfd> pfs; 82 template<class watcher>
38 vector<const io_watcher *> iow; 83 static void reg (watcher &w, io_manager_vec<watcher> &queue);
39 vector<time_watcher *> tw; // actually a heap
40 84
41 void idle_cb (tstamp &ts); time_watcher *idle; 85 template<class watcher>
86 static void unreg (watcher &w, io_manager_vec<watcher> &queue);
87
42public: 88public:
89#if IOM_TIME
90 // fetch time only
91 static tstamp now ();
43 92
44 void reschedule_time_watchers (); 93 // set NOW
94 static void set_now ();
95#endif
45 96
46 // register a watcher 97 // register a watcher
47 void reg (int fd, short events, io_watcher *w); 98#if IOM_IO
48 void unreg (io_watcher *w); 99 static void reg (io_watcher &w); static void unreg (io_watcher &w);
49 void reg (time_watcher *w); 100#endif
50 void unreg (time_watcher *w); 101#if IOM_TIME
102 static void reg (time_watcher &w); static void unreg (time_watcher &w);
103#endif
104#if IOM_CHECK
105 static void reg (check_watcher &w); static void unreg (check_watcher &w);
106#endif
107#if IOM_IDLE
108 static void reg (idle_watcher &w); static void unreg (idle_watcher &w);
109#endif
110#if IOM_SIG
111 static void reg (sig_watcher &w); static void unreg (sig_watcher &w);
112#endif
51 113
52 void loop (); 114 static void loop ();
53
54 io_manager ();
55 ~io_manager ();
56}; 115};
57 116
58extern io_manager iom; 117struct watcher {
118 int active; /* 0 == inactive, else index into respective vector */
59 119
120 watcher () : active (0) { }
121};
122
123#if IOM_IO
124enum { EVENT_READ = 1, EVENT_WRITE = 2 };
125
60struct io_watcher : callback<void, short> { 126struct io_watcher : watcher, callback2<void, io_watcher &, short> {
127 int fd;
128 short events;
129
130 void set (int fd_, short events_) { fd = fd_; events = events_; }
131
132 void set (short events_) { set (fd, events_); }
133 void start () { io_manager::reg (*this); }
134 void start (int fd_, short events_) { set (fd_, events_); io_manager::reg (*this); }
135 void stop () { io_manager::unreg (*this); }
136
61 template<class O1, class O2> 137 template<class O1, class O2>
62 io_watcher (O1 *object, void (O2::*method)(short revents)) 138 io_watcher (O1 *object, void (O2::*method) (io_watcher &, short))
139 : callback2<void, io_watcher &, short> (object,method)
140 { }
141 ~io_watcher () { stop (); }
142};
143#endif
144
145#if IOM_TIME
146struct time_watcher : watcher, callback1<void, time_watcher &> {
147 tstamp at;
148
149 void trigger ();
150
151 void set (tstamp when) { at = when; }
152 void operator () () { trigger (); }
153 void start () { io_manager::reg (*this); }
154 void start (tstamp when) { set (when); io_manager::reg (*this); }
155 void stop () { io_manager::unreg (*this); }
156
157 template<class O1, class O2>
158 time_watcher (O1 *object, void (O2::*method) (time_watcher &))
159 : callback1<void, time_watcher &> (object,method), at (0)
160 { }
161 ~time_watcher () { stop (); }
162};
163#endif
164
165#if IOM_CHECK
166// run before checking for new events
167struct check_watcher : watcher, callback1<void, check_watcher &> {
168 void start () { io_manager::reg (*this); }
169 void stop () { io_manager::unreg (*this); }
170
171 template<class O1, class O2>
172 check_watcher (O1 *object, void (O2::*method) (check_watcher &))
173 : callback1<void, check_watcher &> (object,method)
174 { }
175 ~check_watcher () { stop (); }
176};
177#endif
178
179#if IOM_IDLE
180// run after checking for any i/o, but before waiting
181struct idle_watcher : watcher, callback1<void, idle_watcher &> {
182 void start () { io_manager::reg (*this); }
183 void stop () { io_manager::unreg (*this); }
184
185 template<class O1, class O2>
186 idle_watcher (O1 *object, void (O2::*method) (idle_watcher &))
63 : callback<void, short>(object,method) 187 : callback1<void, idle_watcher &> (object,method)
64 { } 188 { }
65 189 ~idle_watcher () { stop (); }
66 ~io_watcher ()
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}; 190};
191#endif
81 192
82#define TSTAMP_CANCEL -1. 193#if IOM_SIG
194struct sig_watcher : watcher, callback1<void, sig_watcher &> {
195 int signum;
83 196
84struct time_watcher : callback<void, tstamp &> { 197 void start (int signum);
85 bool registered; // already registered? 198 void stop () { io_manager::unreg (*this); }
86 tstamp at;
87 199
88 template<class O1, class O2> 200 template<class O1, class O2>
89 time_watcher (O1 *object, void (O2::*method)(tstamp &)) 201 sig_watcher (O1 *object, void (O2::*method) (sig_watcher &))
90 : callback<void, tstamp &>(object,method) 202 : callback1<void, sig_watcher &> (object,method), signum (-1)
91 , registered(false)
92 { } 203 { }
93 204 ~sig_watcher () { stop (); }
94 ~time_watcher ()
95 {
96 iom.unreg (this);
97 }
98
99 void set (tstamp when);
100 void trigger ();
101
102 void operator ()()
103 {
104 trigger ();
105 }
106
107 void start ();
108 void start (tstamp when)
109 {
110 set (when);
111 }
112
113 void stop ()
114 {
115 iom.unreg (this);
116 }
117
118 void reset (tstamp when = TSTAMP_CANCEL)
119 {
120 stop ();
121 at = when;
122 }
123}; 205};
124
125#endif 206#endif
126 207
208#endif
209

Diff Legend

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