core/sbufs/sbufs.factor

   1 ! Copyright (C) 2004, 2008 Slava Pestov.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: accessors kernel math strings sequences.private sequences
   4 strings growable strings.private ;
   5 IN: sbufs
   6
   7 TUPLE: sbuf
   8 { underlying string }
   9 { length array-capacity } ;
  10
  11 : <sbuf> ( n -- sbuf ) 0 <string> 0 sbuf boa ; inline
  12
  13 M: sbuf set-nth-unsafe
  14     [ >fixnum ] [ >fixnum ] [ underlying>> ] tri* set-string-nth ;
  15
  16 M: sbuf new-sequence
  17     drop [ 0 <string> ] [ >fixnum ] bi sbuf boa ;
  18
  19 : >sbuf ( seq -- sbuf ) SBUF" " clone-like ; inline
  20
  21 M: sbuf like
  22     drop dup sbuf? [
  23         dup string? [ dup length sbuf boa ] [ >sbuf ] if
  24     ] unless ;
  25
  26 M: sbuf new-resizable drop <sbuf> ;
  27
  28 M: sbuf equal?
  29     over sbuf? [ sequence= ] [ 2drop f ] if ;
  30
  31 M: string new-resizable drop <sbuf> ;
  32
  33 M: string like
  34     #! If we have a string, we're done.
  35     #! If we have an sbuf, and it's at full capacity, we're done.
  36     #! Otherwise, call resize-string, which is a relatively
  37     #! fast primitive.
  38     drop dup string? [
  39         dup sbuf? [
  40             [ length ] [ underlying>> ] bi
  41             2dup length eq?
  42             [ nip dup reset-string-hashcode ] [ resize-string ] if
  43         ] [ >string ] if
  44     ] unless ;
  45
  46 INSTANCE: sbuf growable