mirror of
https://github.com/XRPLF/rippled.git
synced 2025-11-19 10:35:50 +00:00
122 lines
2.5 KiB
C++
122 lines
2.5 KiB
C++
//
|
|
// Copyright (c) 2013-2017 Vinnie Falco (vinnie dot falco at gmail dot com)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
#ifndef BEAST_UNIT_TEST_DSTREAM_HPP
|
|
#define BEAST_UNIT_TEST_DSTREAM_HPP
|
|
|
|
#include <boost/config.hpp>
|
|
#include <ios>
|
|
#include <memory>
|
|
#include <ostream>
|
|
#include <streambuf>
|
|
#include <string>
|
|
|
|
#ifdef BOOST_WINDOWS
|
|
#include <boost/detail/winapi/basic_types.hpp>
|
|
//#include <boost/detail/winapi/debugapi.hpp>
|
|
#endif
|
|
|
|
namespace beast {
|
|
namespace unit_test {
|
|
|
|
#ifdef BOOST_WINDOWS
|
|
|
|
namespace detail {
|
|
|
|
template <class CharT, class Traits, class Allocator>
|
|
class dstream_buf : public std::basic_stringbuf<CharT, Traits, Allocator>
|
|
{
|
|
using ostream = std::basic_ostream<CharT, Traits>;
|
|
|
|
bool dbg_;
|
|
ostream& os_;
|
|
|
|
template <class T>
|
|
void
|
|
write(T const*) = delete;
|
|
|
|
void
|
|
write(char const* s)
|
|
{
|
|
if (dbg_)
|
|
/*boost::detail::winapi*/ ::OutputDebugStringA(s);
|
|
os_ << s;
|
|
}
|
|
|
|
void
|
|
write(wchar_t const* s)
|
|
{
|
|
if (dbg_)
|
|
/*boost::detail::winapi*/ ::OutputDebugStringW(s);
|
|
os_ << s;
|
|
}
|
|
|
|
public:
|
|
explicit dstream_buf(ostream& os)
|
|
: os_(os), dbg_(/*boost::detail::winapi*/ ::IsDebuggerPresent() != 0)
|
|
{
|
|
}
|
|
|
|
~dstream_buf()
|
|
{
|
|
sync();
|
|
}
|
|
|
|
int
|
|
sync() override
|
|
{
|
|
write(this->str().c_str());
|
|
this->str("");
|
|
return 0;
|
|
}
|
|
};
|
|
|
|
} // namespace detail
|
|
|
|
/** std::ostream with Visual Studio IDE redirection.
|
|
|
|
Instances of this stream wrap a specified `std::ostream`
|
|
(such as `std::cout` or `std::cerr`). If the IDE debugger
|
|
is attached when the stream is created, output will be
|
|
additionally copied to the Visual Studio Output window.
|
|
*/
|
|
template <
|
|
class CharT,
|
|
class Traits = std::char_traits<CharT>,
|
|
class Allocator = std::allocator<CharT>>
|
|
class basic_dstream : public std::basic_ostream<CharT, Traits>
|
|
{
|
|
detail::dstream_buf<CharT, Traits, Allocator> buf_;
|
|
|
|
public:
|
|
/** Construct a stream.
|
|
|
|
@param os The output stream to wrap.
|
|
*/
|
|
explicit basic_dstream(std::ostream& os)
|
|
: std::basic_ostream<CharT, Traits>(&buf_), buf_(os)
|
|
{
|
|
if (os.flags() & std::ios::unitbuf)
|
|
std::unitbuf(*this);
|
|
}
|
|
};
|
|
|
|
using dstream = basic_dstream<char>;
|
|
using dwstream = basic_dstream<wchar_t>;
|
|
|
|
#else
|
|
|
|
using dstream = std::ostream&;
|
|
using dwstream = std::wostream&;
|
|
|
|
#endif
|
|
|
|
} // namespace unit_test
|
|
} // namespace beast
|
|
|
|
#endif
|