include/c++/stl/stl/_string_sum_methods.h

   1 /*
   2  * Copyright (c) 2003
   3  * Francois Dumont
   4  *
   5  * This material is provided "as is", with absolutely no warranty expressed
   6  * or implied. Any use is at your own risk.
   7  *
   8  * Permission to use or copy this software for any purpose is hereby granted
   9  * without fee, provided the above notices are retained on all copies.
  10  * Permission to modify the code and to distribute modified code is granted,
  11  * provided the above notices are retained, and a notice that the code was
  12  * modified is included with the above copyright notice.
  13  *
  14  */
  15
  16 /*
  17  * All the necessary methods used for template expressions with basic_string
  18  * This file do not have to be macro guarded as it is only used in the _string.h
  19  * file and it is a part of the basic_string definition.
  20  */
  21
  22 public:
  23   template <class _Left, class _Right, class _StorageDir>
  24   basic_string(_STLP_PRIV __bstr_sum<_CharT, _Traits, _Alloc, _Left, _Right, _StorageDir> const& __s)
  25     : _STLP_STRING_SUM_BASE(_Reserve_t(), __s.size(), __s.get_allocator())
  26   { _M_append_sum(__s); }
  27
  28   template <class _Left, class _Right, class _StorageDir>
  29   basic_string(_STLP_PRIV __bstr_sum<_CharT, _Traits, _Alloc, _Left, _Right, _StorageDir> const& __s,
  30                size_type __pos, size_type __n = npos,
  31                const allocator_type& __a = allocator_type())
  32     : _STLP_STRING_SUM_BASE(_Reserve_t(), (__pos <= __s.size()) ? ((min) (__n, __s.size() - __pos)) : 0, __a) {
  33     size_type __size = __s.size();
  34     if (__pos > __size)
  35       this->_M_throw_out_of_range();
  36     else
  37       _M_append_sum_pos(__s, __pos, (min) (__n, __size - __pos));
  38   }
  39
  40 private:
  41   _CharT* _M_append_fast(_STLP_PRIV __char_wrapper<_CharT> __c, _CharT *__buf) {
  42     _STLP_STD::_Copy_Construct(__buf, __c.getValue());
  43     return __buf + 1;
  44   }
  45   _CharT* _M_append_fast(_CharT const* __s, size_type __s_size, _CharT *__buf)
  46   { return uninitialized_copy(__s, __s + __s_size, __buf); }
  47   _CharT* _M_append_fast(_STLP_PRIV __cstr_wrapper<_CharT> const& __s, _CharT *__buf)
  48   { return _M_append_fast(__s.c_str(), __s.size(), __buf); }
  49   _CharT* _M_append_fast(_STLP_PRIV __bstr_wrapper<_CharT, _Traits, _Alloc> __s, _CharT *__buf)
  50   { return _M_append_fast(__s.b_str(), __buf); }
  51   _CharT* _M_append_fast(_Self const& __s, _CharT *__buf)
  52   { return _M_append_fast(__s.data(), __s.size(), __buf); }
  53   _CharT* _M_append_fast(_STLP_PRIV __sum_storage_elem<_CharT, _Traits, _Alloc> const&, _CharT *__buf)
  54   { return __buf; }
  55   template <class _Left, class _Right, class _StorageDir>
  56   _CharT* _M_append_fast(_STLP_PRIV __bstr_sum<_CharT, _Traits, _Alloc, _Left, _Right, _StorageDir> const& __s, _CharT *__buf)
  57   { return _M_append_fast(__s.getRhs(), _M_append_fast(__s.getLhs(), __buf)); }
  58
  59   _CharT* _M_append_fast_pos(_STLP_PRIV __char_wrapper<_CharT> __c, _CharT *__buf, size_type /*__pos*/, size_type __n) {
  60     if (__n == 0)
  61       return __buf;
  62     _STLP_STD::_Copy_Construct(__buf, __c.getValue());
  63     return __buf + 1;
  64   }
  65   _CharT* _M_append_fast_pos(_CharT const* __s, size_type __s_size, _CharT *__buf,
  66                              size_type __pos, size_type __n)
  67   { return uninitialized_copy(__s + __pos, __s + __pos + (min)(__n, __s_size - __pos), __buf); }
  68   _CharT* _M_append_fast_pos(_STLP_PRIV __cstr_wrapper<_CharT> const& __s, _CharT *__buf,
  69                              size_type __pos, size_type __n)
  70   { return _M_append_fast_pos(__s.c_str(), __s.size(), __buf, __pos, __n); }
  71   _CharT* _M_append_fast_pos(_STLP_PRIV __bstr_wrapper<_CharT, _Traits, _Alloc> __s, _CharT *__buf,
  72                              size_type __pos, size_type __n)
  73   { return _M_append_fast_pos(__s.b_str(), __buf, __pos, __n); }
  74   _CharT* _M_append_fast_pos(_Self const& __s, _CharT *__buf,
  75                              size_type __pos, size_type __n)
  76   { return _M_append_fast_pos(__s.data(), __s.size(), __buf, __pos, __n); }
  77   _CharT* _M_append_fast_pos(_STLP_PRIV __sum_storage_elem<_CharT, _Traits, _Alloc> const&, _CharT *__buf,
  78                              size_type, size_type)
  79   { return __buf; }
  80
  81   template <class _Left, class _Right, class _StorageDir>
  82   _CharT* _M_append_fast_pos(_STLP_PRIV __bstr_sum<_CharT, _Traits, _Alloc, _Left, _Right, _StorageDir> const& __s,
  83                              _CharT *__buf, size_type __pos, size_type __n) {
  84     if (__n == 0) {
  85       return __buf;
  86     }
  87     size_type __lhs_size = __s.getLhs().size();
  88     if (__pos < __lhs_size) {
  89       if (__n < (__lhs_size - __pos)) {
  90         return _M_append_fast_pos(__s.getLhs(), __buf, __pos, __n);
  91       } else {
  92         return _M_append_fast_pos(__s.getRhs(), _M_append_fast_pos(__s.getLhs(), __buf, __pos, __n),
  93                                   0, __n - (__lhs_size - __pos));
  94       }
  95     } else {
  96       return _M_append_fast_pos(__s.getRhs(), __buf, __pos - __lhs_size, __n);
  97     }
  98   }
  99
 100   /* Note: We always force use of dynamic buffer if the short string optim option is activated
 101    * to avoid complicated code if the basic_string was instanciated with a non POD type.
 102    * In such a case we should use assignment for objects in the static array something that we
 103    * do not do.
 104    */
 105   size_type _M_get_additional_size(size_type __new_size, const __false_type& /*_Char_Is_POD*/) const {
 106 #if defined (_STLP_USE_SHORT_STRING_OPTIM)
 107     //To avoid problem with the string assumptions, never allocate a dynamic buffer smaller or equal
 108     //than the static one:
 109     if (__new_size < _Base::_DEFAULT_SIZE + 1)
 110       return (_Base::_DEFAULT_SIZE + 1) - __new_size;
 111 #endif /* _STLP_USE_SHORT_STRING_OPTIM */
 112     return 0;
 113   }
 114   size_type _M_get_additional_size(size_type, const __true_type& /*_Char_Is_POD*/) const
 115   { return 0; }
 116
 117   template <class _Left, class _Right, class _StorageDir>
 118   _Self& _M_append_sum (_STLP_PRIV __bstr_sum<_CharT, _Traits, _Alloc, _Left, _Right, _StorageDir> const& __s) {
 119     size_type __s_size = __s.size();
 120     if (__s_size == 0)
 121       return *this;
 122     const size_type __old_size = this->size();
 123     if (__s_size > this->max_size() || __old_size > (this->max_size() - __s_size))
 124       this->_M_throw_length_error();
 125     size_type __offset_size = _M_get_additional_size(__old_size + __s_size, _Char_Is_POD());
 126     if (__old_size + __s_size + __offset_size > this->capacity()) {
 127       const size_type __len = __old_size + __offset_size + (max)(__old_size, __s_size) + 1;
 128       pointer __new_start = this->_M_end_of_storage.allocate(__len);
 129       pointer __new_finish = __new_start;
 130       _STLP_TRY {
 131         __new_finish = uninitialized_copy(this->_M_Start(), this->_M_Finish(), __new_start);
 132         __new_finish = this->_M_append_fast(__s, __new_finish);
 133         this->_M_construct_null(__new_finish);
 134       }
 135       _STLP_UNWIND((_STLP_STD::_Destroy_Range(__new_start,__new_finish),
 136                    this->_M_end_of_storage.deallocate(__new_start,__len)))
 137       this->_M_destroy_range();
 138       this->_M_deallocate_block();
 139       this->_M_reset(__new_start, __new_finish, __new_start + __len);
 140     }
 141     else {
 142       _M_append_sum_no_overflow(__s, 0, __s_size);
 143     }
 144     return *this;
 145   }
 146
 147   template <class _Left, class _Right, class _StorageDir>
 148   _Self& _M_append_sum_pos(_STLP_PRIV __bstr_sum<_CharT, _Traits, _Alloc, _Left, _Right, _StorageDir> const& __s,
 149                            size_type __pos, size_type __n) {
 150     size_type __s_size = (min)(__s.size() - __pos, __n);
 151     if (__s_size == 0)
 152       return *this;
 153     const size_type __old_size = this->size();
 154     if (__s_size > this->max_size() || __old_size > (this->max_size() - __s_size))
 155       this->_M_throw_length_error();
 156     size_type __offset_size = _M_get_additional_size(__old_size + __s_size, _Char_Is_POD());
 157     if (__old_size + __s_size + __offset_size > this->capacity()) {
 158       const size_type __len = __old_size + __offset_size + (max)(__old_size, __s_size) + 1;
 159       pointer __new_start = this->_M_end_of_storage.allocate(__len);
 160       pointer __new_finish = __new_start;
 161       _STLP_TRY {
 162         __new_finish = uninitialized_copy(this->_M_Start(), this->_M_Finish(), __new_start);
 163         __new_finish = _M_append_fast_pos(__s, __new_finish, __pos, __s_size);
 164         this->_M_construct_null(__new_finish);
 165       }
 166       _STLP_UNWIND((_STLP_STD::_Destroy_Range(__new_start,__new_finish),
 167                    this->_M_end_of_storage.deallocate(__new_start,__len)))
 168       this->_M_destroy_range();
 169       this->_M_deallocate_block();
 170       this->_M_reset(__new_start, __new_finish, __new_start + __len);
 171     }
 172     else {
 173       _M_append_sum_no_overflow(__s, __pos, __s_size);
 174     }
 175     return *this;
 176   }
 177
 178   template <class _Left, class _Right, class _StorageDir>
 179   void _M_append_sum_no_overflow(_STLP_PRIV __bstr_sum<_CharT, _Traits, _Alloc, _Left, _Right, _StorageDir> const& __s,
 180                                  size_type __pos, size_type __n) {
 181     pointer __finish = this->_M_Finish();
 182     _M_append_fast_pos(__s, __finish + 1, __pos + 1, __n - 1);
 183     _STLP_TRY {
 184       this->_M_construct_null(__finish + __n);
 185     }
 186     _STLP_UNWIND(this->_M_destroy_ptr_range(__finish + 1, __finish + __n))
 187     /* The call to the traits::assign method is only important for non POD types because the instance
 188      * pointed to by _M_finish has been constructed (default constructor) and should not be constructed
 189      * (copy constructor) once again. For POD it is irrelevant, uninitialized_copy could be fine,
 190      * but we are not going to make two implementations just for that.
 191      */
 192     _Traits::assign(*this->_M_finish, __s[__pos]);
 193     this->_M_finish += __n;
 194   }