aboutsummaryrefslogtreecommitdiff
path: root/backend/include/rcc/CodeGen/MachineCFGUpdater.h
blob: 435953845c1aecf6095a36d5a2943ede86abb0a8 (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
// SPDX-License-Identifier: GPL-3.0-or-later

#ifndef RCC_CODEGEN_MACHINECFGUPDATER_H
#define RCC_CODEGEN_MACHINECFGUPDATER_H

#include "rcc/CodeGen/MachineIRTypes.h"
#include "rcc/Support/Error.h"

namespace rcc {

class MachineBasicBlock;
class MachineFunction;

class MachineCFGUpdater final {
public:
  explicit MachineCFGUpdater(MachineFunction &Function) noexcept : Function(&Function) {}

  [[nodiscard]] Expected<void>
  addSuccessor(MachineBasicBlock &From, MachineBasicBlock &To,
               BranchProbability Probability = BranchProbability::zero());
  [[nodiscard]] Expected<void> removeSuccessor(MachineBasicBlock &From, MachineBasicBlock &To);
  [[nodiscard]] Expected<void> replaceSuccessor(MachineBasicBlock &From,
                                                MachineBasicBlock &OldTarget,
                                                MachineBasicBlock &NewTarget);
  [[nodiscard]] Expected<void> updateBranchProbability(MachineBasicBlock &From,
                                                       MachineBasicBlock &To,
                                                       BranchProbability Probability);
  [[nodiscard]] Expected<void> transferSuccessors(MachineBasicBlock &From, MachineBasicBlock &To);
  [[nodiscard]] Expected<void> disconnectBlock(MachineBasicBlock &Block);

private:
  [[nodiscard]] Expected<void> verifyOwnership(const MachineBasicBlock &Block) const;

  MachineFunction *Function;
};

} // namespace rcc

#endif // RCC_CODEGEN_MACHINECFGUPDATER_H