source: trunk/src/gcc/libstdc++-v3/include/ext/ropeimpl.h@ 2

Last change on this file since 2 was 2, checked in by bird, 23 years ago

Initial revision

  • Property cvs2svn:cvs-rev set to 1.1
  • Property svn:eol-style set to native
  • Property svn:executable set to *
File size: 45.5 KB
Line 
1// SGI's rope class implementation -*- C++ -*-
2
3// Copyright (C) 2001, 2002 Free Software Foundation, Inc.
4//
5// This file is part of the GNU ISO C++ Library. This library is free
6// software; you can redistribute it and/or modify it under the
7// terms of the GNU General Public License as published by the
8// Free Software Foundation; either version 2, or (at your option)
9// any later version.
10
11// This library is distributed in the hope that it will be useful,
12// but WITHOUT ANY WARRANTY; without even the implied warranty of
13// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14// GNU General Public License for more details.
15
16// You should have received a copy of the GNU General Public License along
17// with this library; see the file COPYING. If not, write to the Free
18// Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307,
19// USA.
20
21// As a special exception, you may use this file as part of a free software
22// library without restriction. Specifically, if other files instantiate
23// templates or use macros or inline functions from this file, or you compile
24// this file and link it with other files to produce an executable, this
25// file does not by itself cause the resulting executable to be covered by
26// the GNU General Public License. This exception does not however
27// invalidate any other reasons why the executable file might be covered by
28// the GNU General Public License.
29
30/*
31 * Copyright (c) 1997
32 * Silicon Graphics Computer Systems, Inc.
33 *
34 * Permission to use, copy, modify, distribute and sell this software
35 * and its documentation for any purpose is hereby granted without fee,
36 * provided that the above copyright notice appear in all copies and
37 * that both that copyright notice and this permission notice appear
38 * in supporting documentation. Silicon Graphics makes no
39 * representations about the suitability of this software for any
40 * purpose. It is provided "as is" without express or implied warranty.
41 */
42
43/** @file ropeimpl.h
44 * This is an internal header file, included by other library headers.
45 * You should not attempt to use it directly.
46 */
47
48#include <cstdio>
49#include <iostream>
50#include <bits/functexcept.h>
51
52#include <ext/algorithm> // For copy_n and lexicographical_compare_3way
53#include <ext/memory> // For uninitialized_copy_n
54#include <ext/numeric> // For power
55
56namespace __gnu_cxx
57{
58using std::size_t;
59using std::printf;
60using std::basic_ostream;
61using std::__throw_length_error;
62using std::__alloc;
63using std::_Destroy;
64using std::uninitialized_fill_n;
65
66// Set buf_start, buf_end, and buf_ptr appropriately, filling tmp_buf
67// if necessary. Assumes _M_path_end[leaf_index] and leaf_pos are correct.
68// Results in a valid buf_ptr if the iterator can be legitimately
69// dereferenced.
70template <class _CharT, class _Alloc>
71void _Rope_iterator_base<_CharT,_Alloc>::_S_setbuf(
72 _Rope_iterator_base<_CharT,_Alloc>& __x)
73{
74 const _RopeRep* __leaf = __x._M_path_end[__x._M_leaf_index];
75 size_t __leaf_pos = __x._M_leaf_pos;
76 size_t __pos = __x._M_current_pos;
77
78 switch(__leaf->_M_tag) {
79 case _RopeRep::_S_leaf:
80 __x._M_buf_start =
81 ((_Rope_RopeLeaf<_CharT,_Alloc>*)__leaf)->_M_data;
82 __x._M_buf_ptr = __x._M_buf_start + (__pos - __leaf_pos);
83 __x._M_buf_end = __x._M_buf_start + __leaf->_M_size;
84 break;
85 case _RopeRep::_S_function:
86 case _RopeRep::_S_substringfn:
87 {
88 size_t __len = _S_iterator_buf_len;
89 size_t __buf_start_pos = __leaf_pos;
90 size_t __leaf_end = __leaf_pos + __leaf->_M_size;
91 char_producer<_CharT>* __fn =
92 ((_Rope_RopeFunction<_CharT,_Alloc>*)__leaf)->_M_fn;
93
94 if (__buf_start_pos + __len <= __pos) {
95 __buf_start_pos = __pos - __len/4;
96 if (__buf_start_pos + __len > __leaf_end) {
97 __buf_start_pos = __leaf_end - __len;
98 }
99 }
100 if (__buf_start_pos + __len > __leaf_end) {
101 __len = __leaf_end - __buf_start_pos;
102 }
103 (*__fn)(__buf_start_pos - __leaf_pos, __len, __x._M_tmp_buf);
104 __x._M_buf_ptr = __x._M_tmp_buf + (__pos - __buf_start_pos);
105 __x._M_buf_start = __x._M_tmp_buf;
106 __x._M_buf_end = __x._M_tmp_buf + __len;
107 }
108 break;
109 default:
110 break;
111 }
112}
113
114// Set path and buffer inside a rope iterator. We assume that
115// pos and root are already set.
116template <class _CharT, class _Alloc>
117void _Rope_iterator_base<_CharT,_Alloc>::_S_setcache
118(_Rope_iterator_base<_CharT,_Alloc>& __x)
119{
120 const _RopeRep* __path[_RopeRep::_S_max_rope_depth+1];
121 const _RopeRep* __curr_rope;
122 int __curr_depth = -1; /* index into path */
123 size_t __curr_start_pos = 0;
124 size_t __pos = __x._M_current_pos;
125 unsigned char __dirns = 0; // Bit vector marking right turns in the path
126
127 if (__pos >= __x._M_root->_M_size) {
128 __x._M_buf_ptr = 0;
129 return;
130 }
131 __curr_rope = __x._M_root;
132 if (0 != __curr_rope->_M_c_string) {
133 /* Treat the root as a leaf. */
134 __x._M_buf_start = __curr_rope->_M_c_string;
135 __x._M_buf_end = __curr_rope->_M_c_string + __curr_rope->_M_size;
136 __x._M_buf_ptr = __curr_rope->_M_c_string + __pos;
137 __x._M_path_end[0] = __curr_rope;
138 __x._M_leaf_index = 0;
139 __x._M_leaf_pos = 0;
140 return;
141 }
142 for(;;) {
143 ++__curr_depth;
144 __path[__curr_depth] = __curr_rope;
145 switch(__curr_rope->_M_tag) {
146 case _RopeRep::_S_leaf:
147 case _RopeRep::_S_function:
148 case _RopeRep::_S_substringfn:
149 __x._M_leaf_pos = __curr_start_pos;
150 goto done;
151 case _RopeRep::_S_concat:
152 {
153 _Rope_RopeConcatenation<_CharT,_Alloc>* __c =
154 (_Rope_RopeConcatenation<_CharT,_Alloc>*)__curr_rope;
155 _RopeRep* __left = __c->_M_left;
156 size_t __left_len = __left->_M_size;
157
158 __dirns <<= 1;
159 if (__pos >= __curr_start_pos + __left_len) {
160 __dirns |= 1;
161 __curr_rope = __c->_M_right;
162 __curr_start_pos += __left_len;
163 } else {
164 __curr_rope = __left;
165 }
166 }
167 break;
168 }
169 }
170 done:
171 // Copy last section of path into _M_path_end.
172 {
173 int __i = -1;
174 int __j = __curr_depth + 1 - _S_path_cache_len;
175
176 if (__j < 0) __j = 0;
177 while (__j <= __curr_depth) {
178 __x._M_path_end[++__i] = __path[__j++];
179 }
180 __x._M_leaf_index = __i;
181 }
182 __x._M_path_directions = __dirns;
183 _S_setbuf(__x);
184}
185
186// Specialized version of the above. Assumes that
187// the path cache is valid for the previous position.
188template <class _CharT, class _Alloc>
189void _Rope_iterator_base<_CharT,_Alloc>::_S_setcache_for_incr
190(_Rope_iterator_base<_CharT,_Alloc>& __x)
191{
192 int __current_index = __x._M_leaf_index;
193 const _RopeRep* __current_node = __x._M_path_end[__current_index];
194 size_t __len = __current_node->_M_size;
195 size_t __node_start_pos = __x._M_leaf_pos;
196 unsigned char __dirns = __x._M_path_directions;
197 _Rope_RopeConcatenation<_CharT,_Alloc>* __c;
198
199 if (__x._M_current_pos - __node_start_pos < __len) {
200 /* More stuff in this leaf, we just didn't cache it. */
201 _S_setbuf(__x);
202 return;
203 }
204 // node_start_pos is starting position of last_node.
205 while (--__current_index >= 0) {
206 if (!(__dirns & 1) /* Path turned left */)
207 break;
208 __current_node = __x._M_path_end[__current_index];
209 __c = (_Rope_RopeConcatenation<_CharT,_Alloc>*)__current_node;
210 // Otherwise we were in the right child. Thus we should pop
211 // the concatenation node.
212 __node_start_pos -= __c->_M_left->_M_size;
213 __dirns >>= 1;
214 }
215 if (__current_index < 0) {
216 // We underflowed the cache. Punt.
217 _S_setcache(__x);
218 return;
219 }
220 __current_node = __x._M_path_end[__current_index];
221 __c = (_Rope_RopeConcatenation<_CharT,_Alloc>*)__current_node;
222 // current_node is a concatenation node. We are positioned on the first
223 // character in its right child.
224 // node_start_pos is starting position of current_node.
225 __node_start_pos += __c->_M_left->_M_size;
226 __current_node = __c->_M_right;
227 __x._M_path_end[++__current_index] = __current_node;
228 __dirns |= 1;
229 while (_RopeRep::_S_concat == __current_node->_M_tag) {
230 ++__current_index;
231 if (_S_path_cache_len == __current_index) {
232 int __i;
233 for (__i = 0; __i < _S_path_cache_len-1; __i++) {
234 __x._M_path_end[__i] = __x._M_path_end[__i+1];
235 }
236 --__current_index;
237 }
238 __current_node =
239 ((_Rope_RopeConcatenation<_CharT,_Alloc>*)__current_node)->_M_left;
240 __x._M_path_end[__current_index] = __current_node;
241 __dirns <<= 1;
242 // node_start_pos is unchanged.
243 }
244 __x._M_leaf_index = __current_index;
245 __x._M_leaf_pos = __node_start_pos;
246 __x._M_path_directions = __dirns;
247 _S_setbuf(__x);
248}
249
250template <class _CharT, class _Alloc>
251void _Rope_iterator_base<_CharT,_Alloc>::_M_incr(size_t __n) {
252 _M_current_pos += __n;
253 if (0 != _M_buf_ptr) {
254 size_t __chars_left = _M_buf_end - _M_buf_ptr;
255 if (__chars_left > __n) {
256 _M_buf_ptr += __n;
257 } else if (__chars_left == __n) {
258 _M_buf_ptr += __n;
259 _S_setcache_for_incr(*this);
260 } else {
261 _M_buf_ptr = 0;
262 }
263 }
264}
265
266template <class _CharT, class _Alloc>
267void _Rope_iterator_base<_CharT,_Alloc>::_M_decr(size_t __n) {
268 if (0 != _M_buf_ptr) {
269 size_t __chars_left = _M_buf_ptr - _M_buf_start;
270 if (__chars_left >= __n) {
271 _M_buf_ptr -= __n;
272 } else {
273 _M_buf_ptr = 0;
274 }
275 }
276 _M_current_pos -= __n;
277}
278
279template <class _CharT, class _Alloc>
280void _Rope_iterator<_CharT,_Alloc>::_M_check() {
281 if (_M_root_rope->_M_tree_ptr != _M_root) {
282 // _Rope was modified. Get things fixed up.
283 _RopeRep::_S_unref(_M_root);
284 _M_root = _M_root_rope->_M_tree_ptr;
285 _RopeRep::_S_ref(_M_root);
286 _M_buf_ptr = 0;
287 }
288}
289
290template <class _CharT, class _Alloc>
291inline
292_Rope_const_iterator<_CharT, _Alloc>::_Rope_const_iterator(
293 const _Rope_iterator<_CharT,_Alloc>& __x)
294: _Rope_iterator_base<_CharT,_Alloc>(__x)
295{ }
296
297template <class _CharT, class _Alloc>
298inline _Rope_iterator<_CharT,_Alloc>::_Rope_iterator(
299 rope<_CharT,_Alloc>& __r, size_t __pos)
300: _Rope_iterator_base<_CharT,_Alloc>(__r._M_tree_ptr, __pos),
301 _M_root_rope(&__r)
302{
303 _RopeRep::_S_ref(_M_root);
304}
305
306template <class _CharT, class _Alloc>
307inline size_t
308rope<_CharT,_Alloc>::_S_char_ptr_len(const _CharT* __s)
309{
310 const _CharT* __p = __s;
311
312 while (!_S_is0(*__p)) { ++__p; }
313 return (__p - __s);
314}
315
316
317#ifndef __GC
318
319template <class _CharT, class _Alloc>
320inline void _Rope_RopeRep<_CharT,_Alloc>::_M_free_c_string()
321{
322 _CharT* __cstr = _M_c_string;
323 if (0 != __cstr) {
324 size_t __size = _M_size + 1;
325 _Destroy(__cstr, __cstr + __size);
326 _Data_deallocate(__cstr, __size);
327 }
328}
329
330
331template <class _CharT, class _Alloc>
332 inline void _Rope_RopeRep<_CharT,_Alloc>::_S_free_string(_CharT* __s,
333 size_t __n,
334 allocator_type __a)
335{
336 if (!_S_is_basic_char_type((_CharT*)0)) {
337 _Destroy(__s, __s + __n);
338 }
339// This has to be a static member, so this gets a bit messy
340 __a.deallocate(
341 __s, _Rope_RopeLeaf<_CharT,_Alloc>::_S_rounded_up_size(__n));
342}
343
344
345// There are several reasons for not doing this with virtual destructors
346// and a class specific delete operator:
347// - A class specific delete operator can't easily get access to
348// allocator instances if we need them.
349// - Any virtual function would need a 4 or byte vtable pointer;
350// this only requires a one byte tag per object.
351template <class _CharT, class _Alloc>
352void _Rope_RopeRep<_CharT,_Alloc>::_M_free_tree()
353{
354 switch(_M_tag) {
355 case _S_leaf:
356 {
357 _Rope_RopeLeaf<_CharT,_Alloc>* __l
358 = (_Rope_RopeLeaf<_CharT,_Alloc>*)this;
359 __l->_Rope_RopeLeaf<_CharT,_Alloc>::~_Rope_RopeLeaf();
360 _L_deallocate(__l, 1);
361 break;
362 }
363 case _S_concat:
364 {
365 _Rope_RopeConcatenation<_CharT,_Alloc>* __c
366 = (_Rope_RopeConcatenation<_CharT,_Alloc>*)this;
367 __c->_Rope_RopeConcatenation<_CharT,_Alloc>::
368 ~_Rope_RopeConcatenation();
369 _C_deallocate(__c, 1);
370 break;
371 }
372 case _S_function:
373 {
374 _Rope_RopeFunction<_CharT,_Alloc>* __f
375 = (_Rope_RopeFunction<_CharT,_Alloc>*)this;
376 __f->_Rope_RopeFunction<_CharT,_Alloc>::~_Rope_RopeFunction();
377 _F_deallocate(__f, 1);
378 break;
379 }
380 case _S_substringfn:
381 {
382 _Rope_RopeSubstring<_CharT,_Alloc>* __ss =
383 (_Rope_RopeSubstring<_CharT,_Alloc>*)this;
384 __ss->_Rope_RopeSubstring<_CharT,_Alloc>::
385 ~_Rope_RopeSubstring();
386 _S_deallocate(__ss, 1);
387 break;
388 }
389 }
390}
391#else
392
393template <class _CharT, class _Alloc>
394 inline void _Rope_RopeRep<_CharT,_Alloc>::_S_free_string
395 (const _CharT*, size_t, allocator_type)
396{}
397
398#endif
399
400
401// Concatenate a C string onto a leaf rope by copying the rope data.
402// Used for short ropes.
403template <class _CharT, class _Alloc>
404typename rope<_CharT,_Alloc>::_RopeLeaf*
405rope<_CharT,_Alloc>::_S_leaf_concat_char_iter
406 (_RopeLeaf* __r, const _CharT* __iter, size_t __len)
407{
408 size_t __old_len = __r->_M_size;
409 _CharT* __new_data = (_CharT*)
410 _Data_allocate(_S_rounded_up_size(__old_len + __len));
411 _RopeLeaf* __result;
412
413 uninitialized_copy_n(__r->_M_data, __old_len, __new_data);
414 uninitialized_copy_n(__iter, __len, __new_data + __old_len);
415 _S_cond_store_eos(__new_data[__old_len + __len]);
416 try {
417 __result = _S_new_RopeLeaf(__new_data, __old_len + __len,
418 __r->get_allocator());
419 }
420 catch(...)
421 {
422 _RopeRep::__STL_FREE_STRING(__new_data, __old_len + __len,
423 __r->get_allocator());
424 __throw_exception_again;
425 }
426 return __result;
427}
428
429#ifndef __GC
430// As above, but it's OK to clobber original if refcount is 1
431template <class _CharT, class _Alloc>
432typename rope<_CharT,_Alloc>::_RopeLeaf*
433rope<_CharT,_Alloc>::_S_destr_leaf_concat_char_iter
434 (_RopeLeaf* __r, const _CharT* __iter, size_t __len)
435{
436 if (__r->_M_ref_count > 1)
437 return _S_leaf_concat_char_iter(__r, __iter, __len);
438 size_t __old_len = __r->_M_size;
439 if (_S_allocated_capacity(__old_len) >= __old_len + __len) {
440 // The space has been partially initialized for the standard
441 // character types. But that doesn't matter for those types.
442 uninitialized_copy_n(__iter, __len, __r->_M_data + __old_len);
443 if (_S_is_basic_char_type((_CharT*)0)) {
444 _S_cond_store_eos(__r->_M_data[__old_len + __len]);
445 } else if (__r->_M_c_string != __r->_M_data && 0 != __r->_M_c_string) {
446 __r->_M_free_c_string();
447 __r->_M_c_string = 0;
448 }
449 __r->_M_size = __old_len + __len;
450 __r->_M_ref_count = 2;
451 return __r;
452 } else {
453 _RopeLeaf* __result = _S_leaf_concat_char_iter(__r, __iter, __len);
454 return __result;
455 }
456}
457#endif
458
459// Assumes left and right are not 0.
460// Does not increment (nor decrement on exception) child reference counts.
461// Result has ref count 1.
462template <class _CharT, class _Alloc>
463typename rope<_CharT,_Alloc>::_RopeRep*
464rope<_CharT,_Alloc>::_S_tree_concat (_RopeRep* __left, _RopeRep* __right)
465{
466 _RopeConcatenation* __result = _S_new_RopeConcatenation(__left, __right,
467 __left->get_allocator());
468 size_t __depth = __result->_M_depth;
469
470 if (__depth > 20 && (__result->_M_size < 1000 ||
471 __depth > _RopeRep::_S_max_rope_depth))
472 {
473 _RopeRep* __balanced;
474
475 try
476 {
477 __balanced = _S_balance(__result);
478 __result->_M_unref_nonnil();
479 }
480 catch(...)
481 {
482 _C_deallocate(__result,1);
483 __throw_exception_again;
484 }
485 // In case of exception, we need to deallocate
486 // otherwise dangling result node. But caller
487 // still owns its children. Thus unref is
488 // inappropriate.
489 return __balanced;
490 }
491 else
492 return __result;
493}
494
495template <class _CharT, class _Alloc>
496typename
497rope<_CharT,_Alloc>::_RopeRep* rope<_CharT,_Alloc>::_S_concat_char_iter
498 (_RopeRep* __r, const _CharT*__s, size_t __slen)
499{
500 _RopeRep* __result;
501 if (0 == __slen) {
502 _S_ref(__r);
503 return __r;
504 }
505 if (0 == __r)
506 return __STL_ROPE_FROM_UNOWNED_CHAR_PTR(__s, __slen,
507 __r->get_allocator());
508 if (_RopeRep::_S_leaf == __r->_M_tag &&
509 __r->_M_size + __slen <= _S_copy_max) {
510 __result = _S_leaf_concat_char_iter((_RopeLeaf*)__r, __s, __slen);
511 return __result;
512 }
513 if (_RopeRep::_S_concat == __r->_M_tag
514 && _RopeRep::_S_leaf == ((_RopeConcatenation*)__r)->_M_right->_M_tag) {
515 _RopeLeaf* __right =
516 (_RopeLeaf* )(((_RopeConcatenation* )__r)->_M_right);
517 if (__right->_M_size + __slen <= _S_copy_max) {
518 _RopeRep* __left = ((_RopeConcatenation*)__r)->_M_left;
519 _RopeRep* __nright =
520 _S_leaf_concat_char_iter((_RopeLeaf*)__right, __s, __slen);
521 __left->_M_ref_nonnil();
522 try {
523 __result = _S_tree_concat(__left, __nright);
524 }
525 catch(...)
526 {
527 _S_unref(__left);
528 _S_unref(__nright);
529 __throw_exception_again;
530 }
531 return __result;
532 }
533 }
534 _RopeRep* __nright =
535 __STL_ROPE_FROM_UNOWNED_CHAR_PTR(__s, __slen, __r->get_allocator());
536 try {
537 __r->_M_ref_nonnil();
538 __result = _S_tree_concat(__r, __nright);
539 }
540 catch(...)
541 {
542 _S_unref(__r);
543 _S_unref(__nright);
544 __throw_exception_again;
545 }
546 return __result;
547}
548
549#ifndef __GC
550template <class _CharT, class _Alloc>
551typename rope<_CharT,_Alloc>::_RopeRep*
552rope<_CharT,_Alloc>::_S_destr_concat_char_iter(
553 _RopeRep* __r, const _CharT* __s, size_t __slen)
554{
555 _RopeRep* __result;
556 if (0 == __r)
557 return __STL_ROPE_FROM_UNOWNED_CHAR_PTR(__s, __slen,
558 __r->get_allocator());
559 size_t __count = __r->_M_ref_count;
560 size_t __orig_size = __r->_M_size;
561 if (__count > 1) return _S_concat_char_iter(__r, __s, __slen);
562 if (0 == __slen) {
563 __r->_M_ref_count = 2; // One more than before
564 return __r;
565 }
566 if (__orig_size + __slen <= _S_copy_max &&
567 _RopeRep::_S_leaf == __r->_M_tag) {
568 __result = _S_destr_leaf_concat_char_iter((_RopeLeaf*)__r, __s, __slen);
569 return __result;
570 }
571 if (_RopeRep::_S_concat == __r->_M_tag) {
572 _RopeLeaf* __right = (_RopeLeaf*)(((_RopeConcatenation*)__r)->_M_right);
573 if (_RopeRep::_S_leaf == __right->_M_tag
574 && __right->_M_size + __slen <= _S_copy_max) {
575 _RopeRep* __new_right =
576 _S_destr_leaf_concat_char_iter(__right, __s, __slen);
577 if (__right == __new_right)
578 __new_right->_M_ref_count = 1;
579 else
580 __right->_M_unref_nonnil();
581 __r->_M_ref_count = 2; // One more than before.
582 ((_RopeConcatenation*)__r)->_M_right = __new_right;
583 __r->_M_size = __orig_size + __slen;
584 if (0 != __r->_M_c_string) {
585 __r->_M_free_c_string();
586 __r->_M_c_string = 0;
587 }
588 return __r;
589 }
590 }
591 _RopeRep* __right =
592 __STL_ROPE_FROM_UNOWNED_CHAR_PTR(__s, __slen, __r->get_allocator());
593 __r->_M_ref_nonnil();
594 try {
595 __result = _S_tree_concat(__r, __right);
596 }
597 catch(...)
598 {
599 _S_unref(__r);
600 _S_unref(__right);
601 __throw_exception_again;
602 }
603 return __result;
604}
605#endif /* !__GC */
606
607template <class _CharT, class _Alloc>
608typename rope<_CharT,_Alloc>::_RopeRep*
609rope<_CharT,_Alloc>::_S_concat(_RopeRep* __left, _RopeRep* __right)
610{
611 if (0 == __left) {
612 _S_ref(__right);
613 return __right;
614 }
615 if (0 == __right) {
616 __left->_M_ref_nonnil();
617 return __left;
618 }
619 if (_RopeRep::_S_leaf == __right->_M_tag) {
620 if (_RopeRep::_S_leaf == __left->_M_tag) {
621 if (__right->_M_size + __left->_M_size <= _S_copy_max) {
622 return _S_leaf_concat_char_iter((_RopeLeaf*)__left,
623 ((_RopeLeaf*)__right)->_M_data,
624 __right->_M_size);
625 }
626 } else if (_RopeRep::_S_concat == __left->_M_tag
627 && _RopeRep::_S_leaf ==
628 ((_RopeConcatenation*)__left)->_M_right->_M_tag) {
629 _RopeLeaf* __leftright =
630 (_RopeLeaf*)(((_RopeConcatenation*)__left)->_M_right);
631 if (__leftright->_M_size + __right->_M_size <= _S_copy_max) {
632 _RopeRep* __leftleft = ((_RopeConcatenation*)__left)->_M_left;
633 _RopeRep* __rest = _S_leaf_concat_char_iter(__leftright,
634 ((_RopeLeaf*)__right)->_M_data,
635 __right->_M_size);
636 __leftleft->_M_ref_nonnil();
637 try {
638 return(_S_tree_concat(__leftleft, __rest));
639 }
640 catch(...)
641 {
642 _S_unref(__leftleft);
643 _S_unref(__rest);
644 __throw_exception_again;
645 }
646 }
647 }
648 }
649 __left->_M_ref_nonnil();
650 __right->_M_ref_nonnil();
651 try {
652 return(_S_tree_concat(__left, __right));
653 }
654 catch(...)
655 {
656 _S_unref(__left);
657 _S_unref(__right);
658 __throw_exception_again;
659 }
660}
661
662template <class _CharT, class _Alloc>
663typename rope<_CharT,_Alloc>::_RopeRep*
664rope<_CharT,_Alloc>::_S_substring(_RopeRep* __base,
665 size_t __start, size_t __endp1)
666{
667 if (0 == __base) return 0;
668 size_t __len = __base->_M_size;
669 size_t __adj_endp1;
670 const size_t __lazy_threshold = 128;
671
672 if (__endp1 >= __len) {
673 if (0 == __start) {
674 __base->_M_ref_nonnil();
675 return __base;
676 } else {
677 __adj_endp1 = __len;
678 }
679 } else {
680 __adj_endp1 = __endp1;
681 }
682 switch(__base->_M_tag) {
683 case _RopeRep::_S_concat:
684 {
685 _RopeConcatenation* __c = (_RopeConcatenation*)__base;
686 _RopeRep* __left = __c->_M_left;
687 _RopeRep* __right = __c->_M_right;
688 size_t __left_len = __left->_M_size;
689 _RopeRep* __result;
690
691 if (__adj_endp1 <= __left_len) {
692 return _S_substring(__left, __start, __endp1);
693 } else if (__start >= __left_len) {
694 return _S_substring(__right, __start - __left_len,
695 __adj_endp1 - __left_len);
696 }
697 _Self_destruct_ptr __left_result(
698 _S_substring(__left, __start, __left_len));
699 _Self_destruct_ptr __right_result(
700 _S_substring(__right, 0, __endp1 - __left_len));
701 __result = _S_concat(__left_result, __right_result);
702 return __result;
703 }
704 case _RopeRep::_S_leaf:
705 {
706 _RopeLeaf* __l = (_RopeLeaf*)__base;
707 _RopeLeaf* __result;
708 size_t __result_len;
709 if (__start >= __adj_endp1) return 0;
710 __result_len = __adj_endp1 - __start;
711 if (__result_len > __lazy_threshold) goto lazy;
712# ifdef __GC
713 const _CharT* __section = __l->_M_data + __start;
714 __result = _S_new_RopeLeaf(__section, __result_len,
715 __base->get_allocator());
716 __result->_M_c_string = 0; // Not eos terminated.
717# else
718 // We should sometimes create substring node instead.
719 __result = __STL_ROPE_FROM_UNOWNED_CHAR_PTR(
720 __l->_M_data + __start, __result_len,
721 __base->get_allocator());
722# endif
723 return __result;
724 }
725 case _RopeRep::_S_substringfn:
726 // Avoid introducing multiple layers of substring nodes.
727 {
728 _RopeSubstring* __old = (_RopeSubstring*)__base;
729 size_t __result_len;
730 if (__start >= __adj_endp1) return 0;
731 __result_len = __adj_endp1 - __start;
732 if (__result_len > __lazy_threshold) {
733 _RopeSubstring* __result =
734 _S_new_RopeSubstring(__old->_M_base,
735 __start + __old->_M_start,
736 __adj_endp1 - __start,
737 __base->get_allocator());
738 return __result;
739
740 } // *** else fall through: ***
741 }
742 case _RopeRep::_S_function:
743 {
744 _RopeFunction* __f = (_RopeFunction*)__base;
745 _CharT* __section;
746 size_t __result_len;
747 if (__start >= __adj_endp1) return 0;
748 __result_len = __adj_endp1 - __start;
749
750 if (__result_len > __lazy_threshold) goto lazy;
751 __section = (_CharT*)
752 _Data_allocate(_S_rounded_up_size(__result_len));
753 try {
754 (*(__f->_M_fn))(__start, __result_len, __section);
755 }
756 catch(...)
757 {
758 _RopeRep::__STL_FREE_STRING(
759 __section, __result_len, __base->get_allocator());
760 __throw_exception_again;
761 }
762 _S_cond_store_eos(__section[__result_len]);
763 return _S_new_RopeLeaf(__section, __result_len,
764 __base->get_allocator());
765 }
766 }
767 lazy:
768 {
769 // Create substring node.
770 return _S_new_RopeSubstring(__base, __start, __adj_endp1 - __start,
771 __base->get_allocator());
772 }
773}
774
775template<class _CharT>
776class _Rope_flatten_char_consumer : public _Rope_char_consumer<_CharT> {
777 private:
778 _CharT* _M_buf_ptr;
779 public:
780
781 _Rope_flatten_char_consumer(_CharT* __buffer) {
782 _M_buf_ptr = __buffer;
783 };
784 ~_Rope_flatten_char_consumer() {}
785 bool operator() (const _CharT* __leaf, size_t __n) {
786 uninitialized_copy_n(__leaf, __n, _M_buf_ptr);
787 _M_buf_ptr += __n;
788 return true;
789 }
790};
791
792template<class _CharT>
793class _Rope_find_char_char_consumer : public _Rope_char_consumer<_CharT> {
794 private:
795 _CharT _M_pattern;
796 public:
797 size_t _M_count; // Number of nonmatching characters
798 _Rope_find_char_char_consumer(_CharT __p)
799 : _M_pattern(__p), _M_count(0) {}
800 ~_Rope_find_char_char_consumer() {}
801 bool operator() (const _CharT* __leaf, size_t __n) {
802 size_t __i;
803 for (__i = 0; __i < __n; __i++) {
804 if (__leaf[__i] == _M_pattern) {
805 _M_count += __i; return false;
806 }
807 }
808 _M_count += __n; return true;
809 }
810};
811
812 template<class _CharT, class _Traits>
813 // Here _CharT is both the stream and rope character type.
814class _Rope_insert_char_consumer : public _Rope_char_consumer<_CharT> {
815 private:
816 typedef basic_ostream<_CharT,_Traits> _Insert_ostream;
817 _Insert_ostream& _M_o;
818 public:
819 _Rope_insert_char_consumer(_Insert_ostream& __writer)
820 : _M_o(__writer) {};
821 ~_Rope_insert_char_consumer() { };
822 // Caller is presumed to own the ostream
823 bool operator() (const _CharT* __leaf, size_t __n);
824 // Returns true to continue traversal.
825};
826
827template<class _CharT, class _Traits>
828bool _Rope_insert_char_consumer<_CharT, _Traits>::operator()
829 (const _CharT* __leaf, size_t __n)
830{
831 size_t __i;
832 // We assume that formatting is set up correctly for each element.
833 for (__i = 0; __i < __n; __i++) _M_o.put(__leaf[__i]);
834 return true;
835}
836
837template <class _CharT, class _Alloc>
838bool rope<_CharT, _Alloc>::_S_apply_to_pieces(
839 _Rope_char_consumer<_CharT>& __c,
840 const _RopeRep* __r,
841 size_t __begin, size_t __end)
842{
843 if (0 == __r) return true;
844 switch(__r->_M_tag) {
845 case _RopeRep::_S_concat:
846 {
847 _RopeConcatenation* __conc = (_RopeConcatenation*)__r;
848 _RopeRep* __left = __conc->_M_left;
849 size_t __left_len = __left->_M_size;
850 if (__begin < __left_len) {
851 size_t __left_end = std::min(__left_len, __end);
852 if (!_S_apply_to_pieces(__c, __left, __begin, __left_end))
853 return false;
854 }
855 if (__end > __left_len) {
856 _RopeRep* __right = __conc->_M_right;
857 size_t __right_start = std::max(__left_len, __begin);
858 if (!_S_apply_to_pieces(__c, __right,
859 __right_start - __left_len,
860 __end - __left_len)) {
861 return false;
862 }
863 }
864 }
865 return true;
866 case _RopeRep::_S_leaf:
867 {
868 _RopeLeaf* __l = (_RopeLeaf*)__r;
869 return __c(__l->_M_data + __begin, __end - __begin);
870 }
871 case _RopeRep::_S_function:
872 case _RopeRep::_S_substringfn:
873 {
874 _RopeFunction* __f = (_RopeFunction*)__r;
875 size_t __len = __end - __begin;
876 bool __result;
877 _CharT* __buffer =
878 (_CharT*)__alloc::allocate(__len * sizeof(_CharT));
879 try {
880 (*(__f->_M_fn))(__begin, __len, __buffer);
881 __result = __c(__buffer, __len);
882 __alloc::deallocate(__buffer, __len * sizeof(_CharT));
883 }
884 catch(...)
885 {
886 __alloc::deallocate(__buffer, __len * sizeof(_CharT));
887 __throw_exception_again;
888 }
889 return __result;
890 }
891 default:
892 return false;
893 }
894}
895
896 template<class _CharT, class _Traits>
897 inline void _Rope_fill(basic_ostream<_CharT, _Traits>& __o, size_t __n)
898{
899 char __f = __o.fill();
900 size_t __i;
901
902 for (__i = 0; __i < __n; __i++) __o.put(__f);
903}
904
905
906template <class _CharT> inline bool _Rope_is_simple(_CharT*) { return false; }
907inline bool _Rope_is_simple(char*) { return true; }
908inline bool _Rope_is_simple(wchar_t*) { return true; }
909
910template<class _CharT, class _Traits, class _Alloc>
911basic_ostream<_CharT, _Traits>& operator<< (basic_ostream<_CharT, _Traits>& __o,
912 const rope<_CharT, _Alloc>& __r)
913{
914 size_t __w = __o.width();
915 bool __left = bool(__o.flags() & std::ios::left);
916 size_t __pad_len;
917 size_t __rope_len = __r.size();
918 _Rope_insert_char_consumer<_CharT, _Traits> __c(__o);
919 bool __is_simple = _Rope_is_simple((_CharT*)0);
920
921 if (__rope_len < __w) {
922 __pad_len = __w - __rope_len;
923 } else {
924 __pad_len = 0;
925 }
926 if (!__is_simple) __o.width(__w/__rope_len);
927 try {
928 if (__is_simple && !__left && __pad_len > 0) {
929 _Rope_fill(__o, __pad_len);
930 }
931 __r.apply_to_pieces(0, __r.size(), __c);
932 if (__is_simple && __left && __pad_len > 0) {
933 _Rope_fill(__o, __pad_len);
934 }
935 if (!__is_simple)
936 __o.width(__w);
937 }
938 catch(...)
939 {
940 if (!__is_simple)
941 __o.width(__w);
942 __throw_exception_again;
943 }
944 return __o;
945}
946
947template <class _CharT, class _Alloc>
948_CharT*
949rope<_CharT,_Alloc>::_S_flatten(_RopeRep* __r,
950 size_t __start, size_t __len,
951 _CharT* __buffer)
952{
953 _Rope_flatten_char_consumer<_CharT> __c(__buffer);
954 _S_apply_to_pieces(__c, __r, __start, __start + __len);
955 return(__buffer + __len);
956}
957
958template <class _CharT, class _Alloc>
959size_t
960rope<_CharT,_Alloc>::find(_CharT __pattern, size_t __start) const
961{
962 _Rope_find_char_char_consumer<_CharT> __c(__pattern);
963 _S_apply_to_pieces(__c, _M_tree_ptr, __start, size());
964 size_type __result_pos = __start + __c._M_count;
965# ifndef __STL_OLD_ROPE_SEMANTICS
966 if (__result_pos == size()) __result_pos = npos;
967# endif
968 return __result_pos;
969}
970
971template <class _CharT, class _Alloc>
972_CharT*
973rope<_CharT,_Alloc>::_S_flatten(_RopeRep* __r, _CharT* __buffer)
974{
975 if (0 == __r) return __buffer;
976 switch(__r->_M_tag) {
977 case _RopeRep::_S_concat:
978 {
979 _RopeConcatenation* __c = (_RopeConcatenation*)__r;
980 _RopeRep* __left = __c->_M_left;
981 _RopeRep* __right = __c->_M_right;
982 _CharT* __rest = _S_flatten(__left, __buffer);
983 return _S_flatten(__right, __rest);
984 }
985 case _RopeRep::_S_leaf:
986 {
987 _RopeLeaf* __l = (_RopeLeaf*)__r;
988 return copy_n(__l->_M_data, __l->_M_size, __buffer).second;
989 }
990 case _RopeRep::_S_function:
991 case _RopeRep::_S_substringfn:
992 // We don't yet do anything with substring nodes.
993 // This needs to be fixed before ropefiles will work well.
994 {
995 _RopeFunction* __f = (_RopeFunction*)__r;
996 (*(__f->_M_fn))(0, __f->_M_size, __buffer);
997 return __buffer + __f->_M_size;
998 }
999 default:
1000 return 0;
1001 }
1002}
1003
1004
1005// This needs work for _CharT != char
1006template <class _CharT, class _Alloc>
1007void
1008rope<_CharT,_Alloc>::_S_dump(_RopeRep* __r, int __indent)
1009{
1010 for (int __i = 0; __i < __indent; __i++) putchar(' ');
1011 if (0 == __r) {
1012 printf("NULL\n"); return;
1013 }
1014 if (_RopeRep::_S_concat == __r->_M_tag) {
1015 _RopeConcatenation* __c = (_RopeConcatenation*)__r;
1016 _RopeRep* __left = __c->_M_left;
1017 _RopeRep* __right = __c->_M_right;
1018
1019# ifdef __GC
1020 printf("Concatenation %p (depth = %d, len = %ld, %s balanced)\n",
1021 __r, __r->_M_depth, __r->_M_size, __r->_M_is_balanced? "" : "not");
1022# else
1023 printf("Concatenation %p (rc = %ld, depth = %d, "
1024 "len = %ld, %s balanced)\n",
1025 __r, __r->_M_ref_count, __r->_M_depth, __r->_M_size,
1026 __r->_M_is_balanced? "" : "not");
1027# endif
1028 _S_dump(__left, __indent + 2);
1029 _S_dump(__right, __indent + 2);
1030 return;
1031 } else {
1032 char* __kind;
1033
1034 switch (__r->_M_tag) {
1035 case _RopeRep::_S_leaf:
1036 __kind = "Leaf";
1037 break;
1038 case _RopeRep::_S_function:
1039 __kind = "Function";
1040 break;
1041 case _RopeRep::_S_substringfn:
1042 __kind = "Function representing substring";
1043 break;
1044 default:
1045 __kind = "(corrupted kind field!)";
1046 }
1047# ifdef __GC
1048 printf("%s %p (depth = %d, len = %ld) ",
1049 __kind, __r, __r->_M_depth, __r->_M_size);
1050# else
1051 printf("%s %p (rc = %ld, depth = %d, len = %ld) ",
1052 __kind, __r, __r->_M_ref_count, __r->_M_depth, __r->_M_size);
1053# endif
1054 if (_S_is_one_byte_char_type((_CharT*)0)) {
1055 const int __max_len = 40;
1056 _Self_destruct_ptr __prefix(_S_substring(__r, 0, __max_len));
1057 _CharT __buffer[__max_len + 1];
1058 bool __too_big = __r->_M_size > __prefix->_M_size;
1059
1060 _S_flatten(__prefix, __buffer);
1061 __buffer[__prefix->_M_size] = _S_eos((_CharT*)0);
1062 printf("%s%s\n",
1063 (char*)__buffer, __too_big? "...\n" : "\n");
1064 } else {
1065 printf("\n");
1066 }
1067 }
1068}
1069
1070template <class _CharT, class _Alloc>
1071const unsigned long
1072rope<_CharT,_Alloc>::_S_min_len[
1073 _Rope_RopeRep<_CharT,_Alloc>::_S_max_rope_depth + 1] = {
1074/* 0 */1, /* 1 */2, /* 2 */3, /* 3 */5, /* 4 */8, /* 5 */13, /* 6 */21,
1075/* 7 */34, /* 8 */55, /* 9 */89, /* 10 */144, /* 11 */233, /* 12 */377,
1076/* 13 */610, /* 14 */987, /* 15 */1597, /* 16 */2584, /* 17 */4181,
1077/* 18 */6765, /* 19 */10946, /* 20 */17711, /* 21 */28657, /* 22 */46368,
1078/* 23 */75025, /* 24 */121393, /* 25 */196418, /* 26 */317811,
1079/* 27 */514229, /* 28 */832040, /* 29 */1346269, /* 30 */2178309,
1080/* 31 */3524578, /* 32 */5702887, /* 33 */9227465, /* 34 */14930352,
1081/* 35 */24157817, /* 36 */39088169, /* 37 */63245986, /* 38 */102334155,
1082/* 39 */165580141, /* 40 */267914296, /* 41 */433494437,
1083/* 42 */701408733, /* 43 */1134903170, /* 44 */1836311903,
1084/* 45 */2971215073u };
1085// These are Fibonacci numbers < 2**32.
1086
1087template <class _CharT, class _Alloc>
1088typename rope<_CharT,_Alloc>::_RopeRep*
1089rope<_CharT,_Alloc>::_S_balance(_RopeRep* __r)
1090{
1091 _RopeRep* __forest[_RopeRep::_S_max_rope_depth + 1];
1092 _RopeRep* __result = 0;
1093 int __i;
1094 // Invariant:
1095 // The concatenation of forest in descending order is equal to __r.
1096 // __forest[__i]._M_size >= _S_min_len[__i]
1097 // __forest[__i]._M_depth = __i
1098 // References from forest are included in refcount.
1099
1100 for (__i = 0; __i <= _RopeRep::_S_max_rope_depth; ++__i)
1101 __forest[__i] = 0;
1102 try {
1103 _S_add_to_forest(__r, __forest);
1104 for (__i = 0; __i <= _RopeRep::_S_max_rope_depth; ++__i)
1105 if (0 != __forest[__i]) {
1106# ifndef __GC
1107 _Self_destruct_ptr __old(__result);
1108# endif
1109 __result = _S_concat(__forest[__i], __result);
1110 __forest[__i]->_M_unref_nonnil();
1111# if !defined(__GC) && defined(__EXCEPTIONS)
1112 __forest[__i] = 0;
1113# endif
1114 }
1115 }
1116 catch(...)
1117 {
1118 for(__i = 0; __i <= _RopeRep::_S_max_rope_depth; __i++)
1119 _S_unref(__forest[__i]);
1120 __throw_exception_again;
1121 }