1// (C) Copyright Gennadiy Rozental 2001.
2// Distributed under the Boost Software License, Version 1.0.
3// (See accompanying file LICENSE_1_0.txt or copy at
4// http://www.boost.org/LICENSE_1_0.txt)
5
6// See http://www.boost.org/libs/test for the library home page.
7//
8// File : $RCSfile$
9//
10// Version : $Revision$
11//
12// Description : wraps strstream and stringstream (depends with one is present)
13// to provide the unified interface
14// ***************************************************************************
15
16#ifndef BOOST_TEST_UTILS_WRAP_STRINGSTREAM_HPP
17#define BOOST_TEST_UTILS_WRAP_STRINGSTREAM_HPP
18
19// Boost.Test
20#include <boost/test/detail/config.hpp>
21
22// STL
23#ifdef BOOST_NO_STRINGSTREAM
24#include <strstream> // for std::ostrstream
25#else
26#include <sstream> // for std::ostringstream
27#endif // BOOST_NO_STRINGSTREAM
28
29#include <boost/test/detail/suppress_warnings.hpp>
30
31//____________________________________________________________________________//
32
33namespace boost {
34
35// ************************************************************************** //
36// ************** basic_wrap_stringstream ************** //
37// ************************************************************************** //
38
39template<typename CharT>
40class basic_wrap_stringstream {
41public:
42#if defined(BOOST_CLASSIC_IOSTREAMS)
43 typedef std::ostringstream wrapped_stream;
44#elif defined(BOOST_NO_STRINGSTREAM)
45 typedef std::basic_ostrstream<CharT> wrapped_stream;
46#else
47 typedef std::basic_ostringstream<CharT> wrapped_stream;
48#endif // BOOST_NO_STRINGSTREAM
49 // Access methods
50 basic_wrap_stringstream& ref();
51 wrapped_stream& stream();
52 std::basic_string<CharT> const& str();
53
54private:
55 // Data members
56 wrapped_stream m_stream;
57 std::basic_string<CharT> m_str;
58};
59
60//____________________________________________________________________________//
61
62template <typename CharT, typename T>
63inline basic_wrap_stringstream<CharT>&
64operator<<( basic_wrap_stringstream<CharT>& targ, T const& t )
65{
66 targ.stream() << t;
67 return targ;
68}
69
70//____________________________________________________________________________//
71
72template <typename CharT>
73inline typename basic_wrap_stringstream<CharT>::wrapped_stream&
74basic_wrap_stringstream<CharT>::stream()
75{
76 return m_stream;
77}
78
79//____________________________________________________________________________//
80
81template <typename CharT>
82inline basic_wrap_stringstream<CharT>&
83basic_wrap_stringstream<CharT>::ref()
84{
85 return *this;
86}
87
88//____________________________________________________________________________//
89
90template <typename CharT>
91inline std::basic_string<CharT> const&
92basic_wrap_stringstream<CharT>::str()
93{
94
95#ifdef BOOST_NO_STRINGSTREAM
96 m_str.assign( m_stream.str(), m_stream.pcount() );
97 m_stream.freeze( false );
98#else
99 m_str = m_stream.str();
100#endif
101
102 return m_str;
103}
104
105//____________________________________________________________________________//
106
107template <typename CharT>
108inline basic_wrap_stringstream<CharT>&
109operator<<( basic_wrap_stringstream<CharT>& targ, basic_wrap_stringstream<CharT>& src )
110{
111 targ << src.str();
112 return targ;
113}
114
115//____________________________________________________________________________//
116
117#if BOOST_TEST_USE_STD_LOCALE
118
119template <typename CharT>
120inline basic_wrap_stringstream<CharT>&
121operator<<( basic_wrap_stringstream<CharT>& targ, std::ios_base& (BOOST_TEST_CALL_DECL *man)(std::ios_base&) )
122{
123 targ.stream() << man;
124 return targ;
125}
126
127//____________________________________________________________________________//
128
129template<typename CharT,typename Elem,typename Tr>
130inline basic_wrap_stringstream<CharT>&
131operator<<( basic_wrap_stringstream<CharT>& targ, std::basic_ostream<Elem,Tr>& (BOOST_TEST_CALL_DECL *man)(std::basic_ostream<Elem, Tr>&) )
132{
133 targ.stream() << man;
134 return targ;
135}
136
137//____________________________________________________________________________//
138
139template<typename CharT,typename Elem,typename Tr>
140inline basic_wrap_stringstream<CharT>&
141operator<<( basic_wrap_stringstream<CharT>& targ, std::basic_ios<Elem, Tr>& (BOOST_TEST_CALL_DECL *man)(std::basic_ios<Elem, Tr>&) )
142{
143 targ.stream() << man;
144 return targ;
145}
146
147//____________________________________________________________________________//
148
149#endif
150
151// ************************************************************************** //
152// ************** wrap_stringstream ************** //
153// ************************************************************************** //
154
155typedef basic_wrap_stringstream<char> wrap_stringstream;
156typedef basic_wrap_stringstream<wchar_t> wrap_wstringstream;
157
158} // namespace boost
159
160#include <boost/test/detail/enable_warnings.hpp>
161
162#endif // BOOST_TEST_UTILS_WRAP_STRINGSTREAM_HPP
163