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 class WriteBytecodePass : public Pass {
  23   std::ostream *Out;           // ostream to print on
  24   bool DeleteStream;
  25 public:
  26   WriteBytecodePass() : Out(&std::cout), DeleteStream(false) {}
  27   WriteBytecodePass(std::ostream *o, bool DS = false)
  28     : Out(o), DeleteStream(DS) {
  29   }
  30
  31   inline ~WriteBytecodePass() {
  32     if (DeleteStream) delete Out;
  33   }
  34
  35   bool run(Module &M) {
  36     WriteBytecodeToFile(&M, *Out);
  37     return false;
  38   }
  39 };
  40
  41 #endif