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

Diff Legend

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