From d5fe16c2db0e877f45415c5c7b4eb92897a5f9d1 Mon Sep 17 00:00:00 2001
From: Kevin Enderby <enderby@apple.com>
Date: Mon, 17 May 2010 23:08:19 +0000
Subject: Added support in MC for Directional Local Labels.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@103989 91177308-0d34-0410-b5e6-96231b3b80d8
---
 lib/MC/MCContext.cpp | 45 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

(limited to 'lib/MC/MCContext.cpp')

diff --git a/lib/MC/MCContext.cpp b/lib/MC/MCContext.cpp
index 5a65b8a..5cc72e8 100644
--- a/lib/MC/MCContext.cpp
+++ b/lib/MC/MCContext.cpp
@@ -13,6 +13,7 @@
 #include "llvm/MC/MCSectionELF.h"
 #include "llvm/MC/MCSectionCOFF.h"
 #include "llvm/MC/MCSymbol.h"
+#include "llvm/MC/MCLabel.h"
 #include "llvm/ADT/SmallString.h"
 #include "llvm/ADT/Twine.h"
 using namespace llvm;
@@ -71,6 +72,50 @@ MCSymbol *MCContext::CreateTempSymbol() {
                            "tmp" + Twine(NextUniqueID++));
 }
 
+unsigned MCContext::NextInstance(int64_t LocalLabelVal) {
+  unsigned Instance;
+  MCLabel *Label;
+  Label = Instances[LocalLabelVal];
+  if (Label) {
+    Instance = Label->incInstance();
+  }
+  else {
+    Instance = 1;
+    Label = new MCLabel(Instance);
+    Instances[LocalLabelVal] = Label;
+  }
+  return Instance;
+}
+
+unsigned MCContext::GetInstance(int64_t LocalLabelVal) {
+  int Instance;
+  MCLabel *Label;
+  Label = Instances[LocalLabelVal];
+  if (Label) {
+    Instance = Label->getInstance();
+  }
+  else {
+    Instance = 0;
+    Label = new MCLabel(Instance);
+    Instances[LocalLabelVal] = Label;
+  }
+  return Instance;
+}
+
+MCSymbol *MCContext::CreateDirectionalLocalSymbol(int64_t LocalLabelVal) {
+  return GetOrCreateSymbol(Twine(MAI.getPrivateGlobalPrefix()) +
+                           Twine(LocalLabelVal) +
+                           "\2" +
+			   Twine(NextInstance(LocalLabelVal)));
+}
+MCSymbol *MCContext::GetDirectionalLocalSymbol(int64_t LocalLabelVal,
+                                               int bORf) {
+  return GetOrCreateSymbol(Twine(MAI.getPrivateGlobalPrefix()) +
+                           Twine(LocalLabelVal) +
+                           "\2" +
+			   Twine(GetInstance(LocalLabelVal) + bORf));
+}
+
 MCSymbol *MCContext::LookupSymbol(StringRef Name) const {
   return Symbols.lookup(Name);
 }
-- 
cgit v1.1