diff options
| author | Kevin Cantu <me@kevincantu.org> | 2012-05-29 21:35:12 -0700 |
|---|---|---|
| committer | Brian Anderson <banderson@mozilla.com> | 2012-05-31 11:15:00 -0700 |
| commit | 7dcbaedd329295e1f2692bb2d9eae860a820d0a8 (patch) | |
| tree | be0272a0ec2e259da231f9147aaf1dc09b5758c5 /src/libsyntax/codemap.rs | |
| parent | ff6cde788229484e128849d67e4f32b178b18c84 (diff) | |
| download | rust-7dcbaedd329295e1f2692bb2d9eae860a820d0a8.tar.gz rust-7dcbaedd329295e1f2692bb2d9eae860a820d0a8.zip | |
Rename librustsyntax to libsyntax
Per issue #2418.
Diffstat (limited to 'src/libsyntax/codemap.rs')
| -rw-r--r-- | src/libsyntax/codemap.rs | 236 |
1 files changed, 236 insertions, 0 deletions
diff --git a/src/libsyntax/codemap.rs b/src/libsyntax/codemap.rs new file mode 100644 index 00000000000..259041959d4 --- /dev/null +++ b/src/libsyntax/codemap.rs @@ -0,0 +1,236 @@ +import dvec::{dvec, extensions}; + +export filename; +export filemap; +export span; +export file_substr; +export codemap; +export expn_info; +export expn_info_; +export expanded_from; +export new_filemap; +export new_filemap_w_substr; +export mk_substr_filename; +export lookup_char_pos; +export lookup_char_pos_adj; +export adjust_span; +export span_to_str; +export span_to_filename; +export span_to_lines; +export file_lines; +export get_line; +export next_line; +export span_to_snippet; +export loc; +export get_filemap; +export new_codemap; + +type filename = str; + +type file_pos = {ch: uint, byte: uint}; + +/* A codemap is a thing that maps uints to file/line/column positions + * in a crate. This to make it possible to represent the positions + * with single-word things, rather than passing records all over the + * compiler. + */ + +enum file_substr { + fss_none, + fss_internal(span), + fss_external({filename: str, line: uint, col: uint}) +} + +type filemap = + @{name: filename, substr: file_substr, src: @str, + start_pos: file_pos, mut lines: [file_pos]}; + +type codemap = @{files: dvec<filemap>}; + +type loc = {file: filemap, line: uint, col: uint}; + +fn new_codemap() -> codemap { @{files: dvec()} } + +fn new_filemap_w_substr(filename: filename, substr: file_substr, + src: @str, + start_pos_ch: uint, start_pos_byte: uint) + -> filemap { + ret @{name: filename, substr: substr, src: src, + start_pos: {ch: start_pos_ch, byte: start_pos_byte}, + mut lines: [{ch: start_pos_ch, byte: start_pos_byte}]}; +} + +fn new_filemap(filename: filename, src: @str, + start_pos_ch: uint, start_pos_byte: uint) + -> filemap { + ret new_filemap_w_substr(filename, fss_none, src, + start_pos_ch, start_pos_byte); +} + +fn mk_substr_filename(cm: codemap, sp: span) -> str +{ + let pos = lookup_char_pos(cm, sp.lo); + ret #fmt("<%s:%u:%u>", pos.file.name, pos.line, pos.col); +} + +fn next_line(file: filemap, chpos: uint, byte_pos: uint) { + file.lines += [{ch: chpos, byte: byte_pos + file.start_pos.byte}]; +} + +type lookup_fn = fn@(file_pos) -> uint; + +fn lookup_line(map: codemap, pos: uint, lookup: lookup_fn) + -> {fm: filemap, line: uint} +{ + let len = map.files.len(); + let mut a = 0u; + let mut b = len; + while b - a > 1u { + let m = (a + b) / 2u; + if lookup(map.files[m].start_pos) > pos { b = m; } else { a = m; } + } + if (a >= len) { + fail #fmt("position %u does not resolve to a source location", pos) + } + let f = map.files[a]; + a = 0u; + b = vec::len(f.lines); + while b - a > 1u { + let m = (a + b) / 2u; + if lookup(f.lines[m]) > pos { b = m; } else { a = m; } + } + ret {fm: f, line: a}; +} + +fn lookup_pos(map: codemap, pos: uint, lookup: lookup_fn) -> loc { + let {fm: f, line: a} = lookup_line(map, pos, lookup); + ret {file: f, line: a + 1u, col: pos - lookup(f.lines[a])}; +} + +fn lookup_char_pos(map: codemap, pos: uint) -> loc { + fn lookup(pos: file_pos) -> uint { ret pos.ch; } + ret lookup_pos(map, pos, lookup); +} + +fn lookup_byte_pos(map: codemap, pos: uint) -> loc { + fn lookup(pos: file_pos) -> uint { ret pos.byte; } + ret lookup_pos(map, pos, lookup); +} + +fn lookup_char_pos_adj(map: codemap, pos: uint) + -> {filename: str, line: uint, col: uint, file: option<filemap>} +{ + let loc = lookup_char_pos(map, pos); + alt (loc.file.substr) { + fss_none { + {filename: loc.file.name, line: loc.line, col: loc.col, + file: some(loc.file)} + } + fss_internal(sp) { + lookup_char_pos_adj(map, sp.lo + (pos - loc.file.start_pos.ch)) + } + fss_external(eloc) { + {filename: eloc.filename, + line: eloc.line + loc.line - 1u, + col: if loc.line == 1u {eloc.col + loc.col} else {loc.col}, + file: none} + } + } +} + +fn adjust_span(map: codemap, sp: span) -> span { + fn lookup(pos: file_pos) -> uint { ret pos.ch; } + let line = lookup_line(map, sp.lo, lookup); + alt (line.fm.substr) { + fss_none {sp} + fss_internal(s) { + adjust_span(map, {lo: s.lo + (sp.lo - line.fm.start_pos.ch), + hi: s.lo + (sp.hi - line.fm.start_pos.ch), + expn_info: sp.expn_info})} + fss_external(_) {sp} + } +} + +enum expn_info_ { + expanded_from({call_site: span, + callie: {name: str, span: option<span>}}) +} +type expn_info = option<@expn_info_>; +type span = {lo: uint, hi: uint, expn_info: expn_info}; + +fn span_to_str_no_adj(sp: span, cm: codemap) -> str { + let lo = lookup_char_pos(cm, sp.lo); + let hi = lookup_char_pos(cm, sp.hi); + ret #fmt("%s:%u:%u: %u:%u", lo.file.name, + lo.line, lo.col, hi.line, hi.col) +} + +fn span_to_str(sp: span, cm: codemap) -> str { + let lo = lookup_char_pos_adj(cm, sp.lo); + let hi = lookup_char_pos_adj(cm, sp.hi); + ret #fmt("%s:%u:%u: %u:%u", lo.filename, + lo.line, lo.col, hi.line, hi.col) +} + +type file_lines = {file: filemap, lines: [uint]}; + +fn span_to_filename(sp: span, cm: codemap::codemap) -> filename { + let lo = lookup_char_pos(cm, sp.lo); + ret lo.file.name; +} + +fn span_to_lines(sp: span, cm: codemap::codemap) -> @file_lines { + let lo = lookup_char_pos(cm, sp.lo); + let hi = lookup_char_pos(cm, sp.hi); + let mut lines = []; + for uint::range(lo.line - 1u, hi.line as uint) {|i| lines += [i]; }; + ret @{file: lo.file, lines: lines}; +} + +fn get_line(fm: filemap, line: int) -> str unsafe { + let begin: uint = fm.lines[line].byte - fm.start_pos.byte; + let end = alt str::find_char_from(*fm.src, '\n', begin) { + some(e) { e } + none { str::len(*fm.src) } + }; + str::slice(*fm.src, begin, end) +} + +fn lookup_byte_offset(cm: codemap::codemap, chpos: uint) + -> {fm: filemap, pos: uint} { + let {fm, line} = lookup_line(cm, chpos, {|pos| pos.ch}); + let line_offset = fm.lines[line].byte - fm.start_pos.byte; + let col = chpos - fm.lines[line].ch; + let col_offset = str::count_bytes(*fm.src, line_offset, col); + {fm: fm, pos: line_offset + col_offset} +} + +fn span_to_snippet(sp: span, cm: codemap::codemap) -> str { + let begin = lookup_byte_offset(cm, sp.lo); + let end = lookup_byte_offset(cm, sp.hi); + assert begin.fm == end.fm; + ret str::slice(*begin.fm.src, begin.pos, end.pos); +} + +fn get_snippet(cm: codemap::codemap, fidx: uint, lo: uint, hi: uint) -> str +{ + let fm = cm.files[fidx]; + ret str::slice(*fm.src, lo, hi) +} + +fn get_filemap(cm: codemap, filename: str) -> filemap { + for cm.files.each {|fm| if fm.name == filename { ret fm; } } + //XXjdm the following triggers a mismatched type bug + // (or expected function, found _|_) + fail; // ("asking for " + filename + " which we don't know about"); +} + +// +// Local Variables: +// mode: rust +// fill-column: 78; +// indent-tabs-mode: nil +// c-basic-offset: 4 +// buffer-file-coding-system: utf-8-unix +// End: +// |
