You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and dots ('.'), can be up to 35 characters long. Letters must be lowercase.
216 lines
6.8 KiB
216 lines
6.8 KiB
// (C) Copyright 2008 CodeRage, LLC (turkanis at coderage dot com) |
|
// (C) Copyright 2005-2007 Jonathan Turkanis |
|
// 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.) |
|
|
|
// See http://www.boost.org/libs/iostreams for documentation. |
|
|
|
#ifndef BOOST_IOSTREAMS_TEE_HPP_INCLUDED |
|
#define BOOST_IOSTREAMS_TEE_HPP_INCLUDED |
|
|
|
#if defined(_MSC_VER) && (_MSC_VER >= 1020) |
|
# pragma once |
|
#endif |
|
|
|
#include <boost/assert.hpp> |
|
#include <boost/config.hpp> // BOOST_DEDUCE_TYPENAME. |
|
#include <boost/iostreams/categories.hpp> |
|
#include <boost/iostreams/detail/adapter/device_adapter.hpp> |
|
#include <boost/iostreams/detail/adapter/filter_adapter.hpp> |
|
#include <boost/iostreams/detail/call_traits.hpp> |
|
#include <boost/iostreams/detail/execute.hpp> |
|
#include <boost/iostreams/detail/functional.hpp> // call_close_all |
|
#include <boost/iostreams/operations.hpp> |
|
#include <boost/iostreams/pipeline.hpp> |
|
#include <boost/iostreams/traits.hpp> |
|
#include <boost/static_assert.hpp> |
|
#include <boost/type_traits/is_convertible.hpp> |
|
#include <boost/type_traits/is_same.hpp> |
|
|
|
namespace boost { namespace iostreams { |
|
|
|
// |
|
// Template name: tee_filter. |
|
// Template paramters: |
|
// Device - A blocking Sink. |
|
// |
|
template<typename Device> |
|
class tee_filter : public detail::filter_adapter<Device> { |
|
public: |
|
typedef typename detail::param_type<Device>::type param_type; |
|
typedef typename char_type_of<Device>::type char_type; |
|
struct category |
|
: dual_use_filter_tag, |
|
multichar_tag, |
|
closable_tag, |
|
flushable_tag, |
|
localizable_tag, |
|
optimally_buffered_tag |
|
{ }; |
|
|
|
BOOST_STATIC_ASSERT(is_device<Device>::value); |
|
BOOST_STATIC_ASSERT(( |
|
is_convertible< // Using mode_of causes failures on VC6-7.0. |
|
BOOST_DEDUCED_TYPENAME iostreams::category_of<Device>::type, output |
|
>::value |
|
)); |
|
|
|
explicit tee_filter(param_type dev) |
|
: detail::filter_adapter<Device>(dev) |
|
{ } |
|
|
|
template<typename Source> |
|
std::streamsize read(Source& src, char_type* s, std::streamsize n) |
|
{ |
|
std::streamsize result = iostreams::read(src, s, n); |
|
if (result != -1) { |
|
std::streamsize result2 = iostreams::write(this->component(), s, result); |
|
(void) result2; // Suppress 'unused variable' warning. |
|
BOOST_ASSERT(result == result2); |
|
} |
|
return result; |
|
} |
|
|
|
template<typename Sink> |
|
std::streamsize write(Sink& snk, const char_type* s, std::streamsize n) |
|
{ |
|
std::streamsize result = iostreams::write(snk, s, n); |
|
std::streamsize result2 = iostreams::write(this->component(), s, result); |
|
(void) result2; // Suppress 'unused variable' warning. |
|
BOOST_ASSERT(result == result2); |
|
return result; |
|
} |
|
|
|
template<typename Next> |
|
void close(Next&, BOOST_IOS::openmode) |
|
{ |
|
detail::close_all(this->component()); |
|
} |
|
|
|
template<typename Sink> |
|
bool flush(Sink& snk) |
|
{ |
|
bool r1 = iostreams::flush(snk); |
|
bool r2 = iostreams::flush(this->component()); |
|
return r1 && r2; |
|
} |
|
}; |
|
BOOST_IOSTREAMS_PIPABLE(tee_filter, 1) |
|
|
|
// |
|
// Template name: tee_device. |
|
// Template paramters: |
|
// Device - A blocking Device. |
|
// Sink - A blocking Sink. |
|
// |
|
template<typename Device, typename Sink> |
|
class tee_device { |
|
public: |
|
typedef typename detail::param_type<Device>::type device_param; |
|
typedef typename detail::param_type<Sink>::type sink_param; |
|
typedef typename detail::value_type<Device>::type device_value; |
|
typedef typename detail::value_type<Sink>::type sink_value; |
|
typedef typename char_type_of<Device>::type char_type; |
|
typedef typename |
|
mpl::if_< |
|
is_convertible< |
|
BOOST_DEDUCED_TYPENAME |
|
iostreams::category_of<Device>::type, |
|
output |
|
>, |
|
output, |
|
input |
|
>::type mode; |
|
BOOST_STATIC_ASSERT(is_device<Device>::value); |
|
BOOST_STATIC_ASSERT(is_device<Sink>::value); |
|
BOOST_STATIC_ASSERT(( |
|
is_same< |
|
char_type, |
|
BOOST_DEDUCED_TYPENAME char_type_of<Sink>::type |
|
>::value |
|
)); |
|
BOOST_STATIC_ASSERT(( |
|
is_convertible< |
|
BOOST_DEDUCED_TYPENAME iostreams::category_of<Sink>::type, |
|
output |
|
>::value |
|
)); |
|
struct category |
|
: mode, |
|
device_tag, |
|
closable_tag, |
|
flushable_tag, |
|
localizable_tag, |
|
optimally_buffered_tag |
|
{ }; |
|
tee_device(device_param device, sink_param sink) |
|
: dev_(device), sink_(sink) |
|
{ } |
|
std::streamsize read(char_type* s, std::streamsize n) |
|
{ |
|
BOOST_STATIC_ASSERT(( |
|
is_convertible< |
|
BOOST_DEDUCED_TYPENAME iostreams::category_of<Device>::type, input |
|
>::value |
|
)); |
|
std::streamsize result1 = iostreams::read(dev_, s, n); |
|
if (result1 != -1) { |
|
std::streamsize result2 = iostreams::write(sink_, s, result1); |
|
(void) result1; // Suppress 'unused variable' warning. |
|
(void) result2; |
|
BOOST_ASSERT(result1 == result2); |
|
} |
|
return result1; |
|
} |
|
std::streamsize write(const char_type* s, std::streamsize n) |
|
{ |
|
BOOST_STATIC_ASSERT(( |
|
is_convertible< |
|
BOOST_DEDUCED_TYPENAME iostreams::category_of<Device>::type, output |
|
>::value |
|
)); |
|
std::streamsize result1 = iostreams::write(dev_, s, n); |
|
std::streamsize result2 = iostreams::write(sink_, s, n); |
|
(void) result1; // Suppress 'unused variable' warning. |
|
(void) result2; |
|
BOOST_ASSERT(result1 == n && result2 == n); |
|
return n; |
|
} |
|
void close() |
|
{ |
|
detail::execute_all( detail::call_close_all(dev_), |
|
detail::call_close_all(sink_) ); |
|
} |
|
bool flush() |
|
{ |
|
bool r1 = iostreams::flush(dev_); |
|
bool r2 = iostreams::flush(sink_); |
|
return r1 && r2; |
|
} |
|
template<typename Locale> |
|
void imbue(const Locale& loc) |
|
{ |
|
iostreams::imbue(dev_, loc); |
|
iostreams::imbue(sink_, loc); |
|
} |
|
std::streamsize optimal_buffer_size() const |
|
{ |
|
return (std::max) ( iostreams::optimal_buffer_size(dev_), |
|
iostreams::optimal_buffer_size(sink_) ); |
|
} |
|
private: |
|
device_value dev_; |
|
sink_value sink_; |
|
}; |
|
|
|
template<typename Sink> |
|
tee_filter<Sink> tee(const Sink& snk) |
|
{ return tee_filter<Sink>(snk); } |
|
|
|
template<typename Device, typename Sink> |
|
tee_device<Device, Sink> tee(const Device& dev, const Sink& sink) |
|
{ return tee_device<Device, Sink>(dev, sink); } |
|
|
|
} } // End namespaces iostreams, boost. |
|
|
|
#endif // #ifndef BOOST_IOSTREAMS_TEE_HPP_INCLUDED
|
|
|