[llvm-commits] [llvm] r108888 - in /llvm/trunk: include/llvm/PassRegistry.h lib/VMCore/PassRegistry.cpp

[Owen Anderson]

Author: resistor
Date: Tue Jul 20 14:23:55 2010
New Revision: 108888

URL: http://llvm.org/viewvc/llvm-project?rev=108888&view=rev
Log:
I just fail with SVN today.

Added:
    llvm/trunk/lib/VMCore/PassRegistry.cpp
Modified:
    llvm/trunk/include/llvm/PassRegistry.h

Modified: llvm/trunk/include/llvm/PassRegistry.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/PassRegistry.h?rev=108888&r1=108887&r2=108888&view=diff
==============================================================================
--- llvm/trunk/include/llvm/PassRegistry.h (original)
+++ llvm/trunk/include/llvm/PassRegistry.h Tue Jul 20 14:23:55 2010
@@ -46,8 +46,6 @@
   std::map<const PassInfo*, AnalysisGroupInfo> AnalysisGroupInfoMap;
 
 public:
-  static PassRegistry *getPassRegistry();
-  
   const PassInfo *getPassInfo(intptr_t TI) const;
   const PassInfo *getPassInfo(StringRef Arg) const;
   

Added: llvm/trunk/lib/VMCore/PassRegistry.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/VMCore/PassRegistry.cpp?rev=108888&view=auto
==============================================================================
--- llvm/trunk/lib/VMCore/PassRegistry.cpp (added)
+++ llvm/trunk/lib/VMCore/PassRegistry.cpp Tue Jul 20 14:23:55 2010
@@ -0,0 +1,72 @@
+//===- PassRegistry.cpp - Pass Registration Implementation ----------------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file implements the PassRegistry, with which passes are registered on
+// initialization, and supports the PassManager in dependency resolution.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/PassRegistry.h"
+#include "llvm/System/Mutex.h"
+
+const PassInfo *PassRegistry::getPassInfo(intptr_t TI) const {
+  sys::SmartScopedLock<true> Guard(Lock);
+  MapType::const_iterator I = PassInfoMap.find(TI);
+  return I != PassInfoMap.end() ? I->second : 0;
+}
+
+const PassInfo *PassRegistry::getPassInfo(StringRef Arg) const {
+  sys::SmartScopedLock<true> Guard(Lock);
+  StringMapType::const_iterator I = PassInfoStringMap.find(Arg);
+  return I != PassInfoStringMap.end() ? I->second : 0;
+}
+
+void PassRegistry::registerPass(const PassInfo &PI) {
+  sys::SmartScopedLock<true> Guard(Lock);
+  bool Inserted =
+    PassInfoMap.insert(std::make_pair(PI.getTypeInfo(),&PI)).second;
+  assert(Inserted && "Pass registered multiple times!"); Inserted=Inserted;
+  PassInfoStringMap[PI.getPassArgument()] = &PI;
+}
+
+void PassRegistry::unregisterPass(const PassInfo &PI) {
+  sys::SmartScopedLock<true> Guard(Lock);
+  MapType::iterator I = PassInfoMap.find(PI.getTypeInfo());
+  assert(I != PassInfoMap.end() && "Pass registered but not in map!");
+  
+  // Remove pass from the map.
+  PassInfoMap.erase(I);
+  PassInfoStringMap.erase(PI.getPassArgument());
+}
+
+void PassRegistry::enumerateWith(PassRegistrationListener *L) {
+  sys::SmartScopedLock<true> Guard(Lock);
+  for (MapType::const_iterator I = PassInfoMap.begin(),
+       E = PassInfoMap.end(); I != E; ++I)
+    L->passEnumerate(I->second);
+}
+
+
+/// Analysis Group Mechanisms.
+void PassRegistry::registerAnalysisGroup(PassInfo *InterfaceInfo,
+                                         const PassInfo *ImplementationInfo,
+                                         bool isDefault) {
+  sys::SmartScopedLock<true> Guard(Lock);
+  AnalysisGroupInfo &AGI = AnalysisGroupInfoMap[InterfaceInfo];
+  assert(AGI.Implementations.count(ImplementationInfo) == 0 &&
+         "Cannot add a pass to the same analysis group more than once!");
+  AGI.Implementations.insert(ImplementationInfo);
+  if (isDefault) {
+    assert(InterfaceInfo->getNormalCtor() == 0 &&
+           "Default implementation for analysis group already specified!");
+    assert(ImplementationInfo->getNormalCtor() &&
+         "Cannot specify pass as default if it does not have a default ctor");
+    InterfaceInfo->setNormalCtor(ImplementationInfo->getNormalCtor());
+  }
+}