1 //===- BitcodeWriterPass.cpp - Bitcode writing pass -----------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // BitcodeWriterPass implementation.
12 //===----------------------------------------------------------------------===//
14 #include "llvm/Bitcode/BitcodeWriterPass.h"
15 #include "llvm/Analysis/ModuleSummaryAnalysis.h"
16 #include "llvm/Bitcode/ReaderWriter.h"
17 #include "llvm/IR/Module.h"
18 #include "llvm/IR/PassManager.h"
19 #include "llvm/Pass.h"
22 PreservedAnalyses BitcodeWriterPass::run(Module &M, ModuleAnalysisManager &) {
23 std::unique_ptr<ModuleSummaryIndex> Index;
25 Index = ModuleSummaryIndexBuilder(&M).takeIndex();
26 WriteBitcodeToFile(&M, OS, ShouldPreserveUseListOrder, Index.get(),
28 return PreservedAnalyses::all();
32 class WriteBitcodePass : public ModulePass {
33 raw_ostream &OS; // raw_ostream to print on
34 bool ShouldPreserveUseListOrder;
35 bool EmitSummaryIndex;
39 static char ID; // Pass identification, replacement for typeid
40 WriteBitcodePass() : ModulePass(ID), OS(dbgs()) {
41 initializeWriteBitcodePassPass(*PassRegistry::getPassRegistry());
44 explicit WriteBitcodePass(raw_ostream &o, bool ShouldPreserveUseListOrder,
45 bool EmitSummaryIndex, bool EmitModuleHash)
46 : ModulePass(ID), OS(o),
47 ShouldPreserveUseListOrder(ShouldPreserveUseListOrder),
48 EmitSummaryIndex(EmitSummaryIndex), EmitModuleHash(EmitModuleHash) {
49 initializeWriteBitcodePassPass(*PassRegistry::getPassRegistry());
52 const char *getPassName() const override { return "Bitcode Writer"; }
54 bool runOnModule(Module &M) override {
55 const ModuleSummaryIndex *Index =
57 ? &(getAnalysis<ModuleSummaryIndexWrapperPass>().getIndex())
59 WriteBitcodeToFile(&M, OS, ShouldPreserveUseListOrder, Index,
63 void getAnalysisUsage(AnalysisUsage &AU) const override {
66 AU.addRequired<ModuleSummaryIndexWrapperPass>();
71 char WriteBitcodePass::ID = 0;
72 INITIALIZE_PASS_BEGIN(WriteBitcodePass, "write-bitcode", "Write Bitcode", false,
74 INITIALIZE_PASS_DEPENDENCY(ModuleSummaryIndexWrapperPass)
75 INITIALIZE_PASS_END(WriteBitcodePass, "write-bitcode", "Write Bitcode", false,
78 ModulePass *llvm::createBitcodeWriterPass(raw_ostream &Str,
79 bool ShouldPreserveUseListOrder,
80 bool EmitSummaryIndex, bool EmitModuleHash) {
81 return new WriteBitcodePass(Str, ShouldPreserveUseListOrder,
82 EmitSummaryIndex, EmitModuleHash);