1*2cab237bSDimitry Andric //===--- CodeGenHwModes.cpp -----------------------------------------------===//
2*2cab237bSDimitry Andric //
3*2cab237bSDimitry Andric // The LLVM Compiler Infrastructure
4*2cab237bSDimitry Andric //
5*2cab237bSDimitry Andric // This file is distributed under the University of Illinois Open Source
6*2cab237bSDimitry Andric // License. See LICENSE.TXT for details.
7*2cab237bSDimitry Andric //
8*2cab237bSDimitry Andric //===----------------------------------------------------------------------===//
9*2cab237bSDimitry Andric // Classes to parse and store HW mode information for instruction selection
10*2cab237bSDimitry Andric //===----------------------------------------------------------------------===//
11*2cab237bSDimitry Andric
12*2cab237bSDimitry Andric #include "CodeGenHwModes.h"
13*2cab237bSDimitry Andric #include "llvm/Support/Debug.h"
14*2cab237bSDimitry Andric #include "llvm/Support/raw_ostream.h"
15*2cab237bSDimitry Andric #include "llvm/TableGen/Error.h"
16*2cab237bSDimitry Andric #include "llvm/TableGen/Record.h"
17*2cab237bSDimitry Andric
18*2cab237bSDimitry Andric using namespace llvm;
19*2cab237bSDimitry Andric
20*2cab237bSDimitry Andric StringRef CodeGenHwModes::DefaultModeName = "DefaultMode";
21*2cab237bSDimitry Andric
HwMode(Record * R)22*2cab237bSDimitry Andric HwMode::HwMode(Record *R) {
23*2cab237bSDimitry Andric Name = R->getName();
24*2cab237bSDimitry Andric Features = R->getValueAsString("Features");
25*2cab237bSDimitry Andric }
26*2cab237bSDimitry Andric
27*2cab237bSDimitry Andric LLVM_DUMP_METHOD
dump() const28*2cab237bSDimitry Andric void HwMode::dump() const {
29*2cab237bSDimitry Andric dbgs() << Name << ": " << Features << '\n';
30*2cab237bSDimitry Andric }
31*2cab237bSDimitry Andric
HwModeSelect(Record * R,CodeGenHwModes & CGH)32*2cab237bSDimitry Andric HwModeSelect::HwModeSelect(Record *R, CodeGenHwModes &CGH) {
33*2cab237bSDimitry Andric std::vector<Record*> Modes = R->getValueAsListOfDefs("Modes");
34*2cab237bSDimitry Andric std::vector<Record*> Objects = R->getValueAsListOfDefs("Objects");
35*2cab237bSDimitry Andric if (Modes.size() != Objects.size()) {
36*2cab237bSDimitry Andric PrintError(R->getLoc(), "in record " + R->getName() +
37*2cab237bSDimitry Andric " derived from HwModeSelect: the lists Modes and Objects should "
38*2cab237bSDimitry Andric "have the same size");
39*2cab237bSDimitry Andric report_fatal_error("error in target description.");
40*2cab237bSDimitry Andric }
41*2cab237bSDimitry Andric for (unsigned i = 0, e = Modes.size(); i != e; ++i) {
42*2cab237bSDimitry Andric unsigned ModeId = CGH.getHwModeId(Modes[i]->getName());
43*2cab237bSDimitry Andric Items.push_back(std::make_pair(ModeId, Objects[i]));
44*2cab237bSDimitry Andric }
45*2cab237bSDimitry Andric }
46*2cab237bSDimitry Andric
47*2cab237bSDimitry Andric LLVM_DUMP_METHOD
dump() const48*2cab237bSDimitry Andric void HwModeSelect::dump() const {
49*2cab237bSDimitry Andric dbgs() << '{';
50*2cab237bSDimitry Andric for (const PairType &P : Items)
51*2cab237bSDimitry Andric dbgs() << " (" << P.first << ',' << P.second->getName() << ')';
52*2cab237bSDimitry Andric dbgs() << " }\n";
53*2cab237bSDimitry Andric }
54*2cab237bSDimitry Andric
CodeGenHwModes(RecordKeeper & RK)55*2cab237bSDimitry Andric CodeGenHwModes::CodeGenHwModes(RecordKeeper &RK) : Records(RK) {
56*2cab237bSDimitry Andric std::vector<Record*> MRs = Records.getAllDerivedDefinitions("HwMode");
57*2cab237bSDimitry Andric // The default mode needs a definition in the .td sources for TableGen
58*2cab237bSDimitry Andric // to accept references to it. We need to ignore the definition here.
59*2cab237bSDimitry Andric for (auto I = MRs.begin(), E = MRs.end(); I != E; ++I) {
60*2cab237bSDimitry Andric if ((*I)->getName() != DefaultModeName)
61*2cab237bSDimitry Andric continue;
62*2cab237bSDimitry Andric MRs.erase(I);
63*2cab237bSDimitry Andric break;
64*2cab237bSDimitry Andric }
65*2cab237bSDimitry Andric
66*2cab237bSDimitry Andric for (Record *R : MRs) {
67*2cab237bSDimitry Andric Modes.emplace_back(R);
68*2cab237bSDimitry Andric unsigned NewId = Modes.size();
69*2cab237bSDimitry Andric ModeIds.insert(std::make_pair(Modes[NewId-1].Name, NewId));
70*2cab237bSDimitry Andric }
71*2cab237bSDimitry Andric
72*2cab237bSDimitry Andric std::vector<Record*> MSs = Records.getAllDerivedDefinitions("HwModeSelect");
73*2cab237bSDimitry Andric for (Record *R : MSs) {
74*2cab237bSDimitry Andric auto P = ModeSelects.emplace(std::make_pair(R, HwModeSelect(R, *this)));
75*2cab237bSDimitry Andric assert(P.second);
76*2cab237bSDimitry Andric (void)P;
77*2cab237bSDimitry Andric }
78*2cab237bSDimitry Andric }
79*2cab237bSDimitry Andric
getHwModeId(StringRef Name) const80*2cab237bSDimitry Andric unsigned CodeGenHwModes::getHwModeId(StringRef Name) const {
81*2cab237bSDimitry Andric if (Name == DefaultModeName)
82*2cab237bSDimitry Andric return DefaultMode;
83*2cab237bSDimitry Andric auto F = ModeIds.find(Name);
84*2cab237bSDimitry Andric assert(F != ModeIds.end() && "Unknown mode name");
85*2cab237bSDimitry Andric return F->second;
86*2cab237bSDimitry Andric }
87*2cab237bSDimitry Andric
getHwModeSelect(Record * R) const88*2cab237bSDimitry Andric const HwModeSelect &CodeGenHwModes::getHwModeSelect(Record *R) const {
89*2cab237bSDimitry Andric auto F = ModeSelects.find(R);
90*2cab237bSDimitry Andric assert(F != ModeSelects.end() && "Record is not a \"mode select\"");
91*2cab237bSDimitry Andric return F->second;
92*2cab237bSDimitry Andric }
93*2cab237bSDimitry Andric
94*2cab237bSDimitry Andric LLVM_DUMP_METHOD
dump() const95*2cab237bSDimitry Andric void CodeGenHwModes::dump() const {
96*2cab237bSDimitry Andric dbgs() << "Modes: {\n";
97*2cab237bSDimitry Andric for (const HwMode &M : Modes) {
98*2cab237bSDimitry Andric dbgs() << " ";
99*2cab237bSDimitry Andric M.dump();
100*2cab237bSDimitry Andric }
101*2cab237bSDimitry Andric dbgs() << "}\n";
102*2cab237bSDimitry Andric
103*2cab237bSDimitry Andric dbgs() << "ModeIds: {\n";
104*2cab237bSDimitry Andric for (const auto &P : ModeIds)
105*2cab237bSDimitry Andric dbgs() << " " << P.first() << " -> " << P.second << '\n';
106*2cab237bSDimitry Andric dbgs() << "}\n";
107*2cab237bSDimitry Andric
108*2cab237bSDimitry Andric dbgs() << "ModeSelects: {\n";
109*2cab237bSDimitry Andric for (const auto &P : ModeSelects) {
110*2cab237bSDimitry Andric dbgs() << " " << P.first->getName() << " -> ";
111*2cab237bSDimitry Andric P.second.dump();
112*2cab237bSDimitry Andric }
113*2cab237bSDimitry Andric dbgs() << "}\n";
114*2cab237bSDimitry Andric }
115