@@ -1,8 +1,6 @@
1−use std::borrow::Cow;
2−
31use crate::{
42 chars::char_is_line_ending,
5− graphemes::{ensure_grapheme_boundary_prev, grapheme_width, RopeGraphemes},
3+ graphemes::{ensure_grapheme_boundary_prev, RopeGraphemes},
64 line_ending::line_end_char_index,
75 RopeSlice,
86};
@@ -66,12 +64,7 @@ pub fn coords_at_pos(text: RopeSlice, pos: usize) -> Position {
6664
6765 let line_start = text.line_to_char(line);
6866 let pos = ensure_grapheme_boundary_prev(text, pos);
69− let col = RopeGraphemes::new(text.slice(line_start..pos))
70− .map(|g| {
71− let g: Cow<str> = g.into();
72− grapheme_width(&g)
73− })
74− .sum();
67+ let col = RopeGraphemes::new(text.slice(line_start..pos)).count();
7568
7669 Position::new(line, col)
7770}
@@ -83,6 +76,9 @@ pub fn coords_at_pos(text: RopeSlice, pos: usize) -> Position {
8376/// `false` corresponds to properly round-tripping with `coords_at_pos()`,
8477/// whereas `true` will ensure that block cursors don't jump off the
8578/// end of the line.
79+///
80+/// TODO: this should be changed to work in terms of visual row/column, not
81+/// graphemes.
8682pub fn pos_at_coords(text: RopeSlice, coords: Position, is_1_width: bool) -> usize {
8783 let Position { row, col } = coords;
8884 let line_start = text.line_to_char(row);
@@ -92,17 +88,11 @@ pub fn pos_at_coords(text: RopeSlice, coords: Position, is_1_width: bool) -> usi
9288 text.line_to_char((row + 1).min(text.len_lines()))
9389 };
9490
95− let mut prev_col = 0;
9691 let mut col_char_offset = 0;
97− for g in RopeGraphemes::new(text.slice(line_start..line_end)) {
98− let g: Cow<str> = g.into();
99− let next_col = prev_col + grapheme_width(&g);
100−
101− if next_col > col {
92+ for (i, g) in RopeGraphemes::new(text.slice(line_start..line_end)).enumerate() {
93+ if i == col {
10294 break;
10395 }
104−
105− prev_col = next_col;
10696 col_char_offset += g.chars().count();
10797 }
10898
@@ -131,17 +121,18 @@ mod test {
131121 assert_eq!(coords_at_pos(slice, 10), (1, 4).into()); // position on d
132122
133123 // Test with wide characters.
124+ // TODO: account for character width.
134125 let text = Rope::from("今日はいい\n");
135126 let slice = text.slice(..);
136127 assert_eq!(coords_at_pos(slice, 0), (0, 0).into());
137− assert_eq!(coords_at_pos(slice, 1), (0, 2).into());
138− assert_eq!(coords_at_pos(slice, 2), (0, 4).into());
139− assert_eq!(coords_at_pos(slice, 3), (0, 6).into());
140− assert_eq!(coords_at_pos(slice, 4), (0, 8).into());
141− assert_eq!(coords_at_pos(slice, 5), (0, 10).into());
128+ assert_eq!(coords_at_pos(slice, 1), (0, 1).into());
129+ assert_eq!(coords_at_pos(slice, 2), (0, 2).into());
130+ assert_eq!(coords_at_pos(slice, 3), (0, 3).into());
131+ assert_eq!(coords_at_pos(slice, 4), (0, 4).into());
132+ assert_eq!(coords_at_pos(slice, 5), (0, 5).into());
142133 assert_eq!(coords_at_pos(slice, 6), (1, 0).into());
143134
144− // test with grapheme clusters
135+ // Test with grapheme clusters.
145136 let text = Rope::from("a̐éö̲\r\n");
146137 let slice = text.slice(..);
147138 assert_eq!(coords_at_pos(slice, 0), (0, 0).into());
@@ -150,13 +141,23 @@ mod test {
150141 assert_eq!(coords_at_pos(slice, 7), (0, 3).into());
151142 assert_eq!(coords_at_pos(slice, 9), (1, 0).into());
152143
144+ // Test with wide-character grapheme clusters.
145+ // TODO: account for character width.
153146 let text = Rope::from("किमपि\n");
154147 let slice = text.slice(..);
155148 assert_eq!(coords_at_pos(slice, 0), (0, 0).into());
156− assert_eq!(coords_at_pos(slice, 2), (0, 2).into());
157− assert_eq!(coords_at_pos(slice, 3), (0, 3).into());
158− assert_eq!(coords_at_pos(slice, 5), (0, 5).into());
149+ assert_eq!(coords_at_pos(slice, 2), (0, 1).into());
150+ assert_eq!(coords_at_pos(slice, 3), (0, 2).into());
151+ assert_eq!(coords_at_pos(slice, 5), (0, 3).into());
159152 assert_eq!(coords_at_pos(slice, 6), (1, 0).into());
153+
154+ // Test with tabs.
155+ // Todo: account for tab stops.
156+ let text = Rope::from("\tHello\n");
157+ let slice = text.slice(..);
158+ assert_eq!(coords_at_pos(slice, 0), (0, 0).into());
159+ assert_eq!(coords_at_pos(slice, 1), (0, 1).into());
160+ assert_eq!(coords_at_pos(slice, 2), (0, 2).into());
160161 }
161162
162163 #[test]
@@ -172,21 +173,20 @@ mod test {
172173 assert_eq!(pos_at_coords(slice, (1, 4).into(), false), 10); // position on d
173174
174175 // Test with wide characters.
176+ // TODO: account for character width.
175177 let text = Rope::from("今日はいい\n");
176178 let slice = text.slice(..);
177179 assert_eq!(pos_at_coords(slice, (0, 0).into(), false), 0);
178− assert_eq!(pos_at_coords(slice, (0, 1).into(), false), 0);
179− assert_eq!(pos_at_coords(slice, (0, 2).into(), false), 1);
180− assert_eq!(pos_at_coords(slice, (0, 3).into(), false), 1);
181− assert_eq!(pos_at_coords(slice, (0, 4).into(), false), 2);
182− assert_eq!(pos_at_coords(slice, (0, 6).into(), false), 3);
183− assert_eq!(pos_at_coords(slice, (0, 8).into(), false), 4);
184− assert_eq!(pos_at_coords(slice, (0, 10).into(), false), 5);
185− assert_eq!(pos_at_coords(slice, (0, 11).into(), false), 6);
186− assert_eq!(pos_at_coords(slice, (0, 11).into(), true), 5);
180+ assert_eq!(pos_at_coords(slice, (0, 1).into(), false), 1);
181+ assert_eq!(pos_at_coords(slice, (0, 2).into(), false), 2);
182+ assert_eq!(pos_at_coords(slice, (0, 3).into(), false), 3);
183+ assert_eq!(pos_at_coords(slice, (0, 4).into(), false), 4);
184+ assert_eq!(pos_at_coords(slice, (0, 5).into(), false), 5);
185+ assert_eq!(pos_at_coords(slice, (0, 6).into(), false), 6);
186+ assert_eq!(pos_at_coords(slice, (0, 6).into(), true), 5);
187187 assert_eq!(pos_at_coords(slice, (1, 0).into(), false), 6);
188188
189− // test with grapheme clusters
189+ // Test with grapheme clusters.
190190 let text = Rope::from("a̐éö̲\r\n");
191191 let slice = text.slice(..);
192192 assert_eq!(pos_at_coords(slice, (0, 0).into(), false), 0);
@@ -197,15 +197,24 @@ mod test {
197197 assert_eq!(pos_at_coords(slice, (0, 4).into(), true), 7);
198198 assert_eq!(pos_at_coords(slice, (1, 0).into(), false), 9);
199199
200+ // Test with wide-character grapheme clusters.
201+ // TODO: account for character width.
200202 let text = Rope::from("किमपि");
201203 // 2 - 1 - 2 codepoints
202204 // TODO: delete handling as per https://news.ycombinator.com/item?id=20058454
203205 let slice = text.slice(..);
204206 assert_eq!(pos_at_coords(slice, (0, 0).into(), false), 0);
205− assert_eq!(pos_at_coords(slice, (0, 1).into(), false), 0);
207+ assert_eq!(pos_at_coords(slice, (0, 1).into(), false), 2);
208+ assert_eq!(pos_at_coords(slice, (0, 2).into(), false), 3);
209+ assert_eq!(pos_at_coords(slice, (0, 3).into(), false), 5);
210+ assert_eq!(pos_at_coords(slice, (0, 3).into(), true), 5);
211+
212+ // Test with tabs.
213+ // Todo: account for tab stops.
214+ let text = Rope::from("\tHello\n");
215+ let slice = text.slice(..);
216+ assert_eq!(pos_at_coords(slice, (0, 0).into(), false), 0);
217+ assert_eq!(pos_at_coords(slice, (0, 1).into(), false), 1);
206218 assert_eq!(pos_at_coords(slice, (0, 2).into(), false), 2);
207− assert_eq!(pos_at_coords(slice, (0, 3).into(), false), 3);
208− assert_eq!(pos_at_coords(slice, (0, 4).into(), false), 3);
209− assert_eq!(pos_at_coords(slice, (0, 5).into(), false), 5); // eol
210219 }
211220}