diff --git a/layout/base/nsPresContext.cpp b/layout/base/nsPresContext.cpp index 1393f85847cd..97dd720ff1ca 100644 --- a/layout/base/nsPresContext.cpp +++ b/layout/base/nsPresContext.cpp @@ -259,6 +259,10 @@ nsPresContext::~nsPresContext() NS_PRECONDITION(!mShell, "Presshell forgot to clear our mShell pointer"); SetShell(nsnull); + if (mRefreshDriver) { + mRefreshDriver->Disconnect(); + } + delete mTransitionManager; if (mEventManager) { @@ -873,6 +877,12 @@ nsPresContext::Init(nsIDeviceContext* aDeviceContext) NS_ADDREF(mEventManager); mTransitionManager = new nsTransitionManager(this); + if (!mTransitionManager) + return NS_ERROR_OUT_OF_MEMORY; + + mRefreshDriver = new nsRefreshDriver(this); + if (!mRefreshDriver) + return NS_ERROR_OUT_OF_MEMORY; mLangService = do_GetService(NS_LANGUAGEATOMSERVICE_CONTRACTID); diff --git a/layout/base/nsPresContext.h b/layout/base/nsPresContext.h index 1c4b03702195..407dd37084aa 100644 --- a/layout/base/nsPresContext.h +++ b/layout/base/nsPresContext.h @@ -72,7 +72,6 @@ #include "nsContentUtils.h" #include "nsIWidget.h" #include "mozilla/TimeStamp.h" -#include "nsRefreshDriver.h" class nsImageLoader; #ifdef IBMBIDI @@ -103,6 +102,7 @@ class nsUserFontSet; struct nsFontFaceRuleContainer; class nsObjectFrame; class nsTransitionManager; +class nsRefreshDriver; class imgIContainer; #ifdef MOZ_REFLOW_PERF @@ -233,13 +233,7 @@ public: nsTransitionManager* TransitionManager() { return mTransitionManager; } - nsRefreshDriver* RefreshDriver() { return &mRefreshDriver; } - - static nsPresContext* FromRefreshDriver(nsRefreshDriver* aRefreshDriver) { - return reinterpret_cast( - reinterpret_cast(aRefreshDriver) - - offsetof(nsPresContext, mRefreshDriver)); - } + nsRefreshDriver* RefreshDriver() { return mRefreshDriver; } #endif /** @@ -957,7 +951,7 @@ protected: // from gfx back to layout. nsIEventStateManager* mEventManager; // [STRONG] nsILookAndFeel* mLookAndFeel; // [STRONG] - nsRefreshDriver mRefreshDriver; + nsRefPtr mRefreshDriver; nsTransitionManager* mTransitionManager; // owns; it aggregates our refcount nsIAtom* mMedium; // initialized by subclass ctors; // weak pointer to static atom diff --git a/layout/base/nsRefreshDriver.cpp b/layout/base/nsRefreshDriver.cpp index e6c36da572dd..0cf9fab261d2 100644 --- a/layout/base/nsRefreshDriver.cpp +++ b/layout/base/nsRefreshDriver.cpp @@ -56,7 +56,8 @@ using mozilla::TimeStamp; -nsRefreshDriver::nsRefreshDriver() +nsRefreshDriver::nsRefreshDriver(nsPresContext *aPresContext) + : mPresContext(aPresContext) { } @@ -170,11 +171,7 @@ nsRefreshDriver::ArrayFor(mozFlushType aFlushType) * nsISupports implementation */ -NS_IMPL_ADDREF_USING_AGGREGATOR(nsRefreshDriver, - nsPresContext::FromRefreshDriver(this)) -NS_IMPL_RELEASE_USING_AGGREGATOR(nsRefreshDriver, - nsPresContext::FromRefreshDriver(this)) -NS_IMPL_QUERY_INTERFACE1(nsRefreshDriver, nsITimerCallback) +NS_IMPL_ISUPPORTS1(nsRefreshDriver, nsITimerCallback) /* * nsITimerCallback implementation @@ -185,8 +182,12 @@ nsRefreshDriver::Notify(nsITimer *aTimer) { UpdateMostRecentRefresh(); - nsPresContext *presContext = nsPresContext::FromRefreshDriver(this); - nsCOMPtr presShell = presContext->GetPresShell(); + if (!mPresContext) { + // Things are being destroyed. + NS_ABORT_IF_FALSE(!mTimer, "timer should have been stopped"); + return NS_OK; + } + nsCOMPtr presShell = mPresContext->GetPresShell(); if (!presShell) { // Things are being destroyed. StopTimer(); diff --git a/layout/base/nsRefreshDriver.h b/layout/base/nsRefreshDriver.h index d767b99cb85d..41f535d24260 100644 --- a/layout/base/nsRefreshDriver.h +++ b/layout/base/nsRefreshDriver.h @@ -49,6 +49,8 @@ #include "nsCOMPtr.h" #include "nsTObserverArray.h" +class nsPresContext; + /** * An abstract base class to be implemented by callers wanting to be * notified at refresh times. When nothing needs to be painted, callers @@ -59,16 +61,17 @@ public: virtual void WillRefresh(mozilla::TimeStamp aTime) = 0; }; -/* - * nsRefreshDriver MUST ONLY be constructed as a sub-object of - * nsPresContext (since its reference counting methods forward to the - * pres context of which it is an mRefreshDriver) - */ -class nsRefreshDriver : private nsITimerCallback { +class nsRefreshDriver : public nsITimerCallback { public: - nsRefreshDriver(); + nsRefreshDriver(nsPresContext *aPresContext); ~nsRefreshDriver(); + // nsISupports implementation + NS_DECL_ISUPPORTS + + // nsITimerCallback implementation + NS_DECL_NSITIMERCALLBACK + /** * Return the time of the most recent refresh. This is intended to be * used by callers who want to start an animation now and want to know @@ -95,13 +98,18 @@ public: mozFlushType aFlushType); PRBool RemoveRefreshObserver(nsARefreshObserver *aObserver, mozFlushType aFlushType); + + /** + * Tell the refresh driver that it is done driving refreshes and + * should stop its timer and forget about its pres context. This may + * be called from within a refresh. + */ + void Disconnect() { + StopTimer(); + mPresContext = nsnull; + } + private: - // nsISupports implementation - NS_DECL_ISUPPORTS_INHERITED - - // nsITimerCallback implementation - NS_IMETHOD Notify(nsITimer *aTimer); - typedef nsTObserverArray ObserverArray; void EnsureTimerStarted(); @@ -113,6 +121,9 @@ private: nsCOMPtr mTimer; mozilla::TimeStamp mMostRecentRefresh; // only valid when mTimer non-null + nsPresContext *mPresContext; // weak; pres context passed in constructor + // and unset in Disconnect + // separate arrays for each flush type we support ObserverArray mObservers[3]; };