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 mut selections = HashMap::new();
737 let mut 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 for selection_set_id in view.active_selection_sets(cx).collect::<Vec<_>>() {
742 let replica_selections = view.intersecting_selections(
743 selection_set_id,
744 DisplayPoint::new(start_row, 0)..DisplayPoint::new(end_row, 0),
745 cx,
746 );
747 for selection in &replica_selections {
748 if selection_set_id == view.selection_set_id {
749 let is_empty = selection.start == selection.end;
750 let selection_start = snapshot.prev_row_boundary(selection.start).0;
751 let selection_end = snapshot.next_row_boundary(selection.end).0;
752 for row in cmp::max(selection_start.row(), start_row)
753 ..=cmp::min(selection_end.row(), end_row)
754 {
755 let contains_non_empty_selection =
756 active_rows.entry(row).or_insert(!is_empty);
757 *contains_non_empty_selection |= !is_empty;
758 }
759 }
760 }
761
762 selections.insert(selection_set_id.replica_id, replica_selections);
763 }
764 });
765
766 let line_number_layouts = self.layout_rows(start_row..end_row, &active_rows, &snapshot, cx);
767
768 let mut max_visible_line_width = 0.0;
769 let line_layouts = self.layout_lines(start_row..end_row, &mut snapshot, cx);
770 for line in &line_layouts {
771 if line.width() > max_visible_line_width {
772 max_visible_line_width = line.width();
773 }
774 }
775
776 let blocks = self.layout_blocks(
777 start_row..end_row,
778 &snapshot,
779 text_size.x(),
780 line_height,
781 &style,
782 &line_layouts,
783 cx,
784 );
785
786 let mut layout = LayoutState {
787 size,
788 gutter_size,
789 gutter_padding,
790 text_size,
791 overscroll,
792 text_offset,
793 snapshot,
794 style: self.settings.style.clone(),
795 active_rows,
796 highlighted_row,
797 line_layouts,
798 line_number_layouts,
799 blocks,
800 line_height,
801 em_width,
802 em_advance,
803 selections,
804 max_visible_line_width,
805 };
806
807 let scroll_max = layout.scroll_max(cx.font_cache, cx.text_layout_cache).x();
808 let scroll_width = layout.scroll_width(cx.text_layout_cache);
809 let max_glyph_width = style.text.em_width(&cx.font_cache);
810 self.update_view(cx.app, |view, cx| {
811 let clamped = view.clamp_scroll_left(scroll_max);
812 let autoscrolled;
813 if autoscroll_horizontally {
814 autoscrolled = view.autoscroll_horizontally(
815 start_row,
816 layout.text_size.x(),
817 scroll_width,
818 max_glyph_width,
819 &layout.line_layouts,
820 cx,
821 );
822 } else {
823 autoscrolled = false;
824 }
825
826 if clamped || autoscrolled {
827 layout.snapshot = view.snapshot(cx);
828 }
829 });
830
831 (size, Some(layout))
832 }
833
834 fn paint(
835 &mut self,
836 bounds: RectF,
837 visible_bounds: RectF,
838 layout: &mut Self::LayoutState,
839 cx: &mut PaintContext,
840 ) -> Self::PaintState {
841 if let Some(layout) = layout {
842 cx.scene.push_layer(Some(bounds));
843
844 let gutter_bounds = RectF::new(bounds.origin(), layout.gutter_size);
845 let text_bounds = RectF::new(
846 bounds.origin() + vec2f(layout.gutter_size.x(), 0.0),
847 layout.text_size,
848 );
849
850 self.paint_background(gutter_bounds, text_bounds, layout, cx);
851 if layout.gutter_size.x() > 0. {
852 self.paint_gutter(gutter_bounds, visible_bounds, layout, cx);
853 }
854 self.paint_text(text_bounds, visible_bounds, layout, cx);
855 self.paint_blocks(text_bounds, visible_bounds, layout, cx);
856
857 cx.scene.pop_layer();
858
859 Some(PaintState {
860 bounds,
861 gutter_bounds,
862 text_bounds,
863 })
864 } else {
865 None
866 }
867 }
868
869 fn dispatch_event(
870 &mut self,
871 event: &Event,
872 _: RectF,
873 layout: &mut Self::LayoutState,
874 paint: &mut Self::PaintState,
875 cx: &mut EventContext,
876 ) -> bool {
877 if let (Some(layout), Some(paint)) = (layout, paint) {
878 match event {
879 Event::LeftMouseDown {
880 position,
881 alt,
882 shift,
883 click_count,
884 ..
885 } => self.mouse_down(*position, *alt, *shift, *click_count, layout, paint, cx),
886 Event::LeftMouseUp { position } => self.mouse_up(*position, cx),
887 Event::LeftMouseDragged { position } => {
888 self.mouse_dragged(*position, layout, paint, cx)
889 }
890 Event::ScrollWheel {
891 position,
892 delta,
893 precise,
894 } => self.scroll(*position, *delta, *precise, layout, paint, cx),
895 Event::KeyDown {
896 chars, keystroke, ..
897 } => self.key_down(chars, keystroke, cx),
898 _ => false,
899 }
900 } else {
901 false
902 }
903 }
904
905 fn debug(
906 &self,
907 bounds: RectF,
908 _: &Self::LayoutState,
909 _: &Self::PaintState,
910 _: &gpui::DebugContext,
911 ) -> json::Value {
912 json!({
913 "type": "BufferElement",
914 "bounds": bounds.to_json()
915 })
916 }
917}
918
919pub struct LayoutState {
920 size: Vector2F,
921 gutter_size: Vector2F,
922 gutter_padding: f32,
923 text_size: Vector2F,
924 style: EditorStyle,
925 snapshot: EditorSnapshot,
926 active_rows: BTreeMap<u32, bool>,
927 highlighted_row: Option<u32>,
928 line_layouts: Vec<text_layout::Line>,
929 line_number_layouts: Vec<Option<text_layout::Line>>,
930 blocks: Vec<(u32, ElementBox)>,
931 line_height: f32,
932 em_width: f32,
933 em_advance: f32,
934 selections: HashMap<ReplicaId, Vec<text::Selection<DisplayPoint>>>,
935 overscroll: Vector2F,
936 text_offset: Vector2F,
937 max_visible_line_width: f32,
938}
939
940impl LayoutState {
941 fn scroll_width(&self, layout_cache: &TextLayoutCache) -> f32 {
942 let row = self.snapshot.longest_row();
943 let longest_line_width =
944 layout_line(row, &self.snapshot, &self.style, layout_cache).width();
945 longest_line_width.max(self.max_visible_line_width) + self.overscroll.x()
946 }
947
948 fn scroll_max(&self, font_cache: &FontCache, layout_cache: &TextLayoutCache) -> Vector2F {
949 let text_width = self.text_size.x();
950 let scroll_width = self.scroll_width(layout_cache);
951 let em_width = self.style.text.em_width(font_cache);
952 let max_row = self.snapshot.max_point().row();
953
954 vec2f(
955 ((scroll_width - text_width) / em_width).max(0.0),
956 max_row.saturating_sub(1) as f32,
957 )
958 }
959}
960
961fn layout_line(
962 row: u32,
963 snapshot: &EditorSnapshot,
964 style: &EditorStyle,
965 layout_cache: &TextLayoutCache,
966) -> text_layout::Line {
967 let mut line = snapshot.line(row);
968
969 if line.len() > MAX_LINE_LEN {
970 let mut len = MAX_LINE_LEN;
971 while !line.is_char_boundary(len) {
972 len -= 1;
973 }
974 line.truncate(len);
975 }
976
977 layout_cache.layout_str(
978 &line,
979 style.text.font_size,
980 &[(
981 snapshot.line_len(row) as usize,
982 RunStyle {
983 font_id: style.text.font_id,
984 color: Color::black(),
985 underline: None,
986 },
987 )],
988 )
989}
990
991pub struct PaintState {
992 bounds: RectF,
993 gutter_bounds: RectF,
994 text_bounds: RectF,
995}
996
997impl PaintState {
998 fn point_for_position(
999 &self,
1000 snapshot: &EditorSnapshot,
1001 layout: &LayoutState,
1002 position: Vector2F,
1003 ) -> (DisplayPoint, u32) {
1004 let scroll_position = snapshot.scroll_position();
1005 let position = position - self.text_bounds.origin();
1006 let y = position.y().max(0.0).min(layout.size.y());
1007 let row = ((y / layout.line_height) + scroll_position.y()) as u32;
1008 let row = cmp::min(row, snapshot.max_point().row());
1009 let line = &layout.line_layouts[(row - scroll_position.y() as u32) as usize];
1010 let x = position.x() + (scroll_position.x() * layout.em_width);
1011
1012 let column = if x >= 0.0 {
1013 line.index_for_x(x)
1014 .map(|ix| ix as u32)
1015 .unwrap_or_else(|| snapshot.line_len(row))
1016 } else {
1017 0
1018 };
1019 let overshoot = (0f32.max(x - line.width()) / layout.em_advance) as u32;
1020
1021 (DisplayPoint::new(row, column), overshoot)
1022 }
1023}
1024
1025struct Cursor {
1026 origin: Vector2F,
1027 line_height: f32,
1028 color: Color,
1029}
1030
1031impl Cursor {
1032 fn paint(&self, cx: &mut PaintContext) {
1033 cx.scene.push_quad(Quad {
1034 bounds: RectF::new(self.origin, vec2f(2.0, self.line_height)),
1035 background: Some(self.color),
1036 border: Border::new(0., Color::black()),
1037 corner_radius: 0.,
1038 });
1039 }
1040}
1041
1042#[derive(Debug)]
1043struct Selection {
1044 start_y: f32,
1045 line_height: f32,
1046 lines: Vec<SelectionLine>,
1047 color: Color,
1048}
1049
1050#[derive(Debug)]
1051struct SelectionLine {
1052 start_x: f32,
1053 end_x: f32,
1054}
1055
1056impl Selection {
1057 fn paint(&self, bounds: RectF, scene: &mut Scene) {
1058 if self.lines.len() >= 2 && self.lines[0].start_x > self.lines[1].end_x {
1059 self.paint_lines(self.start_y, &self.lines[0..1], bounds, scene);
1060 self.paint_lines(
1061 self.start_y + self.line_height,
1062 &self.lines[1..],
1063 bounds,
1064 scene,
1065 );
1066 } else {
1067 self.paint_lines(self.start_y, &self.lines, bounds, scene);
1068 }
1069 }
1070
1071 fn paint_lines(&self, start_y: f32, lines: &[SelectionLine], bounds: RectF, scene: &mut Scene) {
1072 if lines.is_empty() {
1073 return;
1074 }
1075
1076 let mut path = PathBuilder::new();
1077 let corner_radius = 0.15 * self.line_height;
1078 let first_line = lines.first().unwrap();
1079 let last_line = lines.last().unwrap();
1080
1081 let first_top_left = vec2f(first_line.start_x, start_y);
1082 let first_top_right = vec2f(first_line.end_x, start_y);
1083
1084 let curve_height = vec2f(0., corner_radius);
1085 let curve_width = |start_x: f32, end_x: f32| {
1086 let max = (end_x - start_x) / 2.;
1087 let width = if max < corner_radius {
1088 max
1089 } else {
1090 corner_radius
1091 };
1092
1093 vec2f(width, 0.)
1094 };
1095
1096 let top_curve_width = curve_width(first_line.start_x, first_line.end_x);
1097 path.reset(first_top_right - top_curve_width);
1098 path.curve_to(first_top_right + curve_height, first_top_right);
1099
1100 let mut iter = lines.iter().enumerate().peekable();
1101 while let Some((ix, line)) = iter.next() {
1102 let bottom_right = vec2f(line.end_x, start_y + (ix + 1) as f32 * self.line_height);
1103
1104 if let Some((_, next_line)) = iter.peek() {
1105 let next_top_right = vec2f(next_line.end_x, bottom_right.y());
1106
1107 match next_top_right.x().partial_cmp(&bottom_right.x()).unwrap() {
1108 Ordering::Equal => {
1109 path.line_to(bottom_right);
1110 }
1111 Ordering::Less => {
1112 let curve_width = curve_width(next_top_right.x(), bottom_right.x());
1113 path.line_to(bottom_right - curve_height);
1114 path.curve_to(bottom_right - curve_width, bottom_right);
1115 path.line_to(next_top_right + curve_width);
1116 path.curve_to(next_top_right + curve_height, next_top_right);
1117 }
1118 Ordering::Greater => {
1119 let curve_width = curve_width(bottom_right.x(), next_top_right.x());
1120 path.line_to(bottom_right - curve_height);
1121 path.curve_to(bottom_right + curve_width, bottom_right);
1122 path.line_to(next_top_right - curve_width);
1123 path.curve_to(next_top_right + curve_height, next_top_right);
1124 }
1125 }
1126 } else {
1127 let curve_width = curve_width(line.start_x, line.end_x);
1128 path.line_to(bottom_right - curve_height);
1129 path.curve_to(bottom_right - curve_width, bottom_right);
1130
1131 let bottom_left = vec2f(line.start_x, bottom_right.y());
1132 path.line_to(bottom_left + curve_width);
1133 path.curve_to(bottom_left - curve_height, bottom_left);
1134 }
1135 }
1136
1137 if first_line.start_x > last_line.start_x {
1138 let curve_width = curve_width(last_line.start_x, first_line.start_x);
1139 let second_top_left = vec2f(last_line.start_x, start_y + self.line_height);
1140 path.line_to(second_top_left + curve_height);
1141 path.curve_to(second_top_left + curve_width, second_top_left);
1142 let first_bottom_left = vec2f(first_line.start_x, second_top_left.y());
1143 path.line_to(first_bottom_left - curve_width);
1144 path.curve_to(first_bottom_left - curve_height, first_bottom_left);
1145 }
1146
1147 path.line_to(first_top_left + curve_height);
1148 path.curve_to(first_top_left + top_curve_width, first_top_left);
1149 path.line_to(first_top_right - top_curve_width);
1150
1151 scene.push_path(path.build(self.color, Some(bounds)));
1152 }
1153}
1154
1155fn scale_vertical_mouse_autoscroll_delta(delta: f32) -> f32 {
1156 delta.powf(1.5) / 100.0
1157}
1158
1159fn scale_horizontal_mouse_autoscroll_delta(delta: f32) -> f32 {
1160 delta.powf(1.2) / 300.0
1161}
1162
1163#[cfg(test)]
1164mod tests {
1165 use super::*;
1166 use crate::{Editor, EditorSettings, MultiBuffer};
1167 use util::test::sample_text;
1168
1169 #[gpui::test]
1170 fn test_layout_line_numbers(cx: &mut gpui::MutableAppContext) {
1171 let settings = EditorSettings::test(cx);
1172 let buffer = MultiBuffer::build_simple(&sample_text(6, 6, 'a'), cx);
1173 let (window_id, editor) = cx.add_window(Default::default(), |cx| {
1174 Editor::for_buffer(
1175 buffer,
1176 {
1177 let settings = settings.clone();
1178 move |_| settings.clone()
1179 },
1180 cx,
1181 )
1182 });
1183 let element = EditorElement::new(editor.downgrade(), settings);
1184
1185 let layouts = editor.update(cx, |editor, cx| {
1186 let snapshot = editor.snapshot(cx);
1187 let mut presenter = cx.build_presenter(window_id, 30.);
1188 let mut layout_cx = presenter.build_layout_context(false, cx);
1189 element.layout_rows(0..6, &Default::default(), &snapshot, &mut layout_cx)
1190 });
1191 assert_eq!(layouts.len(), 6);
1192 }
1193}