multiplexStreamBuf.h

Go to the documentation of this file.

/**
 * PANDA 3D SOFTWARE
 * Copyright (c) Carnegie Mellon University.  All rights reserved.
 *
 * All use of this software is subject to the terms of the revised BSD
 * license.  You should have received a copy of this license along
 * with this source code in a file named "LICENSE."
 *
 * @file multiplexStreamBuf.h
 * @author drose
 * @date 2000-11-27
 */
 
#ifndef MULTIPLEXSTREAMBUF_H
#define MULTIPLEXSTREAMBUF_H
 
#include "pandabase.h"
 
#include "pvector.h"
#include "mutexImpl.h"
#include <stdio.h>
 
/**
 * Used by MultiplexStream to implement an ostream that sends what is written
 * to it to any number of additional sources, like other ostreams.
 */
class EXPCL_PANDA_DOWNLOADER MultiplexStreamBuf : public std::streambuf {
public:
  MultiplexStreamBuf();
  virtual ~MultiplexStreamBuf();
 
  enum BufferType {
    BT_none,
    BT_line,
  };
 
  enum OutputType {
    OT_ostream,
    OT_stdio,
    OT_system_debug,
  };
 
  void add_output(BufferType buffer_type, OutputType output_type,
                  std::ostream *out = nullptr,
                  FILE *fout = nullptr,
                  bool owns_obj = false);
 
  void flush();
 
protected:
  virtual int overflow(int c);
  virtual int sync();
 
private:
  void write_chars(const char *start, int length, bool flush);
 
 
  class Output {
  public:
    void close();
    void write_string(const std::string &str);
 
    BufferType _buffer_type;
    OutputType _output_type;
    std::ostream *_out;
    FILE *_fout;
    bool _owns_obj;
  };
 
  typedef pvector<Output> Outputs;
  Outputs _outputs;
 
  MutexImpl _lock;
  std::string _line_buffer;
};
 
#include "multiplexStreamBuf.I"
 
#endif