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