-//===-- Annotation.cpp - Implement the Annotation Classes --------*- C++ -*--=//
+//===-- Annotation.cpp - Implement the Annotation Classes -----------------===//
+//
+// 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 AnnotationManager class.
//
//===----------------------------------------------------------------------===//
+#include "llvm/Support/Annotation.h"
+#include "llvm/Support/ManagedStatic.h"
#include <map>
-#include "llvm/Annotation.h"
+#include <cstring>
+using namespace llvm;
+
+Annotation::~Annotation() {} // Designed to be subclassed
-typedef map<const string, unsigned> IDMapType;
+Annotable::~Annotable() { // Virtual because it's designed to be subclassed...
+ Annotation *A = AnnotationList;
+ while (A) {
+ Annotation *Next = A->getNext();
+ delete A;
+ A = Next;
+ }
+}
+
+namespace {
+ class StrCmp {
+ public:
+ bool operator()(const char *a, const char *b) const {
+ return strcmp(a, b) < 0;
+ }
+ };
+}
+
+typedef std::map<const char*, unsigned, StrCmp> IDMapType;
static unsigned IDCounter = 0; // Unique ID counter
// Static member to ensure initialiation on demand.
-static IDMapType &getIDMap() { static IDMapType TheMap; return TheMap; }
+static ManagedStatic<IDMapType> IDMap;
// On demand annotation creation support...
-typedef Annotation *(*AnnFactory)(AnnotationID, Annotable *, void *);
-typedef map<unsigned, pair<AnnFactory,void*> > FactMapType;
-static FactMapType &getFactMap() { static FactMapType FactMap; return FactMap; }
+typedef Annotation *(*AnnFactory)(AnnotationID, const Annotable *, void *);
+typedef std::map<unsigned, std::pair<AnnFactory,void*> > FactMapType;
+
+static FactMapType *TheFactMap = 0;
+static FactMapType &getFactMap() {
+ if (TheFactMap == 0)
+ TheFactMap = new FactMapType();
+ return *TheFactMap;
+}
+static void eraseFromFactMap(unsigned ID) {
+ assert(TheFactMap && "No entries found!");
+ TheFactMap->erase(ID);
+ if (TheFactMap->empty()) { // Delete when empty
+ delete TheFactMap;
+ TheFactMap = 0;
+ }
+}
-AnnotationID AnnotationManager::getID(const string &Name) { // Name -> ID
- IDMapType::iterator I = getIDMap().find(Name);
- if (I == getIDMap().end()) {
- getIDMap()[Name] = IDCounter++; // Add a new element
- return IDCounter-1;
+AnnotationID AnnotationManager::getID(const char *Name) { // Name -> ID
+ IDMapType::iterator I = IDMap->find(Name);
+ if (I == IDMap->end()) {
+ (*IDMap)[Name] = IDCounter++; // Add a new element
+ return AnnotationID(IDCounter-1);
}
- return I->second;
+ return AnnotationID(I->second);
+}
+
+// getID - Name -> ID + registration of a factory function for demand driven
+// annotation support.
+AnnotationID AnnotationManager::getID(const char *Name, Factory Fact,
+ void *Data) {
+ AnnotationID Result(getID(Name));
+ registerAnnotationFactory(Result, Fact, Data);
+ return Result;
}
// getName - This function is especially slow, but that's okay because it should
// only be used for debugging.
//
-const string &AnnotationManager::getName(AnnotationID ID) { // ID -> Name
- IDMapType &TheMap = getIDMap();
+const char *AnnotationManager::getName(AnnotationID ID) { // ID -> Name
+ IDMapType &TheMap = *IDMap;
for (IDMapType::iterator I = TheMap.begin(); ; ++I) {
assert(I != TheMap.end() && "Annotation ID is unknown!");
if (I->second == ID.ID) return I->first;
}
}
-
// registerAnnotationFactory - This method is used to register a callback
-// function used to create an annotation on demand if it is needed by the
+// function used to create an annotation on demand if it is needed by the
// Annotable::findOrCreateAnnotation method.
//
-void AnnotationManager::registerAnnotationFactory(AnnotationID ID,
- AnnFactory F,
- void *ExtraData) {
+void AnnotationManager::registerAnnotationFactory(AnnotationID ID, AnnFactory F,
+ void *ExtraData) {
if (F)
- getFactMap()[ID.ID] = make_pair(F, ExtraData);
+ getFactMap()[ID.ID] = std::make_pair(F, ExtraData);
else
- getFactMap().erase(ID.ID);
+ eraseFromFactMap(ID.ID);
}
// createAnnotation - Create an annotation of the specified ID for the
// specified object, using a register annotation creation function.
//
-Annotation *AnnotationManager::createAnnotation(AnnotationID ID,
- Annotable *Obj) {
+Annotation *AnnotationManager::createAnnotation(AnnotationID ID,
+ const Annotable *Obj) {
FactMapType::iterator I = getFactMap().find(ID.ID);
if (I == getFactMap().end()) return 0;
return I->second.first(ID, Obj, I->second.second);