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 : MCStreamer(Context), Assembler(new MCAssembler(Context, TAB,
27 MCObjectStreamer::~MCObjectStreamer() {
28 delete &Assembler->getBackend();
29 delete &Assembler->getEmitter();
33 MCFragment *MCObjectStreamer::getCurrentFragment() const {
34 assert(getCurrentSectionData() && "No current section!");
36 if (!getCurrentSectionData()->empty())
37 return &getCurrentSectionData()->getFragmentList().back();
42 MCDataFragment *MCObjectStreamer::getOrCreateDataFragment() const {
43 MCDataFragment *F = dyn_cast_or_null<MCDataFragment>(getCurrentFragment());
45 F = new MCDataFragment(getCurrentSectionData());
49 const MCExpr *MCObjectStreamer::AddValueSymbols(const MCExpr *Value) {
50 switch (Value->getKind()) {
51 case MCExpr::Target: llvm_unreachable("Can't handle target exprs yet!");
52 case MCExpr::Constant:
55 case MCExpr::Binary: {
56 const MCBinaryExpr *BE = cast<MCBinaryExpr>(Value);
57 AddValueSymbols(BE->getLHS());
58 AddValueSymbols(BE->getRHS());
62 case MCExpr::SymbolRef:
63 Assembler->getOrCreateSymbolData(cast<MCSymbolRefExpr>(Value)->getSymbol());
67 AddValueSymbols(cast<MCUnaryExpr>(Value)->getSubExpr());
74 void MCObjectStreamer::SwitchSection(const MCSection *Section) {
75 assert(Section && "Cannot switch to a null section!");
77 // If already in this section, then this is a noop.
78 if (Section == CurSection) return;
81 CurSectionData = &getAssembler().getOrCreateSectionData(*Section);
84 void MCObjectStreamer::Finish() {
85 getAssembler().Finish();