1 /* -----------------------------------------------------------------------------
2 * $Id: Schedule.h,v 1.27 2002/02/12 15:39:49 sof Exp $
4 * (c) The GHC Team 1998-1999
6 * Prototypes for functions in Schedule.c
7 * (RTS internal scheduler interface)
9 * -------------------------------------------------------------------------*/
10 #ifndef __SCHEDULE_H__
11 #define __SCHEDULE_H__
12 #include "OSThreads.h"
15 //* Scheduler Functions::
16 //* Scheduler Vars and Data Types::
17 //* Some convenient macros::
21 //@node Scheduler Functions, Scheduler Vars and Data Types
22 //@subsection Scheduler Functions
24 //@cindex initScheduler
25 //@cindex exitScheduler
26 /* initScheduler(), exitScheduler(), startTasks()
28 * Called from STG : no
29 * Locks assumed : none
31 extern void initScheduler ( void );
32 extern void exitScheduler ( void );
34 //@cindex awakenBlockedQueue
35 /* awakenBlockedQueue()
37 * Takes a pointer to the beginning of a blocked TSO queue, and
38 * wakes up the entire queue.
40 * Called from STG : yes
41 * Locks assumed : none
44 void awakenBlockedQueue(StgBlockingQueueElement *q, StgClosure *node);
46 void awakenBlockedQueue(StgBlockingQueueElement *q, StgClosure *node);
48 void awakenBlockedQueue(StgTSO *tso);
54 * Takes a pointer to the beginning of a blocked TSO queue, and
55 * removes the first thread, placing it on the runnable queue.
57 * Called from STG : yes
58 * Locks assumed : none
60 #if defined(GRAN) || defined(PAR)
61 StgBlockingQueueElement *unblockOne(StgBlockingQueueElement *bqe, StgClosure *node);
63 StgTSO *unblockOne(StgTSO *tso);
69 * Raises an exception asynchronously in the specified thread.
71 * Called from STG : yes
72 * Locks assumed : none
74 void raiseAsync(StgTSO *tso, StgClosure *exception);
79 * Raises an exception asynchronously in the specified thread.
81 * Called from STG : NO
82 * Locks assumed : sched_mutex
84 void awaitEvent(rtsBool wait); /* In Select.c */
86 /* wakeUpSleepingThreads(nat ticks)
88 * Wakes up any sleeping threads whose timers have expired.
90 * Called from STG : NO
91 * Locks assumed : sched_mutex
93 rtsBool wakeUpSleepingThreads(nat); /* In Select.c */
95 /* GetRoots(evac_fn f)
97 * Call f() for each root known to the scheduler.
99 * Called from STG : NO
100 * Locks assumed : ????
102 void GetRoots(evac_fn);
104 // ToDo: check whether all fcts below are used in the SMP version, too
105 //@cindex awaken_blocked_queue
107 void awaken_blocked_queue(StgBlockingQueueElement *q, StgClosure *node);
108 void unlink_from_bq(StgTSO* tso, StgClosure* node);
109 void initThread(StgTSO *tso, nat stack_size, StgInt pri);
111 nat run_queue_len(void);
112 void awaken_blocked_queue(StgBlockingQueueElement *q, StgClosure *node);
113 void initThread(StgTSO *tso, nat stack_size);
115 char *info_type(StgClosure *closure); // dummy
116 char *info_type_by_ip(StgInfoTable *ip); // dummy
117 void awaken_blocked_queue(StgTSO *q);
118 void initThread(StgTSO *tso, nat stack_size);
121 //@node Scheduler Vars and Data Types, Some convenient macros, Scheduler Functions
122 //@subsection Scheduler Vars and Data Types
124 //@cindex context_switch
125 /* Context switch flag.
126 * Locks required : sched_mutex
128 extern nat context_switch;
129 extern rtsBool interrupted;
132 extern nat timestamp;
134 /* Free capability list.
135 * Locks required: sched_mutex.
138 extern Capability *free_capabilities;
139 extern nat n_free_capabilities;
141 extern Capability MainCapability;
145 * Locks required : sched_mutex
147 * In GranSim we have one run/blocked_queue per PE.
150 // run_queue_hds defined in GranSim.h
152 extern StgTSO *run_queue_hd, *run_queue_tl;
153 extern StgTSO *blocked_queue_hd, *blocked_queue_tl;
154 extern StgTSO *sleeping_queue;
156 /* Linked list of all threads. */
157 extern StgTSO *all_threads;
159 #if defined(RTS_SUPPORTS_THREADS)
160 extern Mutex sched_mutex;
161 extern Condition thread_ready_cond;
163 extern Condition gc_pending_cond;
167 /* Called by shutdown_handler(). */
168 void interruptStgRts ( void );
170 void raiseAsync(StgTSO *tso, StgClosure *exception);
171 nat run_queue_len(void);
173 void resurrectThreads( StgTSO * );
175 //@node Some convenient macros, Index, Scheduler Vars and Data Types
176 //@subsection Some convenient macros
181 void printThreadBlockage(StgTSO *tso);
182 void printThreadStatus(StgTSO *tso);
183 void printAllThreads(void);
185 void print_bq (StgClosure *node);
187 void print_bqe (StgBlockingQueueElement *bqe);
190 /* -----------------------------------------------------------------------------
191 * Some convenient macros...
194 /* END_TSO_QUEUE and friends now defined in includes/StgMiscClosures.h */
196 //@cindex APPEND_TO_RUN_QUEUE
197 /* Add a thread to the end of the run queue.
198 * NOTE: tso->link should be END_TSO_QUEUE before calling this macro.
200 #define APPEND_TO_RUN_QUEUE(tso) \
201 ASSERT(tso->link == END_TSO_QUEUE); \
202 if (run_queue_hd == END_TSO_QUEUE) { \
203 run_queue_hd = tso; \
205 run_queue_tl->link = tso; \
209 //@cindex PUSH_ON_RUN_QUEUE
210 /* Push a thread on the beginning of the run queue. Used for
211 * newly awakened threads, so they get run as soon as possible.
213 #define PUSH_ON_RUN_QUEUE(tso) \
214 tso->link = run_queue_hd; \
215 run_queue_hd = tso; \
216 if (run_queue_tl == END_TSO_QUEUE) { \
217 run_queue_tl = tso; \
220 //@cindex POP_RUN_QUEUE
221 /* Pop the first thread off the runnable queue.
223 #define POP_RUN_QUEUE() \
224 ({ StgTSO *t = run_queue_hd; \
225 if (t != END_TSO_QUEUE) { \
226 run_queue_hd = t->link; \
227 t->link = END_TSO_QUEUE; \
228 if (run_queue_hd == END_TSO_QUEUE) { \
229 run_queue_tl = END_TSO_QUEUE; \
235 //@cindex APPEND_TO_BLOCKED_QUEUE
236 /* Add a thread to the end of the blocked queue.
238 #define APPEND_TO_BLOCKED_QUEUE(tso) \
239 ASSERT(tso->link == END_TSO_QUEUE); \
240 if (blocked_queue_hd == END_TSO_QUEUE) { \
241 blocked_queue_hd = tso; \
243 blocked_queue_tl->link = tso; \
245 blocked_queue_tl = tso;
247 //@cindex THREAD_RUNNABLE
248 /* Signal that a runnable thread has become available, in
249 * case there are any waiting tasks to execute it.
251 #if defined(RTS_SUPPORTS_THREADS)
252 #define THREAD_RUNNABLE() \
253 if ( !noCapabilities() ) { \
254 signalCondition(&thread_ready_cond); \
258 #define THREAD_RUNNABLE() /* nothing */
261 //@cindex EMPTY_RUN_QUEUE
262 /* Check whether the run queue is empty i.e. the PE is idle
264 #define EMPTY_RUN_QUEUE() (run_queue_hd == END_TSO_QUEUE)
265 #define EMPTY_QUEUE(q) (q == END_TSO_QUEUE)
267 #endif /* __SCHEDULE_H__ */
269 //@node Index, , Some convenient macros
273 //* APPEND_TO_BLOCKED_QUEUE:: @cindex\s-+APPEND_TO_BLOCKED_QUEUE
274 //* APPEND_TO_RUN_QUEUE:: @cindex\s-+APPEND_TO_RUN_QUEUE
275 //* POP_RUN_QUEUE :: @cindex\s-+POP_RUN_QUEUE
276 //* PUSH_ON_RUN_QUEUE:: @cindex\s-+PUSH_ON_RUN_QUEUE
277 //* awaitEvent:: @cindex\s-+awaitEvent
278 //* awakenBlockedQueue:: @cindex\s-+awakenBlockedQueue
279 //* awaken_blocked_queue:: @cindex\s-+awaken_blocked_queue
280 //* context_switch:: @cindex\s-+context_switch
281 //* exitScheduler:: @cindex\s-+exitScheduler
282 //* gc_pending_cond:: @cindex\s-+gc_pending_cond
283 //* initScheduler:: @cindex\s-+initScheduler
284 //* raiseAsync:: @cindex\s-+raiseAsync
285 //* startTasks:: @cindex\s-+startTasks
286 //* unblockOne:: @cindex\s-+unblockOne