1 //===-- LLVMContext.cpp - Implement LLVMContext -----------------------===//
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 implements LLVMContext, as a wrapper around the opaque
11 // class LLVMContextImpl.
13 //===----------------------------------------------------------------------===//
15 #include "llvm/LLVMContext.h"
16 #include "llvm/Metadata.h"
17 #include "llvm/Constants.h"
18 #include "llvm/Instruction.h"
19 #include "llvm/Support/ManagedStatic.h"
20 #include "llvm/Support/SourceMgr.h"
21 #include "LLVMContextImpl.h"
24 static ManagedStatic<LLVMContext> GlobalContext;
26 LLVMContext& llvm::getGlobalContext() {
27 return *GlobalContext;
30 LLVMContext::LLVMContext() : pImpl(new LLVMContextImpl(*this)) {
31 // Create the fixed metadata kinds. This is done in the same order as the
32 // MD_* enum values so that they correspond.
34 // Create the 'dbg' metadata kind.
35 unsigned DbgID = getMDKindID("dbg");
36 assert(DbgID == MD_dbg && "dbg kind id drifted"); (void)DbgID;
38 // Create the 'tbaa' metadata kind.
39 unsigned TBAAID = getMDKindID("tbaa");
40 assert(TBAAID == MD_tbaa && "tbaa kind id drifted"); (void)TBAAID;
42 LLVMContext::~LLVMContext() { delete pImpl; }
44 void LLVMContext::addModule(Module *M) {
45 pImpl->OwnedModules.insert(M);
48 void LLVMContext::removeModule(Module *M) {
49 pImpl->OwnedModules.erase(M);
52 //===----------------------------------------------------------------------===//
53 // Recoverable Backend Errors
54 //===----------------------------------------------------------------------===//
57 setInlineAsmDiagnosticHandler(InlineAsmDiagHandlerTy DiagHandler,
59 pImpl->InlineAsmDiagHandler = DiagHandler;
60 pImpl->InlineAsmDiagContext = DiagContext;
63 /// getInlineAsmDiagnosticHandler - Return the diagnostic handler set by
64 /// setInlineAsmDiagnosticHandler.
65 LLVMContext::InlineAsmDiagHandlerTy
66 LLVMContext::getInlineAsmDiagnosticHandler() const {
67 return pImpl->InlineAsmDiagHandler;
70 /// getInlineAsmDiagnosticContext - Return the diagnostic context set by
71 /// setInlineAsmDiagnosticHandler.
72 void *LLVMContext::getInlineAsmDiagnosticContext() const {
73 return pImpl->InlineAsmDiagContext;
76 void LLVMContext::emitError(StringRef ErrorStr) {
77 emitError(0U, ErrorStr);
80 void LLVMContext::emitError(const Instruction *I, StringRef ErrorStr) {
81 unsigned LocCookie = 0;
82 if (const MDNode *SrcLoc = I->getMetadata("srcloc")) {
83 if (SrcLoc->getNumOperands() != 0)
84 if (const ConstantInt *CI = dyn_cast<ConstantInt>(SrcLoc->getOperand(0)))
85 LocCookie = CI->getZExtValue();
87 return emitError(LocCookie, ErrorStr);
90 void LLVMContext::emitError(unsigned LocCookie, StringRef ErrorStr) {
91 // If there is no error handler installed, just print the error and exit.
92 if (pImpl->InlineAsmDiagHandler == 0) {
93 errs() << "error: " << ErrorStr << "\n";
97 // If we do have an error handler, we can report the error and keep going.
98 SMDiagnostic Diag("", "error: " + ErrorStr.str());
100 pImpl->InlineAsmDiagHandler(Diag, pImpl->InlineAsmDiagContext, LocCookie);
104 //===----------------------------------------------------------------------===//
105 // Metadata Kind Uniquing
106 //===----------------------------------------------------------------------===//
109 /// isValidName - Return true if Name is a valid custom metadata handler name.
110 static bool isValidName(StringRef MDName) {
114 if (!isalpha(MDName[0]))
117 for (StringRef::iterator I = MDName.begin() + 1, E = MDName.end(); I != E;
119 if (!isalnum(*I) && *I != '_' && *I != '-' && *I != '.')
126 /// getMDKindID - Return a unique non-zero ID for the specified metadata kind.
127 unsigned LLVMContext::getMDKindID(StringRef Name) const {
128 assert(isValidName(Name) && "Invalid MDNode name");
130 // If this is new, assign it its ID.
132 pImpl->CustomMDKindNames.GetOrCreateValue(
133 Name, pImpl->CustomMDKindNames.size()).second;
136 /// getHandlerNames - Populate client supplied smallvector using custome
137 /// metadata name and ID.
138 void LLVMContext::getMDKindNames(SmallVectorImpl<StringRef> &Names) const {
139 Names.resize(pImpl->CustomMDKindNames.size());
140 for (StringMap<unsigned>::const_iterator I = pImpl->CustomMDKindNames.begin(),
141 E = pImpl->CustomMDKindNames.end(); I != E; ++I)
142 Names[I->second] = I->first();