1 //===- PassInstrumentation.cpp - Pass Instrumentation interface -*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 /// \file
9 ///
10 /// This file provides the implementation of PassInstrumentation class.
11 ///
12 //===----------------------------------------------------------------------===//
13 
14 #include "llvm/IR/PassInstrumentation.h"
15 #include "llvm/ADT/STLExtras.h"
16 #include "llvm/IR/PassManager.h"
17 
18 namespace llvm {
19 
20 void PassInstrumentationCallbacks::addClassToPassName(StringRef ClassName,
21                                                       StringRef PassName) {
22   if (ClassToPassName[ClassName].empty())
23     ClassToPassName[ClassName] = PassName.str();
24 }
25 
26 StringRef
27 PassInstrumentationCallbacks::getPassNameForClassName(StringRef ClassName) {
28   return ClassToPassName[ClassName];
29 }
30 
31 AnalysisKey PassInstrumentationAnalysis::Key;
32 
33 bool isSpecialPass(StringRef PassID, const std::vector<StringRef> &Specials) {
34   size_t Pos = PassID.find('<');
35   StringRef Prefix = PassID;
36   if (Pos != StringRef::npos)
37     Prefix = PassID.substr(0, Pos);
38   return any_of(Specials, [Prefix](StringRef S) { return Prefix.endswith(S); });
39 }
40 
41 } // namespace llvm
42