Chris@16: // (C) Copyright 2008 CodeRage, LLC (turkanis at coderage dot com)
Chris@16: // (C) Copyright 2005-2007 Jonathan Turkanis
Chris@16: // Distributed under the Boost Software License, Version 1.0. (See accompanying
Chris@16: // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt.)
Chris@16: 
Chris@16: // See http://www.boost.org/libs/iostreams for documentation.
Chris@16: 
Chris@16: #ifndef BOOST_IOSTREAMS_TEE_HPP_INCLUDED
Chris@16: #define BOOST_IOSTREAMS_TEE_HPP_INCLUDED
Chris@16: 
Chris@16: #if defined(_MSC_VER) && (_MSC_VER >= 1020)
Chris@16: # pragma once
Chris@16: #endif
Chris@16: 
Chris@16: #include <boost/assert.hpp>
Chris@16: #include <boost/config.hpp>  // BOOST_DEDUCE_TYPENAME.
Chris@16: #include <boost/iostreams/categories.hpp>
Chris@16: #include <boost/iostreams/detail/adapter/device_adapter.hpp>
Chris@16: #include <boost/iostreams/detail/adapter/filter_adapter.hpp>
Chris@16: #include <boost/iostreams/detail/call_traits.hpp>
Chris@16: #include <boost/iostreams/detail/execute.hpp>
Chris@16: #include <boost/iostreams/detail/functional.hpp>  // call_close_all 
Chris@16: #include <boost/iostreams/operations.hpp>
Chris@16: #include <boost/iostreams/pipeline.hpp>
Chris@16: #include <boost/iostreams/traits.hpp>
Chris@16: #include <boost/static_assert.hpp>
Chris@16: #include <boost/type_traits/is_convertible.hpp>
Chris@16: #include <boost/type_traits/is_same.hpp>
Chris@16: 
Chris@16: namespace boost { namespace iostreams {
Chris@16: 
Chris@16: //
Chris@16: // Template name: tee_filter.
Chris@16: // Template parameters:
Chris@16: //      Device - A blocking Sink.
Chris@16: //
Chris@16: template<typename Device>
Chris@16: class tee_filter : public detail::filter_adapter<Device> {
Chris@16: public:
Chris@16:     typedef typename detail::param_type<Device>::type  param_type;
Chris@16:     typedef typename char_type_of<Device>::type        char_type;
Chris@16:     struct category
Chris@16:         : dual_use_filter_tag,
Chris@16:           multichar_tag,
Chris@16:           closable_tag,
Chris@16:           flushable_tag,
Chris@16:           localizable_tag,
Chris@16:           optimally_buffered_tag
Chris@16:         { };
Chris@16: 
Chris@16:     BOOST_STATIC_ASSERT(is_device<Device>::value);
Chris@16:     BOOST_STATIC_ASSERT((
Chris@16:         is_convertible< // Using mode_of causes failures on VC6-7.0.
Chris@16:             BOOST_DEDUCED_TYPENAME iostreams::category_of<Device>::type, output
Chris@16:         >::value
Chris@16:     ));
Chris@16: 
Chris@16:     explicit tee_filter(param_type dev) 
Chris@16:         : detail::filter_adapter<Device>(dev) 
Chris@16:         { }
Chris@16: 
Chris@16:     template<typename Source>
Chris@16:     std::streamsize read(Source& src, char_type* s, std::streamsize n)
Chris@16:     {
Chris@16:         std::streamsize result = iostreams::read(src, s, n);
Chris@16:         if (result != -1) {
Chris@16:             std::streamsize result2 = iostreams::write(this->component(), s, result);
Chris@16:             (void) result2; // Suppress 'unused variable' warning.
Chris@16:             BOOST_ASSERT(result == result2);
Chris@16:         }
Chris@16:         return result;
Chris@16:     }
Chris@16: 
Chris@16:     template<typename Sink>
Chris@16:     std::streamsize write(Sink& snk, const char_type* s, std::streamsize n)
Chris@16:     {
Chris@16:         std::streamsize result = iostreams::write(snk, s, n);
Chris@16:         std::streamsize result2 = iostreams::write(this->component(), s, result);
Chris@16:         (void) result2; // Suppress 'unused variable' warning.
Chris@16:         BOOST_ASSERT(result == result2);
Chris@16:         return result;
Chris@16:     }
Chris@16: 
Chris@16:     template<typename Next>
Chris@16:     void close(Next&, BOOST_IOS::openmode)
Chris@16:     { 
Chris@16:         detail::close_all(this->component());
Chris@16:     }
Chris@16: 
Chris@16:     template<typename Sink>
Chris@16:     bool flush(Sink& snk)
Chris@16:     {
Chris@16:         bool r1 = iostreams::flush(snk);
Chris@16:         bool r2 = iostreams::flush(this->component());
Chris@16:         return r1 && r2;
Chris@16:     }
Chris@16: };
Chris@16: BOOST_IOSTREAMS_PIPABLE(tee_filter, 1)
Chris@16: 
Chris@16: //
Chris@16: // Template name: tee_device.
Chris@16: // Template parameters:
Chris@16: //      Device - A blocking Device.
Chris@16: //      Sink - A blocking Sink.
Chris@16: //
Chris@16: template<typename Device, typename Sink>
Chris@16: class tee_device {
Chris@16: public:
Chris@16:     typedef typename detail::param_type<Device>::type  device_param;
Chris@16:     typedef typename detail::param_type<Sink>::type    sink_param;
Chris@16:     typedef typename detail::value_type<Device>::type  device_value;
Chris@16:     typedef typename detail::value_type<Sink>::type    sink_value;
Chris@16:     typedef typename char_type_of<Device>::type        char_type;
Chris@16:     typedef typename
Chris@16:             mpl::if_<
Chris@16:                  is_convertible<
Chris@16:                      BOOST_DEDUCED_TYPENAME 
Chris@16:                          iostreams::category_of<Device>::type, 
Chris@16:                      output
Chris@16:                  >,
Chris@16:                  output,
Chris@16:                  input
Chris@16:             >::type                                    mode;
Chris@16:     BOOST_STATIC_ASSERT(is_device<Device>::value);
Chris@16:     BOOST_STATIC_ASSERT(is_device<Sink>::value);
Chris@16:     BOOST_STATIC_ASSERT((
Chris@16:         is_same<
Chris@16:             char_type, 
Chris@16:             BOOST_DEDUCED_TYPENAME char_type_of<Sink>::type
Chris@16:         >::value
Chris@16:     ));
Chris@16:     BOOST_STATIC_ASSERT((
Chris@16:         is_convertible<
Chris@16:             BOOST_DEDUCED_TYPENAME iostreams::category_of<Sink>::type, 
Chris@16:             output
Chris@16:         >::value
Chris@16:     ));
Chris@16:     struct category
Chris@16:         : mode,
Chris@16:           device_tag,
Chris@16:           closable_tag,
Chris@16:           flushable_tag,
Chris@16:           localizable_tag,
Chris@16:           optimally_buffered_tag
Chris@16:         { };
Chris@16:     tee_device(device_param device, sink_param sink) 
Chris@16:         : dev_(device), sink_(sink)
Chris@16:         { }
Chris@16:     std::streamsize read(char_type* s, std::streamsize n)
Chris@16:     {
Chris@16:         BOOST_STATIC_ASSERT((
Chris@16:             is_convertible<
Chris@16:                 BOOST_DEDUCED_TYPENAME iostreams::category_of<Device>::type, input
Chris@16:             >::value
Chris@16:         ));
Chris@16:         std::streamsize result1 = iostreams::read(dev_, s, n);
Chris@16:         if (result1 != -1) {
Chris@16:             std::streamsize result2 = iostreams::write(sink_, s, result1);
Chris@16:             (void) result1; // Suppress 'unused variable' warning.
Chris@16:             (void) result2;
Chris@16:             BOOST_ASSERT(result1 == result2);
Chris@16:         }
Chris@16:         return result1;
Chris@16:     }
Chris@16:     std::streamsize write(const char_type* s, std::streamsize n)
Chris@16:     {
Chris@16:         BOOST_STATIC_ASSERT((
Chris@16:             is_convertible<
Chris@16:                 BOOST_DEDUCED_TYPENAME iostreams::category_of<Device>::type, output
Chris@16:             >::value
Chris@16:         ));
Chris@16:         std::streamsize result1 = iostreams::write(dev_, s, n);
Chris@16:         std::streamsize result2 = iostreams::write(sink_, s, n);
Chris@16:         (void) result1; // Suppress 'unused variable' warning.
Chris@16:         (void) result2;
Chris@16:         BOOST_ASSERT(result1 == n && result2 == n);
Chris@16:         return n;
Chris@16:     }
Chris@16:     void close()
Chris@16:     {
Chris@16:         detail::execute_all( detail::call_close_all(dev_),
Chris@16:                              detail::call_close_all(sink_) );
Chris@16:     }
Chris@16:     bool flush()
Chris@16:     {
Chris@16:         bool r1 = iostreams::flush(dev_);
Chris@16:         bool r2 = iostreams::flush(sink_);
Chris@16:         return r1 && r2;
Chris@16:     }
Chris@16:     template<typename Locale>
Chris@16:     void imbue(const Locale& loc)
Chris@16:     {
Chris@16:         iostreams::imbue(dev_, loc);
Chris@16:         iostreams::imbue(sink_, loc);
Chris@16:     }
Chris@16:     std::streamsize optimal_buffer_size() const 
Chris@16:     {
Chris@16:         return (std::max) ( iostreams::optimal_buffer_size(dev_), 
Chris@16:                             iostreams::optimal_buffer_size(sink_) );
Chris@16:     }
Chris@16: private:
Chris@16:     device_value  dev_;
Chris@16:     sink_value    sink_;
Chris@16: };
Chris@16: 
Chris@16: template<typename Sink>
Chris@16: tee_filter<Sink> tee(Sink& snk) 
Chris@16: { return tee_filter<Sink>(snk); }
Chris@16: 
Chris@16: template<typename Sink>
Chris@16: tee_filter<Sink> tee(const Sink& snk) 
Chris@16: { return tee_filter<Sink>(snk); }
Chris@16: 
Chris@16: template<typename Device, typename Sink>
Chris@16: tee_device<Device, Sink> tee(Device& dev, Sink& sink) 
Chris@16: { return tee_device<Device, Sink>(dev, sink); }
Chris@16: 
Chris@16: template<typename Device, typename Sink>
Chris@16: tee_device<Device, Sink> tee(const Device& dev, Sink& sink) 
Chris@16: { return tee_device<Device, Sink>(dev, sink); }
Chris@16: 
Chris@16: template<typename Device, typename Sink>
Chris@16: tee_device<Device, Sink> tee(Device& dev, const Sink& sink) 
Chris@16: { return tee_device<Device, Sink>(dev, sink); }
Chris@16: 
Chris@16: template<typename Device, typename Sink>
Chris@16: tee_device<Device, Sink> tee(const Device& dev, const Sink& sink) 
Chris@16: { return tee_device<Device, Sink>(dev, sink); }
Chris@16: 
Chris@16: } } // End namespaces iostreams, boost.
Chris@16: 
Chris@16: #endif // #ifndef BOOST_IOSTREAMS_TEE_HPP_INCLUDED