83 lines
2.0 KiB
C++
83 lines
2.0 KiB
C++
// Filename: pfstreamBuf.h
|
|
// Created by: cary (12Dec00)
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
//
|
|
// PANDA 3D SOFTWARE
|
|
// Copyright (c) 2001 - 2004, Disney Enterprises, Inc. All rights reserved
|
|
//
|
|
// All use of this software is subject to the terms of the Panda 3d
|
|
// Software license. You should have received a copy of this license
|
|
// along with this source code; you will also find a current copy of
|
|
// the license at http://etc.cmu.edu/panda3d/docs/license/ .
|
|
//
|
|
// To contact the maintainers of this program write to
|
|
// panda3d-general@lists.sourceforge.net .
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef __PFSTREAMBUF_H__
|
|
#define __PFSTREAMBUF_H__
|
|
|
|
#include "dtoolbase.h"
|
|
#include <string>
|
|
#include <stdio.h>
|
|
|
|
// By default, we'll use the Windows flavor of pipe functions if we're
|
|
// compiling under Windows. Turn this off to use popen(), even on
|
|
// Windows. (popen() doesn't seem to work on Win9x, although it does
|
|
// work on NT-based variants.)
|
|
#ifdef WIN32_VC
|
|
#define WIN_PIPE_CALLS 1
|
|
#endif
|
|
|
|
#ifdef WIN_PIPE_CALLS
|
|
#include <windows.h>
|
|
|
|
#else // WIN_PIPE_CALLS
|
|
|
|
#ifdef WIN32_VC
|
|
#define popen _popen
|
|
#define pclose _pclose
|
|
#endif
|
|
|
|
#endif // WIN_PIPE_CALLS
|
|
|
|
class EXPCL_DTOOL PipeStreamBuf : public streambuf {
|
|
public:
|
|
enum Direction { Input, Output };
|
|
|
|
PipeStreamBuf(Direction);
|
|
virtual ~PipeStreamBuf(void);
|
|
|
|
void flush();
|
|
void command(const string);
|
|
|
|
protected:
|
|
virtual int overflow(int c);
|
|
virtual int sync(void);
|
|
virtual int underflow(void);
|
|
private:
|
|
void init_pipe();
|
|
bool is_open() const;
|
|
bool eof_pipe() const;
|
|
bool open_pipe(const string &cmd);
|
|
void close_pipe();
|
|
size_t write_pipe(const char *data, size_t len);
|
|
size_t read_pipe(char *data, size_t len);
|
|
|
|
Direction _dir;
|
|
string _line_buffer;
|
|
|
|
#ifndef WIN_PIPE_CALLS
|
|
FILE *_pipe;
|
|
|
|
#else // WIN_PIPE_CALLS
|
|
HANDLE _child_out;
|
|
#endif // WIN_PIPE_CALLS
|
|
|
|
void write_chars(const char*, int, bool);
|
|
};
|
|
|
|
#endif /* __PFSTREAMBUF_H__ */
|