include/llvm/Bytecode/WriteBytecodePass.h

   1 //===- llvm/Bytecode/WriteBytecodePass.h - Bytecode Writer Pass -*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file defines a simple pass to write the working module to a file after
  11 // pass processing is completed.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef LLVM_BYTECODE_WRITEBYTECODEPASS_H
  16 #define LLVM_BYTECODE_WRITEBYTECODEPASS_H
  17
  18 #include "llvm/Pass.h"
  19 #include "llvm/Bytecode/Writer.h"
  20 #include <iostream>
  21
  22 namespace llvm {
  23
  24 class WriteBytecodePass : public ModulePass {
  25   std::ostream *Out;           // ostream to print on
  26   bool DeleteStream;
  27   bool CompressFile;
  28 public:
  29   WriteBytecodePass()
  30     : Out(&std::cout), DeleteStream(false), CompressFile(true) {}
  31   WriteBytecodePass(std::ostream *o, bool DS = false, bool CF = true )
  32     : Out(o), DeleteStream(DS),  CompressFile(CF) {}
  33
  34   inline ~WriteBytecodePass() {
  35     if (DeleteStream) delete Out;
  36   }
  37
  38   bool runOnModule(Module &M) {
  39     WriteBytecodeToFile(&M, *Out, CompressFile );
  40     return false;
  41   }
  42 };
  43
  44 } // End llvm namespace
  45
  46 #endif