skia2/tools/trace/SkEventTracingPriv.cpp
Brian Osman 1e09686840 Add --traceMatch argument to filter tracing categories
Bug: skia:
Change-Id: I1449f0f4d7d9ab6225d98c601eafa7461a2a7dde
Reviewed-on: https://skia-review.googlesource.com/27120
Commit-Queue: Brian Osman <brianosman@google.com>
Commit-Queue: Mike Klein <mtklein@google.com>
Reviewed-by: Mike Klein <mtklein@google.com>
2017-07-26 20:12:22 +00:00

85 lines
3.0 KiB
C++

/*
* Copyright 2017 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "SkEventTracingPriv.h"
#include "SkATrace.h"
#include "SkChromeTracingTracer.h"
#include "SkCommandLineFlags.h"
#include "SkDebugfTracer.h"
#include "SkEventTracer.h"
#include "SkTraceEvent.h"
DEFINE_string(trace, "",
"Log trace events in one of several modes:\n"
" debugf : Show events using SkDebugf\n"
" atrace : Send events to Android ATrace\n"
" <filename> : Any other string is interpreted as a filename. Writes\n"
" trace events to specified file as JSON, for viewing\n"
" with chrome://tracing");
DEFINE_string(traceMatch, "",
"Filter which categories are traced.\n"
"Uses same format as --match\n");
void initializeEventTracingForTools(const char* traceFlag) {
if (!traceFlag) {
if (FLAGS_trace.isEmpty()) {
return;
}
traceFlag = FLAGS_trace[0];
}
SkEventTracer* eventTracer = nullptr;
if (0 == strcmp(traceFlag, "atrace")) {
eventTracer = new SkATrace();
} else if (0 == strcmp(traceFlag, "debugf")) {
eventTracer = new SkDebugfTracer();
} else {
eventTracer = new SkChromeTracingTracer(traceFlag);
}
SkAssertResult(SkEventTracer::SetInstance(eventTracer));
}
uint8_t* SkEventTracingCategories::getCategoryGroupEnabled(const char* name) {
static_assert(0 == offsetof(CategoryState, fEnabled), "CategoryState");
// We ignore the "disabled-by-default-" prefix in our internal tools
if (SkStrStartsWith(name, TRACE_CATEGORY_PREFIX)) {
name += strlen(TRACE_CATEGORY_PREFIX);
}
// Chrome's implementation of this API does a two-phase lookup (once without a lock, then again
// with a lock. But the tracing macros avoid calling these functions more than once per site,
// so just do something simple (and easier to reason about):
SkAutoMutexAcquire lock(&fMutex);
for (int i = 0; i < fNumCategories; ++i) {
if (0 == strcmp(name, fCategories[i].fName)) {
return reinterpret_cast<uint8_t*>(&fCategories[i]);
}
}
if (fNumCategories >= kMaxCategories) {
SkDEBUGFAIL("Exhausted event tracing categories. Increase kMaxCategories.");
return reinterpret_cast<uint8_t*>(&fCategories[0]);
}
fCategories[fNumCategories].fEnabled = SkCommandLineFlags::ShouldSkip(FLAGS_traceMatch, name)
? 0 : SkEventTracer::kEnabledForRecording_CategoryGroupEnabledFlags;
fCategories[fNumCategories].fName = name;
return reinterpret_cast<uint8_t*>(&fCategories[fNumCategories++]);
}
const char* SkEventTracingCategories::getCategoryGroupName(const uint8_t* categoryEnabledFlag) {
if (categoryEnabledFlag) {
return reinterpret_cast<const CategoryState*>(categoryEnabledFlag)->fName;
}
return nullptr;
}