1 //===----------------------- CodeRegionGenerator.h --------------*- 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 ///
11 /// This file declares classes responsible for generating llvm-mca
12 /// CodeRegions from various types of input. llvm-mca only analyzes CodeRegions,
13 /// so the classes here provide the input-to-CodeRegions translation.
14 //
15 //===----------------------------------------------------------------------===//
16 
17 #ifndef LLVM_TOOLS_LLVM_MCA_CODEREGION_GENERATOR_H
18 #define LLVM_TOOLS_LLVM_MCA_CODEREGION_GENERATOR_H
19 
20 #include "CodeRegion.h"
21 #include "llvm/MC/MCAsmInfo.h"
22 #include "llvm/MC/MCContext.h"
23 #include "llvm/MC/MCSubtargetInfo.h"
24 #include "llvm/Support/Error.h"
25 #include "llvm/Support/SourceMgr.h"
26 #include "llvm/Support/TargetRegistry.h"
27 #include <memory>
28 
29 namespace llvm {
30 namespace mca {
31 
32 /// This class is responsible for parsing the input given to the llvm-mca
33 /// driver, and converting that into a CodeRegions instance.
34 class CodeRegionGenerator {
35 protected:
36   CodeRegions Regions;
37   CodeRegionGenerator(const CodeRegionGenerator &) = delete;
38   CodeRegionGenerator &operator=(const CodeRegionGenerator &) = delete;
39 
40 public:
CodeRegionGenerator(SourceMgr & SM)41   CodeRegionGenerator(SourceMgr &SM) : Regions(SM) {}
42   virtual ~CodeRegionGenerator();
43   virtual Expected<const CodeRegions &> parseCodeRegions() = 0;
44 };
45 
46 /// This class is responsible for parsing input ASM and generating
47 /// a CodeRegions instance.
48 class AsmCodeRegionGenerator final : public CodeRegionGenerator {
49   const Target &TheTarget;
50   MCContext &Ctx;
51   const MCAsmInfo &MAI;
52   const MCSubtargetInfo &STI;
53   const MCInstrInfo &MCII;
54   unsigned AssemblerDialect; // This is set during parsing.
55 
56 public:
AsmCodeRegionGenerator(const Target & T,SourceMgr & SM,MCContext & C,const MCAsmInfo & A,const MCSubtargetInfo & S,const MCInstrInfo & I)57   AsmCodeRegionGenerator(const Target &T, SourceMgr &SM, MCContext &C,
58                          const MCAsmInfo &A, const MCSubtargetInfo &S,
59                          const MCInstrInfo &I)
60       : CodeRegionGenerator(SM), TheTarget(T), Ctx(C), MAI(A), STI(S), MCII(I),
61         AssemblerDialect(0) {}
62 
getAssemblerDialect()63   unsigned getAssemblerDialect() const { return AssemblerDialect; }
64   Expected<const CodeRegions &> parseCodeRegions() override;
65 };
66 
67 } // namespace mca
68 } // namespace llvm
69 
70 #endif // LLVM_TOOLS_LLVM_MCA_CODEREGION_GENERATOR_H
71