]> git.saurik.com Git - wxWidgets.git/blame_incremental - src/msw/thread.cpp
Various small fixes and tweaks
[wxWidgets.git] / src / msw / thread.cpp
... / ...
CommitLineData
1/////////////////////////////////////////////////////////////////////////////
2// Name: thread.cpp
3// Purpose: wxThread Implementation
4// Author: Original from Wolfram Gloger/Guilhem Lavaux
5// Modified by: Vadim Zeitlin to make it work :-)
6// Created: 04/22/98
7// RCS-ID: $Id$
8// Copyright: (c) Wolfram Gloger (1996, 1997); Guilhem Lavaux (1998),
9// Vadim Zeitlin (1999)
10// Licence: wxWindows licence
11/////////////////////////////////////////////////////////////////////////////
12
13#ifdef __GNUG__
14 #pragma implementation "thread.h"
15#endif
16
17// ----------------------------------------------------------------------------
18// headers
19// ----------------------------------------------------------------------------
20
21// For compilers that support precompilation, includes "wx.h".
22#include "wx/wxprec.h"
23
24#if defined(__BORLANDC__)
25 #pragma hdrstop
26#endif
27
28#ifndef WX_PRECOMP
29 #include "wx/wx.h"
30#endif
31
32#if wxUSE_THREADS
33
34#include "wx/msw/private.h"
35
36#include "wx/module.h"
37#include "wx/thread.h"
38
39// must have this symbol defined to get _beginthread/_endthread declarations
40#ifndef _MT
41 #define _MT
42#endif
43
44#ifdef __VISUALC__
45 #include <process.h>
46#endif
47
48// ----------------------------------------------------------------------------
49// constants
50// ----------------------------------------------------------------------------
51
52// the possible states of the thread ("=>" shows all possible transitions from
53// this state)
54enum wxThreadState
55{
56 STATE_NEW, // didn't start execution yet (=> RUNNING)
57 STATE_RUNNING, // thread is running (=> PAUSED, CANCELED)
58 STATE_PAUSED, // thread is temporarily suspended (=> RUNNING)
59 STATE_CANCELED, // thread should terminate a.s.a.p. (=> EXITED)
60 STATE_EXITED // thread is terminating
61};
62
63// ----------------------------------------------------------------------------
64// this module globals
65// ----------------------------------------------------------------------------
66
67// TLS index of the slot where we store the pointer to the current thread
68static DWORD gs_tlsThisThread = 0xFFFFFFFF;
69
70// id of the main thread - the one which can call GUI functions without first
71// calling wxMutexGuiEnter()
72static DWORD gs_idMainThread = 0;
73
74// if it's FALSE, some secondary thread is holding the GUI lock
75static bool gs_bGuiOwnedByMainThread = TRUE;
76
77// critical section which controls access to all GUI functions: any secondary
78// thread (i.e. except the main one) must enter this crit section before doing
79// any GUI calls
80static wxCriticalSection *gs_critsectGui = NULL;
81
82// critical section which protects gs_nWaitingForGui variable
83static wxCriticalSection *gs_critsectWaitingForGui = NULL;
84
85// number of threads waiting for GUI in wxMutexGuiEnter()
86static size_t gs_nWaitingForGui = 0;
87
88// are we waiting for a thread termination?
89static bool gs_waitingForThread = FALSE;
90
91// ============================================================================
92// Windows implementation of thread classes
93// ============================================================================
94
95// ----------------------------------------------------------------------------
96// wxMutex implementation
97// ----------------------------------------------------------------------------
98
99class wxMutexInternal
100{
101public:
102 HANDLE p_mutex;
103};
104
105wxMutex::wxMutex()
106{
107 m_internal = new wxMutexInternal;
108 m_internal->p_mutex = CreateMutex(NULL, FALSE, NULL);
109 if ( !m_internal->p_mutex )
110 {
111 wxLogSysError(_("Can not create mutex."));
112 }
113
114 m_locked = 0;
115}
116
117wxMutex::~wxMutex()
118{
119 if (m_locked > 0)
120 wxLogDebug(wxT("Warning: freeing a locked mutex (%d locks)."), m_locked);
121 CloseHandle(m_internal->p_mutex);
122}
123
124wxMutexError wxMutex::Lock()
125{
126 DWORD ret;
127
128 ret = WaitForSingleObject(m_internal->p_mutex, INFINITE);
129 switch ( ret )
130 {
131 case WAIT_ABANDONED:
132 return wxMUTEX_BUSY;
133
134 case WAIT_OBJECT_0:
135 // ok
136 break;
137
138 case WAIT_FAILED:
139 wxLogSysError(_("Couldn't acquire a mutex lock"));
140 return wxMUTEX_MISC_ERROR;
141
142 case WAIT_TIMEOUT:
143 default:
144 wxFAIL_MSG(wxT("impossible return value in wxMutex::Lock"));
145 }
146
147 m_locked++;
148 return wxMUTEX_NO_ERROR;
149}
150
151wxMutexError wxMutex::TryLock()
152{
153 DWORD ret;
154
155 ret = WaitForSingleObject(m_internal->p_mutex, 0);
156 if (ret == WAIT_TIMEOUT || ret == WAIT_ABANDONED)
157 return wxMUTEX_BUSY;
158
159 m_locked++;
160 return wxMUTEX_NO_ERROR;
161}
162
163wxMutexError wxMutex::Unlock()
164{
165 if (m_locked > 0)
166 m_locked--;
167
168 BOOL ret = ReleaseMutex(m_internal->p_mutex);
169 if ( ret == 0 )
170 {
171 wxLogSysError(_("Couldn't release a mutex"));
172 return wxMUTEX_MISC_ERROR;
173 }
174
175 return wxMUTEX_NO_ERROR;
176}
177
178// ----------------------------------------------------------------------------
179// wxCondition implementation
180// ----------------------------------------------------------------------------
181
182class wxConditionInternal
183{
184public:
185 wxConditionInternal()
186 {
187 event = ::CreateEvent(
188 NULL, // default secutiry
189 FALSE, // not manual reset
190 FALSE, // nonsignaled initially
191 NULL // nameless event
192 );
193 if ( !event )
194 {
195 wxLogSysError(_("Can not create event object."));
196 }
197 waiters = 0;
198 }
199
200 bool Wait(DWORD timeout)
201 {
202 waiters++;
203
204 // FIXME this should be MsgWaitForMultipleObjects() as well probably
205 DWORD rc = ::WaitForSingleObject(event, timeout);
206
207 waiters--;
208
209 return rc != WAIT_TIMEOUT;
210 }
211
212 ~wxConditionInternal()
213 {
214 if ( event )
215 {
216 if ( !::CloseHandle(event) )
217 {
218 wxLogLastError("CloseHandle(event)");
219 }
220 }
221 }
222
223 HANDLE event;
224 int waiters;
225};
226
227wxCondition::wxCondition()
228{
229 m_internal = new wxConditionInternal;
230}
231
232wxCondition::~wxCondition()
233{
234 delete m_internal;
235}
236
237void wxCondition::Wait()
238{
239 (void)m_internal->Wait(INFINITE);
240}
241
242bool wxCondition::Wait(unsigned long sec,
243 unsigned long nsec)
244{
245 return m_internal->Wait(sec*1000 + nsec/1000000);
246}
247
248void wxCondition::Signal()
249{
250 // set the event to signaled: if a thread is already waiting on it, it will
251 // be woken up, otherwise the event will remain in the signaled state until
252 // someone waits on it. In any case, the system will return it to a non
253 // signalled state afterwards. If multiple threads are waiting, only one
254 // will be woken up.
255 if ( !::SetEvent(m_internal->event) )
256 {
257 wxLogLastError("SetEvent");
258 }
259}
260
261void wxCondition::Broadcast()
262{
263 // this works because all these threads are already waiting and so each
264 // SetEvent() inside Signal() is really a PulseEvent() because the event
265 // state is immediately returned to non-signaled
266 for ( int i = 0; i < m_internal->waiters; i++ )
267 {
268 Signal();
269 }
270}
271
272// ----------------------------------------------------------------------------
273// wxCriticalSection implementation
274// ----------------------------------------------------------------------------
275
276wxCriticalSection::wxCriticalSection()
277{
278 wxASSERT_MSG( sizeof(CRITICAL_SECTION) <= sizeof(m_buffer),
279 _T("must increase buffer size in wx/thread.h") );
280
281 ::InitializeCriticalSection((CRITICAL_SECTION *)m_buffer);
282}
283
284wxCriticalSection::~wxCriticalSection()
285{
286 ::DeleteCriticalSection((CRITICAL_SECTION *)m_buffer);
287}
288
289void wxCriticalSection::Enter()
290{
291 ::EnterCriticalSection((CRITICAL_SECTION *)m_buffer);
292}
293
294void wxCriticalSection::Leave()
295{
296 ::LeaveCriticalSection((CRITICAL_SECTION *)m_buffer);
297}
298
299// ----------------------------------------------------------------------------
300// wxThread implementation
301// ----------------------------------------------------------------------------
302
303// wxThreadInternal class
304// ----------------------
305
306class wxThreadInternal
307{
308public:
309 wxThreadInternal()
310 {
311 m_hThread = 0;
312 m_state = STATE_NEW;
313 m_priority = WXTHREAD_DEFAULT_PRIORITY;
314 }
315
316 ~wxThreadInternal()
317 {
318 Free();
319 }
320
321 void Free()
322 {
323 if ( m_hThread )
324 {
325 if ( !::CloseHandle(m_hThread) )
326 {
327 wxLogLastError("CloseHandle(thread)");
328 }
329
330 m_hThread = 0;
331 }
332 }
333
334 // create a new (suspended) thread (for the given thread object)
335 bool Create(wxThread *thread);
336
337 // suspend/resume/terminate
338 bool Suspend();
339 bool Resume();
340 void Cancel() { m_state = STATE_CANCELED; }
341
342 // thread state
343 void SetState(wxThreadState state) { m_state = state; }
344 wxThreadState GetState() const { return m_state; }
345
346 // thread priority
347 void SetPriority(unsigned int priority);
348 unsigned int GetPriority() const { return m_priority; }
349
350 // thread handle and id
351 HANDLE GetHandle() const { return m_hThread; }
352 DWORD GetId() const { return m_tid; }
353
354 // thread function
355 static DWORD WinThreadStart(wxThread *thread);
356
357private:
358 HANDLE m_hThread; // handle of the thread
359 wxThreadState m_state; // state, see wxThreadState enum
360 unsigned int m_priority; // thread priority in "wx" units
361 DWORD m_tid; // thread id
362};
363
364DWORD wxThreadInternal::WinThreadStart(wxThread *thread)
365{
366 // store the thread object in the TLS
367 if ( !::TlsSetValue(gs_tlsThisThread, thread) )
368 {
369 wxLogSysError(_("Can not start thread: error writing TLS."));
370
371 return (DWORD)-1;
372 }
373
374 DWORD rc = (DWORD)thread->Entry();
375
376 // enter m_critsect before changing the thread state
377 thread->m_critsect.Enter();
378 bool wasCancelled = thread->m_internal->GetState() == STATE_CANCELED;
379 thread->m_internal->SetState(STATE_EXITED);
380 thread->m_critsect.Leave();
381
382 thread->OnExit();
383
384 // if the thread was cancelled (from Delete()), then it the handle is still
385 // needed there
386 if ( thread->IsDetached() && !wasCancelled )
387 {
388 // auto delete
389 delete thread;
390 }
391 //else: the joinable threads handle will be closed when Wait() is done
392
393 return rc;
394}
395
396void wxThreadInternal::SetPriority(unsigned int priority)
397{
398 m_priority = priority;
399
400 // translate wxWindows priority to the Windows one
401 int win_priority;
402 if (m_priority <= 20)
403 win_priority = THREAD_PRIORITY_LOWEST;
404 else if (m_priority <= 40)
405 win_priority = THREAD_PRIORITY_BELOW_NORMAL;
406 else if (m_priority <= 60)
407 win_priority = THREAD_PRIORITY_NORMAL;
408 else if (m_priority <= 80)
409 win_priority = THREAD_PRIORITY_ABOVE_NORMAL;
410 else if (m_priority <= 100)
411 win_priority = THREAD_PRIORITY_HIGHEST;
412 else
413 {
414 wxFAIL_MSG(wxT("invalid value of thread priority parameter"));
415 win_priority = THREAD_PRIORITY_NORMAL;
416 }
417
418 if ( !::SetThreadPriority(m_hThread, win_priority) )
419 {
420 wxLogSysError(_("Can't set thread priority"));
421 }
422}
423
424bool wxThreadInternal::Create(wxThread *thread)
425{
426 // for compilers which have it, we should use C RTL function for thread
427 // creation instead of Win32 API one because otherwise we will have memory
428 // leaks if the thread uses C RTL (and most threads do)
429#ifdef __VISUALC__
430 typedef unsigned (__stdcall *RtlThreadStart)(void *);
431
432 m_hThread = (HANDLE)_beginthreadex(NULL, 0,
433 (RtlThreadStart)
434 wxThreadInternal::WinThreadStart,
435 thread, CREATE_SUSPENDED,
436 (unsigned int *)&m_tid);
437#else // !VC++
438 m_hThread = ::CreateThread
439 (
440 NULL, // default security
441 0, // default stack size
442 (LPTHREAD_START_ROUTINE) // thread entry point
443 wxThreadInternal::WinThreadStart, //
444 (LPVOID)thread, // parameter
445 CREATE_SUSPENDED, // flags
446 &m_tid // [out] thread id
447 );
448#endif // VC++/!VC++
449
450 if ( m_hThread == NULL )
451 {
452 wxLogSysError(_("Can't create thread"));
453
454 return FALSE;
455 }
456
457 if ( m_priority != WXTHREAD_DEFAULT_PRIORITY )
458 {
459 SetPriority(m_priority);
460 }
461
462 return TRUE;
463}
464
465bool wxThreadInternal::Suspend()
466{
467 DWORD nSuspendCount = ::SuspendThread(m_hThread);
468 if ( nSuspendCount == (DWORD)-1 )
469 {
470 wxLogSysError(_("Can not suspend thread %x"), m_hThread);
471
472 return FALSE;
473 }
474
475 m_state = STATE_PAUSED;
476
477 return TRUE;
478}
479
480bool wxThreadInternal::Resume()
481{
482 DWORD nSuspendCount = ::ResumeThread(m_hThread);
483 if ( nSuspendCount == (DWORD)-1 )
484 {
485 wxLogSysError(_("Can not resume thread %x"), m_hThread);
486
487 return FALSE;
488 }
489
490 m_state = STATE_RUNNING;
491
492 return TRUE;
493}
494
495// static functions
496// ----------------
497
498wxThread *wxThread::This()
499{
500 wxThread *thread = (wxThread *)::TlsGetValue(gs_tlsThisThread);
501
502 // be careful, 0 may be a valid return value as well
503 if ( !thread && (::GetLastError() != NO_ERROR) )
504 {
505 wxLogSysError(_("Couldn't get the current thread pointer"));
506
507 // return NULL...
508 }
509
510 return thread;
511}
512
513bool wxThread::IsMain()
514{
515 return ::GetCurrentThreadId() == gs_idMainThread;
516}
517
518void wxThread::Yield()
519{
520 // 0 argument to Sleep() is special and means to just give away the rest of
521 // our timeslice
522 ::Sleep(0);
523}
524
525void wxThread::Sleep(unsigned long milliseconds)
526{
527 ::Sleep(milliseconds);
528}
529
530// ctor and dtor
531// -------------
532
533wxThread::wxThread(wxThreadKind kind)
534{
535 m_internal = new wxThreadInternal();
536
537 m_isDetached = kind == wxTHREAD_DETACHED;
538}
539
540wxThread::~wxThread()
541{
542 delete m_internal;
543}
544
545// create/start thread
546// -------------------
547
548wxThreadError wxThread::Create()
549{
550 wxCriticalSectionLocker lock(m_critsect);
551
552 if ( !m_internal->Create(this) )
553 return wxTHREAD_NO_RESOURCE;
554
555 return wxTHREAD_NO_ERROR;
556}
557
558wxThreadError wxThread::Run()
559{
560 wxCriticalSectionLocker lock(m_critsect);
561
562 if ( m_internal->GetState() != STATE_NEW )
563 {
564 // actually, it may be almost any state at all, not only STATE_RUNNING
565 return wxTHREAD_RUNNING;
566 }
567
568 // the thread has just been created and is still suspended - let it run
569 return Resume();
570}
571
572// suspend/resume thread
573// ---------------------
574
575wxThreadError wxThread::Pause()
576{
577 wxCriticalSectionLocker lock(m_critsect);
578
579 return m_internal->Suspend() ? wxTHREAD_NO_ERROR : wxTHREAD_MISC_ERROR;
580}
581
582wxThreadError wxThread::Resume()
583{
584 wxCriticalSectionLocker lock(m_critsect);
585
586 return m_internal->Resume() ? wxTHREAD_NO_ERROR : wxTHREAD_MISC_ERROR;
587}
588
589// stopping thread
590// ---------------
591
592wxThread::ExitCode wxThread::Wait()
593{
594 // although under Windows we can wait for any thread, it's an error to
595 // wait for a detached one in wxWin API
596 wxCHECK_MSG( !IsDetached(), (ExitCode)-1,
597 _T("can't wait for detached thread") );
598
599 ExitCode rc = (ExitCode)-1;
600
601 (void)Delete(&rc);
602
603 m_internal->Free();
604
605 return rc;
606}
607
608wxThreadError wxThread::Delete(ExitCode *pRc)
609{
610 ExitCode rc = 0;
611
612 // Delete() is always safe to call, so consider all possible states
613 if ( IsPaused() )
614 Resume();
615
616 HANDLE hThread = m_internal->GetHandle();
617
618 if ( IsRunning() )
619 {
620 if ( IsMain() )
621 {
622 // set flag for wxIsWaitingForThread()
623 gs_waitingForThread = TRUE;
624
625#if wxUSE_GUI
626 wxBeginBusyCursor();
627#endif // wxUSE_GUI
628 }
629
630 // ask the thread to terminate
631 {
632 wxCriticalSectionLocker lock(m_critsect);
633
634 m_internal->Cancel();
635 }
636
637#if wxUSE_GUI
638 // we can't just wait for the thread to terminate because it might be
639 // calling some GUI functions and so it will never terminate before we
640 // process the Windows messages that result from these functions
641 DWORD result;
642 do
643 {
644 result = ::MsgWaitForMultipleObjects
645 (
646 1, // number of objects to wait for
647 &hThread, // the objects
648 FALSE, // don't wait for all objects
649 INFINITE, // no timeout
650 QS_ALLEVENTS // return as soon as there are any events
651 );
652
653 switch ( result )
654 {
655 case 0xFFFFFFFF:
656 // error
657 wxLogSysError(_("Can not wait for thread termination"));
658 Kill();
659 return wxTHREAD_KILLED;
660
661 case WAIT_OBJECT_0:
662 // thread we're waiting for terminated
663 break;
664
665 case WAIT_OBJECT_0 + 1:
666 // new message arrived, process it
667 if ( !wxTheApp->DoMessage() )
668 {
669 // WM_QUIT received: kill the thread
670 Kill();
671
672 return wxTHREAD_KILLED;
673 }
674
675 if ( IsMain() )
676 {
677 // give the thread we're waiting for chance to exit
678 // from the GUI call it might have been in
679 if ( (gs_nWaitingForGui > 0) && wxGuiOwnedByMainThread() )
680 {
681 wxMutexGuiLeave();
682 }
683 }
684
685 break;
686
687 default:
688 wxFAIL_MSG(wxT("unexpected result of MsgWaitForMultipleObject"));
689 }
690 } while ( result != WAIT_OBJECT_0 );
691#else // !wxUSE_GUI
692 // simply wait for the thread to terminate
693 //
694 // OTOH, even console apps create windows (in wxExecute, for WinSock
695 // &c), so may be use MsgWaitForMultipleObject() too here?
696 if ( WaitForSingleObject(hThread, INFINITE) != WAIT_OBJECT_0 )
697 {
698 wxFAIL_MSG(wxT("unexpected result of WaitForSingleObject"));
699 }
700#endif // wxUSE_GUI/!wxUSE_GUI
701
702 if ( IsMain() )
703 {
704 gs_waitingForThread = FALSE;
705
706#if wxUSE_GUI
707 wxEndBusyCursor();
708#endif // wxUSE_GUI
709 }
710 }
711
712 if ( !::GetExitCodeThread(hThread, (LPDWORD)&rc) )
713 {
714 wxLogLastError("GetExitCodeThread");
715
716 rc = (ExitCode)-1;
717 }
718
719 if ( IsDetached() )
720 {
721 // if the thread exits normally, this is done in WinThreadStart, but in
722 // this case it would have been too early because
723 // MsgWaitForMultipleObject() would fail if the therad handle was
724 // closed while we were waiting on it, so we must do it here
725 delete this;
726 }
727
728 wxASSERT_MSG( (DWORD)rc != STILL_ACTIVE,
729 wxT("thread must be already terminated.") );
730
731 if ( pRc )
732 *pRc = rc;
733
734 return rc == (ExitCode)-1 ? wxTHREAD_MISC_ERROR : wxTHREAD_NO_ERROR;
735}
736
737wxThreadError wxThread::Kill()
738{
739 if ( !IsRunning() )
740 return wxTHREAD_NOT_RUNNING;
741
742 if ( !::TerminateThread(m_internal->GetHandle(), (DWORD)-1) )
743 {
744 wxLogSysError(_("Couldn't terminate thread"));
745
746 return wxTHREAD_MISC_ERROR;
747 }
748
749 m_internal->Free();
750
751 if ( IsDetached() )
752 {
753 delete this;
754 }
755
756 return wxTHREAD_NO_ERROR;
757}
758
759void wxThread::Exit(ExitCode status)
760{
761 m_internal->Free();
762
763 if ( IsDetached() )
764 {
765 delete this;
766 }
767
768#ifdef __VISUALC__
769 _endthreadex((unsigned)status);
770#else // !VC++
771 ::ExitThread((DWORD)status);
772#endif // VC++/!VC++
773
774 wxFAIL_MSG(wxT("Couldn't return from ExitThread()!"));
775}
776
777// priority setting
778// ----------------
779
780void wxThread::SetPriority(unsigned int prio)
781{
782 wxCriticalSectionLocker lock(m_critsect);
783
784 m_internal->SetPriority(prio);
785}
786
787unsigned int wxThread::GetPriority() const
788{
789 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
790
791 return m_internal->GetPriority();
792}
793
794unsigned long wxThread::GetId() const
795{
796 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
797
798 return (unsigned long)m_internal->GetId();
799}
800
801bool wxThread::IsRunning() const
802{
803 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
804
805 return m_internal->GetState() == STATE_RUNNING;
806}
807
808bool wxThread::IsAlive() const
809{
810 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
811
812 return (m_internal->GetState() == STATE_RUNNING) ||
813 (m_internal->GetState() == STATE_PAUSED);
814}
815
816bool wxThread::IsPaused() const
817{
818 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
819
820 return m_internal->GetState() == STATE_PAUSED;
821}
822
823bool wxThread::TestDestroy()
824{
825 wxCriticalSectionLocker lock((wxCriticalSection &)m_critsect); // const_cast
826
827 return m_internal->GetState() == STATE_CANCELED;
828}
829
830// ----------------------------------------------------------------------------
831// Automatic initialization for thread module
832// ----------------------------------------------------------------------------
833
834class wxThreadModule : public wxModule
835{
836public:
837 virtual bool OnInit();
838 virtual void OnExit();
839
840private:
841 DECLARE_DYNAMIC_CLASS(wxThreadModule)
842};
843
844IMPLEMENT_DYNAMIC_CLASS(wxThreadModule, wxModule)
845
846bool wxThreadModule::OnInit()
847{
848 // allocate TLS index for storing the pointer to the current thread
849 gs_tlsThisThread = ::TlsAlloc();
850 if ( gs_tlsThisThread == 0xFFFFFFFF )
851 {
852 // in normal circumstances it will only happen if all other
853 // TLS_MINIMUM_AVAILABLE (>= 64) indices are already taken - in other
854 // words, this should never happen
855 wxLogSysError(_("Thread module initialization failed: "
856 "impossible to allocate index in thread "
857 "local storage"));
858
859 return FALSE;
860 }
861
862 // main thread doesn't have associated wxThread object, so store 0 in the
863 // TLS instead
864 if ( !::TlsSetValue(gs_tlsThisThread, (LPVOID)0) )
865 {
866 ::TlsFree(gs_tlsThisThread);
867 gs_tlsThisThread = 0xFFFFFFFF;
868
869 wxLogSysError(_("Thread module initialization failed: "
870 "can not store value in thread local storage"));
871
872 return FALSE;
873 }
874
875 gs_critsectWaitingForGui = new wxCriticalSection();
876
877 gs_critsectGui = new wxCriticalSection();
878 gs_critsectGui->Enter();
879
880 // no error return for GetCurrentThreadId()
881 gs_idMainThread = ::GetCurrentThreadId();
882
883 return TRUE;
884}
885
886void wxThreadModule::OnExit()
887{
888 if ( !::TlsFree(gs_tlsThisThread) )
889 {
890 wxLogLastError("TlsFree failed.");
891 }
892
893 if ( gs_critsectGui )
894 {
895 gs_critsectGui->Leave();
896 delete gs_critsectGui;
897 gs_critsectGui = NULL;
898 }
899
900 delete gs_critsectWaitingForGui;
901 gs_critsectWaitingForGui = NULL;
902}
903
904// ----------------------------------------------------------------------------
905// under Windows, these functions are implemented using a critical section and
906// not a mutex, so the names are a bit confusing
907// ----------------------------------------------------------------------------
908
909void WXDLLEXPORT wxMutexGuiEnter()
910{
911 // this would dead lock everything...
912 wxASSERT_MSG( !wxThread::IsMain(),
913 wxT("main thread doesn't want to block in wxMutexGuiEnter()!") );
914
915 // the order in which we enter the critical sections here is crucial!!
916
917 // set the flag telling to the main thread that we want to do some GUI
918 {
919 wxCriticalSectionLocker enter(*gs_critsectWaitingForGui);
920
921 gs_nWaitingForGui++;
922 }
923
924 wxWakeUpMainThread();
925
926 // now we may block here because the main thread will soon let us in
927 // (during the next iteration of OnIdle())
928 gs_critsectGui->Enter();
929}
930
931void WXDLLEXPORT wxMutexGuiLeave()
932{
933 wxCriticalSectionLocker enter(*gs_critsectWaitingForGui);
934
935 if ( wxThread::IsMain() )
936 {
937 gs_bGuiOwnedByMainThread = FALSE;
938 }
939 else
940 {
941 // decrement the number of waiters now
942 wxASSERT_MSG( gs_nWaitingForGui > 0,
943 wxT("calling wxMutexGuiLeave() without entering it first?") );
944
945 gs_nWaitingForGui--;
946
947 wxWakeUpMainThread();
948 }
949
950 gs_critsectGui->Leave();
951}
952
953void WXDLLEXPORT wxMutexGuiLeaveOrEnter()
954{
955 wxASSERT_MSG( wxThread::IsMain(),
956 wxT("only main thread may call wxMutexGuiLeaveOrEnter()!") );
957
958 wxCriticalSectionLocker enter(*gs_critsectWaitingForGui);
959
960 if ( gs_nWaitingForGui == 0 )
961 {
962 // no threads are waiting for GUI - so we may acquire the lock without
963 // any danger (but only if we don't already have it)
964 if ( !wxGuiOwnedByMainThread() )
965 {
966 gs_critsectGui->Enter();
967
968 gs_bGuiOwnedByMainThread = TRUE;
969 }
970 //else: already have it, nothing to do
971 }
972 else
973 {
974 // some threads are waiting, release the GUI lock if we have it
975 if ( wxGuiOwnedByMainThread() )
976 {
977 wxMutexGuiLeave();
978 }
979 //else: some other worker thread is doing GUI
980 }
981}
982
983bool WXDLLEXPORT wxGuiOwnedByMainThread()
984{
985 return gs_bGuiOwnedByMainThread;
986}
987
988// wake up the main thread if it's in ::GetMessage()
989void WXDLLEXPORT wxWakeUpMainThread()
990{
991 // sending any message would do - hopefully WM_NULL is harmless enough
992 if ( !::PostThreadMessage(gs_idMainThread, WM_NULL, 0, 0) )
993 {
994 // should never happen
995 wxLogLastError("PostThreadMessage(WM_NULL)");
996 }
997}
998
999bool WXDLLEXPORT wxIsWaitingForThread()
1000{
1001 return gs_waitingForThread;
1002}
1003
1004#endif // wxUSE_THREADS