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