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

Diff Legend

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