1//===- PassRegistry.cpp - Pass Registration Implementation ----------------===//
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//
9// This file implements the PassRegistry, with which passes are registered on
10// initialization, and supports the PassManager in dependency resolution.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/PassRegistry.h"
15#include "llvm/ADT/STLExtras.h"
16#include "llvm/Pass.h"
17#include "llvm/PassInfo.h"
18#include <cassert>
19#include <memory>
20#include <utility>
21
22using namespace llvm;
23
24PassRegistry *PassRegistry::getPassRegistry() {
25 static PassRegistry PassRegistryObj;
26 return &PassRegistryObj;
27}
28
29//===----------------------------------------------------------------------===//
30// Accessors
31//
32
33PassRegistry::~PassRegistry() = default;
34
35const PassInfo *PassRegistry::getPassInfo(const void *TI) const {
36 sys::SmartScopedReader<true> Guard(Lock);
37 return PassInfoMap.lookup(Val: TI);
38}
39
40const PassInfo *PassRegistry::getPassInfo(StringRef Arg) const {
41 sys::SmartScopedReader<true> Guard(Lock);
42 return PassInfoStringMap.lookup(Key: Arg);
43}
44
45//===----------------------------------------------------------------------===//
46// Pass Registration mechanism
47//
48
49void PassRegistry::registerPass(const PassInfo &PI, bool ShouldFree) {
50 sys::SmartScopedWriter<true> Guard(Lock);
51 bool Inserted =
52 PassInfoMap.insert(KV: std::make_pair(x: PI.getTypeInfo(), y: &PI)).second;
53 assert(Inserted && "Pass registered multiple times!");
54 (void)Inserted;
55 PassInfoStringMap[PI.getPassArgument()] = &PI;
56
57 // Notify any listeners.
58 for (auto *Listener : Listeners)
59 Listener->passRegistered(&PI);
60
61 if (ShouldFree)
62 ToFree.push_back(x: std::unique_ptr<const PassInfo>(&PI));
63}
64
65void PassRegistry::enumerateWith(PassRegistrationListener *L) {
66 sys::SmartScopedReader<true> Guard(Lock);
67 for (auto PassInfoPair : PassInfoMap)
68 L->passEnumerate(PassInfoPair.second);
69}
70
71void PassRegistry::addRegistrationListener(PassRegistrationListener *L) {
72 sys::SmartScopedWriter<true> Guard(Lock);
73 Listeners.push_back(x: L);
74}
75
76void PassRegistry::removeRegistrationListener(PassRegistrationListener *L) {
77 sys::SmartScopedWriter<true> Guard(Lock);
78
79 auto I = llvm::find(Range&: Listeners, Val: L);
80 Listeners.erase(position: I);
81}
82