Add support for BlockAddress values in ARM constant pools.
[oota-llvm.git] / lib / Target / ARM / ARMConstantPoolValue.cpp
1 //===- ARMConstantPoolValue.cpp - ARM constantpool value --------*- C++ -*-===//
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 ARM specific constantpool value class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "ARMConstantPoolValue.h"
15 #include "llvm/ADT/FoldingSet.h"
16 #include "llvm/Constant.h"
17 #include "llvm/Constants.h"
18 #include "llvm/GlobalValue.h"
19 #include "llvm/Type.h"
20 #include "llvm/Support/raw_ostream.h"
21 #include <cstdlib>
22 using namespace llvm;
23
24 ARMConstantPoolValue::ARMConstantPoolValue(Constant *cval, unsigned id,
25                                            ARMCP::ARMCPKind K,
26                                            unsigned char PCAdj,
27                                            const char *Modif,
28                                            bool AddCA)
29   : MachineConstantPoolValue((const Type*)cval->getType()),
30     CVal(cval), S(NULL), LabelId(id), Kind(K), PCAdjust(PCAdj),
31     Modifier(Modif), AddCurrentAddress(AddCA) {}
32
33 ARMConstantPoolValue::ARMConstantPoolValue(LLVMContext &C,
34                                            const char *s, unsigned id,
35                                            unsigned char PCAdj,
36                                            const char *Modif,
37                                            bool AddCA)
38   : MachineConstantPoolValue((const Type*)Type::getInt32Ty(C)),
39     CVal(NULL), S(strdup(s)), LabelId(id), Kind(ARMCP::CPExtSymbol),
40     PCAdjust(PCAdj), Modifier(Modif), AddCurrentAddress(AddCA) {}
41
42 ARMConstantPoolValue::ARMConstantPoolValue(GlobalValue *gv, const char *Modif)
43   : MachineConstantPoolValue((const Type*)Type::getInt32Ty(gv->getContext())),
44     CVal(gv), S(NULL), LabelId(0), Kind(ARMCP::CPValue), PCAdjust(0),
45     Modifier(Modif) {}
46
47 GlobalValue *ARMConstantPoolValue::getGV() const {
48   return dyn_cast_or_null<GlobalValue>(CVal);
49 }
50
51 BlockAddress *ARMConstantPoolValue::getBlockAddress() const {
52   return dyn_cast_or_null<BlockAddress>(CVal);
53 }
54
55 int ARMConstantPoolValue::getExistingMachineCPValue(MachineConstantPool *CP,
56                                                     unsigned Alignment) {
57   unsigned AlignMask = Alignment - 1;
58   const std::vector<MachineConstantPoolEntry> Constants = CP->getConstants();
59   for (unsigned i = 0, e = Constants.size(); i != e; ++i) {
60     if (Constants[i].isMachineConstantPoolEntry() &&
61         (Constants[i].getAlignment() & AlignMask) == 0) {
62       ARMConstantPoolValue *CPV =
63         (ARMConstantPoolValue *)Constants[i].Val.MachineCPVal;
64       if (CPV->CVal == CVal &&
65           CPV->S == S &&
66           CPV->LabelId == LabelId &&
67           CPV->PCAdjust == PCAdjust)
68         return i;
69     }
70   }
71
72   return -1;
73 }
74
75 ARMConstantPoolValue::~ARMConstantPoolValue() {
76   free((void*)S);
77 }
78
79 void
80 ARMConstantPoolValue::AddSelectionDAGCSEId(FoldingSetNodeID &ID) {
81   ID.AddPointer(CVal);
82   ID.AddPointer(S);
83   ID.AddInteger(LabelId);
84   ID.AddInteger(PCAdjust);
85 }
86
87 void ARMConstantPoolValue::dump() const {
88   errs() << "  " << *this;
89 }
90
91
92 void ARMConstantPoolValue::print(raw_ostream &O) const {
93   if (CVal)
94     O << CVal->getName();
95   else
96     O << S;
97   if (Modifier) O << "(" << Modifier << ")";
98   if (PCAdjust != 0) {
99     O << "-(LPC" << LabelId << "+" << (unsigned)PCAdjust;
100     if (AddCurrentAddress) O << "-.";
101     O << ")";
102   }
103 }