sl@0: // (C) Copyright Jonathan Turkanis 2003.
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: // Contains: The function template copy, which reads data from a Source 
sl@0: // and writes it to a Sink until the end of the sequence is reached, returning 
sl@0: // the number of characters transfered.
sl@0: 
sl@0: // The implementation is complicated by the need to handle smart adapters
sl@0: // and direct devices.
sl@0: 
sl@0: #ifndef BOOST_IOSTREAMS_COPY_HPP_INCLUDED
sl@0: #define BOOST_IOSTREAMS_COPY_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 <algorithm>                        // copy.
sl@0: #include <utility>                          // pair.
sl@0: #include <boost/detail/workaround.hpp>
sl@0: #include <boost/iostreams/chain.hpp>
sl@0: #include <boost/iostreams/constants.hpp>
sl@0: #include <boost/iostreams/detail/adapter/non_blocking_adapter.hpp>        
sl@0: #include <boost/iostreams/detail/buffer.hpp>       
sl@0: #include <boost/iostreams/detail/closer.hpp>    
sl@0: #include <boost/iostreams/detail/enable_if_stream.hpp>  
sl@0: #include <boost/iostreams/detail/ios.hpp>   // failure, streamsize.                   
sl@0: #include <boost/iostreams/detail/resolve.hpp>                   
sl@0: #include <boost/iostreams/detail/wrap_unwrap.hpp>
sl@0: #include <boost/iostreams/operations.hpp>  // read, write, close.
sl@0: #include <boost/iostreams/pipeline.hpp>
sl@0: #include <boost/static_assert.hpp>  
sl@0: #include <boost/type_traits/is_same.hpp> 
sl@0: 
sl@0: namespace boost { namespace iostreams {
sl@0: 
sl@0: namespace detail {
sl@0: 
sl@0: template<typename Source, typename Sink>
sl@0: std::streamsize copy_impl( Source& src, Sink& snk, 
sl@0:                            std::streamsize /* buffer_size */,
sl@0:                            mpl::true_, mpl::true_ )
sl@0: {   // Copy from a direct Source to a direct Sink.
sl@0:     using namespace std;
sl@0:     typedef typename char_type_of<Source>::type  char_type;
sl@0:     typedef pair<char_type*, char_type*>         pair_type;
sl@0:     pair_type p1 = iostreams::input_sequence(src);
sl@0:     pair_type p2 = iostreams::output_sequence(snk);
sl@0:     if (p1.second - p1.first < p2.second - p2.first) {
sl@0:         std::copy(p1.first, p1.second, p2.first);
sl@0:         return static_cast<streamsize>(p1.second - p1.first);
sl@0:     } else {
sl@0:         throw BOOST_IOSTREAMS_FAILURE("destination too small");
sl@0:     }
sl@0: }
sl@0: 
sl@0: template<typename Source, typename Sink>
sl@0: std::streamsize copy_impl( Source& src, Sink& snk, 
sl@0:                            std::streamsize /* buffer_size */,
sl@0:                            mpl::true_, mpl::false_ )
sl@0: {   // Copy from a direct Source to an indirect Sink.
sl@0:     using namespace std;
sl@0:     typedef typename char_type_of<Source>::type  char_type;
sl@0:     typedef pair<char_type*, char_type*>         pair_type;
sl@0:     pair_type p = iostreams::input_sequence(src);
sl@0:     std::streamsize size, total;
sl@0:     for ( total = 0, size = static_cast<streamsize>(p.second - p.first);
sl@0:           total < size; )
sl@0:     {
sl@0:         std::streamsize amt = 
sl@0:             iostreams::write(snk, p.first + total, size - total); 
sl@0:         total += amt;
sl@0:     }
sl@0:     return size;
sl@0: }
sl@0: 
sl@0: template<typename Source, typename Sink>
sl@0: std::streamsize copy_impl( Source& src, Sink& snk, 
sl@0:                            std::streamsize buffer_size,
sl@0:                            mpl::false_, mpl::true_ )
sl@0: {   // Copy from an indirect Source to a direct Sink.
sl@0:     using namespace std;
sl@0:     typedef typename char_type_of<Source>::type  char_type;
sl@0:     typedef pair<char_type*, char_type*>         pair_type;
sl@0:     detail::basic_buffer<char_type>  buf(buffer_size);
sl@0:     pair_type                        p = snk.output_sequence();
sl@0:     streamsize                       total = 0;
sl@0:     bool                             done  = false;
sl@0:     while (!done) {
sl@0:         streamsize amt;
sl@0:         done = (amt = iostreams::read(src, buf.data(), buffer_size)) == -1;
sl@0:         std::copy(buf.data(), buf.data() + amt, p.first + total);
sl@0:         if (amt != -1)
sl@0:             total += amt;
sl@0:     }
sl@0:     return total;
sl@0: }
sl@0: 
sl@0: template<typename Source, typename Sink>
sl@0: std::streamsize copy_impl( Source& src, Sink& snk, 
sl@0:                            std::streamsize buffer_size,
sl@0:                            mpl::false_, mpl::false_ )
sl@0: {   // Copy from an indirect Source to a indirect Sink. This algorithm
sl@0:     // can be improved by eliminating the non_blocking_adapter.
sl@0:     typedef typename char_type_of<Source>::type char_type;
sl@0:     detail::basic_buffer<char_type>  buf(buffer_size);
sl@0:     non_blocking_adapter<Sink>       nb(snk);
sl@0:     std::streamsize                  total = 0;
sl@0:     bool                             done = false;
sl@0:     while (!done) {
sl@0:         std::streamsize amt;
sl@0:         done = (amt = iostreams::read(src, buf.data(), buffer_size)) == -1;
sl@0:         if (amt != -1) {
sl@0:             iostreams::write(nb, buf.data(), amt);
sl@0:             total += amt;
sl@0:         }
sl@0:     }
sl@0:     return total;
sl@0: }
sl@0: 
sl@0: template<typename Source, typename Sink>
sl@0: std::streamsize copy_impl(Source src, Sink snk, std::streamsize buffer_size)
sl@0: {
sl@0:     using namespace std;
sl@0:     typedef typename char_type_of<Source>::type  src_char;
sl@0:     typedef typename char_type_of<Sink>::type    snk_char;
sl@0:     BOOST_STATIC_ASSERT((is_same<src_char, snk_char>::value));
sl@0:     bool                     nothrow = false;
sl@0:     external_closer<Source>  close_source(src, BOOST_IOS::in, nothrow);
sl@0:     external_closer<Sink>    close_sink(snk, BOOST_IOS::out, nothrow);
sl@0:     streamsize result =
sl@0:         copy_impl( src, snk, buffer_size, 
sl@0:                    is_direct<Source>(), is_direct<Sink>() );
sl@0:     return result; 
sl@0: }
sl@0: 
sl@0: } // End namespace detail.
sl@0:                     
sl@0: //------------------Definition of copy----------------------------------------//
sl@0: 
sl@0: template<typename Source, typename Sink>
sl@0: std::streamsize
sl@0: copy( const Source& src, const Sink& snk,
sl@0:       std::streamsize buffer_size = default_device_buffer_size
sl@0:       BOOST_IOSTREAMS_DISABLE_IF_STREAM(Source)
sl@0:       BOOST_IOSTREAMS_DISABLE_IF_STREAM(Sink) )
sl@0: { 
sl@0:     typedef typename char_type_of<Source>::type char_type;
sl@0:     return detail::copy_impl( detail::resolve<input, char_type>(src), 
sl@0:                               detail::resolve<output, char_type>(snk), 
sl@0:                               buffer_size ); 
sl@0: }
sl@0: 
sl@0: #if !BOOST_WORKAROUND(BOOST_MSVC, <= 1300) //---------------------------------//
sl@0: 
sl@0: template<typename Source, typename Sink>
sl@0: std::streamsize
sl@0: copy( Source& src, const Sink& snk,
sl@0:       std::streamsize buffer_size = default_device_buffer_size
sl@0:       BOOST_IOSTREAMS_ENABLE_IF_STREAM(Source)
sl@0:       BOOST_IOSTREAMS_DISABLE_IF_STREAM(Sink) ) 
sl@0: { 
sl@0:     typedef typename char_type_of<Source>::type char_type;
sl@0:     return detail::copy_impl( detail::wrap(src), 
sl@0:                               detail::resolve<output, char_type>(snk), 
sl@0:                               buffer_size );
sl@0: }
sl@0: 
sl@0: template<typename Source, typename Sink>
sl@0: std::streamsize
sl@0: copy( const Source& src, Sink& snk,
sl@0:       std::streamsize buffer_size = default_device_buffer_size
sl@0:       BOOST_IOSTREAMS_DISABLE_IF_STREAM(Source)
sl@0:       BOOST_IOSTREAMS_ENABLE_IF_STREAM(Sink) ) 
sl@0: { 
sl@0:     typedef typename char_type_of<Source>::type char_type;
sl@0:     return detail::copy_impl( detail::resolve<input, char_type>(src), 
sl@0:                               detail::wrap(snk), buffer_size);
sl@0: }
sl@0: 
sl@0: template<typename Source, typename Sink>
sl@0: std::streamsize
sl@0: copy( Source& src, Sink& snk,
sl@0:       std::streamsize buffer_size = default_device_buffer_size
sl@0:       BOOST_IOSTREAMS_ENABLE_IF_STREAM(Source)
sl@0:       BOOST_IOSTREAMS_ENABLE_IF_STREAM(Sink) ) 
sl@0: { 
sl@0:     return detail::copy_impl(detail::wrap(src), detail::wrap(snk), buffer_size);
sl@0: }
sl@0: 
sl@0: #endif // #if !BOOST_WORKAROUND(BOOST_MSVC, <= 1300) //-----------------------//
sl@0: 
sl@0: } } // End namespaces iostreams, boost.
sl@0: 
sl@0: #endif // #ifndef BOOST_IOSTREAMS_COPY_HPP_INCLUDED