]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/llvm/include/llvm/CodeGen/RegisterUsageInfo.h
Merge llvm trunk r321414 to contrib/llvm.
[FreeBSD/FreeBSD.git] / contrib / llvm / include / llvm / CodeGen / RegisterUsageInfo.h
1 //==- RegisterUsageInfo.h - Register Usage Informartion Storage --*- C++ -*-==//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 /// \file
10 /// This pass is required to take advantage of the interprocedural register
11 /// allocation infrastructure.
12 ///
13 /// This pass is simple immutable pass which keeps RegMasks (calculated based on
14 /// actual register allocation) for functions in a module and provides simple
15 /// API to query this information.
16 ///
17 //===----------------------------------------------------------------------===//
18
19 #ifndef LLVM_CODEGEN_PHYSICALREGISTERUSAGEINFO_H
20 #define LLVM_CODEGEN_PHYSICALREGISTERUSAGEINFO_H
21
22 #include "llvm/ADT/DenseMap.h"
23 #include "llvm/IR/Instructions.h"
24 #include "llvm/Pass.h"
25 #include <cstdint>
26 #include <vector>
27
28 namespace llvm {
29
30 class Function;
31 class TargetMachine;
32
33 class PhysicalRegisterUsageInfo : public ImmutablePass {
34   virtual void anchor();
35
36 public:
37   static char ID;
38
39   PhysicalRegisterUsageInfo() : ImmutablePass(ID) {
40     PassRegistry &Registry = *PassRegistry::getPassRegistry();
41     initializePhysicalRegisterUsageInfoPass(Registry);
42   }
43
44   void getAnalysisUsage(AnalysisUsage &AU) const override {
45     AU.setPreservesAll();
46   }
47
48   /// To set TargetMachine *, which is used to print
49   /// analysis when command line option -print-regusage is used.
50   void setTargetMachine(const TargetMachine *TM_) { TM = TM_; }
51
52   bool doInitialization(Module &M) override;
53
54   bool doFinalization(Module &M) override;
55
56   /// To store RegMask for given Function *.
57   void storeUpdateRegUsageInfo(const Function *FP,
58                                std::vector<uint32_t> RegMask);
59
60   /// To query stored RegMask for given Function *, it will return nullptr if
61   /// function is not known.
62   const std::vector<uint32_t> *getRegUsageInfo(const Function *FP);
63
64   void print(raw_ostream &OS, const Module *M = nullptr) const override;
65
66 private:
67   /// A Dense map from Function * to RegMask.
68   /// In RegMask 0 means register used (clobbered) by function.
69   /// and 1 means content of register will be preserved around function call.
70   DenseMap<const Function *, std::vector<uint32_t>> RegMasks;
71
72   const TargetMachine *TM;
73 };
74
75 } // end namespace llvm
76
77 #endif // LLVM_CODEGEN_PHYSICALREGISTERUSAGEINFO_H