llvm/lib/Target/DirectX/DXILWriter/DXILDebugInfoMap.h

//===- DXILDebugInfoMap.h - DXIL debug info serialization ------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_LIB_TARGET_DIRECTX_DXILWRITER_DXILDEBUGINFOMAP_H
#define LLVM_LIB_TARGET_DIRECTX_DXILWRITER_DXILDEBUGINFOMAP_H

#include "llvm/ADT/DenseMap.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/Instruction.h"
#include <memory>

namespace llvm {
namespace dxil {

class DXILDebugInfoMap {
  struct InstructionDeleter {
    void operator()(Instruction *I) { I->deleteValue(); }
  };

public:
  using FuncMap = DenseMap<const Function *, std::unique_ptr<Function>>;
  using InstMap = DenseMap<const Instruction *,
                           std::unique_ptr<Instruction, InstructionDeleter>>;
  using MDMap = DenseMap<const Metadata *, const Metadata *>;

  DXILDebugInfoMap() = default;
  DXILDebugInfoMap(const DXILDebugInfoMap &) = delete;
  DXILDebugInfoMap(DXILDebugInfoMap &&) = default;

  /// Completely replace one function with another in ValueEnumerator.
  FuncMap FuncReplace;

  /// Completely replace one instruction with another in ValueEnumerator.
  InstMap InstReplace;

  /// Enumerate extra metadata when Key is encountered in ValueEnumerator.
  MDMap MDExtra;

  /// Completely replace one metadata with another in ValueEnumerator.
  MDMap MDReplace;

  const Function &getDXILFunction(const Function &F) const {
    auto It = FuncReplace.find(&F);
    if (It != FuncReplace.end())
      return *It->second.get();
    return F;
  }

  const Instruction &getDXILInstruction(const Instruction &I) const {
    auto It = InstReplace.find(&I);
    if (It != InstReplace.end())
      return *It->second.get();
    return I;
  }

  const Metadata *getDXILMetadata(const Metadata *M) const {
    return MDReplace.lookup_or(M, M);
  }
};

DXILDebugInfoMap collectDXILDebugInfo(Module &M);

} // namespace dxil
} // namespace llvm

#endif // LLVM_LIB_TARGET_DIRECTX_DXILWRITER_DXILDEBUGINFOMAP_H