1 //===- lib/MC/MCContext.cpp - Machine Code Context ------------------------===//
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 #include "llvm/MC/MCContext.h"
11 #include "llvm/MC/MCAsmInfo.h"
12 #include "llvm/MC/MCSection.h"
13 #include "llvm/MC/MCSymbol.h"
14 #include "llvm/ADT/SmallString.h"
15 #include "llvm/ADT/Twine.h"
18 MCContext::MCContext(const MCAsmInfo &mai) : MAI(mai), NextUniqueID(0) {
21 MCContext::~MCContext() {
22 // NOTE: The sections are all allocated out of a bump pointer allocator,
23 // we don't need to free them here.
26 MCSymbol *MCContext::GetOrCreateSymbol(StringRef Name) {
27 assert(!Name.empty() && "Normal symbols cannot be unnamed!");
28 MCSymbol *&Entry = Symbols[Name];
29 if (Entry) return Entry;
31 return Entry = new (*this) MCSymbol(Name, false);
34 MCSymbol *MCContext::GetOrCreateSymbol(const Twine &Name) {
35 SmallString<128> NameSV;
36 Name.toVector(NameSV);
37 return GetOrCreateSymbol(NameSV.str());
40 MCSymbol *MCContext::CreateTempSymbol() {
41 return GetOrCreateTemporarySymbol(Twine(MAI.getPrivateGlobalPrefix()) +
42 "tmp" + Twine(NextUniqueID++));
46 MCSymbol *MCContext::GetOrCreateTemporarySymbol(StringRef Name) {
47 // If there is no name, create a new anonymous symbol.
48 // FIXME: Remove this. This form of the method should always take a name.
50 return GetOrCreateTemporarySymbol(Twine(MAI.getPrivateGlobalPrefix()) +
51 "tmp" + Twine(NextUniqueID++));
53 // Otherwise create as usual.
54 MCSymbol *&Entry = Symbols[Name];
55 if (Entry) return Entry;
56 return Entry = new (*this) MCSymbol(Name, true);
59 MCSymbol *MCContext::GetOrCreateTemporarySymbol(const Twine &Name) {
60 SmallString<128> NameSV;
61 Name.toVector(NameSV);
62 return GetOrCreateTemporarySymbol(NameSV.str());
66 MCSymbol *MCContext::LookupSymbol(StringRef Name) const {
67 return Symbols.lookup(Name);