mirror of
https://github.com/boostorg/nowide.git
synced 2026-02-26 04:42:24 +00:00
Enforce uniform formatting via clang-format
This commit is contained in:
Flamefire
469
src/iostream.cpp
469
src/iostream.cpp
@@ -12,9 +12,10 @@
|
||||
|
||||
namespace boost {
|
||||
namespace nowide {
|
||||
BOOST_NOWIDE_DECL void dummy_exported_function(){}
|
||||
}
|
||||
}
|
||||
BOOST_NOWIDE_DECL void dummy_exported_function()
|
||||
{}
|
||||
} // namespace nowide
|
||||
} // namespace boost
|
||||
|
||||
#else
|
||||
|
||||
@@ -23,248 +24,248 @@ namespace nowide {
|
||||
#include <vector>
|
||||
|
||||
#ifndef NOMINMAX
|
||||
# define NOMINMAX
|
||||
#define NOMINMAX
|
||||
#endif
|
||||
|
||||
|
||||
#include <windows.h>
|
||||
|
||||
namespace boost {
|
||||
namespace nowide {
|
||||
namespace details {
|
||||
namespace details {
|
||||
|
||||
namespace {
|
||||
bool is_atty_handle(HANDLE h)
|
||||
{
|
||||
if(h)
|
||||
{
|
||||
DWORD dummy;
|
||||
return GetConsoleMode(h, &dummy) == TRUE;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
} // namespace
|
||||
|
||||
class console_output_buffer : public std::streambuf
|
||||
{
|
||||
public:
|
||||
console_output_buffer(HANDLE h) : handle_(h)
|
||||
{}
|
||||
|
||||
protected:
|
||||
int sync()
|
||||
{
|
||||
return overflow(traits_type::eof());
|
||||
}
|
||||
int overflow(int c)
|
||||
{
|
||||
if(!handle_)
|
||||
return -1;
|
||||
int n = static_cast<int>(pptr() - pbase());
|
||||
int r = 0;
|
||||
|
||||
if(n > 0 && (r = write(pbase(), n)) < 0)
|
||||
return -1;
|
||||
if(r < n)
|
||||
{
|
||||
std::memmove(pbase(), pbase() + r, n - r);
|
||||
}
|
||||
setp(buffer_, buffer_ + buffer_size);
|
||||
pbump(n - r);
|
||||
if(c != traits_type::eof())
|
||||
sputc(traits_type::to_char_type(c));
|
||||
return 0;
|
||||
}
|
||||
|
||||
private:
|
||||
int write(char const *p, int n)
|
||||
{
|
||||
namespace uf = detail::utf;
|
||||
char const *b = p;
|
||||
char const *e = p + n;
|
||||
DWORD size = 0;
|
||||
if(n > buffer_size)
|
||||
return -1;
|
||||
wchar_t *out = wbuffer_;
|
||||
uf::code_point c;
|
||||
size_t decoded = 0;
|
||||
while(p < e && (c = uf::utf_traits<char>::decode(p, e)) != uf::incomplete)
|
||||
{
|
||||
if(c == uf::illegal)
|
||||
c = BOOST_NOWIDE_REPLACEMENT_CHARACTER;
|
||||
out = uf::utf_traits<wchar_t>::encode(c, out);
|
||||
decoded = p - b;
|
||||
}
|
||||
if(!WriteConsoleW(handle_, wbuffer_, static_cast<DWORD>(out - wbuffer_), &size, 0))
|
||||
return -1;
|
||||
return static_cast<int>(decoded);
|
||||
}
|
||||
|
||||
static const int buffer_size = 1024;
|
||||
char buffer_[buffer_size];
|
||||
wchar_t wbuffer_[buffer_size]; // for null
|
||||
HANDLE handle_;
|
||||
};
|
||||
|
||||
class console_input_buffer : public std::streambuf
|
||||
{
|
||||
public:
|
||||
console_input_buffer(HANDLE h) : handle_(h), wsize_(0)
|
||||
{}
|
||||
|
||||
protected:
|
||||
int pbackfail(int c)
|
||||
{
|
||||
if(c == traits_type::eof())
|
||||
return traits_type::eof();
|
||||
|
||||
if(gptr() != eback())
|
||||
{
|
||||
gbump(-1);
|
||||
*gptr() = traits_type::to_char_type(c);
|
||||
return 0;
|
||||
}
|
||||
|
||||
if(pback_buffer_.empty())
|
||||
{
|
||||
pback_buffer_.resize(4);
|
||||
char *b = &pback_buffer_[0];
|
||||
char *e = b + pback_buffer_.size();
|
||||
setg(b, e - 1, e);
|
||||
*gptr() = traits_type::to_char_type(c);
|
||||
} else
|
||||
{
|
||||
size_t n = pback_buffer_.size();
|
||||
std::vector<char> tmp;
|
||||
tmp.resize(n * 2);
|
||||
memcpy(&tmp[n], &pback_buffer_[0], n);
|
||||
tmp.swap(pback_buffer_);
|
||||
char *b = &pback_buffer_[0];
|
||||
char *e = b + n * 2;
|
||||
char *p = b + n - 1;
|
||||
*p = traits_type::to_char_type(c);
|
||||
setg(b, p, e);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int underflow()
|
||||
{
|
||||
if(!handle_)
|
||||
return -1;
|
||||
if(!pback_buffer_.empty())
|
||||
pback_buffer_.clear();
|
||||
|
||||
size_t n = read();
|
||||
setg(buffer_, buffer_, buffer_ + n);
|
||||
if(n == 0)
|
||||
return traits_type::eof();
|
||||
return std::char_traits<char>::to_int_type(*gptr());
|
||||
}
|
||||
|
||||
private:
|
||||
size_t read()
|
||||
{
|
||||
namespace uf = detail::utf;
|
||||
DWORD read_wchars = 0;
|
||||
size_t n = wbuffer_size - wsize_;
|
||||
if(!ReadConsoleW(handle_, wbuffer_, static_cast<DWORD>(n), &read_wchars, 0))
|
||||
return 0;
|
||||
wsize_ += read_wchars;
|
||||
char *out = buffer_;
|
||||
wchar_t *b = wbuffer_;
|
||||
wchar_t *e = b + wsize_;
|
||||
wchar_t *p = b;
|
||||
uf::code_point c = 0;
|
||||
wsize_ = e - p;
|
||||
while(p < e && (c = uf::utf_traits<wchar_t>::decode(p, e)) != uf::illegal && c != uf::incomplete)
|
||||
{
|
||||
out = uf::utf_traits<char>::encode(c, out);
|
||||
wsize_ = e - p;
|
||||
}
|
||||
|
||||
if(c == uf::illegal)
|
||||
return 0;
|
||||
|
||||
if(c == uf::incomplete)
|
||||
{
|
||||
std::memmove(b, e - wsize_, sizeof(wchar_t) * wsize_);
|
||||
}
|
||||
|
||||
return out - buffer_;
|
||||
}
|
||||
|
||||
static const size_t buffer_size = 1024 * 3;
|
||||
static const size_t wbuffer_size = 1024;
|
||||
char buffer_[buffer_size];
|
||||
wchar_t wbuffer_[buffer_size]; // for null
|
||||
HANDLE handle_;
|
||||
size_t wsize_;
|
||||
std::vector<char> pback_buffer_;
|
||||
};
|
||||
|
||||
winconsole_ostream::winconsole_ostream(int fd) : std::ostream(0)
|
||||
{
|
||||
HANDLE h = 0;
|
||||
switch(fd)
|
||||
{
|
||||
case 1: h = GetStdHandle(STD_OUTPUT_HANDLE); break;
|
||||
case 2: h = GetStdHandle(STD_ERROR_HANDLE); break;
|
||||
}
|
||||
if(is_atty_handle(h))
|
||||
{
|
||||
d.reset(new console_output_buffer(h));
|
||||
std::ostream::rdbuf(d.get());
|
||||
} else
|
||||
{
|
||||
std::ostream::rdbuf(fd == 1 ? std::cout.rdbuf() : std::cerr.rdbuf());
|
||||
}
|
||||
}
|
||||
winconsole_ostream::~winconsole_ostream()
|
||||
{
|
||||
try
|
||||
{
|
||||
flush();
|
||||
} catch(...)
|
||||
{}
|
||||
}
|
||||
|
||||
winconsole_istream::winconsole_istream() : std::istream(0)
|
||||
{
|
||||
HANDLE h = GetStdHandle(STD_INPUT_HANDLE);
|
||||
if(is_atty_handle(h))
|
||||
{
|
||||
d.reset(new console_input_buffer(h));
|
||||
std::istream::rdbuf(d.get());
|
||||
} else
|
||||
{
|
||||
std::istream::rdbuf(std::cin.rdbuf());
|
||||
}
|
||||
}
|
||||
|
||||
winconsole_istream::~winconsole_istream()
|
||||
{}
|
||||
|
||||
} // namespace details
|
||||
|
||||
details::winconsole_istream cin;
|
||||
details::winconsole_ostream cout(1);
|
||||
details::winconsole_ostream cerr(2);
|
||||
details::winconsole_ostream clog(2);
|
||||
|
||||
namespace {
|
||||
bool is_atty_handle(HANDLE h)
|
||||
struct initialize
|
||||
{
|
||||
if(h) {
|
||||
DWORD dummy;
|
||||
return GetConsoleMode(h,&dummy) == TRUE;
|
||||
initialize()
|
||||
{
|
||||
boost::nowide::cin.tie(&boost::nowide::cout);
|
||||
boost::nowide::cerr.tie(&boost::nowide::cout);
|
||||
boost::nowide::clog.tie(&boost::nowide::cout);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
}
|
||||
} inst;
|
||||
} // namespace
|
||||
|
||||
class console_output_buffer : public std::streambuf {
|
||||
public:
|
||||
console_output_buffer(HANDLE h) :
|
||||
handle_(h)
|
||||
{
|
||||
}
|
||||
protected:
|
||||
int sync()
|
||||
{
|
||||
return overflow(traits_type::eof());
|
||||
}
|
||||
int overflow(int c)
|
||||
{
|
||||
if(!handle_)
|
||||
return -1;
|
||||
int n = static_cast<int>(pptr() - pbase());
|
||||
int r = 0;
|
||||
|
||||
if(n > 0 && (r=write(pbase(),n)) < 0)
|
||||
return -1;
|
||||
if(r < n) {
|
||||
std::memmove(pbase(),pbase() + r,n-r);
|
||||
}
|
||||
setp(buffer_, buffer_ + buffer_size);
|
||||
pbump(n-r);
|
||||
if(c!=traits_type::eof())
|
||||
sputc(traits_type::to_char_type(c));
|
||||
return 0;
|
||||
}
|
||||
private:
|
||||
|
||||
int write(char const *p,int n)
|
||||
{
|
||||
namespace uf = detail::utf;
|
||||
char const *b = p;
|
||||
char const *e = p+n;
|
||||
DWORD size=0;
|
||||
if(n > buffer_size)
|
||||
return -1;
|
||||
wchar_t *out = wbuffer_;
|
||||
uf::code_point c;
|
||||
size_t decoded = 0;
|
||||
while(p < e && (c = uf::utf_traits<char>::decode(p,e))!=uf::incomplete) {
|
||||
if(c == uf::illegal)
|
||||
c = BOOST_NOWIDE_REPLACEMENT_CHARACTER;
|
||||
out = uf::utf_traits<wchar_t>::encode(c,out);
|
||||
decoded = p-b;
|
||||
}
|
||||
if(!WriteConsoleW(handle_,wbuffer_,static_cast<DWORD>(out - wbuffer_),&size,0))
|
||||
return -1;
|
||||
return static_cast<int>(decoded);
|
||||
}
|
||||
|
||||
static const int buffer_size = 1024;
|
||||
char buffer_[buffer_size];
|
||||
wchar_t wbuffer_[buffer_size]; // for null
|
||||
HANDLE handle_;
|
||||
};
|
||||
|
||||
class console_input_buffer: public std::streambuf {
|
||||
public:
|
||||
console_input_buffer(HANDLE h) :
|
||||
handle_(h),
|
||||
wsize_(0)
|
||||
{
|
||||
}
|
||||
|
||||
protected:
|
||||
int pbackfail(int c)
|
||||
{
|
||||
if(c==traits_type::eof())
|
||||
return traits_type::eof();
|
||||
|
||||
if(gptr()!=eback()) {
|
||||
gbump(-1);
|
||||
*gptr() = traits_type::to_char_type(c);
|
||||
return 0;
|
||||
}
|
||||
|
||||
if(pback_buffer_.empty()) {
|
||||
pback_buffer_.resize(4);
|
||||
char *b = &pback_buffer_[0];
|
||||
char *e = b + pback_buffer_.size();
|
||||
setg(b,e-1,e);
|
||||
*gptr() = traits_type::to_char_type(c);
|
||||
}
|
||||
else {
|
||||
size_t n = pback_buffer_.size();
|
||||
std::vector<char> tmp;
|
||||
tmp.resize(n*2);
|
||||
memcpy(&tmp[n],&pback_buffer_[0],n);
|
||||
tmp.swap(pback_buffer_);
|
||||
char *b = &pback_buffer_[0];
|
||||
char *e = b + n * 2;
|
||||
char *p = b+n-1;
|
||||
*p = traits_type::to_char_type(c);
|
||||
setg(b,p,e);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int underflow()
|
||||
{
|
||||
if(!handle_)
|
||||
return -1;
|
||||
if(!pback_buffer_.empty())
|
||||
pback_buffer_.clear();
|
||||
|
||||
size_t n = read();
|
||||
setg(buffer_,buffer_,buffer_+n);
|
||||
if(n == 0)
|
||||
return traits_type::eof();
|
||||
return std::char_traits<char>::to_int_type(*gptr());
|
||||
}
|
||||
|
||||
private:
|
||||
|
||||
size_t read()
|
||||
{
|
||||
namespace uf = detail::utf;
|
||||
DWORD read_wchars = 0;
|
||||
size_t n = wbuffer_size - wsize_;
|
||||
if(!ReadConsoleW(handle_,wbuffer_,static_cast<DWORD>(n),&read_wchars,0))
|
||||
return 0;
|
||||
wsize_ += read_wchars;
|
||||
char *out = buffer_;
|
||||
wchar_t *b = wbuffer_;
|
||||
wchar_t *e = b + wsize_;
|
||||
wchar_t *p = b;
|
||||
uf::code_point c = 0;
|
||||
wsize_ = e-p;
|
||||
while(p < e && (c = uf::utf_traits<wchar_t>::decode(p,e))!=uf::illegal && c!=uf::incomplete) {
|
||||
out = uf::utf_traits<char>::encode(c,out);
|
||||
wsize_ = e-p;
|
||||
}
|
||||
|
||||
if(c==uf::illegal)
|
||||
return 0;
|
||||
|
||||
|
||||
if(c==uf::incomplete) {
|
||||
std::memmove(b,e-wsize_,sizeof(wchar_t)*wsize_);
|
||||
}
|
||||
|
||||
return out - buffer_;
|
||||
}
|
||||
|
||||
static const size_t buffer_size = 1024 * 3;
|
||||
static const size_t wbuffer_size = 1024;
|
||||
char buffer_[buffer_size];
|
||||
wchar_t wbuffer_[buffer_size]; // for null
|
||||
HANDLE handle_;
|
||||
size_t wsize_;
|
||||
std::vector<char> pback_buffer_;
|
||||
};
|
||||
|
||||
winconsole_ostream::winconsole_ostream(int fd) : std::ostream(0)
|
||||
{
|
||||
HANDLE h = 0;
|
||||
switch(fd) {
|
||||
case 1:
|
||||
h = GetStdHandle(STD_OUTPUT_HANDLE);
|
||||
break;
|
||||
case 2:
|
||||
h = GetStdHandle(STD_ERROR_HANDLE);
|
||||
break;
|
||||
}
|
||||
if(is_atty_handle(h)) {
|
||||
d.reset(new console_output_buffer(h));
|
||||
std::ostream::rdbuf(d.get());
|
||||
}
|
||||
else {
|
||||
std::ostream::rdbuf( fd == 1 ? std::cout.rdbuf() : std::cerr.rdbuf() );
|
||||
}
|
||||
}
|
||||
winconsole_ostream::~winconsole_ostream()
|
||||
{
|
||||
try {
|
||||
flush();
|
||||
}
|
||||
catch(...){}
|
||||
}
|
||||
|
||||
winconsole_istream::winconsole_istream() : std::istream(0)
|
||||
{
|
||||
HANDLE h = GetStdHandle(STD_INPUT_HANDLE);
|
||||
if(is_atty_handle(h)) {
|
||||
d.reset(new console_input_buffer(h));
|
||||
std::istream::rdbuf(d.get());
|
||||
}
|
||||
else {
|
||||
std::istream::rdbuf(std::cin.rdbuf());
|
||||
}
|
||||
}
|
||||
|
||||
winconsole_istream::~winconsole_istream()
|
||||
{
|
||||
}
|
||||
|
||||
} // details
|
||||
|
||||
details::winconsole_istream cin;
|
||||
details::winconsole_ostream cout(1);
|
||||
details::winconsole_ostream cerr(2);
|
||||
details::winconsole_ostream clog(2);
|
||||
|
||||
namespace {
|
||||
struct initialize {
|
||||
initialize()
|
||||
{
|
||||
boost::nowide::cin.tie(&boost::nowide::cout);
|
||||
boost::nowide::cerr.tie(&boost::nowide::cout);
|
||||
boost::nowide::clog.tie(&boost::nowide::cout);
|
||||
}
|
||||
} inst;
|
||||
}
|
||||
|
||||
|
||||
|
||||
} // nowide
|
||||
} // namespace nowide
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
|
||||
Reference in New Issue
Block a user