PrintContext.h

Go to the documentation of this file.

//===-- PrintContext.h ------------------------------------------*- C++ -*-===//
//
//                     The KLEE Symbolic Virtual Machine
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
 
#ifndef KLEE_PRINTCONTEXT_H
#define KLEE_PRINTCONTEXT_H
 
#include "klee/Expr/Expr.h"
 
#include "llvm/Support/raw_ostream.h"
 
#include <sstream>
#include <string>
#include <stack>
 
class PrintContext {
private:
  llvm::raw_ostream &os;
  std::string newline;
 
  std::stack<unsigned int> indentStack;
 
public:
  unsigned pos;
 
  PrintContext(llvm::raw_ostream &_os) : os(_os), newline("\n"), indentStack(), pos()
  {
          indentStack.push(pos);
  }
 
  void setNewline(const std::string &_newline) {
    newline = _newline;
  }
 
  void breakLine(unsigned indent=0) {
    os << newline;
    if (indent)
      os.indent(indent) << ' ';
    pos = indent;
  }
 
  PrintContext& breakLineI()
  {
          breakLine(indentStack.top());
          return *this;
  }
 
  PrintContext& pushIndent()
  {
          indentStack.push(pos);
          return *this;
  }
 
  PrintContext& popIndent()
  {
          indentStack.pop();
          return *this;
  }
 
  void write(const std::string &s) {
    os << s;
    pos += s.length();
  }
 
  template <typename T>
  PrintContext &operator<<(T elt) {
    std::string str;
    llvm::raw_string_ostream ss(str);
    ss << elt;
    write(ss.str());
    return *this;
  }
 
};
 
 
#endif /* KLEE_PRINTCONTEXT_H */