1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/tools/profiler/TracedTaskCommon.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,178 @@
1.4 +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
1.5 +/* vim:set ts=2 sw=2 sts=2 et cindent: */
1.6 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.7 + * License, v. 2.0. If a copy of the MPL was not distributed with this
1.8 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
1.9 +
1.10 +#include "GeckoTaskTracerImpl.h"
1.11 +#include "TracedTaskCommon.h"
1.12 +
1.13 +namespace mozilla {
1.14 +namespace tasktracer {
1.15 +
1.16 +TracedTaskCommon::TracedTaskCommon()
1.17 + : mSourceEventId(0)
1.18 + , mSourceEventType(SourceEventType::UNKNOWN)
1.19 +{
1.20 + Init();
1.21 +}
1.22 +
1.23 +void
1.24 +TracedTaskCommon::Init()
1.25 +{
1.26 + TraceInfo* info = GetOrCreateTraceInfo();
1.27 + NS_ENSURE_TRUE_VOID(info);
1.28 +
1.29 + mTaskId = GenNewUniqueTaskId();
1.30 + mSourceEventId = info->mCurTraceSourceId;
1.31 + mSourceEventType = info->mCurTraceSourceType;
1.32 +
1.33 + LogDispatch(mTaskId, info->mCurTaskId, mSourceEventId, mSourceEventType);
1.34 +}
1.35 +
1.36 +void
1.37 +TracedTaskCommon::SetTraceInfo()
1.38 +{
1.39 + TraceInfo* info = GetOrCreateTraceInfo();
1.40 + NS_ENSURE_TRUE_VOID(info);
1.41 +
1.42 + info->mCurTraceSourceId = mSourceEventId;
1.43 + info->mCurTraceSourceType = mSourceEventType;
1.44 + info->mCurTaskId = mTaskId;
1.45 +}
1.46 +
1.47 +void
1.48 +TracedTaskCommon::ClearTraceInfo()
1.49 +{
1.50 + TraceInfo* info = GetOrCreateTraceInfo();
1.51 + NS_ENSURE_TRUE_VOID(info);
1.52 +
1.53 + info->mCurTraceSourceId = 0;
1.54 + info->mCurTraceSourceType = SourceEventType::UNKNOWN;
1.55 + info->mCurTaskId = 0;
1.56 +}
1.57 +
1.58 +/**
1.59 + * Implementation of class TracedRunnable.
1.60 + */
1.61 +TracedRunnable::TracedRunnable(nsIRunnable* aOriginalObj)
1.62 + : TracedTaskCommon()
1.63 + , mOriginalObj(aOriginalObj)
1.64 +{
1.65 + LogVirtualTablePtr(mTaskId, mSourceEventId, *(int**)(aOriginalObj));
1.66 +}
1.67 +
1.68 +NS_IMETHODIMP
1.69 +TracedRunnable::Run()
1.70 +{
1.71 + LogBegin(mTaskId, mSourceEventId);
1.72 +
1.73 + SetTraceInfo();
1.74 + nsresult rv = mOriginalObj->Run();
1.75 + ClearTraceInfo();
1.76 +
1.77 + LogEnd(mTaskId, mSourceEventId);
1.78 + return rv;
1.79 +}
1.80 +
1.81 +/**
1.82 + * Implementation of class TracedTask.
1.83 + */
1.84 +TracedTask::TracedTask(Task* aOriginalObj)
1.85 + : TracedTaskCommon()
1.86 + , mOriginalObj(aOriginalObj)
1.87 +{
1.88 + LogVirtualTablePtr(mTaskId, mSourceEventId, *(int**)(aOriginalObj));
1.89 +}
1.90 +
1.91 +void
1.92 +TracedTask::Run()
1.93 +{
1.94 + LogBegin(mTaskId, mSourceEventId);
1.95 +
1.96 + SetTraceInfo();
1.97 + mOriginalObj->Run();
1.98 + ClearTraceInfo();
1.99 +
1.100 + LogEnd(mTaskId, mSourceEventId);
1.101 +}
1.102 +
1.103 +FakeTracedTask::FakeTracedTask(int* aVptr)
1.104 + : TracedTaskCommon()
1.105 +{
1.106 + LogVirtualTablePtr(mTaskId, mSourceEventId, aVptr);
1.107 +}
1.108 +
1.109 +FakeTracedTask::FakeTracedTask(const FakeTracedTask& aTask)
1.110 +{
1.111 + mTaskId = aTask.mTaskId;
1.112 + mSourceEventId = aTask.mSourceEventId;
1.113 + mSourceEventType = aTask.mSourceEventType;
1.114 +}
1.115 +
1.116 +void
1.117 +FakeTracedTask::BeginFakeTracedTask()
1.118 +{
1.119 + LogBegin(mTaskId, mSourceEventId);
1.120 + SetTraceInfo();
1.121 +}
1.122 +
1.123 +void
1.124 +FakeTracedTask::EndFakeTracedTask()
1.125 +{
1.126 + ClearTraceInfo();
1.127 + LogEnd(mTaskId, mSourceEventId);
1.128 +}
1.129 +
1.130 +AutoRunFakeTracedTask::AutoRunFakeTracedTask(FakeTracedTask* aFakeTracedTask)
1.131 + : mInitialized(false)
1.132 +{
1.133 + if (aFakeTracedTask) {
1.134 + mInitialized = true;
1.135 + mFakeTracedTask = *aFakeTracedTask;
1.136 + mFakeTracedTask.BeginFakeTracedTask();
1.137 + }
1.138 +}
1.139 +
1.140 +AutoRunFakeTracedTask::~AutoRunFakeTracedTask()
1.141 +{
1.142 + if (mInitialized) {
1.143 + mFakeTracedTask.EndFakeTracedTask();
1.144 + }
1.145 +}
1.146 +
1.147 +/**
1.148 + * CreateTracedRunnable() returns a TracedRunnable wrapping the original
1.149 + * nsIRunnable object, aRunnable.
1.150 + */
1.151 +already_AddRefed<nsIRunnable>
1.152 +CreateTracedRunnable(nsIRunnable* aRunnable)
1.153 +{
1.154 + nsCOMPtr<nsIRunnable> runnable = new TracedRunnable(aRunnable);
1.155 + return runnable.forget();
1.156 +}
1.157 +
1.158 +/**
1.159 + * CreateTracedTask() returns a TracedTask wrapping the original Task object,
1.160 + * aTask.
1.161 + */
1.162 +Task*
1.163 +CreateTracedTask(Task* aTask)
1.164 +{
1.165 + Task* task = new TracedTask(aTask);
1.166 + return task;
1.167 +}
1.168 +
1.169 +/**
1.170 + * CreateFakeTracedTask() returns a FakeTracedTask tracking the event which is
1.171 + * not dispatched from its parent task directly, such as timer events.
1.172 + */
1.173 +FakeTracedTask*
1.174 +CreateFakeTracedTask(int* aVptr)
1.175 +{
1.176 + nsAutoPtr<FakeTracedTask> task(new FakeTracedTask(aVptr));
1.177 + return task.forget();
1.178 +}
1.179 +
1.180 +} // namespace tasktracer
1.181 +} // namespace mozilla
