1 //===-- llvm/CodeGen/SSARegMap.h --------------------------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file was developed by the LLVM research group and is distributed under
6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // Map register numbers to register classes that are correctly sized (typed) to
11 // hold the information. Assists register allocation. Contained by
12 // MachineFunction, should be deleted by register allocator when it is no
15 //===----------------------------------------------------------------------===//
17 #ifndef LLVM_CODEGEN_SSAREGMAP_H
18 #define LLVM_CODEGEN_SSAREGMAP_H
20 #include "llvm/Target/MRegisterInfo.h"
21 #include "llvm/ADT/DenseMap.h"
25 class TargetRegisterClass;
28 DenseMap<const TargetRegisterClass*, VirtReg2IndexFunctor> RegClassMap;
32 SSARegMap() : NextRegNum(MRegisterInfo::FirstVirtualRegister) { }
34 const TargetRegisterClass* getRegClass(unsigned Reg) {
35 return RegClassMap[Reg];
38 /// createVirtualRegister - Create and return a new virtual register in the
39 /// function with the specified register class.
41 unsigned createVirtualRegister(const TargetRegisterClass *RegClass) {
42 assert(RegClass && "Cannot create register without RegClass!");
43 RegClassMap.grow(NextRegNum);
44 RegClassMap[NextRegNum] = RegClass;
48 unsigned getLastVirtReg() const {
49 return NextRegNum - 1;
53 } // End llvm namespace