Chris@16: // (C) Copyright 2008 CodeRage, LLC (turkanis at coderage dot com)
Chris@16: // (C) Copyright 2003-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_CLOSE_HPP_INCLUDED
Chris@16: #define BOOST_IOSTREAMS_CLOSE_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/config.hpp>  // DEDUCED_TYPENAME, MSVC.
Chris@16: #include <boost/detail/workaround.hpp>
Chris@16: #include <boost/iostreams/categories.hpp>
Chris@16: #include <boost/iostreams/flush.hpp>
Chris@16: #include <boost/iostreams/detail/adapter/non_blocking_adapter.hpp>
Chris@16: #include <boost/iostreams/detail/ios.hpp> // BOOST_IOS
Chris@16: #include <boost/iostreams/detail/select.hpp>
Chris@16: #include <boost/iostreams/detail/wrap_unwrap.hpp>
Chris@16: #include <boost/iostreams/operations_fwd.hpp>
Chris@16: #include <boost/iostreams/traits.hpp>
Chris@16: #include <boost/mpl/identity.hpp>
Chris@16: #include <boost/mpl/if.hpp>
Chris@16: #include <boost/type_traits/is_convertible.hpp>
Chris@16: #include <boost/type_traits/is_integral.hpp>
Chris@16: #include <boost/type_traits/remove_cv.hpp>
Chris@16: #include <boost/type_traits/remove_reference.hpp>
Chris@16: 
Chris@16: // Must come last.
Chris@16: #include <boost/iostreams/detail/config/disable_warnings.hpp>
Chris@16: 
Chris@16: namespace boost { namespace iostreams {
Chris@16: 
Chris@16: template<typename T>
Chris@16: void close(T& t);
Chris@16: 
Chris@16: template<typename T>
Chris@16: void close(T& t, BOOST_IOS::openmode which);
Chris@16: 
Chris@16: template<typename T, typename Sink>
Chris@16: void close(T& t, Sink& snk, BOOST_IOS::openmode which);
Chris@16:     
Chris@16: namespace detail {
Chris@16: 
Chris@16: template<typename T>
Chris@16: void close_all(T& t)
Chris@16: { 
Chris@16:     try {
Chris@16:         boost::iostreams::close(t, BOOST_IOS::in);
Chris@16:     } catch (...) {
Chris@16:         try {
Chris@16:             boost::iostreams::close(t, BOOST_IOS::out);
Chris@16:         } catch (...) { }
Chris@16:         throw;
Chris@16:     }
Chris@16:     boost::iostreams::close(t, BOOST_IOS::out);
Chris@16: }
Chris@16: 
Chris@16: template<typename T, typename Sink>
Chris@16: void close_all(T& t, Sink& snk)
Chris@16: { 
Chris@16:     try {
Chris@16:         boost::iostreams::close(t, snk, BOOST_IOS::in);
Chris@16:     } catch (...) {
Chris@16:         try {
Chris@16:             boost::iostreams::close(t, snk, BOOST_IOS::out);
Chris@16:         } catch (...) { }
Chris@16:         throw;
Chris@16:     }
Chris@16:     boost::iostreams::close(t, snk, BOOST_IOS::out);
Chris@16: }
Chris@16: 
Chris@16: } // End namespace detail. 
Chris@16: 
Chris@16: } } // End namespaces iostreams, boost.
Chris@16: 
Chris@16: #if BOOST_WORKAROUND(BOOST_MSVC, < 1300) //-----------------------------------//
Chris@16: # include <boost/iostreams/detail/vc6/close.hpp>
Chris@16: #else // #if BOOST_WORKAROUND(BOOST_MSVC, < 1300) //--------------------------//
Chris@16: 
Chris@16: namespace boost { namespace iostreams {
Chris@16: 
Chris@16: namespace detail {
Chris@16: 
Chris@16: template<typename T>
Chris@16: struct close_impl;
Chris@16: 
Chris@16: } // End namespace detail.
Chris@16: 
Chris@16: template<typename T>
Chris@16: void close(T& t) { detail::close_all(t); }
Chris@16: 
Chris@16: template<typename T>
Chris@16: void close(T& t, BOOST_IOS::openmode which)
Chris@16: { 
Chris@16: #ifdef BOOST_IOSTREAMS_STRICT
Chris@16:     BOOST_ASSERT(which == BOOST_IOS::in || which == BOOST_IOS::out);
Chris@16: #else
Chris@16:     if (which == (BOOST_IOS::in | BOOST_IOS::out)) {
Chris@16:         detail::close_all(t);
Chris@16:         return;
Chris@16:     }
Chris@16: #endif
Chris@16:     detail::close_impl<T>::close(detail::unwrap(t), which); 
Chris@16: }
Chris@16: 
Chris@16: template<typename T, typename Sink>
Chris@16: void close(T& t, Sink& snk, BOOST_IOS::openmode which)
Chris@16: { 
Chris@16: #ifdef BOOST_IOSTREAMS_STRICT
Chris@16:     BOOST_ASSERT(which == BOOST_IOS::in || which == BOOST_IOS::out);
Chris@16: #else
Chris@16:     if (which == (BOOST_IOS::in | BOOST_IOS::out)) {
Chris@16:         detail::close_all(t, snk);
Chris@16:         return;
Chris@16:     }
Chris@16: #endif
Chris@16:     detail::close_impl<T>::close(detail::unwrap(t), snk, which); 
Chris@16: }
Chris@16: 
Chris@16: namespace detail {
Chris@16: 
Chris@16: //------------------Definition of close_impl----------------------------------//
Chris@16: 
Chris@16: struct close_boost_stream { };
Chris@16: struct close_filtering_stream { };
Chris@16: 
Chris@16: template<typename T>
Chris@16: struct close_tag {
Chris@16:     typedef typename category_of<T>::type             category;
Chris@16:     typedef typename detail::unwrapped_type<T>::type  unwrapped;
Chris@16:     typedef typename
Chris@16:             iostreams::select<
Chris@16:                 mpl::not_< is_convertible<category, closable_tag> >,
Chris@16:                 any_tag,
Chris@16:                 mpl::or_<
Chris@16:                     is_boost_stream<unwrapped>,
Chris@16:                     is_boost_stream_buffer<unwrapped>
Chris@16:                 >,
Chris@16:                 close_boost_stream,
Chris@16:                 mpl::or_<
Chris@16:                     is_filtering_stream<unwrapped>,
Chris@16:                     is_filtering_streambuf<unwrapped>
Chris@16:                 >,
Chris@16:                 close_filtering_stream,
Chris@16:                 mpl::or_<
Chris@16:                     is_convertible<category, two_sequence>,
Chris@16:                     is_convertible<category, dual_use>
Chris@16:                 >,
Chris@16:                 two_sequence,
Chris@16:                 else_,
Chris@16:                 closable_tag
Chris@16:             >::type type;
Chris@16: };
Chris@16: 
Chris@16: template<typename T>
Chris@16: struct close_impl
Chris@16:     : mpl::if_<
Chris@16:           is_custom<T>,
Chris@16:           operations<T>,
Chris@16:           close_impl<BOOST_DEDUCED_TYPENAME close_tag<T>::type>
Chris@16:       >::type
Chris@16:     { };
Chris@16: 
Chris@16: template<>
Chris@16: struct close_impl<any_tag> {
Chris@16:     template<typename T>
Chris@16:     static void close(T& t, BOOST_IOS::openmode which)
Chris@16:     {
Chris@16:         if (which == BOOST_IOS::out)
Chris@16:             iostreams::flush(t);
Chris@16:     }
Chris@16: 
Chris@16:     template<typename T, typename Sink>
Chris@16:     static void close(T& t, Sink& snk, BOOST_IOS::openmode which)
Chris@16:     {
Chris@16:         if (which == BOOST_IOS::out) {
Chris@16:             non_blocking_adapter<Sink> nb(snk);
Chris@16:             iostreams::flush(t, nb);
Chris@16:         }
Chris@16:     }
Chris@16: };
Chris@16: 
Chris@16: template<>
Chris@16: struct close_impl<close_boost_stream> {
Chris@16:     template<typename T>
Chris@16:     static void close(T& t)
Chris@16:     {
Chris@16:         t.close();
Chris@16:     }
Chris@16:     template<typename T>
Chris@16:     static void close(T& t, BOOST_IOS::openmode which)
Chris@16:     {
Chris@16:         if (which == BOOST_IOS::out)
Chris@16:             t.close();
Chris@16:     }
Chris@16: };
Chris@16: 
Chris@16: template<>
Chris@16: struct close_impl<close_filtering_stream> {
Chris@16:     template<typename T>
Chris@16:     static void close(T& t, BOOST_IOS::openmode which)
Chris@16:     {
Chris@16:         typedef typename category_of<T>::type category;
Chris@16:         const bool in =  is_convertible<category, input>::value &&
Chris@16:                         !is_convertible<category, output>::value;
Chris@16:         if (in == (which == BOOST_IOS::in) && t.is_complete())
Chris@16:             t.pop();
Chris@16:     }
Chris@16: };
Chris@16: 
Chris@16: template<>
Chris@16: struct close_impl<closable_tag> {
Chris@16:     template<typename T>
Chris@16:     static void close(T& t, BOOST_IOS::openmode which)
Chris@16:     {
Chris@16:         typedef typename category_of<T>::type category;
Chris@16:         const bool in =  is_convertible<category, input>::value &&
Chris@16:                         !is_convertible<category, output>::value;
Chris@16:         if (in == (which == BOOST_IOS::in))
Chris@16:             t.close();
Chris@16:     }
Chris@16:     template<typename T, typename Sink>
Chris@16:     static void close(T& t, Sink& snk, BOOST_IOS::openmode which)
Chris@16:     {
Chris@16:         typedef typename category_of<T>::type category;
Chris@16:         const bool in =  is_convertible<category, input>::value &&
Chris@16:                         !is_convertible<category, output>::value;
Chris@16:         if (in == (which == BOOST_IOS::in)) {
Chris@16:             non_blocking_adapter<Sink> nb(snk);
Chris@16:             t.close(nb);
Chris@16:         }
Chris@16:     }
Chris@16: };
Chris@16: 
Chris@16: template<>
Chris@16: struct close_impl<two_sequence> {
Chris@16:     template<typename T>
Chris@16:     static void close(T& t, BOOST_IOS::openmode which) { t.close(which); }
Chris@16:     template<typename T, typename Sink>
Chris@16:     static void close(T& t, Sink& snk, BOOST_IOS::openmode which)
Chris@16:     {
Chris@16:         non_blocking_adapter<Sink> nb(snk);
Chris@16:         t.close(nb, which);
Chris@16:     }
Chris@16: };
Chris@16: 
Chris@16: } // End namespace detail.
Chris@16: 
Chris@16: } } // End namespaces iostreams, boost.
Chris@16: 
Chris@16: #endif // #if BOOST_WORKAROUND(BOOST_MSVC, < 1300) //-------------------------//
Chris@16: 
Chris@16: #include <boost/iostreams/detail/config/enable_warnings.hpp>
Chris@16: 
Chris@16: #endif // #ifndef BOOST_IOSTREAMS_CLOSE_HPP_INCLUDED