]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
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 | ||
33 | namespace boost { | |
34 | ||
35 | // ************************************************************************** // | |
36 | // ************** basic_wrap_stringstream ************** // | |
37 | // ************************************************************************** // | |
38 | ||
39 | template<typename CharT> | |
40 | class basic_wrap_stringstream { | |
41 | public: | |
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 | ||
54 | private: | |
55 | // Data members | |
56 | wrapped_stream m_stream; | |
57 | std::basic_string<CharT> m_str; | |
58 | }; | |
59 | ||
60 | //____________________________________________________________________________// | |
61 | ||
62 | template <typename CharT, typename T> | |
63 | inline basic_wrap_stringstream<CharT>& | |
64 | operator<<( basic_wrap_stringstream<CharT>& targ, T const& t ) | |
65 | { | |
66 | targ.stream() << t; | |
67 | return targ; | |
68 | } | |
69 | ||
70 | //____________________________________________________________________________// | |
71 | ||
72 | template <typename CharT> | |
73 | inline typename basic_wrap_stringstream<CharT>::wrapped_stream& | |
74 | basic_wrap_stringstream<CharT>::stream() | |
75 | { | |
76 | return m_stream; | |
77 | } | |
78 | ||
79 | //____________________________________________________________________________// | |
80 | ||
81 | template <typename CharT> | |
82 | inline basic_wrap_stringstream<CharT>& | |
83 | basic_wrap_stringstream<CharT>::ref() | |
84 | { | |
85 | return *this; | |
86 | } | |
87 | ||
88 | //____________________________________________________________________________// | |
89 | ||
90 | template <typename CharT> | |
91 | inline std::basic_string<CharT> const& | |
92 | basic_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 | ||
107 | template <typename CharT> | |
108 | inline basic_wrap_stringstream<CharT>& | |
109 | operator<<( 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 | ||
119 | template <typename CharT> | |
120 | inline basic_wrap_stringstream<CharT>& | |
121 | operator<<( 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 | ||
129 | template<typename CharT,typename Elem,typename Tr> | |
130 | inline basic_wrap_stringstream<CharT>& | |
131 | operator<<( 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 | ||
139 | template<typename CharT,typename Elem,typename Tr> | |
140 | inline basic_wrap_stringstream<CharT>& | |
141 | operator<<( 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 | ||
155 | typedef basic_wrap_stringstream<char> wrap_stringstream; | |
156 | typedef 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 |