1 //===- lib/MC/MCObjectStreamer.cpp - Object File MCStreamer Interface -----===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/MC/MCObjectStreamer.h"
12 #include "llvm/Support/ErrorHandling.h"
13 #include "llvm/MC/MCAssembler.h"
14 #include "llvm/MC/MCCodeEmitter.h"
15 #include "llvm/MC/MCExpr.h"
16 #include "llvm/Target/TargetAsmBackend.h"
19 MCObjectStreamer::MCObjectStreamer(MCContext &Context, TargetAsmBackend &TAB,
20 raw_ostream &_OS, MCCodeEmitter *_Emitter,
21 bool _PadSectionToAlignment)
22 : MCStreamer(Context), Assembler(new MCAssembler(Context, TAB,
24 _PadSectionToAlignment,
30 MCObjectStreamer::~MCObjectStreamer() {
31 delete &Assembler->getBackend();
32 delete &Assembler->getEmitter();
36 MCFragment *MCObjectStreamer::getCurrentFragment() const {
37 assert(getCurrentSectionData() && "No current section!");
39 if (!getCurrentSectionData()->empty())
40 return &getCurrentSectionData()->getFragmentList().back();
45 MCDataFragment *MCObjectStreamer::getOrCreateDataFragment() const {
46 MCDataFragment *F = dyn_cast_or_null<MCDataFragment>(getCurrentFragment());
48 F = new MCDataFragment(getCurrentSectionData());
52 const MCExpr *MCObjectStreamer::AddValueSymbols(const MCExpr *Value) {
53 switch (Value->getKind()) {
54 case MCExpr::Target: llvm_unreachable("Can't handle target exprs yet!");
55 case MCExpr::Constant:
58 case MCExpr::Binary: {
59 const MCBinaryExpr *BE = cast<MCBinaryExpr>(Value);
60 AddValueSymbols(BE->getLHS());
61 AddValueSymbols(BE->getRHS());
65 case MCExpr::SymbolRef:
66 Assembler->getOrCreateSymbolData(cast<MCSymbolRefExpr>(Value)->getSymbol());
70 AddValueSymbols(cast<MCUnaryExpr>(Value)->getSubExpr());
77 void MCObjectStreamer::SwitchSection(const MCSection *Section) {
78 assert(Section && "Cannot switch to a null section!");
80 // If already in this section, then this is a noop.
81 if (Section == CurSection) return;
83 PrevSection = CurSection;
85 CurSectionData = &getAssembler().getOrCreateSectionData(*Section);
88 void MCObjectStreamer::Finish() {
89 getAssembler().Finish();