michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* vim:set ts=2 sw=2 sts=2 et cindent: */
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 "GeckoTaskTracerImpl.h"
michael@0: #include "TracedTaskCommon.h"
michael@0: 
michael@0: namespace mozilla {
michael@0: namespace tasktracer {
michael@0: 
michael@0: TracedTaskCommon::TracedTaskCommon()
michael@0:   : mSourceEventId(0)
michael@0:   , mSourceEventType(SourceEventType::UNKNOWN)
michael@0: {
michael@0:   Init();
michael@0: }
michael@0: 
michael@0: void
michael@0: TracedTaskCommon::Init()
michael@0: {
michael@0:   TraceInfo* info = GetOrCreateTraceInfo();
michael@0:   NS_ENSURE_TRUE_VOID(info);
michael@0: 
michael@0:   mTaskId = GenNewUniqueTaskId();
michael@0:   mSourceEventId = info->mCurTraceSourceId;
michael@0:   mSourceEventType = info->mCurTraceSourceType;
michael@0: 
michael@0:   LogDispatch(mTaskId, info->mCurTaskId, mSourceEventId, mSourceEventType);
michael@0: }
michael@0: 
michael@0: void
michael@0: TracedTaskCommon::SetTraceInfo()
michael@0: {
michael@0:   TraceInfo* info = GetOrCreateTraceInfo();
michael@0:   NS_ENSURE_TRUE_VOID(info);
michael@0: 
michael@0:   info->mCurTraceSourceId = mSourceEventId;
michael@0:   info->mCurTraceSourceType = mSourceEventType;
michael@0:   info->mCurTaskId = mTaskId;
michael@0: }
michael@0: 
michael@0: void
michael@0: TracedTaskCommon::ClearTraceInfo()
michael@0: {
michael@0:   TraceInfo* info = GetOrCreateTraceInfo();
michael@0:   NS_ENSURE_TRUE_VOID(info);
michael@0: 
michael@0:   info->mCurTraceSourceId = 0;
michael@0:   info->mCurTraceSourceType = SourceEventType::UNKNOWN;
michael@0:   info->mCurTaskId = 0;
michael@0: }
michael@0: 
michael@0: /**
michael@0:  * Implementation of class TracedRunnable.
michael@0:  */
michael@0: TracedRunnable::TracedRunnable(nsIRunnable* aOriginalObj)
michael@0:   : TracedTaskCommon()
michael@0:   , mOriginalObj(aOriginalObj)
michael@0: {
michael@0:   LogVirtualTablePtr(mTaskId, mSourceEventId, *(int**)(aOriginalObj));
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: TracedRunnable::Run()
michael@0: {
michael@0:   LogBegin(mTaskId, mSourceEventId);
michael@0: 
michael@0:   SetTraceInfo();
michael@0:   nsresult rv = mOriginalObj->Run();
michael@0:   ClearTraceInfo();
michael@0: 
michael@0:   LogEnd(mTaskId, mSourceEventId);
michael@0:   return rv;
michael@0: }
michael@0: 
michael@0: /**
michael@0:  * Implementation of class TracedTask.
michael@0:  */
michael@0: TracedTask::TracedTask(Task* aOriginalObj)
michael@0:   : TracedTaskCommon()
michael@0:   , mOriginalObj(aOriginalObj)
michael@0: {
michael@0:   LogVirtualTablePtr(mTaskId, mSourceEventId, *(int**)(aOriginalObj));
michael@0: }
michael@0: 
michael@0: void
michael@0: TracedTask::Run()
michael@0: {
michael@0:   LogBegin(mTaskId, mSourceEventId);
michael@0: 
michael@0:   SetTraceInfo();
michael@0:   mOriginalObj->Run();
michael@0:   ClearTraceInfo();
michael@0: 
michael@0:   LogEnd(mTaskId, mSourceEventId);
michael@0: }
michael@0: 
michael@0: FakeTracedTask::FakeTracedTask(int* aVptr)
michael@0:   : TracedTaskCommon()
michael@0: {
michael@0:   LogVirtualTablePtr(mTaskId, mSourceEventId, aVptr);
michael@0: }
michael@0: 
michael@0: FakeTracedTask::FakeTracedTask(const FakeTracedTask& aTask)
michael@0: {
michael@0:   mTaskId = aTask.mTaskId;
michael@0:   mSourceEventId = aTask.mSourceEventId;
michael@0:   mSourceEventType = aTask.mSourceEventType;
michael@0: }
michael@0: 
michael@0: void
michael@0: FakeTracedTask::BeginFakeTracedTask()
michael@0: {
michael@0:   LogBegin(mTaskId, mSourceEventId);
michael@0:   SetTraceInfo();
michael@0: }
michael@0: 
michael@0: void
michael@0: FakeTracedTask::EndFakeTracedTask()
michael@0: {
michael@0:   ClearTraceInfo();
michael@0:   LogEnd(mTaskId, mSourceEventId);
michael@0: }
michael@0: 
michael@0: AutoRunFakeTracedTask::AutoRunFakeTracedTask(FakeTracedTask* aFakeTracedTask)
michael@0:   : mInitialized(false)
michael@0: {
michael@0:   if (aFakeTracedTask) {
michael@0:     mInitialized = true;
michael@0:     mFakeTracedTask = *aFakeTracedTask;
michael@0:     mFakeTracedTask.BeginFakeTracedTask();
michael@0:   }
michael@0: }
michael@0: 
michael@0: AutoRunFakeTracedTask::~AutoRunFakeTracedTask()
michael@0: {
michael@0:   if (mInitialized) {
michael@0:     mFakeTracedTask.EndFakeTracedTask();
michael@0:   }
michael@0: }
michael@0: 
michael@0: /**
michael@0:  * CreateTracedRunnable() returns a TracedRunnable wrapping the original
michael@0:  * nsIRunnable object, aRunnable.
michael@0:  */
michael@0: already_AddRefed<nsIRunnable>
michael@0: CreateTracedRunnable(nsIRunnable* aRunnable)
michael@0: {
michael@0:   nsCOMPtr<nsIRunnable> runnable = new TracedRunnable(aRunnable);
michael@0:   return runnable.forget();
michael@0: }
michael@0: 
michael@0: /**
michael@0:  * CreateTracedTask() returns a TracedTask wrapping the original Task object,
michael@0:  * aTask.
michael@0:  */
michael@0: Task*
michael@0: CreateTracedTask(Task* aTask)
michael@0: {
michael@0:   Task* task = new TracedTask(aTask);
michael@0:   return task;
michael@0: }
michael@0: 
michael@0: /**
michael@0:  * CreateFakeTracedTask() returns a FakeTracedTask tracking the event which is
michael@0:  * not dispatched from its parent task directly, such as timer events.
michael@0:  */
michael@0: FakeTracedTask*
michael@0: CreateFakeTracedTask(int* aVptr)
michael@0: {
michael@0:   nsAutoPtr<FakeTracedTask> task(new FakeTracedTask(aVptr));
michael@0:   return task.forget();
michael@0: }
michael@0: 
michael@0: } // namespace tasktracer
michael@0: } // namespace mozilla