1 //===-- Annotation.cpp - Implement the Annotation Classes --------*- C++ -*--=//
3 // This file implements the AnnotationManager class.
5 //===----------------------------------------------------------------------===//
8 #include "llvm/Annotation.h"
14 typedef map<const string, unsigned> IDMapType;
15 static unsigned IDCounter = 0; // Unique ID counter
17 // Static member to ensure initialiation on demand.
18 static IDMapType &getIDMap() { static IDMapType TheMap; return TheMap; }
20 // On demand annotation creation support...
21 typedef Annotation *(*AnnFactory)(AnnotationID, const Annotable *, void *);
22 typedef map<unsigned, pair<AnnFactory,void*> > FactMapType;
23 static FactMapType &getFactMap() { static FactMapType FactMap; return FactMap; }
26 AnnotationID AnnotationManager::getID(const string &Name) { // Name -> ID
27 IDMapType::iterator I = getIDMap().find(Name);
28 if (I == getIDMap().end()) {
29 getIDMap()[Name] = IDCounter++; // Add a new element
35 // getID - Name -> ID + registration of a factory function for demand driven
36 // annotation support.
37 AnnotationID AnnotationManager::getID(const string &Name, Factory Fact,
39 AnnotationID Result(getID(Name));
40 registerAnnotationFactory(Result, Fact, Data);
45 // getName - This function is especially slow, but that's okay because it should
46 // only be used for debugging.
48 const string &AnnotationManager::getName(AnnotationID ID) { // ID -> Name
49 IDMapType &TheMap = getIDMap();
50 for (IDMapType::iterator I = TheMap.begin(); ; ++I) {
51 assert(I != TheMap.end() && "Annotation ID is unknown!");
52 if (I->second == ID.ID) return I->first;
57 // registerAnnotationFactory - This method is used to register a callback
58 // function used to create an annotation on demand if it is needed by the
59 // Annotable::findOrCreateAnnotation method.
61 void AnnotationManager::registerAnnotationFactory(AnnotationID ID,
65 getFactMap()[ID.ID] = make_pair(F, ExtraData);
67 getFactMap().erase(ID.ID);
70 // createAnnotation - Create an annotation of the specified ID for the
71 // specified object, using a register annotation creation function.
73 Annotation *AnnotationManager::createAnnotation(AnnotationID ID,
74 const Annotable *Obj) {
75 FactMapType::iterator I = getFactMap().find(ID.ID);
76 if (I == getFactMap().end()) return 0;
77 return I->second.first(ID, Obj, I->second.second);