1 //===- llvm/PassRegistry.h - Pass Information Registry ----------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file defines PassRegistry, a class that is used in the initialization
11 // and registration of passes. At application startup, passes are registered
12 // with the PassRegistry, which is later provided to the PassManager for
13 // dependency resolution and similar tasks.
15 //===----------------------------------------------------------------------===//
17 #ifndef LLVM_PASSREGISTRY_H
18 #define LLVM_PASSREGISTRY_H
20 #include "llvm-c/Core.h"
21 #include "llvm/ADT/DenseMap.h"
22 #include "llvm/ADT/SmallPtrSet.h"
23 #include "llvm/ADT/StringMap.h"
24 #include "llvm/ADT/StringRef.h"
25 #include "llvm/PassInfo.h"
26 #include "llvm/Support/CBindingWrapping.h"
27 #include "llvm/Support/RWMutex.h"
34 struct PassRegistrationListener;
36 /// PassRegistry - This class manages the registration and intitialization of
37 /// the pass subsystem as application startup, and assists the PassManager
38 /// in resolving pass dependencies.
39 /// NOTE: PassRegistry is NOT thread-safe. If you want to use LLVM on multiple
40 /// threads simultaneously, you will need to use a separate PassRegistry on
43 mutable sys::SmartRWMutex<true> Lock;
45 /// Only if false, synchronization must use the Lock mutex.
46 std::atomic<bool> locked;
48 /// PassInfoMap - Keep track of the PassInfo object for each registered pass.
49 typedef DenseMap<const void *, const PassInfo *> MapType;
52 typedef StringMap<const PassInfo *> StringMapType;
53 StringMapType PassInfoStringMap;
55 std::vector<std::unique_ptr<const PassInfo>> ToFree;
56 std::vector<PassRegistrationListener *> Listeners;
59 PassRegistry() : locked(false) {}
62 /// getPassRegistry - Access the global registry object, which is
63 /// automatically initialized at application launch and destroyed by
65 static PassRegistry *getPassRegistry();
67 /// Enables fast thread synchronization in getPassInfo().
68 /// After calling lock() no more passes may be registered.
69 void lock() { locked = true; }
71 /// getPassInfo - Look up a pass' corresponding PassInfo, indexed by the pass'
72 /// type identifier (&MyPass::ID).
73 const PassInfo *getPassInfo(const void *TI) const;
75 /// getPassInfo - Look up a pass' corresponding PassInfo, indexed by the pass'
77 const PassInfo *getPassInfo(StringRef Arg) const;
79 /// registerPass - Register a pass (by means of its PassInfo) with the
80 /// registry. Required in order to use the pass with a PassManager.
81 void registerPass(const PassInfo &PI, bool ShouldFree = false);
83 /// registerAnalysisGroup - Register an analysis group (or a pass implementing
84 // an analysis group) with the registry. Like registerPass, this is required
85 // in order for a PassManager to be able to use this group/pass.
86 void registerAnalysisGroup(const void *InterfaceID, const void *PassID,
87 PassInfo &Registeree, bool isDefault,
88 bool ShouldFree = false);
90 /// enumerateWith - Enumerate the registered passes, calling the provided
91 /// PassRegistrationListener's passEnumerate() callback on each of them.
92 void enumerateWith(PassRegistrationListener *L);
94 /// addRegistrationListener - Register the given PassRegistrationListener
95 /// to receive passRegistered() callbacks whenever a new pass is registered.
96 void addRegistrationListener(PassRegistrationListener *L);
98 /// removeRegistrationListener - Unregister a PassRegistrationListener so that
99 /// it no longer receives passRegistered() callbacks.
100 void removeRegistrationListener(PassRegistrationListener *L);
103 // Create wrappers for C Binding types (see CBindingWrapping.h).
104 DEFINE_STDCXX_CONVERSION_FUNCTIONS(PassRegistry, LLVMPassRegistryRef)