1 //===- DXILWriterPass.cpp - Bitcode writing pass --------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // DXILWriterPass implementation.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "DXILWriterPass.h"
14 #include "DXILBitcodeWriter.h"
15 #include "llvm/ADT/DenseMap.h"
16 #include "llvm/ADT/StringRef.h"
17 #include "llvm/Analysis/ModuleSummaryAnalysis.h"
18 #include "llvm/IR/Constants.h"
19 #include "llvm/IR/GlobalVariable.h"
20 #include "llvm/IR/Module.h"
21 #include "llvm/IR/PassManager.h"
22 #include "llvm/InitializePasses.h"
23 #include "llvm/Pass.h"
24 #include "llvm/Support/Alignment.h"
25 #include "llvm/Transforms/Utils/ModuleUtils.h"
26 
27 using namespace llvm;
28 using namespace llvm::dxil;
29 
30 namespace {
31 class WriteDXILPass : public llvm::ModulePass {
32   raw_ostream &OS; // raw_ostream to print on
33 
34 public:
35   static char ID; // Pass identification, replacement for typeid
36   WriteDXILPass() : ModulePass(ID), OS(dbgs()) {
37     initializeWriteDXILPassPass(*PassRegistry::getPassRegistry());
38   }
39 
40   explicit WriteDXILPass(raw_ostream &o) : ModulePass(ID), OS(o) {
41     initializeWriteDXILPassPass(*PassRegistry::getPassRegistry());
42   }
43 
44   StringRef getPassName() const override { return "Bitcode Writer"; }
45 
46   bool runOnModule(Module &M) override {
47     WriteDXILToFile(M, OS);
48     return false;
49   }
50   void getAnalysisUsage(AnalysisUsage &AU) const override {
51     AU.setPreservesAll();
52   }
53 };
54 
55 class EmbedDXILPass : public llvm::ModulePass {
56 public:
57   static char ID; // Pass identification, replacement for typeid
58   EmbedDXILPass() : ModulePass(ID) {
59     initializeEmbedDXILPassPass(*PassRegistry::getPassRegistry());
60   }
61 
62   StringRef getPassName() const override { return "DXIL Embedder"; }
63 
64   bool runOnModule(Module &M) override {
65     std::string Data;
66     llvm::raw_string_ostream OS(Data);
67     WriteDXILToFile(M, OS);
68 
69     Constant *ModuleConstant =
70         ConstantDataArray::get(M.getContext(), arrayRefFromStringRef(Data));
71     auto *GV = new llvm::GlobalVariable(M, ModuleConstant->getType(), true,
72                                         GlobalValue::PrivateLinkage,
73                                         ModuleConstant, "dx.dxil");
74     GV->setSection("DXIL");
75     GV->setAlignment(Align(4));
76     appendToCompilerUsed(M, {GV});
77     return true;
78   }
79 
80   void getAnalysisUsage(AnalysisUsage &AU) const override {
81     AU.setPreservesAll();
82   }
83 };
84 } // namespace
85 
86 char WriteDXILPass::ID = 0;
87 INITIALIZE_PASS_BEGIN(WriteDXILPass, "write-bitcode", "Write Bitcode", false,
88                       true)
89 INITIALIZE_PASS_DEPENDENCY(ModuleSummaryIndexWrapperPass)
90 INITIALIZE_PASS_END(WriteDXILPass, "write-bitcode", "Write Bitcode", false,
91                     true)
92 
93 ModulePass *llvm::createDXILWriterPass(raw_ostream &Str) {
94   return new WriteDXILPass(Str);
95 }
96 
97 char EmbedDXILPass::ID = 0;
98 INITIALIZE_PASS(EmbedDXILPass, "dxil-embed", "Embed DXIL", false, true)
99 
100 ModulePass *llvm::createDXILEmbedderPass() { return new EmbedDXILPass(); }
101