simplify getSectionForMergableConstant to take a SectionKind.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@77134 91177308-0d34-0410-b5e6-96231b3b80d8

include/llvm/Target/DarwinTargetAsmInfo.h

@@ -40,8 +40,7 @@ namespace llvm {
                                       Mangler *Mang) const;
 
     
-    virtual const Section *
-    getSectionForMergableConstant(uint64_t Size, unsigned ReloInfo) const;
+    virtual const Section *getSectionForMergableConstant(SectionKind Kind)const;
     
   private:
     const Section* MergeableStringSection(const GlobalVariable *GV) const;

include/llvm/Target/ELFTargetAsmInfo.h

@@ -29,7 +29,7 @@ namespace llvm {
     /// specified size and relocation information, return a section that it
     /// should be placed in.
     virtual const Section *
-    getSectionForMergableConstant(uint64_t Size, unsigned ReloInfo) const;
+    getSectionForMergableConstant(SectionKind Kind) const;
     
     /// getFlagsForNamedSection - If this target wants to be able to infer
     /// section flags based on the name of the section specified for a global

include/llvm/Target/TargetAsmInfo.h

@@ -710,8 +710,7 @@ namespace llvm {
     /// getSectionForMergableConstant - Given a mergable constant with the
     /// specified size and relocation information, return a section that it
     /// should be placed in.
-    virtual const Section *
-    getSectionForMergableConstant(uint64_t Size, unsigned ReloInfo) const;
+    virtual const Section *getSectionForMergableConstant(SectionKind Kind)const;
 
     
     /// getSectionPrefixForUniqueGlobal - Return a string that we should prepend

lib/CodeGen/AsmPrinter/AsmPrinter.cpp

@@ -301,17 +301,28 @@ void AsmPrinter::EmitConstantPool(MachineConstantPool *MCP) {
   const std::vector<MachineConstantPoolEntry> &CP = MCP->getConstants();
   if (CP.empty()) return;
 
-  const TargetData &TD = *TM.getTargetData();
-  
   // Calculate sections for constant pool entries. We collect entries to go into
   // the same section together to reduce amount of section switch statements.
   SmallVector<SectionCPs, 4> CPSections;
   for (unsigned i = 0, e = CP.size(); i != e; ++i) {
     const MachineConstantPoolEntry &CPE = CP[i];
     unsigned Align = CPE.getAlignment();
-    uint64_t Size = TD.getTypeAllocSize(CPE.getType());
-    const Section *S =
-      TAI->getSectionForMergableConstant(Size, CPE.getRelocationInfo());
+    
+    SectionKind Kind;
+    switch (CPE.getRelocationInfo()) {
+    default: llvm_unreachable("Unknown section kind");
+    case 2: Kind = SectionKind::getReadOnlyWithRel(); break;
+    case 1: Kind = SectionKind::getReadOnlyWithRelLocal(); break;
+    case 0:
+      switch (TM.getTargetData()->getTypeAllocSize(CPE.getType())) {
+      case 4:   Kind = SectionKind::getMergableConst4(); break;
+      case 8:   Kind = SectionKind::getMergableConst8(); break;
+      case 16:  Kind = SectionKind::getMergableConst16(); break;
+      default:  Kind = SectionKind::getMergableConst(); break;
+      }
+    }
+
+    const Section *S = TAI->getSectionForMergableConstant(Kind);
     
     // The number of sections are small, just do a linear search from the
     // last section to the first.

lib/CodeGen/ELFWriter.cpp

@@ -29,7 +29,6 @@
 //===----------------------------------------------------------------------===//
 
 #define DEBUG_TYPE "elfwriter"
-
 #include "ELF.h"
 #include "ELFWriter.h"
 #include "ELFCodeEmitter.h"
@@ -155,10 +154,21 @@ ELFSection &ELFWriter::getJumpTableSection() {
 
 // Get a constant pool section based on the section name returned by TAI
 ELFSection &ELFWriter::getConstantPoolSection(MachineConstantPoolEntry &CPE) {
-  uint64_t Size = TM.getTargetData()->getTypeAllocSize(CPE.getType());
+  SectionKind Kind;
+  switch (CPE.getRelocationInfo()) {
+  default: llvm_unreachable("Unknown section kind");
+  case 2: Kind = SectionKind::getReadOnlyWithRel(); break;
+  case 1: Kind = SectionKind::getReadOnlyWithRelLocal(); break;
+  case 0:
+    switch (TM.getTargetData()->getTypeAllocSize(CPE.getType())) {
+    case 4:   Kind = SectionKind::getMergableConst4(); break;
+    case 8:   Kind = SectionKind::getMergableConst8(); break;
+    case 16:  Kind = SectionKind::getMergableConst16(); break;
+    default:  Kind = SectionKind::getMergableConst(); break;
+    }
+  }
   
-  std::string CstPoolName =
-    TAI->getSectionForMergableConstant(Size,CPE.getRelocationInfo())->getName();
+  std::string CstPoolName = TAI->getSectionForMergableConstant(Kind)->getName();
   return getSection(CstPoolName,
                     ELFSection::SHT_PROGBITS,
                     ELFSection::SHF_MERGE | ELFSection::SHF_ALLOC,

lib/Target/DarwinTargetAsmInfo.cpp

@@ -188,25 +188,18 @@ DarwinTargetAsmInfo::MergeableStringSection(const GlobalVariable *GV) const {
 }
 
 const Section *
-DarwinTargetAsmInfo::getSectionForMergableConstant(uint64_t Size,
-                                                   unsigned ReloInfo) const {
+DarwinTargetAsmInfo::getSectionForMergableConstant(SectionKind Kind) const {
   // If this constant requires a relocation, we have to put it in the data
   // segment, not in the text segment.
-  if (ReloInfo != 0)
+  if (Kind.isDataRel())
     return ConstDataSection;
   
-  switch (Size) {
-  default: break;
-  case 4:
+  if (Kind.isMergableConst4())
     return FourByteConstantSection;
-  case 8:
+  if (Kind.isMergableConst8())
     return EightByteConstantSection;
-  case 16:
-    if (SixteenByteConstantSection)
-      return SixteenByteConstantSection;
-    break;
-  }
-  
+  if (Kind.isMergableConst16() && SixteenByteConstantSection)
+    return SixteenByteConstantSection;
   return ReadOnlySection;  // .const
 }
 

lib/Target/ELFTargetAsmInfo.cpp

@@ -92,21 +92,8 @@ ELFTargetAsmInfo::SelectSectionForGlobal(const GlobalValue *GV,
 /// specified size and relocation information, return a section that it
 /// should be placed in.
 const Section *
-ELFTargetAsmInfo::getSectionForMergableConstant(uint64_t Size,
-                                                unsigned ReloInfo) const {
-  // If this constant pool entry has relocations, stick it into a relocatable
-  // section.
-  if (ReloInfo == 2)
-    return DataRelROSection;
-  if (ReloInfo == 1)
-    return DataRelROLocalSection;
-  
-  switch (Size) {
-  default: return ReadOnlySection;   // .rodata
-  case 4:  return MergableConst4Section;
-  case 8:  return MergableConst8Section;
-  case 16: return MergableConst16Section;
-  }
+ELFTargetAsmInfo::getSectionForMergableConstant(SectionKind Kind) const {
+  return SelectSectionForGlobal(0, Kind);
 }
 
 /// getFlagsForNamedSection - If this target wants to be able to infer

lib/Target/TargetAsmInfo.cpp

@@ -372,9 +372,8 @@ TargetAsmInfo::SelectSectionForGlobal(const GlobalValue *GV,
 /// specified size and relocation information, return a section that it
 /// should be placed in.
 const Section *
-TargetAsmInfo::getSectionForMergableConstant(uint64_t Size,
-                                             unsigned ReloInfo) const {
-  if (ReloInfo == 0)
+TargetAsmInfo::getSectionForMergableConstant(SectionKind Kind) const {
+  if (Kind.isReadOnly())
     if (const Section *S = getReadOnlySection())
       return S;