Line data    Source code

       1             : //===- ConstantPools.cpp - ConstantPool class -----------------------------===//
       2             : //
       3             : //                     The LLVM Compiler Infrastructure
       4             : //
       5             : // This file is distributed under the University of Illinois Open Source
       6             : // License. See LICENSE.TXT for details.
       7             : //
       8             : //===----------------------------------------------------------------------===//
       9             : //
      10             : // This file implements the ConstantPool and  AssemblerConstantPools classes.
      11             : //
      12             : //===----------------------------------------------------------------------===//
      13             : 
      14             : #include "llvm/MC/ConstantPools.h"
      15             : #include "llvm/MC/MCContext.h"
      16             : #include "llvm/MC/MCDirectives.h"
      17             : #include "llvm/MC/MCExpr.h"
      18             : #include "llvm/MC/MCStreamer.h"
      19             : #include "llvm/Support/Casting.h"
      20             : 
      21             : using namespace llvm;
      22             : 
      23             : //
      24             : // ConstantPool implementation
      25             : //
      26             : // Emit the contents of the constant pool using the provided streamer.
      27         191 : void ConstantPool::emitEntries(MCStreamer &Streamer) {
      28         191 :   if (Entries.empty())
      29             :     return;
      30         191 :   Streamer.EmitDataRegion(MCDR_DataRegion);
      31         491 :   for (const ConstantPoolEntry &Entry : Entries) {
      32         300 :     Streamer.EmitCodeAlignment(Entry.Size); // align naturally
      33         300 :     Streamer.EmitLabel(Entry.Label);
      34         300 :     Streamer.EmitValue(Entry.Value, Entry.Size, Entry.Loc);
      35             :   }
      36         191 :   Streamer.EmitDataRegion(MCDR_DataRegionEnd);
      37             :   Entries.clear();
      38             : }
      39             : 
      40         324 : const MCExpr *ConstantPool::addEntry(const MCExpr *Value, MCContext &Context,
      41             :                                      unsigned Size, SMLoc Loc) {
      42             :   const MCConstantExpr *C = dyn_cast<MCConstantExpr>(Value);
      43             : 
      44             :   // Check if there is existing entry for the same constant. If so, reuse it.
      45         235 :   auto Itr = C ? CachedEntries.find(C->getValue()) : CachedEntries.end();
      46         324 :   if (Itr != CachedEntries.end())
      47          20 :     return Itr->second;
      48             : 
      49         304 :   MCSymbol *CPEntryLabel = Context.createTempSymbol();
      50             : 
      51         608 :   Entries.push_back(ConstantPoolEntry(CPEntryLabel, Value, Size, Loc));
      52             :   const auto SymRef = MCSymbolRefExpr::create(CPEntryLabel, Context);
      53         304 :   if (C)
      54         215 :     CachedEntries[C->getValue()] = SymRef;
      55             :   return SymRef;
      56             : }
      57             : 
      58         422 : bool ConstantPool::empty() { return Entries.empty(); }
      59             : 
      60          34 : void ConstantPool::clearCache() {
      61             :   CachedEntries.clear();
      62          34 : }
      63             : 
      64             : //
      65             : // AssemblerConstantPools implementation
      66             : //
      67          88 : ConstantPool *AssemblerConstantPools::getConstantPool(MCSection *Section) {
      68          88 :   ConstantPoolMapTy::iterator CP = ConstantPools.find(Section);
      69          88 :   if (CP == ConstantPools.end())
      70             :     return nullptr;
      71             : 
      72          68 :   return &CP->second;
      73             : }
      74             : 
      75             : ConstantPool &
      76         324 : AssemblerConstantPools::getOrCreateConstantPool(MCSection *Section) {
      77         324 :   return ConstantPools[Section];
      78             : }
      79             : 
      80         211 : static void emitConstantPool(MCStreamer &Streamer, MCSection *Section,
      81             :                              ConstantPool &CP) {
      82         211 :   if (!CP.empty()) {
      83         191 :     Streamer.SwitchSection(Section);
      84         191 :     CP.emitEntries(Streamer);
      85             :   }
      86         211 : }
      87             : 
      88        6051 : void AssemblerConstantPools::emitAll(MCStreamer &Streamer) {
      89             :   // Dump contents of assembler constant pools.
      90        6228 :   for (auto &CPI : ConstantPools) {
      91         177 :     MCSection *Section = CPI.first;
      92         177 :     ConstantPool &CP = CPI.second;
      93             : 
      94         177 :     emitConstantPool(Streamer, Section, CP);
      95             :   }
      96        6051 : }
      97             : 
      98          44 : void AssemblerConstantPools::emitForCurrentSection(MCStreamer &Streamer) {
      99             :   MCSection *Section = Streamer.getCurrentSectionOnly();
     100          44 :   if (ConstantPool *CP = getConstantPool(Section))
     101          34 :     emitConstantPool(Streamer, Section, *CP);
     102          44 : }
     103             : 
     104          44 : void AssemblerConstantPools::clearCacheForCurrentSection(MCStreamer &Streamer) {
     105             :   MCSection *Section = Streamer.getCurrentSectionOnly();
     106          44 :   if (ConstantPool *CP = getConstantPool(Section))
     107          34 :     CP->clearCache();
     108          44 : }
     109             : 
     110         324 : const MCExpr *AssemblerConstantPools::addEntry(MCStreamer &Streamer,
     111             :                                                const MCExpr *Expr,
     112             :                                                unsigned Size, SMLoc Loc) {
     113             :   MCSection *Section = Streamer.getCurrentSectionOnly();
     114         324 :   return getOrCreateConstantPool(Section).addEntry(Expr, Streamer.getContext(),
     115         324 :                                                    Size, Loc);
     116             : }