summaryrefslogtreecommitdiffstats
path: root/tools/profiler/core/VTuneProfiler.cpp
blob: 58a39c51eeff8ad0659c0d6e3f055877c9384ff6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* 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/. */

#ifdef XP_WIN
#  undef UNICODE
#  undef _UNICODE
#endif

#include "VTuneProfiler.h"
#include "mozilla/Bootstrap.h"
#include <memory>

VTuneProfiler* VTuneProfiler::mInstance = nullptr;

void VTuneProfiler::Initialize() {
  // This is just a 'dirty trick' to find out if the ittnotify DLL was found.
  // If it wasn't this function always returns 0, otherwise it returns
  // incrementing numbers, if the library was found this wastes 2 events but
  // that should be okay.
  __itt_event testEvent =
      __itt_event_create("Test event", strlen("Test event"));
  testEvent = __itt_event_create("Test event 2", strlen("Test event 2"));

  if (testEvent) {
    mInstance = new VTuneProfiler();
  }
}

void VTuneProfiler::Shutdown() {}

void VTuneProfiler::TraceInternal(const char* aName, TracingKind aKind) {
  std::string str(aName);

  auto iter = mStrings.find(str);

  __itt_event event;
  if (iter != mStrings.end()) {
    event = iter->second;
  } else {
    event = __itt_event_create(aName, str.length());
    mStrings.insert({str, event});
  }

  if (aKind == TRACING_INTERVAL_START || aKind == TRACING_EVENT) {
    // VTune will consider starts not matched with an end to be single point in
    // time events.
    __itt_event_start(event);
  } else {
    __itt_event_end(event);
  }
}

void VTuneProfiler::RegisterThreadInternal(const char* aName) {
  std::string str(aName);

  if (!str.compare("GeckoMain")) {
    // Process main thread.
    switch (XRE_GetProcessType()) {
      case GeckoProcessType::GeckoProcessType_Default:
        __itt_thread_set_name("Main Process");
        break;
      case GeckoProcessType::GeckoProcessType_Content:
        __itt_thread_set_name("Content Process");
        break;
      case GeckoProcessType::GeckoProcessType_GMPlugin:
        __itt_thread_set_name("Plugin Process");
        break;
      case GeckoProcessType::GeckoProcessType_GPU:
        __itt_thread_set_name("GPU Process");
        break;
      default:
        __itt_thread_set_name("Unknown Process");
    }
    return;
  }
  __itt_thread_set_name(aName);
}