/*========================== begin_copyright_notice ============================

Copyright (C) 2017-2023 Intel Corporation

SPDX-License-Identifier: MIT

============================= end_copyright_notice ===========================*/

/*========================== begin_copyright_notice ============================

This file is distributed under the University of Illinois Open Source License.
See LICENSE.TXT for details.

============================= end_copyright_notice ===========================*/

///////////////////////////////////////////////////////////////////////////////
// This file is based on llvm-3.4\lib\CodeGen\AsmPrinter\DwarfCompilerUnit.h
///////////////////////////////////////////////////////////////////////////////

#pragma once

// clang-format off
#include "common/LLVMWarningsPush.hpp"
#include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/SmallVector.h"
#include <llvm/ADT/ArrayRef.h>
#include "llvm/ADT/StringMap.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Config/llvm-config.h"
#include "llvm/IR/DebugInfo.h"
#include "llvm/IR/GlobalValue.h"
#include "common/LLVMWarningsPop.hpp"
#include <llvmWrapper/ADT/Optional.h>
// clang-format on

#include "DIE.hpp"
#include "DwarfDebug.hpp"
#include "EmitterOpts.hpp"
#include "Utils.hpp"

namespace llvm {
class MachineLocation;
class ConstantInt;
class ConstantFP;
class MCExpr;
} // namespace llvm

namespace IGC {
class DbgVariable;

//===----------------------------------------------------------------------===//
/// PieceBuilder - Helper class used to emit location info for register based
/// variables from pieces. Current implementation supports only consecutive
/// registers.
class PieceBuilder {
public:
  struct PieceInfo {
    unsigned regNum;
    uint64_t sizeBits;
    uint64_t offsetBits;
  };

  PieceBuilder(uint16_t RegNum, size_t NumGRFs, uint64_t RegSizeBits, uint64_t VarSizeBits, uint64_t SubRegOffsetBits);
  unsigned pieceCount() const;
  PieceInfo get(unsigned index) const;

private:
  unsigned RegNum;
  size_t NumGRFs;
  uint64_t RegSizeBits;
  uint64_t VariableSizeInBits;
  uint64_t SubRegOffsetInBits;
};

//===----------------------------------------------------------------------===//
/// CompileUnit - This dwarf writer support class manages information associated
/// with a source file.
class CompileUnit {
  /// UniqueID - a numeric ID unique among all CUs in the module
  ///
  unsigned UniqueID;

  /// Node - llvm::MDNode for the compile unit.
  llvm::DICompileUnit *Node;

  /// CUDie - Compile unit debug information entry.
  ///
  DIE *CUDie;

  /// Asm - Target of Dwarf emission.
  StreamEmitter *Asm;

  const DebugEmitterOpts &EmitSettings;

  // Holders for some common dwarf information.
  IGC::DwarfDebug *DD;

  /// IndexTyDie - An anonymous type for index type.  Owned by CUDie.
  DIE *IndexTyDie;

  /// MDNodeToDieMap - Tracks the mapping of unit level debug information
  /// variables to debug information entries.
  llvm::DenseMap<const llvm::MDNode *, DIE *> MDNodeToDieMap;

  /// MDNodeToDIEEntryMap - Tracks the mapping of unit level debug information
  /// descriptors to debug information entries using a DIEEntry proxy.
  llvm::DenseMap<const llvm::MDNode *, DIEEntry *> MDNodeToDIEEntryMap;

  /// DIEBlocks - A list of all the DIEBlocks in use.
  std::vector<DIEBlock *> DIEBlocks;

  /// DIEInlinedStrings - A list of all the DIEInlinedStrings in use.
  std::vector<DIEInlinedString *> DIEInlinedStrings;

  /// ContainingTypeMap - This map is used to keep track of subprogram DIEs that
  /// need DW_AT_containing_type attribute. This attribute points to a DIE that
  /// corresponds to the llvm::MDNode mapped with the subprogram DIE.
  llvm::DenseMap<DIE *, const llvm::MDNode *> ContainingTypeMap;

  // DIEValueAllocator - All DIEValues are allocated through this allocator.
  llvm::BumpPtrAllocator DIEValueAllocator;

  // DIEIntegerOne - A preallocated DIEValue because 1 is used frequently.
  DIEInteger *DIEIntegerOne;

public:
  CompileUnit(unsigned UID, DIE *D, llvm::DICompileUnit *CU, StreamEmitter *A, IGC::DwarfDebug *DW);
  ~CompileUnit();
  CompileUnit(const CompileUnit &) = delete;
  CompileUnit &operator=(const CompileUnit &) = delete;

  using ImportedEntityList = llvm::SmallVector<const llvm::MDNode *, 8>;
  using ImportedEntityMap = llvm::DenseMap<const llvm::MDNode *, ImportedEntityList>;

  ImportedEntityMap ImportedEntities;

  // Accessors.
  unsigned getUniqueID() const { return UniqueID; }
  uint16_t getLanguage() const { return getSourceLanguage(Node, DD->GetVISAModule()->GetModule()); }
  llvm::DICompileUnit *getNode() const { return Node; }
  DIE *getCUDie() const { return CUDie; }

  CompileUnit &getCU() { return *this; }

  unsigned getDebugInfoOffset() const { return DebugInfoOffset; }
  void setDebugInfoOffset(unsigned DbgInfoOff) { DebugInfoOffset = DbgInfoOff; }

  /// hasContent - Return true if this compile unit has something to write out.
  ///
  bool hasContent() const { return !CUDie->getChildren().empty(); }

  /// getParentContextString - Get a string containing the language specific
  /// context for a global name.
  std::string getParentContextString(llvm::DIScope *Context) const;

  /// getDIE - Returns the debug information entry map slot for the
  /// specified debug variable. We delegate the request to DwarfDebug
  /// when the llvm::MDNode can be part of the type system, since DIEs for
  /// the type system can be shared across CUs and the mappings are
  /// kept in DwarfDebug.
  DIE *getDIE(llvm::DINode *D) const;

  DIEBlock *getDIEBlock() { return new (DIEValueAllocator) DIEBlock(); }

  /// A pair of GlobalVariable and DIExpression.
  struct GlobalExpr {
    const llvm::GlobalVariable *Var;
    const llvm::DIExpression *Expr;
  };

  /// insertDIE - Insert DIE into the map. We delegate the request to DwarfDebug
  /// when the llvm::MDNode can be part of the type system, since DIEs for
  /// the type system can be shared across CUs and the mappings are
  /// kept in DwarfDebug.
  void insertDIE(llvm::MDNode *Desc, DIE *D);

  /// addDie - Adds or interns the DIE to the compile unit.
  ///
  void addDie(DIE *Buffer) { CUDie->addChild(Buffer); }

  /// addFlag - Add a flag that is true to the DIE.
  void addFlag(DIE *Die, llvm::dwarf::Attribute Attribute);

  /// addUInt - Add an unsigned integer attribute data and value.
  ///
  void addUInt(DIE *Die, llvm::dwarf::Attribute Attribute, std::optional<llvm::dwarf::Form> Form, uint64_t Integer);

  void addUInt(DIEBlock *Block, llvm::dwarf::Form Form, uint64_t Integer);

  void addBitPiece(IGC::DIEBlock *Block, uint64_t SizeBits, uint64_t OffsetBits);

  /// addSInt - Add an signed integer attribute data and value.
  ///
  void addSInt(DIE *Die, llvm::dwarf::Attribute Attribute, std::optional<llvm::dwarf::Form> Form, int64_t Integer);

  void addSInt(DIEBlock *Die, std::optional<llvm::dwarf::Form> Form, int64_t Integer);

  /// addString - Add a string attribute data and value.
  ///
  void addString(DIE *Die, llvm::dwarf::Attribute Attribute, const llvm::StringRef Str);

  /// addExpr - Add a Dwarf expression attribute data and value.
  ///
  void addExpr(DIEBlock *Die, llvm::dwarf::Form Form, const llvm::MCExpr *Expr);

  /// addLabel - Add a Dwarf label attribute data and value.
  ///
  void addLabel(DIE *Die, llvm::dwarf::Attribute Attribute, llvm::dwarf::Form Form, const llvm::MCSymbol *Label);

  void addLabel(DIEBlock *Die, llvm::dwarf::Form Form, const llvm::MCSymbol *Label);

  /// addLabelAddress - Add a dwarf label attribute data and value using
  /// either DW_FORM_addr, DW_FORM_GNU_addr_index or DW_FORM_addrx.
  ///
  void addLabelAddress(DIE *Die, llvm::dwarf::Attribute Attribute, llvm::MCSymbol *Label);

  // addLabelLoc - Add dwarf label attribute data and value using
  // DW_FORM_sec_offset.
  void addLabelLoc(DIE *Die, llvm::dwarf::Attribute Attribute, llvm::MCSymbol *Label);

  /// addOpAddress - Add a dwarf op address data and value using the
  /// form given and an op of either DW_FORM_addr or DW_FORM_GNU_addr_index.
  ///
  void addOpAddress(DIEBlock *Die, const llvm::MCSymbol *Label);

  /// addDelta - Add a label delta attribute data and value.
  ///
  void addDelta(DIE *Die, llvm::dwarf::Attribute Attribute, llvm::dwarf::Form Form, const llvm::MCSymbol *Hi,
                const llvm::MCSymbol *Lo);

  /// addDIEEntry - Add a DIE attribute data and value.
  ///
  void addDIEEntry(DIE *Die, llvm::dwarf::Attribute Attribute, DIE *Entry);

  /// addDIEEntry - Add a DIE attribute data and value.
  ///
  void addDIEEntry(DIE *Die, llvm::dwarf::Attribute Attribute, DIEEntry *Entry);

  /// addBlock - Add block data.
  ///
  void addBlock(DIE *Die, llvm::dwarf::Attribute Attribute, DIEBlock *Block);

  /// addSourceLine - Add location information to specified debug information
  /// entry.
  void addSourceLine(DIE *Die, llvm::DIScope *S, unsigned Line);
  void addSourceLine(DIE *Die, unsigned Line, const llvm::DIFile *File);
  void addSourceLine(DIE *Die, llvm::DIImportedEntity *IE, unsigned Line);
  void addSourceLine(DIE *Die, llvm::DIVariable *V);
  void addSourceLine(DIE *Die, llvm::DISubprogram *SP);
  void addSourceLine(DIE *Die, llvm::DIType *Ty);
  void addSourceLine(DIE *Die, llvm::DIGlobalVariable *GV);

  /// addConstantValue - Add constant value entry in variable DIE.
  void addConstantValue(DIE *Die, const llvm::ConstantInt *CI, bool Unsigned);
  void addConstantValue(DIE *Die, const llvm::APInt &Val, bool Unsigned);

  /// addConstantFPValue - Add constant value entry in variable DIE.
  void addConstantFPValue(DIE *Die, const llvm::ConstantFP *CFP);

  /// addConstantUValue - Add constant unsigned value entry in variable
  /// DIEBlock.
  void addConstantUValue(DIEBlock *TheDie, uint64_t Val);

  /// addConstantData - Add constant data entry in variable DIE.
  void addConstantData(DIE *Die, const unsigned char *Ptr8, int NumBytes);

  /// addTemplateParams - Add template parameters in buffer.
  void addTemplateParams(DIE &Buffer, llvm::DINodeArray TParams);

  ///  addRegOrConst - Decide whether to emit regx or const
  void addRegOrConst(DIEBlock *TheDie, unsigned DWReg);

  /// addRegisterOp - Add register operand.
  void addRegisterOp(DIEBlock *TheDie, unsigned Reg);

  /// addRegisterOffset - Add register offset.
  void addRegisterOffset(DIEBlock *TheDie, unsigned Reg, int64_t Offset);

  /// addType - Add a new type attribute to the specified entity. This takes
  /// and attribute parameter because DW_AT_friend attributes are also
  /// type references.
  void addType(DIE *Entity, llvm::DIType *Ty, llvm::dwarf::Attribute Attribute = llvm::dwarf::DW_AT_type);

  // addSimdWidth - add SIMD width
  void addSimdWidth(DIE *Die, uint16_t SimdWidth);

  // addBindlessOrStatelessLocation - add a sequence of attributes to calculate
  // stateless or bindless location of variable. baseAddr is one of the
  // following base addreses:
  // - General State Base Address when variable located in stateless surface
  // - Bindless Surface State Base Address when variable located in bindless
  // surface
  // - Bindless Sampler State Base Addres when variable located in bindless
  // sampler
  void addBindlessOrStatelessLocation(DIEBlock *Block, const VISAVariableLocation &Loc, uint32_t baseAddr);

  // addBindlessSamplerLocation - add a sequence of attributes to calculate
  // bindless sampler location of variable
  void addBindlessSamplerLocation(DIEBlock *Block, const VISAVariableLocation &Loc);

  // addBE_FP - emits operations to add contents of BE_FP to current top of
  // dwarf stack
  void addBE_FP(IGC::DIEBlock *Block);

  // addScratchLocation - add a sequence of attributes to emit scratch space
  // location of variable
  void addScratchLocation(DIEBlock *Block, uint32_t memoryOffset, int32_t vectorOffset);

  // addSimdLane - add a sequence of attributes to calculate location of
  // variable among SIMD lanes, e.g. a GRF subregister.
  void addSimdLane(DIEBlock *Block, const DbgVariable &DV, const VISAVariableLocation &Loc,
                   const DbgDecoder::LiveIntervalsVISA *lr, uint16_t regOffset, bool isPacked, bool isSecondHalf);

  // addSimdLaneScalar - add a sequence of attributes to calculate location of
  // scalar variable e.g. a GRF subregister.
  void addSimdLaneScalar(DIEBlock *Block, const DbgVariable &DV, const VISAVariableLocation &Loc,
                         const DbgDecoder::LiveIntervalsVISA &lr);

  // emitBitPiecesForRegVal - emit bitPieces DW_OP_bit_piece sequence for
  // register value. It is used to describe vector variables in registers.
  void emitBitPiecesForRegVal(IGC::DIEBlock *Block, const PieceBuilder &pieceBuilder);

  // addSimdLaneRegionBase - add a sequence of attributes to calculate location
  // of region base address variable for vc-backend
  void addSimdLaneRegionBase(IGC::DIEBlock *Block, const DbgVariable &DV, const VISAVariableLocation &Loc,
                             const DbgDecoder::LiveIntervalsVISA *lr);

  // emit expression to extract sub-reg value
  void extractSubRegValue(IGC::DIEBlock *Block, unsigned char Sz);

  /// Construct import_module DIE.
  IGC::DIE *constructImportedEntityDIE(llvm::DIImportedEntity *Module);

  /// getOrCreateNameSpace - Create a DIE for DINameSpace.
  DIE *getOrCreateNameSpace(llvm::DINamespace *NS);

  /// Get or create global variable DIE.
  DIE *getOrCreateGlobalVariableDIE(llvm::DIGlobalVariable *GV, llvm::ArrayRef<GlobalExpr> GlobalExprs);

  /// getOrCreateSubprogramDIE - Create new DIE using SP.
  DIE *getOrCreateSubprogramDIE(llvm::DISubprogram *SP);

  /// getOrCreateModuleDIE - Create new DIE for DIModule.
  DIE *getOrCreateModuleDIE(llvm::DIModule *MD);

  /// getOrCreateTypeDIE - Find existing DIE or create new DIE for the
  /// given llvm::DIType.
  DIE *getOrCreateTypeDIE(const llvm::MDNode *N);

  /// getOrCreateContextDIE - Get context owner's DIE.
  DIE *getOrCreateContextDIE(llvm::DIScope *Context);

  /// constructContainingTypeDIEs - Construct DIEs for types that contain
  /// vtables.
  void constructContainingTypeDIEs();

  /// constructVariableDIE - Construct a DIE for the given DbgVariable.
  DIE *constructVariableDIE(DbgVariable &DV, bool isScopeAbstract);

  /// Apply attributes for DIE created in constructVariableDIE
  void applyVariableAttributes(DbgVariable &DV, DIE *VariableDie, bool isScopeAbstract, DIE *AbsDIE = nullptr);

  /// Create a DIE with the given Tag, add the DIE to its parent, and
  /// call insertDIE if MD is not null.
  DIE *createAndAddDIE(unsigned Tag, DIE &Parent, llvm::DINode *N = nullptr);

  void addImportedEntity(const llvm::DIImportedEntity *IE) {
    llvm::DIScope *Scope = IE->getScope();

    // Temporary workaround
    if (!Scope) {
      return;
    }
    // assert(Scope && "Invalid Scope encoding!");

    if (!llvm::isa<llvm::DILocalScope>(Scope)) {
      // No need to add imported enities that are not local declaration.
      return;
    }

    auto *LocalScope = llvm::cast<llvm::DILocalScope>(Scope)->getNonLexicalBlockFileScope();
    ImportedEntities[LocalScope].push_back(IE);
  }

  /// Compute the size of a header for this unit, not including the initial
  /// length field.
  unsigned getHeaderSize() const {
    if (DD->getDwarfVersion() > 4) {
      return sizeof(int16_t) + // DWARF version number
             sizeof(int8_t) +  // Unit Type (in bytes)
             sizeof(int8_t) +  // Pointer Size (in bytes)
             sizeof(int32_t);  // Offset Into Abbrev. Section
    } else {
      return sizeof(int16_t) + // DWARF version number
             sizeof(int32_t) + // Offset Into Abbrev. Section
             sizeof(int8_t);   // Pointer Size (in bytes)
    }
  }

  /// Emit the header for this unit, not including the initial length field.
  void emitHeader(const llvm::MCSection *ASection, const llvm::MCSymbol *ASectionSym);

private:
  /// constructTypeDIE - Construct basic type die from llvm::DIBasicType.
  void constructTypeDIE(DIE &Buffer, llvm::DIBasicType *BTy);

  /// constructTypeDIE - Construct derived type die from llvm::DIDerivedType.
  void constructTypeDIE(DIE &Buffer, llvm::DIDerivedType *DTy);

  /// constructTypeDIE - Construct derived type die from llvm::DIStringType.
  void constructTypeDIE(DIE &Buffer, llvm::DIStringType *DTy);

  /// constructTypeDIE - Construct type DIE from DICompositeType.
  void constructTypeDIE(DIE &Buffer, llvm::DICompositeType *CTy);

  /// constructTypeDIE - Construct type DIE from DISubroutineType
  /// This was added for LLVM 3.8 since DISubroutineType is no longer
  /// derived from DICompositeType.
  void constructTypeDIE(DIE &Buffer, llvm::DISubroutineType *STy);

  /// constructSubrangeDIE - Construct subrange DIE from DISubrange.
  void constructSubrangeDIE(DIE &Buffer, llvm::DISubrange *SR, DIE *IndexTy);

  /// constructArrayTypeDIE - Construct array type DIE from DICompositeType.
  void constructArrayTypeDIE(DIE &Buffer, llvm::DICompositeType *CTy);

  /// constructEnumTypeDIE - Construct enum type DIE from DIEnumerator.
  void constructEnumTypeDIE(DIE &Buffer, llvm::DICompositeType *CTy);

  /// constructMemberDIE - Construct member DIE from llvm::DIDerivedType.
  void constructMemberDIE(DIE &Buffer, llvm::DIDerivedType *DT);

  /// constructTemplateTypeParameterDIE - Construct new DIE for the given
  /// llvm::DITemplateTypeParameter.
  void constructTemplateTypeParameterDIE(DIE &Buffer, llvm::DITemplateTypeParameter *TP);

  /// constructTemplateValueParameterDIE - Construct new DIE for the given
  /// llvm::DITemplateValueParameter.
  void constructTemplateValueParameterDIE(DIE &Buffer, llvm::DITemplateValueParameter *TVP);

  /// getOrCreateStaticMemberDIE - Create new static data member DIE.
  DIE *getOrCreateStaticMemberDIE(llvm::DIDerivedType *DT);

  /// Offset of the CUDie from beginning of debug info section.
  unsigned DebugInfoOffset;

  /// getLowerBoundDefault - Return the default lower bound for an array. If the
  /// DWARF version doesn't handle the language, return -1.
  int64_t getDefaultLowerBound() const;

  /// getDIEEntry - Returns the debug information entry for the specified
  /// debug variable.
  DIEEntry *getDIEEntry(const llvm::MDNode *N) const { return MDNodeToDIEEntryMap.lookup(N); }

  /// insertDIEEntry - Insert debug information entry into the map.
  void insertDIEEntry(const llvm::MDNode *N, DIEEntry *E) { MDNodeToDIEEntryMap.insert(std::make_pair(N, E)); }

  // getIndexTyDie - Get an anonymous type for index type.
  DIE *getIndexTyDie() { return IndexTyDie; }

  // setIndexTyDie - Set D as anonymous type for index which can be reused
  // later.
  void setIndexTyDie(DIE *D) { IndexTyDie = D; }

  /// createDIEEntry - Creates a new DIEEntry to be a proxy for a debug
  /// information entry.
  DIEEntry *createDIEEntry(DIE *Entry);

  /// resolve - Look in the DwarfDebug map for the llvm::MDNode that
  /// corresponds to the reference.
  template <typename T> inline T *resolve(T *Ref) const { return DD->resolve(Ref); }

public:
  // Added for 1-step elf
  void buildLocation(const llvm::Instruction *, IGC::DbgVariable &, IGC::DIE *);
  DIEBlock *buildPointer(const DbgVariable &, const VISAVariableLocation &);
  DIEBlock *buildSampler(const DbgVariable &, const VISAVariableLocation &);

  // buildSLM - Build expression for location described as offset in SLM memory.
  DIEBlock *buildSLM(const DbgVariable &, const VISAVariableLocation &, IGC::DIE *);
  DIEBlock *buildGeneral(DbgVariable &, const VISAVariableLocation &,
                         const std::vector<DbgDecoder::LiveIntervalsVISA> *);
  unsigned int stackValueOffset = 0;

private:
  bool buildPrivateBaseRegBased(const DbgVariable &, IGC::DIEBlock *, const VISAVariableLocation &);
  bool buildFpBasedLoc(const DbgVariable &, IGC::DIEBlock *, const VISAVariableLocation &);
  bool buildSlicedLoc(DbgVariable &, IGC::DIEBlock *, const VISAVariableLocation &,
                      const std::vector<DbgDecoder::LiveIntervalsVISA> *);
  bool buildValidVar(DbgVariable &, IGC::DIEBlock *, const VISAVariableLocation &,
                     const std::vector<DbgDecoder::LiveIntervalsVISA> *, DbgRegisterType);

  // Variables, used in buildGeneral-algorithm:
  bool emitLocation = false;
  DIEValue *skipOff = nullptr;
  unsigned int offsetTaken = 0;

  // Hold DISubprogram nodes that correspond to functions with out-of-line
  // definition.
  llvm::SmallPtrSet<llvm::DISubprogram *, 10> ExtFunc;
};

} // namespace IGC
