Source code

Revision control

Other Tools

1
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
2
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
3
/* This Source Code Form is subject to the terms of the Mozilla Public
4
* License, v. 2.0. If a copy of the MPL was not distributed with this
5
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
6
7
#ifdef XP_WIN
8
# undef UNICODE
9
# undef _UNICODE
10
#endif
11
12
#include "VTuneProfiler.h"
13
#include "mozilla/Bootstrap.h"
14
#include <memory>
15
16
VTuneProfiler* VTuneProfiler::mInstance = nullptr;
17
18
void VTuneProfiler::Initialize() {
19
// This is just a 'dirty trick' to find out if the ittnotify DLL was found.
20
// If it wasn't this function always returns 0, otherwise it returns
21
// incrementing numbers, if the library was found this wastes 2 events but
22
// that should be okay.
23
__itt_event testEvent =
24
__itt_event_create("Test event", strlen("Test event"));
25
testEvent = __itt_event_create("Test event 2", strlen("Test event 2"));
26
27
if (testEvent) {
28
mInstance = new VTuneProfiler();
29
}
30
}
31
32
void VTuneProfiler::Shutdown() {}
33
34
void VTuneProfiler::TraceInternal(const char* aName, TracingKind aKind) {
35
std::string str(aName);
36
37
auto iter = mStrings.find(str);
38
39
__itt_event event;
40
if (iter != mStrings.end()) {
41
event = iter->second;
42
} else {
43
event = __itt_event_create(aName, str.length());
44
mStrings.insert({str, event});
45
}
46
47
if (aKind == TRACING_INTERVAL_START || aKind == TRACING_EVENT) {
48
// VTune will consider starts not matched with an end to be single point in
49
// time events.
50
__itt_event_start(event);
51
} else {
52
__itt_event_end(event);
53
}
54
}
55
56
void VTuneProfiler::RegisterThreadInternal(const char* aName) {
57
std::string str(aName);
58
59
if (!str.compare("GeckoMain")) {
60
// Process main thread.
61
switch (XRE_GetProcessType()) {
62
case GeckoProcessType::GeckoProcessType_Default:
63
__itt_thread_set_name("Main Process");
64
break;
65
case GeckoProcessType::GeckoProcessType_Content:
66
__itt_thread_set_name("Content Process");
67
break;
68
case GeckoProcessType::GeckoProcessType_GMPlugin:
69
__itt_thread_set_name("Plugin Process");
70
break;
71
case GeckoProcessType::GeckoProcessType_GPU:
72
__itt_thread_set_name("GPU Process");
73
break;
74
default:
75
__itt_thread_set_name("Unknown Process");
76
}
77
return;
78
}
79
__itt_thread_set_name(aName);
80
}