sl@0: // (C) Copyright Jonathan Turkanis 2005.
sl@0: // Distributed under the Boost Software License, Version 1.0. (See accompanying
sl@0: // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt.)
sl@0: 
sl@0: // See http://www.boost.org/libs/iostreams for documentation.
sl@0: 
sl@0: #ifndef BOOST_IOSTREAMS_TEE_HPP_INCLUDED
sl@0: #define BOOST_IOSTREAMS_TEE_HPP_INCLUDED
sl@0: 
sl@0: #if defined(_MSC_VER) && (_MSC_VER >= 1020)
sl@0: # pragma once
sl@0: #endif
sl@0: 
sl@0: #include <cassert>
sl@0: #include <boost/config.hpp>  // BOOST_DEDUCE_TYPENAME.
sl@0: #include <boost/iostreams/categories.hpp>
sl@0: #include <boost/iostreams/detail/adapter/basic_adapter.hpp>
sl@0: #include <boost/iostreams/detail/call_traits.hpp>
sl@0: #include <boost/iostreams/detail/closer.hpp>
sl@0: #include <boost/iostreams/operations.hpp>
sl@0: #include <boost/iostreams/pipeline.hpp>
sl@0: #include <boost/iostreams/traits.hpp>
sl@0: #include <boost/static_assert.hpp>
sl@0: #include <boost/type_traits/is_convertible.hpp>
sl@0: #include <boost/type_traits/is_same.hpp>
sl@0: 
sl@0: namespace boost { namespace iostreams {
sl@0: 
sl@0: //
sl@0: // Template name: tee_filter.
sl@0: // Template paramters:
sl@0: //      Device - A blocking Sink.
sl@0: //
sl@0: template<typename Device>
sl@0: class tee_filter : public detail::basic_adapter<Device> {
sl@0: public:
sl@0:     typedef typename detail::param_type<Device>::type  param_type;
sl@0:     typedef typename char_type_of<Device>::type        char_type;
sl@0:     struct category
sl@0:         : multichar_output_filter_tag,
sl@0:           closable_tag,
sl@0:           flushable_tag,
sl@0:           localizable_tag,
sl@0:           optimally_buffered_tag
sl@0:         { };
sl@0: 
sl@0:     BOOST_STATIC_ASSERT((
sl@0:         is_convertible< // Using mode_of causes failures on VC6-7.0.
sl@0:             BOOST_DEDUCED_TYPENAME iostreams::category_of<Device>::type, output
sl@0:         >::value
sl@0:     ));
sl@0: 
sl@0:     explicit tee_filter(param_type dev) 
sl@0:         : detail::basic_adapter<Device>(dev) 
sl@0:         { }
sl@0: 
sl@0:     template<typename Sink>
sl@0:     std::streamsize write(Sink& snk, const char_type* s, std::streamsize n)
sl@0:     {
sl@0:         std::streamsize result = iostreams::write(snk, s, n);
sl@0:         std::streamsize result2 = iostreams::write(this->component(), s, result);
sl@0:         (void) result2; // Suppress 'unused variable' warning.
sl@0:         assert(result == result2);
sl@0:         return result;
sl@0:     }
sl@0: 
sl@0:     template<typename Next>
sl@0:     void close( Next&,
sl@0:                 BOOST_IOS::openmode which =
sl@0:                     BOOST_IOS::in | BOOST_IOS::out )
sl@0:     { iostreams::close(this->component(), which); }
sl@0: 
sl@0:     template<typename Sink>
sl@0:     bool flush(Sink& snk)
sl@0:     {
sl@0:         bool r1 = iostreams::flush(snk);
sl@0:         bool r2 = iostreams::flush(this->component());
sl@0:         return r1 && r2;
sl@0:     }
sl@0: };
sl@0: BOOST_IOSTREAMS_PIPABLE(tee_filter, 1)
sl@0: 
sl@0: //
sl@0: // Template name: tee_device.
sl@0: // Template paramters:
sl@0: //      Sink1 - A blocking Sink.
sl@0: //      Sink2 - A blocking Sink.
sl@0: //
sl@0: template<typename Sink1, typename Sink2>
sl@0: class tee_device {
sl@0: public:
sl@0:     typedef typename detail::param_type<Sink1>::type  param_type1;
sl@0:     typedef typename detail::param_type<Sink2>::type  param_type2;
sl@0:     typedef typename detail::value_type<Sink1>::type  value_type1;
sl@0:     typedef typename detail::value_type<Sink2>::type  value_type2;
sl@0:     typedef typename char_type_of<Sink1>::type        char_type;
sl@0:     BOOST_STATIC_ASSERT((
sl@0:         is_same<
sl@0:             char_type, 
sl@0:             BOOST_DEDUCED_TYPENAME char_type_of<Sink2>::type
sl@0:         >::value
sl@0:     ));
sl@0:     BOOST_STATIC_ASSERT((
sl@0:         is_convertible< // Using mode_of causes failures on VC6-7.0.
sl@0:             BOOST_DEDUCED_TYPENAME iostreams::category_of<Sink1>::type, output
sl@0:         >::value
sl@0:     ));
sl@0:     BOOST_STATIC_ASSERT((
sl@0:         is_convertible< // Using mode_of causes failures on VC6-7.0.
sl@0:             BOOST_DEDUCED_TYPENAME iostreams::category_of<Sink2>::type, output
sl@0:         >::value
sl@0:     ));
sl@0:     struct category
sl@0:         : output,
sl@0:           device_tag,
sl@0:           closable_tag,
sl@0:           flushable_tag,
sl@0:           localizable_tag,
sl@0:           optimally_buffered_tag
sl@0:         { };
sl@0:     tee_device(param_type1 sink1, param_type2 sink2) 
sl@0:         : sink1_(sink1), sink2_(sink2)
sl@0:         { }
sl@0:     std::streamsize write(const char_type* s, std::streamsize n)
sl@0:     {
sl@0:         std::streamsize result1 = iostreams::write(sink1_, s, n);
sl@0:         std::streamsize result2 = iostreams::write(sink2_, s, n);
sl@0:         (void) result1; // Suppress 'unused variable' warning.
sl@0:         (void) result2;
sl@0:         assert(result1 == n && result2 == n);
sl@0:         return n;
sl@0:     }
sl@0:     void close(BOOST_IOS::openmode which = BOOST_IOS::in | BOOST_IOS::out)
sl@0:     { 
sl@0:         detail::external_closer<Sink2> close2(sink2_, which);
sl@0:         detail::external_closer<Sink1> close1(sink1_, which);
sl@0:     }
sl@0:     bool flush()
sl@0:     {
sl@0:         bool r1 = iostreams::flush(sink1_);
sl@0:         bool r2 = iostreams::flush(sink2_);
sl@0:         return r1 && r2;
sl@0:     }
sl@0:     template<typename Locale>
sl@0:     void imbue(const Locale& loc)
sl@0:     {
sl@0:         iostreams::imbue(sink1_, loc);
sl@0:         iostreams::imbue(sink2_, loc);
sl@0:     }
sl@0:     std::streamsize optimal_buffer_size() const 
sl@0:     {
sl@0:         return (std::max) ( iostreams::optimal_buffer_size(sink1_), 
sl@0:                             iostreams::optimal_buffer_size(sink2_) );
sl@0:     }
sl@0: private:
sl@0:     value_type1 sink1_;
sl@0:     value_type2 sink2_;
sl@0: };
sl@0: 
sl@0: template<typename Sink>
sl@0: tee_filter<Sink> tee(const Sink& snk) 
sl@0: { return tee_filter<Sink>(snk); }
sl@0: 
sl@0: template<typename Sink1, typename Sink2>
sl@0: tee_device<Sink1, Sink2> tee(const Sink1& sink1, const Sink2& sink2) 
sl@0: { return tee_device<Sink1, Sink2>(sink1, sink2); }
sl@0: 
sl@0: } } // End namespaces iostreams, boost.
sl@0: 
sl@0: #endif // #ifndef BOOST_IOSTREAMS_TEE_HPP_INCLUDED