Try harder to evaluate expressions when printing assembly.
[oota-llvm.git] / lib / Target / Mips / MCTargetDesc / MipsMCExpr.cpp
1 //===-- MipsMCExpr.cpp - Mips specific MC expression classes --------------===//
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 #define DEBUG_TYPE "mipsmcexpr"
11 #include "MipsMCExpr.h"
12 #include "llvm/MC/MCAsmInfo.h"
13 #include "llvm/MC/MCAssembler.h"
14 #include "llvm/MC/MCContext.h"
15
16 using namespace llvm;
17
18 const MipsMCExpr*
19 MipsMCExpr::Create(VariantKind Kind, const MCExpr *Expr,
20                    MCContext &Ctx) {
21   return new (Ctx) MipsMCExpr(Kind, Expr);
22 }
23
24 void MipsMCExpr::PrintImpl(raw_ostream &OS) const {
25   switch (Kind) {
26   default: llvm_unreachable("Invalid kind!");
27   case VK_Mips_ABS_LO: OS << "%lo"; break;
28   case VK_Mips_ABS_HI: OS << "%hi"; break;
29   }
30
31   OS << '(';
32   Expr->print(OS);
33   OS << ')';
34 }
35
36 bool
37 MipsMCExpr::EvaluateAsRelocatableImpl(MCValue &Res,
38                                       const MCAsmLayout *Layout) const {
39   return getSubExpr()->EvaluateAsRelocatable(Res, Layout);
40 }
41
42 // FIXME: This basically copies MCObjectStreamer::AddValueSymbols. Perhaps
43 // that method should be made public?
44 static void AddValueSymbolsImpl(const MCExpr *Value, MCAssembler *Asm) {
45   switch (Value->getKind()) {
46   case MCExpr::Target:
47     llvm_unreachable("Can't handle nested target expr!");
48
49   case MCExpr::Constant:
50     break;
51
52   case MCExpr::Binary: {
53     const MCBinaryExpr *BE = cast<MCBinaryExpr>(Value);
54     AddValueSymbolsImpl(BE->getLHS(), Asm);
55     AddValueSymbolsImpl(BE->getRHS(), Asm);
56     break;
57   }
58
59   case MCExpr::SymbolRef:
60     Asm->getOrCreateSymbolData(cast<MCSymbolRefExpr>(Value)->getSymbol());
61     break;
62
63   case MCExpr::Unary:
64     AddValueSymbolsImpl(cast<MCUnaryExpr>(Value)->getSubExpr(), Asm);
65     break;
66   }
67 }
68
69 void MipsMCExpr::AddValueSymbols(MCAssembler *Asm) const {
70   AddValueSymbolsImpl(getSubExpr(), Asm);
71 }