Added support in MC for Directional Local Labels.

llvm-svn: 103989

1 files changed, 45 insertions, 0 deletions

diff --git a/llvm/lib/MC/MCContext.cpp b/llvm/lib/MC/MCContext.cpp
index 5a65b8ae380..5cc72e8e9f5 100644
--- a/llvm/lib/MC/MCContext.cpp
+++ b/llvm/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);
 }