168d6d8abSJakob Stoklund Olesen //===- CodeGenRegisters.cpp - Register and RegisterClass Info -------------===//
268d6d8abSJakob Stoklund Olesen //
368d6d8abSJakob Stoklund Olesen //                     The LLVM Compiler Infrastructure
468d6d8abSJakob Stoklund Olesen //
568d6d8abSJakob Stoklund Olesen // This file is distributed under the University of Illinois Open Source
668d6d8abSJakob Stoklund Olesen // License. See LICENSE.TXT for details.
768d6d8abSJakob Stoklund Olesen //
868d6d8abSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
968d6d8abSJakob Stoklund Olesen //
1068d6d8abSJakob Stoklund Olesen // This file defines structures to encapsulate information gleaned from the
1168d6d8abSJakob Stoklund Olesen // target register and register class definitions.
1268d6d8abSJakob Stoklund Olesen //
1368d6d8abSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
1468d6d8abSJakob Stoklund Olesen 
1568d6d8abSJakob Stoklund Olesen #include "CodeGenRegisters.h"
1668d6d8abSJakob Stoklund Olesen #include "CodeGenTarget.h"
171d7a2c57SAndrew Trick #include "llvm/ADT/IntEqClasses.h"
18c0fc173dSJakob Stoklund Olesen #include "llvm/ADT/STLExtras.h"
1991d19d8eSChandler Carruth #include "llvm/ADT/SmallVector.h"
2068d6d8abSJakob Stoklund Olesen #include "llvm/ADT/StringExtras.h"
219a7f4b76SJim Grosbach #include "llvm/ADT/Twine.h"
2291d19d8eSChandler Carruth #include "llvm/TableGen/Error.h"
2368d6d8abSJakob Stoklund Olesen 
2468d6d8abSJakob Stoklund Olesen using namespace llvm;
2568d6d8abSJakob Stoklund Olesen 
2668d6d8abSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
27f1bb1519SJakob Stoklund Olesen //                             CodeGenSubRegIndex
28f1bb1519SJakob Stoklund Olesen //===----------------------------------------------------------------------===//
29f1bb1519SJakob Stoklund Olesen 
30f1bb1519SJakob Stoklund Olesen CodeGenSubRegIndex::CodeGenSubRegIndex(Record *R, unsigned Enum)
31d346d487SJakob Stoklund Olesen   : TheDef(R), EnumValue(Enum), LaneMask(0) {
3270a0bbcaSJakob Stoklund Olesen   Name = R->getName();
3370a0bbcaSJakob Stoklund Olesen   if (R->getValue("Namespace"))
3470a0bbcaSJakob Stoklund Olesen     Namespace = R->getValueAsString("Namespace");
35f1bb1519SJakob Stoklund Olesen }
36f1bb1519SJakob Stoklund Olesen 
3770a0bbcaSJakob Stoklund Olesen CodeGenSubRegIndex::CodeGenSubRegIndex(StringRef N, StringRef Nspace,
3870a0bbcaSJakob Stoklund Olesen                                        unsigned Enum)
39d346d487SJakob Stoklund Olesen   : TheDef(0), Name(N), Namespace(Nspace), EnumValue(Enum), LaneMask(0) {
40f1bb1519SJakob Stoklund Olesen }
41f1bb1519SJakob Stoklund Olesen 
42f1bb1519SJakob Stoklund Olesen std::string CodeGenSubRegIndex::getQualifiedName() const {
43f1bb1519SJakob Stoklund Olesen   std::string N = getNamespace();
44f1bb1519SJakob Stoklund Olesen   if (!N.empty())
45f1bb1519SJakob Stoklund Olesen     N += "::";
46f1bb1519SJakob Stoklund Olesen   N += getName();
47f1bb1519SJakob Stoklund Olesen   return N;
48f1bb1519SJakob Stoklund Olesen }
49f1bb1519SJakob Stoklund Olesen 
5021231609SJakob Stoklund Olesen void CodeGenSubRegIndex::updateComponents(CodeGenRegBank &RegBank) {
5170a0bbcaSJakob Stoklund Olesen   if (!TheDef)
5270a0bbcaSJakob Stoklund Olesen     return;
533697143aSJakob Stoklund Olesen 
5421231609SJakob Stoklund Olesen   std::vector<Record*> Comps = TheDef->getValueAsListOfDefs("ComposedOf");
553697143aSJakob Stoklund Olesen   if (!Comps.empty()) {
5621231609SJakob Stoklund Olesen     if (Comps.size() != 2)
57635debe8SJoerg Sonnenberger       PrintFatalError(TheDef->getLoc(),
58635debe8SJoerg Sonnenberger                       "ComposedOf must have exactly two entries");
5921231609SJakob Stoklund Olesen     CodeGenSubRegIndex *A = RegBank.getSubRegIdx(Comps[0]);
6021231609SJakob Stoklund Olesen     CodeGenSubRegIndex *B = RegBank.getSubRegIdx(Comps[1]);
6121231609SJakob Stoklund Olesen     CodeGenSubRegIndex *X = A->addComposite(B, this);
6221231609SJakob Stoklund Olesen     if (X)
63635debe8SJoerg Sonnenberger       PrintFatalError(TheDef->getLoc(), "Ambiguous ComposedOf entries");
6421231609SJakob Stoklund Olesen   }
6521231609SJakob Stoklund Olesen 
663697143aSJakob Stoklund Olesen   std::vector<Record*> Parts =
673697143aSJakob Stoklund Olesen     TheDef->getValueAsListOfDefs("CoveringSubRegIndices");
683697143aSJakob Stoklund Olesen   if (!Parts.empty()) {
693697143aSJakob Stoklund Olesen     if (Parts.size() < 2)
70635debe8SJoerg Sonnenberger       PrintFatalError(TheDef->getLoc(),
713697143aSJakob Stoklund Olesen                     "CoveredBySubRegs must have two or more entries");
723697143aSJakob Stoklund Olesen     SmallVector<CodeGenSubRegIndex*, 8> IdxParts;
733697143aSJakob Stoklund Olesen     for (unsigned i = 0, e = Parts.size(); i != e; ++i)
743697143aSJakob Stoklund Olesen       IdxParts.push_back(RegBank.getSubRegIdx(Parts[i]));
753697143aSJakob Stoklund Olesen     RegBank.addConcatSubRegIndex(IdxParts, this);
763697143aSJakob Stoklund Olesen   }
773697143aSJakob Stoklund Olesen }
783697143aSJakob Stoklund Olesen 
79d346d487SJakob Stoklund Olesen unsigned CodeGenSubRegIndex::computeLaneMask() {
80d346d487SJakob Stoklund Olesen   // Already computed?
81d346d487SJakob Stoklund Olesen   if (LaneMask)
82d346d487SJakob Stoklund Olesen     return LaneMask;
83d346d487SJakob Stoklund Olesen 
84d346d487SJakob Stoklund Olesen   // Recursion guard, shouldn't be required.
85d346d487SJakob Stoklund Olesen   LaneMask = ~0u;
86d346d487SJakob Stoklund Olesen 
87d346d487SJakob Stoklund Olesen   // The lane mask is simply the union of all sub-indices.
88d346d487SJakob Stoklund Olesen   unsigned M = 0;
89d346d487SJakob Stoklund Olesen   for (CompMap::iterator I = Composed.begin(), E = Composed.end(); I != E; ++I)
90d346d487SJakob Stoklund Olesen     M |= I->second->computeLaneMask();
91d346d487SJakob Stoklund Olesen   assert(M && "Missing lane mask, sub-register cycle?");
92d346d487SJakob Stoklund Olesen   LaneMask = M;
93d346d487SJakob Stoklund Olesen   return LaneMask;
94d346d487SJakob Stoklund Olesen }
95d346d487SJakob Stoklund Olesen 
96f1bb1519SJakob Stoklund Olesen //===----------------------------------------------------------------------===//
9768d6d8abSJakob Stoklund Olesen //                              CodeGenRegister
9868d6d8abSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
9968d6d8abSJakob Stoklund Olesen 
10084bd44ebSJakob Stoklund Olesen CodeGenRegister::CodeGenRegister(Record *R, unsigned Enum)
10184bd44ebSJakob Stoklund Olesen   : TheDef(R),
10284bd44ebSJakob Stoklund Olesen     EnumValue(Enum),
10384bd44ebSJakob Stoklund Olesen     CostPerUse(R->getValueAsInt("CostPerUse")),
104f43b5995SJakob Stoklund Olesen     CoveredBySubRegs(R->getValueAsBit("CoveredBySubRegs")),
1057f381bd2SJakob Stoklund Olesen     NumNativeRegUnits(0),
1063f3eb180SJakob Stoklund Olesen     SubRegsComplete(false),
10750ecd0ffSJakob Stoklund Olesen     SuperRegsComplete(false),
10850ecd0ffSJakob Stoklund Olesen     TopoSig(~0u)
10984bd44ebSJakob Stoklund Olesen {}
11068d6d8abSJakob Stoklund Olesen 
111c1e9087fSJakob Stoklund Olesen void CodeGenRegister::buildObjectGraph(CodeGenRegBank &RegBank) {
112c1e9087fSJakob Stoklund Olesen   std::vector<Record*> SRIs = TheDef->getValueAsListOfDefs("SubRegIndices");
113c1e9087fSJakob Stoklund Olesen   std::vector<Record*> SRs = TheDef->getValueAsListOfDefs("SubRegs");
114c1e9087fSJakob Stoklund Olesen 
115c1e9087fSJakob Stoklund Olesen   if (SRIs.size() != SRs.size())
116635debe8SJoerg Sonnenberger     PrintFatalError(TheDef->getLoc(),
117c1e9087fSJakob Stoklund Olesen                     "SubRegs and SubRegIndices must have the same size");
118c1e9087fSJakob Stoklund Olesen 
119c1e9087fSJakob Stoklund Olesen   for (unsigned i = 0, e = SRIs.size(); i != e; ++i) {
120c1e9087fSJakob Stoklund Olesen     ExplicitSubRegIndices.push_back(RegBank.getSubRegIdx(SRIs[i]));
121c1e9087fSJakob Stoklund Olesen     ExplicitSubRegs.push_back(RegBank.getReg(SRs[i]));
122c1e9087fSJakob Stoklund Olesen   }
123c08df9e5SJakob Stoklund Olesen 
124c08df9e5SJakob Stoklund Olesen   // Also compute leading super-registers. Each register has a list of
125c08df9e5SJakob Stoklund Olesen   // covered-by-subregs super-registers where it appears as the first explicit
126c08df9e5SJakob Stoklund Olesen   // sub-register.
127c08df9e5SJakob Stoklund Olesen   //
128c08df9e5SJakob Stoklund Olesen   // This is used by computeSecondarySubRegs() to find candidates.
129c08df9e5SJakob Stoklund Olesen   if (CoveredBySubRegs && !ExplicitSubRegs.empty())
130c08df9e5SJakob Stoklund Olesen     ExplicitSubRegs.front()->LeadingSuperRegs.push_back(this);
131534848b1SJakob Stoklund Olesen 
132bde91766SBenjamin Kramer   // Add ad hoc alias links. This is a symmetric relationship between two
133534848b1SJakob Stoklund Olesen   // registers, so build a symmetric graph by adding links in both ends.
134534848b1SJakob Stoklund Olesen   std::vector<Record*> Aliases = TheDef->getValueAsListOfDefs("Aliases");
135534848b1SJakob Stoklund Olesen   for (unsigned i = 0, e = Aliases.size(); i != e; ++i) {
136534848b1SJakob Stoklund Olesen     CodeGenRegister *Reg = RegBank.getReg(Aliases[i]);
137534848b1SJakob Stoklund Olesen     ExplicitAliases.push_back(Reg);
138534848b1SJakob Stoklund Olesen     Reg->ExplicitAliases.push_back(this);
139534848b1SJakob Stoklund Olesen   }
140c1e9087fSJakob Stoklund Olesen }
141c1e9087fSJakob Stoklund Olesen 
14268d6d8abSJakob Stoklund Olesen const std::string &CodeGenRegister::getName() const {
14368d6d8abSJakob Stoklund Olesen   return TheDef->getName();
14468d6d8abSJakob Stoklund Olesen }
14568d6d8abSJakob Stoklund Olesen 
1461d7a2c57SAndrew Trick namespace {
1471d7a2c57SAndrew Trick // Iterate over all register units in a set of registers.
1481d7a2c57SAndrew Trick class RegUnitIterator {
1491d7a2c57SAndrew Trick   CodeGenRegister::Set::const_iterator RegI, RegE;
1501d7a2c57SAndrew Trick   CodeGenRegister::RegUnitList::const_iterator UnitI, UnitE;
1511d7a2c57SAndrew Trick 
1521d7a2c57SAndrew Trick public:
1531d7a2c57SAndrew Trick   RegUnitIterator(const CodeGenRegister::Set &Regs):
1541d7a2c57SAndrew Trick     RegI(Regs.begin()), RegE(Regs.end()), UnitI(), UnitE() {
1551d7a2c57SAndrew Trick 
1561d7a2c57SAndrew Trick     if (RegI != RegE) {
1571d7a2c57SAndrew Trick       UnitI = (*RegI)->getRegUnits().begin();
1581d7a2c57SAndrew Trick       UnitE = (*RegI)->getRegUnits().end();
1591d7a2c57SAndrew Trick       advance();
1601d7a2c57SAndrew Trick     }
1611d7a2c57SAndrew Trick   }
1621d7a2c57SAndrew Trick 
1631d7a2c57SAndrew Trick   bool isValid() const { return UnitI != UnitE; }
1641d7a2c57SAndrew Trick 
165393f432dSBill Wendling   unsigned operator* () const { assert(isValid()); return *UnitI; }
1661d7a2c57SAndrew Trick 
1671d7a2c57SAndrew Trick   const CodeGenRegister *getReg() const { assert(isValid()); return *RegI; }
1681d7a2c57SAndrew Trick 
1691d7a2c57SAndrew Trick   /// Preincrement.  Move to the next unit.
1701d7a2c57SAndrew Trick   void operator++() {
1711d7a2c57SAndrew Trick     assert(isValid() && "Cannot advance beyond the last operand");
1721d7a2c57SAndrew Trick     ++UnitI;
1731d7a2c57SAndrew Trick     advance();
1741d7a2c57SAndrew Trick   }
1751d7a2c57SAndrew Trick 
1761d7a2c57SAndrew Trick protected:
1771d7a2c57SAndrew Trick   void advance() {
1781d7a2c57SAndrew Trick     while (UnitI == UnitE) {
1791d7a2c57SAndrew Trick       if (++RegI == RegE)
1801d7a2c57SAndrew Trick         break;
1811d7a2c57SAndrew Trick       UnitI = (*RegI)->getRegUnits().begin();
1821d7a2c57SAndrew Trick       UnitE = (*RegI)->getRegUnits().end();
1831d7a2c57SAndrew Trick     }
1841d7a2c57SAndrew Trick   }
1851d7a2c57SAndrew Trick };
1861d7a2c57SAndrew Trick } // namespace
1871d7a2c57SAndrew Trick 
188cdefdf1fSAndrew Trick // Merge two RegUnitLists maintaining the order and removing duplicates.
1891a004ca0SAndrew Trick // Overwrites MergedRU in the process.
1901a004ca0SAndrew Trick static void mergeRegUnits(CodeGenRegister::RegUnitList &MergedRU,
191c544e7c0SAndrew Trick                           const CodeGenRegister::RegUnitList &RRU) {
1921a004ca0SAndrew Trick   CodeGenRegister::RegUnitList LRU = MergedRU;
1931a004ca0SAndrew Trick   MergedRU.clear();
194c544e7c0SAndrew Trick   std::set_union(LRU.begin(), LRU.end(), RRU.begin(), RRU.end(),
195a890e3c6SAndrew Trick                  std::back_inserter(MergedRU));
1961a004ca0SAndrew Trick }
1971a004ca0SAndrew Trick 
1981d7a2c57SAndrew Trick // Return true of this unit appears in RegUnits.
1991d7a2c57SAndrew Trick static bool hasRegUnit(CodeGenRegister::RegUnitList &RegUnits, unsigned Unit) {
2001d7a2c57SAndrew Trick   return std::count(RegUnits.begin(), RegUnits.end(), Unit);
2011d7a2c57SAndrew Trick }
2021d7a2c57SAndrew Trick 
2031d7a2c57SAndrew Trick // Inherit register units from subregisters.
2041d7a2c57SAndrew Trick // Return true if the RegUnits changed.
2051d7a2c57SAndrew Trick bool CodeGenRegister::inheritRegUnits(CodeGenRegBank &RegBank) {
2061d7a2c57SAndrew Trick   unsigned OldNumUnits = RegUnits.size();
2071d7a2c57SAndrew Trick   for (SubRegMap::const_iterator I = SubRegs.begin(), E = SubRegs.end();
2081d7a2c57SAndrew Trick        I != E; ++I) {
2091d7a2c57SAndrew Trick     CodeGenRegister *SR = I->second;
2101d7a2c57SAndrew Trick     // Merge the subregister's units into this register's RegUnits.
2111d7a2c57SAndrew Trick     mergeRegUnits(RegUnits, SR->RegUnits);
2121d7a2c57SAndrew Trick   }
2131d7a2c57SAndrew Trick   return OldNumUnits != RegUnits.size();
2141d7a2c57SAndrew Trick }
2151d7a2c57SAndrew Trick 
21684bd44ebSJakob Stoklund Olesen const CodeGenRegister::SubRegMap &
2177d1fa380SJakob Stoklund Olesen CodeGenRegister::computeSubRegs(CodeGenRegBank &RegBank) {
21884bd44ebSJakob Stoklund Olesen   // Only compute this map once.
21984bd44ebSJakob Stoklund Olesen   if (SubRegsComplete)
22084bd44ebSJakob Stoklund Olesen     return SubRegs;
22184bd44ebSJakob Stoklund Olesen   SubRegsComplete = true;
22284bd44ebSJakob Stoklund Olesen 
223c1e9087fSJakob Stoklund Olesen   // First insert the explicit subregs and make sure they are fully indexed.
224c1e9087fSJakob Stoklund Olesen   for (unsigned i = 0, e = ExplicitSubRegs.size(); i != e; ++i) {
225c1e9087fSJakob Stoklund Olesen     CodeGenRegister *SR = ExplicitSubRegs[i];
226c1e9087fSJakob Stoklund Olesen     CodeGenSubRegIndex *Idx = ExplicitSubRegIndices[i];
227f1bb1519SJakob Stoklund Olesen     if (!SubRegs.insert(std::make_pair(Idx, SR)).second)
228635debe8SJoerg Sonnenberger       PrintFatalError(TheDef->getLoc(), "SubRegIndex " + Idx->getName() +
22984bd44ebSJakob Stoklund Olesen                       " appears twice in Register " + getName());
2309b41e5dbSJakob Stoklund Olesen     // Map explicit sub-registers first, so the names take precedence.
2319b41e5dbSJakob Stoklund Olesen     // The inherited sub-registers are mapped below.
2329b41e5dbSJakob Stoklund Olesen     SubReg2Idx.insert(std::make_pair(SR, Idx));
23384bd44ebSJakob Stoklund Olesen   }
23484bd44ebSJakob Stoklund Olesen 
23584bd44ebSJakob Stoklund Olesen   // Keep track of inherited subregs and how they can be reached.
23621231609SJakob Stoklund Olesen   SmallPtrSet<CodeGenRegister*, 8> Orphans;
23784bd44ebSJakob Stoklund Olesen 
23821231609SJakob Stoklund Olesen   // Clone inherited subregs and place duplicate entries in Orphans.
23984bd44ebSJakob Stoklund Olesen   // Here the order is important - earlier subregs take precedence.
240c1e9087fSJakob Stoklund Olesen   for (unsigned i = 0, e = ExplicitSubRegs.size(); i != e; ++i) {
241c1e9087fSJakob Stoklund Olesen     CodeGenRegister *SR = ExplicitSubRegs[i];
2427d1fa380SJakob Stoklund Olesen     const SubRegMap &Map = SR->computeSubRegs(RegBank);
243d2b4713eSJakob Stoklund Olesen 
24484bd44ebSJakob Stoklund Olesen     for (SubRegMap::const_iterator SI = Map.begin(), SE = Map.end(); SI != SE;
245d2b4713eSJakob Stoklund Olesen          ++SI) {
24684bd44ebSJakob Stoklund Olesen       if (!SubRegs.insert(*SI).second)
24721231609SJakob Stoklund Olesen         Orphans.insert(SI->second);
248d2b4713eSJakob Stoklund Olesen     }
24984bd44ebSJakob Stoklund Olesen   }
25084bd44ebSJakob Stoklund Olesen 
25121231609SJakob Stoklund Olesen   // Expand any composed subreg indices.
25221231609SJakob Stoklund Olesen   // If dsub_2 has ComposedOf = [qsub_1, dsub_0], and this register has a
25321231609SJakob Stoklund Olesen   // qsub_1 subreg, add a dsub_2 subreg.  Keep growing Indices and process
25421231609SJakob Stoklund Olesen   // expanded subreg indices recursively.
255c1e9087fSJakob Stoklund Olesen   SmallVector<CodeGenSubRegIndex*, 8> Indices = ExplicitSubRegIndices;
25621231609SJakob Stoklund Olesen   for (unsigned i = 0; i != Indices.size(); ++i) {
25721231609SJakob Stoklund Olesen     CodeGenSubRegIndex *Idx = Indices[i];
25821231609SJakob Stoklund Olesen     const CodeGenSubRegIndex::CompMap &Comps = Idx->getComposites();
25921231609SJakob Stoklund Olesen     CodeGenRegister *SR = SubRegs[Idx];
2607d1fa380SJakob Stoklund Olesen     const SubRegMap &Map = SR->computeSubRegs(RegBank);
26121231609SJakob Stoklund Olesen 
26221231609SJakob Stoklund Olesen     // Look at the possible compositions of Idx.
26321231609SJakob Stoklund Olesen     // They may not all be supported by SR.
26421231609SJakob Stoklund Olesen     for (CodeGenSubRegIndex::CompMap::const_iterator I = Comps.begin(),
26521231609SJakob Stoklund Olesen            E = Comps.end(); I != E; ++I) {
26621231609SJakob Stoklund Olesen       SubRegMap::const_iterator SRI = Map.find(I->first);
26721231609SJakob Stoklund Olesen       if (SRI == Map.end())
26821231609SJakob Stoklund Olesen         continue; // Idx + I->first doesn't exist in SR.
26921231609SJakob Stoklund Olesen       // Add I->second as a name for the subreg SRI->second, assuming it is
27021231609SJakob Stoklund Olesen       // orphaned, and the name isn't already used for something else.
27121231609SJakob Stoklund Olesen       if (SubRegs.count(I->second) || !Orphans.erase(SRI->second))
27221231609SJakob Stoklund Olesen         continue;
27321231609SJakob Stoklund Olesen       // We found a new name for the orphaned sub-register.
27421231609SJakob Stoklund Olesen       SubRegs.insert(std::make_pair(I->second, SRI->second));
27521231609SJakob Stoklund Olesen       Indices.push_back(I->second);
27621231609SJakob Stoklund Olesen     }
27721231609SJakob Stoklund Olesen   }
27821231609SJakob Stoklund Olesen 
27984bd44ebSJakob Stoklund Olesen   // Now Orphans contains the inherited subregisters without a direct index.
28084bd44ebSJakob Stoklund Olesen   // Create inferred indexes for all missing entries.
28121231609SJakob Stoklund Olesen   // Work backwards in the Indices vector in order to compose subregs bottom-up.
28221231609SJakob Stoklund Olesen   // Consider this subreg sequence:
28321231609SJakob Stoklund Olesen   //
28421231609SJakob Stoklund Olesen   //   qsub_1 -> dsub_0 -> ssub_0
28521231609SJakob Stoklund Olesen   //
28621231609SJakob Stoklund Olesen   // The qsub_1 -> dsub_0 composition becomes dsub_2, so the ssub_0 register
28721231609SJakob Stoklund Olesen   // can be reached in two different ways:
28821231609SJakob Stoklund Olesen   //
28921231609SJakob Stoklund Olesen   //   qsub_1 -> ssub_0
29021231609SJakob Stoklund Olesen   //   dsub_2 -> ssub_0
29121231609SJakob Stoklund Olesen   //
29221231609SJakob Stoklund Olesen   // We pick the latter composition because another register may have [dsub_0,
293bde91766SBenjamin Kramer   // dsub_1, dsub_2] subregs without necessarily having a qsub_1 subreg.  The
29421231609SJakob Stoklund Olesen   // dsub_2 -> ssub_0 composition can be shared.
29521231609SJakob Stoklund Olesen   while (!Indices.empty() && !Orphans.empty()) {
29621231609SJakob Stoklund Olesen     CodeGenSubRegIndex *Idx = Indices.pop_back_val();
29721231609SJakob Stoklund Olesen     CodeGenRegister *SR = SubRegs[Idx];
2987d1fa380SJakob Stoklund Olesen     const SubRegMap &Map = SR->computeSubRegs(RegBank);
29921231609SJakob Stoklund Olesen     for (SubRegMap::const_iterator SI = Map.begin(), SE = Map.end(); SI != SE;
30021231609SJakob Stoklund Olesen          ++SI)
30121231609SJakob Stoklund Olesen       if (Orphans.erase(SI->second))
30221231609SJakob Stoklund Olesen         SubRegs[RegBank.getCompositeSubRegIndex(Idx, SI->first)] = SI->second;
30384bd44ebSJakob Stoklund Olesen   }
3041a004ca0SAndrew Trick 
3059b41e5dbSJakob Stoklund Olesen   // Compute the inverse SubReg -> Idx map.
3069b41e5dbSJakob Stoklund Olesen   for (SubRegMap::const_iterator SI = SubRegs.begin(), SE = SubRegs.end();
3079b41e5dbSJakob Stoklund Olesen        SI != SE; ++SI) {
30859959363SJakob Stoklund Olesen     if (SI->second == this) {
309d7b66968SJakob Stoklund Olesen       ArrayRef<SMLoc> Loc;
31059959363SJakob Stoklund Olesen       if (TheDef)
31159959363SJakob Stoklund Olesen         Loc = TheDef->getLoc();
312635debe8SJoerg Sonnenberger       PrintFatalError(Loc, "Register " + getName() +
31359959363SJakob Stoklund Olesen                       " has itself as a sub-register");
31459959363SJakob Stoklund Olesen     }
31559959363SJakob Stoklund Olesen     // Ensure that every sub-register has a unique name.
31659959363SJakob Stoklund Olesen     DenseMap<const CodeGenRegister*, CodeGenSubRegIndex*>::iterator Ins =
31759959363SJakob Stoklund Olesen       SubReg2Idx.insert(std::make_pair(SI->second, SI->first)).first;
31859959363SJakob Stoklund Olesen     if (Ins->second == SI->first)
3199b41e5dbSJakob Stoklund Olesen       continue;
32059959363SJakob Stoklund Olesen     // Trouble: Two different names for SI->second.
321d7b66968SJakob Stoklund Olesen     ArrayRef<SMLoc> Loc;
32259959363SJakob Stoklund Olesen     if (TheDef)
32359959363SJakob Stoklund Olesen       Loc = TheDef->getLoc();
324635debe8SJoerg Sonnenberger     PrintFatalError(Loc, "Sub-register can't have two names: " +
32559959363SJakob Stoklund Olesen                   SI->second->getName() + " available as " +
32659959363SJakob Stoklund Olesen                   SI->first->getName() + " and " + Ins->second->getName());
3279b41e5dbSJakob Stoklund Olesen   }
3289b41e5dbSJakob Stoklund Olesen 
329c08df9e5SJakob Stoklund Olesen   // Derive possible names for sub-register concatenations from any explicit
330c08df9e5SJakob Stoklund Olesen   // sub-registers. By doing this before computeSecondarySubRegs(), we ensure
331c08df9e5SJakob Stoklund Olesen   // that getConcatSubRegIndex() won't invent any concatenated indices that the
332c08df9e5SJakob Stoklund Olesen   // user already specified.
333c08df9e5SJakob Stoklund Olesen   for (unsigned i = 0, e = ExplicitSubRegs.size(); i != e; ++i) {
334c08df9e5SJakob Stoklund Olesen     CodeGenRegister *SR = ExplicitSubRegs[i];
335c08df9e5SJakob Stoklund Olesen     if (!SR->CoveredBySubRegs || SR->ExplicitSubRegs.size() <= 1)
336c08df9e5SJakob Stoklund Olesen       continue;
337c08df9e5SJakob Stoklund Olesen 
338c08df9e5SJakob Stoklund Olesen     // SR is composed of multiple sub-regs. Find their names in this register.
339c08df9e5SJakob Stoklund Olesen     SmallVector<CodeGenSubRegIndex*, 8> Parts;
340c08df9e5SJakob Stoklund Olesen     for (unsigned j = 0, e = SR->ExplicitSubRegs.size(); j != e; ++j)
341c08df9e5SJakob Stoklund Olesen       Parts.push_back(getSubRegIndex(SR->ExplicitSubRegs[j]));
342c08df9e5SJakob Stoklund Olesen 
343c08df9e5SJakob Stoklund Olesen     // Offer this as an existing spelling for the concatenation of Parts.
344c08df9e5SJakob Stoklund Olesen     RegBank.addConcatSubRegIndex(Parts, ExplicitSubRegIndices[i]);
345c08df9e5SJakob Stoklund Olesen   }
346c08df9e5SJakob Stoklund Olesen 
347066fba1aSJakob Stoklund Olesen   // Initialize RegUnitList. Because getSubRegs is called recursively, this
348066fba1aSJakob Stoklund Olesen   // processes the register hierarchy in postorder.
3491a004ca0SAndrew Trick   //
350066fba1aSJakob Stoklund Olesen   // Inherit all sub-register units. It is good enough to look at the explicit
351066fba1aSJakob Stoklund Olesen   // sub-registers, the other registers won't contribute any more units.
352066fba1aSJakob Stoklund Olesen   for (unsigned i = 0, e = ExplicitSubRegs.size(); i != e; ++i) {
353066fba1aSJakob Stoklund Olesen     CodeGenRegister *SR = ExplicitSubRegs[i];
354066fba1aSJakob Stoklund Olesen     // Explicit sub-registers are usually disjoint, so this is a good way of
355066fba1aSJakob Stoklund Olesen     // computing the union. We may pick up a few duplicates that will be
356066fba1aSJakob Stoklund Olesen     // eliminated below.
357066fba1aSJakob Stoklund Olesen     unsigned N = RegUnits.size();
358066fba1aSJakob Stoklund Olesen     RegUnits.append(SR->RegUnits.begin(), SR->RegUnits.end());
359066fba1aSJakob Stoklund Olesen     std::inplace_merge(RegUnits.begin(), RegUnits.begin() + N, RegUnits.end());
360066fba1aSJakob Stoklund Olesen   }
361066fba1aSJakob Stoklund Olesen   RegUnits.erase(std::unique(RegUnits.begin(), RegUnits.end()), RegUnits.end());
362066fba1aSJakob Stoklund Olesen 
363066fba1aSJakob Stoklund Olesen   // Absent any ad hoc aliasing, we create one register unit per leaf register.
364066fba1aSJakob Stoklund Olesen   // These units correspond to the maximal cliques in the register overlap
365066fba1aSJakob Stoklund Olesen   // graph which is optimal.
366066fba1aSJakob Stoklund Olesen   //
367066fba1aSJakob Stoklund Olesen   // When there is ad hoc aliasing, we simply create one unit per edge in the
368066fba1aSJakob Stoklund Olesen   // undirected ad hoc aliasing graph. Technically, we could do better by
369066fba1aSJakob Stoklund Olesen   // identifying maximal cliques in the ad hoc graph, but cliques larger than 2
370066fba1aSJakob Stoklund Olesen   // are extremely rare anyway (I've never seen one), so we don't bother with
371066fba1aSJakob Stoklund Olesen   // the added complexity.
372066fba1aSJakob Stoklund Olesen   for (unsigned i = 0, e = ExplicitAliases.size(); i != e; ++i) {
373066fba1aSJakob Stoklund Olesen     CodeGenRegister *AR = ExplicitAliases[i];
374066fba1aSJakob Stoklund Olesen     // Only visit each edge once.
375066fba1aSJakob Stoklund Olesen     if (AR->SubRegsComplete)
376066fba1aSJakob Stoklund Olesen       continue;
377066fba1aSJakob Stoklund Olesen     // Create a RegUnit representing this alias edge, and add it to both
378066fba1aSJakob Stoklund Olesen     // registers.
379095f22afSJakob Stoklund Olesen     unsigned Unit = RegBank.newRegUnit(this, AR);
380066fba1aSJakob Stoklund Olesen     RegUnits.push_back(Unit);
381066fba1aSJakob Stoklund Olesen     AR->RegUnits.push_back(Unit);
382066fba1aSJakob Stoklund Olesen   }
383066fba1aSJakob Stoklund Olesen 
384066fba1aSJakob Stoklund Olesen   // Finally, create units for leaf registers without ad hoc aliases. Note that
385066fba1aSJakob Stoklund Olesen   // a leaf register with ad hoc aliases doesn't get its own unit - it isn't
386066fba1aSJakob Stoklund Olesen   // necessary. This means the aliasing leaf registers can share a single unit.
387066fba1aSJakob Stoklund Olesen   if (RegUnits.empty())
388095f22afSJakob Stoklund Olesen     RegUnits.push_back(RegBank.newRegUnit(this));
389066fba1aSJakob Stoklund Olesen 
3907f381bd2SJakob Stoklund Olesen   // We have now computed the native register units. More may be adopted later
3917f381bd2SJakob Stoklund Olesen   // for balancing purposes.
3927f381bd2SJakob Stoklund Olesen   NumNativeRegUnits = RegUnits.size();
3937f381bd2SJakob Stoklund Olesen 
39484bd44ebSJakob Stoklund Olesen   return SubRegs;
39584bd44ebSJakob Stoklund Olesen }
39684bd44ebSJakob Stoklund Olesen 
397c08df9e5SJakob Stoklund Olesen // In a register that is covered by its sub-registers, try to find redundant
398c08df9e5SJakob Stoklund Olesen // sub-registers. For example:
399c08df9e5SJakob Stoklund Olesen //
400c08df9e5SJakob Stoklund Olesen //   QQ0 = {Q0, Q1}
401c08df9e5SJakob Stoklund Olesen //   Q0 = {D0, D1}
402c08df9e5SJakob Stoklund Olesen //   Q1 = {D2, D3}
403c08df9e5SJakob Stoklund Olesen //
404c08df9e5SJakob Stoklund Olesen // We can infer that D1_D2 is also a sub-register, even if it wasn't named in
405c08df9e5SJakob Stoklund Olesen // the register definition.
406c08df9e5SJakob Stoklund Olesen //
407c08df9e5SJakob Stoklund Olesen // The explicitly specified registers form a tree. This function discovers
408c08df9e5SJakob Stoklund Olesen // sub-register relationships that would force a DAG.
409c08df9e5SJakob Stoklund Olesen //
410c08df9e5SJakob Stoklund Olesen void CodeGenRegister::computeSecondarySubRegs(CodeGenRegBank &RegBank) {
411c08df9e5SJakob Stoklund Olesen   // Collect new sub-registers first, add them later.
412c08df9e5SJakob Stoklund Olesen   SmallVector<SubRegMap::value_type, 8> NewSubRegs;
413c08df9e5SJakob Stoklund Olesen 
414c08df9e5SJakob Stoklund Olesen   // Look at the leading super-registers of each sub-register. Those are the
415c08df9e5SJakob Stoklund Olesen   // candidates for new sub-registers, assuming they are fully contained in
416c08df9e5SJakob Stoklund Olesen   // this register.
417c08df9e5SJakob Stoklund Olesen   for (SubRegMap::iterator I = SubRegs.begin(), E = SubRegs.end(); I != E; ++I){
418c08df9e5SJakob Stoklund Olesen     const CodeGenRegister *SubReg = I->second;
419c08df9e5SJakob Stoklund Olesen     const CodeGenRegister::SuperRegList &Leads = SubReg->LeadingSuperRegs;
420c08df9e5SJakob Stoklund Olesen     for (unsigned i = 0, e = Leads.size(); i != e; ++i) {
421c08df9e5SJakob Stoklund Olesen       CodeGenRegister *Cand = const_cast<CodeGenRegister*>(Leads[i]);
422c08df9e5SJakob Stoklund Olesen       // Already got this sub-register?
423c08df9e5SJakob Stoklund Olesen       if (Cand == this || getSubRegIndex(Cand))
424c08df9e5SJakob Stoklund Olesen         continue;
425c08df9e5SJakob Stoklund Olesen       // Check if each component of Cand is already a sub-register.
426c08df9e5SJakob Stoklund Olesen       // We know that the first component is I->second, and is present with the
427c08df9e5SJakob Stoklund Olesen       // name I->first.
428c08df9e5SJakob Stoklund Olesen       SmallVector<CodeGenSubRegIndex*, 8> Parts(1, I->first);
429c08df9e5SJakob Stoklund Olesen       assert(!Cand->ExplicitSubRegs.empty() &&
430c08df9e5SJakob Stoklund Olesen              "Super-register has no sub-registers");
431c08df9e5SJakob Stoklund Olesen       for (unsigned j = 1, e = Cand->ExplicitSubRegs.size(); j != e; ++j) {
432c08df9e5SJakob Stoklund Olesen         if (CodeGenSubRegIndex *Idx = getSubRegIndex(Cand->ExplicitSubRegs[j]))
433c08df9e5SJakob Stoklund Olesen           Parts.push_back(Idx);
434c08df9e5SJakob Stoklund Olesen         else {
435c08df9e5SJakob Stoklund Olesen           // Sub-register doesn't exist.
436c08df9e5SJakob Stoklund Olesen           Parts.clear();
437c08df9e5SJakob Stoklund Olesen           break;
438c08df9e5SJakob Stoklund Olesen         }
439c08df9e5SJakob Stoklund Olesen       }
440c08df9e5SJakob Stoklund Olesen       // If some Cand sub-register is not part of this register, or if Cand only
441c08df9e5SJakob Stoklund Olesen       // has one sub-register, there is nothing to do.
442c08df9e5SJakob Stoklund Olesen       if (Parts.size() <= 1)
443c08df9e5SJakob Stoklund Olesen         continue;
444c08df9e5SJakob Stoklund Olesen 
445c08df9e5SJakob Stoklund Olesen       // Each part of Cand is a sub-register of this. Make the full Cand also
446c08df9e5SJakob Stoklund Olesen       // a sub-register with a concatenated sub-register index.
447c08df9e5SJakob Stoklund Olesen       CodeGenSubRegIndex *Concat= RegBank.getConcatSubRegIndex(Parts);
448c08df9e5SJakob Stoklund Olesen       NewSubRegs.push_back(std::make_pair(Concat, Cand));
449c08df9e5SJakob Stoklund Olesen     }
450c08df9e5SJakob Stoklund Olesen   }
451c08df9e5SJakob Stoklund Olesen 
452c08df9e5SJakob Stoklund Olesen   // Now add all the new sub-registers.
453c08df9e5SJakob Stoklund Olesen   for (unsigned i = 0, e = NewSubRegs.size(); i != e; ++i) {
454c08df9e5SJakob Stoklund Olesen     // Don't add Cand if another sub-register is already using the index.
455c08df9e5SJakob Stoklund Olesen     if (!SubRegs.insert(NewSubRegs[i]).second)
456c08df9e5SJakob Stoklund Olesen       continue;
457c08df9e5SJakob Stoklund Olesen 
458c08df9e5SJakob Stoklund Olesen     CodeGenSubRegIndex *NewIdx = NewSubRegs[i].first;
459c08df9e5SJakob Stoklund Olesen     CodeGenRegister *NewSubReg = NewSubRegs[i].second;
460c08df9e5SJakob Stoklund Olesen     SubReg2Idx.insert(std::make_pair(NewSubReg, NewIdx));
461c08df9e5SJakob Stoklund Olesen   }
462c08df9e5SJakob Stoklund Olesen 
463c08df9e5SJakob Stoklund Olesen   // Create sub-register index composition maps for the synthesized indices.
464c08df9e5SJakob Stoklund Olesen   for (unsigned i = 0, e = NewSubRegs.size(); i != e; ++i) {
465c08df9e5SJakob Stoklund Olesen     CodeGenSubRegIndex *NewIdx = NewSubRegs[i].first;
466c08df9e5SJakob Stoklund Olesen     CodeGenRegister *NewSubReg = NewSubRegs[i].second;
467c08df9e5SJakob Stoklund Olesen     for (SubRegMap::const_iterator SI = NewSubReg->SubRegs.begin(),
468c08df9e5SJakob Stoklund Olesen            SE = NewSubReg->SubRegs.end(); SI != SE; ++SI) {
469c08df9e5SJakob Stoklund Olesen       CodeGenSubRegIndex *SubIdx = getSubRegIndex(SI->second);
470c08df9e5SJakob Stoklund Olesen       if (!SubIdx)
471635debe8SJoerg Sonnenberger         PrintFatalError(TheDef->getLoc(), "No SubRegIndex for " +
472c08df9e5SJakob Stoklund Olesen                         SI->second->getName() + " in " + getName());
473c08df9e5SJakob Stoklund Olesen       NewIdx->addComposite(SI->first, SubIdx);
474c08df9e5SJakob Stoklund Olesen     }
475c08df9e5SJakob Stoklund Olesen   }
476c08df9e5SJakob Stoklund Olesen }
477c08df9e5SJakob Stoklund Olesen 
47850ecd0ffSJakob Stoklund Olesen void CodeGenRegister::computeSuperRegs(CodeGenRegBank &RegBank) {
4793f3eb180SJakob Stoklund Olesen   // Only visit each register once.
4803f3eb180SJakob Stoklund Olesen   if (SuperRegsComplete)
4813f3eb180SJakob Stoklund Olesen     return;
4823f3eb180SJakob Stoklund Olesen   SuperRegsComplete = true;
4833f3eb180SJakob Stoklund Olesen 
4843f3eb180SJakob Stoklund Olesen   // Make sure all sub-registers have been visited first, so the super-reg
4853f3eb180SJakob Stoklund Olesen   // lists will be topologically ordered.
4863f3eb180SJakob Stoklund Olesen   for (SubRegMap::const_iterator I = SubRegs.begin(), E = SubRegs.end();
4873f3eb180SJakob Stoklund Olesen        I != E; ++I)
48850ecd0ffSJakob Stoklund Olesen     I->second->computeSuperRegs(RegBank);
4893f3eb180SJakob Stoklund Olesen 
4903f3eb180SJakob Stoklund Olesen   // Now add this as a super-register on all sub-registers.
49150ecd0ffSJakob Stoklund Olesen   // Also compute the TopoSigId in post-order.
49250ecd0ffSJakob Stoklund Olesen   TopoSigId Id;
4933f3eb180SJakob Stoklund Olesen   for (SubRegMap::const_iterator I = SubRegs.begin(), E = SubRegs.end();
4943f3eb180SJakob Stoklund Olesen        I != E; ++I) {
49550ecd0ffSJakob Stoklund Olesen     // Topological signature computed from SubIdx, TopoId(SubReg).
49650ecd0ffSJakob Stoklund Olesen     // Loops and idempotent indices have TopoSig = ~0u.
49750ecd0ffSJakob Stoklund Olesen     Id.push_back(I->first->EnumValue);
49850ecd0ffSJakob Stoklund Olesen     Id.push_back(I->second->TopoSig);
49950ecd0ffSJakob Stoklund Olesen 
5003f3eb180SJakob Stoklund Olesen     // Don't add duplicate entries.
5013f3eb180SJakob Stoklund Olesen     if (!I->second->SuperRegs.empty() && I->second->SuperRegs.back() == this)
5023f3eb180SJakob Stoklund Olesen       continue;
5033f3eb180SJakob Stoklund Olesen     I->second->SuperRegs.push_back(this);
5043f3eb180SJakob Stoklund Olesen   }
50550ecd0ffSJakob Stoklund Olesen   TopoSig = RegBank.getTopoSig(Id);
5063f3eb180SJakob Stoklund Olesen }
5073f3eb180SJakob Stoklund Olesen 
508d2b4713eSJakob Stoklund Olesen void
50900296815SJakob Stoklund Olesen CodeGenRegister::addSubRegsPreOrder(SetVector<const CodeGenRegister*> &OSet,
510f1bb1519SJakob Stoklund Olesen                                     CodeGenRegBank &RegBank) const {
511d2b4713eSJakob Stoklund Olesen   assert(SubRegsComplete && "Must precompute sub-registers");
512c1e9087fSJakob Stoklund Olesen   for (unsigned i = 0, e = ExplicitSubRegs.size(); i != e; ++i) {
513c1e9087fSJakob Stoklund Olesen     CodeGenRegister *SR = ExplicitSubRegs[i];
514d2b4713eSJakob Stoklund Olesen     if (OSet.insert(SR))
515f1bb1519SJakob Stoklund Olesen       SR->addSubRegsPreOrder(OSet, RegBank);
516d2b4713eSJakob Stoklund Olesen   }
517c08df9e5SJakob Stoklund Olesen   // Add any secondary sub-registers that weren't part of the explicit tree.
518c08df9e5SJakob Stoklund Olesen   for (SubRegMap::const_iterator I = SubRegs.begin(), E = SubRegs.end();
519c08df9e5SJakob Stoklund Olesen        I != E; ++I)
520c08df9e5SJakob Stoklund Olesen     OSet.insert(I->second);
521d2b4713eSJakob Stoklund Olesen }
522d2b4713eSJakob Stoklund Olesen 
5236d808257SJakob Stoklund Olesen // Compute overlapping registers.
5246d808257SJakob Stoklund Olesen //
5256d808257SJakob Stoklund Olesen // The standard set is all super-registers and all sub-registers, but the
5266d808257SJakob Stoklund Olesen // target description can add arbitrary overlapping registers via the 'Aliases'
5276d808257SJakob Stoklund Olesen // field. This complicates things, but we can compute overlapping sets using
5286d808257SJakob Stoklund Olesen // the following rules:
5296d808257SJakob Stoklund Olesen //
5306d808257SJakob Stoklund Olesen // 1. The relation overlap(A, B) is reflexive and symmetric but not transitive.
5316d808257SJakob Stoklund Olesen //
5326d808257SJakob Stoklund Olesen // 2. overlap(A, B) implies overlap(A, S) for all S in supers(B).
5336d808257SJakob Stoklund Olesen //
5346d808257SJakob Stoklund Olesen // Alternatively:
5356d808257SJakob Stoklund Olesen //
53691ce36c9SSylvestre Ledru //    overlap(A, B) iff there exists:
5376d808257SJakob Stoklund Olesen //    A' in { A, subregs(A) } and B' in { B, subregs(B) } such that:
5386d808257SJakob Stoklund Olesen //    A' = B' or A' in aliases(B') or B' in aliases(A').
5396d808257SJakob Stoklund Olesen //
5406d808257SJakob Stoklund Olesen // Here subregs(A) is the full flattened sub-register set returned by
5416d808257SJakob Stoklund Olesen // A.getSubRegs() while aliases(A) is simply the special 'Aliases' field in the
5426d808257SJakob Stoklund Olesen // description of register A.
5436d808257SJakob Stoklund Olesen //
5446d808257SJakob Stoklund Olesen // This also implies that registers with a common sub-register are considered
5456d808257SJakob Stoklund Olesen // overlapping. This can happen when forming register pairs:
5466d808257SJakob Stoklund Olesen //
5476d808257SJakob Stoklund Olesen //    P0 = (R0, R1)
5486d808257SJakob Stoklund Olesen //    P1 = (R1, R2)
5496d808257SJakob Stoklund Olesen //    P2 = (R2, R3)
5506d808257SJakob Stoklund Olesen //
5516d808257SJakob Stoklund Olesen // In this case, we will infer an overlap between P0 and P1 because of the
5526d808257SJakob Stoklund Olesen // shared sub-register R1. There is no overlap between P0 and P2.
5536d808257SJakob Stoklund Olesen //
5546d808257SJakob Stoklund Olesen void CodeGenRegister::computeOverlaps(CodeGenRegister::Set &Overlaps,
5556d808257SJakob Stoklund Olesen                                       const CodeGenRegBank &RegBank) const {
5566d808257SJakob Stoklund Olesen   assert(!RegUnits.empty() && "Compute register units before overlaps.");
5576d808257SJakob Stoklund Olesen 
5586d808257SJakob Stoklund Olesen   // Register units are assigned such that the overlapping registers are the
5596d808257SJakob Stoklund Olesen   // super-registers of the root registers of the register units.
5606d808257SJakob Stoklund Olesen   for (unsigned rui = 0, rue = RegUnits.size(); rui != rue; ++rui) {
5616d808257SJakob Stoklund Olesen     const RegUnit &RU = RegBank.getRegUnit(RegUnits[rui]);
5626d808257SJakob Stoklund Olesen     ArrayRef<const CodeGenRegister*> Roots = RU.getRoots();
5636d808257SJakob Stoklund Olesen     for (unsigned ri = 0, re = Roots.size(); ri != re; ++ri) {
5646d808257SJakob Stoklund Olesen       const CodeGenRegister *Root = Roots[ri];
5656d808257SJakob Stoklund Olesen       Overlaps.insert(Root);
5666d808257SJakob Stoklund Olesen       ArrayRef<const CodeGenRegister*> Supers = Root->getSuperRegs();
5676d808257SJakob Stoklund Olesen       Overlaps.insert(Supers.begin(), Supers.end());
5686d808257SJakob Stoklund Olesen     }
5696d808257SJakob Stoklund Olesen   }
5706d808257SJakob Stoklund Olesen }
5716d808257SJakob Stoklund Olesen 
5721d7a2c57SAndrew Trick // Get the sum of this register's unit weights.
5731d7a2c57SAndrew Trick unsigned CodeGenRegister::getWeight(const CodeGenRegBank &RegBank) const {
5741d7a2c57SAndrew Trick   unsigned Weight = 0;
5751d7a2c57SAndrew Trick   for (RegUnitList::const_iterator I = RegUnits.begin(), E = RegUnits.end();
5761d7a2c57SAndrew Trick        I != E; ++I) {
577095f22afSJakob Stoklund Olesen     Weight += RegBank.getRegUnit(*I).Weight;
5781d7a2c57SAndrew Trick   }
5791d7a2c57SAndrew Trick   return Weight;
5801d7a2c57SAndrew Trick }
5811d7a2c57SAndrew Trick 
58268d6d8abSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
5833bd1b65eSJakob Stoklund Olesen //                               RegisterTuples
5843bd1b65eSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
5853bd1b65eSJakob Stoklund Olesen 
5863bd1b65eSJakob Stoklund Olesen // A RegisterTuples def is used to generate pseudo-registers from lists of
5873bd1b65eSJakob Stoklund Olesen // sub-registers. We provide a SetTheory expander class that returns the new
5883bd1b65eSJakob Stoklund Olesen // registers.
5893bd1b65eSJakob Stoklund Olesen namespace {
5903bd1b65eSJakob Stoklund Olesen struct TupleExpander : SetTheory::Expander {
5913bd1b65eSJakob Stoklund Olesen   void expand(SetTheory &ST, Record *Def, SetTheory::RecSet &Elts) {
5923bd1b65eSJakob Stoklund Olesen     std::vector<Record*> Indices = Def->getValueAsListOfDefs("SubRegIndices");
5933bd1b65eSJakob Stoklund Olesen     unsigned Dim = Indices.size();
594af8ee2cdSDavid Greene     ListInit *SubRegs = Def->getValueAsListInit("SubRegs");
5953bd1b65eSJakob Stoklund Olesen     if (Dim != SubRegs->getSize())
596635debe8SJoerg Sonnenberger       PrintFatalError(Def->getLoc(), "SubRegIndices and SubRegs size mismatch");
5973bd1b65eSJakob Stoklund Olesen     if (Dim < 2)
598635debe8SJoerg Sonnenberger       PrintFatalError(Def->getLoc(),
599635debe8SJoerg Sonnenberger                       "Tuples must have at least 2 sub-registers");
6003bd1b65eSJakob Stoklund Olesen 
6013bd1b65eSJakob Stoklund Olesen     // Evaluate the sub-register lists to be zipped.
6023bd1b65eSJakob Stoklund Olesen     unsigned Length = ~0u;
6033bd1b65eSJakob Stoklund Olesen     SmallVector<SetTheory::RecSet, 4> Lists(Dim);
6043bd1b65eSJakob Stoklund Olesen     for (unsigned i = 0; i != Dim; ++i) {
60570909373SJoerg Sonnenberger       ST.evaluate(SubRegs->getElement(i), Lists[i], Def->getLoc());
6063bd1b65eSJakob Stoklund Olesen       Length = std::min(Length, unsigned(Lists[i].size()));
6073bd1b65eSJakob Stoklund Olesen     }
6083bd1b65eSJakob Stoklund Olesen 
6093bd1b65eSJakob Stoklund Olesen     if (Length == 0)
6103bd1b65eSJakob Stoklund Olesen       return;
6113bd1b65eSJakob Stoklund Olesen 
6123bd1b65eSJakob Stoklund Olesen     // Precompute some types.
6133bd1b65eSJakob Stoklund Olesen     Record *RegisterCl = Def->getRecords().getClass("Register");
614abcfdceaSJakob Stoklund Olesen     RecTy *RegisterRecTy = RecordRecTy::get(RegisterCl);
615af8ee2cdSDavid Greene     StringInit *BlankName = StringInit::get("");
6163bd1b65eSJakob Stoklund Olesen 
6173bd1b65eSJakob Stoklund Olesen     // Zip them up.
6183bd1b65eSJakob Stoklund Olesen     for (unsigned n = 0; n != Length; ++n) {
6193bd1b65eSJakob Stoklund Olesen       std::string Name;
6203bd1b65eSJakob Stoklund Olesen       Record *Proto = Lists[0][n];
621af8ee2cdSDavid Greene       std::vector<Init*> Tuple;
6223bd1b65eSJakob Stoklund Olesen       unsigned CostPerUse = 0;
6233bd1b65eSJakob Stoklund Olesen       for (unsigned i = 0; i != Dim; ++i) {
6243bd1b65eSJakob Stoklund Olesen         Record *Reg = Lists[i][n];
6253bd1b65eSJakob Stoklund Olesen         if (i) Name += '_';
6263bd1b65eSJakob Stoklund Olesen         Name += Reg->getName();
627abcfdceaSJakob Stoklund Olesen         Tuple.push_back(DefInit::get(Reg));
6283bd1b65eSJakob Stoklund Olesen         CostPerUse = std::max(CostPerUse,
6293bd1b65eSJakob Stoklund Olesen                               unsigned(Reg->getValueAsInt("CostPerUse")));
6303bd1b65eSJakob Stoklund Olesen       }
6313bd1b65eSJakob Stoklund Olesen 
6323bd1b65eSJakob Stoklund Olesen       // Create a new Record representing the synthesized register. This record
6333bd1b65eSJakob Stoklund Olesen       // is only for consumption by CodeGenRegister, it is not added to the
6343bd1b65eSJakob Stoklund Olesen       // RecordKeeper.
6353bd1b65eSJakob Stoklund Olesen       Record *NewReg = new Record(Name, Def->getLoc(), Def->getRecords());
6363bd1b65eSJakob Stoklund Olesen       Elts.insert(NewReg);
6373bd1b65eSJakob Stoklund Olesen 
6383bd1b65eSJakob Stoklund Olesen       // Copy Proto super-classes.
639f12e8a93SJordan Rose       ArrayRef<Record *> Supers = Proto->getSuperClasses();
640f12e8a93SJordan Rose       ArrayRef<SMRange> Ranges = Proto->getSuperClassRanges();
641f12e8a93SJordan Rose       for (unsigned i = 0, e = Supers.size(); i != e; ++i)
642f12e8a93SJordan Rose         NewReg->addSuperClass(Supers[i], Ranges[i]);
6433bd1b65eSJakob Stoklund Olesen 
6443bd1b65eSJakob Stoklund Olesen       // Copy Proto fields.
6453bd1b65eSJakob Stoklund Olesen       for (unsigned i = 0, e = Proto->getValues().size(); i != e; ++i) {
6463bd1b65eSJakob Stoklund Olesen         RecordVal RV = Proto->getValues()[i];
6473bd1b65eSJakob Stoklund Olesen 
648f43b5995SJakob Stoklund Olesen         // Skip existing fields, like NAME.
649f43b5995SJakob Stoklund Olesen         if (NewReg->getValue(RV.getNameInit()))
650071c69cdSJakob Stoklund Olesen           continue;
651071c69cdSJakob Stoklund Olesen 
652f43b5995SJakob Stoklund Olesen         StringRef Field = RV.getName();
653f43b5995SJakob Stoklund Olesen 
6543bd1b65eSJakob Stoklund Olesen         // Replace the sub-register list with Tuple.
655f43b5995SJakob Stoklund Olesen         if (Field == "SubRegs")
656e32ebf22SDavid Greene           RV.setValue(ListInit::get(Tuple, RegisterRecTy));
6573bd1b65eSJakob Stoklund Olesen 
6583bd1b65eSJakob Stoklund Olesen         // Provide a blank AsmName. MC hacks are required anyway.
659f43b5995SJakob Stoklund Olesen         if (Field == "AsmName")
6603bd1b65eSJakob Stoklund Olesen           RV.setValue(BlankName);
6613bd1b65eSJakob Stoklund Olesen 
6623bd1b65eSJakob Stoklund Olesen         // CostPerUse is aggregated from all Tuple members.
663f43b5995SJakob Stoklund Olesen         if (Field == "CostPerUse")
664e32ebf22SDavid Greene           RV.setValue(IntInit::get(CostPerUse));
6653bd1b65eSJakob Stoklund Olesen 
666f43b5995SJakob Stoklund Olesen         // Composite registers are always covered by sub-registers.
667f43b5995SJakob Stoklund Olesen         if (Field == "CoveredBySubRegs")
668f43b5995SJakob Stoklund Olesen           RV.setValue(BitInit::get(true));
669f43b5995SJakob Stoklund Olesen 
6703bd1b65eSJakob Stoklund Olesen         // Copy fields from the RegisterTuples def.
671f43b5995SJakob Stoklund Olesen         if (Field == "SubRegIndices" ||
672f43b5995SJakob Stoklund Olesen             Field == "CompositeIndices") {
673f43b5995SJakob Stoklund Olesen           NewReg->addValue(*Def->getValue(Field));
6743bd1b65eSJakob Stoklund Olesen           continue;
6753bd1b65eSJakob Stoklund Olesen         }
6763bd1b65eSJakob Stoklund Olesen 
6773bd1b65eSJakob Stoklund Olesen         // Some fields get their default uninitialized value.
678f43b5995SJakob Stoklund Olesen         if (Field == "DwarfNumbers" ||
679f43b5995SJakob Stoklund Olesen             Field == "DwarfAlias" ||
680f43b5995SJakob Stoklund Olesen             Field == "Aliases") {
681f43b5995SJakob Stoklund Olesen           if (const RecordVal *DefRV = RegisterCl->getValue(Field))
682d9149a45SJakob Stoklund Olesen             NewReg->addValue(*DefRV);
6833bd1b65eSJakob Stoklund Olesen           continue;
6843bd1b65eSJakob Stoklund Olesen         }
6853bd1b65eSJakob Stoklund Olesen 
6863bd1b65eSJakob Stoklund Olesen         // Everything else is copied from Proto.
6873bd1b65eSJakob Stoklund Olesen         NewReg->addValue(RV);
6883bd1b65eSJakob Stoklund Olesen       }
6893bd1b65eSJakob Stoklund Olesen     }
6903bd1b65eSJakob Stoklund Olesen   }
6913bd1b65eSJakob Stoklund Olesen };
6923bd1b65eSJakob Stoklund Olesen }
6933bd1b65eSJakob Stoklund Olesen 
6943bd1b65eSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
69568d6d8abSJakob Stoklund Olesen //                            CodeGenRegisterClass
69668d6d8abSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
69768d6d8abSJakob Stoklund Olesen 
698d7bc5c26SJakob Stoklund Olesen CodeGenRegisterClass::CodeGenRegisterClass(CodeGenRegBank &RegBank, Record *R)
69950ecd0ffSJakob Stoklund Olesen   : TheDef(R),
70050ecd0ffSJakob Stoklund Olesen     Name(R->getName()),
70150ecd0ffSJakob Stoklund Olesen     TopoSigs(RegBank.getNumTopoSigs()),
70250ecd0ffSJakob Stoklund Olesen     EnumValue(-1) {
70368d6d8abSJakob Stoklund Olesen   // Rename anonymous register classes.
70468d6d8abSJakob Stoklund Olesen   if (R->getName().size() > 9 && R->getName()[9] == '.') {
70568d6d8abSJakob Stoklund Olesen     static unsigned AnonCounter = 0;
706*2bb7db9dSMichael J. Spencer     R->setName("AnonRegClass_" + utostr(AnonCounter));
707*2bb7db9dSMichael J. Spencer     // MSVC2012 ICEs if AnonCounter++ is directly passed to utostr.
708*2bb7db9dSMichael J. Spencer     ++AnonCounter;
70968d6d8abSJakob Stoklund Olesen   }
71068d6d8abSJakob Stoklund Olesen 
71168d6d8abSJakob Stoklund Olesen   std::vector<Record*> TypeList = R->getValueAsListOfDefs("RegTypes");
71268d6d8abSJakob Stoklund Olesen   for (unsigned i = 0, e = TypeList.size(); i != e; ++i) {
71368d6d8abSJakob Stoklund Olesen     Record *Type = TypeList[i];
71468d6d8abSJakob Stoklund Olesen     if (!Type->isSubClassOf("ValueType"))
715635debe8SJoerg Sonnenberger       PrintFatalError("RegTypes list member '" + Type->getName() +
716635debe8SJoerg Sonnenberger         "' does not derive from the ValueType class!");
71768d6d8abSJakob Stoklund Olesen     VTs.push_back(getValueType(Type));
71868d6d8abSJakob Stoklund Olesen   }
71968d6d8abSJakob Stoklund Olesen   assert(!VTs.empty() && "RegisterClass must contain at least one ValueType!");
72068d6d8abSJakob Stoklund Olesen 
721331534e5SJakob Stoklund Olesen   // Allocation order 0 is the full set. AltOrders provides others.
722331534e5SJakob Stoklund Olesen   const SetTheory::RecVec *Elements = RegBank.getSets().expand(R);
723331534e5SJakob Stoklund Olesen   ListInit *AltOrders = R->getValueAsListInit("AltOrders");
724331534e5SJakob Stoklund Olesen   Orders.resize(1 + AltOrders->size());
725331534e5SJakob Stoklund Olesen 
72635cea3daSJakob Stoklund Olesen   // Default allocation order always contains all registers.
727331534e5SJakob Stoklund Olesen   for (unsigned i = 0, e = Elements->size(); i != e; ++i) {
728331534e5SJakob Stoklund Olesen     Orders[0].push_back((*Elements)[i]);
72950ecd0ffSJakob Stoklund Olesen     const CodeGenRegister *Reg = RegBank.getReg((*Elements)[i]);
73050ecd0ffSJakob Stoklund Olesen     Members.insert(Reg);
73150ecd0ffSJakob Stoklund Olesen     TopoSigs.set(Reg->getTopoSig());
732331534e5SJakob Stoklund Olesen   }
73368d6d8abSJakob Stoklund Olesen 
73435cea3daSJakob Stoklund Olesen   // Alternative allocation orders may be subsets.
73535cea3daSJakob Stoklund Olesen   SetTheory::RecSet Order;
736331534e5SJakob Stoklund Olesen   for (unsigned i = 0, e = AltOrders->size(); i != e; ++i) {
73770909373SJoerg Sonnenberger     RegBank.getSets().evaluate(AltOrders->getElement(i), Order, R->getLoc());
738331534e5SJakob Stoklund Olesen     Orders[1 + i].append(Order.begin(), Order.end());
73935cea3daSJakob Stoklund Olesen     // Verify that all altorder members are regclass members.
74035cea3daSJakob Stoklund Olesen     while (!Order.empty()) {
74135cea3daSJakob Stoklund Olesen       CodeGenRegister *Reg = RegBank.getReg(Order.back());
74235cea3daSJakob Stoklund Olesen       Order.pop_back();
74335cea3daSJakob Stoklund Olesen       if (!contains(Reg))
744635debe8SJoerg Sonnenberger         PrintFatalError(R->getLoc(), " AltOrder register " + Reg->getName() +
74535cea3daSJakob Stoklund Olesen                       " is not a class member");
74635cea3daSJakob Stoklund Olesen     }
74735cea3daSJakob Stoklund Olesen   }
74835cea3daSJakob Stoklund Olesen 
74968d6d8abSJakob Stoklund Olesen   // Allow targets to override the size in bits of the RegisterClass.
75068d6d8abSJakob Stoklund Olesen   unsigned Size = R->getValueAsInt("Size");
75168d6d8abSJakob Stoklund Olesen 
75268d6d8abSJakob Stoklund Olesen   Namespace = R->getValueAsString("Namespace");
75368d6d8abSJakob Stoklund Olesen   SpillSize = Size ? Size : EVT(VTs[0]).getSizeInBits();
75468d6d8abSJakob Stoklund Olesen   SpillAlignment = R->getValueAsInt("Alignment");
75568d6d8abSJakob Stoklund Olesen   CopyCost = R->getValueAsInt("CopyCost");
75668d6d8abSJakob Stoklund Olesen   Allocatable = R->getValueAsBit("isAllocatable");
757dd8fbf57SJakob Stoklund Olesen   AltOrderSelect = R->getValueAsString("AltOrderSelect");
75868d6d8abSJakob Stoklund Olesen }
75968d6d8abSJakob Stoklund Olesen 
76003efe84dSJakob Stoklund Olesen // Create an inferred register class that was missing from the .td files.
76103efe84dSJakob Stoklund Olesen // Most properties will be inherited from the closest super-class after the
76203efe84dSJakob Stoklund Olesen // class structure has been computed.
763eebd5bc6SJakob Stoklund Olesen CodeGenRegisterClass::CodeGenRegisterClass(CodeGenRegBank &RegBank,
764eebd5bc6SJakob Stoklund Olesen                                            StringRef Name, Key Props)
76503efe84dSJakob Stoklund Olesen   : Members(*Props.Members),
76603efe84dSJakob Stoklund Olesen     TheDef(0),
76703efe84dSJakob Stoklund Olesen     Name(Name),
768eebd5bc6SJakob Stoklund Olesen     TopoSigs(RegBank.getNumTopoSigs()),
76903efe84dSJakob Stoklund Olesen     EnumValue(-1),
77003efe84dSJakob Stoklund Olesen     SpillSize(Props.SpillSize),
77103efe84dSJakob Stoklund Olesen     SpillAlignment(Props.SpillAlignment),
77203efe84dSJakob Stoklund Olesen     CopyCost(0),
77303efe84dSJakob Stoklund Olesen     Allocatable(true) {
774eebd5bc6SJakob Stoklund Olesen   for (CodeGenRegister::Set::iterator I = Members.begin(), E = Members.end();
775eebd5bc6SJakob Stoklund Olesen        I != E; ++I)
776eebd5bc6SJakob Stoklund Olesen     TopoSigs.set((*I)->getTopoSig());
77703efe84dSJakob Stoklund Olesen }
77803efe84dSJakob Stoklund Olesen 
77903efe84dSJakob Stoklund Olesen // Compute inherited propertied for a synthesized register class.
78003efe84dSJakob Stoklund Olesen void CodeGenRegisterClass::inheritProperties(CodeGenRegBank &RegBank) {
78103efe84dSJakob Stoklund Olesen   assert(!getDef() && "Only synthesized classes can inherit properties");
78203efe84dSJakob Stoklund Olesen   assert(!SuperClasses.empty() && "Synthesized class without super class");
78303efe84dSJakob Stoklund Olesen 
78403efe84dSJakob Stoklund Olesen   // The last super-class is the smallest one.
78503efe84dSJakob Stoklund Olesen   CodeGenRegisterClass &Super = *SuperClasses.back();
78603efe84dSJakob Stoklund Olesen 
78703efe84dSJakob Stoklund Olesen   // Most properties are copied directly.
78803efe84dSJakob Stoklund Olesen   // Exceptions are members, size, and alignment
78903efe84dSJakob Stoklund Olesen   Namespace = Super.Namespace;
79003efe84dSJakob Stoklund Olesen   VTs = Super.VTs;
79103efe84dSJakob Stoklund Olesen   CopyCost = Super.CopyCost;
79203efe84dSJakob Stoklund Olesen   Allocatable = Super.Allocatable;
79303efe84dSJakob Stoklund Olesen   AltOrderSelect = Super.AltOrderSelect;
79403efe84dSJakob Stoklund Olesen 
79503efe84dSJakob Stoklund Olesen   // Copy all allocation orders, filter out foreign registers from the larger
79603efe84dSJakob Stoklund Olesen   // super-class.
79703efe84dSJakob Stoklund Olesen   Orders.resize(Super.Orders.size());
79803efe84dSJakob Stoklund Olesen   for (unsigned i = 0, ie = Super.Orders.size(); i != ie; ++i)
79903efe84dSJakob Stoklund Olesen     for (unsigned j = 0, je = Super.Orders[i].size(); j != je; ++j)
80003efe84dSJakob Stoklund Olesen       if (contains(RegBank.getReg(Super.Orders[i][j])))
80103efe84dSJakob Stoklund Olesen         Orders[i].push_back(Super.Orders[i][j]);
80203efe84dSJakob Stoklund Olesen }
80303efe84dSJakob Stoklund Olesen 
804d7bc5c26SJakob Stoklund Olesen bool CodeGenRegisterClass::contains(const CodeGenRegister *Reg) const {
805d7bc5c26SJakob Stoklund Olesen   return Members.count(Reg);
806d7bc5c26SJakob Stoklund Olesen }
807d7bc5c26SJakob Stoklund Olesen 
80803efe84dSJakob Stoklund Olesen namespace llvm {
80903efe84dSJakob Stoklund Olesen   raw_ostream &operator<<(raw_ostream &OS, const CodeGenRegisterClass::Key &K) {
81003efe84dSJakob Stoklund Olesen     OS << "{ S=" << K.SpillSize << ", A=" << K.SpillAlignment;
81103efe84dSJakob Stoklund Olesen     for (CodeGenRegister::Set::const_iterator I = K.Members->begin(),
81203efe84dSJakob Stoklund Olesen          E = K.Members->end(); I != E; ++I)
81303efe84dSJakob Stoklund Olesen       OS << ", " << (*I)->getName();
81403efe84dSJakob Stoklund Olesen     return OS << " }";
81503efe84dSJakob Stoklund Olesen   }
81603efe84dSJakob Stoklund Olesen }
81703efe84dSJakob Stoklund Olesen 
81803efe84dSJakob Stoklund Olesen // This is a simple lexicographical order that can be used to search for sets.
81903efe84dSJakob Stoklund Olesen // It is not the same as the topological order provided by TopoOrderRC.
82003efe84dSJakob Stoklund Olesen bool CodeGenRegisterClass::Key::
82103efe84dSJakob Stoklund Olesen operator<(const CodeGenRegisterClass::Key &B) const {
82203efe84dSJakob Stoklund Olesen   assert(Members && B.Members);
82303efe84dSJakob Stoklund Olesen   if (*Members != *B.Members)
82403efe84dSJakob Stoklund Olesen     return *Members < *B.Members;
82503efe84dSJakob Stoklund Olesen   if (SpillSize != B.SpillSize)
82603efe84dSJakob Stoklund Olesen     return SpillSize < B.SpillSize;
82703efe84dSJakob Stoklund Olesen   return SpillAlignment < B.SpillAlignment;
82803efe84dSJakob Stoklund Olesen }
82903efe84dSJakob Stoklund Olesen 
830d7bc5c26SJakob Stoklund Olesen // Returns true if RC is a strict subclass.
831d7bc5c26SJakob Stoklund Olesen // RC is a sub-class of this class if it is a valid replacement for any
832d7bc5c26SJakob Stoklund Olesen // instruction operand where a register of this classis required. It must
833d7bc5c26SJakob Stoklund Olesen // satisfy these conditions:
834d7bc5c26SJakob Stoklund Olesen //
835d7bc5c26SJakob Stoklund Olesen // 1. All RC registers are also in this.
836d7bc5c26SJakob Stoklund Olesen // 2. The RC spill size must not be smaller than our spill size.
837d7bc5c26SJakob Stoklund Olesen // 3. RC spill alignment must be compatible with ours.
838d7bc5c26SJakob Stoklund Olesen //
8396417395dSJakob Stoklund Olesen static bool testSubClass(const CodeGenRegisterClass *A,
8406417395dSJakob Stoklund Olesen                          const CodeGenRegisterClass *B) {
8416417395dSJakob Stoklund Olesen   return A->SpillAlignment && B->SpillAlignment % A->SpillAlignment == 0 &&
8426417395dSJakob Stoklund Olesen     A->SpillSize <= B->SpillSize &&
8436417395dSJakob Stoklund Olesen     std::includes(A->getMembers().begin(), A->getMembers().end(),
8446417395dSJakob Stoklund Olesen                   B->getMembers().begin(), B->getMembers().end(),
845afef4222SJakob Stoklund Olesen                   CodeGenRegister::Less());
846d7bc5c26SJakob Stoklund Olesen }
847d7bc5c26SJakob Stoklund Olesen 
848c0fc173dSJakob Stoklund Olesen /// Sorting predicate for register classes.  This provides a topological
849c0fc173dSJakob Stoklund Olesen /// ordering that arranges all register classes before their sub-classes.
850c0fc173dSJakob Stoklund Olesen ///
851c0fc173dSJakob Stoklund Olesen /// Register classes with the same registers, spill size, and alignment form a
852c0fc173dSJakob Stoklund Olesen /// clique.  They will be ordered alphabetically.
853c0fc173dSJakob Stoklund Olesen ///
854c0fc173dSJakob Stoklund Olesen static int TopoOrderRC(const void *PA, const void *PB) {
855c0fc173dSJakob Stoklund Olesen   const CodeGenRegisterClass *A = *(const CodeGenRegisterClass* const*)PA;
856c0fc173dSJakob Stoklund Olesen   const CodeGenRegisterClass *B = *(const CodeGenRegisterClass* const*)PB;
857c0fc173dSJakob Stoklund Olesen   if (A == B)
858c0fc173dSJakob Stoklund Olesen     return 0;
859c0fc173dSJakob Stoklund Olesen 
860c0fc173dSJakob Stoklund Olesen   // Order by ascending spill size.
861c0fc173dSJakob Stoklund Olesen   if (A->SpillSize < B->SpillSize)
862c0fc173dSJakob Stoklund Olesen     return -1;
863c0fc173dSJakob Stoklund Olesen   if (A->SpillSize > B->SpillSize)
864c0fc173dSJakob Stoklund Olesen     return 1;
865c0fc173dSJakob Stoklund Olesen 
866c0fc173dSJakob Stoklund Olesen   // Order by ascending spill alignment.
867c0fc173dSJakob Stoklund Olesen   if (A->SpillAlignment < B->SpillAlignment)
868c0fc173dSJakob Stoklund Olesen     return -1;
869c0fc173dSJakob Stoklund Olesen   if (A->SpillAlignment > B->SpillAlignment)
870c0fc173dSJakob Stoklund Olesen     return 1;
871c0fc173dSJakob Stoklund Olesen 
8724fd600b6SJakob Stoklund Olesen   // Order by descending set size.  Note that the classes' allocation order may
8734fd600b6SJakob Stoklund Olesen   // not have been computed yet.  The Members set is always vaild.
8744fd600b6SJakob Stoklund Olesen   if (A->getMembers().size() > B->getMembers().size())
8754fd600b6SJakob Stoklund Olesen     return -1;
8764fd600b6SJakob Stoklund Olesen   if (A->getMembers().size() < B->getMembers().size())
8774fd600b6SJakob Stoklund Olesen     return 1;
8784fd600b6SJakob Stoklund Olesen 
879c0fc173dSJakob Stoklund Olesen   // Finally order by name as a tie breaker.
880fff0dfd8SJakob Stoklund Olesen   return StringRef(A->getName()).compare(B->getName());
881c0fc173dSJakob Stoklund Olesen }
882c0fc173dSJakob Stoklund Olesen 
883bd92dc60SJakob Stoklund Olesen std::string CodeGenRegisterClass::getQualifiedName() const {
884bd92dc60SJakob Stoklund Olesen   if (Namespace.empty())
885bd92dc60SJakob Stoklund Olesen     return getName();
886bd92dc60SJakob Stoklund Olesen   else
887bd92dc60SJakob Stoklund Olesen     return Namespace + "::" + getName();
88868d6d8abSJakob Stoklund Olesen }
88968d6d8abSJakob Stoklund Olesen 
8902c024b2dSJakob Stoklund Olesen // Compute sub-classes of all register classes.
8912c024b2dSJakob Stoklund Olesen // Assume the classes are ordered topologically.
89203efe84dSJakob Stoklund Olesen void CodeGenRegisterClass::computeSubClasses(CodeGenRegBank &RegBank) {
89303efe84dSJakob Stoklund Olesen   ArrayRef<CodeGenRegisterClass*> RegClasses = RegBank.getRegClasses();
89403efe84dSJakob Stoklund Olesen 
8952c024b2dSJakob Stoklund Olesen   // Visit backwards so sub-classes are seen first.
8962c024b2dSJakob Stoklund Olesen   for (unsigned rci = RegClasses.size(); rci; --rci) {
8972c024b2dSJakob Stoklund Olesen     CodeGenRegisterClass &RC = *RegClasses[rci - 1];
8982c024b2dSJakob Stoklund Olesen     RC.SubClasses.resize(RegClasses.size());
8992c024b2dSJakob Stoklund Olesen     RC.SubClasses.set(RC.EnumValue);
9002c024b2dSJakob Stoklund Olesen 
9012c024b2dSJakob Stoklund Olesen     // Normally, all subclasses have IDs >= rci, unless RC is part of a clique.
9022c024b2dSJakob Stoklund Olesen     for (unsigned s = rci; s != RegClasses.size(); ++s) {
9032c024b2dSJakob Stoklund Olesen       if (RC.SubClasses.test(s))
9042c024b2dSJakob Stoklund Olesen         continue;
9052c024b2dSJakob Stoklund Olesen       CodeGenRegisterClass *SubRC = RegClasses[s];
9066417395dSJakob Stoklund Olesen       if (!testSubClass(&RC, SubRC))
9072c024b2dSJakob Stoklund Olesen         continue;
9082c024b2dSJakob Stoklund Olesen       // SubRC is a sub-class. Grap all its sub-classes so we won't have to
9092c024b2dSJakob Stoklund Olesen       // check them again.
9102c024b2dSJakob Stoklund Olesen       RC.SubClasses |= SubRC->SubClasses;
9112c024b2dSJakob Stoklund Olesen     }
9122c024b2dSJakob Stoklund Olesen 
913bde91766SBenjamin Kramer     // Sweep up missed clique members.  They will be immediately preceding RC.
9146417395dSJakob Stoklund Olesen     for (unsigned s = rci - 1; s && testSubClass(&RC, RegClasses[s - 1]); --s)
9152c024b2dSJakob Stoklund Olesen       RC.SubClasses.set(s - 1);
9162c024b2dSJakob Stoklund Olesen   }
917b15fad9dSJakob Stoklund Olesen 
918b15fad9dSJakob Stoklund Olesen   // Compute the SuperClasses lists from the SubClasses vectors.
919b15fad9dSJakob Stoklund Olesen   for (unsigned rci = 0; rci != RegClasses.size(); ++rci) {
920b15fad9dSJakob Stoklund Olesen     const BitVector &SC = RegClasses[rci]->getSubClasses();
921b15fad9dSJakob Stoklund Olesen     for (int s = SC.find_first(); s >= 0; s = SC.find_next(s)) {
922b15fad9dSJakob Stoklund Olesen       if (unsigned(s) == rci)
923b15fad9dSJakob Stoklund Olesen         continue;
924b15fad9dSJakob Stoklund Olesen       RegClasses[s]->SuperClasses.push_back(RegClasses[rci]);
925b15fad9dSJakob Stoklund Olesen     }
926b15fad9dSJakob Stoklund Olesen   }
92703efe84dSJakob Stoklund Olesen 
92803efe84dSJakob Stoklund Olesen   // With the class hierarchy in place, let synthesized register classes inherit
92903efe84dSJakob Stoklund Olesen   // properties from their closest super-class. The iteration order here can
93003efe84dSJakob Stoklund Olesen   // propagate properties down multiple levels.
93103efe84dSJakob Stoklund Olesen   for (unsigned rci = 0; rci != RegClasses.size(); ++rci)
93203efe84dSJakob Stoklund Olesen     if (!RegClasses[rci]->getDef())
93303efe84dSJakob Stoklund Olesen       RegClasses[rci]->inheritProperties(RegBank);
9342c024b2dSJakob Stoklund Olesen }
9352c024b2dSJakob Stoklund Olesen 
936c7b437aeSJakob Stoklund Olesen void
937f1bb1519SJakob Stoklund Olesen CodeGenRegisterClass::getSuperRegClasses(CodeGenSubRegIndex *SubIdx,
938f1bb1519SJakob Stoklund Olesen                                          BitVector &Out) const {
939f1bb1519SJakob Stoklund Olesen   DenseMap<CodeGenSubRegIndex*,
940f1bb1519SJakob Stoklund Olesen            SmallPtrSet<CodeGenRegisterClass*, 8> >::const_iterator
941c7b437aeSJakob Stoklund Olesen     FindI = SuperRegClasses.find(SubIdx);
942c7b437aeSJakob Stoklund Olesen   if (FindI == SuperRegClasses.end())
943c7b437aeSJakob Stoklund Olesen     return;
944c7b437aeSJakob Stoklund Olesen   for (SmallPtrSet<CodeGenRegisterClass*, 8>::const_iterator I =
945c7b437aeSJakob Stoklund Olesen        FindI->second.begin(), E = FindI->second.end(); I != E; ++I)
946c7b437aeSJakob Stoklund Olesen     Out.set((*I)->EnumValue);
947c7b437aeSJakob Stoklund Olesen }
948c7b437aeSJakob Stoklund Olesen 
94997254150SAndrew Trick // Populate a unique sorted list of units from a register set.
95097254150SAndrew Trick void CodeGenRegisterClass::buildRegUnitSet(
95197254150SAndrew Trick   std::vector<unsigned> &RegUnits) const {
95297254150SAndrew Trick   std::vector<unsigned> TmpUnits;
95397254150SAndrew Trick   for (RegUnitIterator UnitI(Members); UnitI.isValid(); ++UnitI)
95497254150SAndrew Trick     TmpUnits.push_back(*UnitI);
95597254150SAndrew Trick   std::sort(TmpUnits.begin(), TmpUnits.end());
95697254150SAndrew Trick   std::unique_copy(TmpUnits.begin(), TmpUnits.end(),
95797254150SAndrew Trick                    std::back_inserter(RegUnits));
95897254150SAndrew Trick }
959c7b437aeSJakob Stoklund Olesen 
96076a5a71eSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
96176a5a71eSJakob Stoklund Olesen //                               CodeGenRegBank
96276a5a71eSJakob Stoklund Olesen //===----------------------------------------------------------------------===//
96376a5a71eSJakob Stoklund Olesen 
96470a0bbcaSJakob Stoklund Olesen CodeGenRegBank::CodeGenRegBank(RecordKeeper &Records) {
9653bd1b65eSJakob Stoklund Olesen   // Configure register Sets to understand register classes and tuples.
9665ee87726SJakob Stoklund Olesen   Sets.addFieldExpander("RegisterClass", "MemberList");
967c3abb0f6SJakob Stoklund Olesen   Sets.addFieldExpander("CalleeSavedRegs", "SaveList");
9683bd1b65eSJakob Stoklund Olesen   Sets.addExpander("RegisterTuples", new TupleExpander());
9695ee87726SJakob Stoklund Olesen 
97084bd44ebSJakob Stoklund Olesen   // Read in the user-defined (named) sub-register indices.
97184bd44ebSJakob Stoklund Olesen   // More indices will be synthesized later.
972f1bb1519SJakob Stoklund Olesen   std::vector<Record*> SRIs = Records.getAllDerivedDefinitions("SubRegIndex");
973f1bb1519SJakob Stoklund Olesen   std::sort(SRIs.begin(), SRIs.end(), LessRecord());
974f1bb1519SJakob Stoklund Olesen   for (unsigned i = 0, e = SRIs.size(); i != e; ++i)
975f1bb1519SJakob Stoklund Olesen     getSubRegIdx(SRIs[i]);
97621231609SJakob Stoklund Olesen   // Build composite maps from ComposedOf fields.
97721231609SJakob Stoklund Olesen   for (unsigned i = 0, e = SubRegIndices.size(); i != e; ++i)
97821231609SJakob Stoklund Olesen     SubRegIndices[i]->updateComponents(*this);
97984bd44ebSJakob Stoklund Olesen 
98084bd44ebSJakob Stoklund Olesen   // Read in the register definitions.
98184bd44ebSJakob Stoklund Olesen   std::vector<Record*> Regs = Records.getAllDerivedDefinitions("Register");
98284bd44ebSJakob Stoklund Olesen   std::sort(Regs.begin(), Regs.end(), LessRecord());
98384bd44ebSJakob Stoklund Olesen   Registers.reserve(Regs.size());
98484bd44ebSJakob Stoklund Olesen   // Assign the enumeration values.
98584bd44ebSJakob Stoklund Olesen   for (unsigned i = 0, e = Regs.size(); i != e; ++i)
9868e188be0SJakob Stoklund Olesen     getReg(Regs[i]);
98722ea424dSJakob Stoklund Olesen 
9883bd1b65eSJakob Stoklund Olesen   // Expand tuples and number the new registers.
9893bd1b65eSJakob Stoklund Olesen   std::vector<Record*> Tups =
9903bd1b65eSJakob Stoklund Olesen     Records.getAllDerivedDefinitions("RegisterTuples");
9913bd1b65eSJakob Stoklund Olesen   for (unsigned i = 0, e = Tups.size(); i != e; ++i) {
9923bd1b65eSJakob Stoklund Olesen     const std::vector<Record*> *TupRegs = Sets.expand(Tups[i]);
9933bd1b65eSJakob Stoklund Olesen     for (unsigned j = 0, je = TupRegs->size(); j != je; ++j)
9943bd1b65eSJakob Stoklund Olesen       getReg((*TupRegs)[j]);
9953bd1b65eSJakob Stoklund Olesen   }
9963bd1b65eSJakob Stoklund Olesen 
997c1e9087fSJakob Stoklund Olesen   // Now all the registers are known. Build the object graph of explicit
998c1e9087fSJakob Stoklund Olesen   // register-register references.
999c1e9087fSJakob Stoklund Olesen   for (unsigned i = 0, e = Registers.size(); i != e; ++i)
1000c1e9087fSJakob Stoklund Olesen     Registers[i]->buildObjectGraph(*this);
1001c1e9087fSJakob Stoklund Olesen 
1002ccd682c6SOwen Anderson   // Compute register name map.
1003ccd682c6SOwen Anderson   for (unsigned i = 0, e = Registers.size(); i != e; ++i)
1004ccd682c6SOwen Anderson     RegistersByName.GetOrCreateValue(
1005ccd682c6SOwen Anderson                        Registers[i]->TheDef->getValueAsString("AsmName"),
1006ccd682c6SOwen Anderson                        Registers[i]);
1007ccd682c6SOwen Anderson 
1008c1e9087fSJakob Stoklund Olesen   // Precompute all sub-register maps.
100903efe84dSJakob Stoklund Olesen   // This will create Composite entries for all inferred sub-register indices.
101003efe84dSJakob Stoklund Olesen   for (unsigned i = 0, e = Registers.size(); i != e; ++i)
10117d1fa380SJakob Stoklund Olesen     Registers[i]->computeSubRegs(*this);
101203efe84dSJakob Stoklund Olesen 
1013c08df9e5SJakob Stoklund Olesen   // Infer even more sub-registers by combining leading super-registers.
1014c08df9e5SJakob Stoklund Olesen   for (unsigned i = 0, e = Registers.size(); i != e; ++i)
1015c08df9e5SJakob Stoklund Olesen     if (Registers[i]->CoveredBySubRegs)
1016c08df9e5SJakob Stoklund Olesen       Registers[i]->computeSecondarySubRegs(*this);
1017c08df9e5SJakob Stoklund Olesen 
10183f3eb180SJakob Stoklund Olesen   // After the sub-register graph is complete, compute the topologically
10193f3eb180SJakob Stoklund Olesen   // ordered SuperRegs list.
10203f3eb180SJakob Stoklund Olesen   for (unsigned i = 0, e = Registers.size(); i != e; ++i)
102150ecd0ffSJakob Stoklund Olesen     Registers[i]->computeSuperRegs(*this);
10223f3eb180SJakob Stoklund Olesen 
10231d7a2c57SAndrew Trick   // Native register units are associated with a leaf register. They've all been
10241d7a2c57SAndrew Trick   // discovered now.
1025095f22afSJakob Stoklund Olesen   NumNativeRegUnits = RegUnits.size();
10261d7a2c57SAndrew Trick 
102722ea424dSJakob Stoklund Olesen   // Read in register class definitions.
102822ea424dSJakob Stoklund Olesen   std::vector<Record*> RCs = Records.getAllDerivedDefinitions("RegisterClass");
102922ea424dSJakob Stoklund Olesen   if (RCs.empty())
1030635debe8SJoerg Sonnenberger     PrintFatalError(std::string("No 'RegisterClass' subclasses defined!"));
103122ea424dSJakob Stoklund Olesen 
103203efe84dSJakob Stoklund Olesen   // Allocate user-defined register classes.
103322ea424dSJakob Stoklund Olesen   RegClasses.reserve(RCs.size());
103403efe84dSJakob Stoklund Olesen   for (unsigned i = 0, e = RCs.size(); i != e; ++i)
103503efe84dSJakob Stoklund Olesen     addToMaps(new CodeGenRegisterClass(*this, RCs[i]));
103603efe84dSJakob Stoklund Olesen 
103703efe84dSJakob Stoklund Olesen   // Infer missing classes to create a full algebra.
103803efe84dSJakob Stoklund Olesen   computeInferredRegisterClasses();
103903efe84dSJakob Stoklund Olesen 
1040c0fc173dSJakob Stoklund Olesen   // Order register classes topologically and assign enum values.
1041c0fc173dSJakob Stoklund Olesen   array_pod_sort(RegClasses.begin(), RegClasses.end(), TopoOrderRC);
1042c0fc173dSJakob Stoklund Olesen   for (unsigned i = 0, e = RegClasses.size(); i != e; ++i)
1043c0fc173dSJakob Stoklund Olesen     RegClasses[i]->EnumValue = i;
104403efe84dSJakob Stoklund Olesen   CodeGenRegisterClass::computeSubClasses(*this);
104576a5a71eSJakob Stoklund Olesen }
104676a5a71eSJakob Stoklund Olesen 
104770a0bbcaSJakob Stoklund Olesen // Create a synthetic CodeGenSubRegIndex without a corresponding Record.
104870a0bbcaSJakob Stoklund Olesen CodeGenSubRegIndex*
104970a0bbcaSJakob Stoklund Olesen CodeGenRegBank::createSubRegIndex(StringRef Name, StringRef Namespace) {
105070a0bbcaSJakob Stoklund Olesen   CodeGenSubRegIndex *Idx = new CodeGenSubRegIndex(Name, Namespace,
105170a0bbcaSJakob Stoklund Olesen                                                    SubRegIndices.size() + 1);
105270a0bbcaSJakob Stoklund Olesen   SubRegIndices.push_back(Idx);
105370a0bbcaSJakob Stoklund Olesen   return Idx;
105470a0bbcaSJakob Stoklund Olesen }
105570a0bbcaSJakob Stoklund Olesen 
1056f1bb1519SJakob Stoklund Olesen CodeGenSubRegIndex *CodeGenRegBank::getSubRegIdx(Record *Def) {
1057f1bb1519SJakob Stoklund Olesen   CodeGenSubRegIndex *&Idx = Def2SubRegIdx[Def];
1058f1bb1519SJakob Stoklund Olesen   if (Idx)
1059f1bb1519SJakob Stoklund Olesen     return Idx;
1060f1bb1519SJakob Stoklund Olesen   Idx = new CodeGenSubRegIndex(Def, SubRegIndices.size() + 1);
1061f1bb1519SJakob Stoklund Olesen   SubRegIndices.push_back(Idx);
1062f1bb1519SJakob Stoklund Olesen   return Idx;
1063f1bb1519SJakob Stoklund Olesen }
1064f1bb1519SJakob Stoklund Olesen 
106584bd44ebSJakob Stoklund Olesen CodeGenRegister *CodeGenRegBank::getReg(Record *Def) {
10668e188be0SJakob Stoklund Olesen   CodeGenRegister *&Reg = Def2Reg[Def];
10678e188be0SJakob Stoklund Olesen   if (Reg)
106884bd44ebSJakob Stoklund Olesen     return Reg;
10698e188be0SJakob Stoklund Olesen   Reg = new CodeGenRegister(Def, Registers.size() + 1);
10708e188be0SJakob Stoklund Olesen   Registers.push_back(Reg);
10718e188be0SJakob Stoklund Olesen   return Reg;
107284bd44ebSJakob Stoklund Olesen }
107384bd44ebSJakob Stoklund Olesen 
107403efe84dSJakob Stoklund Olesen void CodeGenRegBank::addToMaps(CodeGenRegisterClass *RC) {
107503efe84dSJakob Stoklund Olesen   RegClasses.push_back(RC);
107603efe84dSJakob Stoklund Olesen 
107703efe84dSJakob Stoklund Olesen   if (Record *Def = RC->getDef())
107803efe84dSJakob Stoklund Olesen     Def2RC.insert(std::make_pair(Def, RC));
107903efe84dSJakob Stoklund Olesen 
108003efe84dSJakob Stoklund Olesen   // Duplicate classes are rejected by insert().
108103efe84dSJakob Stoklund Olesen   // That's OK, we only care about the properties handled by CGRC::Key.
108203efe84dSJakob Stoklund Olesen   CodeGenRegisterClass::Key K(*RC);
108303efe84dSJakob Stoklund Olesen   Key2RC.insert(std::make_pair(K, RC));
108403efe84dSJakob Stoklund Olesen }
108503efe84dSJakob Stoklund Olesen 
10867ebc6b05SJakob Stoklund Olesen // Create a synthetic sub-class if it is missing.
10877ebc6b05SJakob Stoklund Olesen CodeGenRegisterClass*
10887ebc6b05SJakob Stoklund Olesen CodeGenRegBank::getOrCreateSubClass(const CodeGenRegisterClass *RC,
10897ebc6b05SJakob Stoklund Olesen                                     const CodeGenRegister::Set *Members,
10907ebc6b05SJakob Stoklund Olesen                                     StringRef Name) {
10917ebc6b05SJakob Stoklund Olesen   // Synthetic sub-class has the same size and alignment as RC.
10927ebc6b05SJakob Stoklund Olesen   CodeGenRegisterClass::Key K(Members, RC->SpillSize, RC->SpillAlignment);
10937ebc6b05SJakob Stoklund Olesen   RCKeyMap::const_iterator FoundI = Key2RC.find(K);
10947ebc6b05SJakob Stoklund Olesen   if (FoundI != Key2RC.end())
10957ebc6b05SJakob Stoklund Olesen     return FoundI->second;
10967ebc6b05SJakob Stoklund Olesen 
10977ebc6b05SJakob Stoklund Olesen   // Sub-class doesn't exist, create a new one.
1098eebd5bc6SJakob Stoklund Olesen   CodeGenRegisterClass *NewRC = new CodeGenRegisterClass(*this, Name, K);
10997ebc6b05SJakob Stoklund Olesen   addToMaps(NewRC);
11007ebc6b05SJakob Stoklund Olesen   return NewRC;
11017ebc6b05SJakob Stoklund Olesen }
11027ebc6b05SJakob Stoklund Olesen 
110322ea424dSJakob Stoklund Olesen CodeGenRegisterClass *CodeGenRegBank::getRegClass(Record *Def) {
110422ea424dSJakob Stoklund Olesen   if (CodeGenRegisterClass *RC = Def2RC[Def])
110522ea424dSJakob Stoklund Olesen     return RC;
110622ea424dSJakob Stoklund Olesen 
1107635debe8SJoerg Sonnenberger   PrintFatalError(Def->getLoc(), "Not a known RegisterClass!");
110822ea424dSJakob Stoklund Olesen }
110922ea424dSJakob Stoklund Olesen 
1110f1bb1519SJakob Stoklund Olesen CodeGenSubRegIndex*
1111f1bb1519SJakob Stoklund Olesen CodeGenRegBank::getCompositeSubRegIndex(CodeGenSubRegIndex *A,
11129a44ad70SJakob Stoklund Olesen                                         CodeGenSubRegIndex *B) {
111384bd44ebSJakob Stoklund Olesen   // Look for an existing entry.
11149a44ad70SJakob Stoklund Olesen   CodeGenSubRegIndex *Comp = A->compose(B);
11159a44ad70SJakob Stoklund Olesen   if (Comp)
111684bd44ebSJakob Stoklund Olesen     return Comp;
111784bd44ebSJakob Stoklund Olesen 
111884bd44ebSJakob Stoklund Olesen   // None exists, synthesize one.
111976a5a71eSJakob Stoklund Olesen   std::string Name = A->getName() + "_then_" + B->getName();
112070a0bbcaSJakob Stoklund Olesen   Comp = createSubRegIndex(Name, A->getNamespace());
11219a44ad70SJakob Stoklund Olesen   A->addComposite(B, Comp);
112284bd44ebSJakob Stoklund Olesen   return Comp;
112376a5a71eSJakob Stoklund Olesen }
112476a5a71eSJakob Stoklund Olesen 
1125c08df9e5SJakob Stoklund Olesen CodeGenSubRegIndex *CodeGenRegBank::
1126c08df9e5SJakob Stoklund Olesen getConcatSubRegIndex(const SmallVector<CodeGenSubRegIndex*, 8> &Parts) {
1127c08df9e5SJakob Stoklund Olesen   assert(Parts.size() > 1 && "Need two parts to concatenate");
1128c08df9e5SJakob Stoklund Olesen 
1129c08df9e5SJakob Stoklund Olesen   // Look for an existing entry.
1130c08df9e5SJakob Stoklund Olesen   CodeGenSubRegIndex *&Idx = ConcatIdx[Parts];
1131c08df9e5SJakob Stoklund Olesen   if (Idx)
1132c08df9e5SJakob Stoklund Olesen     return Idx;
1133c08df9e5SJakob Stoklund Olesen 
1134c08df9e5SJakob Stoklund Olesen   // None exists, synthesize one.
1135c08df9e5SJakob Stoklund Olesen   std::string Name = Parts.front()->getName();
1136c08df9e5SJakob Stoklund Olesen   for (unsigned i = 1, e = Parts.size(); i != e; ++i) {
1137c08df9e5SJakob Stoklund Olesen     Name += '_';
1138c08df9e5SJakob Stoklund Olesen     Name += Parts[i]->getName();
1139c08df9e5SJakob Stoklund Olesen   }
114070a0bbcaSJakob Stoklund Olesen   return Idx = createSubRegIndex(Name, Parts.front()->getNamespace());
1141c08df9e5SJakob Stoklund Olesen }
1142c08df9e5SJakob Stoklund Olesen 
114384bd44ebSJakob Stoklund Olesen void CodeGenRegBank::computeComposites() {
114450ecd0ffSJakob Stoklund Olesen   // Keep track of TopoSigs visited. We only need to visit each TopoSig once,
114550ecd0ffSJakob Stoklund Olesen   // and many registers will share TopoSigs on regular architectures.
114650ecd0ffSJakob Stoklund Olesen   BitVector TopoSigs(getNumTopoSigs());
114750ecd0ffSJakob Stoklund Olesen 
114884bd44ebSJakob Stoklund Olesen   for (unsigned i = 0, e = Registers.size(); i != e; ++i) {
11498e188be0SJakob Stoklund Olesen     CodeGenRegister *Reg1 = Registers[i];
115050ecd0ffSJakob Stoklund Olesen 
115150ecd0ffSJakob Stoklund Olesen     // Skip identical subreg structures already processed.
115250ecd0ffSJakob Stoklund Olesen     if (TopoSigs.test(Reg1->getTopoSig()))
115350ecd0ffSJakob Stoklund Olesen       continue;
115450ecd0ffSJakob Stoklund Olesen     TopoSigs.set(Reg1->getTopoSig());
115550ecd0ffSJakob Stoklund Olesen 
1156d2b4713eSJakob Stoklund Olesen     const CodeGenRegister::SubRegMap &SRM1 = Reg1->getSubRegs();
115784bd44ebSJakob Stoklund Olesen     for (CodeGenRegister::SubRegMap::const_iterator i1 = SRM1.begin(),
115884bd44ebSJakob Stoklund Olesen          e1 = SRM1.end(); i1 != e1; ++i1) {
1159f1bb1519SJakob Stoklund Olesen       CodeGenSubRegIndex *Idx1 = i1->first;
116084bd44ebSJakob Stoklund Olesen       CodeGenRegister *Reg2 = i1->second;
116184bd44ebSJakob Stoklund Olesen       // Ignore identity compositions.
116284bd44ebSJakob Stoklund Olesen       if (Reg1 == Reg2)
116384bd44ebSJakob Stoklund Olesen         continue;
1164d2b4713eSJakob Stoklund Olesen       const CodeGenRegister::SubRegMap &SRM2 = Reg2->getSubRegs();
116584bd44ebSJakob Stoklund Olesen       // Try composing Idx1 with another SubRegIndex.
116684bd44ebSJakob Stoklund Olesen       for (CodeGenRegister::SubRegMap::const_iterator i2 = SRM2.begin(),
116784bd44ebSJakob Stoklund Olesen            e2 = SRM2.end(); i2 != e2; ++i2) {
11689a44ad70SJakob Stoklund Olesen         CodeGenSubRegIndex *Idx2 = i2->first;
116984bd44ebSJakob Stoklund Olesen         CodeGenRegister *Reg3 = i2->second;
117084bd44ebSJakob Stoklund Olesen         // Ignore identity compositions.
117184bd44ebSJakob Stoklund Olesen         if (Reg2 == Reg3)
117284bd44ebSJakob Stoklund Olesen           continue;
117384bd44ebSJakob Stoklund Olesen         // OK Reg1:IdxPair == Reg3. Find the index with Reg:Idx == Reg3.
11742d247c80SJakob Stoklund Olesen         CodeGenSubRegIndex *Idx3 = Reg1->getSubRegIndex(Reg3);
11752d247c80SJakob Stoklund Olesen         assert(Idx3 && "Sub-register doesn't have an index");
11762d247c80SJakob Stoklund Olesen 
117784bd44ebSJakob Stoklund Olesen         // Conflicting composition? Emit a warning but allow it.
11782d247c80SJakob Stoklund Olesen         if (CodeGenSubRegIndex *Prev = Idx1->addComposite(Idx2, Idx3))
11799a7f4b76SJim Grosbach           PrintWarning(Twine("SubRegIndex ") + Idx1->getQualifiedName() +
11809a7f4b76SJim Grosbach                        " and " + Idx2->getQualifiedName() +
11819a7f4b76SJim Grosbach                        " compose ambiguously as " + Prev->getQualifiedName() +
11822d247c80SJakob Stoklund Olesen                        " or " + Idx3->getQualifiedName());
118384bd44ebSJakob Stoklund Olesen       }
118484bd44ebSJakob Stoklund Olesen     }
118584bd44ebSJakob Stoklund Olesen   }
118684bd44ebSJakob Stoklund Olesen }
118784bd44ebSJakob Stoklund Olesen 
1188d346d487SJakob Stoklund Olesen // Compute lane masks. This is similar to register units, but at the
1189d346d487SJakob Stoklund Olesen // sub-register index level. Each bit in the lane mask is like a register unit
1190d346d487SJakob Stoklund Olesen // class, and two lane masks will have a bit in common if two sub-register
1191d346d487SJakob Stoklund Olesen // indices overlap in some register.
1192d346d487SJakob Stoklund Olesen //
1193d346d487SJakob Stoklund Olesen // Conservatively share a lane mask bit if two sub-register indices overlap in
1194d346d487SJakob Stoklund Olesen // some registers, but not in others. That shouldn't happen a lot.
1195d346d487SJakob Stoklund Olesen void CodeGenRegBank::computeSubRegIndexLaneMasks() {
1196d346d487SJakob Stoklund Olesen   // First assign individual bits to all the leaf indices.
1197d346d487SJakob Stoklund Olesen   unsigned Bit = 0;
1198d346d487SJakob Stoklund Olesen   for (unsigned i = 0, e = SubRegIndices.size(); i != e; ++i) {
1199d346d487SJakob Stoklund Olesen     CodeGenSubRegIndex *Idx = SubRegIndices[i];
1200d346d487SJakob Stoklund Olesen     if (Idx->getComposites().empty()) {
1201d346d487SJakob Stoklund Olesen       Idx->LaneMask = 1u << Bit;
1202d346d487SJakob Stoklund Olesen       // Share bit 31 in the unlikely case there are more than 32 leafs.
12036b1eda0aSJakob Stoklund Olesen       //
12046b1eda0aSJakob Stoklund Olesen       // Sharing bits is harmless; it allows graceful degradation in targets
12056b1eda0aSJakob Stoklund Olesen       // with more than 32 vector lanes. They simply get a limited resolution
12066b1eda0aSJakob Stoklund Olesen       // view of lanes beyond the 32nd.
12076b1eda0aSJakob Stoklund Olesen       //
12086b1eda0aSJakob Stoklund Olesen       // See also the comment for getSubRegIndexLaneMask().
1209d346d487SJakob Stoklund Olesen       if (Bit < 31) ++Bit;
1210d346d487SJakob Stoklund Olesen     } else {
1211d346d487SJakob Stoklund Olesen       Idx->LaneMask = 0;
1212d346d487SJakob Stoklund Olesen     }
1213d346d487SJakob Stoklund Olesen   }
1214d346d487SJakob Stoklund Olesen 
1215d346d487SJakob Stoklund Olesen   // FIXME: What if ad-hoc aliasing introduces overlaps that aren't represented
1216d346d487SJakob Stoklund Olesen   // by the sub-register graph? This doesn't occur in any known targets.
1217d346d487SJakob Stoklund Olesen 
1218d346d487SJakob Stoklund Olesen   // Inherit lanes from composites.
1219d346d487SJakob Stoklund Olesen   for (unsigned i = 0, e = SubRegIndices.size(); i != e; ++i)
1220d346d487SJakob Stoklund Olesen     SubRegIndices[i]->computeLaneMask();
1221d346d487SJakob Stoklund Olesen }
1222d346d487SJakob Stoklund Olesen 
12231d7a2c57SAndrew Trick namespace {
12241d7a2c57SAndrew Trick // UberRegSet is a helper class for computeRegUnitWeights. Each UberRegSet is
12251d7a2c57SAndrew Trick // the transitive closure of the union of overlapping register
12261d7a2c57SAndrew Trick // classes. Together, the UberRegSets form a partition of the registers. If we
12271d7a2c57SAndrew Trick // consider overlapping register classes to be connected, then each UberRegSet
12281d7a2c57SAndrew Trick // is a set of connected components.
12291d7a2c57SAndrew Trick //
12301d7a2c57SAndrew Trick // An UberRegSet will likely be a horizontal slice of register names of
12311d7a2c57SAndrew Trick // the same width. Nontrivial subregisters should then be in a separate
12321d7a2c57SAndrew Trick // UberRegSet. But this property isn't required for valid computation of
12331d7a2c57SAndrew Trick // register unit weights.
12341d7a2c57SAndrew Trick //
12351d7a2c57SAndrew Trick // A Weight field caches the max per-register unit weight in each UberRegSet.
12361d7a2c57SAndrew Trick //
12371d7a2c57SAndrew Trick // A set of SingularDeterminants flags single units of some register in this set
12381d7a2c57SAndrew Trick // for which the unit weight equals the set weight. These units should not have
12391d7a2c57SAndrew Trick // their weight increased.
12401d7a2c57SAndrew Trick struct UberRegSet {
12411d7a2c57SAndrew Trick   CodeGenRegister::Set Regs;
12421d7a2c57SAndrew Trick   unsigned Weight;
12431d7a2c57SAndrew Trick   CodeGenRegister::RegUnitList SingularDeterminants;
12441d7a2c57SAndrew Trick 
12451d7a2c57SAndrew Trick   UberRegSet(): Weight(0) {}
12461d7a2c57SAndrew Trick };
12471d7a2c57SAndrew Trick } // namespace
12481d7a2c57SAndrew Trick 
12491d7a2c57SAndrew Trick // Partition registers into UberRegSets, where each set is the transitive
12501d7a2c57SAndrew Trick // closure of the union of overlapping register classes.
12511d7a2c57SAndrew Trick //
12521d7a2c57SAndrew Trick // UberRegSets[0] is a special non-allocatable set.
12531d7a2c57SAndrew Trick static void computeUberSets(std::vector<UberRegSet> &UberSets,
12541d7a2c57SAndrew Trick                             std::vector<UberRegSet*> &RegSets,
12551d7a2c57SAndrew Trick                             CodeGenRegBank &RegBank) {
12561d7a2c57SAndrew Trick 
12571d7a2c57SAndrew Trick   const std::vector<CodeGenRegister*> &Registers = RegBank.getRegisters();
12581d7a2c57SAndrew Trick 
12591d7a2c57SAndrew Trick   // The Register EnumValue is one greater than its index into Registers.
12601d7a2c57SAndrew Trick   assert(Registers.size() == Registers[Registers.size()-1]->EnumValue &&
12611d7a2c57SAndrew Trick          "register enum value mismatch");
12621d7a2c57SAndrew Trick 
12631d7a2c57SAndrew Trick   // For simplicitly make the SetID the same as EnumValue.
12641d7a2c57SAndrew Trick   IntEqClasses UberSetIDs(Registers.size()+1);
12650d94c73cSAndrew Trick   std::set<unsigned> AllocatableRegs;
12661d7a2c57SAndrew Trick   for (unsigned i = 0, e = RegBank.getRegClasses().size(); i != e; ++i) {
12670d94c73cSAndrew Trick 
12681d7a2c57SAndrew Trick     CodeGenRegisterClass *RegClass = RegBank.getRegClasses()[i];
12690d94c73cSAndrew Trick     if (!RegClass->Allocatable)
12700d94c73cSAndrew Trick       continue;
12710d94c73cSAndrew Trick 
12721d7a2c57SAndrew Trick     const CodeGenRegister::Set &Regs = RegClass->getMembers();
12730d94c73cSAndrew Trick     if (Regs.empty())
12740d94c73cSAndrew Trick       continue;
12751d7a2c57SAndrew Trick 
12761d7a2c57SAndrew Trick     unsigned USetID = UberSetIDs.findLeader((*Regs.begin())->EnumValue);
12771d7a2c57SAndrew Trick     assert(USetID && "register number 0 is invalid");
12781d7a2c57SAndrew Trick 
12790d94c73cSAndrew Trick     AllocatableRegs.insert((*Regs.begin())->EnumValue);
12801d7a2c57SAndrew Trick     for (CodeGenRegister::Set::const_iterator I = llvm::next(Regs.begin()),
12810d94c73cSAndrew Trick            E = Regs.end(); I != E; ++I) {
12820d94c73cSAndrew Trick       AllocatableRegs.insert((*I)->EnumValue);
12831d7a2c57SAndrew Trick       UberSetIDs.join(USetID, (*I)->EnumValue);
12841d7a2c57SAndrew Trick     }
12850d94c73cSAndrew Trick   }
12860d94c73cSAndrew Trick   // Combine non-allocatable regs.
12870d94c73cSAndrew Trick   for (unsigned i = 0, e = Registers.size(); i != e; ++i) {
12880d94c73cSAndrew Trick     unsigned RegNum = Registers[i]->EnumValue;
12890d94c73cSAndrew Trick     if (AllocatableRegs.count(RegNum))
12900d94c73cSAndrew Trick       continue;
12910d94c73cSAndrew Trick 
12920d94c73cSAndrew Trick     UberSetIDs.join(0, RegNum);
12930d94c73cSAndrew Trick   }
12941d7a2c57SAndrew Trick   UberSetIDs.compress();
12951d7a2c57SAndrew Trick 
12961d7a2c57SAndrew Trick   // Make the first UberSet a special unallocatable set.
12971d7a2c57SAndrew Trick   unsigned ZeroID = UberSetIDs[0];
12981d7a2c57SAndrew Trick 
12991d7a2c57SAndrew Trick   // Insert Registers into the UberSets formed by union-find.
13001d7a2c57SAndrew Trick   // Do not resize after this.
13011d7a2c57SAndrew Trick   UberSets.resize(UberSetIDs.getNumClasses());
13021d7a2c57SAndrew Trick   for (unsigned i = 0, e = Registers.size(); i != e; ++i) {
13031d7a2c57SAndrew Trick     const CodeGenRegister *Reg = Registers[i];
13041d7a2c57SAndrew Trick     unsigned USetID = UberSetIDs[Reg->EnumValue];
13051d7a2c57SAndrew Trick     if (!USetID)
13061d7a2c57SAndrew Trick       USetID = ZeroID;
13071d7a2c57SAndrew Trick     else if (USetID == ZeroID)
13081d7a2c57SAndrew Trick       USetID = 0;
13091d7a2c57SAndrew Trick 
13101d7a2c57SAndrew Trick     UberRegSet *USet = &UberSets[USetID];
13111d7a2c57SAndrew Trick     USet->Regs.insert(Reg);
13121d7a2c57SAndrew Trick     RegSets[i] = USet;
13131d7a2c57SAndrew Trick   }
13141d7a2c57SAndrew Trick }
13151d7a2c57SAndrew Trick 
13161d7a2c57SAndrew Trick // Recompute each UberSet weight after changing unit weights.
13171d7a2c57SAndrew Trick static void computeUberWeights(std::vector<UberRegSet> &UberSets,
13181d7a2c57SAndrew Trick                                CodeGenRegBank &RegBank) {
13191d7a2c57SAndrew Trick   // Skip the first unallocatable set.
13201d7a2c57SAndrew Trick   for (std::vector<UberRegSet>::iterator I = llvm::next(UberSets.begin()),
13211d7a2c57SAndrew Trick          E = UberSets.end(); I != E; ++I) {
13221d7a2c57SAndrew Trick 
13231d7a2c57SAndrew Trick     // Initialize all unit weights in this set, and remember the max units/reg.
13241d7a2c57SAndrew Trick     const CodeGenRegister *Reg = 0;
13251d7a2c57SAndrew Trick     unsigned MaxWeight = 0, Weight = 0;
13261d7a2c57SAndrew Trick     for (RegUnitIterator UnitI(I->Regs); UnitI.isValid(); ++UnitI) {
13271d7a2c57SAndrew Trick       if (Reg != UnitI.getReg()) {
13281d7a2c57SAndrew Trick         if (Weight > MaxWeight)
13291d7a2c57SAndrew Trick           MaxWeight = Weight;
13301d7a2c57SAndrew Trick         Reg = UnitI.getReg();
13311d7a2c57SAndrew Trick         Weight = 0;
13321d7a2c57SAndrew Trick       }
1333095f22afSJakob Stoklund Olesen       unsigned UWeight = RegBank.getRegUnit(*UnitI).Weight;
13341d7a2c57SAndrew Trick       if (!UWeight) {
13351d7a2c57SAndrew Trick         UWeight = 1;
13361d7a2c57SAndrew Trick         RegBank.increaseRegUnitWeight(*UnitI, UWeight);
13371d7a2c57SAndrew Trick       }
13381d7a2c57SAndrew Trick       Weight += UWeight;
13391d7a2c57SAndrew Trick     }
13401d7a2c57SAndrew Trick     if (Weight > MaxWeight)
13411d7a2c57SAndrew Trick       MaxWeight = Weight;
13421d7a2c57SAndrew Trick 
13431d7a2c57SAndrew Trick     // Update the set weight.
13441d7a2c57SAndrew Trick     I->Weight = MaxWeight;
13451d7a2c57SAndrew Trick 
13461d7a2c57SAndrew Trick     // Find singular determinants.
13471d7a2c57SAndrew Trick     for (CodeGenRegister::Set::iterator RegI = I->Regs.begin(),
13481d7a2c57SAndrew Trick            RegE = I->Regs.end(); RegI != RegE; ++RegI) {
13491d7a2c57SAndrew Trick       if ((*RegI)->getRegUnits().size() == 1
13501d7a2c57SAndrew Trick           && (*RegI)->getWeight(RegBank) == I->Weight)
13511d7a2c57SAndrew Trick         mergeRegUnits(I->SingularDeterminants, (*RegI)->getRegUnits());
13521d7a2c57SAndrew Trick     }
13531d7a2c57SAndrew Trick   }
13541d7a2c57SAndrew Trick }
13551d7a2c57SAndrew Trick 
13561d7a2c57SAndrew Trick // normalizeWeight is a computeRegUnitWeights helper that adjusts the weight of
13571d7a2c57SAndrew Trick // a register and its subregisters so that they have the same weight as their
13581d7a2c57SAndrew Trick // UberSet. Self-recursion processes the subregister tree in postorder so
13591d7a2c57SAndrew Trick // subregisters are normalized first.
13601d7a2c57SAndrew Trick //
13611d7a2c57SAndrew Trick // Side effects:
13621d7a2c57SAndrew Trick // - creates new adopted register units
13631d7a2c57SAndrew Trick // - causes superregisters to inherit adopted units
13641d7a2c57SAndrew Trick // - increases the weight of "singular" units
13651d7a2c57SAndrew Trick // - induces recomputation of UberWeights.
13661d7a2c57SAndrew Trick static bool normalizeWeight(CodeGenRegister *Reg,
13671d7a2c57SAndrew Trick                             std::vector<UberRegSet> &UberSets,
13681d7a2c57SAndrew Trick                             std::vector<UberRegSet*> &RegSets,
13695d133998SAndrew Trick                             std::set<unsigned> &NormalRegs,
13701d7a2c57SAndrew Trick                             CodeGenRegister::RegUnitList &NormalUnits,
13711d7a2c57SAndrew Trick                             CodeGenRegBank &RegBank) {
13721d7a2c57SAndrew Trick   bool Changed = false;
13735d133998SAndrew Trick   if (!NormalRegs.insert(Reg->EnumValue).second)
13745d133998SAndrew Trick     return Changed;
13755d133998SAndrew Trick 
13761d7a2c57SAndrew Trick   const CodeGenRegister::SubRegMap &SRM = Reg->getSubRegs();
13771d7a2c57SAndrew Trick   for (CodeGenRegister::SubRegMap::const_iterator SRI = SRM.begin(),
13781d7a2c57SAndrew Trick          SRE = SRM.end(); SRI != SRE; ++SRI) {
13791d7a2c57SAndrew Trick     if (SRI->second == Reg)
13801d7a2c57SAndrew Trick       continue; // self-cycles happen
13811d7a2c57SAndrew Trick 
13825d133998SAndrew Trick     Changed |= normalizeWeight(SRI->second, UberSets, RegSets,
13835d133998SAndrew Trick                                NormalRegs, NormalUnits, RegBank);
13841d7a2c57SAndrew Trick   }
13851d7a2c57SAndrew Trick   // Postorder register normalization.
13861d7a2c57SAndrew Trick 
13871d7a2c57SAndrew Trick   // Inherit register units newly adopted by subregisters.
13881d7a2c57SAndrew Trick   if (Reg->inheritRegUnits(RegBank))
13891d7a2c57SAndrew Trick     computeUberWeights(UberSets, RegBank);
13901d7a2c57SAndrew Trick 
13911d7a2c57SAndrew Trick   // Check if this register is too skinny for its UberRegSet.
13921d7a2c57SAndrew Trick   UberRegSet *UberSet = RegSets[RegBank.getRegIndex(Reg)];
13931d7a2c57SAndrew Trick 
13941d7a2c57SAndrew Trick   unsigned RegWeight = Reg->getWeight(RegBank);
13951d7a2c57SAndrew Trick   if (UberSet->Weight > RegWeight) {
13961d7a2c57SAndrew Trick     // A register unit's weight can be adjusted only if it is the singular unit
13971d7a2c57SAndrew Trick     // for this register, has not been used to normalize a subregister's set,
13981d7a2c57SAndrew Trick     // and has not already been used to singularly determine this UberRegSet.
13991d7a2c57SAndrew Trick     unsigned AdjustUnit = Reg->getRegUnits().front();
14001d7a2c57SAndrew Trick     if (Reg->getRegUnits().size() != 1
14011d7a2c57SAndrew Trick         || hasRegUnit(NormalUnits, AdjustUnit)
14021d7a2c57SAndrew Trick         || hasRegUnit(UberSet->SingularDeterminants, AdjustUnit)) {
14031d7a2c57SAndrew Trick       // We don't have an adjustable unit, so adopt a new one.
14041d7a2c57SAndrew Trick       AdjustUnit = RegBank.newRegUnit(UberSet->Weight - RegWeight);
14051d7a2c57SAndrew Trick       Reg->adoptRegUnit(AdjustUnit);
14061d7a2c57SAndrew Trick       // Adopting a unit does not immediately require recomputing set weights.
14071d7a2c57SAndrew Trick     }
14081d7a2c57SAndrew Trick     else {
14091d7a2c57SAndrew Trick       // Adjust the existing single unit.
14101d7a2c57SAndrew Trick       RegBank.increaseRegUnitWeight(AdjustUnit, UberSet->Weight - RegWeight);
14111d7a2c57SAndrew Trick       // The unit may be shared among sets and registers within this set.
14121d7a2c57SAndrew Trick       computeUberWeights(UberSets, RegBank);
14131d7a2c57SAndrew Trick     }
14141d7a2c57SAndrew Trick     Changed = true;
14151d7a2c57SAndrew Trick   }
14161d7a2c57SAndrew Trick 
14171d7a2c57SAndrew Trick   // Mark these units normalized so superregisters can't change their weights.
14181d7a2c57SAndrew Trick   mergeRegUnits(NormalUnits, Reg->getRegUnits());
14191d7a2c57SAndrew Trick 
14201d7a2c57SAndrew Trick   return Changed;
14211d7a2c57SAndrew Trick }
14221d7a2c57SAndrew Trick 
14231d7a2c57SAndrew Trick // Compute a weight for each register unit created during getSubRegs.
14241d7a2c57SAndrew Trick //
14251d7a2c57SAndrew Trick // The goal is that two registers in the same class will have the same weight,
14261d7a2c57SAndrew Trick // where each register's weight is defined as sum of its units' weights.
14271d7a2c57SAndrew Trick void CodeGenRegBank::computeRegUnitWeights() {
14281d7a2c57SAndrew Trick   std::vector<UberRegSet> UberSets;
14291d7a2c57SAndrew Trick   std::vector<UberRegSet*> RegSets(Registers.size());
14301d7a2c57SAndrew Trick   computeUberSets(UberSets, RegSets, *this);
14311d7a2c57SAndrew Trick   // UberSets and RegSets are now immutable.
14321d7a2c57SAndrew Trick 
14331d7a2c57SAndrew Trick   computeUberWeights(UberSets, *this);
14341d7a2c57SAndrew Trick 
14351d7a2c57SAndrew Trick   // Iterate over each Register, normalizing the unit weights until reaching
14361d7a2c57SAndrew Trick   // a fix point.
14371d7a2c57SAndrew Trick   unsigned NumIters = 0;
14381d7a2c57SAndrew Trick   for (bool Changed = true; Changed; ++NumIters) {
14391d7a2c57SAndrew Trick     assert(NumIters <= NumNativeRegUnits && "Runaway register unit weights");
14401d7a2c57SAndrew Trick     Changed = false;
14411d7a2c57SAndrew Trick     for (unsigned i = 0, e = Registers.size(); i != e; ++i) {
14421d7a2c57SAndrew Trick       CodeGenRegister::RegUnitList NormalUnits;
14435d133998SAndrew Trick       std::set<unsigned> NormalRegs;
14445d133998SAndrew Trick       Changed |= normalizeWeight(Registers[i], UberSets, RegSets,
14455d133998SAndrew Trick                                  NormalRegs, NormalUnits, *this);
14461d7a2c57SAndrew Trick     }
14471d7a2c57SAndrew Trick   }
14481d7a2c57SAndrew Trick }
14491d7a2c57SAndrew Trick 
1450739a0038SAndrew Trick // Find a set in UniqueSets with the same elements as Set.
1451739a0038SAndrew Trick // Return an iterator into UniqueSets.
1452739a0038SAndrew Trick static std::vector<RegUnitSet>::const_iterator
1453739a0038SAndrew Trick findRegUnitSet(const std::vector<RegUnitSet> &UniqueSets,
1454739a0038SAndrew Trick                const RegUnitSet &Set) {
1455739a0038SAndrew Trick   std::vector<RegUnitSet>::const_iterator
1456739a0038SAndrew Trick     I = UniqueSets.begin(), E = UniqueSets.end();
1457739a0038SAndrew Trick   for(;I != E; ++I) {
1458739a0038SAndrew Trick     if (I->Units == Set.Units)
1459739a0038SAndrew Trick       break;
1460739a0038SAndrew Trick   }
1461739a0038SAndrew Trick   return I;
1462739a0038SAndrew Trick }
1463739a0038SAndrew Trick 
1464739a0038SAndrew Trick // Return true if the RUSubSet is a subset of RUSuperSet.
1465739a0038SAndrew Trick static bool isRegUnitSubSet(const std::vector<unsigned> &RUSubSet,
1466739a0038SAndrew Trick                             const std::vector<unsigned> &RUSuperSet) {
14679002c315SAndrew Trick   return std::includes(RUSuperSet.begin(), RUSuperSet.end(),
14689002c315SAndrew Trick                        RUSubSet.begin(), RUSubSet.end());
1469739a0038SAndrew Trick }
1470739a0038SAndrew Trick 
1471739a0038SAndrew Trick // Iteratively prune unit sets.
1472739a0038SAndrew Trick void CodeGenRegBank::pruneUnitSets() {
1473739a0038SAndrew Trick   assert(RegClassUnitSets.empty() && "this invalidates RegClassUnitSets");
1474739a0038SAndrew Trick 
1475739a0038SAndrew Trick   // Form an equivalence class of UnitSets with no significant difference.
1476a5eee987SAndrew Trick   std::vector<unsigned> SuperSetIDs;
1477739a0038SAndrew Trick   for (unsigned SubIdx = 0, EndIdx = RegUnitSets.size();
1478739a0038SAndrew Trick        SubIdx != EndIdx; ++SubIdx) {
1479739a0038SAndrew Trick     const RegUnitSet &SubSet = RegUnitSets[SubIdx];
14800d94c73cSAndrew Trick     unsigned SuperIdx = 0;
14810d94c73cSAndrew Trick     for (; SuperIdx != EndIdx; ++SuperIdx) {
1482739a0038SAndrew Trick       if (SuperIdx == SubIdx)
1483739a0038SAndrew Trick         continue;
1484a5eee987SAndrew Trick 
1485a5eee987SAndrew Trick       const RegUnitSet &SuperSet = RegUnitSets[SuperIdx];
1486a5eee987SAndrew Trick       if (isRegUnitSubSet(SubSet.Units, SuperSet.Units)
1487a5eee987SAndrew Trick           && (SubSet.Units.size() + 3 > SuperSet.Units.size())) {
14880d94c73cSAndrew Trick         break;
1489739a0038SAndrew Trick       }
1490739a0038SAndrew Trick     }
1491a5eee987SAndrew Trick     if (SuperIdx == EndIdx)
1492a5eee987SAndrew Trick       SuperSetIDs.push_back(SubIdx);
1493a5eee987SAndrew Trick   }
1494a5eee987SAndrew Trick   // Populate PrunedUnitSets with each equivalence class's superset.
1495a5eee987SAndrew Trick   std::vector<RegUnitSet> PrunedUnitSets(SuperSetIDs.size());
1496a5eee987SAndrew Trick   for (unsigned i = 0, e = SuperSetIDs.size(); i != e; ++i) {
1497a5eee987SAndrew Trick     unsigned SuperIdx = SuperSetIDs[i];
1498a5eee987SAndrew Trick     PrunedUnitSets[i].Name = RegUnitSets[SuperIdx].Name;
1499a5eee987SAndrew Trick     PrunedUnitSets[i].Units.swap(RegUnitSets[SuperIdx].Units);
1500739a0038SAndrew Trick   }
1501739a0038SAndrew Trick   RegUnitSets.swap(PrunedUnitSets);
1502739a0038SAndrew Trick }
1503739a0038SAndrew Trick 
1504739a0038SAndrew Trick // Create a RegUnitSet for each RegClass that contains all units in the class
1505739a0038SAndrew Trick // including adopted units that are necessary to model register pressure. Then
1506739a0038SAndrew Trick // iteratively compute RegUnitSets such that the union of any two overlapping
1507739a0038SAndrew Trick // RegUnitSets is repreresented.
1508739a0038SAndrew Trick //
1509739a0038SAndrew Trick // RegisterInfoEmitter will map each RegClass to its RegUnitClass and any
1510739a0038SAndrew Trick // RegUnitSet that is a superset of that RegUnitClass.
1511739a0038SAndrew Trick void CodeGenRegBank::computeRegUnitSets() {
1512739a0038SAndrew Trick 
1513739a0038SAndrew Trick   // Compute a unique RegUnitSet for each RegClass.
1514739a0038SAndrew Trick   const ArrayRef<CodeGenRegisterClass*> &RegClasses = getRegClasses();
1515739a0038SAndrew Trick   unsigned NumRegClasses = RegClasses.size();
1516739a0038SAndrew Trick   for (unsigned RCIdx = 0, RCEnd = NumRegClasses; RCIdx != RCEnd; ++RCIdx) {
15170d94c73cSAndrew Trick     if (!RegClasses[RCIdx]->Allocatable)
15180d94c73cSAndrew Trick       continue;
1519739a0038SAndrew Trick 
1520739a0038SAndrew Trick     // Speculatively grow the RegUnitSets to hold the new set.
1521739a0038SAndrew Trick     RegUnitSets.resize(RegUnitSets.size() + 1);
1522739a0038SAndrew Trick     RegUnitSets.back().Name = RegClasses[RCIdx]->getName();
15237d52db98SAndrew Trick 
15247d52db98SAndrew Trick     // Compute a sorted list of units in this class.
152597254150SAndrew Trick     RegClasses[RCIdx]->buildRegUnitSet(RegUnitSets.back().Units);
1526739a0038SAndrew Trick 
1527739a0038SAndrew Trick     // Find an existing RegUnitSet.
1528739a0038SAndrew Trick     std::vector<RegUnitSet>::const_iterator SetI =
1529739a0038SAndrew Trick       findRegUnitSet(RegUnitSets, RegUnitSets.back());
1530739a0038SAndrew Trick     if (SetI != llvm::prior(RegUnitSets.end()))
1531739a0038SAndrew Trick       RegUnitSets.pop_back();
1532739a0038SAndrew Trick   }
1533739a0038SAndrew Trick 
1534739a0038SAndrew Trick   // Iteratively prune unit sets.
1535739a0038SAndrew Trick   pruneUnitSets();
1536739a0038SAndrew Trick 
1537739a0038SAndrew Trick   // Iterate over all unit sets, including new ones added by this loop.
1538739a0038SAndrew Trick   unsigned NumRegUnitSubSets = RegUnitSets.size();
1539739a0038SAndrew Trick   for (unsigned Idx = 0, EndIdx = RegUnitSets.size(); Idx != EndIdx; ++Idx) {
1540739a0038SAndrew Trick     // In theory, this is combinatorial. In practice, it needs to be bounded
1541739a0038SAndrew Trick     // by a small number of sets for regpressure to be efficient.
1542739a0038SAndrew Trick     // If the assert is hit, we need to implement pruning.
1543739a0038SAndrew Trick     assert(Idx < (2*NumRegUnitSubSets) && "runaway unit set inference");
1544739a0038SAndrew Trick 
1545739a0038SAndrew Trick     // Compare new sets with all original classes.
1546f8b1a666SAndrew Trick     for (unsigned SearchIdx = (Idx >= NumRegUnitSubSets) ? 0 : Idx+1;
1547739a0038SAndrew Trick          SearchIdx != EndIdx; ++SearchIdx) {
1548739a0038SAndrew Trick       std::set<unsigned> Intersection;
1549739a0038SAndrew Trick       std::set_intersection(RegUnitSets[Idx].Units.begin(),
1550739a0038SAndrew Trick                             RegUnitSets[Idx].Units.end(),
1551739a0038SAndrew Trick                             RegUnitSets[SearchIdx].Units.begin(),
1552739a0038SAndrew Trick                             RegUnitSets[SearchIdx].Units.end(),
1553739a0038SAndrew Trick                             std::inserter(Intersection, Intersection.begin()));
1554739a0038SAndrew Trick       if (Intersection.empty())
1555739a0038SAndrew Trick         continue;
1556739a0038SAndrew Trick 
1557739a0038SAndrew Trick       // Speculatively grow the RegUnitSets to hold the new set.
1558739a0038SAndrew Trick       RegUnitSets.resize(RegUnitSets.size() + 1);
1559739a0038SAndrew Trick       RegUnitSets.back().Name =
1560739a0038SAndrew Trick         RegUnitSets[Idx].Name + "+" + RegUnitSets[SearchIdx].Name;
1561739a0038SAndrew Trick 
1562739a0038SAndrew Trick       std::set_union(RegUnitSets[Idx].Units.begin(),
1563739a0038SAndrew Trick                      RegUnitSets[Idx].Units.end(),
1564739a0038SAndrew Trick                      RegUnitSets[SearchIdx].Units.begin(),
1565739a0038SAndrew Trick                      RegUnitSets[SearchIdx].Units.end(),
1566739a0038SAndrew Trick                      std::inserter(RegUnitSets.back().Units,
1567739a0038SAndrew Trick                                    RegUnitSets.back().Units.begin()));
1568739a0038SAndrew Trick 
1569739a0038SAndrew Trick       // Find an existing RegUnitSet, or add the union to the unique sets.
1570739a0038SAndrew Trick       std::vector<RegUnitSet>::const_iterator SetI =
1571739a0038SAndrew Trick         findRegUnitSet(RegUnitSets, RegUnitSets.back());
1572739a0038SAndrew Trick       if (SetI != llvm::prior(RegUnitSets.end()))
1573739a0038SAndrew Trick         RegUnitSets.pop_back();
1574739a0038SAndrew Trick     }
1575739a0038SAndrew Trick   }
1576739a0038SAndrew Trick 
15770d94c73cSAndrew Trick   // Iteratively prune unit sets after inferring supersets.
1578739a0038SAndrew Trick   pruneUnitSets();
1579739a0038SAndrew Trick 
1580739a0038SAndrew Trick   // For each register class, list the UnitSets that are supersets.
1581739a0038SAndrew Trick   RegClassUnitSets.resize(NumRegClasses);
1582739a0038SAndrew Trick   for (unsigned RCIdx = 0, RCEnd = NumRegClasses; RCIdx != RCEnd; ++RCIdx) {
15830d94c73cSAndrew Trick     if (!RegClasses[RCIdx]->Allocatable)
15840d94c73cSAndrew Trick       continue;
15850d94c73cSAndrew Trick 
1586739a0038SAndrew Trick     // Recompute the sorted list of units in this class.
1587739a0038SAndrew Trick     std::vector<unsigned> RegUnits;
158897254150SAndrew Trick     RegClasses[RCIdx]->buildRegUnitSet(RegUnits);
1589739a0038SAndrew Trick 
1590739a0038SAndrew Trick     // Don't increase pressure for unallocatable regclasses.
1591739a0038SAndrew Trick     if (RegUnits.empty())
1592739a0038SAndrew Trick       continue;
1593739a0038SAndrew Trick 
1594739a0038SAndrew Trick     // Find all supersets.
1595739a0038SAndrew Trick     for (unsigned USIdx = 0, USEnd = RegUnitSets.size();
1596739a0038SAndrew Trick          USIdx != USEnd; ++USIdx) {
1597739a0038SAndrew Trick       if (isRegUnitSubSet(RegUnits, RegUnitSets[USIdx].Units))
1598739a0038SAndrew Trick         RegClassUnitSets[RCIdx].push_back(USIdx);
1599739a0038SAndrew Trick     }
16000d94c73cSAndrew Trick     assert(!RegClassUnitSets[RCIdx].empty() && "missing unit set for regclass");
1601739a0038SAndrew Trick   }
1602510e606eSAndrew Trick 
1603510e606eSAndrew Trick   // For each register unit, ensure that we have the list of UnitSets that
1604510e606eSAndrew Trick   // contain the unit. Normally, this matches an existing list of UnitSets for a
1605510e606eSAndrew Trick   // register class. If not, we create a new entry in RegClassUnitSets as a
1606510e606eSAndrew Trick   // "fake" register class.
1607510e606eSAndrew Trick   for (unsigned UnitIdx = 0, UnitEnd = NumNativeRegUnits;
1608510e606eSAndrew Trick        UnitIdx < UnitEnd; ++UnitIdx) {
1609510e606eSAndrew Trick     std::vector<unsigned> RUSets;
1610510e606eSAndrew Trick     for (unsigned i = 0, e = RegUnitSets.size(); i != e; ++i) {
1611510e606eSAndrew Trick       RegUnitSet &RUSet = RegUnitSets[i];
1612510e606eSAndrew Trick       if (std::find(RUSet.Units.begin(), RUSet.Units.end(), UnitIdx)
1613510e606eSAndrew Trick           == RUSet.Units.end())
1614510e606eSAndrew Trick         continue;
1615510e606eSAndrew Trick       RUSets.push_back(i);
1616510e606eSAndrew Trick     }
1617510e606eSAndrew Trick     unsigned RCUnitSetsIdx = 0;
1618510e606eSAndrew Trick     for (unsigned e = RegClassUnitSets.size();
1619510e606eSAndrew Trick          RCUnitSetsIdx != e; ++RCUnitSetsIdx) {
1620510e606eSAndrew Trick       if (RegClassUnitSets[RCUnitSetsIdx] == RUSets) {
1621510e606eSAndrew Trick         break;
1622510e606eSAndrew Trick       }
1623510e606eSAndrew Trick     }
1624510e606eSAndrew Trick     RegUnits[UnitIdx].RegClassUnitSetsIdx = RCUnitSetsIdx;
1625510e606eSAndrew Trick     if (RCUnitSetsIdx == RegClassUnitSets.size()) {
1626510e606eSAndrew Trick       // Create a new list of UnitSets as a "fake" register class.
1627510e606eSAndrew Trick       RegClassUnitSets.resize(RCUnitSetsIdx + 1);
1628510e606eSAndrew Trick       RegClassUnitSets[RCUnitSetsIdx].swap(RUSets);
1629510e606eSAndrew Trick     }
1630510e606eSAndrew Trick   }
1631739a0038SAndrew Trick }
1632739a0038SAndrew Trick 
163384bd44ebSJakob Stoklund Olesen void CodeGenRegBank::computeDerivedInfo() {
163484bd44ebSJakob Stoklund Olesen   computeComposites();
1635d346d487SJakob Stoklund Olesen   computeSubRegIndexLaneMasks();
16361d7a2c57SAndrew Trick 
16371d7a2c57SAndrew Trick   // Compute a weight for each register unit created during getSubRegs.
16381d7a2c57SAndrew Trick   // This may create adopted register units (with unit # >= NumNativeRegUnits).
16391d7a2c57SAndrew Trick   computeRegUnitWeights();
1640739a0038SAndrew Trick 
1641739a0038SAndrew Trick   // Compute a unique set of RegUnitSets. One for each RegClass and inferred
1642739a0038SAndrew Trick   // supersets for the union of overlapping sets.
1643739a0038SAndrew Trick   computeRegUnitSets();
164484bd44ebSJakob Stoklund Olesen }
164584bd44ebSJakob Stoklund Olesen 
1646c0f97e3dSJakob Stoklund Olesen //
1647c0f97e3dSJakob Stoklund Olesen // Synthesize missing register class intersections.
1648c0f97e3dSJakob Stoklund Olesen //
1649c0f97e3dSJakob Stoklund Olesen // Make sure that sub-classes of RC exists such that getCommonSubClass(RC, X)
1650c0f97e3dSJakob Stoklund Olesen // returns a maximal register class for all X.
1651c0f97e3dSJakob Stoklund Olesen //
1652c0f97e3dSJakob Stoklund Olesen void CodeGenRegBank::inferCommonSubClass(CodeGenRegisterClass *RC) {
1653c0f97e3dSJakob Stoklund Olesen   for (unsigned rci = 0, rce = RegClasses.size(); rci != rce; ++rci) {
1654c0f97e3dSJakob Stoklund Olesen     CodeGenRegisterClass *RC1 = RC;
1655c0f97e3dSJakob Stoklund Olesen     CodeGenRegisterClass *RC2 = RegClasses[rci];
1656c0f97e3dSJakob Stoklund Olesen     if (RC1 == RC2)
1657c0f97e3dSJakob Stoklund Olesen       continue;
1658c0f97e3dSJakob Stoklund Olesen 
1659c0f97e3dSJakob Stoklund Olesen     // Compute the set intersection of RC1 and RC2.
1660c0f97e3dSJakob Stoklund Olesen     const CodeGenRegister::Set &Memb1 = RC1->getMembers();
1661c0f97e3dSJakob Stoklund Olesen     const CodeGenRegister::Set &Memb2 = RC2->getMembers();
1662c0f97e3dSJakob Stoklund Olesen     CodeGenRegister::Set Intersection;
1663c0f97e3dSJakob Stoklund Olesen     std::set_intersection(Memb1.begin(), Memb1.end(),
1664c0f97e3dSJakob Stoklund Olesen                           Memb2.begin(), Memb2.end(),
1665f94cd193SJakob Stoklund Olesen                           std::inserter(Intersection, Intersection.begin()),
1666f94cd193SJakob Stoklund Olesen                           CodeGenRegister::Less());
1667c0f97e3dSJakob Stoklund Olesen 
1668c0f97e3dSJakob Stoklund Olesen     // Skip disjoint class pairs.
1669c0f97e3dSJakob Stoklund Olesen     if (Intersection.empty())
1670c0f97e3dSJakob Stoklund Olesen       continue;
1671c0f97e3dSJakob Stoklund Olesen 
1672c0f97e3dSJakob Stoklund Olesen     // If RC1 and RC2 have different spill sizes or alignments, use the
1673c0f97e3dSJakob Stoklund Olesen     // larger size for sub-classing.  If they are equal, prefer RC1.
1674c0f97e3dSJakob Stoklund Olesen     if (RC2->SpillSize > RC1->SpillSize ||
1675c0f97e3dSJakob Stoklund Olesen         (RC2->SpillSize == RC1->SpillSize &&
1676c0f97e3dSJakob Stoklund Olesen          RC2->SpillAlignment > RC1->SpillAlignment))
1677c0f97e3dSJakob Stoklund Olesen       std::swap(RC1, RC2);
1678c0f97e3dSJakob Stoklund Olesen 
1679c0f97e3dSJakob Stoklund Olesen     getOrCreateSubClass(RC1, &Intersection,
1680c0f97e3dSJakob Stoklund Olesen                         RC1->getName() + "_and_" + RC2->getName());
1681c0f97e3dSJakob Stoklund Olesen   }
1682c0f97e3dSJakob Stoklund Olesen }
1683c0f97e3dSJakob Stoklund Olesen 
168403efe84dSJakob Stoklund Olesen //
16856a5f0a19SJakob Stoklund Olesen // Synthesize missing sub-classes for getSubClassWithSubReg().
16866a5f0a19SJakob Stoklund Olesen //
16876a5f0a19SJakob Stoklund Olesen // Make sure that the set of registers in RC with a given SubIdx sub-register
16886a5f0a19SJakob Stoklund Olesen // form a register class.  Update RC->SubClassWithSubReg.
16896a5f0a19SJakob Stoklund Olesen //
16906a5f0a19SJakob Stoklund Olesen void CodeGenRegBank::inferSubClassWithSubReg(CodeGenRegisterClass *RC) {
16916a5f0a19SJakob Stoklund Olesen   // Map SubRegIndex to set of registers in RC supporting that SubRegIndex.
1692f1bb1519SJakob Stoklund Olesen   typedef std::map<CodeGenSubRegIndex*, CodeGenRegister::Set,
1693f1bb1519SJakob Stoklund Olesen                    CodeGenSubRegIndex::Less> SubReg2SetMap;
169403efe84dSJakob Stoklund Olesen 
169503efe84dSJakob Stoklund Olesen   // Compute the set of registers supporting each SubRegIndex.
169603efe84dSJakob Stoklund Olesen   SubReg2SetMap SRSets;
16976a5f0a19SJakob Stoklund Olesen   for (CodeGenRegister::Set::const_iterator RI = RC->getMembers().begin(),
16986a5f0a19SJakob Stoklund Olesen        RE = RC->getMembers().end(); RI != RE; ++RI) {
1699b1147c46SJakob Stoklund Olesen     const CodeGenRegister::SubRegMap &SRM = (*RI)->getSubRegs();
1700b1147c46SJakob Stoklund Olesen     for (CodeGenRegister::SubRegMap::const_iterator I = SRM.begin(),
1701b1147c46SJakob Stoklund Olesen          E = SRM.end(); I != E; ++I)
170203efe84dSJakob Stoklund Olesen       SRSets[I->first].insert(*RI);
170303efe84dSJakob Stoklund Olesen   }
170403efe84dSJakob Stoklund Olesen 
170503efe84dSJakob Stoklund Olesen   // Find matching classes for all SRSets entries.  Iterate in SubRegIndex
170603efe84dSJakob Stoklund Olesen   // numerical order to visit synthetic indices last.
170703efe84dSJakob Stoklund Olesen   for (unsigned sri = 0, sre = SubRegIndices.size(); sri != sre; ++sri) {
1708f1bb1519SJakob Stoklund Olesen     CodeGenSubRegIndex *SubIdx = SubRegIndices[sri];
17093a541b04SJakob Stoklund Olesen     SubReg2SetMap::const_iterator I = SRSets.find(SubIdx);
171003efe84dSJakob Stoklund Olesen     // Unsupported SubRegIndex. Skip it.
171103efe84dSJakob Stoklund Olesen     if (I == SRSets.end())
171203efe84dSJakob Stoklund Olesen       continue;
17133a541b04SJakob Stoklund Olesen     // In most cases, all RC registers support the SubRegIndex.
17146a5f0a19SJakob Stoklund Olesen     if (I->second.size() == RC->getMembers().size()) {
17156a5f0a19SJakob Stoklund Olesen       RC->setSubClassWithSubReg(SubIdx, RC);
171603efe84dSJakob Stoklund Olesen       continue;
17173a541b04SJakob Stoklund Olesen     }
171803efe84dSJakob Stoklund Olesen     // This is a real subset.  See if we have a matching class.
17197ebc6b05SJakob Stoklund Olesen     CodeGenRegisterClass *SubRC =
17206a5f0a19SJakob Stoklund Olesen       getOrCreateSubClass(RC, &I->second,
17216a5f0a19SJakob Stoklund Olesen                           RC->getName() + "_with_" + I->first->getName());
17226a5f0a19SJakob Stoklund Olesen     RC->setSubClassWithSubReg(SubIdx, SubRC);
17236a5f0a19SJakob Stoklund Olesen   }
172403efe84dSJakob Stoklund Olesen }
1725c0f97e3dSJakob Stoklund Olesen 
17266a5f0a19SJakob Stoklund Olesen //
1727b92f557cSJakob Stoklund Olesen // Synthesize missing sub-classes of RC for getMatchingSuperRegClass().
1728b92f557cSJakob Stoklund Olesen //
1729b92f557cSJakob Stoklund Olesen // Create sub-classes of RC such that getMatchingSuperRegClass(RC, SubIdx, X)
1730b92f557cSJakob Stoklund Olesen // has a maximal result for any SubIdx and any X >= FirstSubRegRC.
1731b92f557cSJakob Stoklund Olesen //
1732b92f557cSJakob Stoklund Olesen 
1733b92f557cSJakob Stoklund Olesen void CodeGenRegBank::inferMatchingSuperRegClass(CodeGenRegisterClass *RC,
1734b92f557cSJakob Stoklund Olesen                                                 unsigned FirstSubRegRC) {
1735b92f557cSJakob Stoklund Olesen   SmallVector<std::pair<const CodeGenRegister*,
1736b92f557cSJakob Stoklund Olesen                         const CodeGenRegister*>, 16> SSPairs;
173750ecd0ffSJakob Stoklund Olesen   BitVector TopoSigs(getNumTopoSigs());
1738b92f557cSJakob Stoklund Olesen 
1739b92f557cSJakob Stoklund Olesen   // Iterate in SubRegIndex numerical order to visit synthetic indices last.
1740b92f557cSJakob Stoklund Olesen   for (unsigned sri = 0, sre = SubRegIndices.size(); sri != sre; ++sri) {
1741f1bb1519SJakob Stoklund Olesen     CodeGenSubRegIndex *SubIdx = SubRegIndices[sri];
1742b92f557cSJakob Stoklund Olesen     // Skip indexes that aren't fully supported by RC's registers. This was
1743b92f557cSJakob Stoklund Olesen     // computed by inferSubClassWithSubReg() above which should have been
1744b92f557cSJakob Stoklund Olesen     // called first.
1745b92f557cSJakob Stoklund Olesen     if (RC->getSubClassWithSubReg(SubIdx) != RC)
1746b92f557cSJakob Stoklund Olesen       continue;
1747b92f557cSJakob Stoklund Olesen 
1748b92f557cSJakob Stoklund Olesen     // Build list of (Super, Sub) pairs for this SubIdx.
1749b92f557cSJakob Stoklund Olesen     SSPairs.clear();
175050ecd0ffSJakob Stoklund Olesen     TopoSigs.reset();
1751b92f557cSJakob Stoklund Olesen     for (CodeGenRegister::Set::const_iterator RI = RC->getMembers().begin(),
1752b92f557cSJakob Stoklund Olesen          RE = RC->getMembers().end(); RI != RE; ++RI) {
1753b92f557cSJakob Stoklund Olesen       const CodeGenRegister *Super = *RI;
1754b92f557cSJakob Stoklund Olesen       const CodeGenRegister *Sub = Super->getSubRegs().find(SubIdx)->second;
1755b92f557cSJakob Stoklund Olesen       assert(Sub && "Missing sub-register");
1756b92f557cSJakob Stoklund Olesen       SSPairs.push_back(std::make_pair(Super, Sub));
175750ecd0ffSJakob Stoklund Olesen       TopoSigs.set(Sub->getTopoSig());
1758b92f557cSJakob Stoklund Olesen     }
1759b92f557cSJakob Stoklund Olesen 
1760b92f557cSJakob Stoklund Olesen     // Iterate over sub-register class candidates.  Ignore classes created by
1761b92f557cSJakob Stoklund Olesen     // this loop. They will never be useful.
1762b92f557cSJakob Stoklund Olesen     for (unsigned rci = FirstSubRegRC, rce = RegClasses.size(); rci != rce;
1763b92f557cSJakob Stoklund Olesen          ++rci) {
1764b92f557cSJakob Stoklund Olesen       CodeGenRegisterClass *SubRC = RegClasses[rci];
176550ecd0ffSJakob Stoklund Olesen       // Topological shortcut: SubRC members have the wrong shape.
176650ecd0ffSJakob Stoklund Olesen       if (!TopoSigs.anyCommon(SubRC->getTopoSigs()))
176750ecd0ffSJakob Stoklund Olesen         continue;
1768b92f557cSJakob Stoklund Olesen       // Compute the subset of RC that maps into SubRC.
1769b92f557cSJakob Stoklund Olesen       CodeGenRegister::Set SubSet;
1770b92f557cSJakob Stoklund Olesen       for (unsigned i = 0, e = SSPairs.size(); i != e; ++i)
1771b92f557cSJakob Stoklund Olesen         if (SubRC->contains(SSPairs[i].second))
1772b92f557cSJakob Stoklund Olesen           SubSet.insert(SSPairs[i].first);
1773b92f557cSJakob Stoklund Olesen       if (SubSet.empty())
1774b92f557cSJakob Stoklund Olesen         continue;
1775b92f557cSJakob Stoklund Olesen       // RC injects completely into SubRC.
1776c7b437aeSJakob Stoklund Olesen       if (SubSet.size() == SSPairs.size()) {
1777c7b437aeSJakob Stoklund Olesen         SubRC->addSuperRegClass(SubIdx, RC);
1778b92f557cSJakob Stoklund Olesen         continue;
1779c7b437aeSJakob Stoklund Olesen       }
1780b92f557cSJakob Stoklund Olesen       // Only a subset of RC maps into SubRC. Make sure it is represented by a
1781b92f557cSJakob Stoklund Olesen       // class.
1782b92f557cSJakob Stoklund Olesen       getOrCreateSubClass(RC, &SubSet, RC->getName() +
1783b92f557cSJakob Stoklund Olesen                           "_with_" + SubIdx->getName() +
1784b92f557cSJakob Stoklund Olesen                           "_in_" + SubRC->getName());
1785b92f557cSJakob Stoklund Olesen     }
1786b92f557cSJakob Stoklund Olesen   }
1787b92f557cSJakob Stoklund Olesen }
1788b92f557cSJakob Stoklund Olesen 
1789b92f557cSJakob Stoklund Olesen 
1790b92f557cSJakob Stoklund Olesen //
17916a5f0a19SJakob Stoklund Olesen // Infer missing register classes.
17926a5f0a19SJakob Stoklund Olesen //
17936a5f0a19SJakob Stoklund Olesen void CodeGenRegBank::computeInferredRegisterClasses() {
17946a5f0a19SJakob Stoklund Olesen   // When this function is called, the register classes have not been sorted
17956a5f0a19SJakob Stoklund Olesen   // and assigned EnumValues yet.  That means getSubClasses(),
17966a5f0a19SJakob Stoklund Olesen   // getSuperClasses(), and hasSubClass() functions are defunct.
1797b92f557cSJakob Stoklund Olesen   unsigned FirstNewRC = RegClasses.size();
17986a5f0a19SJakob Stoklund Olesen 
17996a5f0a19SJakob Stoklund Olesen   // Visit all register classes, including the ones being added by the loop.
18006a5f0a19SJakob Stoklund Olesen   for (unsigned rci = 0; rci != RegClasses.size(); ++rci) {
18016a5f0a19SJakob Stoklund Olesen     CodeGenRegisterClass *RC = RegClasses[rci];
18026a5f0a19SJakob Stoklund Olesen 
18036a5f0a19SJakob Stoklund Olesen     // Synthesize answers for getSubClassWithSubReg().
18046a5f0a19SJakob Stoklund Olesen     inferSubClassWithSubReg(RC);
18056a5f0a19SJakob Stoklund Olesen 
1806c0f97e3dSJakob Stoklund Olesen     // Synthesize answers for getCommonSubClass().
18076a5f0a19SJakob Stoklund Olesen     inferCommonSubClass(RC);
1808b92f557cSJakob Stoklund Olesen 
1809b92f557cSJakob Stoklund Olesen     // Synthesize answers for getMatchingSuperRegClass().
1810b92f557cSJakob Stoklund Olesen     inferMatchingSuperRegClass(RC);
1811b92f557cSJakob Stoklund Olesen 
1812b92f557cSJakob Stoklund Olesen     // New register classes are created while this loop is running, and we need
1813b92f557cSJakob Stoklund Olesen     // to visit all of them.  I  particular, inferMatchingSuperRegClass needs
1814b92f557cSJakob Stoklund Olesen     // to match old super-register classes with sub-register classes created
1815b92f557cSJakob Stoklund Olesen     // after inferMatchingSuperRegClass was called.  At this point,
1816b92f557cSJakob Stoklund Olesen     // inferMatchingSuperRegClass has checked SuperRC = [0..rci] with SubRC =
1817b92f557cSJakob Stoklund Olesen     // [0..FirstNewRC).  We need to cover SubRC = [FirstNewRC..rci].
1818b92f557cSJakob Stoklund Olesen     if (rci + 1 == FirstNewRC) {
1819b92f557cSJakob Stoklund Olesen       unsigned NextNewRC = RegClasses.size();
1820b92f557cSJakob Stoklund Olesen       for (unsigned rci2 = 0; rci2 != FirstNewRC; ++rci2)
1821b92f557cSJakob Stoklund Olesen         inferMatchingSuperRegClass(RegClasses[rci2], FirstNewRC);
1822b92f557cSJakob Stoklund Olesen       FirstNewRC = NextNewRC;
1823b92f557cSJakob Stoklund Olesen     }
182403efe84dSJakob Stoklund Olesen   }
182503efe84dSJakob Stoklund Olesen }
182603efe84dSJakob Stoklund Olesen 
182722ea424dSJakob Stoklund Olesen /// getRegisterClassForRegister - Find the register class that contains the
182822ea424dSJakob Stoklund Olesen /// specified physical register.  If the register is not in a register class,
182922ea424dSJakob Stoklund Olesen /// return null. If the register is in multiple classes, and the classes have a
183022ea424dSJakob Stoklund Olesen /// superset-subset relationship and the same set of types, return the
183122ea424dSJakob Stoklund Olesen /// superclass.  Otherwise return null.
183222ea424dSJakob Stoklund Olesen const CodeGenRegisterClass*
183322ea424dSJakob Stoklund Olesen CodeGenRegBank::getRegClassForRegister(Record *R) {
1834d7bc5c26SJakob Stoklund Olesen   const CodeGenRegister *Reg = getReg(R);
183519be2ab3SJakob Stoklund Olesen   ArrayRef<CodeGenRegisterClass*> RCs = getRegClasses();
183622ea424dSJakob Stoklund Olesen   const CodeGenRegisterClass *FoundRC = 0;
183722ea424dSJakob Stoklund Olesen   for (unsigned i = 0, e = RCs.size(); i != e; ++i) {
183819be2ab3SJakob Stoklund Olesen     const CodeGenRegisterClass &RC = *RCs[i];
1839d7bc5c26SJakob Stoklund Olesen     if (!RC.contains(Reg))
184022ea424dSJakob Stoklund Olesen       continue;
184122ea424dSJakob Stoklund Olesen 
184222ea424dSJakob Stoklund Olesen     // If this is the first class that contains the register,
184322ea424dSJakob Stoklund Olesen     // make a note of it and go on to the next class.
184422ea424dSJakob Stoklund Olesen     if (!FoundRC) {
184522ea424dSJakob Stoklund Olesen       FoundRC = &RC;
184622ea424dSJakob Stoklund Olesen       continue;
184722ea424dSJakob Stoklund Olesen     }
184822ea424dSJakob Stoklund Olesen 
184922ea424dSJakob Stoklund Olesen     // If a register's classes have different types, return null.
185022ea424dSJakob Stoklund Olesen     if (RC.getValueTypes() != FoundRC->getValueTypes())
185122ea424dSJakob Stoklund Olesen       return 0;
185222ea424dSJakob Stoklund Olesen 
185322ea424dSJakob Stoklund Olesen     // Check to see if the previously found class that contains
185422ea424dSJakob Stoklund Olesen     // the register is a subclass of the current class. If so,
185522ea424dSJakob Stoklund Olesen     // prefer the superclass.
1856d7bc5c26SJakob Stoklund Olesen     if (RC.hasSubClass(FoundRC)) {
185722ea424dSJakob Stoklund Olesen       FoundRC = &RC;
185822ea424dSJakob Stoklund Olesen       continue;
185922ea424dSJakob Stoklund Olesen     }
186022ea424dSJakob Stoklund Olesen 
186122ea424dSJakob Stoklund Olesen     // Check to see if the previously found class that contains
186222ea424dSJakob Stoklund Olesen     // the register is a superclass of the current class. If so,
186322ea424dSJakob Stoklund Olesen     // prefer the superclass.
1864d7bc5c26SJakob Stoklund Olesen     if (FoundRC->hasSubClass(&RC))
186522ea424dSJakob Stoklund Olesen       continue;
186622ea424dSJakob Stoklund Olesen 
186722ea424dSJakob Stoklund Olesen     // Multiple classes, and neither is a superclass of the other.
186822ea424dSJakob Stoklund Olesen     // Return null.
186922ea424dSJakob Stoklund Olesen     return 0;
187022ea424dSJakob Stoklund Olesen   }
187122ea424dSJakob Stoklund Olesen   return FoundRC;
187222ea424dSJakob Stoklund Olesen }
1873c3abb0f6SJakob Stoklund Olesen 
1874c3abb0f6SJakob Stoklund Olesen BitVector CodeGenRegBank::computeCoveredRegisters(ArrayRef<Record*> Regs) {
187500296815SJakob Stoklund Olesen   SetVector<const CodeGenRegister*> Set;
1876c3abb0f6SJakob Stoklund Olesen 
1877c3abb0f6SJakob Stoklund Olesen   // First add Regs with all sub-registers.
1878c3abb0f6SJakob Stoklund Olesen   for (unsigned i = 0, e = Regs.size(); i != e; ++i) {
1879c3abb0f6SJakob Stoklund Olesen     CodeGenRegister *Reg = getReg(Regs[i]);
1880c3abb0f6SJakob Stoklund Olesen     if (Set.insert(Reg))
1881c3abb0f6SJakob Stoklund Olesen       // Reg is new, add all sub-registers.
1882c3abb0f6SJakob Stoklund Olesen       // The pre-ordering is not important here.
1883f1bb1519SJakob Stoklund Olesen       Reg->addSubRegsPreOrder(Set, *this);
1884c3abb0f6SJakob Stoklund Olesen   }
1885c3abb0f6SJakob Stoklund Olesen 
1886c3abb0f6SJakob Stoklund Olesen   // Second, find all super-registers that are completely covered by the set.
1887f43b5995SJakob Stoklund Olesen   for (unsigned i = 0; i != Set.size(); ++i) {
1888f43b5995SJakob Stoklund Olesen     const CodeGenRegister::SuperRegList &SR = Set[i]->getSuperRegs();
1889f43b5995SJakob Stoklund Olesen     for (unsigned j = 0, e = SR.size(); j != e; ++j) {
189000296815SJakob Stoklund Olesen       const CodeGenRegister *Super = SR[j];
1891f43b5995SJakob Stoklund Olesen       if (!Super->CoveredBySubRegs || Set.count(Super))
1892f43b5995SJakob Stoklund Olesen         continue;
1893f43b5995SJakob Stoklund Olesen       // This new super-register is covered by its sub-registers.
1894f43b5995SJakob Stoklund Olesen       bool AllSubsInSet = true;
1895f43b5995SJakob Stoklund Olesen       const CodeGenRegister::SubRegMap &SRM = Super->getSubRegs();
1896f43b5995SJakob Stoklund Olesen       for (CodeGenRegister::SubRegMap::const_iterator I = SRM.begin(),
1897f43b5995SJakob Stoklund Olesen              E = SRM.end(); I != E; ++I)
1898f43b5995SJakob Stoklund Olesen         if (!Set.count(I->second)) {
1899f43b5995SJakob Stoklund Olesen           AllSubsInSet = false;
1900f43b5995SJakob Stoklund Olesen           break;
1901f43b5995SJakob Stoklund Olesen         }
1902f43b5995SJakob Stoklund Olesen       // All sub-registers in Set, add Super as well.
1903f43b5995SJakob Stoklund Olesen       // We will visit Super later to recheck its super-registers.
1904f43b5995SJakob Stoklund Olesen       if (AllSubsInSet)
1905f43b5995SJakob Stoklund Olesen         Set.insert(Super);
1906f43b5995SJakob Stoklund Olesen     }
1907f43b5995SJakob Stoklund Olesen   }
1908c3abb0f6SJakob Stoklund Olesen 
1909c3abb0f6SJakob Stoklund Olesen   // Convert to BitVector.
1910c3abb0f6SJakob Stoklund Olesen   BitVector BV(Registers.size() + 1);
1911c3abb0f6SJakob Stoklund Olesen   for (unsigned i = 0, e = Set.size(); i != e; ++i)
1912c3abb0f6SJakob Stoklund Olesen     BV.set(Set[i]->EnumValue);
1913c3abb0f6SJakob Stoklund Olesen   return BV;
1914c3abb0f6SJakob Stoklund Olesen }
1915