3rdparty/boost/boost/format/feed_args.hpp

   1 // ----------------------------------------------------------------------------
   2 //  feed_args.hpp :  functions for processing each argument
   3 //                      (feed, feed_manip, and distribute)
   4 // ----------------------------------------------------------------------------
   5
   6 //  Copyright Samuel Krempp 2003. Use, modification, and distribution are
   7 //  subject to the Boost Software License, Version 1.0. (See accompanying
   8 //  file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
   9
  10 //  See http://www.boost.org/libs/format for library home page
  11
  12 // ----------------------------------------------------------------------------
  13
  14 #ifndef BOOST_FORMAT_FEED_ARGS_HPP
  15 #define BOOST_FORMAT_FEED_ARGS_HPP
  16
  17 #include <boost/config.hpp>
  18 #include <boost/assert.hpp>
  19 #include <boost/throw_exception.hpp>
  20
  21 #include <boost/format/format_class.hpp>
  22 #include <boost/format/group.hpp>
  23 #include <boost/format/detail/msvc_disambiguater.hpp>
  24
  25 namespace boost {
  26 namespace io {
  27 namespace detail {
  28
  29     template<class Ch, class Tr, class Alloc>
  30     void mk_str( std::basic_string<Ch,Tr, Alloc> & res,
  31                  const Ch * beg,
  32                  typename std::basic_string<Ch,Tr,Alloc>::size_type size,
  33                  std::streamsize w,
  34                  const Ch fill_char,
  35                  std::ios_base::fmtflags f,
  36                  const Ch prefix_space, // 0 if no space-padding
  37                  bool center)
  38     // applies centered/left/right  padding  to the string  [beg, beg+size[
  39     // Effects : the result is placed in res.
  40     {
  41         typedef typename std::basic_string<Ch,Tr,Alloc>::size_type size_type;
  42         res.resize(0);
  43         if(w<=0 || static_cast<size_type>(w) <=size) {
  44             // no need to pad.
  45             res.reserve(size + !!prefix_space);
  46             if(prefix_space)
  47               res.append(1, prefix_space);
  48             if (size)
  49               res.append(beg, size);
  50         }
  51         else {
  52             std::streamsize n=static_cast<std::streamsize>(w-size-!!prefix_space);
  53             std::streamsize n_after = 0, n_before = 0;
  54             res.reserve(static_cast<size_type>(w)); // allocate once for the 2 inserts
  55             if(center)
  56                 n_after = n/2, n_before = n - n_after;
  57             else
  58                 if(f & std::ios_base::left)
  59                     n_after = n;
  60                 else
  61                     n_before = n;
  62             // now make the res string :
  63             if(n_before) res.append(static_cast<size_type>(n_before), fill_char);
  64             if(prefix_space)
  65               res.append(1, prefix_space);
  66             if (size)
  67               res.append(beg, size);
  68             if(n_after) res.append(static_cast<size_type>(n_after), fill_char);
  69         }
  70     } // -mk_str(..)
  71
  72
  73 #if BOOST_WORKAROUND(__DECCXX_VER, BOOST_TESTED_AT(60590042))
  74 // __DECCXX needs to be tricked to disambiguate this simple overload..
  75 // the trick is in "boost/format/msvc_disambiguater.hpp"
  76
  77     template< class Ch, class Tr, class T> inline
  78     void put_head (BOOST_IO_STD basic_ostream<Ch, Tr> & os, const T& x ) {
  79         disambiguater<Ch, Tr, T>::put_head(os, x, 1L);
  80     }
  81     template< class Ch, class Tr, class T> inline
  82     void put_last (BOOST_IO_STD basic_ostream<Ch, Tr> & os, const T& x ) {
  83         disambiguater<Ch, Tr, T>::put_last(os, x, 1L);
  84     }
  85
  86 #else
  87
  88     template< class Ch, class Tr, class T> inline
  89     void put_head (BOOST_IO_STD basic_ostream<Ch, Tr> &, const T& ) {
  90     }
  91
  92     template< class Ch, class Tr, class T> inline
  93     void put_head( BOOST_IO_STD basic_ostream<Ch, Tr> & os, const group1<T>& x ) {
  94         os << group_head(x.a1_); // send the first N-1 items, not the last
  95     }
  96
  97     template< class Ch, class Tr, class T> inline
  98     void put_last( BOOST_IO_STD basic_ostream<Ch, Tr> & os, const T& x ) {
  99         os << x ;
 100     }
 101
 102     template< class Ch, class Tr, class T> inline
 103     void put_last( BOOST_IO_STD basic_ostream<Ch, Tr> & os, const group1<T>& x ) {
 104         os << group_last(x.a1_); // this selects the last element
 105     }
 106
 107 #ifndef BOOST_NO_OVERLOAD_FOR_NON_CONST
 108     template< class Ch, class Tr, class T> inline
 109     void put_head( BOOST_IO_STD basic_ostream<Ch, Tr> &, T& ) {
 110     }
 111
 112     template< class Ch, class Tr, class T> inline
 113     void put_last( BOOST_IO_STD basic_ostream<Ch, Tr> & os, T& x) {
 114         os << x ;
 115     }
 116 #endif
 117 #endif  // -__DECCXX workaround
 118
 119     template< class Ch, class Tr, class T>
 120     void call_put_head(BOOST_IO_STD basic_ostream<Ch, Tr> & os, const void* x) {
 121         put_head(os, *(typename ::boost::remove_reference<T>::type*)x);
 122     }
 123
 124     template< class Ch, class Tr, class T>
 125     void call_put_last(BOOST_IO_STD basic_ostream<Ch, Tr> & os, const void* x) {
 126         put_last(os, *(T*)x);
 127     }
 128
 129     template< class Ch, class Tr>
 130     struct put_holder {
 131         template<class T>
 132         put_holder(T& t)
 133           : arg(&t),
 134             put_head(&call_put_head<Ch, Tr, T>),
 135             put_last(&call_put_last<Ch, Tr, T>)
 136         {}
 137         const void* arg;
 138         void (*put_head)(BOOST_IO_STD basic_ostream<Ch, Tr> & os, const void* x);
 139         void (*put_last)(BOOST_IO_STD basic_ostream<Ch, Tr> & os, const void* x);
 140     };
 141
 142     template< class Ch, class Tr> inline
 143     void put_head( BOOST_IO_STD basic_ostream<Ch, Tr> & os, const put_holder<Ch, Tr>& t) {
 144         t.put_head(os, t.arg);
 145     }
 146
 147     template< class Ch, class Tr> inline
 148     void put_last( BOOST_IO_STD basic_ostream<Ch, Tr> & os, const put_holder<Ch, Tr>& t) {
 149         t.put_last(os, t.arg);
 150     }
 151
 152
 153     template< class Ch, class Tr, class Alloc, class T>
 154     void put( T x,
 155               const format_item<Ch, Tr, Alloc>& specs,
 156               typename basic_format<Ch, Tr, Alloc>::string_type& res,
 157               typename basic_format<Ch, Tr, Alloc>::internal_streambuf_t & buf,
 158               io::detail::locale_t *loc_p = NULL)
 159     {
 160 #ifdef BOOST_MSVC
 161        // If std::min<unsigned> or std::max<unsigned> are already instantiated
 162        // at this point then we get a blizzard of warning messages when we call
 163        // those templates with std::size_t as arguments.  Weird and very annoyning...
 164 #pragma warning(push)
 165 #pragma warning(disable:4267)
 166 #endif
 167         // does the actual conversion of x, with given params, into a string
 168         // using the supplied stringbuf.
 169
 170         typedef typename basic_format<Ch, Tr, Alloc>::string_type   string_type;
 171         typedef typename basic_format<Ch, Tr, Alloc>::format_item_t format_item_t;
 172         typedef typename string_type::size_type size_type;
 173
 174         basic_oaltstringstream<Ch, Tr, Alloc>  oss( &buf);
 175
 176         if(loc_p != NULL)
 177             oss.imbue(*loc_p);
 178
 179         specs.fmtstate_.apply_on(oss, loc_p);
 180
 181         // the stream format state can be modified by manipulators in the argument :
 182         put_head( oss, x );
 183         // in case x is a group, apply the manip part of it,
 184         // in order to find width
 185
 186         const std::ios_base::fmtflags fl=oss.flags();
 187         const bool internal = (fl & std::ios_base::internal) != 0;
 188         const std::streamsize w = oss.width();
 189         const bool two_stepped_padding= internal && (w!=0);
 190
 191         res.resize(0);
 192         if(! two_stepped_padding) {
 193             if(w>0) // handle padding via mk_str, not natively in stream
 194                 oss.width(0);
 195             put_last( oss, x);
 196             const Ch * res_beg = buf.pbase();
 197             Ch prefix_space = 0;
 198             if(specs.pad_scheme_ & format_item_t::spacepad)
 199                 if(buf.pcount()== 0 ||
 200                    (res_beg[0] !=oss.widen('+') && res_beg[0] !=oss.widen('-')  ))
 201                     prefix_space = oss.widen(' ');
 202             size_type res_size = (std::min)(
 203                 static_cast<size_type>(specs.truncate_ - !!prefix_space),
 204                 buf.pcount() );
 205             mk_str(res, res_beg, res_size, w, oss.fill(), fl,
 206                    prefix_space, (specs.pad_scheme_ & format_item_t::centered) !=0 );
 207         }
 208         else  { // 2-stepped padding
 209             // internal can be implied by zeropad, or user-set.
 210             // left, right, and centered alignment overrule internal,
 211             // but spacepad or truncate might be mixed with internal (using manipulator)
 212             put_last( oss, x); // may pad
 213             const Ch * res_beg = buf.pbase();
 214             size_type res_size = buf.pcount();
 215             bool prefix_space=false;
 216             if(specs.pad_scheme_ & format_item_t::spacepad)
 217                 if(buf.pcount()== 0 ||
 218                    (res_beg[0] !=oss.widen('+') && res_beg[0] !=oss.widen('-')  ))
 219                     prefix_space = true;
 220             if(res_size == static_cast<size_type>(w) && w<=specs.truncate_ && !prefix_space) {
 221                 // okay, only one thing was printed and padded, so res is fine
 222                 res.assign(res_beg, res_size);
 223             }
 224             else { //   length w exceeded
 225                 // either it was multi-output with first output padding up all width..
 226                 // either it was one big arg and we are fine.
 227                 // Note that res_size<w is possible  (in case of bad user-defined formatting)
 228                 res.assign(res_beg, res_size);
 229                 res_beg=NULL;  // invalidate pointers.
 230
 231                 // make a new stream, to start re-formatting from scratch :
 232                 buf.clear_buffer();
 233                 basic_oaltstringstream<Ch, Tr, Alloc>  oss2( &buf);
 234                 specs.fmtstate_.apply_on(oss2, loc_p);
 235                 put_head( oss2, x );
 236
 237                 oss2.width(0);
 238                 if(prefix_space)
 239                     oss2 << ' ';
 240                 put_last(oss2, x );
 241                 if(buf.pcount()==0 && specs.pad_scheme_ & format_item_t::spacepad) {
 242                     prefix_space =true;
 243                     oss2 << ' ';
 244                 }
 245                 // we now have the minimal-length output
 246                 const Ch * tmp_beg = buf.pbase();
 247                 size_type tmp_size = (std::min)(static_cast<size_type>(specs.truncate_),
 248                                                 buf.pcount() );
 249
 250
 251                 if(static_cast<size_type>(w) <= tmp_size) {
 252                     // minimal length is already >= w, so no padding (cool!)
 253                         res.assign(tmp_beg, tmp_size);
 254                 }
 255                 else { // hum..  we need to pad (multi_output, or spacepad present)
 256                     //find where we should pad
 257                     size_type sz = (std::min)(res_size + (prefix_space ? 1 : 0), tmp_size);
 258                     size_type i = prefix_space;
 259                     for(; i<sz && tmp_beg[i] == res[i - (prefix_space ? 1 : 0)]; ++i) {}
 260                     if(i>=tmp_size) i=prefix_space;
 261                     res.assign(tmp_beg, i);
 262                                         std::streamsize d = w - static_cast<std::streamsize>(tmp_size);
 263                                         BOOST_ASSERT(d>0);
 264                     res.append(static_cast<size_type>( d ), oss2.fill());
 265                     res.append(tmp_beg+i, tmp_size-i);
 266                     BOOST_ASSERT(i+(tmp_size-i)+(std::max)(d,(std::streamsize)0)
 267                                  == static_cast<size_type>(w));
 268                     BOOST_ASSERT(res.size() == static_cast<size_type>(w));
 269                 }
 270             }
 271         }
 272         buf.clear_buffer();
 273 #ifdef BOOST_MSVC
 274 #pragma warning(pop)
 275 #endif
 276     } // end- put(..)
 277
 278
 279     template< class Ch, class Tr, class Alloc, class T>
 280     void distribute (basic_format<Ch,Tr, Alloc>& self, T x) {
 281         // call put(x, ..) on every occurrence of the current argument :
 282         if(self.cur_arg_ >= self.num_args_)  {
 283             if( self.exceptions() & too_many_args_bit )
 284                 boost::throw_exception(too_many_args(self.cur_arg_, self.num_args_));
 285             else return;
 286         }
 287         for(unsigned long i=0; i < self.items_.size(); ++i) {
 288             if(self.items_[i].argN_ == self.cur_arg_) {
 289                 put<Ch, Tr, Alloc, T> (x, self.items_[i], self.items_[i].res_,
 290                                 self.buf_, boost::get_pointer(self.loc_) );
 291             }
 292         }
 293     }
 294
 295     template<class Ch, class Tr, class Alloc, class T>
 296     basic_format<Ch, Tr, Alloc>&
 297     feed_impl (basic_format<Ch,Tr, Alloc>& self, T x) {
 298         if(self.dumped_) self.clear();
 299         distribute<Ch, Tr, Alloc, T> (self, x);
 300         ++self.cur_arg_;
 301         if(self.bound_.size() != 0) {
 302                 while( self.cur_arg_ < self.num_args_ && self.bound_[self.cur_arg_] )
 303                     ++self.cur_arg_;
 304         }
 305         return self;
 306     }
 307
 308     template<class Ch, class Tr, class Alloc, class T> inline
 309     basic_format<Ch, Tr, Alloc>&
 310     feed (basic_format<Ch,Tr, Alloc>& self, T x) {
 311         return feed_impl<Ch, Tr, Alloc, const put_holder<Ch, Tr>&>(self, put_holder<Ch, Tr>(x));
 312     }
 313
 314 } // namespace detail
 315 } // namespace io
 316 } // namespace boost
 317
 318
 319 #endif //  BOOST_FORMAT_FEED_ARGS_HPP