lib/Bitcode/Writer/BitcodeWriterPass.cpp

   1 //===- BitcodeWriterPass.cpp - Bitcode writing pass -----------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // BitcodeWriterPass implementation.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "llvm/Bitcode/BitcodeWriterPass.h"
  15 #include "llvm/Bitcode/ReaderWriter.h"
  16 #include "llvm/IR/Module.h"
  17 #include "llvm/IR/PassManager.h"
  18 #include "llvm/Pass.h"
  19 using namespace llvm;
  20
  21 PreservedAnalyses BitcodeWriterPass::run(Module &M) {
  22   WriteBitcodeToFile(&M, OS, ShouldPreserveUseListOrder, EmitFunctionSummary);
  23   return PreservedAnalyses::all();
  24 }
  25
  26 namespace {
  27   class WriteBitcodePass : public ModulePass {
  28     raw_ostream &OS; // raw_ostream to print on
  29     bool ShouldPreserveUseListOrder;
  30     bool EmitFunctionSummary;
  31
  32   public:
  33     static char ID; // Pass identification, replacement for typeid
  34     explicit WriteBitcodePass(raw_ostream &o, bool ShouldPreserveUseListOrder,
  35                               bool EmitFunctionSummary)
  36         : ModulePass(ID), OS(o),
  37           ShouldPreserveUseListOrder(ShouldPreserveUseListOrder),
  38           EmitFunctionSummary(EmitFunctionSummary) {}
  39
  40     const char *getPassName() const override { return "Bitcode Writer"; }
  41
  42     bool runOnModule(Module &M) override {
  43       WriteBitcodeToFile(&M, OS, ShouldPreserveUseListOrder,
  44                          EmitFunctionSummary);
  45       return false;
  46     }
  47   };
  48 }
  49
  50 char WriteBitcodePass::ID = 0;
  51
  52 ModulePass *llvm::createBitcodeWriterPass(raw_ostream &Str,
  53                                           bool ShouldPreserveUseListOrder,
  54                                           bool EmitFunctionSummary) {
  55   return new WriteBitcodePass(Str, ShouldPreserveUseListOrder,
  56                               EmitFunctionSummary);
  57 }