X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/5d644707ca96dcaa63dc5a24c9eb5c2b7ee0b50d..fb8d7eb7a880f1f2e32d8830f9c5e12b2536e05f:/src/os2/timer.cpp

diff --git a/src/os2/timer.cpp b/src/os2/timer.cpp
index bb897e1f78..764afc22ab 100644
--- a/src/os2/timer.cpp
+++ b/src/os2/timer.cpp
@@ -1,5 +1,5 @@
 /////////////////////////////////////////////////////////////////////////////
-// Name:        timer.cpp
+// Name:        src/os2/timer.cpp
 // Purpose:     wxTimer implementation
 // Author:      David Webster
 // Modified by:
@@ -12,80 +12,64 @@
 // For compilers that support precompilation, includes "wx.h".
 #include "wx/wxprec.h"
 
-#include "wx/window.h"
-#include "wx/os2/private.h"
+#include "wx/os2/private/timer.h"
 
 #ifndef WX_PRECOMP
-    #include "wx/setup.h"
     #include "wx/list.h"
+    #include "wx/window.h"
     #include "wx/event.h"
     #include "wx/app.h"
+    #include "wx/intl.h"
+    #include "wx/log.h"
 #endif
 
-#include "wx/intl.h"
-#include "wx/log.h"
-
-#include "wx/timer.h"
+#include "wx/os2/private.h"
 
 #include <time.h>
 #include <sys/types.h>
 
 #include <sys/timeb.h>
-// ----------------------------------------------------------------------------
-// private functions
-// ----------------------------------------------------------------------------
-
-wxList wxTimerList(wxKEY_INTEGER);
-ULONG wxTimerProc(HWND hwnd, ULONG, int nIdTimer, ULONG);
 
 // ----------------------------------------------------------------------------
-// macros
+// private globals
 // ----------------------------------------------------------------------------
 
-IMPLEMENT_ABSTRACT_CLASS(wxTimer, wxObject)
+// define a hash containing all the timers: it is indexed by timer id and
+// contains the corresponding timer
+WX_DECLARE_HASH_MAP(unsigned long, wxOS2TimerImpl *, wxIntegerHash, wxIntegerEqual,
+                    wxTimerMap);
 
-void wxTimer::Init()
+// instead of using a global here, wrap it in a static function as otherwise it
+// could have been used before being initialized if a timer object were created
+// globally
+static wxTimerMap& TimerMap()
 {
-    m_ulId = 0;
-}
+    static wxTimerMap s_timerMap;
 
-wxTimer::~wxTimer()
-{
-    Stop();
-    wxTimer::Stop();
-    wxTimerList.DeleteObject(this);
+    return s_timerMap;
 }
 
-void wxTimer::Notify()
-{
-    //
-    // The base class version generates an event if it has owner - which it
-    // should because otherwise nobody can process timer events, but it does
-    // not use the OS's ID, which OS/2 must have to figure out which timer fired
-    //
-    wxCHECK_RET( m_owner, _T("wxTimer::Notify() should be overridden.") );
+// ----------------------------------------------------------------------------
+// private functions
+// ----------------------------------------------------------------------------
 
-    wxTimerEvent                    vEvent( m_idTimer
-                                           ,m_milli
-                                          );
+// timer callback used for all timers
+ULONG wxTimerProc(HWND hwnd, ULONG, int nIdTimer, ULONG);
 
-    (void)m_owner->ProcessEvent(vEvent);
-} // end of wxTimer::Notify
+// ============================================================================
+// implementation
+// ============================================================================
 
-bool wxTimer::Start(
-  int                               nMilliseconds
-, bool                              bOneShot
-)
-{
-    (void)wxTimerBase::Start( nMilliseconds
-                             ,bOneShot
-                            );
-
-    wxCHECK_MSG( m_milli > 0L, FALSE, wxT("invalid value for timer") );
+// ----------------------------------------------------------------------------
+// wxTimer class
+// ----------------------------------------------------------------------------
 
-    wxTimerList.DeleteObject(this);
+bool wxOS2TimerImpl::Start( int nMilliseconds, bool bOneShot )
+{
+    if ( !wxTimerImpl::Start( nMilliseconds, bOneShot ) )
+        return false;
 
-    wxWindow*                       pWin = NULL;
+    wxWindow* pWin = NULL;
 
     if (m_owner)
     {
@@ -97,27 +81,43 @@ bool wxTimer::Start(
                                 );
     }
     else
+    {
         m_ulId = ::WinStartTimer( m_Hab
                                  ,NULLHANDLE
                                  ,0
                                  ,(ULONG)nMilliseconds
                                 );
+    }
+
     if (m_ulId > 0L)
     {
-        wxTimerList.Append( m_ulId
-                           ,this
-                          );
-        return(TRUE);
+        // check that SetTimer() didn't reuse an existing id: according to
+        // the MSDN this can happen and this would be catastrophic to us as
+        // we rely on ids uniquely identifying the timers because we use
+        // them as keys in the hash
+        if ( TimerMap().find(m_ulId) != TimerMap().end() )
+        {
+            wxLogError(_("Timer creation failed."));
+
+            ::WinStopTimer(m_Hab, pWin?(pWin->GetHWND()):NULL, m_ulId);
+            m_ulId = 0;
+
+            return false;
+        }
+
+        TimerMap()[m_ulId] = this;
+
+        return true;
     }
     else
     {
         wxLogSysError(_("Couldn't create a timer"));
 
-        return(FALSE);
+        return false;
     }
 }
 
-void wxTimer::Stop()
+void wxOS2TimerImpl::Stop()
 {
     if ( m_ulId )
     {
@@ -129,7 +129,8 @@ void wxTimer::Stop()
         }
         else
             ::WinStopTimer(m_Hab, NULLHANDLE, m_ulId);
-        wxTimerList.DeleteObject(this);
+
+        TimerMap().erase(m_ulId);
     }
     m_ulId = 0L;
 }
@@ -139,7 +140,7 @@ void wxTimer::Stop()
 // ----------------------------------------------------------------------------
 
 void wxProcessTimer(
-  wxTimer&                          rTimer
+  wxOS2TimerImpl&                          rTimer
 )
 {
     //
@@ -161,11 +162,10 @@ ULONG wxTimerProc(
 , ULONG
 )
 {
-    wxNode*                         pNode = wxTimerList.Find((ULONG)nIdTimer);
+    wxTimerMap::iterator node = TimerMap().find((ULONG)nIdTimer);
 
-    wxCHECK_MSG(pNode, 0, wxT("bogus timer id in wxTimerProc") );
-
-    wxProcessTimer(*(wxTimer *)pNode->Data());
+    wxCHECK_MSG(node != TimerMap().end(), 0,
+                wxT("bogus timer id in wxTimerProc") );
+    wxProcessTimer(*(node->second));
     return 0;
 }
-