1 /////////////////////////////////////////////////////////////////////////////
3 // Purpose: wxThread Implementation
4 // Author: Original from Wolfram Gloger/Guilhem Lavaux
5 // Modified by: Vadim Zeitlin to make it work :-)
8 // Copyright: (c) Wolfram Gloger (1996, 1997); Guilhem Lavaux (1998),
9 // Vadim Zeitlin (1999)
10 // Licence: wxWindows licence
11 /////////////////////////////////////////////////////////////////////////////
14 #pragma implementation "thread.h"
17 // ----------------------------------------------------------------------------
19 // ----------------------------------------------------------------------------
21 // For compilers that support precompilation, includes "wx.h".
22 #include "wx/wxprec.h"
24 #if defined(__BORLANDC__)
38 #include "wx/module.h"
39 #include "wx/thread.h"
41 // the possible states of the thread ("=>" shows all possible transitions from
45 STATE_NEW
, // didn't start execution yet (=> RUNNING)
46 STATE_RUNNING
, // thread is running (=> PAUSED, CANCELED)
47 STATE_PAUSED
, // thread is temporarily suspended (=> RUNNING)
48 STATE_CANCELED
, // thread should terminate a.s.a.p. (=> EXITED)
49 STATE_EXITED
// thread is terminating
52 // ----------------------------------------------------------------------------
54 // ----------------------------------------------------------------------------
56 // TLS index of the slot where we store the pointer to the current thread
57 static DWORD s_tlsThisThread
= 0xFFFFFFFF;
59 // id of the main thread - the one which can call GUI functions without first
60 // calling wxMutexGuiEnter()
61 static DWORD s_idMainThread
= 0;
63 // if it's FALSE, some secondary thread is holding the GUI lock
64 static bool s_bGuiOwnedByMainThread
= TRUE
;
66 // critical section which controls access to all GUI functions: any secondary
67 // thread (i.e. except the main one) must enter this crit section before doing
69 static wxCriticalSection
*s_critsectGui
= NULL
;
71 // critical section which protects s_nWaitingForGui variable
72 static wxCriticalSection
*s_critsectWaitingForGui
= NULL
;
74 // number of threads waiting for GUI in wxMutexGuiEnter()
75 static size_t s_nWaitingForGui
= 0;
77 // are we waiting for a thread termination?
78 static bool s_waitingForThread
= FALSE
;
80 // ============================================================================
81 // Windows implementation of thread classes
82 // ============================================================================
84 // ----------------------------------------------------------------------------
85 // wxMutex implementation
86 // ----------------------------------------------------------------------------
95 p_internal
= new wxMutexInternal
;
96 p_internal
->p_mutex
= CreateMutex(NULL
, FALSE
, NULL
);
97 if ( !p_internal
->p_mutex
)
99 wxLogSysError(_("Can not create mutex."));
108 wxLogDebug("Warning: freeing a locked mutex (%d locks).", m_locked
);
109 CloseHandle(p_internal
->p_mutex
);
112 wxMutexError
wxMutex::Lock()
116 ret
= WaitForSingleObject(p_internal
->p_mutex
, INFINITE
);
127 wxLogSysError(_("Couldn't acquire a mutex lock"));
128 return wxMUTEX_MISC_ERROR
;
132 wxFAIL_MSG("impossible return value in wxMutex::Lock");
136 return wxMUTEX_NO_ERROR
;
139 wxMutexError
wxMutex::TryLock()
143 ret
= WaitForSingleObject(p_internal
->p_mutex
, 0);
144 if (ret
== WAIT_TIMEOUT
|| ret
== WAIT_ABANDONED
)
148 return wxMUTEX_NO_ERROR
;
151 wxMutexError
wxMutex::Unlock()
156 BOOL ret
= ReleaseMutex(p_internal
->p_mutex
);
159 wxLogSysError(_("Couldn't release a mutex"));
160 return wxMUTEX_MISC_ERROR
;
163 return wxMUTEX_NO_ERROR
;
166 // ----------------------------------------------------------------------------
167 // wxCondition implementation
168 // ----------------------------------------------------------------------------
170 class wxConditionInternal
177 wxCondition::wxCondition()
179 p_internal
= new wxConditionInternal
;
180 p_internal
->event
= CreateEvent(NULL
, FALSE
, FALSE
, NULL
);
181 if ( !p_internal
->event
)
183 wxLogSysError(_("Can not create event object."));
186 p_internal
->waiters
= 0;
189 wxCondition::~wxCondition()
191 CloseHandle(p_internal
->event
);
194 void wxCondition::Wait(wxMutex
& mutex
)
197 p_internal
->waiters
++;
198 WaitForSingleObject(p_internal
->event
, INFINITE
);
199 p_internal
->waiters
--;
203 bool wxCondition::Wait(wxMutex
& mutex
,
210 p_internal
->waiters
++;
211 ret
= WaitForSingleObject(p_internal
->event
, (sec
*1000)+(nsec
/1000000));
212 p_internal
->waiters
--;
215 return (ret
!= WAIT_TIMEOUT
);
218 void wxCondition::Signal()
220 SetEvent(p_internal
->event
);
223 void wxCondition::Broadcast()
227 for (i
=0;i
<p_internal
->waiters
;i
++)
229 if ( SetEvent(p_internal
->event
) == 0 )
231 wxLogSysError(_("Couldn't change the state of event object."));
236 // ----------------------------------------------------------------------------
237 // wxCriticalSection implementation
238 // ----------------------------------------------------------------------------
240 class wxCriticalSectionInternal
243 // init the critical section object
244 wxCriticalSectionInternal()
245 { ::InitializeCriticalSection(&m_data
); }
247 // implicit cast to the associated data
248 operator CRITICAL_SECTION
*() { return &m_data
; }
250 // free the associated ressources
251 ~wxCriticalSectionInternal()
252 { ::DeleteCriticalSection(&m_data
); }
255 CRITICAL_SECTION m_data
;
258 wxCriticalSection::wxCriticalSection()
263 wxCriticalSection::~wxCriticalSection()
265 wxASSERT_MSG( !m_critsect
, "Forgot to Leave() critical section" );
268 void wxCriticalSection::Enter()
270 m_critsect
= new wxCriticalSectionInternal
;
272 ::EnterCriticalSection(*m_critsect
);
275 void wxCriticalSection::Leave()
277 wxCHECK_RET( m_critsect
, "Leave() without matching Enter()" );
279 ::LeaveCriticalSection(*m_critsect
);
285 // ----------------------------------------------------------------------------
286 // wxThread implementation
287 // ----------------------------------------------------------------------------
289 // wxThreadInternal class
290 // ----------------------
292 class wxThreadInternal
299 m_priority
= WXTHREAD_DEFAULT_PRIORITY
;
302 // create a new (suspended) thread (for the given thread object)
303 bool Create(wxThread
*thread
);
305 // suspend/resume/terminate
308 void Cancel() { m_state
= STATE_CANCELED
; }
311 void SetState(wxThreadState state
) { m_state
= state
; }
312 wxThreadState
GetState() const { return m_state
; }
315 void SetPriority(unsigned int priority
) { m_priority
= priority
; }
316 unsigned int GetPriority() const { return m_priority
; }
318 // thread handle and id
319 HANDLE
GetHandle() const { return m_hThread
; }
320 DWORD
GetId() const { return m_tid
; }
323 static DWORD
WinThreadStart(wxThread
*thread
);
326 HANDLE m_hThread
; // handle of the thread
327 wxThreadState m_state
; // state, see wxThreadState enum
328 unsigned int m_priority
; // thread priority in "wx" units
329 DWORD m_tid
; // thread id
332 DWORD
wxThreadInternal::WinThreadStart(wxThread
*thread
)
334 // store the thread object in the TLS
335 if ( !::TlsSetValue(s_tlsThisThread
, thread
) )
337 wxLogSysError(_("Can not start thread: error writing TLS."));
342 DWORD ret
= (DWORD
)thread
->Entry();
343 thread
->p_internal
->SetState(STATE_EXITED
);
351 bool wxThreadInternal::Create(wxThread
*thread
)
353 m_hThread
= ::CreateThread
355 NULL
, // default security
356 0, // default stack size
357 (LPTHREAD_START_ROUTINE
) // thread entry point
358 wxThreadInternal::WinThreadStart
, //
359 (LPVOID
)thread
, // parameter
360 CREATE_SUSPENDED
, // flags
361 &m_tid
// [out] thread id
364 if ( m_hThread
== NULL
)
366 wxLogSysError(_("Can't create thread"));
371 // translate wxWindows priority to the Windows one
373 if (m_priority
<= 20)
374 win_priority
= THREAD_PRIORITY_LOWEST
;
375 else if (m_priority
<= 40)
376 win_priority
= THREAD_PRIORITY_BELOW_NORMAL
;
377 else if (m_priority
<= 60)
378 win_priority
= THREAD_PRIORITY_NORMAL
;
379 else if (m_priority
<= 80)
380 win_priority
= THREAD_PRIORITY_ABOVE_NORMAL
;
381 else if (m_priority
<= 100)
382 win_priority
= THREAD_PRIORITY_HIGHEST
;
385 wxFAIL_MSG("invalid value of thread priority parameter");
386 win_priority
= THREAD_PRIORITY_NORMAL
;
389 if ( ::SetThreadPriority(m_hThread
, win_priority
) == 0 )
391 wxLogSysError(_("Can't set thread priority"));
397 bool wxThreadInternal::Suspend()
399 DWORD nSuspendCount
= ::SuspendThread(m_hThread
);
400 if ( nSuspendCount
== (DWORD
)-1 )
402 wxLogSysError(_("Can not suspend thread %x"), m_hThread
);
407 m_state
= STATE_PAUSED
;
412 bool wxThreadInternal::Resume()
414 DWORD nSuspendCount
= ::ResumeThread(m_hThread
);
415 if ( nSuspendCount
== (DWORD
)-1 )
417 wxLogSysError(_("Can not resume thread %x"), m_hThread
);
422 m_state
= STATE_RUNNING
;
430 wxThread
*wxThread::This()
432 wxThread
*thread
= (wxThread
*)::TlsGetValue(s_tlsThisThread
);
434 // be careful, 0 may be a valid return value as well
435 if ( !thread
&& (::GetLastError() != NO_ERROR
) )
437 wxLogSysError(_("Couldn't get the current thread pointer"));
445 bool wxThread::IsMain()
447 return ::GetCurrentThreadId() == s_idMainThread
;
450 void wxThread::Yield()
452 // 0 argument to Sleep() is special
456 void wxThread::Sleep(unsigned long milliseconds
)
458 ::Sleep(milliseconds
);
461 // create/start thread
462 // -------------------
464 wxThreadError
wxThread::Create()
466 if ( !p_internal
->Create(this) )
467 return wxTHREAD_NO_RESOURCE
;
469 return wxTHREAD_NO_ERROR
;
472 wxThreadError
wxThread::Run()
474 wxCriticalSectionLocker
lock(m_critsect
);
476 if ( p_internal
->GetState() != STATE_NEW
)
478 // actually, it may be almost any state at all, not only STATE_RUNNING
479 return wxTHREAD_RUNNING
;
485 // suspend/resume thread
486 // ---------------------
488 wxThreadError
wxThread::Pause()
490 wxCriticalSectionLocker
lock(m_critsect
);
492 return p_internal
->Suspend() ? wxTHREAD_NO_ERROR
: wxTHREAD_MISC_ERROR
;
495 wxThreadError
wxThread::Resume()
497 wxCriticalSectionLocker
lock(m_critsect
);
499 return p_internal
->Resume() ? wxTHREAD_NO_ERROR
: wxTHREAD_MISC_ERROR
;
505 wxThread::ExitCode
wxThread::Delete()
509 // Delete() is always safe to call, so consider all possible states
517 // set flag for wxIsWaitingForThread()
518 s_waitingForThread
= TRUE
;
525 wxCriticalSectionLocker
lock(m_critsect
);
527 p_internal
->Cancel();
528 hThread
= p_internal
->GetHandle();
531 // we can't just wait for the thread to terminate because it might be
532 // calling some GUI functions and so it will never terminate before we
533 // process the Windows messages that result from these functions
537 result
= ::MsgWaitForMultipleObjects
539 1, // number of objects to wait for
540 &hThread
, // the objects
541 FALSE
, // don't wait for all objects
542 INFINITE
, // no timeout
543 QS_ALLEVENTS
// return as soon as there are any events
550 wxLogSysError(_("Can not wait for thread termination"));
555 // thread we're waiting for terminated
558 case WAIT_OBJECT_0
+ 1:
559 // new message arrived, process it
560 if ( !wxTheApp
->DoMessage() )
562 // WM_QUIT received: kill the thread
570 // give the thread we're waiting for chance to exit
571 // from the GUI call it might have been in
572 if ( (s_nWaitingForGui
> 0) && wxGuiOwnedByMainThread() )
581 wxFAIL_MSG("unexpected result of MsgWaitForMultipleObject");
583 } while ( result
!= WAIT_OBJECT_0
);
587 s_waitingForThread
= FALSE
;
592 if ( !::GetExitCodeThread(hThread
, (LPDWORD
)&rc
) )
594 wxLogLastError("GetExitCodeThread");
599 wxASSERT_MSG( (LPVOID
)rc
!= (LPVOID
)STILL_ACTIVE
,
600 "thread must be already terminated." );
602 ::CloseHandle(hThread
);
608 wxThreadError
wxThread::Kill()
611 return wxTHREAD_NOT_RUNNING
;
613 if ( !::TerminateThread(p_internal
->GetHandle(), (DWORD
)-1) )
615 wxLogSysError(_("Couldn't terminate thread"));
617 return wxTHREAD_MISC_ERROR
;
622 return wxTHREAD_NO_ERROR
;
625 void wxThread::Exit(void *status
)
629 ::ExitThread((DWORD
)status
);
631 wxFAIL_MSG("Couldn't return from ExitThread()!");
634 void wxThread::SetPriority(unsigned int prio
)
636 wxCriticalSectionLocker
lock(m_critsect
);
638 p_internal
->SetPriority(prio
);
641 unsigned int wxThread::GetPriority() const
643 wxCriticalSectionLocker
lock((wxCriticalSection
&)m_critsect
);
645 return p_internal
->GetPriority();
648 unsigned long wxThread::GetID() const
650 wxCriticalSectionLocker
lock((wxCriticalSection
&)m_critsect
);
652 return (unsigned long)p_internal
->GetId();
655 bool wxThread::IsRunning() const
657 wxCriticalSectionLocker
lock((wxCriticalSection
&)m_critsect
);
659 return p_internal
->GetState() == STATE_RUNNING
;
662 bool wxThread::IsAlive() const
664 wxCriticalSectionLocker
lock((wxCriticalSection
&)m_critsect
);
666 return (p_internal
->GetState() == STATE_RUNNING
) ||
667 (p_internal
->GetState() == STATE_PAUSED
);
670 bool wxThread::TestDestroy()
672 wxCriticalSectionLocker
lock((wxCriticalSection
&)m_critsect
);
674 return p_internal
->GetState() == STATE_CANCELED
;
679 p_internal
= new wxThreadInternal();
682 wxThread::~wxThread()
687 // ----------------------------------------------------------------------------
688 // Automatic initialization for thread module
689 // ----------------------------------------------------------------------------
691 class wxThreadModule
: public wxModule
694 virtual bool OnInit();
695 virtual void OnExit();
698 DECLARE_DYNAMIC_CLASS(wxThreadModule
)
701 IMPLEMENT_DYNAMIC_CLASS(wxThreadModule
, wxModule
)
703 bool wxThreadModule::OnInit()
705 // allocate TLS index for storing the pointer to the current thread
706 s_tlsThisThread
= ::TlsAlloc();
707 if ( s_tlsThisThread
== 0xFFFFFFFF )
709 // in normal circumstances it will only happen if all other
710 // TLS_MINIMUM_AVAILABLE (>= 64) indices are already taken - in other
711 // words, this should never happen
712 wxLogSysError(_("Thread module initialization failed: "
713 "impossible to allocate index in thread "
719 // main thread doesn't have associated wxThread object, so store 0 in the
721 if ( !::TlsSetValue(s_tlsThisThread
, (LPVOID
)0) )
723 ::TlsFree(s_tlsThisThread
);
724 s_tlsThisThread
= 0xFFFFFFFF;
726 wxLogSysError(_("Thread module initialization failed: "
727 "can not store value in thread local storage"));
732 s_critsectWaitingForGui
= new wxCriticalSection();
734 s_critsectGui
= new wxCriticalSection();
735 s_critsectGui
->Enter();
737 // no error return for GetCurrentThreadId()
738 s_idMainThread
= ::GetCurrentThreadId();
743 void wxThreadModule::OnExit()
745 if ( !::TlsFree(s_tlsThisThread
) )
747 wxLogLastError("TlsFree failed.");
752 s_critsectGui
->Leave();
753 delete s_critsectGui
;
754 s_critsectGui
= NULL
;
757 wxDELETE(s_critsectWaitingForGui
);
760 // ----------------------------------------------------------------------------
761 // under Windows, these functions are implemented usign a critical section and
762 // not a mutex, so the names are a bit confusing
763 // ----------------------------------------------------------------------------
765 void WXDLLEXPORT
wxMutexGuiEnter()
767 // this would dead lock everything...
768 wxASSERT_MSG( !wxThread::IsMain(),
769 "main thread doesn't want to block in wxMutexGuiEnter()!" );
771 // the order in which we enter the critical sections here is crucial!!
773 // set the flag telling to the main thread that we want to do some GUI
775 wxCriticalSectionLocker
enter(*s_critsectWaitingForGui
);
780 wxWakeUpMainThread();
782 // now we may block here because the main thread will soon let us in
783 // (during the next iteration of OnIdle())
784 s_critsectGui
->Enter();
787 void WXDLLEXPORT
wxMutexGuiLeave()
789 wxCriticalSectionLocker
enter(*s_critsectWaitingForGui
);
791 if ( wxThread::IsMain() )
793 s_bGuiOwnedByMainThread
= FALSE
;
797 // decrement the number of waiters now
798 wxASSERT_MSG( s_nWaitingForGui
> 0,
799 "calling wxMutexGuiLeave() without entering it first?" );
803 wxWakeUpMainThread();
806 s_critsectGui
->Leave();
809 void WXDLLEXPORT
wxMutexGuiLeaveOrEnter()
811 wxASSERT_MSG( wxThread::IsMain(),
812 "only main thread may call wxMutexGuiLeaveOrEnter()!" );
814 wxCriticalSectionLocker
enter(*s_critsectWaitingForGui
);
816 if ( s_nWaitingForGui
== 0 )
818 // no threads are waiting for GUI - so we may acquire the lock without
819 // any danger (but only if we don't already have it)
820 if ( !wxGuiOwnedByMainThread() )
822 s_critsectGui
->Enter();
824 s_bGuiOwnedByMainThread
= TRUE
;
826 //else: already have it, nothing to do
830 // some threads are waiting, release the GUI lock if we have it
831 if ( wxGuiOwnedByMainThread() )
835 //else: some other worker thread is doing GUI
839 bool WXDLLEXPORT
wxGuiOwnedByMainThread()
841 return s_bGuiOwnedByMainThread
;
844 // wake up the main thread if it's in ::GetMessage()
845 void WXDLLEXPORT
wxWakeUpMainThread()
847 // sending any message would do - hopefully WM_NULL is harmless enough
848 if ( !::PostThreadMessage(s_idMainThread
, WM_NULL
, 0, 0) )
850 // should never happen
851 wxLogLastError("PostThreadMessage(WM_NULL)");
855 bool WXDLLEXPORT
wxIsWaitingForThread()
857 return s_waitingForThread
;
860 #endif // wxUSE_THREADS