57 lines
2.0 KiB
Rust
57 lines
2.0 KiB
Rust
|
use superslice::Ext;
|
||
|
|
use ::{TextUnit};
|
||
|
|
|
||
|
|
pub struct LineIndex {
|
||
|
|
newlines: Vec<TextUnit>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
|
||
|
|
pub struct LineCol {
|
||
|
|
pub line: u32,
|
||
|
|
pub col: TextUnit,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl LineIndex {
|
||
|
|
pub fn new(text: &str) -> LineIndex {
|
||
|
|
let mut newlines = vec![0.into()];
|
||
|
|
let mut curr = 0.into();
|
||
|
|
for c in text.chars() {
|
||
|
|
curr += TextUnit::of_char(c);
|
||
|
|
if c == '\n' {
|
||
|
|
newlines.push(curr);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
LineIndex { newlines }
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn translate(&self, offset: TextUnit) -> LineCol {
|
||
|
|
let line = self.newlines.upper_bound(&offset) - 1;
|
||
|
|
let line_start_offset = self.newlines[line];
|
||
|
|
let col = offset - line_start_offset;
|
||
|
|
return LineCol { line: line as u32, col }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_line_index() {
|
||
|
|
let text = "hello\nworld";
|
||
|
|
let index = LineIndex::new(text);
|
||
|
|
assert_eq!(index.translate(0.into()), LineCol { line: 0, col: 0.into()});
|
||
|
|
assert_eq!(index.translate(1.into()), LineCol { line: 0, col: 1.into()});
|
||
|
|
assert_eq!(index.translate(5.into()), LineCol { line: 0, col: 5.into()});
|
||
|
|
assert_eq!(index.translate(6.into()), LineCol { line: 1, col: 0.into()});
|
||
|
|
assert_eq!(index.translate(7.into()), LineCol { line: 1, col: 1.into()});
|
||
|
|
assert_eq!(index.translate(8.into()), LineCol { line: 1, col: 2.into()});
|
||
|
|
assert_eq!(index.translate(10.into()), LineCol { line: 1, col: 4.into()});
|
||
|
|
assert_eq!(index.translate(11.into()), LineCol { line: 1, col: 5.into()});
|
||
|
|
assert_eq!(index.translate(12.into()), LineCol { line: 1, col: 6.into()});
|
||
|
|
|
||
|
|
let text = "\nhello\nworld";
|
||
|
|
let index = LineIndex::new(text);
|
||
|
|
assert_eq!(index.translate(0.into()), LineCol { line: 0, col: 0.into()});
|
||
|
|
assert_eq!(index.translate(1.into()), LineCol { line: 1, col: 0.into()});
|
||
|
|
assert_eq!(index.translate(2.into()), LineCol { line: 1, col: 1.into()});
|
||
|
|
assert_eq!(index.translate(6.into()), LineCol { line: 1, col: 5.into()});
|
||
|
|
assert_eq!(index.translate(7.into()), LineCol { line: 2, col: 0.into()});
|
||
|
|
}
|