108 lines
2.9 KiB
C++
108 lines
2.9 KiB
C++
// Filename: pandaFileStreamBuf.h
|
|
// Created by: drose (08Sep08)
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
//
|
|
// 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."
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef PANDAFILESTREAMBUF_H
|
|
#define PANDAFILESTREAMBUF_H
|
|
|
|
#include "dtoolbase.h"
|
|
|
|
#ifdef USE_PANDAFILESTREAM
|
|
|
|
#if defined(_WIN32)
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Class : PandaFileStreamBuf
|
|
// Description : The streambuf object that implements
|
|
// pifstream and pofstream.
|
|
////////////////////////////////////////////////////////////////////
|
|
class EXPCL_DTOOL PandaFileStreamBuf : public streambuf {
|
|
public:
|
|
PandaFileStreamBuf();
|
|
virtual ~PandaFileStreamBuf();
|
|
|
|
void open(const char *filename, ios::openmode mode);
|
|
#ifdef _WIN32
|
|
void attach(const char *filename, HANDLE handle, ios::openmode mode);
|
|
#else
|
|
void attach(const char *filename, int fd, ios::openmode mode);
|
|
#endif
|
|
|
|
bool is_open() const;
|
|
void close();
|
|
|
|
enum NewlineMode {
|
|
NM_native,
|
|
NM_binary,
|
|
NM_msdos,
|
|
NM_unix,
|
|
NM_mac,
|
|
};
|
|
static NewlineMode _newline_mode;
|
|
|
|
protected:
|
|
virtual streampos seekoff(streamoff off, ios_seekdir dir, ios_openmode which);
|
|
virtual streampos seekpos(streampos pos, ios_openmode which);
|
|
|
|
virtual int overflow(int c);
|
|
virtual int sync();
|
|
virtual int underflow();
|
|
|
|
private:
|
|
size_t read_chars(char *start, size_t length);
|
|
size_t write_chars(const char *start, size_t length);
|
|
|
|
size_t read_chars_raw(char *start, size_t length);
|
|
size_t write_chars_raw(const char *start, size_t length);
|
|
|
|
size_t decode_newlines(char *dest, size_t dest_length,
|
|
const char *source, size_t source_length);
|
|
|
|
size_t encode_newlines_msdos(char *dest, size_t dest_length,
|
|
const char *source, size_t source_length);
|
|
size_t encode_newlines_unix(char *dest, size_t dest_length,
|
|
const char *source, size_t source_length);
|
|
size_t encode_newlines_mac(char *dest, size_t dest_length,
|
|
const char *source, size_t source_length);
|
|
|
|
private:
|
|
string _filename;
|
|
bool _is_open;
|
|
ios::openmode _open_mode;
|
|
|
|
char _last_read_nl;
|
|
|
|
#ifdef _WIN32
|
|
HANDLE _handle;
|
|
#else
|
|
int _fd; // Posix file descriptor
|
|
#endif // _WIN32
|
|
|
|
char *_buffer;
|
|
streampos _ppos;
|
|
streampos _gpos;
|
|
};
|
|
|
|
EXPCL_DTOOL ostream &
|
|
operator << (ostream &out, PandaFileStreamBuf::NewlineMode newline_mode);
|
|
|
|
EXPCL_DTOOL istream &
|
|
operator >> (istream &in, PandaFileStreamBuf::NewlineMode &newline_mode);
|
|
|
|
#endif // USE_PANDAFILESTREAM
|
|
|
|
#endif
|