#include "wx/debug.h"
#include "wx/filefn.h"
#include "wx/log.h"
+ #include "wx/thread.h"
#endif
+#include "wx/init.h"
+
+#include "wx/ptr_scpd.h"
#include "wx/module.h"
+#include "wx/except.h"
+
+#if defined(__WXMSW__) && defined(__WXDEBUG__)
+ #include "wx/msw/msvcrt.h"
+
+ static struct EnableMemLeakChecking
+ {
+ EnableMemLeakChecking()
+ {
+ // do check for memory leaks on program exit (another useful flag
+ // is _CRTDBG_DELAY_FREE_MEM_DF which doesn't free deallocated
+ // memory which may be used to simulate low-memory condition)
+ wxCrtSetDbgFlag(_CRTDBG_LEAK_CHECK_DF);
+ }
+ } gs_enableLeakChecks;
+#endif // __WXMSW__ && __WXDEBUG__
// ----------------------------------------------------------------------------
// private classes
// ----------------------------------------------------------------------------
// we need a dummy app object if the user doesn't want to create a real one
-class wxDummyConsoleApp : public wxApp
+class wxDummyConsoleApp : public wxAppConsole
{
public:
+ wxDummyConsoleApp() { }
+
virtual int OnRun() { wxFAIL_MSG( _T("unreachable code") ); return 0; }
+
+ DECLARE_NO_COPY_CLASS(wxDummyConsoleApp)
+};
+
+// we need a special kind of auto pointer to wxApp which not only deletes the
+// pointer it holds in its dtor but also resets the global application pointer
+wxDECLARE_SCOPED_PTR(wxAppConsole, wxAppPtrBase);
+wxDEFINE_SCOPED_PTR(wxAppConsole, wxAppPtrBase);
+
+class wxAppPtr : public wxAppPtrBase
+{
+public:
+ wxEXPLICIT wxAppPtr(wxAppConsole *ptr = NULL) : wxAppPtrBase(ptr) { }
+ ~wxAppPtr()
+ {
+ if ( get() )
+ {
+ // the pointer is going to be deleted in the base class dtor, don't
+ // leave the dangling pointer!
+ wxApp::SetInstance(NULL);
+ }
+ }
+
+ void Set(wxAppConsole *ptr)
+ {
+ reset(ptr);
+
+ wxApp::SetInstance(ptr);
+ }
+
+ DECLARE_NO_COPY_CLASS(wxAppPtr)
+};
+
+// class to ensure that wxAppBase::CleanUp() is called if our Initialize()
+// fails
+class wxCallAppCleanup
+{
+public:
+ wxCallAppCleanup(wxAppConsole *app) : m_app(app) { }
+ ~wxCallAppCleanup() { if ( m_app ) m_app->CleanUp(); }
+
+ void Dismiss() { m_app = NULL; }
+
+private:
+ wxAppConsole *m_app;
+};
+
+// another tiny class which simply exists to ensure that wxEntryCleanup is
+// always called
+class wxCleanupOnExit
+{
+public:
+ ~wxCleanupOnExit() { wxEntryCleanup(); }
};
// ----------------------------------------------------------------------------
// private functions
// ----------------------------------------------------------------------------
-static bool DoInit();
-static void DoCleanUp();
+// suppress warnings about unused variables
+static inline void Use(void *) { }
+
+#define WX_SUPPRESS_UNUSED_WARN(x) Use(&x)
// ----------------------------------------------------------------------------
-// private vars
+// initialization data
// ----------------------------------------------------------------------------
-static size_t gs_nInitCount = 0;
+static struct InitData
+{
+ InitData()
+ {
+ nInitCount = 0;
+
+#if wxUSE_UNICODE
+ argc = 0;
+ // argv = NULL; -- not even really needed
+#endif // wxUSE_UNICODE
+ }
+
+ // critical section protecting this struct
+ wxCRIT_SECT_DECLARE_MEMBER(csInit);
+
+ // number of times wxInitialize() was called minus the number of times
+ // wxUninitialize() was
+ size_t nInitCount;
+
+#if wxUSE_UNICODE
+ int argc;
+
+ // if we receive the command line arguments as ASCII and have to convert
+ // them to Unicode ourselves (this is the case under Unix but not Windows,
+ // for example), we remember the converted argv here because we'll have to
+ // free it when doing cleanup to avoid memory leaks
+ wchar_t **argv;
+#endif // wxUSE_UNICODE
+
+ DECLARE_NO_COPY_CLASS(InitData)
+} gs_initData;
// ============================================================================
// implementation
// ============================================================================
// ----------------------------------------------------------------------------
-// wxBase-specific functions
+// command line arguments ANSI -> Unicode conversion
// ----------------------------------------------------------------------------
-bool WXDLLEXPORT wxInitialize()
+#if wxUSE_UNICODE
+
+static void ConvertArgsToUnicode(int argc, char **argv)
{
- if ( gs_nInitCount )
+ gs_initData.argv = new wchar_t *[argc + 1];
+ for ( int i = 0; i < argc; i++ )
{
- // already initialized
- return TRUE;
+ gs_initData.argv[i] = wxStrdup(wxConvLocal.cMB2WX(argv[i]));
}
- wxASSERT_MSG( !wxTheApp,
- wxT("either call wxInitialize or create app, not both!") );
+ gs_initData.argc = argc;
+ gs_initData.argv[argc] = NULL;
+}
- if ( !DoInit() )
+static void FreeConvertedArgs()
+{
+ if ( gs_initData.argv )
{
- return FALSE;
+ for ( int i = 0; i < gs_initData.argc; i++ )
+ {
+ free(gs_initData.argv[i]);
+ }
+
+ delete [] gs_initData.argv;
+ gs_initData.argv = NULL;
+ gs_initData.argc = 0;
}
+}
- wxTheApp = new wxDummyConsoleApp;
-
- if ( !wxTheApp )
- {
- return FALSE;
- }
+#endif // wxUSE_UNICODE
- gs_nInitCount++;
+// ----------------------------------------------------------------------------
+// start up
+// ----------------------------------------------------------------------------
- return TRUE;
+// initialization which is always done (not customizable) before wxApp creation
+static bool DoCommonPreInit()
+{
+ return true;
}
-void WXDLLEXPORT wxUninitialize()
+// non customizable initialization done after wxApp creation and initialization
+static bool DoCommonPostInit()
{
- if ( !--gs_nInitCount )
- {
- DoCleanUp();
- }
+ wxModule::RegisterModules();
+
+ return wxModule::InitializeModules();
}
-int wxEntry(int argc, char **argv)
+bool wxEntryStart(int& argc, wxChar **argv)
{
- // library initialization
- if ( !DoInit() )
+ // do minimal, always necessary, initialization
+ // --------------------------------------------
+
+ // initialize wxRTTI
+ if ( !DoCommonPreInit() )
{
- return -1;
+ return false;
}
- // create the app
- if ( !wxTheApp )
- {
- wxCHECK_MSG( wxApp::GetInitializerFunction(), -1,
- wxT("No application object: use IMPLEMENT_APP macro.") );
+ // first of all, we need an application object
+ // -------------------------------------------
+
+ // the user might have already created it himself somehow
+ wxAppPtr app(wxTheApp);
+ if ( !app.get() )
+ {
+ // if not, he might have used IMPLEMENT_APP() to give us a function to
+ // create it
wxAppInitializerFunction fnCreate = wxApp::GetInitializerFunction();
- wxTheApp = (wxApp *)fnCreate();
+ if ( fnCreate )
+ {
+ // he did, try to create the custom wxApp object
+ app.Set((*fnCreate)());
+ }
}
- wxCHECK_MSG( wxTheApp, -1, wxT("wxWindows error: no application object") );
+ if ( !app.get() )
+ {
+ // either IMPLEMENT_APP() was not used at all or it failed -- in any
+ // case we still need something
+ app.Set(new wxDummyConsoleApp);
+ }
- // app preinitialization
- wxTheApp->argc = argc;
-#if wxUSE_UNICODE
- wxTheApp->argv = new wxChar*[argc+1];
- for ( int mb_argc = 0; mb_argc < argc; mb_argc++ )
+ // wxApp initialization: this can be customized
+ // --------------------------------------------
+
+ if ( !app->Initialize(argc, argv) )
{
- wxTheApp->argv[mb_argc] = wxStrdup(wxConvLocal.cMB2WX(argv[mb_argc]));
+ return false;
}
- wxTheApp->argv[mb_argc] = (wxChar *)NULL;
-#else
- wxTheApp->argv = argv;
-#endif
- wxString name = wxFileNameFromPath(wxTheApp->argv[0]);
- wxStripExtension(name);
- wxTheApp->SetAppName(name);
+ wxCallAppCleanup callAppCleanup(app.get());
- int retValue = 0;
+ // for compatibility call the old initialization function too
+ if ( !app->OnInitGui() )
+ return false;
- // app initialization
- if ( !wxTheApp->OnInit() )
- retValue = -1;
- // app execution
- if ( retValue == 0 )
- {
- retValue = wxTheApp->OnRun();
+ // common initialization after wxTheApp creation
+ // ---------------------------------------------
- // app clean up
- wxTheApp->OnExit();
- }
+ if ( !DoCommonPostInit() )
+ return false;
+
+
+ // prevent the smart pointer from destroying its contents
+ app.release();
- // library clean up
- DoCleanUp();
+ // and the cleanup object from doing cleanup
+ callAppCleanup.Dismiss();
- return retValue;
+ return true;
}
-// ----------------------------------------------------------------------------
-// private functions
-// ----------------------------------------------------------------------------
+#if wxUSE_UNICODE
-static bool DoInit()
+// we provide a wxEntryStart() wrapper taking "char *" pointer too
+bool wxEntryStart(int& argc, char **argv)
{
- wxClassInfo::InitializeClasses();
+ ConvertArgsToUnicode(argc, argv);
- wxModule::RegisterModules();
- if ( !wxModule::InitializeModules() )
+ if ( !wxEntryStart(argc, gs_initData.argv) )
{
- return FALSE;
+ FreeConvertedArgs();
+
+ return false;
}
- return TRUE;
+ return true;
}
-static void DoCleanUp()
+#endif // wxUSE_UNICODE
+
+// ----------------------------------------------------------------------------
+// clean up
+// ----------------------------------------------------------------------------
+
+// cleanup done before destroying wxTheApp
+static void DoCommonPreCleanup()
{
#if wxUSE_LOG
- // flush the logged messages if any
- wxLog *log = wxLog::GetActiveTarget();
- if (log != NULL && log->HasPendingMessages())
- log->Flush();
-
- // continuing to use user defined log target is unsafe from now on because
- // some resources may be already unavailable, so replace it by something
- // more safe
+ // flush the logged messages if any and install a 'safer' log target: the
+ // default one (wxLogGui) can't be used after the resources are freed just
+ // below and the user supplied one might be even more unsafe (using any
+ // wxWidgets GUI function is unsafe starting from now)
wxLog::DontCreateOnDemand();
+
+ // this will flush the old messages if any
delete wxLog::SetActiveTarget(new wxLogStderr);
#endif // wxUSE_LOG
+}
+// cleanup done after destroying wxTheApp
+static void DoCommonPostCleanup()
+{
wxModule::CleanUpModules();
- wxClassInfo::CleanUpClasses();
+ wxClassInfo::CleanUp();
- // TODO: this should really be done in ~wxApp
+ // we can't do this in wxApp itself because it doesn't know if argv had
+ // been allocated
#if wxUSE_UNICODE
- for ( int mb_argc = 0; mb_argc < wxTheApp->argc; mb_argc++ )
- {
- free(wxTheApp->argv[mb_argc]);
- }
+ FreeConvertedArgs();
#endif // wxUSE_UNICODE
- // delete the application object
- delete wxTheApp;
- wxTheApp = (wxApp *)NULL;
-
+ // Note: check for memory leaks is now done via wxDebugContextDumpDelayCounter
#if wxUSE_LOG
// and now delete the last logger as well
delete wxLog::SetActiveTarget(NULL);
#endif // wxUSE_LOG
}
+void wxEntryCleanup()
+{
+ DoCommonPreCleanup();
+
+
+ // delete the application object
+ if ( wxTheApp )
+ {
+ wxTheApp->CleanUp();
+
+ delete wxTheApp;
+ wxApp::SetInstance(NULL);
+ }
+
+
+ DoCommonPostCleanup();
+}
+
+// ----------------------------------------------------------------------------
+// wxEntry
+// ----------------------------------------------------------------------------
+
+// for MSW with wxUSE_ON_FATAL_EXCEPTION on, wxEntry is defined in msw/main.cpp
+#if !defined(__WXMSW__) || !wxUSE_ON_FATAL_EXCEPTION
+ #define wxEntryReal wxEntry
+#endif // !(__WXMSW__ && wxUSE_ON_FATAL_EXCEPTION)
+
+int wxEntryReal(int& argc, wxChar **argv)
+{
+ // library initialization
+ if ( !wxEntryStart(argc, argv) )
+ {
+ return -1;
+ }
+
+ // if wxEntryStart succeeded, we must call wxEntryCleanup even if the code
+ // below returns or throws
+ wxCleanupOnExit cleanupOnExit;
+
+ WX_SUPPRESS_UNUSED_WARN(cleanupOnExit);
+
+ wxTRY
+ {
+
+ // app initialization
+ if ( !wxTheApp->CallOnInit() )
+ {
+ // don't call OnExit() if OnInit() failed
+ return -1;
+ }
+
+ // ensure that OnExit() is called if OnInit() had succeeded
+ class CallOnExit
+ {
+ public:
+ ~CallOnExit() { wxTheApp->OnExit(); }
+ } callOnExit;
+
+ WX_SUPPRESS_UNUSED_WARN(callOnExit);
+
+ // app execution
+ return wxTheApp->OnRun();
+ }
+ wxCATCH_ALL( wxTheApp->OnUnhandledException(); return -1; )
+}
+
+#if wxUSE_UNICODE
+
+// as with wxEntryStart, we provide an ANSI wrapper
+int wxEntry(int& argc, char **argv)
+{
+ ConvertArgsToUnicode(argc, argv);
+
+ return wxEntry(argc, gs_initData.argv);
+}
+
+#endif // wxUSE_UNICODE
+
+// ----------------------------------------------------------------------------
+// wxInitialize/wxUninitialize
+// ----------------------------------------------------------------------------
+
+bool wxInitialize(int argc, wxChar **argv)
+{
+ wxCRIT_SECT_LOCKER(lockInit, gs_initData.csInit);
+
+ if ( gs_initData.nInitCount++ )
+ {
+ // already initialized
+ return true;
+ }
+
+ return wxEntryStart(argc, argv);
+}
+
+void wxUninitialize()
+{
+ wxCRIT_SECT_LOCKER(lockInit, gs_initData.csInit);
+
+ if ( !--gs_initData.nInitCount )
+ {
+ wxEntryCleanup();
+ }
+}
+