michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "mozilla/WidgetTraceEvent.h"
michael@0: #include "mozilla/StaticPtr.h"
michael@0: #include "nsThreadUtils.h"
michael@0: #include <mozilla/CondVar.h>
michael@0: #include <mozilla/Mutex.h>
michael@0: 
michael@0: using mozilla::CondVar;
michael@0: using mozilla::Mutex;
michael@0: using mozilla::MutexAutoLock;
michael@0: 
michael@0: namespace mozilla {
michael@0:   class TracerRunnable : public nsRunnable {
michael@0:     public:
michael@0:       TracerRunnable() {
michael@0:         mTracerLock = new Mutex("TracerRunnable");
michael@0:         mTracerCondVar = new CondVar(*mTracerLock, "TracerRunnable");
michael@0:         mMainThread = do_GetMainThread();
michael@0:       }
michael@0: 
michael@0:       ~TracerRunnable() {
michael@0:         delete mTracerCondVar;
michael@0:         delete mTracerLock;
michael@0:         mTracerLock = nullptr;
michael@0:         mTracerCondVar = nullptr;
michael@0:       }
michael@0: 
michael@0:       virtual nsresult Run() {
michael@0:         MutexAutoLock lock(*mTracerLock);
michael@0:         mHasRun = true;
michael@0:         mTracerCondVar->Notify();
michael@0:         return NS_OK;
michael@0:       }
michael@0: 
michael@0:       bool Fire() {
michael@0:         if (!mTracerLock || !mTracerCondVar) {
michael@0:           return false;
michael@0:         }
michael@0: 
michael@0:         MutexAutoLock lock(*mTracerLock);
michael@0:         mHasRun = false;
michael@0:         mMainThread->Dispatch(this, NS_DISPATCH_NORMAL);
michael@0:         while (!mHasRun) {
michael@0:           mTracerCondVar->Wait();
michael@0:         }
michael@0:         return true;
michael@0:       }
michael@0: 
michael@0:       void Signal() {
michael@0:         MutexAutoLock lock(*mTracerLock);
michael@0:         mHasRun = true;
michael@0:         mTracerCondVar->Notify();
michael@0:       }
michael@0: 
michael@0:     private:
michael@0:       Mutex* mTracerLock;
michael@0:       CondVar* mTracerCondVar;
michael@0:       bool mHasRun;
michael@0:       nsCOMPtr<nsIThread> mMainThread;
michael@0:   };
michael@0: 
michael@0:   StaticRefPtr<TracerRunnable> sTracerRunnable;
michael@0: 
michael@0:   bool InitWidgetTracing()
michael@0:   {
michael@0:     if (!sTracerRunnable) {
michael@0:       sTracerRunnable = new TracerRunnable();
michael@0:     }
michael@0:     return true;
michael@0:   }
michael@0: 
michael@0:   void CleanUpWidgetTracing()
michael@0:   {
michael@0:     sTracerRunnable = nullptr;
michael@0:   }
michael@0: 
michael@0:   bool FireAndWaitForTracerEvent()
michael@0:   {
michael@0:     if (sTracerRunnable) {
michael@0:       return sTracerRunnable->Fire();
michael@0:     }
michael@0: 
michael@0:     return false;
michael@0:   }
michael@0: 
michael@0:   void SignalTracerThread()
michael@0:   {
michael@0:     if (sTracerRunnable) {
michael@0:       return sTracerRunnable->Signal();
michael@0:     }
michael@0:   }
michael@0: }  // namespace mozilla
michael@0: