2020-02-26 15:22:12 +03:00
|
|
|
module strings
|
|
|
|
|
|
|
|
// strings.repeat - fill a string with `n` repetitions of the character `c`
|
|
|
|
pub fn repeat(c byte, n int) string {
|
|
|
|
if n <= 0 {
|
|
|
|
return ''
|
|
|
|
}
|
2021-06-15 14:47:11 +03:00
|
|
|
mut bytes := unsafe { malloc_noscan(n + 1) }
|
2020-07-15 22:56:50 +03:00
|
|
|
unsafe {
|
2020-12-21 23:00:32 +03:00
|
|
|
C.memset(bytes, c, n)
|
2020-07-15 22:56:50 +03:00
|
|
|
bytes[n] = `0`
|
|
|
|
}
|
2020-12-23 21:13:42 +03:00
|
|
|
return unsafe { bytes.vstring_with_len(n) }
|
2020-02-26 15:22:12 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// strings.repeat_string - gives you `n` repetitions of the substring `s`
|
|
|
|
// NB: strings.repeat, that repeats a single byte, is between 2x
|
|
|
|
// and 24x faster than strings.repeat_string called for a 1 char string.
|
|
|
|
pub fn repeat_string(s string, n int) string {
|
|
|
|
if n <= 0 || s.len == 0 {
|
|
|
|
return ''
|
|
|
|
}
|
|
|
|
slen := s.len
|
2020-12-21 23:00:32 +03:00
|
|
|
blen := slen * n
|
2021-06-15 14:47:11 +03:00
|
|
|
mut bytes := unsafe { malloc_noscan(blen + 1) }
|
2020-12-21 23:00:32 +03:00
|
|
|
for bi in 0 .. n {
|
|
|
|
bislen := bi * slen
|
|
|
|
for si in 0 .. slen {
|
2020-07-15 22:56:50 +03:00
|
|
|
unsafe {
|
2020-12-21 23:00:32 +03:00
|
|
|
bytes[bislen + si] = s[si]
|
2020-07-15 22:56:50 +03:00
|
|
|
}
|
2020-02-26 15:22:12 +03:00
|
|
|
}
|
|
|
|
}
|
2020-07-15 22:56:50 +03:00
|
|
|
unsafe {
|
|
|
|
bytes[blen] = `0`
|
|
|
|
}
|
2020-12-23 21:13:42 +03:00
|
|
|
return unsafe { bytes.vstring_with_len(blen) }
|
2020-02-26 15:22:12 +03:00
|
|
|
}
|