llvm-mc: Factor getSectionData out of SwitchSection.

llvm-svn: 80344

1 files changed, 11 insertions, 7 deletions

diff --git a/llvm/lib/MC/MCMachOStreamer.cpp b/llvm/lib/MC/MCMachOStreamer.cpp
index e89f7d8dbd3..9f813fdc49e 100644
--- a/llvm/lib/MC/MCMachOStreamer.cpp
+++ b/llvm/lib/MC/MCMachOStreamer.cpp
@@ -63,6 +63,15 @@ private:
     return 0;
   }
 
+  MCSectionData &getSectionData(const MCSection &Section) {
+    MCSectionData *&Entry = SectionMap[&Section];
+
+    if (!Entry)
+      Entry = new MCSectionData(Section, &Assembler);
+
+    return *Entry;
+  }
+
   MCSymbolData &getSymbolData(MCSymbol &Symbol) {
     MCSymbolData *&Entry = SymbolMap[&Symbol];
 
@@ -135,14 +144,9 @@ void MCMachOStreamer::SwitchSection(const MCSection *Section) {
   
   // If already in this section, then this is a noop.
   if (Section == CurSection) return;
-  
-  CurSection = Section;
-  MCSectionData *&Entry = SectionMap[Section];
 
-  if (!Entry)
-    Entry = new MCSectionData(*Section, &Assembler);
-
-  CurSectionData = Entry;
+  CurSection = Section;
+  CurSectionData = &getSectionData(*Section);
 }
 
 void MCMachOStreamer::EmitLabel(MCSymbol *Symbol) {