//===- AArch64InstrInfo.h - AArch64 Instruction Information -----*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file contains the AArch64 implementation of the TargetInstrInfo class.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_TARGET_AARCH64INSTRINFO_H
#define LLVM_TARGET_AARCH64INSTRINFO_H

#include "llvm/Target/TargetInstrInfo.h"
#include "AArch64RegisterInfo.h"

#define GET_INSTRINFO_HEADER
#include "AArch64GenInstrInfo.inc"

namespace llvm {

class AArch64Subtarget;

class AArch64InstrInfo : public AArch64GenInstrInfo {
  const AArch64RegisterInfo RI;
  const AArch64Subtarget &Subtarget;
public:
  explicit AArch64InstrInfo(const AArch64Subtarget &TM);

  /// getRegisterInfo - TargetInstrInfo is a superset of MRegister info.  As
  /// such, whenever a client has an instance of instruction info, it should
  /// always be able to get register info as well (through this method).
  ///
  const TargetRegisterInfo &getRegisterInfo() const { return RI; }

  const AArch64Subtarget &getSubTarget() const { return Subtarget; }

  void copyPhysReg(MachineBasicBlock &MBB,
                   MachineBasicBlock::iterator I, DebugLoc DL,
                   unsigned DestReg, unsigned SrcReg,
                   bool KillSrc) const;

  void storeRegToStackSlot(MachineBasicBlock &MBB,
                           MachineBasicBlock::iterator MI,
                           unsigned SrcReg, bool isKill, int FrameIndex,
                           const TargetRegisterClass *RC,
                           const TargetRegisterInfo *TRI) const;
  void loadRegFromStackSlot(MachineBasicBlock &MBB,
                            MachineBasicBlock::iterator MBBI,
                            unsigned DestReg, int FrameIdx,
                            const TargetRegisterClass *RC,
                            const TargetRegisterInfo *TRI) const;

  bool AnalyzeBranch(MachineBasicBlock &MBB, MachineBasicBlock *&TBB,
                     MachineBasicBlock *&FBB,
                     SmallVectorImpl<MachineOperand> &Cond,
                     bool AllowModify = false) const;
  unsigned InsertBranch(MachineBasicBlock &MBB, MachineBasicBlock *TBB,
                        MachineBasicBlock *FBB,
                        const SmallVectorImpl<MachineOperand> &Cond,
                        DebugLoc DL) const;
  unsigned RemoveBranch(MachineBasicBlock &MBB) const;
  bool ReverseBranchCondition(SmallVectorImpl<MachineOperand> &Cond) const;

  bool expandPostRAPseudo(MachineBasicBlock::iterator MI) const;

  /// Look through the instructions in this function and work out the largest
  /// the stack frame can be while maintaining the ability to address local
  /// slots with no complexities.
  unsigned estimateRSStackLimit(MachineFunction &MF) const;

  /// getAddressConstraints - For loads and stores (and PRFMs) taking an
  /// immediate offset, this function determines the constraints required for
  /// the immediate. It must satisfy:
  ///    + MinOffset <= imm <= MaxOffset
  ///    + imm % OffsetScale == 0
  void getAddressConstraints(const MachineInstr &MI, int &AccessScale,
                             int &MinOffset, int &MaxOffset) const;


  unsigned getInstSizeInBytes(const MachineInstr &MI) const;

  unsigned getInstBundleLength(const MachineInstr &MI) const;

};

bool rewriteA64FrameIndex(MachineInstr &MI, unsigned FrameRegIdx,
                          unsigned FrameReg, int &Offset,
                          const AArch64InstrInfo &TII);


void emitRegUpdate(MachineBasicBlock &MBB, MachineBasicBlock::iterator MI,
                   DebugLoc dl, const TargetInstrInfo &TII,
                   unsigned DstReg, unsigned SrcReg, unsigned ScratchReg,
                   int64_t NumBytes,
                   MachineInstr::MIFlag MIFlags = MachineInstr::NoFlags);

void emitSPUpdate(MachineBasicBlock &MBB, MachineBasicBlock::iterator MI,
                  DebugLoc dl, const TargetInstrInfo &TII,
                  unsigned ScratchReg, int64_t NumBytes,
                  MachineInstr::MIFlag MIFlags = MachineInstr::NoFlags);

}

#endif
