authorgravatar for frmdstryr@protonmail.comfrmdstryr <frmdstryr@protonmail.com> 2023-11-08 20:45:25-05:00
committergravatar for noreply@github.comGitHub <noreply@github.com> 2023-11-09 01:45:25+00:00
logf258a391daa31b3ba2c37d879db96fadc0c058f3
tree07b32cd0d52119e8b1cf17d1679726d43ee7c54e
parent2ac0ba03a60a133e7ffbab9ad6aaf3b54f6c747b
signature Signed by PGP key 4AEE18F83AFDEB23

Speed up ast.tokenLocation

1 files changed, 14 insertions(+), 3 deletions(-)

lib/std/zig/Ast.zig+14-3
...@@ -140,9 +140,20 @@ pub fn tokenLocation(self: Ast, start_offset: ByteOffset, token_index: TokenInde...@@ -140,9 +140,20 @@ pub fn tokenLocation(self: Ast, start_offset: ByteOffset, token_index: TokenInde
140 .line_end = self.source.len,140 .line_end = self.source.len,
141 };141 };
142 const token_start = self.tokens.items(.start)[token_index];142 const token_start = self.tokens.items(.start)[token_index];
143 for (self.source[start_offset..], 0..) |c, i| {143
144 if (i + start_offset == token_start) {144 // Scan to by line until we go past the token start
145 loc.line_end = i + start_offset;145 while (std.mem.indexOfScalarPos(u8, self.source, loc.line_start, '\n')) |i| {
146 if (i >= token_start) {
147 break; // Went past
148 }
149 loc.line += 1;
150 loc.line_start = i + 1;
151 }
152
153 const offset = loc.line_start;
154 for (self.source[offset..], 0..) |c, i| {
155 if (i + offset == token_start) {
156 loc.line_end = i + offset;
146 while (loc.line_end < self.source.len and self.source[loc.line_end] != '\n') {157 while (loc.line_end < self.source.len and self.source[loc.line_end] != '\n') {
147 loc.line_end += 1;158 loc.line_end += 1;
148 }159 }