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/IR/LLVMContext.h"
16 #include "LLVMContextImpl.h"
17 #include "llvm/IR/Constants.h"
18 #include "llvm/IR/Instruction.h"
19 #include "llvm/IR/Metadata.h"
20 #include "llvm/Support/ManagedStatic.h"
21 #include "llvm/Support/SourceMgr.h"
25 static ManagedStatic<LLVMContext> GlobalContext;
27 LLVMContext& llvm::getGlobalContext() {
28 return *GlobalContext;
31 LLVMContext::LLVMContext() : pImpl(new LLVMContextImpl(*this)) {
32 // Create the fixed metadata kinds. This is done in the same order as the
33 // MD_* enum values so that they correspond.
35 // Create the 'dbg' metadata kind.
36 unsigned DbgID = getMDKindID("dbg");
37 assert(DbgID == MD_dbg && "dbg kind id drifted"); (void)DbgID;
39 // Create the 'tbaa' metadata kind.
40 unsigned TBAAID = getMDKindID("tbaa");
41 assert(TBAAID == MD_tbaa && "tbaa kind id drifted"); (void)TBAAID;
43 // Create the 'prof' metadata kind.
44 unsigned ProfID = getMDKindID("prof");
45 assert(ProfID == MD_prof && "prof kind id drifted"); (void)ProfID;
47 // Create the 'fpmath' metadata kind.
48 unsigned FPAccuracyID = getMDKindID("fpmath");
49 assert(FPAccuracyID == MD_fpmath && "fpmath kind id drifted");
52 // Create the 'range' metadata kind.
53 unsigned RangeID = getMDKindID("range");
54 assert(RangeID == MD_range && "range kind id drifted");
57 // Create the 'tbaa.struct' metadata kind.
58 unsigned TBAAStructID = getMDKindID("tbaa.struct");
59 assert(TBAAStructID == MD_tbaa_struct && "tbaa.struct kind id drifted");
62 LLVMContext::~LLVMContext() { delete pImpl; }
64 void LLVMContext::addModule(Module *M) {
65 pImpl->OwnedModules.insert(M);
68 void LLVMContext::removeModule(Module *M) {
69 pImpl->OwnedModules.erase(M);
72 //===----------------------------------------------------------------------===//
73 // Recoverable Backend Errors
74 //===----------------------------------------------------------------------===//
76 void LLVMContext::setDiagnosticHandler(DiagHandlerTy DiagHandler,
78 pImpl->DiagHandler = DiagHandler;
79 pImpl->DiagContext = DiagContext;
82 /// getDiagnosticHandler - Return the diagnostic handler set by
83 /// setDiagnosticHandler.
84 LLVMContext::DiagHandlerTy LLVMContext::getDiagnosticHandler() const {
85 return pImpl->DiagHandler;
88 /// getDiagnosticContext - Return the diagnostic context set by
89 /// setDiagnosticHandler.
90 void *LLVMContext::getDiagnosticContext() const {
91 return pImpl->DiagContext;
94 void LLVMContext::emitError(const Twine &ErrorStr) {
95 emitError(0U, ErrorStr);
98 void LLVMContext::emitError(const Instruction *I, const Twine &ErrorStr) {
99 unsigned LocCookie = 0;
100 if (const MDNode *SrcLoc = I->getMetadata("srcloc")) {
101 if (SrcLoc->getNumOperands() != 0)
102 if (const ConstantInt *CI = dyn_cast<ConstantInt>(SrcLoc->getOperand(0)))
103 LocCookie = CI->getZExtValue();
105 return emitError(LocCookie, ErrorStr);
108 void LLVMContext::emitError(unsigned LocCookie, const Twine &ErrorStr) {
109 // If there is no error handler installed, just print the error and exit.
110 if (pImpl->DiagHandler == 0) {
111 errs() << "error: " << ErrorStr << "\n";
115 // If we do have an error handler, we can report the error and keep going.
116 SMDiagnostic Diag("", SourceMgr::DK_Error, ErrorStr.str());
118 pImpl->DiagHandler(Diag, pImpl->DiagContext, LocCookie);
121 //===----------------------------------------------------------------------===//
122 // Metadata Kind Uniquing
123 //===----------------------------------------------------------------------===//
126 /// isValidName - Return true if Name is a valid custom metadata handler name.
127 static bool isValidName(StringRef MDName) {
131 if (!std::isalpha(MDName[0]))
134 for (StringRef::iterator I = MDName.begin() + 1, E = MDName.end(); I != E;
136 if (!std::isalnum(*I) && *I != '_' && *I != '-' && *I != '.')
143 /// getMDKindID - Return a unique non-zero ID for the specified metadata kind.
144 unsigned LLVMContext::getMDKindID(StringRef Name) const {
145 assert(isValidName(Name) && "Invalid MDNode name");
147 // If this is new, assign it its ID.
149 pImpl->CustomMDKindNames.GetOrCreateValue(
150 Name, pImpl->CustomMDKindNames.size()).second;
153 /// getHandlerNames - Populate client supplied smallvector using custome
154 /// metadata name and ID.
155 void LLVMContext::getMDKindNames(SmallVectorImpl<StringRef> &Names) const {
156 Names.resize(pImpl->CustomMDKindNames.size());
157 for (StringMap<unsigned>::const_iterator I = pImpl->CustomMDKindNames.begin(),
158 E = pImpl->CustomMDKindNames.end(); I != E; ++I)
159 Names[I->second] = I->first();