TreeStream.h

Go to the documentation of this file.

//===-- TreeStream.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_TREESTREAM_H
#define KLEE_TREESTREAM_H
 
#include <string>
#include <vector>
 
namespace klee {
 
  typedef unsigned TreeStreamID;
  class TreeOStream;
 
  class TreeStreamWriter {
    static const unsigned bufferSize = 4*4096;
 
    friend class TreeOStream;
 
  private:
    char buffer[bufferSize];
    unsigned lastID, bufferCount;
 
    std::string path;
    std::ofstream *output;
    unsigned ids;
 
    void write(TreeOStream &os, const char *s, unsigned size);
    void flushBuffer();
 
  public:
    TreeStreamWriter(const std::string &_path);
    ~TreeStreamWriter();
 
    bool good();
 
    TreeOStream open();
    TreeOStream open(const TreeOStream &node);
 
    void flush();
 
    // hack, to be replace by proper stream capabilities
    void readStream(TreeStreamID id,
                    std::vector<unsigned char> &out);
  };
 
  class TreeOStream {
    friend class TreeStreamWriter;
 
  private:
    TreeStreamWriter *writer;
    unsigned id;
    
    TreeOStream(TreeStreamWriter &_writer, unsigned _id);
 
  public:
    TreeOStream();
    ~TreeOStream();
 
    unsigned getID() const;
 
    void write(const char *buffer, unsigned size);
 
    TreeOStream &operator<<(const std::string &s);
 
    void flush();
  };
}
 
#endif /* KLEE_TREESTREAM_H */