aboutsummaryrefslogtreecommitdiff
path: root/backend/lib/CodeGen/RegAllocBase.h
blob: 4ef0bb7f917bc05434f440859c49c0f299c2f098 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
// SPDX-License-Identifier: GPL-3.0-or-later

// The private header has no stable include-root-relative guard name.
// NOLINTNEXTLINE(llvm-header-guard)
#ifndef RCC_LIB_CODEGEN_REGALLOCBASE_H
#define RCC_LIB_CODEGEN_REGALLOCBASE_H

#include "RegAllocInternal.h"
#include "rcc/CodeGen/LiveIntervals.h"
#include "rcc/CodeGen/RegAllocCommon.h"
#include "rcc/CodeGen/TargetRegisterAllocationInfo.h"
#include "rcc/CodeGen/VirtRegMap.h"

#include <vector>

namespace rcc {

struct AllocationRoundResult final {
  std::vector<Register> Spilled;
};

class RegAllocBase {
public:
  virtual ~RegAllocBase() = default;

  [[nodiscard]] Expected<regalloc_detail::AllocationRunResult> run();

protected:
  RegAllocBase(MachineFunction &Function, const TargetDescription &Target,
               const TargetRegisterAllocationInfo &TargetInfo,
               MachineFunctionAnalysisManager &Analyses, RegisterAllocationOptions Options,
               MachineActionReplay *Actions = nullptr) noexcept
      : Function(Function), Target(Target), TargetInfo(TargetInfo), Analyses(Analyses),
        Options(Options), Actions(Actions) {}

  [[nodiscard]] virtual Expected<AllocationRoundResult>
  allocateRound(const LiveIntervalsAnalysis::Result &Intervals, VirtRegMap &Mapping) = 0;

  MachineFunction &Function;
  const TargetDescription &Target;
  const TargetRegisterAllocationInfo &TargetInfo;
  MachineFunctionAnalysisManager &Analyses;
  RegisterAllocationOptions Options;
  MachineActionReplay *Actions = nullptr;
};

[[nodiscard]] std::vector<const LiveInterval *>
orderedAllocationIntervals(const LiveIntervalsAnalysis::Result &Intervals,
                           const MachineFunction &Function);

} // namespace rcc

#endif // RCC_LIB_CODEGEN_REGALLOCBASE_H