1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/strings/builder.v

147 lines
3.5 KiB
V
Raw Normal View History

// Copyright (c) 2019-2021 Alexander Medvednikov. All rights reserved.
2019-06-23 05:21:30 +03:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-09-14 23:48:30 +03:00
module strings
2019-06-22 21:20:28 +03:00
// strings.Builder is used to efficiently append many strings to a large
2020-12-21 23:00:32 +03:00
// dynamically growing buffer, then use the resulting large string. Using
// a string builder is much better for performance/memory usage than doing
// constantly string concatenation.
2019-10-24 12:47:21 +03:00
pub struct Builder {
2020-05-09 16:16:48 +03:00
pub mut:
2019-12-20 00:29:37 +03:00
buf []byte
len int
initial_size int = 1
2019-06-22 21:20:28 +03:00
}
2020-12-21 23:00:32 +03:00
// new_builder returns a new string builder, with an initial capacity of `initial_size`
pub fn new_builder(initial_size int) Builder {
2019-12-20 00:29:37 +03:00
return Builder{
2020-12-21 23:00:32 +03:00
// buf: make(0, initial_size)
buf: []byte{cap: initial_size}
2020-05-18 22:38:06 +03:00
len: 0
initial_size: initial_size
2019-06-22 21:20:28 +03:00
}
}
2020-12-21 23:00:32 +03:00
// write_bytes appends `bytes` to the accumulated buffer
[unsafe]
2020-05-17 14:51:18 +03:00
pub fn (mut b Builder) write_bytes(bytes byteptr, howmany int) {
2019-12-11 19:20:46 +03:00
b.buf.push_many(bytes, howmany)
b.len += howmany
}
2020-12-21 23:00:32 +03:00
// write_b appends a single `data` byte to the accumulated buffer
2020-05-17 14:51:18 +03:00
pub fn (mut b Builder) write_b(data byte) {
2019-12-06 23:02:09 +03:00
b.buf << data
2019-12-08 14:11:19 +03:00
b.len++
2019-12-06 23:02:09 +03:00
}
// write implements the Writer interface
pub fn (mut b Builder) write(data []byte) ?int {
b.buf << data
b.len += data.len
return data.len
}
2020-12-21 23:00:32 +03:00
// write appends the string `s` to the buffer
[inline]
pub fn (mut b Builder) write_string(s string) {
2019-12-24 20:54:43 +03:00
if s == '' {
return
}
2019-10-10 01:15:19 +03:00
b.buf.push_many(s.str, s.len)
2019-12-20 00:29:37 +03:00
// for c in s {
// b.buf << c
// }
// b.buf << []byte(s) // TODO
2019-06-22 21:20:28 +03:00
b.len += s.len
}
2020-12-21 23:00:32 +03:00
// go_back discards the last `n` bytes from the buffer
2020-05-17 14:51:18 +03:00
pub fn (mut b Builder) go_back(n int) {
2020-12-21 23:00:32 +03:00
b.buf.trim(b.buf.len - n)
b.len -= n
}
2020-03-11 01:21:26 +03:00
fn bytes2string(b []byte) string {
mut copy := b.clone()
2020-08-27 07:46:18 +03:00
copy << byte(`\0`)
return unsafe { tos(copy.data, copy.len - 1) }
}
2020-12-21 23:00:32 +03:00
// cut_last cuts the last `n` bytes from the buffer and returns them
2020-06-17 01:59:33 +03:00
pub fn (mut b Builder) cut_last(n int) string {
2020-12-21 23:00:32 +03:00
res := bytes2string(b.buf[b.len - n..])
b.buf.trim(b.buf.len - n)
2020-06-17 01:59:33 +03:00
b.len -= n
return res
2020-06-17 01:59:33 +03:00
}
/*
pub fn (mut b Builder) cut_to(pos int) string {
res := bytes2string( b.buf[pos..] )
2020-06-17 01:59:33 +03:00
b.buf.trim(pos)
b.len = pos
return res
2020-06-17 01:59:33 +03:00
}
*/
2020-12-21 23:00:32 +03:00
// go_back_to resets the buffer to the given position `pos`
// NB: pos should be < than the existing buffer length.
2020-05-17 14:51:18 +03:00
pub fn (mut b Builder) go_back_to(pos int) {
2020-04-28 14:15:37 +03:00
b.buf.trim(pos)
b.len = pos
}
2020-12-21 23:00:32 +03:00
// writeln appends the string `s`, and then a newline character.
[inline]
2020-05-17 14:51:18 +03:00
pub fn (mut b Builder) writeln(s string) {
2019-12-20 00:29:37 +03:00
// for c in s {
// b.buf << c
// }
2019-10-10 01:15:19 +03:00
b.buf.push_many(s.str, s.len)
2019-12-20 00:29:37 +03:00
// b.buf << []byte(s) // TODO
2019-06-22 21:20:28 +03:00
b.buf << `\n`
b.len += s.len + 1
}
2020-03-21 09:01:06 +03:00
// buf == 'hello world'
// last_n(5) returns 'world'
2020-03-21 09:04:53 +03:00
pub fn (b &Builder) last_n(n int) string {
2020-03-21 09:01:06 +03:00
if n > b.len {
return ''
}
2020-12-21 23:00:32 +03:00
return bytes2string(b.buf[b.len - n..])
2020-03-21 09:01:06 +03:00
}
// buf == 'hello world'
// after(6) returns 'world'
2020-03-21 09:04:53 +03:00
pub fn (b &Builder) after(n int) string {
2020-03-21 09:01:06 +03:00
if n >= b.len {
return ''
}
2020-12-21 23:00:32 +03:00
return bytes2string(b.buf[n..])
2020-03-21 09:01:06 +03:00
}
// str returns a copy of all of the accumulated buffer content.
// NB: after a call to b.str(), the builder b should not be
// used again, you need to call b.free() first, or just leave
// it to be freed by -autofree when it goes out of scope.
// The returned string *owns* its own separate copy of the
// accumulated data that was in the string builder, before the
// .str() call.
2020-05-17 14:51:18 +03:00
pub fn (mut b Builder) str() string {
2019-11-17 02:22:43 +03:00
b.buf << `\0`
s := unsafe { byteptr(memdup(b.buf.data, b.len)).vstring_with_len(b.len) }
b.len = 0
b.buf.trim(0)
return s
2019-06-22 21:20:28 +03:00
}
// free - manually free the contents of the buffer
[unsafe]
2020-05-17 14:51:18 +03:00
pub fn (mut b Builder) free() {
unsafe { free(b.buf.data) }
b.len = 0
2019-06-24 23:34:21 +03:00
}