mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
374 lines
11 KiB
374 lines
11 KiB
// fstream standard header
|
|
#ifndef _FSTREAM_
|
|
#define _FSTREAM_
|
|
#include <istream>
|
|
#include <new>
|
|
|
|
#ifdef _MSC_VER
|
|
#pragma pack(push,8)
|
|
#endif /* _MSC_VER */
|
|
_STD_BEGIN
|
|
|
|
extern _CRTIMP FILE *__cdecl __Fiopen(const char *,ios_base::openmode);
|
|
|
|
// TEMPLATE FUNCTION _Fgetc
|
|
template<class _E> inline
|
|
bool _Fgetc(_E& _C, _Filet *_Fi)
|
|
{return (fread(&_C, sizeof (_E), 1, _Fi) == 1); }
|
|
inline bool _Fgetc(char& _C, _Filet *_Fi)
|
|
{int _Ch;
|
|
if ((_Ch = fgetc(_Fi)) == EOF)
|
|
return (false);
|
|
else
|
|
{_C = (char)_Ch;
|
|
return (true); }}
|
|
inline bool _Fgetc(wchar_t& _C, _Filet *_Fi)
|
|
{_Wint_t _Ch;
|
|
if ((_Ch = fgetwc(_Fi)) == WEOF)
|
|
return (false);
|
|
else
|
|
{_C = _Ch;
|
|
return (true); }}
|
|
// TEMPLATE FUNCTION _Fputc
|
|
template<class _E> inline
|
|
bool _Fputc(_E _C, _Filet *_Fi)
|
|
{return (fwrite(&_C, sizeof (_E), 1, _Fi) == 1); }
|
|
inline bool _Fputc(char _C, _Filet *_Fi)
|
|
{return (fputc(_C, _Fi) != EOF); }
|
|
inline bool _Fputc(wchar_t _C, _Filet *_Fi)
|
|
{return (fputwc(_C, _Fi) != WEOF); }
|
|
// TEMPLATE FUNCTION _Ungetc
|
|
template<class _E> inline
|
|
bool _Ungetc(const _E& _C, _Filet *_Fi, size_t _N)
|
|
{const unsigned char *_P = (const unsigned char *)&_C;
|
|
for (_P += _N; 0 < _N && ungetc(*--_P, _Fi) != EOF; --_N)
|
|
;
|
|
if (_N == 0)
|
|
return (true);
|
|
else
|
|
{for (; _N < sizeof (_E); ++_N)
|
|
fgetc(_Fi);
|
|
return (false); }}
|
|
template<class _E> inline
|
|
bool _Ungetc(const _E& _C, _Filet *_Fi)
|
|
{return (_Ungetc(_C, _Fi, sizeof (_E))); }
|
|
inline bool _Ungetc(char _C, _Filet *_Fi)
|
|
{return (ungetc((unsigned char)_C, _Fi) != EOF); }
|
|
inline bool _Ungetc(wchar_t _C, _Filet *_Fi)
|
|
{return (ungetwc(_C, _Fi) != WEOF); }
|
|
// TEMPLATE CLASS basic_filebuf
|
|
template<class _E, class _Tr = char_traits<_E> >
|
|
class basic_filebuf : public basic_streambuf<_E, _Tr> {
|
|
public:
|
|
typedef basic_filebuf<_E, _Tr> _Myt;
|
|
typedef basic_streambuf<_E, _Tr> _Mysb;
|
|
typedef codecvt<_E, char, _Tr::state_type> _Cvt;
|
|
basic_filebuf(_Filet *_F = 0)
|
|
: _Loc(), _Mysb() {_Init(_F, _Newfl); }
|
|
basic_filebuf(_Uninitialized)
|
|
: _Loc(_Noinit), _Mysb(_Noinit) {}
|
|
virtual ~basic_filebuf()
|
|
{if (_Closef)
|
|
close();
|
|
delete _Str; }
|
|
enum _Initfl {_Newfl, _Openfl, _Closefl};
|
|
bool is_open() const
|
|
{return (_File != 0); }
|
|
_Myt *open(const char *_S, ios_base::openmode _M)
|
|
{_Filet *_Fp;
|
|
if (_File != 0 || (_Fp = __Fiopen(_S, _M)) == 0)
|
|
return (0);
|
|
_Init(_Fp, _Openfl);
|
|
_Initcvt();
|
|
return (this); }
|
|
_Myt *open(const char *_N, ios::open_mode _M)
|
|
{return (open(_N, (ios::openmode)_M)); }
|
|
_Myt *close()
|
|
{if (_File != 0 && fclose(_File) == 0)
|
|
{_Init(0, _Closefl);
|
|
return (this); }
|
|
else
|
|
return (0); }
|
|
protected:
|
|
virtual int_type overflow(int_type _C = _Tr::eof())
|
|
{if (_Tr::eq_int_type(_Tr::eof(), _C))
|
|
return (_Tr::not_eof(_C));
|
|
else if (pptr() != 0 && pptr() < epptr())
|
|
{*_Pninc() = _Tr::to_char_type(_C);
|
|
return (_C); }
|
|
else if (_File == 0)
|
|
return (_Tr::eof());
|
|
else if (_Pcvt == 0)
|
|
return (_Fputc(_Tr::to_char_type(_C), _File)
|
|
? _C : _Tr::eof());
|
|
else
|
|
{const int _NC = 8;
|
|
const _E _X = _Tr::to_char_type(_C);
|
|
const _E *_S;
|
|
char *_D;
|
|
_Str->erase();
|
|
for (size_t _I = _NC; ; _I += _NC)
|
|
{_Str->append(_NC, '\0');
|
|
switch (_Pcvt->out(_State,
|
|
&_X, &_X + 1, _S,
|
|
_Str->begin(), _Str->end(), _D))
|
|
{case codecvt_base::partial:
|
|
if (_S == &_X)
|
|
return (_Tr::eof());
|
|
case codecvt_base::ok: // can fall through
|
|
{size_t _N = _D - _Str->begin();
|
|
return (fwrite(_Str->begin(), 1, _N, _File)
|
|
== _N ? _C : _Tr::eof()); }
|
|
case codecvt_base::noconv:
|
|
return (_Fputc(_X, _File) ? _C : _Tr::eof());
|
|
default:
|
|
return (_Tr::eof()); }}}}
|
|
virtual int_type pbackfail(int_type _C = _Tr::eof())
|
|
{if (gptr() != 0 && eback() < gptr()
|
|
&& (_Tr::eq_int_type(_Tr::eof(), _C)
|
|
|| _Tr::eq_int_type(_Tr::to_int_type(gptr()[-1]),
|
|
_C)))
|
|
{_Gndec();
|
|
return (_Tr::not_eof(_C)); }
|
|
else if (_File == 0 || _Tr::eq_int_type(_Tr::eof(), _C))
|
|
return (_Tr::eof());
|
|
else if (_Pcvt == 0)
|
|
return (_Ungetc(_Tr::to_char_type(_C), _File)
|
|
? _C : _Tr::eof());
|
|
else if (0 < _Str->size()
|
|
&& _Ungetc(*_Str->begin(), _File, _Str->size()))
|
|
{_Str->erase();
|
|
_State = _State0;
|
|
return (_C); }
|
|
else
|
|
return (_Tr::eof()); }
|
|
virtual int_type underflow()
|
|
{if (gptr() != 0 && gptr() < egptr())
|
|
return (_Tr::to_int_type(*gptr()));
|
|
else
|
|
return (pbackfail(uflow())); }
|
|
virtual int_type uflow()
|
|
{if (gptr() != 0 && gptr() < egptr())
|
|
return (_Tr::to_int_type(*_Gninc()));
|
|
else if (_File == 0)
|
|
return (_Tr::eof());
|
|
else if (_Pcvt == 0)
|
|
{_E _C;
|
|
return (_Fgetc(_C, _File) ? _Tr::to_int_type(_C)
|
|
: _Tr::eof()); }
|
|
else
|
|
for (_State0 = _State, _Str->erase(); ; )
|
|
{_E _X, *_D;
|
|
const char *_S;
|
|
int _C = fgetc(_File);
|
|
if (_C == EOF)
|
|
return (_Tr::eof()); // partial char?
|
|
_Str->append(1, (char)_C);
|
|
_State = _State0;
|
|
switch (_Pcvt->in(_State,
|
|
_Str->begin(), _Str->end(), _S,
|
|
&_X, &_X + 1, _D))
|
|
{case codecvt_base::partial:
|
|
break;
|
|
case codecvt_base::noconv:
|
|
if (_Str->size() < sizeof (_E))
|
|
break;
|
|
memcpy(&_X, _Str->begin(), sizeof (_E));
|
|
case codecvt_base::ok: // can fall through
|
|
return (_Tr::to_int_type(_X));
|
|
default:
|
|
return (_Tr::eof()); }}}
|
|
virtual pos_type seekoff(off_type _O, ios_base::seekdir _Way,
|
|
ios_base::openmode =
|
|
(ios_base::openmode)(ios_base::in | ios_base::out))
|
|
{fpos_t _Fp;
|
|
if (_File == 0 || fseek(_File, (long)_O, _Way) != 0
|
|
|| fgetpos(_File, &_Fp) != 0)
|
|
return (pos_type(_BADOFF));
|
|
return (pos_type(_State, _Fp)); }
|
|
virtual pos_type seekpos(pos_type _P,
|
|
ios_base::openmode =
|
|
(ios_base::openmode)(ios_base::in | ios_base::out))
|
|
{fpos_t _Fp = _P.get_fpos_t();
|
|
off_type _Off = (off_type)_P - _FPOSOFF(_Fp);
|
|
if (_File == 0
|
|
|| fsetpos(_File, &_Fp) != 0
|
|
|| _Off != 0 && fseek(_File, (long)_Off, SEEK_CUR) != 0
|
|
|| fgetpos(_File, &_Fp) != 0)
|
|
return (pos_type(_BADOFF));
|
|
if (_Str != 0)
|
|
_State = _P.state(), _Str->erase();
|
|
return (pos_type(_State, _Fp)); }
|
|
virtual _Mysb *setbuf(_E *_S, streamsize _N)
|
|
{return (_File == 0 || setvbuf(_File, (char *)_S,
|
|
_IOFBF, _N * sizeof (_E)) != 0 ? 0 : this); }
|
|
virtual int sync()
|
|
{return (_File == 0 || 0 <= fflush(_File) ? 0 : -1); }
|
|
void _Init(_Filet *_Fp, _Initfl _Which)
|
|
{static _Tr::state_type _Stinit;
|
|
_Closef = _Which == _Openfl;
|
|
if (_Which == _Newfl)
|
|
{_Loc.locale::~locale();
|
|
new (&_Loc) locale;
|
|
_Str = 0; }
|
|
_Mysb::_Init();
|
|
if (_Fp != 0 && !_Closef && sizeof (_E) == 1)
|
|
{_Mysb::_Init((_E **)&_Fp->_base,
|
|
(_E **)&_Fp->_ptr, &_Fp->_cnt,
|
|
(_E **)&_Fp->_base, (_E **)&_Fp->_ptr,
|
|
&_Fp->_cnt); }
|
|
_File = _Fp;
|
|
_State = _Stinit;
|
|
_State0 = _Stinit;
|
|
_Pcvt = 0; }
|
|
void _Initcvt()
|
|
{_Pcvt = (_Cvt *)&_USE(getloc(), _Cvt);
|
|
_Loc = _ADDFAC(_Loc, _Pcvt);
|
|
if (_Pcvt->always_noconv())
|
|
_Pcvt = 0;
|
|
if (_Str == 0)
|
|
_Str = new string; }
|
|
private:
|
|
_Cvt *_Pcvt;
|
|
_Tr::state_type _State0;
|
|
_Tr::state_type _State;
|
|
string *_Str;
|
|
bool _Closef;
|
|
locale _Loc;
|
|
_Filet *_File;
|
|
};
|
|
|
|
#ifdef _DLL
|
|
#pragma warning(disable:4231) /* the extern before template is a non-standard extension */
|
|
extern template class _CRTIMP basic_filebuf<char, char_traits<char> >;
|
|
extern template class _CRTIMP basic_filebuf<wchar_t, char_traits<wchar_t> >;
|
|
#pragma warning(default:4231) /* restore previous warning */
|
|
#endif // _DLL
|
|
|
|
// TEMPLATE CLASS basic_ifstream
|
|
template<class _E, class _Tr = char_traits<_E> >
|
|
class basic_ifstream : public basic_istream<_E, _Tr> {
|
|
public:
|
|
typedef basic_ifstream<_E, _Tr> _Myt;
|
|
typedef basic_filebuf<_E, _Tr> _Myfb;
|
|
basic_ifstream()
|
|
: basic_istream<_E, _Tr>(&_Fb) {}
|
|
explicit basic_ifstream(const char *_S,
|
|
ios_base::openmode _M = in)
|
|
: basic_istream<_E, _Tr>(&_Fb)
|
|
{if (_Fb.open(_S, _M | in) == 0)
|
|
setstate(failbit); }
|
|
virtual ~basic_ifstream()
|
|
{}
|
|
_Myfb *rdbuf() const
|
|
{return ((_Myfb *)&_Fb); }
|
|
bool is_open() const
|
|
{return (_Fb.is_open()); }
|
|
void open(const char *_S, ios_base::openmode _M = in)
|
|
{if (_Fb.open(_S, _M | in) == 0)
|
|
setstate(failbit); }
|
|
void open(const char *_S, ios_base::open_mode _M)
|
|
{open(_S, (openmode)_M); }
|
|
void close()
|
|
{if (_Fb.close() == 0)
|
|
setstate(failbit); }
|
|
private:
|
|
_Myfb _Fb;
|
|
};
|
|
|
|
#ifdef _DLL
|
|
#pragma warning(disable:4231) /* the extern before template is a non-standard extension */
|
|
extern template class _CRTIMP basic_ifstream<char, char_traits<char> >;
|
|
extern template class _CRTIMP basic_ifstream<wchar_t, char_traits<wchar_t> >;
|
|
#pragma warning(default:4231) /* restore previous warning */
|
|
#endif // _DLL
|
|
|
|
// TEMPLATE CLASS basic_ofstream
|
|
template<class _E, class _Tr = char_traits<_E> >
|
|
class basic_ofstream : public basic_ostream<_E, _Tr> {
|
|
public:
|
|
typedef basic_ofstream<_E, _Tr> _Myt;
|
|
typedef basic_filebuf<_E, _Tr> _Myfb;
|
|
basic_ofstream()
|
|
: basic_ostream<_E, _Tr>(&_Fb) {}
|
|
explicit basic_ofstream(const char *_S,
|
|
ios_base::openmode _M = out | trunc)
|
|
: basic_ostream<_E, _Tr>(&_Fb)
|
|
{if (_Fb.open(_S, _M | out) == 0)
|
|
setstate(failbit); }
|
|
virtual ~basic_ofstream()
|
|
{}
|
|
_Myfb *rdbuf() const
|
|
{return ((_Myfb *)&_Fb); }
|
|
bool is_open() const
|
|
{return (_Fb.is_open()); }
|
|
void open(const char *_S, ios_base::openmode _M = out | trunc)
|
|
{if (_Fb.open(_S, _M | out) == 0)
|
|
setstate(failbit); }
|
|
void open(const char *_S, ios_base::open_mode _M)
|
|
{open(_S, (openmode)_M); }
|
|
void close()
|
|
{if (_Fb.close() == 0)
|
|
setstate(failbit); }
|
|
private:
|
|
_Myfb _Fb;
|
|
};
|
|
|
|
#ifdef _DLL
|
|
#pragma warning(disable:4231) /* the extern before template is a non-standard extension */
|
|
extern template class _CRTIMP basic_ofstream<char, char_traits<char> >;
|
|
extern template class _CRTIMP basic_ofstream<wchar_t, char_traits<wchar_t> >;
|
|
#pragma warning(default:4231) /* restore previous warning */
|
|
#endif // _DLL
|
|
|
|
|
|
// TEMPLATE CLASS basic_fstream
|
|
template<class _E, class _Tr = char_traits<_E> >
|
|
class basic_fstream : public basic_iostream<_E, _Tr> {
|
|
public:
|
|
basic_fstream()
|
|
: basic_iostream<_E, _Tr>(&_Fb) {}
|
|
explicit basic_fstream(const char *_S,
|
|
ios_base::openmode _M = in | out)
|
|
: basic_iostream<_E, _Tr>(&_Fb)
|
|
{if (_Fb.open(_S, _M) == 0)
|
|
setstate(failbit); }
|
|
virtual ~basic_fstream()
|
|
{}
|
|
basic_filebuf<_E, _Tr> *rdbuf() const
|
|
{return ((basic_filebuf<_E, _Tr> *)&_Fb); }
|
|
bool is_open() const
|
|
{return (_Fb.is_open()); }
|
|
void open(const char *_S, ios_base::openmode _M = in | out)
|
|
{if (_Fb.open(_S, _M) == 0)
|
|
setstate(failbit); }
|
|
void open(const char *_S, ios_base::open_mode _M)
|
|
{open(_S, (openmode)_M); }
|
|
void close()
|
|
{if (_Fb.close() == 0)
|
|
setstate(failbit); }
|
|
private:
|
|
basic_filebuf<_E, _Tr> _Fb;
|
|
};
|
|
|
|
#ifdef _DLL
|
|
#pragma warning(disable:4231) /* the extern before template is a non-standard extension */
|
|
extern template class _CRTIMP basic_fstream<char, char_traits<char> >;
|
|
extern template class _CRTIMP basic_fstream<wchar_t, char_traits<wchar_t> >;
|
|
#pragma warning(default:4231) /* restore previous warning */
|
|
#endif // _DLL
|
|
|
|
|
|
|
|
|
|
_STD_END
|
|
#ifdef _MSC_VER
|
|
#pragma pack(pop)
|
|
#endif /* _MSC_VER */
|
|
|
|
#endif /* _FSTREAM_ */
|
|
|
|
/*
|
|
* Copyright (c) 1994 by P.J. Plauger. ALL RIGHTS RESERVED.
|
|
* Consult your license regarding permissions and restrictions.
|
|
*/
|