86 строки
3.4 KiB
C++
86 строки
3.4 KiB
C++
///////////////////////////////////////////////////////////////////////////////
|
|
// //
|
|
// PixPassHelpers.h
|
|
// // Copyright (C) Microsoft Corporation. All rights reserved. // This file is
|
|
// distributed under the University of Illinois Open Source // License. See
|
|
// LICENSE.TXT for details. //
|
|
// //
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#pragma once
|
|
|
|
#include <vector>
|
|
|
|
#include "dxc/DXIL/DxilModule.h"
|
|
#include "llvm/IR/DebugInfoMetadata.h"
|
|
#include "llvm/IR/IRBuilder.h"
|
|
#include "llvm/IR/Instructions.h"
|
|
|
|
//#define PIX_DEBUG_DUMP_HELPER
|
|
#ifdef PIX_DEBUG_DUMP_HELPER
|
|
#include "dxc/Support/Global.h"
|
|
#endif
|
|
|
|
namespace PIXPassHelpers {
|
|
|
|
class ScopedInstruction {
|
|
llvm::Instruction *m_Instruction;
|
|
|
|
public:
|
|
ScopedInstruction(llvm::Instruction *I) : m_Instruction(I) {}
|
|
~ScopedInstruction() { delete m_Instruction; }
|
|
llvm::Instruction *Get() const { return m_Instruction; }
|
|
};
|
|
|
|
void FindRayQueryHandlesForFunction(
|
|
llvm::Function *F, llvm::SmallPtrSetImpl<llvm::Value *> &RayQueryHandles);
|
|
enum class PixUAVHandleMode { NonLib, Lib };
|
|
llvm::CallInst *CreateUAVOnceForModule(hlsl::DxilModule &DM,
|
|
llvm::IRBuilder<> &Builder,
|
|
unsigned int hlslBindIndex,
|
|
const char *name);
|
|
hlsl::DxilResource *CreateGlobalUAVResource(hlsl::DxilModule &DM,
|
|
unsigned int hlslBindIndex,
|
|
const char *name);
|
|
llvm::CallInst *CreateHandleForResource(hlsl::DxilModule &DM,
|
|
llvm::IRBuilder<> &Builder,
|
|
hlsl::DxilResourceBase *resource,
|
|
const char *name);
|
|
llvm::Function *GetEntryFunction(hlsl::DxilModule &DM);
|
|
std::vector<llvm::BasicBlock *> GetAllBlocks(hlsl::DxilModule &DM);
|
|
std::vector<llvm::Function *>
|
|
GetAllInstrumentableFunctions(hlsl::DxilModule &DM);
|
|
hlsl::DXIL::ShaderKind GetFunctionShaderKind(hlsl::DxilModule &DM,
|
|
llvm::Function *fn);
|
|
#ifdef PIX_DEBUG_DUMP_HELPER
|
|
void Log(const char *format, ...);
|
|
void LogPartialLine(const char *format, ...);
|
|
void IncreaseLogIndent();
|
|
void DecreaseLogIndent();
|
|
void DumpFullType(llvm::DIType const *type);
|
|
#else
|
|
inline void DumpFullType(llvm::DIType const *) {}
|
|
inline void Log(const char *, ...) {}
|
|
inline void LogPartialLine(const char *format, ...) {}
|
|
inline void IncreaseLogIndent() {}
|
|
inline void DecreaseLogIndent() {}
|
|
#endif
|
|
class ScopedIndenter {
|
|
public:
|
|
ScopedIndenter() { IncreaseLogIndent(); }
|
|
~ScopedIndenter() { DecreaseLogIndent(); }
|
|
};
|
|
|
|
struct ExpandedStruct {
|
|
llvm::Type *ExpandedPayloadStructType = nullptr;
|
|
llvm::Type *ExpandedPayloadStructPtrType = nullptr;
|
|
};
|
|
|
|
ExpandedStruct ExpandStructType(llvm::LLVMContext &Ctx,
|
|
llvm::Type *OriginalPayloadStructType);
|
|
void ReplaceAllUsesOfInstructionWithNewValueAndDeleteInstruction(
|
|
llvm::Instruction *Instr, llvm::Value *newValue, llvm::Type *newType);
|
|
unsigned int FindOrAddSV_Position(hlsl::DxilModule &DM,
|
|
unsigned UpStreamSVPosRow);
|
|
} // namespace PIXPassHelpers
|