1use crate::display_map::{BlockContext, ToDisplayPoint};
2
3use super::{
4 DisplayPoint, Editor, EditorMode, EditorSettings, EditorSnapshot, EditorStyle, Input, Scroll,
5 Select, SelectPhase, SoftWrap, MAX_LINE_LEN,
6};
7use crate::ToPoint;
8use clock::ReplicaId;
9use gpui::{
10 color::Color,
11 geometry::{
12 rect::RectF,
13 vector::{vec2f, Vector2F},
14 PathBuilder,
15 },
16 json::{self, ToJson},
17 keymap::Keystroke,
18 text_layout::{self, RunStyle, TextLayoutCache},
19 AppContext, Axis, Border, Element, ElementBox, Event, EventContext, FontCache, LayoutContext,
20 MutableAppContext, PaintContext, Quad, Scene, SizeConstraint, ViewContext, WeakViewHandle,
21};
22use json::json;
23use language::Chunk;
24use smallvec::SmallVec;
25use std::{
26 cmp::{self, Ordering},
27 collections::{BTreeMap, HashMap},
28 fmt::Write,
29 ops::Range,
30};
31
32pub struct EditorElement {
33 view: WeakViewHandle<Editor>,
34 settings: EditorSettings,
35}
36
37impl EditorElement {
38 pub fn new(view: WeakViewHandle<Editor>, settings: EditorSettings) -> Self {
39 Self { view, settings }
40 }
41
42 fn view<'a>(&self, cx: &'a AppContext) -> &'a Editor {
43 self.view.upgrade(cx).unwrap().read(cx)
44 }
45
46 fn update_view<F, T>(&self, cx: &mut MutableAppContext, f: F) -> T
47 where
48 F: FnOnce(&mut Editor, &mut ViewContext<Editor>) -> T,
49 {
50 self.view.upgrade(cx).unwrap().update(cx, f)
51 }
52
53 fn snapshot(&self, cx: &mut MutableAppContext) -> EditorSnapshot {
54 self.update_view(cx, |view, cx| view.snapshot(cx))
55 }
56
57 fn mouse_down(
58 &self,
59 position: Vector2F,
60 alt: bool,
61 shift: bool,
62 mut click_count: usize,
63 layout: &mut LayoutState,
64 paint: &mut PaintState,
65 cx: &mut EventContext,
66 ) -> bool {
67 if paint.gutter_bounds.contains_point(position) {
68 click_count = 3; // Simulate triple-click when clicking the gutter to select lines
69 } else if !paint.text_bounds.contains_point(position) {
70 return false;
71 }
72
73 let snapshot = self.snapshot(cx.app);
74 let (position, overshoot) = paint.point_for_position(&snapshot, layout, position);
75
76 if shift && alt {
77 cx.dispatch_action(Select(SelectPhase::BeginColumnar {
78 position,
79 overshoot,
80 }));
81 } else if shift {
82 cx.dispatch_action(Select(SelectPhase::Extend {
83 position,
84 click_count,
85 }));
86 } else {
87 cx.dispatch_action(Select(SelectPhase::Begin {
88 position,
89 add: alt,
90 click_count,
91 }));
92 }
93
94 true
95 }
96
97 fn mouse_up(&self, _position: Vector2F, cx: &mut EventContext) -> bool {
98 if self.view(cx.app.as_ref()).is_selecting() {
99 cx.dispatch_action(Select(SelectPhase::End));
100 true
101 } else {
102 false
103 }
104 }
105
106 fn mouse_dragged(
107 &self,
108 position: Vector2F,
109 layout: &mut LayoutState,
110 paint: &mut PaintState,
111 cx: &mut EventContext,
112 ) -> bool {
113 let view = self.view(cx.app.as_ref());
114
115 if view.is_selecting() {
116 let rect = paint.text_bounds;
117 let mut scroll_delta = Vector2F::zero();
118
119 let vertical_margin = layout.line_height.min(rect.height() / 3.0);
120 let top = rect.origin_y() + vertical_margin;
121 let bottom = rect.lower_left().y() - vertical_margin;
122 if position.y() < top {
123 scroll_delta.set_y(-scale_vertical_mouse_autoscroll_delta(top - position.y()))
124 }
125 if position.y() > bottom {
126 scroll_delta.set_y(scale_vertical_mouse_autoscroll_delta(position.y() - bottom))
127 }
128
129 let horizontal_margin = layout.line_height.min(rect.width() / 3.0);
130 let left = rect.origin_x() + horizontal_margin;
131 let right = rect.upper_right().x() - horizontal_margin;
132 if position.x() < left {
133 scroll_delta.set_x(-scale_horizontal_mouse_autoscroll_delta(
134 left - position.x(),
135 ))
136 }
137 if position.x() > right {
138 scroll_delta.set_x(scale_horizontal_mouse_autoscroll_delta(
139 position.x() - right,
140 ))
141 }
142
143 let font_cache = cx.font_cache.clone();
144 let text_layout_cache = cx.text_layout_cache.clone();
145 let snapshot = self.snapshot(cx.app);
146 let (position, overshoot) = paint.point_for_position(&snapshot, layout, position);
147
148 cx.dispatch_action(Select(SelectPhase::Update {
149 position,
150 overshoot,
151 scroll_position: (snapshot.scroll_position() + scroll_delta).clamp(
152 Vector2F::zero(),
153 layout.scroll_max(&font_cache, &text_layout_cache),
154 ),
155 }));
156 true
157 } else {
158 false
159 }
160 }
161
162 fn key_down(&self, chars: &str, keystroke: &Keystroke, cx: &mut EventContext) -> bool {
163 let view = self.view.upgrade(cx.app).unwrap();
164
165 if view.is_focused(cx.app) {
166 if chars.is_empty() {
167 false
168 } else {
169 if chars.chars().any(|c| c.is_control()) || keystroke.cmd || keystroke.ctrl {
170 false
171 } else {
172 cx.dispatch_action(Input(chars.to_string()));
173 true
174 }
175 }
176 } else {
177 false
178 }
179 }
180
181 fn scroll(
182 &self,
183 position: Vector2F,
184 mut delta: Vector2F,
185 precise: bool,
186 layout: &mut LayoutState,
187 paint: &mut PaintState,
188 cx: &mut EventContext,
189 ) -> bool {
190 if !paint.bounds.contains_point(position) {
191 return false;
192 }
193
194 let snapshot = self.snapshot(cx.app);
195 let font_cache = &cx.font_cache;
196 let layout_cache = &cx.text_layout_cache;
197 let max_glyph_width = layout.em_width;
198 if !precise {
199 delta *= vec2f(max_glyph_width, layout.line_height);
200 }
201
202 let scroll_position = snapshot.scroll_position();
203 let x = (scroll_position.x() * max_glyph_width - delta.x()) / max_glyph_width;
204 let y = (scroll_position.y() * layout.line_height - delta.y()) / layout.line_height;
205 let scroll_position = vec2f(x, y).clamp(
206 Vector2F::zero(),
207 layout.scroll_max(font_cache, layout_cache),
208 );
209
210 cx.dispatch_action(Scroll(scroll_position));
211
212 true
213 }
214
215 fn paint_background(
216 &self,
217 gutter_bounds: RectF,
218 text_bounds: RectF,
219 layout: &LayoutState,
220 cx: &mut PaintContext,
221 ) {
222 let bounds = gutter_bounds.union_rect(text_bounds);
223 let scroll_top = layout.snapshot.scroll_position().y() * layout.line_height;
224 let editor = self.view(cx.app);
225 let style = &self.settings.style;
226 cx.scene.push_quad(Quad {
227 bounds: gutter_bounds,
228 background: Some(style.gutter_background),
229 border: Border::new(0., Color::transparent_black()),
230 corner_radius: 0.,
231 });
232 cx.scene.push_quad(Quad {
233 bounds: text_bounds,
234 background: Some(style.background),
235 border: Border::new(0., Color::transparent_black()),
236 corner_radius: 0.,
237 });
238
239 if let EditorMode::Full = editor.mode {
240 let mut active_rows = layout.active_rows.iter().peekable();
241 while let Some((start_row, contains_non_empty_selection)) = active_rows.next() {
242 let mut end_row = *start_row;
243 while active_rows.peek().map_or(false, |r| {
244 *r.0 == end_row + 1 && r.1 == contains_non_empty_selection
245 }) {
246 active_rows.next().unwrap();
247 end_row += 1;
248 }
249
250 if !contains_non_empty_selection {
251 let origin = vec2f(
252 bounds.origin_x(),
253 bounds.origin_y() + (layout.line_height * *start_row as f32) - scroll_top,
254 );
255 let size = vec2f(
256 bounds.width(),
257 layout.line_height * (end_row - start_row + 1) as f32,
258 );
259 cx.scene.push_quad(Quad {
260 bounds: RectF::new(origin, size),
261 background: Some(style.active_line_background),
262 border: Border::default(),
263 corner_radius: 0.,
264 });
265 }
266 }
267
268 if let Some(highlighted_row) = layout.highlighted_row {
269 let origin = vec2f(
270 bounds.origin_x(),
271 bounds.origin_y() + (layout.line_height * highlighted_row as f32) - scroll_top,
272 );
273 let size = vec2f(bounds.width(), layout.line_height);
274 cx.scene.push_quad(Quad {
275 bounds: RectF::new(origin, size),
276 background: Some(style.highlighted_line_background),
277 border: Border::default(),
278 corner_radius: 0.,
279 });
280 }
281 }
282 }
283
284 fn paint_gutter(
285 &mut self,
286 bounds: RectF,
287 visible_bounds: RectF,
288 layout: &LayoutState,
289 cx: &mut PaintContext,
290 ) {
291 let scroll_top = layout.snapshot.scroll_position().y() * layout.line_height;
292 for (ix, line) in layout.line_number_layouts.iter().enumerate() {
293 if let Some(line) = line {
294 let line_origin = bounds.origin()
295 + vec2f(
296 bounds.width() - line.width() - layout.gutter_padding,
297 ix as f32 * layout.line_height - (scroll_top % layout.line_height),
298 );
299 line.paint(line_origin, visible_bounds, layout.line_height, cx);
300 }
301 }
302 }
303
304 fn paint_text(
305 &mut self,
306 bounds: RectF,
307 visible_bounds: RectF,
308 layout: &LayoutState,
309 cx: &mut PaintContext,
310 ) {
311 let view = self.view(cx.app);
312 let style = &self.settings.style;
313 let local_replica_id = view.replica_id(cx);
314 let scroll_position = layout.snapshot.scroll_position();
315 let start_row = scroll_position.y() as u32;
316 let scroll_top = scroll_position.y() * layout.line_height;
317 let end_row = ((scroll_top + bounds.height()) / layout.line_height).ceil() as u32 + 1; // Add 1 to ensure selections bleed off screen
318 let max_glyph_width = layout.em_width;
319 let scroll_left = scroll_position.x() * max_glyph_width;
320
321 cx.scene.push_layer(Some(bounds));
322
323 // Draw selections
324 let corner_radius = 2.5;
325 let mut cursors = SmallVec::<[Cursor; 32]>::new();
326
327 let content_origin = bounds.origin() + layout.text_offset;
328
329 for (replica_id, selections) in &layout.selections {
330 let style = style.replica_selection_style(*replica_id);
331
332 for selection in selections {
333 if selection.start != selection.end {
334 let row_range = if selection.end.column() == 0 {
335 cmp::max(selection.start.row(), start_row)
336 ..cmp::min(selection.end.row(), end_row)
337 } else {
338 cmp::max(selection.start.row(), start_row)
339 ..cmp::min(selection.end.row() + 1, end_row)
340 };
341
342 let selection = Selection {
343 color: style.selection,
344 line_height: layout.line_height,
345 start_y: content_origin.y() + row_range.start as f32 * layout.line_height
346 - scroll_top,
347 lines: row_range
348 .into_iter()
349 .map(|row| {
350 let line_layout = &layout.line_layouts[(row - start_row) as usize];
351 SelectionLine {
352 start_x: if row == selection.start.row() {
353 content_origin.x()
354 + line_layout
355 .x_for_index(selection.start.column() as usize)
356 - scroll_left
357 } else {
358 content_origin.x() - scroll_left
359 },
360 end_x: if row == selection.end.row() {
361 content_origin.x()
362 + line_layout
363 .x_for_index(selection.end.column() as usize)
364 - scroll_left
365 } else {
366 content_origin.x()
367 + line_layout.width()
368 + corner_radius * 2.0
369 - scroll_left
370 },
371 }
372 })
373 .collect(),
374 };
375
376 selection.paint(bounds, cx.scene);
377 }
378
379 if view.show_local_cursors() || *replica_id != local_replica_id {
380 let cursor_position = selection.head();
381 if (start_row..end_row).contains(&cursor_position.row()) {
382 let cursor_row_layout =
383 &layout.line_layouts[(cursor_position.row() - start_row) as usize];
384 let x = cursor_row_layout.x_for_index(cursor_position.column() as usize)
385 - scroll_left;
386 let y = cursor_position.row() as f32 * layout.line_height - scroll_top;
387 cursors.push(Cursor {
388 color: style.cursor,
389 origin: content_origin + vec2f(x, y),
390 line_height: layout.line_height,
391 });
392 }
393 }
394 }
395 }
396
397 if let Some(visible_text_bounds) = bounds.intersection(visible_bounds) {
398 // Draw glyphs
399 for (ix, line) in layout.line_layouts.iter().enumerate() {
400 let row = start_row + ix as u32;
401 line.paint(
402 content_origin
403 + vec2f(-scroll_left, row as f32 * layout.line_height - scroll_top),
404 visible_text_bounds,
405 layout.line_height,
406 cx,
407 );
408 }
409 }
410
411 cx.scene.push_layer(Some(bounds));
412 for cursor in cursors {
413 cursor.paint(cx);
414 }
415 cx.scene.pop_layer();
416
417 cx.scene.pop_layer();
418 }
419
420 fn paint_blocks(
421 &mut self,
422 text_bounds: RectF,
423 visible_bounds: RectF,
424 layout: &mut LayoutState,
425 cx: &mut PaintContext,
426 ) {
427 let scroll_position = layout.snapshot.scroll_position();
428 let scroll_left = scroll_position.x() * layout.em_width;
429 let scroll_top = scroll_position.y() * layout.line_height;
430
431 for (row, element) in &mut layout.blocks {
432 let origin = text_bounds.origin()
433 + vec2f(-scroll_left, *row as f32 * layout.line_height - scroll_top);
434 element.paint(origin, visible_bounds, cx);
435 }
436 }
437
438 fn max_line_number_width(&self, snapshot: &EditorSnapshot, cx: &LayoutContext) -> f32 {
439 let digit_count = (snapshot.buffer_row_count() as f32).log10().floor() as usize + 1;
440 let style = &self.settings.style;
441
442 cx.text_layout_cache
443 .layout_str(
444 "1".repeat(digit_count).as_str(),
445 style.text.font_size,
446 &[(
447 digit_count,
448 RunStyle {
449 font_id: style.text.font_id,
450 color: Color::black(),
451 underline: None,
452 },
453 )],
454 )
455 .width()
456 }
457
458 fn layout_rows(
459 &self,
460 rows: Range<u32>,
461 active_rows: &BTreeMap<u32, bool>,
462 snapshot: &EditorSnapshot,
463 cx: &LayoutContext,
464 ) -> Vec<Option<text_layout::Line>> {
465 let style = &self.settings.style;
466 let include_line_numbers = snapshot.mode == EditorMode::Full;
467 let mut line_number_layouts = Vec::with_capacity(rows.len());
468 let mut line_number = String::new();
469 for (ix, row) in snapshot
470 .buffer_rows(rows.start)
471 .take((rows.end - rows.start) as usize)
472 .enumerate()
473 {
474 let display_row = rows.start + ix as u32;
475 let color = if active_rows.contains_key(&display_row) {
476 style.line_number_active
477 } else {
478 style.line_number
479 };
480 if let Some(buffer_row) = row {
481 if include_line_numbers {
482 line_number.clear();
483 write!(&mut line_number, "{}", buffer_row + 1).unwrap();
484 line_number_layouts.push(Some(cx.text_layout_cache.layout_str(
485 &line_number,
486 style.text.font_size,
487 &[(
488 line_number.len(),
489 RunStyle {
490 font_id: style.text.font_id,
491 color,
492 underline: None,
493 },
494 )],
495 )));
496 }
497 } else {
498 line_number_layouts.push(None);
499 }
500 }
501
502 line_number_layouts
503 }
504
505 fn layout_lines(
506 &mut self,
507 mut rows: Range<u32>,
508 snapshot: &mut EditorSnapshot,
509 cx: &LayoutContext,
510 ) -> Vec<text_layout::Line> {
511 rows.end = cmp::min(rows.end, snapshot.max_point().row() + 1);
512 if rows.start >= rows.end {
513 return Vec::new();
514 }
515
516 // When the editor is empty and unfocused, then show the placeholder.
517 if snapshot.is_empty() && !snapshot.is_focused() {
518 let placeholder_style = self.settings.style.placeholder_text();
519 let placeholder_text = snapshot.placeholder_text();
520 let placeholder_lines = placeholder_text
521 .as_ref()
522 .map_or("", AsRef::as_ref)
523 .split('\n')
524 .skip(rows.start as usize)
525 .take(rows.len());
526 return placeholder_lines
527 .map(|line| {
528 cx.text_layout_cache.layout_str(
529 line,
530 placeholder_style.font_size,
531 &[(
532 line.len(),
533 RunStyle {
534 font_id: placeholder_style.font_id,
535 color: placeholder_style.color,
536 underline: None,
537 },
538 )],
539 )
540 })
541 .collect();
542 }
543
544 let style = &self.settings.style;
545 let mut prev_font_properties = style.text.font_properties.clone();
546 let mut prev_font_id = style.text.font_id;
547
548 let mut layouts = Vec::with_capacity(rows.len());
549 let mut line = String::new();
550 let mut styles = Vec::new();
551 let mut row = rows.start;
552 let mut line_exceeded_max_len = false;
553 let chunks = snapshot.chunks(rows.clone(), Some(&style.syntax));
554
555 let newline_chunk = Chunk {
556 text: "\n",
557 ..Default::default()
558 };
559 'outer: for chunk in chunks.chain([newline_chunk]) {
560 for (ix, mut line_chunk) in chunk.text.split('\n').enumerate() {
561 if ix > 0 {
562 layouts.push(cx.text_layout_cache.layout_str(
563 &line,
564 style.text.font_size,
565 &styles,
566 ));
567 line.clear();
568 styles.clear();
569 row += 1;
570 line_exceeded_max_len = false;
571 if row == rows.end {
572 break 'outer;
573 }
574 }
575
576 if !line_chunk.is_empty() && !line_exceeded_max_len {
577 let highlight_style =
578 chunk.highlight_style.unwrap_or(style.text.clone().into());
579 // Avoid a lookup if the font properties match the previous ones.
580 let font_id = if highlight_style.font_properties == prev_font_properties {
581 prev_font_id
582 } else {
583 cx.font_cache
584 .select_font(
585 style.text.font_family_id,
586 &highlight_style.font_properties,
587 )
588 .unwrap_or(style.text.font_id)
589 };
590
591 if line.len() + line_chunk.len() > MAX_LINE_LEN {
592 let mut chunk_len = MAX_LINE_LEN - line.len();
593 while !line_chunk.is_char_boundary(chunk_len) {
594 chunk_len -= 1;
595 }
596 line_chunk = &line_chunk[..chunk_len];
597 line_exceeded_max_len = true;
598 }
599
600 let underline = if let Some(severity) = chunk.diagnostic {
601 Some(super::diagnostic_style(severity, true, style).text)
602 } else {
603 highlight_style.underline
604 };
605
606 line.push_str(line_chunk);
607 styles.push((
608 line_chunk.len(),
609 RunStyle {
610 font_id,
611 color: highlight_style.color,
612 underline,
613 },
614 ));
615 prev_font_id = font_id;
616 prev_font_properties = highlight_style.font_properties;
617 }
618 }
619 }
620
621 layouts
622 }
623
624 fn layout_blocks(
625 &mut self,
626 rows: Range<u32>,
627 snapshot: &EditorSnapshot,
628 text_width: f32,
629 line_height: f32,
630 style: &EditorStyle,
631 line_layouts: &[text_layout::Line],
632 cx: &mut LayoutContext,
633 ) -> Vec<(u32, ElementBox)> {
634 snapshot
635 .blocks_in_range(rows.clone())
636 .map(|(start_row, block)| {
637 let anchor_row = block
638 .position()
639 .to_point(&snapshot.buffer_snapshot)
640 .to_display_point(snapshot)
641 .row();
642
643 let anchor_x = if rows.contains(&anchor_row) {
644 line_layouts[(anchor_row - rows.start) as usize]
645 .x_for_index(block.column() as usize)
646 } else {
647 layout_line(anchor_row, snapshot, style, cx.text_layout_cache)
648 .x_for_index(block.column() as usize)
649 };
650
651 let mut element = block.render(&BlockContext { cx, anchor_x });
652 element.layout(
653 SizeConstraint {
654 min: Vector2F::zero(),
655 max: vec2f(text_width, block.height() as f32 * line_height),
656 },
657 cx,
658 );
659 (start_row, element)
660 })
661 .collect()
662 }
663}
664
665impl Element for EditorElement {
666 type LayoutState = Option<LayoutState>;
667 type PaintState = Option<PaintState>;
668
669 fn layout(
670 &mut self,
671 constraint: SizeConstraint,
672 cx: &mut LayoutContext,
673 ) -> (Vector2F, Self::LayoutState) {
674 let mut size = constraint.max;
675 if size.x().is_infinite() {
676 unimplemented!("we don't yet handle an infinite width constraint on buffer elements");
677 }
678
679 let snapshot = self.snapshot(cx.app);
680 let style = self.settings.style.clone();
681 let line_height = style.text.line_height(cx.font_cache);
682
683 let gutter_padding;
684 let gutter_width;
685 if snapshot.mode == EditorMode::Full {
686 gutter_padding = style.text.em_width(cx.font_cache);
687 gutter_width = self.max_line_number_width(&snapshot, cx) + gutter_padding * 2.0;
688 } else {
689 gutter_padding = 0.0;
690 gutter_width = 0.0
691 };
692
693 let text_width = size.x() - gutter_width;
694 let text_offset = vec2f(-style.text.descent(cx.font_cache), 0.);
695 let em_width = style.text.em_width(cx.font_cache);
696 let em_advance = style.text.em_advance(cx.font_cache);
697 let overscroll = vec2f(em_width, 0.);
698 let wrap_width = match self.settings.soft_wrap {
699 SoftWrap::None => None,
700 SoftWrap::EditorWidth => Some(text_width - text_offset.x() - overscroll.x() - em_width),
701 SoftWrap::Column(column) => Some(column as f32 * em_advance),
702 };
703 let snapshot = self.update_view(cx.app, |view, cx| {
704 if view.set_wrap_width(wrap_width, cx) {
705 view.snapshot(cx)
706 } else {
707 snapshot
708 }
709 });
710
711 let scroll_height = (snapshot.max_point().row() + 1) as f32 * line_height;
712 if let EditorMode::AutoHeight { max_lines } = snapshot.mode {
713 size.set_y(
714 scroll_height
715 .min(constraint.max_along(Axis::Vertical))
716 .max(constraint.min_along(Axis::Vertical))
717 .min(line_height * max_lines as f32),
718 )
719 } else if size.y().is_infinite() {
720 size.set_y(scroll_height);
721 }
722 let gutter_size = vec2f(gutter_width, size.y());
723 let text_size = vec2f(text_width, size.y());
724
725 let (autoscroll_horizontally, mut snapshot) = self.update_view(cx.app, |view, cx| {
726 let autoscroll_horizontally = view.autoscroll_vertically(size.y(), line_height, cx);
727 let snapshot = view.snapshot(cx);
728 (autoscroll_horizontally, snapshot)
729 });
730
731 let scroll_position = snapshot.scroll_position();
732 let start_row = scroll_position.y() as u32;
733 let scroll_top = scroll_position.y() * line_height;
734 let end_row = ((scroll_top + size.y()) / line_height).ceil() as u32 + 1; // Add 1 to ensure selections bleed off screen
735
736 let selections = HashMap::new();
737 let active_rows = BTreeMap::new();
738 let mut highlighted_row = None;
739 self.update_view(cx.app, |view, cx| {
740 highlighted_row = view.highlighted_row();
741
742 // TODO: Get this working with editors owning their own selections
743
744 // for selection_set_id in view.active_selection_sets(cx).collect::<Vec<_>>() {
745 // let replica_selections = view.intersecting_selections(
746 // selection_set_id,
747 // DisplayPoint::new(start_row, 0)..DisplayPoint::new(end_row, 0),
748 // cx,
749 // );
750 // for selection in &replica_selections {
751 // if selection_set_id == view.selection_set_id {
752 // let is_empty = selection.start == selection.end;
753 // let selection_start = snapshot.prev_row_boundary(selection.start).0;
754 // let selection_end = snapshot.next_row_boundary(selection.end).0;
755 // for row in cmp::max(selection_start.row(), start_row)
756 // ..=cmp::min(selection_end.row(), end_row)
757 // {
758 // let contains_non_empty_selection =
759 // active_rows.entry(row).or_insert(!is_empty);
760 // *contains_non_empty_selection |= !is_empty;
761 // }
762 // }
763 // }
764
765 // selections.insert(selection_set_id.replica_id, replica_selections);
766 // }
767 });
768
769 let line_number_layouts = self.layout_rows(start_row..end_row, &active_rows, &snapshot, cx);
770
771 let mut max_visible_line_width = 0.0;
772 let line_layouts = self.layout_lines(start_row..end_row, &mut snapshot, cx);
773 for line in &line_layouts {
774 if line.width() > max_visible_line_width {
775 max_visible_line_width = line.width();
776 }
777 }
778
779 let blocks = self.layout_blocks(
780 start_row..end_row,
781 &snapshot,
782 text_size.x(),
783 line_height,
784 &style,
785 &line_layouts,
786 cx,
787 );
788
789 let mut layout = LayoutState {
790 size,
791 gutter_size,
792 gutter_padding,
793 text_size,
794 overscroll,
795 text_offset,
796 snapshot,
797 style: self.settings.style.clone(),
798 active_rows,
799 highlighted_row,
800 line_layouts,
801 line_number_layouts,
802 blocks,
803 line_height,
804 em_width,
805 em_advance,
806 selections,
807 max_visible_line_width,
808 };
809
810 let scroll_max = layout.scroll_max(cx.font_cache, cx.text_layout_cache).x();
811 let scroll_width = layout.scroll_width(cx.text_layout_cache);
812 let max_glyph_width = style.text.em_width(&cx.font_cache);
813 self.update_view(cx.app, |view, cx| {
814 let clamped = view.clamp_scroll_left(scroll_max);
815 let autoscrolled;
816 if autoscroll_horizontally {
817 autoscrolled = view.autoscroll_horizontally(
818 start_row,
819 layout.text_size.x(),
820 scroll_width,
821 max_glyph_width,
822 &layout.line_layouts,
823 cx,
824 );
825 } else {
826 autoscrolled = false;
827 }
828
829 if clamped || autoscrolled {
830 layout.snapshot = view.snapshot(cx);
831 }
832 });
833
834 (size, Some(layout))
835 }
836
837 fn paint(
838 &mut self,
839 bounds: RectF,
840 visible_bounds: RectF,
841 layout: &mut Self::LayoutState,
842 cx: &mut PaintContext,
843 ) -> Self::PaintState {
844 if let Some(layout) = layout {
845 cx.scene.push_layer(Some(bounds));
846
847 let gutter_bounds = RectF::new(bounds.origin(), layout.gutter_size);
848 let text_bounds = RectF::new(
849 bounds.origin() + vec2f(layout.gutter_size.x(), 0.0),
850 layout.text_size,
851 );
852
853 self.paint_background(gutter_bounds, text_bounds, layout, cx);
854 if layout.gutter_size.x() > 0. {
855 self.paint_gutter(gutter_bounds, visible_bounds, layout, cx);
856 }
857 self.paint_text(text_bounds, visible_bounds, layout, cx);
858 self.paint_blocks(text_bounds, visible_bounds, layout, cx);
859
860 cx.scene.pop_layer();
861
862 Some(PaintState {
863 bounds,
864 gutter_bounds,
865 text_bounds,
866 })
867 } else {
868 None
869 }
870 }
871
872 fn dispatch_event(
873 &mut self,
874 event: &Event,
875 _: RectF,
876 layout: &mut Self::LayoutState,
877 paint: &mut Self::PaintState,
878 cx: &mut EventContext,
879 ) -> bool {
880 if let (Some(layout), Some(paint)) = (layout, paint) {
881 match event {
882 Event::LeftMouseDown {
883 position,
884 alt,
885 shift,
886 click_count,
887 ..
888 } => self.mouse_down(*position, *alt, *shift, *click_count, layout, paint, cx),
889 Event::LeftMouseUp { position } => self.mouse_up(*position, cx),
890 Event::LeftMouseDragged { position } => {
891 self.mouse_dragged(*position, layout, paint, cx)
892 }
893 Event::ScrollWheel {
894 position,
895 delta,
896 precise,
897 } => self.scroll(*position, *delta, *precise, layout, paint, cx),
898 Event::KeyDown {
899 chars, keystroke, ..
900 } => self.key_down(chars, keystroke, cx),
901 _ => false,
902 }
903 } else {
904 false
905 }
906 }
907
908 fn debug(
909 &self,
910 bounds: RectF,
911 _: &Self::LayoutState,
912 _: &Self::PaintState,
913 _: &gpui::DebugContext,
914 ) -> json::Value {
915 json!({
916 "type": "BufferElement",
917 "bounds": bounds.to_json()
918 })
919 }
920}
921
922pub struct LayoutState {
923 size: Vector2F,
924 gutter_size: Vector2F,
925 gutter_padding: f32,
926 text_size: Vector2F,
927 style: EditorStyle,
928 snapshot: EditorSnapshot,
929 active_rows: BTreeMap<u32, bool>,
930 highlighted_row: Option<u32>,
931 line_layouts: Vec<text_layout::Line>,
932 line_number_layouts: Vec<Option<text_layout::Line>>,
933 blocks: Vec<(u32, ElementBox)>,
934 line_height: f32,
935 em_width: f32,
936 em_advance: f32,
937 selections: HashMap<ReplicaId, Vec<text::Selection<DisplayPoint>>>,
938 overscroll: Vector2F,
939 text_offset: Vector2F,
940 max_visible_line_width: f32,
941}
942
943impl LayoutState {
944 fn scroll_width(&self, layout_cache: &TextLayoutCache) -> f32 {
945 let row = self.snapshot.longest_row();
946 let longest_line_width =
947 layout_line(row, &self.snapshot, &self.style, layout_cache).width();
948 longest_line_width.max(self.max_visible_line_width) + self.overscroll.x()
949 }
950
951 fn scroll_max(&self, font_cache: &FontCache, layout_cache: &TextLayoutCache) -> Vector2F {
952 let text_width = self.text_size.x();
953 let scroll_width = self.scroll_width(layout_cache);
954 let em_width = self.style.text.em_width(font_cache);
955 let max_row = self.snapshot.max_point().row();
956
957 vec2f(
958 ((scroll_width - text_width) / em_width).max(0.0),
959 max_row.saturating_sub(1) as f32,
960 )
961 }
962}
963
964fn layout_line(
965 row: u32,
966 snapshot: &EditorSnapshot,
967 style: &EditorStyle,
968 layout_cache: &TextLayoutCache,
969) -> text_layout::Line {
970 let mut line = snapshot.line(row);
971
972 if line.len() > MAX_LINE_LEN {
973 let mut len = MAX_LINE_LEN;
974 while !line.is_char_boundary(len) {
975 len -= 1;
976 }
977 line.truncate(len);
978 }
979
980 layout_cache.layout_str(
981 &line,
982 style.text.font_size,
983 &[(
984 snapshot.line_len(row) as usize,
985 RunStyle {
986 font_id: style.text.font_id,
987 color: Color::black(),
988 underline: None,
989 },
990 )],
991 )
992}
993
994pub struct PaintState {
995 bounds: RectF,
996 gutter_bounds: RectF,
997 text_bounds: RectF,
998}
999
1000impl PaintState {
1001 fn point_for_position(
1002 &self,
1003 snapshot: &EditorSnapshot,
1004 layout: &LayoutState,
1005 position: Vector2F,
1006 ) -> (DisplayPoint, u32) {
1007 let scroll_position = snapshot.scroll_position();
1008 let position = position - self.text_bounds.origin();
1009 let y = position.y().max(0.0).min(layout.size.y());
1010 let row = ((y / layout.line_height) + scroll_position.y()) as u32;
1011 let row = cmp::min(row, snapshot.max_point().row());
1012 let line = &layout.line_layouts[(row - scroll_position.y() as u32) as usize];
1013 let x = position.x() + (scroll_position.x() * layout.em_width);
1014
1015 let column = if x >= 0.0 {
1016 line.index_for_x(x)
1017 .map(|ix| ix as u32)
1018 .unwrap_or_else(|| snapshot.line_len(row))
1019 } else {
1020 0
1021 };
1022 let overshoot = (0f32.max(x - line.width()) / layout.em_advance) as u32;
1023
1024 (DisplayPoint::new(row, column), overshoot)
1025 }
1026}
1027
1028struct Cursor {
1029 origin: Vector2F,
1030 line_height: f32,
1031 color: Color,
1032}
1033
1034impl Cursor {
1035 fn paint(&self, cx: &mut PaintContext) {
1036 cx.scene.push_quad(Quad {
1037 bounds: RectF::new(self.origin, vec2f(2.0, self.line_height)),
1038 background: Some(self.color),
1039 border: Border::new(0., Color::black()),
1040 corner_radius: 0.,
1041 });
1042 }
1043}
1044
1045#[derive(Debug)]
1046struct Selection {
1047 start_y: f32,
1048 line_height: f32,
1049 lines: Vec<SelectionLine>,
1050 color: Color,
1051}
1052
1053#[derive(Debug)]
1054struct SelectionLine {
1055 start_x: f32,
1056 end_x: f32,
1057}
1058
1059impl Selection {
1060 fn paint(&self, bounds: RectF, scene: &mut Scene) {
1061 if self.lines.len() >= 2 && self.lines[0].start_x > self.lines[1].end_x {
1062 self.paint_lines(self.start_y, &self.lines[0..1], bounds, scene);
1063 self.paint_lines(
1064 self.start_y + self.line_height,
1065 &self.lines[1..],
1066 bounds,
1067 scene,
1068 );
1069 } else {
1070 self.paint_lines(self.start_y, &self.lines, bounds, scene);
1071 }
1072 }
1073
1074 fn paint_lines(&self, start_y: f32, lines: &[SelectionLine], bounds: RectF, scene: &mut Scene) {
1075 if lines.is_empty() {
1076 return;
1077 }
1078
1079 let mut path = PathBuilder::new();
1080 let corner_radius = 0.15 * self.line_height;
1081 let first_line = lines.first().unwrap();
1082 let last_line = lines.last().unwrap();
1083
1084 let first_top_left = vec2f(first_line.start_x, start_y);
1085 let first_top_right = vec2f(first_line.end_x, start_y);
1086
1087 let curve_height = vec2f(0., corner_radius);
1088 let curve_width = |start_x: f32, end_x: f32| {
1089 let max = (end_x - start_x) / 2.;
1090 let width = if max < corner_radius {
1091 max
1092 } else {
1093 corner_radius
1094 };
1095
1096 vec2f(width, 0.)
1097 };
1098
1099 let top_curve_width = curve_width(first_line.start_x, first_line.end_x);
1100 path.reset(first_top_right - top_curve_width);
1101 path.curve_to(first_top_right + curve_height, first_top_right);
1102
1103 let mut iter = lines.iter().enumerate().peekable();
1104 while let Some((ix, line)) = iter.next() {
1105 let bottom_right = vec2f(line.end_x, start_y + (ix + 1) as f32 * self.line_height);
1106
1107 if let Some((_, next_line)) = iter.peek() {
1108 let next_top_right = vec2f(next_line.end_x, bottom_right.y());
1109
1110 match next_top_right.x().partial_cmp(&bottom_right.x()).unwrap() {
1111 Ordering::Equal => {
1112 path.line_to(bottom_right);
1113 }
1114 Ordering::Less => {
1115 let curve_width = curve_width(next_top_right.x(), bottom_right.x());
1116 path.line_to(bottom_right - curve_height);
1117 path.curve_to(bottom_right - curve_width, bottom_right);
1118 path.line_to(next_top_right + curve_width);
1119 path.curve_to(next_top_right + curve_height, next_top_right);
1120 }
1121 Ordering::Greater => {
1122 let curve_width = curve_width(bottom_right.x(), next_top_right.x());
1123 path.line_to(bottom_right - curve_height);
1124 path.curve_to(bottom_right + curve_width, bottom_right);
1125 path.line_to(next_top_right - curve_width);
1126 path.curve_to(next_top_right + curve_height, next_top_right);
1127 }
1128 }
1129 } else {
1130 let curve_width = curve_width(line.start_x, line.end_x);
1131 path.line_to(bottom_right - curve_height);
1132 path.curve_to(bottom_right - curve_width, bottom_right);
1133
1134 let bottom_left = vec2f(line.start_x, bottom_right.y());
1135 path.line_to(bottom_left + curve_width);
1136 path.curve_to(bottom_left - curve_height, bottom_left);
1137 }
1138 }
1139
1140 if first_line.start_x > last_line.start_x {
1141 let curve_width = curve_width(last_line.start_x, first_line.start_x);
1142 let second_top_left = vec2f(last_line.start_x, start_y + self.line_height);
1143 path.line_to(second_top_left + curve_height);
1144 path.curve_to(second_top_left + curve_width, second_top_left);
1145 let first_bottom_left = vec2f(first_line.start_x, second_top_left.y());
1146 path.line_to(first_bottom_left - curve_width);
1147 path.curve_to(first_bottom_left - curve_height, first_bottom_left);
1148 }
1149
1150 path.line_to(first_top_left + curve_height);
1151 path.curve_to(first_top_left + top_curve_width, first_top_left);
1152 path.line_to(first_top_right - top_curve_width);
1153
1154 scene.push_path(path.build(self.color, Some(bounds)));
1155 }
1156}
1157
1158fn scale_vertical_mouse_autoscroll_delta(delta: f32) -> f32 {
1159 delta.powf(1.5) / 100.0
1160}
1161
1162fn scale_horizontal_mouse_autoscroll_delta(delta: f32) -> f32 {
1163 delta.powf(1.2) / 300.0
1164}
1165
1166#[cfg(test)]
1167mod tests {
1168 use super::*;
1169 use crate::{Editor, EditorSettings, MultiBuffer};
1170 use util::test::sample_text;
1171
1172 #[gpui::test]
1173 fn test_layout_line_numbers(cx: &mut gpui::MutableAppContext) {
1174 let settings = EditorSettings::test(cx);
1175 let buffer = MultiBuffer::build_simple(&sample_text(6, 6, 'a'), cx);
1176 let (window_id, editor) = cx.add_window(Default::default(), |cx| {
1177 Editor::for_buffer(
1178 buffer,
1179 {
1180 let settings = settings.clone();
1181 move |_| settings.clone()
1182 },
1183 cx,
1184 )
1185 });
1186 let element = EditorElement::new(editor.downgrade(), settings);
1187
1188 let layouts = editor.update(cx, |editor, cx| {
1189 let snapshot = editor.snapshot(cx);
1190 let mut presenter = cx.build_presenter(window_id, 30.);
1191 let mut layout_cx = presenter.build_layout_context(false, cx);
1192 element.layout_rows(0..6, &Default::default(), &snapshot, &mut layout_cx)
1193 });
1194 assert_eq!(layouts.len(), 6);
1195 }
1196}