blob: d4b92a4133e11a7e459bff19e8b6196e2c433ab4 [file] [log] [blame]
// (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.
// Based on the work of Christopher Diggins.
#ifndef BOOST_IOSTREAMS_STDIO_FILTER_HPP_INCLUDED
#define BOOST_IOSTREAMS_STDIO_FILTER_HPP_INCLUDED
#if defined(_MSC_VER) && (_MSC_VER >= 1020)
# pragma once
#endif
#include <iostream>
#include <memory> // allocator.
#include <vector>
#include <boost/iostreams/detail/config/wide_streams.hpp>
#include <boost/iostreams/detail/char_traits.hpp>
#include <boost/iostreams/detail/ios.hpp>
#include <boost/iostreams/device/array.hpp>
#include <boost/iostreams/device/back_inserter.hpp>
#include <boost/iostreams/filter/aggregate.hpp>
#include <boost/iostreams/pipeline.hpp>
#include <boost/iostreams/stream_buffer.hpp>
namespace boost { namespace iostreams {
namespace detail {
} // End namespace detail.
template<typename Ch, typename Alloc = std::allocator<Ch> >
class basic_stdio_filter : public aggregate_filter<Ch, Alloc> {
private:
typedef aggregate_filter<Ch, Alloc> base_type;
public:
typedef typename base_type::char_type char_type;
typedef typename base_type::category category;
typedef typename base_type::vector_type vector_type;
private:
static std::istream& standard_input(char*) { return std::cin; }
static std::ostream& standard_output(char*) { return std::cout; }
#ifndef BOOST_IOSTREAMS_NO_WIDE_STREAMS
static std::wistream& standard_input(wchar_t*) { return std::wcin; }
static std::wostream& standard_output(wchar_t*) { return std::wcout; }
#endif // BOOST_IOSTREAMS_NO_WIDE_STREAMS
struct scoped_redirector { // Thanks to Maxim Egorushkin.
typedef BOOST_IOSTREAMS_CHAR_TRAITS(Ch) traits_type;
typedef BOOST_IOSTREAMS_BASIC_IOS(Ch, traits_type) ios_type;
typedef BOOST_IOSTREAMS_BASIC_STREAMBUF(Ch, traits_type) streambuf_type;
scoped_redirector( ios_type& ios,
streambuf_type* newbuf )
: ios_(ios), old_(ios.rdbuf(newbuf))
{ }
~scoped_redirector() { ios_.rdbuf(old_); }
scoped_redirector& operator=(const scoped_redirector&);
ios_type& ios_;
streambuf_type* old_;
};
virtual void do_filter() = 0;
virtual void do_filter(const vector_type& src, vector_type& dest)
{
stream_buffer< basic_array_source<Ch> >
srcbuf(&src[0], &src[0] + src.size());
stream_buffer< back_insert_device<vector_type> >
destbuf(iostreams::back_inserter(dest));
scoped_redirector redirect_input(standard_input((Ch*)0), &srcbuf);
scoped_redirector redirect_output(standard_output((Ch*)0), &destbuf);
do_filter();
}
};
BOOST_IOSTREAMS_PIPABLE(basic_stdio_filter, 2)
typedef basic_stdio_filter<char> stdio_filter;
typedef basic_stdio_filter<wchar_t> wstdio_wfilter;
} } // End namespaces iostreams, boost.
#endif // #ifndef BOOST_IOSTREAMS_STDIO_FILTER_HPP_INCLUDED