]> git.saurik.com Git - wxWidgets.git/blame - src/msw/thread.cpp
added support for wxALWAYS_SHOW_SB (finally closes patch 410865 -- first still opened...)
[wxWidgets.git] / src / msw / thread.cpp
CommitLineData
2bda0e17 1/////////////////////////////////////////////////////////////////////////////
9e84b847 2// Name: src/msw/thread.cpp
2bda0e17
KB
3// Purpose: wxThread Implementation
4// Author: Original from Wolfram Gloger/Guilhem Lavaux
bee503b0 5// Modified by: Vadim Zeitlin to make it work :-)
2bda0e17
KB
6// Created: 04/22/98
7// RCS-ID: $Id$
9e84b847
VZ
8// Copyright: (c) Wolfram Gloger (1996, 1997), Guilhem Lavaux (1998);
9// Vadim Zeitlin (1999-2002)
2bda0e17
KB
10// Licence: wxWindows licence
11/////////////////////////////////////////////////////////////////////////////
12
13#ifdef __GNUG__
3222fde2 14 #pragma implementation "thread.h"
2bda0e17
KB
15#endif
16
3222fde2
VZ
17// ----------------------------------------------------------------------------
18// headers
19// ----------------------------------------------------------------------------
20
a3b46648 21// For compilers that support precompilation, includes "wx.h".
2bda0e17
KB
22#include "wx/wxprec.h"
23
24#if defined(__BORLANDC__)
3222fde2 25 #pragma hdrstop
2bda0e17
KB
26#endif
27
28#ifndef WX_PRECOMP
60ab3272
VS
29 #include "wx/intl.h"
30 #include "wx/app.h"
2bda0e17
KB
31#endif
32
3222fde2
VZ
33#if wxUSE_THREADS
34
e2478fde
VZ
35#include "wx/apptrait.h"
36
0d0512bd 37#include "wx/msw/private.h"
17b439e8 38#include "wx/msw/missing.h"
3222fde2 39
2bda0e17
KB
40#include "wx/module.h"
41#include "wx/thread.h"
42
b568d04f
VZ
43// must have this symbol defined to get _beginthread/_endthread declarations
44#ifndef _MT
45 #define _MT
46#endif
47
9f51f2b6
VZ
48#if defined(__BORLANDC__)
49 #if !defined(__MT__)
50 // I can't set -tWM in the IDE (anyone?) so have to do this
51 #define __MT__
52 #endif
53
54 #if !defined(__MFC_COMPAT__)
55 // Needed to know about _beginthreadex etc..
56 #define __MFC_COMPAT__
57 #endif
58#endif // BC++
59
60// define wxUSE_BEGIN_THREAD if the compiler has _beginthreadex() function
61// which should be used instead of Win32 ::CreateThread() if possible
e0256755
GRG
62#if defined(__VISUALC__) || \
63 (defined(__BORLANDC__) && (__BORLANDC__ >= 0x500)) || \
9f51f2b6 64 (defined(__GNUG__) && defined(__MSVCRT__)) || \
ba14d986 65 defined(__WATCOMC__) || defined(__MWERKS__)
ccebc98a 66
9f51f2b6
VZ
67 #undef wxUSE_BEGIN_THREAD
68 #define wxUSE_BEGIN_THREAD
8536082d
BJ
69#endif
70
9f51f2b6
VZ
71#ifdef wxUSE_BEGIN_THREAD
72 // this is where _beginthreadex() is declared
b568d04f 73 #include <process.h>
9f51f2b6
VZ
74
75 // the return type of the thread function entry point
76 typedef unsigned THREAD_RETVAL;
77
78 // the calling convention of the thread function entry point
79 #define THREAD_CALLCONV __stdcall
80#else
81 // the settings for CreateThread()
82 typedef DWORD THREAD_RETVAL;
83 #define THREAD_CALLCONV WINAPI
b568d04f
VZ
84#endif
85
86// ----------------------------------------------------------------------------
87// constants
88// ----------------------------------------------------------------------------
89
bf1852e1
VZ
90// the possible states of the thread ("=>" shows all possible transitions from
91// this state)
92enum wxThreadState
93{
94 STATE_NEW, // didn't start execution yet (=> RUNNING)
95 STATE_RUNNING, // thread is running (=> PAUSED, CANCELED)
96 STATE_PAUSED, // thread is temporarily suspended (=> RUNNING)
97 STATE_CANCELED, // thread should terminate a.s.a.p. (=> EXITED)
98 STATE_EXITED // thread is terminating
2bda0e17
KB
99};
100
3222fde2 101// ----------------------------------------------------------------------------
b568d04f 102// this module globals
3222fde2 103// ----------------------------------------------------------------------------
2bda0e17 104
bf1852e1 105// TLS index of the slot where we store the pointer to the current thread
b568d04f 106static DWORD gs_tlsThisThread = 0xFFFFFFFF;
bf1852e1 107
3222fde2
VZ
108// id of the main thread - the one which can call GUI functions without first
109// calling wxMutexGuiEnter()
b568d04f 110static DWORD gs_idMainThread = 0;
bee503b0
VZ
111
112// if it's FALSE, some secondary thread is holding the GUI lock
b568d04f 113static bool gs_bGuiOwnedByMainThread = TRUE;
2bda0e17 114
3222fde2
VZ
115// critical section which controls access to all GUI functions: any secondary
116// thread (i.e. except the main one) must enter this crit section before doing
117// any GUI calls
b568d04f 118static wxCriticalSection *gs_critsectGui = NULL;
bee503b0 119
b568d04f
VZ
120// critical section which protects gs_nWaitingForGui variable
121static wxCriticalSection *gs_critsectWaitingForGui = NULL;
bee503b0
VZ
122
123// number of threads waiting for GUI in wxMutexGuiEnter()
b568d04f 124static size_t gs_nWaitingForGui = 0;
2bda0e17 125
bf1852e1 126// are we waiting for a thread termination?
b568d04f 127static bool gs_waitingForThread = FALSE;
bf1852e1 128
3222fde2 129// ============================================================================
9e84b847 130// Windows implementation of thread and related classes
3222fde2
VZ
131// ============================================================================
132
133// ----------------------------------------------------------------------------
9e84b847
VZ
134// wxCriticalSection
135// ----------------------------------------------------------------------------
136
137wxCriticalSection::wxCriticalSection()
138{
dac348b9 139 wxCOMPILE_TIME_ASSERT( sizeof(CRITICAL_SECTION) <= sizeof(wxCritSectBuffer),
9e84b847
VZ
140 wxCriticalSectionBufferTooSmall );
141
142 ::InitializeCriticalSection((CRITICAL_SECTION *)m_buffer);
143}
144
145wxCriticalSection::~wxCriticalSection()
146{
147 ::DeleteCriticalSection((CRITICAL_SECTION *)m_buffer);
148}
149
150void wxCriticalSection::Enter()
151{
152 ::EnterCriticalSection((CRITICAL_SECTION *)m_buffer);
153}
154
155void wxCriticalSection::Leave()
156{
157 ::LeaveCriticalSection((CRITICAL_SECTION *)m_buffer);
158}
159
160// ----------------------------------------------------------------------------
161// wxMutex
3222fde2 162// ----------------------------------------------------------------------------
0d0512bd 163
3222fde2
VZ
164class wxMutexInternal
165{
2bda0e17 166public:
9e84b847
VZ
167 wxMutexInternal(wxMutexType mutexType);
168 ~wxMutexInternal();
7f684264 169
9e84b847
VZ
170 bool IsOk() const { return m_mutex != NULL; }
171
172 wxMutexError Lock() { return LockTimeout(INFINITE); }
173 wxMutexError TryLock() { return LockTimeout(0); }
174 wxMutexError Unlock();
175
176private:
177 wxMutexError LockTimeout(DWORD milliseconds);
7f684264 178
7f684264 179 HANDLE m_mutex;
22f3361e
VZ
180
181 DECLARE_NO_COPY_CLASS(wxMutexInternal)
2bda0e17
KB
182};
183
9e84b847
VZ
184// all mutexes are recursive under Win32 so we don't use mutexType
185wxMutexInternal::wxMutexInternal(wxMutexType WXUNUSED(mutexType))
2bda0e17 186{
9e84b847
VZ
187 // create a nameless (hence intra process and always private) mutex
188 m_mutex = ::CreateMutex
189 (
190 NULL, // default secutiry attributes
191 FALSE, // not initially locked
192 NULL // no name
193 );
a6b0bd49 194
9e84b847
VZ
195 if ( !m_mutex )
196 {
197 wxLogLastError(_T("CreateMutex()"));
198 }
2bda0e17
KB
199}
200
9e84b847 201wxMutexInternal::~wxMutexInternal()
2bda0e17 202{
9e84b847 203 if ( m_mutex )
7f684264 204 {
9e84b847
VZ
205 if ( !::CloseHandle(m_mutex) )
206 {
207 wxLogLastError(_T("CloseHandle(mutex)"));
208 }
7f684264 209 }
2bda0e17
KB
210}
211
9e84b847 212wxMutexError wxMutexInternal::LockTimeout(DWORD milliseconds)
2bda0e17 213{
9e84b847
VZ
214 DWORD rc = ::WaitForSingleObject(m_mutex, milliseconds);
215 if ( rc == WAIT_ABANDONED )
3222fde2 216 {
9e84b847
VZ
217 // the previous caller died without releasing the mutex, but now we can
218 // really lock it
219 wxLogDebug(_T("WaitForSingleObject() returned WAIT_ABANDONED"));
220
221 // use 0 timeout, normally we should always get it
222 rc = ::WaitForSingleObject(m_mutex, 0);
223 }
3222fde2 224
9e84b847
VZ
225 switch ( rc )
226 {
3222fde2
VZ
227 case WAIT_OBJECT_0:
228 // ok
229 break;
2bda0e17 230
3222fde2 231 case WAIT_TIMEOUT:
9e84b847
VZ
232 return wxMUTEX_BUSY;
233
234 case WAIT_ABANDONED: // checked for above
3222fde2 235 default:
223d09f6 236 wxFAIL_MSG(wxT("impossible return value in wxMutex::Lock"));
9e84b847 237 // fall through
3222fde2 238
9e84b847
VZ
239 case WAIT_FAILED:
240 wxLogLastError(_T("WaitForSingleObject(mutex)"));
241 return wxMUTEX_MISC_ERROR;
242 }
2bda0e17 243
3222fde2 244 return wxMUTEX_NO_ERROR;
2bda0e17
KB
245}
246
9e84b847 247wxMutexError wxMutexInternal::Unlock()
2bda0e17 248{
9e84b847 249 if ( !::ReleaseMutex(m_mutex) )
3222fde2 250 {
2b5f62a0 251 wxLogLastError(_T("ReleaseMutex()"));
9e84b847 252
3222fde2
VZ
253 return wxMUTEX_MISC_ERROR;
254 }
a6b0bd49 255
3222fde2 256 return wxMUTEX_NO_ERROR;
2bda0e17
KB
257}
258
be809868 259// --------------------------------------------------------------------------
9e84b847 260// wxSemaphore
be809868
VZ
261// --------------------------------------------------------------------------
262
9e84b847 263// a trivial wrapper around Win32 semaphore
be809868 264class wxSemaphoreInternal
3222fde2 265{
2bda0e17 266public:
9e84b847 267 wxSemaphoreInternal(int initialcount, int maxcount);
be809868
VZ
268 ~wxSemaphoreInternal();
269
9e84b847 270 bool IsOk() const { return m_semaphore != NULL; }
be809868 271
9e84b847 272 wxSemaError Wait() { return WaitTimeout(INFINITE); }
918dc519
VZ
273
274 wxSemaError TryWait()
275 {
276 wxSemaError rc = WaitTimeout(0);
277 if ( rc == wxSEMA_TIMEOUT )
278 rc = wxSEMA_BUSY;
279
280 return rc;
281 }
282
9e84b847 283 wxSemaError WaitTimeout(unsigned long milliseconds);
be809868 284
9e84b847 285 wxSemaError Post();
be809868
VZ
286
287private:
288 HANDLE m_semaphore;
22f3361e
VZ
289
290 DECLARE_NO_COPY_CLASS(wxSemaphoreInternal)
be809868
VZ
291};
292
9e84b847 293wxSemaphoreInternal::wxSemaphoreInternal(int initialcount, int maxcount)
be809868
VZ
294{
295 if ( maxcount == 0 )
b568d04f 296 {
be809868
VZ
297 // make it practically infinite
298 maxcount = INT_MAX;
299 }
4d1c1c3c 300
9e84b847
VZ
301 m_semaphore = ::CreateSemaphore
302 (
303 NULL, // default security attributes
304 initialcount,
305 maxcount,
306 NULL // no name
307 );
308
be809868
VZ
309 if ( !m_semaphore )
310 {
311 wxLogLastError(_T("CreateSemaphore()"));
b568d04f 312 }
be809868
VZ
313}
314
315wxSemaphoreInternal::~wxSemaphoreInternal()
316{
9e84b847 317 if ( m_semaphore )
b568d04f 318 {
9e84b847
VZ
319 if ( !::CloseHandle(m_semaphore) )
320 {
321 wxLogLastError(_T("CloseHandle(semaphore)"));
322 }
be809868
VZ
323 }
324}
b568d04f 325
9e84b847 326wxSemaError wxSemaphoreInternal::WaitTimeout(unsigned long milliseconds)
be809868 327{
9e84b847 328 DWORD rc = ::WaitForSingleObject( m_semaphore, milliseconds );
be809868 329
9e84b847 330 switch ( rc )
be809868
VZ
331 {
332 case WAIT_OBJECT_0:
9e84b847 333 return wxSEMA_NO_ERROR;
b568d04f 334
be809868 335 case WAIT_TIMEOUT:
918dc519 336 return wxSEMA_TIMEOUT;
be809868
VZ
337
338 default:
9e84b847 339 wxLogLastError(_T("WaitForSingleObject(semaphore)"));
b568d04f
VZ
340 }
341
9e84b847 342 return wxSEMA_MISC_ERROR;
be809868
VZ
343}
344
9e84b847 345wxSemaError wxSemaphoreInternal::Post()
be809868 346{
9e84b847 347 if ( !::ReleaseSemaphore(m_semaphore, 1, NULL /* ptr to previous count */) )
4d1c1c3c 348 {
be809868 349 wxLogLastError(_T("ReleaseSemaphore"));
be809868 350
9e84b847
VZ
351 return wxSEMA_MISC_ERROR;
352 }
be809868 353
9e84b847 354 return wxSEMA_NO_ERROR;
be809868 355}
4d1c1c3c 356
be809868 357// --------------------------------------------------------------------------
9e84b847 358// wxCondition
be809868
VZ
359// --------------------------------------------------------------------------
360
9e84b847
VZ
361// Win32 doesn't have explicit support for the POSIX condition variables and
362// the Win32 events have quite different semantics, so we reimplement the
363// conditions from scratch using the mutexes and semaphores
be809868
VZ
364class wxConditionInternal
365{
366public:
c112e100 367 wxConditionInternal(wxMutex& mutex);
be809868 368
9e84b847 369 bool IsOk() const { return m_mutex.IsOk() && m_semaphore.IsOk(); }
be809868 370
9e84b847
VZ
371 wxCondError Wait();
372 wxCondError WaitTimeout(unsigned long milliseconds);
be809868 373
9e84b847
VZ
374 wxCondError Signal();
375 wxCondError Broadcast();
be809868
VZ
376
377private:
9e84b847
VZ
378 // the number of threads currently waiting for this condition
379 LONG m_numWaiters;
be809868 380
9e84b847
VZ
381 // the critical section protecting m_numWaiters
382 wxCriticalSection m_csWaiters;
be809868 383
9e84b847 384 wxMutex& m_mutex;
be809868
VZ
385 wxSemaphore m_semaphore;
386};
387
c112e100
VZ
388wxConditionInternal::wxConditionInternal(wxMutex& mutex)
389 : m_mutex(mutex)
be809868 390{
9e84b847
VZ
391 // another thread can't access it until we return from ctor, so no need to
392 // protect access to m_numWaiters here
be809868
VZ
393 m_numWaiters = 0;
394}
395
9e84b847 396wxCondError wxConditionInternal::Wait()
be809868
VZ
397{
398 // increment the number of waiters
9e84b847 399 ::InterlockedIncrement(&m_numWaiters);
be809868 400
c112e100 401 m_mutex.Unlock();
be809868
VZ
402
403 // a potential race condition can occur here
404 //
405 // after a thread increments nwaiters, and unlocks the mutex and before the
406 // semaphore.Wait() is called, if another thread can cause a signal to be
407 // generated
408 //
409 // this race condition is handled by using a semaphore and incrementing the
410 // semaphore only if 'nwaiters' is greater that zero since the semaphore,
411 // can 'remember' signals the race condition will not occur
412
413 // wait ( if necessary ) and decrement semaphore
9e84b847 414 wxSemaError err = m_semaphore.Wait();
c112e100 415 m_mutex.Lock();
9e84b847
VZ
416
417 return err == wxSEMA_NO_ERROR ? wxCOND_NO_ERROR : wxCOND_MISC_ERROR;
be809868
VZ
418}
419
9e84b847 420wxCondError wxConditionInternal::WaitTimeout(unsigned long milliseconds)
be809868 421{
9e84b847 422 ::InterlockedIncrement(&m_numWaiters);
be809868 423
c112e100 424 m_mutex.Unlock();
be809868
VZ
425
426 // a race condition can occur at this point in the code
427 //
428 // please see the comments in Wait(), for details
429
9e84b847 430 wxSemaError err = m_semaphore.WaitTimeout(milliseconds);
be809868 431
9e84b847 432 if ( err == wxSEMA_BUSY )
4d1c1c3c 433 {
be809868
VZ
434 // another potential race condition exists here it is caused when a
435 // 'waiting' thread timesout, and returns from WaitForSingleObject, but
436 // has not yet decremented 'nwaiters'.
437 //
438 // at this point if another thread calls signal() then the semaphore
439 // will be incremented, but the waiting thread will miss it.
440 //
441 // to handle this particular case, the waiting thread calls
442 // WaitForSingleObject again with a timeout of 0, after locking
443 // 'nwaiters_mutex'. this call does not block because of the zero
444 // timeout, but will allow the waiting thread to catch the missed
445 // signals.
9e84b847
VZ
446 wxCriticalSectionLocker lock(m_csWaiters);
447
448 err = m_semaphore.WaitTimeout(0);
be809868 449
9e84b847 450 if ( err != wxSEMA_NO_ERROR )
4d1c1c3c 451 {
be809868 452 m_numWaiters--;
4d1c1c3c
VZ
453 }
454 }
455
c112e100 456 m_mutex.Lock();
be809868 457
9e84b847 458 return err == wxSEMA_NO_ERROR ? wxCOND_NO_ERROR : wxCOND_MISC_ERROR;
be809868
VZ
459}
460
9e84b847 461wxCondError wxConditionInternal::Signal()
be809868 462{
9e84b847 463 wxCriticalSectionLocker lock(m_csWaiters);
be809868
VZ
464
465 if ( m_numWaiters > 0 )
b568d04f 466 {
be809868 467 // increment the semaphore by 1
9e84b847
VZ
468 if ( m_semaphore.Post() != wxSEMA_NO_ERROR )
469 return wxCOND_MISC_ERROR;
be809868
VZ
470
471 m_numWaiters--;
b568d04f
VZ
472 }
473
9e84b847 474 return wxCOND_NO_ERROR;
be809868 475}
4d1c1c3c 476
9e84b847 477wxCondError wxConditionInternal::Broadcast()
be809868 478{
9e84b847 479 wxCriticalSectionLocker lock(m_csWaiters);
2bda0e17 480
be809868
VZ
481 while ( m_numWaiters > 0 )
482 {
9e84b847
VZ
483 if ( m_semaphore.Post() != wxSEMA_NO_ERROR )
484 return wxCOND_MISC_ERROR;
485
be809868
VZ
486 m_numWaiters--;
487 }
488
9e84b847 489 return wxCOND_NO_ERROR;
3222fde2
VZ
490}
491
492// ----------------------------------------------------------------------------
493// wxThread implementation
494// ----------------------------------------------------------------------------
495
bf1852e1
VZ
496// wxThreadInternal class
497// ----------------------
498
3222fde2
VZ
499class wxThreadInternal
500{
2bda0e17 501public:
bf1852e1
VZ
502 wxThreadInternal()
503 {
504 m_hThread = 0;
505 m_state = STATE_NEW;
506 m_priority = WXTHREAD_DEFAULT_PRIORITY;
507 }
508
b568d04f
VZ
509 ~wxThreadInternal()
510 {
511 Free();
512 }
513
514 void Free()
515 {
516 if ( m_hThread )
517 {
518 if ( !::CloseHandle(m_hThread) )
519 {
f6bcfd97 520 wxLogLastError(wxT("CloseHandle(thread)"));
b568d04f
VZ
521 }
522
523 m_hThread = 0;
524 }
525 }
526
bf1852e1 527 // create a new (suspended) thread (for the given thread object)
6fe73788 528 bool Create(wxThread *thread, unsigned int stackSize);
bf1852e1
VZ
529
530 // suspend/resume/terminate
531 bool Suspend();
532 bool Resume();
533 void Cancel() { m_state = STATE_CANCELED; }
534
535 // thread state
536 void SetState(wxThreadState state) { m_state = state; }
537 wxThreadState GetState() const { return m_state; }
538
539 // thread priority
b568d04f 540 void SetPriority(unsigned int priority);
bf1852e1
VZ
541 unsigned int GetPriority() const { return m_priority; }
542
543 // thread handle and id
544 HANDLE GetHandle() const { return m_hThread; }
545 DWORD GetId() const { return m_tid; }
546
547 // thread function
9f51f2b6 548 static THREAD_RETVAL THREAD_CALLCONV WinThreadStart(void *thread);
2bda0e17 549
bf1852e1
VZ
550private:
551 HANDLE m_hThread; // handle of the thread
552 wxThreadState m_state; // state, see wxThreadState enum
553 unsigned int m_priority; // thread priority in "wx" units
554 DWORD m_tid; // thread id
22f3361e
VZ
555
556 DECLARE_NO_COPY_CLASS(wxThreadInternal)
2bda0e17
KB
557};
558
08d7397c 559THREAD_RETVAL THREAD_CALLCONV wxThreadInternal::WinThreadStart(void *param)
2bda0e17 560{
9f51f2b6 561 THREAD_RETVAL rc;
f6bcfd97
BP
562 bool wasCancelled;
563
564 // first of all, check whether we hadn't been cancelled already and don't
565 // start the user code at all then
9f51f2b6 566 wxThread *thread = (wxThread *)param;
696e1ea0
VZ
567 if ( thread->m_internal->GetState() == STATE_EXITED )
568 {
9f51f2b6 569 rc = (THREAD_RETVAL)-1;
f6bcfd97 570 wasCancelled = TRUE;
696e1ea0 571 }
f6bcfd97 572 else // do run thread
bf1852e1 573 {
f6bcfd97
BP
574 // store the thread object in the TLS
575 if ( !::TlsSetValue(gs_tlsThisThread, thread) )
576 {
577 wxLogSysError(_("Can not start thread: error writing TLS."));
bf1852e1 578
f6bcfd97
BP
579 return (DWORD)-1;
580 }
bf1852e1 581
9f51f2b6 582 rc = (THREAD_RETVAL)thread->Entry();
b568d04f 583
f6bcfd97
BP
584 // enter m_critsect before changing the thread state
585 thread->m_critsect.Enter();
586 wasCancelled = thread->m_internal->GetState() == STATE_CANCELED;
587 thread->m_internal->SetState(STATE_EXITED);
588 thread->m_critsect.Leave();
589 }
b568d04f 590
bee503b0 591 thread->OnExit();
2bda0e17 592
f6bcfd97 593 // if the thread was cancelled (from Delete()), then its handle is still
b568d04f
VZ
594 // needed there
595 if ( thread->IsDetached() && !wasCancelled )
596 {
597 // auto delete
598 delete thread;
599 }
600 //else: the joinable threads handle will be closed when Wait() is done
bf1852e1 601
b568d04f 602 return rc;
2bda0e17
KB
603}
604
b568d04f 605void wxThreadInternal::SetPriority(unsigned int priority)
2bda0e17 606{
b568d04f 607 m_priority = priority;
3222fde2 608
bf1852e1
VZ
609 // translate wxWindows priority to the Windows one
610 int win_priority;
611 if (m_priority <= 20)
612 win_priority = THREAD_PRIORITY_LOWEST;
613 else if (m_priority <= 40)
614 win_priority = THREAD_PRIORITY_BELOW_NORMAL;
615 else if (m_priority <= 60)
616 win_priority = THREAD_PRIORITY_NORMAL;
617 else if (m_priority <= 80)
618 win_priority = THREAD_PRIORITY_ABOVE_NORMAL;
619 else if (m_priority <= 100)
620 win_priority = THREAD_PRIORITY_HIGHEST;
3222fde2
VZ
621 else
622 {
223d09f6 623 wxFAIL_MSG(wxT("invalid value of thread priority parameter"));
bf1852e1 624 win_priority = THREAD_PRIORITY_NORMAL;
3222fde2
VZ
625 }
626
b568d04f 627 if ( !::SetThreadPriority(m_hThread, win_priority) )
bee503b0
VZ
628 {
629 wxLogSysError(_("Can't set thread priority"));
630 }
b568d04f
VZ
631}
632
6fe73788 633bool wxThreadInternal::Create(wxThread *thread, unsigned int stackSize)
b568d04f 634{
50ef4401
VZ
635 wxASSERT_MSG( m_state == STATE_NEW && !m_hThread,
636 _T("Create()ing thread twice?") );
637
b568d04f
VZ
638 // for compilers which have it, we should use C RTL function for thread
639 // creation instead of Win32 API one because otherwise we will have memory
640 // leaks if the thread uses C RTL (and most threads do)
9f51f2b6 641#ifdef wxUSE_BEGIN_THREAD
414c639c
VZ
642
643 // Watcom is reported to not like 0 stack size (which means "use default"
644 // for the other compilers and is also the default value for stackSize)
645#ifdef __WATCOMC__
646 if ( !stackSize )
647 stackSize = 10240;
648#endif // __WATCOMC__
649
9f51f2b6
VZ
650 m_hThread = (HANDLE)_beginthreadex
651 (
6fe73788
RL
652 NULL, // default security
653 stackSize,
654 wxThreadInternal::WinThreadStart, // entry point
655 thread,
656 CREATE_SUSPENDED,
657 (unsigned int *)&m_tid
9f51f2b6 658 );
611cb666 659#else // compiler doesn't have _beginthreadex
b568d04f
VZ
660 m_hThread = ::CreateThread
661 (
662 NULL, // default security
414c639c 663 stackSize, // stack size
9f51f2b6 664 wxThreadInternal::WinThreadStart, // thread entry point
b568d04f
VZ
665 (LPVOID)thread, // parameter
666 CREATE_SUSPENDED, // flags
667 &m_tid // [out] thread id
668 );
611cb666 669#endif // _beginthreadex/CreateThread
b568d04f
VZ
670
671 if ( m_hThread == NULL )
672 {
673 wxLogSysError(_("Can't create thread"));
674
675 return FALSE;
676 }
677
678 if ( m_priority != WXTHREAD_DEFAULT_PRIORITY )
679 {
680 SetPriority(m_priority);
681 }
3222fde2 682
bf1852e1 683 return TRUE;
2bda0e17
KB
684}
685
bf1852e1 686bool wxThreadInternal::Suspend()
2bda0e17 687{
bf1852e1
VZ
688 DWORD nSuspendCount = ::SuspendThread(m_hThread);
689 if ( nSuspendCount == (DWORD)-1 )
bee503b0 690 {
bf1852e1
VZ
691 wxLogSysError(_("Can not suspend thread %x"), m_hThread);
692
693 return FALSE;
bee503b0 694 }
2bda0e17 695
bf1852e1
VZ
696 m_state = STATE_PAUSED;
697
698 return TRUE;
a6b0bd49
VZ
699}
700
bf1852e1 701bool wxThreadInternal::Resume()
a6b0bd49 702{
bf1852e1 703 DWORD nSuspendCount = ::ResumeThread(m_hThread);
a6b0bd49
VZ
704 if ( nSuspendCount == (DWORD)-1 )
705 {
bf1852e1 706 wxLogSysError(_("Can not resume thread %x"), m_hThread);
a6b0bd49 707
bf1852e1 708 return FALSE;
a6b0bd49
VZ
709 }
710
f6bcfd97
BP
711 // don't change the state from STATE_EXITED because it's special and means
712 // we are going to terminate without running any user code - if we did it,
713 // the codei n Delete() wouldn't work
714 if ( m_state != STATE_EXITED )
715 {
716 m_state = STATE_RUNNING;
717 }
bee503b0 718
bf1852e1 719 return TRUE;
a6b0bd49
VZ
720}
721
bf1852e1
VZ
722// static functions
723// ----------------
724
725wxThread *wxThread::This()
a6b0bd49 726{
b568d04f 727 wxThread *thread = (wxThread *)::TlsGetValue(gs_tlsThisThread);
bf1852e1
VZ
728
729 // be careful, 0 may be a valid return value as well
730 if ( !thread && (::GetLastError() != NO_ERROR) )
a6b0bd49 731 {
bf1852e1 732 wxLogSysError(_("Couldn't get the current thread pointer"));
a6b0bd49 733
bf1852e1 734 // return NULL...
a6b0bd49
VZ
735 }
736
bf1852e1
VZ
737 return thread;
738}
739
740bool wxThread::IsMain()
741{
b568d04f 742 return ::GetCurrentThreadId() == gs_idMainThread;
bf1852e1
VZ
743}
744
c25a510b
JS
745#ifdef Yield
746#undef Yield
747#endif
748
bf1852e1
VZ
749void wxThread::Yield()
750{
b568d04f
VZ
751 // 0 argument to Sleep() is special and means to just give away the rest of
752 // our timeslice
bf1852e1
VZ
753 ::Sleep(0);
754}
755
756void wxThread::Sleep(unsigned long milliseconds)
757{
758 ::Sleep(milliseconds);
759}
760
ef8d96c2
VZ
761int wxThread::GetCPUCount()
762{
28a4627c
VZ
763 SYSTEM_INFO si;
764 GetSystemInfo(&si);
765
766 return si.dwNumberOfProcessors;
ef8d96c2
VZ
767}
768
4958ea8f
RD
769unsigned long wxThread::GetCurrentId()
770{
771 return (unsigned long)::GetCurrentThreadId();
772}
773
ef8d96c2
VZ
774bool wxThread::SetConcurrency(size_t level)
775{
28a4627c
VZ
776 wxASSERT_MSG( IsMain(), _T("should only be called from the main thread") );
777
ef8d96c2 778 // ok only for the default one
28a4627c
VZ
779 if ( level == 0 )
780 return 0;
781
782 // get system affinity mask first
783 HANDLE hProcess = ::GetCurrentProcess();
784 DWORD dwProcMask, dwSysMask;
785 if ( ::GetProcessAffinityMask(hProcess, &dwProcMask, &dwSysMask) == 0 )
786 {
787 wxLogLastError(_T("GetProcessAffinityMask"));
788
789 return FALSE;
790 }
791
792 // how many CPUs have we got?
793 if ( dwSysMask == 1 )
794 {
795 // don't bother with all this complicated stuff - on a single
796 // processor system it doesn't make much sense anyhow
797 return level == 1;
798 }
799
800 // calculate the process mask: it's a bit vector with one bit per
801 // processor; we want to schedule the process to run on first level
802 // CPUs
803 DWORD bit = 1;
804 while ( bit )
805 {
806 if ( dwSysMask & bit )
807 {
808 // ok, we can set this bit
809 dwProcMask |= bit;
810
811 // another process added
812 if ( !--level )
813 {
814 // and that's enough
815 break;
816 }
817 }
818
819 // next bit
820 bit <<= 1;
821 }
822
823 // could we set all bits?
824 if ( level != 0 )
825 {
826 wxLogDebug(_T("bad level %u in wxThread::SetConcurrency()"), level);
827
828 return FALSE;
829 }
830
831 // set it: we can't link to SetProcessAffinityMask() because it doesn't
832 // exist in Win9x, use RT binding instead
833
696e1ea0 834 typedef BOOL (*SETPROCESSAFFINITYMASK)(HANDLE, DWORD);
28a4627c
VZ
835
836 // can use static var because we're always in the main thread here
837 static SETPROCESSAFFINITYMASK pfnSetProcessAffinityMask = NULL;
838
839 if ( !pfnSetProcessAffinityMask )
840 {
841 HMODULE hModKernel = ::LoadLibrary(_T("kernel32"));
842 if ( hModKernel )
843 {
844 pfnSetProcessAffinityMask = (SETPROCESSAFFINITYMASK)
f6bcfd97 845 ::GetProcAddress(hModKernel, "SetProcessAffinityMask");
28a4627c
VZ
846 }
847
848 // we've discovered a MT version of Win9x!
849 wxASSERT_MSG( pfnSetProcessAffinityMask,
f6bcfd97 850 _T("this system has several CPUs but no SetProcessAffinityMask function?") );
28a4627c
VZ
851 }
852
853 if ( !pfnSetProcessAffinityMask )
854 {
855 // msg given above - do it only once
856 return FALSE;
857 }
858
696e1ea0 859 if ( pfnSetProcessAffinityMask(hProcess, dwProcMask) == 0 )
28a4627c
VZ
860 {
861 wxLogLastError(_T("SetProcessAffinityMask"));
862
863 return FALSE;
864 }
865
866 return TRUE;
ef8d96c2
VZ
867}
868
b568d04f
VZ
869// ctor and dtor
870// -------------
871
872wxThread::wxThread(wxThreadKind kind)
873{
9fc3ad34 874 m_internal = new wxThreadInternal();
b568d04f
VZ
875
876 m_isDetached = kind == wxTHREAD_DETACHED;
877}
878
879wxThread::~wxThread()
880{
9fc3ad34 881 delete m_internal;
b568d04f
VZ
882}
883
bf1852e1
VZ
884// create/start thread
885// -------------------
886
6fe73788 887wxThreadError wxThread::Create(unsigned int stackSize)
bf1852e1 888{
b568d04f
VZ
889 wxCriticalSectionLocker lock(m_critsect);
890
6fe73788 891 if ( !m_internal->Create(this, stackSize) )
bf1852e1 892 return wxTHREAD_NO_RESOURCE;
bee503b0 893
a6b0bd49 894 return wxTHREAD_NO_ERROR;
2bda0e17
KB
895}
896
bf1852e1 897wxThreadError wxThread::Run()
2bda0e17 898{
bf1852e1
VZ
899 wxCriticalSectionLocker lock(m_critsect);
900
9fc3ad34 901 if ( m_internal->GetState() != STATE_NEW )
bf1852e1
VZ
902 {
903 // actually, it may be almost any state at all, not only STATE_RUNNING
904 return wxTHREAD_RUNNING;
905 }
906
b568d04f 907 // the thread has just been created and is still suspended - let it run
bf1852e1 908 return Resume();
2bda0e17
KB
909}
910
bf1852e1
VZ
911// suspend/resume thread
912// ---------------------
913
914wxThreadError wxThread::Pause()
2bda0e17 915{
bf1852e1
VZ
916 wxCriticalSectionLocker lock(m_critsect);
917
9fc3ad34 918 return m_internal->Suspend() ? wxTHREAD_NO_ERROR : wxTHREAD_MISC_ERROR;
2bda0e17
KB
919}
920
bf1852e1 921wxThreadError wxThread::Resume()
2bda0e17 922{
bf1852e1
VZ
923 wxCriticalSectionLocker lock(m_critsect);
924
9fc3ad34 925 return m_internal->Resume() ? wxTHREAD_NO_ERROR : wxTHREAD_MISC_ERROR;
2bda0e17
KB
926}
927
bf1852e1
VZ
928// stopping thread
929// ---------------
930
b568d04f
VZ
931wxThread::ExitCode wxThread::Wait()
932{
933 // although under Windows we can wait for any thread, it's an error to
934 // wait for a detached one in wxWin API
935 wxCHECK_MSG( !IsDetached(), (ExitCode)-1,
936 _T("can't wait for detached thread") );
937
938 ExitCode rc = (ExitCode)-1;
939
940 (void)Delete(&rc);
941
9fc3ad34 942 m_internal->Free();
b568d04f
VZ
943
944 return rc;
945}
946
947wxThreadError wxThread::Delete(ExitCode *pRc)
2bda0e17 948{
bf1852e1
VZ
949 ExitCode rc = 0;
950
951 // Delete() is always safe to call, so consider all possible states
696e1ea0 952
f6bcfd97
BP
953 // we might need to resume the thread, but we might also not need to cancel
954 // it if it doesn't run yet
955 bool shouldResume = FALSE,
956 shouldCancel = TRUE,
957 isRunning = FALSE;
696e1ea0 958
f6bcfd97 959 // check if the thread already started to run
696e1ea0
VZ
960 {
961 wxCriticalSectionLocker lock(m_critsect);
962
963 if ( m_internal->GetState() == STATE_NEW )
964 {
f6bcfd97
BP
965 // WinThreadStart() will see it and terminate immediately, no need
966 // to cancel the thread - but we still need to resume it to let it
967 // run
696e1ea0
VZ
968 m_internal->SetState(STATE_EXITED);
969
f6bcfd97
BP
970 Resume(); // it knows about STATE_EXITED special case
971
972 shouldCancel = FALSE;
973 isRunning = TRUE;
974
975 // shouldResume is correctly set to FALSE here
976 }
977 else
978 {
979 shouldResume = IsPaused();
696e1ea0
VZ
980 }
981 }
982
f6bcfd97
BP
983 // resume the thread if it is paused
984 if ( shouldResume )
bf1852e1
VZ
985 Resume();
986
9fc3ad34 987 HANDLE hThread = m_internal->GetHandle();
b568d04f 988
0ef2ebba
VZ
989 // does is still run?
990 if ( isRunning || IsRunning() )
bf1852e1
VZ
991 {
992 if ( IsMain() )
993 {
994 // set flag for wxIsWaitingForThread()
b568d04f 995 gs_waitingForThread = TRUE;
bf1852e1
VZ
996 }
997
b568d04f 998 // ask the thread to terminate
f6bcfd97 999 if ( shouldCancel )
bf1852e1
VZ
1000 {
1001 wxCriticalSectionLocker lock(m_critsect);
1002
9fc3ad34 1003 m_internal->Cancel();
bf1852e1
VZ
1004 }
1005
1006 // we can't just wait for the thread to terminate because it might be
1007 // calling some GUI functions and so it will never terminate before we
1008 // process the Windows messages that result from these functions
e2478fde
VZ
1009 // (note that even in console applications we might have to process
1010 // messages if we use wxExecute() or timers or ...)
6cd4655b 1011 DWORD result = 0; // suppress warnings from broken compilers
bf1852e1
VZ
1012 do
1013 {
14205189
VZ
1014 if ( IsMain() )
1015 {
1016 // give the thread we're waiting for chance to do the GUI call
1017 // it might be in
1018 if ( (gs_nWaitingForGui > 0) && wxGuiOwnedByMainThread() )
1019 {
1020 wxMutexGuiLeave();
1021 }
1022 }
1023
bf1852e1
VZ
1024 result = ::MsgWaitForMultipleObjects
1025 (
1026 1, // number of objects to wait for
1027 &hThread, // the objects
1028 FALSE, // don't wait for all objects
1029 INFINITE, // no timeout
14205189
VZ
1030 QS_ALLINPUT | // return as soon as there are any events
1031 QS_ALLPOSTMESSAGE
bf1852e1
VZ
1032 );
1033
1034 switch ( result )
1035 {
1036 case 0xFFFFFFFF:
1037 // error
1038 wxLogSysError(_("Can not wait for thread termination"));
1039 Kill();
b568d04f 1040 return wxTHREAD_KILLED;
bf1852e1
VZ
1041
1042 case WAIT_OBJECT_0:
1043 // thread we're waiting for terminated
1044 break;
1045
1046 case WAIT_OBJECT_0 + 1:
0ef2ebba 1047 // new message arrived, process it
bf1852e1 1048 {
e2478fde
VZ
1049 wxAppTraits *traits = wxTheApp ? wxTheApp->GetTraits()
1050 : NULL;
1051
1052 if ( traits && !traits->DoMessageFromThreadWait() )
1053 {
1054 // WM_QUIT received: kill the thread
1055 Kill();
0ef2ebba 1056
e2478fde
VZ
1057 return wxTHREAD_KILLED;
1058 }
bf1852e1 1059 }
bf1852e1
VZ
1060 break;
1061
1062 default:
223d09f6 1063 wxFAIL_MSG(wxT("unexpected result of MsgWaitForMultipleObject"));
bf1852e1
VZ
1064 }
1065 } while ( result != WAIT_OBJECT_0 );
1066
1067 if ( IsMain() )
1068 {
b568d04f 1069 gs_waitingForThread = FALSE;
bf1852e1 1070 }
b568d04f 1071 }
bf1852e1 1072
10671c67
VZ
1073 // although the thread might be already in the EXITED state it might not
1074 // have terminated yet and so we are not sure that it has actually
1075 // terminated if the "if" above hadn't been taken
1076 do
b568d04f 1077 {
10671c67
VZ
1078 if ( !::GetExitCodeThread(hThread, (LPDWORD)&rc) )
1079 {
1080 wxLogLastError(wxT("GetExitCodeThread"));
bf1852e1 1081
10671c67
VZ
1082 rc = (ExitCode)-1;
1083 }
1084 } while ( (DWORD)rc == STILL_ACTIVE );
bf1852e1 1085
b568d04f
VZ
1086 if ( IsDetached() )
1087 {
1088 // if the thread exits normally, this is done in WinThreadStart, but in
1089 // this case it would have been too early because
f6bcfd97 1090 // MsgWaitForMultipleObject() would fail if the thread handle was
b568d04f
VZ
1091 // closed while we were waiting on it, so we must do it here
1092 delete this;
bf1852e1
VZ
1093 }
1094
b568d04f
VZ
1095 if ( pRc )
1096 *pRc = rc;
1097
1098 return rc == (ExitCode)-1 ? wxTHREAD_MISC_ERROR : wxTHREAD_NO_ERROR;
2bda0e17
KB
1099}
1100
bf1852e1 1101wxThreadError wxThread::Kill()
2bda0e17 1102{
bf1852e1
VZ
1103 if ( !IsRunning() )
1104 return wxTHREAD_NOT_RUNNING;
1105
9fc3ad34 1106 if ( !::TerminateThread(m_internal->GetHandle(), (DWORD)-1) )
bf1852e1
VZ
1107 {
1108 wxLogSysError(_("Couldn't terminate thread"));
1109
1110 return wxTHREAD_MISC_ERROR;
1111 }
1112
9fc3ad34 1113 m_internal->Free();
b568d04f
VZ
1114
1115 if ( IsDetached() )
1116 {
1117 delete this;
1118 }
bf1852e1
VZ
1119
1120 return wxTHREAD_NO_ERROR;
2bda0e17
KB
1121}
1122
b568d04f 1123void wxThread::Exit(ExitCode status)
2bda0e17 1124{
9fc3ad34 1125 m_internal->Free();
2bda0e17 1126
b568d04f
VZ
1127 if ( IsDetached() )
1128 {
1129 delete this;
1130 }
1131
9f51f2b6 1132#ifdef wxUSE_BEGIN_THREAD
b568d04f
VZ
1133 _endthreadex((unsigned)status);
1134#else // !VC++
bf1852e1 1135 ::ExitThread((DWORD)status);
b568d04f 1136#endif // VC++/!VC++
2bda0e17 1137
223d09f6 1138 wxFAIL_MSG(wxT("Couldn't return from ExitThread()!"));
bf1852e1 1139}
2bda0e17 1140
b568d04f
VZ
1141// priority setting
1142// ----------------
1143
bf1852e1
VZ
1144void wxThread::SetPriority(unsigned int prio)
1145{
1146 wxCriticalSectionLocker lock(m_critsect);
1147
9fc3ad34 1148 m_internal->SetPriority(prio);
bf1852e1 1149}
2bda0e17 1150
bf1852e1
VZ
1151unsigned int wxThread::GetPriority() const
1152{
b568d04f 1153 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
2bda0e17 1154
9fc3ad34 1155 return m_internal->GetPriority();
2bda0e17
KB
1156}
1157
b568d04f 1158unsigned long wxThread::GetId() const
2bda0e17 1159{
b568d04f 1160 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
bf1852e1 1161
9fc3ad34 1162 return (unsigned long)m_internal->GetId();
2bda0e17
KB
1163}
1164
72fd19a1
JS
1165bool wxThread::IsRunning() const
1166{
b568d04f 1167 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
bf1852e1 1168
9fc3ad34 1169 return m_internal->GetState() == STATE_RUNNING;
72fd19a1
JS
1170}
1171
1172bool wxThread::IsAlive() const
1173{
b568d04f 1174 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
bf1852e1 1175
9fc3ad34
VZ
1176 return (m_internal->GetState() == STATE_RUNNING) ||
1177 (m_internal->GetState() == STATE_PAUSED);
72fd19a1
JS
1178}
1179
a737331d
GL
1180bool wxThread::IsPaused() const
1181{
b568d04f 1182 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
a737331d 1183
9fc3ad34 1184 return m_internal->GetState() == STATE_PAUSED;
a737331d
GL
1185}
1186
8c10faf1 1187bool wxThread::TestDestroy()
2bda0e17 1188{
b568d04f 1189 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
bf1852e1 1190
9fc3ad34 1191 return m_internal->GetState() == STATE_CANCELED;
2bda0e17
KB
1192}
1193
3222fde2
VZ
1194// ----------------------------------------------------------------------------
1195// Automatic initialization for thread module
1196// ----------------------------------------------------------------------------
2bda0e17 1197
3222fde2 1198class wxThreadModule : public wxModule
a6b0bd49 1199{
3222fde2
VZ
1200public:
1201 virtual bool OnInit();
1202 virtual void OnExit();
d524867f 1203
3222fde2
VZ
1204private:
1205 DECLARE_DYNAMIC_CLASS(wxThreadModule)
1206};
d524867f
RR
1207
1208IMPLEMENT_DYNAMIC_CLASS(wxThreadModule, wxModule)
1209
3222fde2 1210bool wxThreadModule::OnInit()
d524867f 1211{
bf1852e1 1212 // allocate TLS index for storing the pointer to the current thread
b568d04f
VZ
1213 gs_tlsThisThread = ::TlsAlloc();
1214 if ( gs_tlsThisThread == 0xFFFFFFFF )
bf1852e1
VZ
1215 {
1216 // in normal circumstances it will only happen if all other
1217 // TLS_MINIMUM_AVAILABLE (>= 64) indices are already taken - in other
1218 // words, this should never happen
f6bcfd97 1219 wxLogSysError(_("Thread module initialization failed: impossible to allocate index in thread local storage"));
bf1852e1
VZ
1220
1221 return FALSE;
1222 }
1223
1224 // main thread doesn't have associated wxThread object, so store 0 in the
1225 // TLS instead
b568d04f 1226 if ( !::TlsSetValue(gs_tlsThisThread, (LPVOID)0) )
bf1852e1 1227 {
b568d04f
VZ
1228 ::TlsFree(gs_tlsThisThread);
1229 gs_tlsThisThread = 0xFFFFFFFF;
bf1852e1 1230
f6bcfd97 1231 wxLogSysError(_("Thread module initialization failed: can not store value in thread local storage"));
bf1852e1
VZ
1232
1233 return FALSE;
1234 }
1235
b568d04f 1236 gs_critsectWaitingForGui = new wxCriticalSection();
bee503b0 1237
b568d04f
VZ
1238 gs_critsectGui = new wxCriticalSection();
1239 gs_critsectGui->Enter();
bee503b0 1240
bf1852e1 1241 // no error return for GetCurrentThreadId()
b568d04f 1242 gs_idMainThread = ::GetCurrentThreadId();
3222fde2 1243
d524867f
RR
1244 return TRUE;
1245}
1246
3222fde2 1247void wxThreadModule::OnExit()
d524867f 1248{
b568d04f 1249 if ( !::TlsFree(gs_tlsThisThread) )
bf1852e1 1250 {
f6bcfd97 1251 wxLogLastError(wxT("TlsFree failed."));
bf1852e1
VZ
1252 }
1253
b568d04f 1254 if ( gs_critsectGui )
3222fde2 1255 {
b568d04f
VZ
1256 gs_critsectGui->Leave();
1257 delete gs_critsectGui;
1258 gs_critsectGui = NULL;
3222fde2 1259 }
bee503b0 1260
b568d04f
VZ
1261 delete gs_critsectWaitingForGui;
1262 gs_critsectWaitingForGui = NULL;
3222fde2
VZ
1263}
1264
bee503b0 1265// ----------------------------------------------------------------------------
b568d04f 1266// under Windows, these functions are implemented using a critical section and
3222fde2 1267// not a mutex, so the names are a bit confusing
bee503b0
VZ
1268// ----------------------------------------------------------------------------
1269
0c2f35dc 1270void WXDLLIMPEXP_BASE wxMutexGuiEnter()
3222fde2 1271{
bee503b0
VZ
1272 // this would dead lock everything...
1273 wxASSERT_MSG( !wxThread::IsMain(),
223d09f6 1274 wxT("main thread doesn't want to block in wxMutexGuiEnter()!") );
bee503b0
VZ
1275
1276 // the order in which we enter the critical sections here is crucial!!
1277
1278 // set the flag telling to the main thread that we want to do some GUI
1279 {
b568d04f 1280 wxCriticalSectionLocker enter(*gs_critsectWaitingForGui);
bee503b0 1281
b568d04f 1282 gs_nWaitingForGui++;
bee503b0
VZ
1283 }
1284
1285 wxWakeUpMainThread();
1286
1287 // now we may block here because the main thread will soon let us in
1288 // (during the next iteration of OnIdle())
b568d04f 1289 gs_critsectGui->Enter();
3222fde2
VZ
1290}
1291
0c2f35dc 1292void WXDLLIMPEXP_BASE wxMutexGuiLeave()
3222fde2 1293{
b568d04f 1294 wxCriticalSectionLocker enter(*gs_critsectWaitingForGui);
bee503b0
VZ
1295
1296 if ( wxThread::IsMain() )
1297 {
b568d04f 1298 gs_bGuiOwnedByMainThread = FALSE;
bee503b0
VZ
1299 }
1300 else
1301 {
4d1c1c3c 1302 // decrement the number of threads waiting for GUI access now
b568d04f 1303 wxASSERT_MSG( gs_nWaitingForGui > 0,
223d09f6 1304 wxT("calling wxMutexGuiLeave() without entering it first?") );
bee503b0 1305
b568d04f 1306 gs_nWaitingForGui--;
bee503b0
VZ
1307
1308 wxWakeUpMainThread();
1309 }
1310
b568d04f 1311 gs_critsectGui->Leave();
bee503b0
VZ
1312}
1313
0c2f35dc 1314void WXDLLIMPEXP_BASE wxMutexGuiLeaveOrEnter()
bee503b0
VZ
1315{
1316 wxASSERT_MSG( wxThread::IsMain(),
223d09f6 1317 wxT("only main thread may call wxMutexGuiLeaveOrEnter()!") );
bee503b0 1318
b568d04f 1319 wxCriticalSectionLocker enter(*gs_critsectWaitingForGui);
bee503b0 1320
b568d04f 1321 if ( gs_nWaitingForGui == 0 )
bee503b0
VZ
1322 {
1323 // no threads are waiting for GUI - so we may acquire the lock without
1324 // any danger (but only if we don't already have it)
1325 if ( !wxGuiOwnedByMainThread() )
1326 {
b568d04f 1327 gs_critsectGui->Enter();
bee503b0 1328
b568d04f 1329 gs_bGuiOwnedByMainThread = TRUE;
bee503b0
VZ
1330 }
1331 //else: already have it, nothing to do
1332 }
1333 else
1334 {
1335 // some threads are waiting, release the GUI lock if we have it
1336 if ( wxGuiOwnedByMainThread() )
1337 {
1338 wxMutexGuiLeave();
1339 }
1340 //else: some other worker thread is doing GUI
1341 }
1342}
1343
0c2f35dc 1344bool WXDLLIMPEXP_BASE wxGuiOwnedByMainThread()
bee503b0 1345{
b568d04f 1346 return gs_bGuiOwnedByMainThread;
bee503b0
VZ
1347}
1348
1349// wake up the main thread if it's in ::GetMessage()
0c2f35dc 1350void WXDLLIMPEXP_BASE wxWakeUpMainThread()
bee503b0
VZ
1351{
1352 // sending any message would do - hopefully WM_NULL is harmless enough
b568d04f 1353 if ( !::PostThreadMessage(gs_idMainThread, WM_NULL, 0, 0) )
bee503b0
VZ
1354 {
1355 // should never happen
f6bcfd97 1356 wxLogLastError(wxT("PostThreadMessage(WM_NULL)"));
bee503b0 1357 }
3222fde2 1358}
d524867f 1359
0c2f35dc 1360bool WXDLLIMPEXP_BASE wxIsWaitingForThread()
bf1852e1 1361{
b568d04f 1362 return gs_waitingForThread;
bf1852e1
VZ
1363}
1364
9e84b847
VZ
1365// ----------------------------------------------------------------------------
1366// include common implementation code
1367// ----------------------------------------------------------------------------
1368
1369#include "wx/thrimpl.cpp"
1370
3222fde2 1371#endif // wxUSE_THREADS
6fe73788 1372