summaryrefslogtreecommitdiff
path: root/include/llvm/Assembly/CachedWriter.h
blob: c1ef1037ee3f1f16ae9b7e03ec43bed74526956b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
//===-- llvm/Assembly/CachedWriter.h - Printer Accellerator ------*- C++ -*--=//
//
// This file defines a 'CacheWriter' class that is used to accelerate printing
// chunks of LLVM.  This is used when a module is not being changed, but random
// parts of it need to be printed.  This can greatly speed up printing of LLVM
// output.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_ASSEMBLY_CACHED_WRITER_H
#define LLVM_ASSEMBLY_CACHED_WRITER_H

#include "llvm/Value.h"
#include <iostream>

class Module;
class PointerType;
class SlotCalculator;
class AssemblyWriter;  // Internal private class

class CachedWriter {
  AssemblyWriter *AW;
  SlotCalculator *SC;
public:
  std::ostream &Out;
public:
  CachedWriter(std::ostream &O = std::cout) : AW(0), SC(0), Out(O) { }
  CachedWriter(const Module *M, std::ostream &O = std::cout)
    : AW(0), SC(0), Out(O) {
    setModule(M);
  }
  ~CachedWriter();

  // setModule - Invalidate internal state, use the new module instead.
  void setModule(const Module *M);

  CachedWriter &operator<<(const Value *V);

  inline CachedWriter &operator<<(Value *X) {
    return *this << (const Value*)X;
  }
  inline CachedWriter &operator<<(const GlobalVariable *X) {
    return *this << (const Value*)X;
  }
  inline CachedWriter &operator<<(const Function *X) {
    return *this << (const Value*)X;
  }
  inline CachedWriter &operator<<(const Argument *X) {
    return *this << (const Value*)X;
  }
  inline CachedWriter &operator<<(const BasicBlock *X) {
    return *this << (const Value*)X;
  }
  inline CachedWriter &operator<<(const Instruction *X) {
    return *this << (const Value*)X; 
  }
  inline CachedWriter &operator<<(const Constant *X) {
    return *this << (const Value*)X; 
  }
  inline CachedWriter &operator<<(const Type *X) {
    return *this << (const Value*)X;
  }
  inline CachedWriter &operator<<(const PointerType *X) {
    return *this << (const Value*)X; 
  }

  inline CachedWriter &operator<<(std::ostream &(&Manip)(std::ostream &)) {
    Out << Manip; return *this;
  }

  template<class X>
  inline CachedWriter &operator<<(const X &v) {
    Out << v;
    return *this;
  }
};

#endif