10b57cec5SDimitry Andric //===-- lib/CodeGen/GlobalISel/GISelChangeObserver.cpp --------------------===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric // 90b57cec5SDimitry Andric // This file constains common code to combine machine functions at generic 100b57cec5SDimitry Andric // level. 110b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 120b57cec5SDimitry Andric 130b57cec5SDimitry Andric #include "llvm/CodeGen/GlobalISel/GISelChangeObserver.h" 140b57cec5SDimitry Andric #include "llvm/CodeGen/MachineRegisterInfo.h" 150b57cec5SDimitry Andric 160b57cec5SDimitry Andric using namespace llvm; 170b57cec5SDimitry Andric 180b57cec5SDimitry Andric void GISelChangeObserver::changingAllUsesOfReg( 190b57cec5SDimitry Andric const MachineRegisterInfo &MRI, unsigned Reg) { 200b57cec5SDimitry Andric for (auto &ChangingMI : MRI.use_instructions(Reg)) { 210b57cec5SDimitry Andric changingInstr(ChangingMI); 220b57cec5SDimitry Andric ChangingAllUsesOfReg.insert(&ChangingMI); 230b57cec5SDimitry Andric } 240b57cec5SDimitry Andric } 250b57cec5SDimitry Andric 260b57cec5SDimitry Andric void GISelChangeObserver::finishedChangingAllUsesOfReg() { 270b57cec5SDimitry Andric for (auto *ChangedMI : ChangingAllUsesOfReg) 280b57cec5SDimitry Andric changedInstr(*ChangedMI); 290b57cec5SDimitry Andric ChangingAllUsesOfReg.clear(); 300b57cec5SDimitry Andric } 310b57cec5SDimitry Andric 320b57cec5SDimitry Andric RAIIDelegateInstaller::RAIIDelegateInstaller(MachineFunction &MF, 330b57cec5SDimitry Andric MachineFunction::Delegate *Del) 340b57cec5SDimitry Andric : MF(MF), Delegate(Del) { 350b57cec5SDimitry Andric // Register this as the delegate for handling insertions and deletions of 360b57cec5SDimitry Andric // instructions. 370b57cec5SDimitry Andric MF.setDelegate(Del); 380b57cec5SDimitry Andric } 390b57cec5SDimitry Andric 400b57cec5SDimitry Andric RAIIDelegateInstaller::~RAIIDelegateInstaller() { MF.resetDelegate(Delegate); } 41*5ffd83dbSDimitry Andric 42*5ffd83dbSDimitry Andric RAIIMFObserverInstaller::RAIIMFObserverInstaller(MachineFunction &MF, 43*5ffd83dbSDimitry Andric GISelChangeObserver &Observer) 44*5ffd83dbSDimitry Andric : MF(MF) { 45*5ffd83dbSDimitry Andric MF.setObserver(&Observer); 46*5ffd83dbSDimitry Andric } 47*5ffd83dbSDimitry Andric 48*5ffd83dbSDimitry Andric RAIIMFObserverInstaller::~RAIIMFObserverInstaller() { MF.setObserver(nullptr); } 49