X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/77f62aa7b6d354c46fce7829fe90a08b0a0ee803..f8e1a81f3e2fdad47eb014c8d74d919c7a9faaa4:/src/common/dcbufcmn.cpp

diff --git a/src/common/dcbufcmn.cpp b/src/common/dcbufcmn.cpp
index eeea10859f..ed6bb0e7bb 100644
--- a/src/common/dcbufcmn.cpp
+++ b/src/common/dcbufcmn.cpp
@@ -24,50 +24,49 @@
     #pragma hdrstop
 #endif
 
-#ifndef WX_PRECOMP
-#endif
-
 #include "wx/dcbuffer.h"
 
+#ifndef WX_PRECOMP
+    #include "wx/module.h"
+#endif
 
 // ============================================================================
 // implementation
 // ============================================================================
 
-// ============================================================================
-// wxSharedDCBufferManager
-//   Helper class to free shared buffer when the app exists.
-// ============================================================================
+// ----------------------------------------------------------------------------
+// wxSharedDCBufferManager: helper class maintaining backing store bitmap
+// ----------------------------------------------------------------------------
 
-class wxSharedDCBufferManager
+class wxSharedDCBufferManager : public wxModule
 {
-    friend class wxBufferedDC;
 public:
-
     wxSharedDCBufferManager() { }
-    ~wxSharedDCBufferManager() { }
 
-    wxBitmap* GetBuffer(int w, int h)
+    virtual bool OnInit() { return true; }
+    virtual void OnExit() { wxDELETE(ms_buffer); }
+
+    static wxBitmap* GetBuffer(int w, int h)
     {
-        if ( !m_buffer.IsOk() ||
-             w > m_buffer.GetWidth() ||
-             h > m_buffer.GetHeight() )
+        if ( !ms_buffer ||
+                w > ms_buffer->GetWidth() ||
+                    h > ms_buffer->GetHeight() )
         {
-            // Create slightly larger bitmap so we don't need to
-            // be reallocating constantly when the user enlarges
-            // the frame for the first time.
-            m_buffer = wxBitmap(w, h);
+            delete ms_buffer;
+            ms_buffer = new wxBitmap(w, h);
         }
-
-        return &m_buffer;
+        return ms_buffer;
     }
 
 private:
-    wxBitmap    m_buffer;
+    static wxBitmap *ms_buffer;
+
+    DECLARE_DYNAMIC_CLASS(wxSharedDCBufferManager)
 };
 
-static wxSharedDCBufferManager gs_sharedDCBufferManager;
+wxBitmap* wxSharedDCBufferManager::ms_buffer = NULL;
 
+IMPLEMENT_DYNAMIC_CLASS(wxSharedDCBufferManager, wxModule)
 
 // ============================================================================
 // wxBufferedDC
@@ -75,12 +74,12 @@ static wxSharedDCBufferManager gs_sharedDCBufferManager;
 
 void wxBufferedDC::UseBuffer(wxCoord w, wxCoord h)
 {
-    if ( !m_buffer )
+    if ( !m_buffer || !m_buffer->IsOk() )
     {
         if ( w == -1 || h == -1 )
             m_dc->GetSize(&w, &h);
 
-        m_buffer = gs_sharedDCBufferManager.GetBuffer(w, h);
+        m_buffer = wxSharedDCBufferManager::GetBuffer(w, h);
     }
 
     SelectObject(*m_buffer);