diff options
author | Andreas Rumpf <rumpf_a@web.de> | 2021-04-08 00:54:47 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-04-08 00:54:47 +0200 |
commit | 643dbc743bf6436cdc659ae57e2841e125aaa9de (patch) | |
tree | 707fd9b64985fb4456d37a40027fc1aef329d392 /lib/pure/strutils.nim | |
parent | bd7807de4c7f8474b029f36b4d3527f7a03de6e4 (diff) | |
download | Nim-643dbc743bf6436cdc659ae57e2841e125aaa9de.tar.gz |
use strstr for a faster find implementation (#17672)
* use strstr for a faster find implementation * stress the -d:release and -d:danger switches
Diffstat (limited to 'lib/pure/strutils.nim')
-rw-r--r-- | lib/pure/strutils.nim | 29 |
1 files changed, 26 insertions, 3 deletions
diff --git a/lib/pure/strutils.nim b/lib/pure/strutils.nim index 3680b11de..bfd53a3a0 100644 --- a/lib/pure/strutils.nim +++ b/lib/pure/strutils.nim @@ -1826,6 +1826,9 @@ func find*(a: SkipTable, s, sub: string, start: Natural = 0, last = 0): int {. when not (defined(js) or defined(nimdoc) or defined(nimscript)): func c_memchr(cstr: pointer, c: char, n: csize_t): pointer {. importc: "memchr", header: "<string.h>".} + func c_strstr(haystack, needle: cstring): cstring {. + importc: "strstr", header: "<string.h>".} + const hasCStringBuiltin = true else: const hasCStringBuiltin = false @@ -1889,9 +1892,29 @@ func find*(s, sub: string, start: Natural = 0, last = 0): int {.rtl, ## * `replace func<#replace,string,string,string>`_ if sub.len > s.len: return -1 if sub.len == 1: return find(s, sub[0], start, last) - var a {.noinit.}: SkipTable - initSkipTable(a, sub) - result = find(a, s, sub, start, last) + + template useSkipTable {.dirty.} = + var a {.noinit.}: SkipTable + initSkipTable(a, sub) + result = find(a, s, sub, start, last) + + when not hasCStringBuiltin: + useSkipTable() + else: + when nimvm: + useSkipTable() + else: + when hasCStringBuiltin: + if last == 0 and s.len > start: + let found = c_strstr(s[start].unsafeAddr, sub) + if not found.isNil: + result = cast[ByteAddress](found) -% cast[ByteAddress](s.cstring) + else: + result = -1 + else: + useSkipTable() + else: + useSkipTable() func rfind*(s: string, sub: char, start: Natural = 0, last = -1): int {.rtl, extern: "nsuRFindChar".} = |