/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ /* This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ #include "WebGLContextLossHandler.h" #include "mozilla/DebugOnly.h" #include "nsISupportsImpl.h" #include "nsITimer.h" #include "nsThreadUtils.h" #include "WebGLContext.h" namespace mozilla { class WatchdogTimerEvent final : public nsITimerCallback { const WeakPtr<WebGLContextLossHandler> mHandler; public: NS_DECL_ISUPPORTS explicit WatchdogTimerEvent(WebGLContextLossHandler* handler) : mHandler(handler) { } private: virtual ~WatchdogTimerEvent() { } NS_IMETHOD Notify(nsITimer*) override { if (mHandler) { mHandler->TimerCallback(); } return NS_OK; } }; NS_IMPL_ISUPPORTS(WatchdogTimerEvent, nsITimerCallback, nsISupports) //////////////////////////////////////// WebGLContextLossHandler::WebGLContextLossHandler(WebGLContext* webgl) : mWebGL(webgl) , mTimer(do_CreateInstance(NS_TIMER_CONTRACTID)) , mTimerPending(false) , mShouldRunTimerAgain(false) #ifdef DEBUG , mThread(NS_GetCurrentThread()) #endif { MOZ_ASSERT(mThread); } WebGLContextLossHandler::~WebGLContextLossHandler() { // NS_GetCurrentThread() returns null during shutdown. const DebugOnly<nsIThread*> callingThread = NS_GetCurrentThread(); MOZ_ASSERT(callingThread == mThread || !callingThread); } //////////////////// void WebGLContextLossHandler::RunTimer() { MOZ_ASSERT(NS_GetCurrentThread() == mThread); // If the timer was already running, don't restart it here. Instead, // wait until the previous call is done, then fire it one more time. // This is also an optimization to prevent unnecessary // cross-communication between threads. if (mTimerPending) { mShouldRunTimerAgain = true; return; } const RefPtr<WatchdogTimerEvent> event = new WatchdogTimerEvent(this); const uint32_t kDelayMS = 1000; mTimer->InitWithCallback(event, kDelayMS, nsITimer::TYPE_ONE_SHOT); mTimerPending = true; } //////////////////// void WebGLContextLossHandler::TimerCallback() { MOZ_ASSERT(NS_GetCurrentThread() == mThread); mTimerPending = false; const bool runOnceMore = mShouldRunTimerAgain; mShouldRunTimerAgain = false; mWebGL->UpdateContextLossStatus(); if (runOnceMore && !mTimerPending) { RunTimer(); } } } // namespace mozilla