Bench app context phase 2

crates/editor/Cargo.toml

@@ -108,7 +108,8 @@ zlog.workspace = true
 [dev-dependencies]
 criterion.workspace = true
 ctor.workspace = true
-gpui = { workspace = true, features = ["test-support"] }
+gpui = { workspace = true, features = ["bench"] }
+gpui_platform = { workspace = true, features = ["font-kit", "wayland", "x11"] }
 language = { workspace = true, features = ["test-support"] }
 languages = {workspace = true, features = ["test-support"] }
 lsp = { workspace = true, features = ["test-support"] }

crates/editor/benches/editor_render.rs

@@ -6,18 +6,18 @@ use editor::{
 use gpui::{AppContext as _, BenchAppContext, Focusable as _, TestAppContext, TestDispatcher};
 use rand::{Rng as _, SeedableRng as _, rngs::StdRng};
 use settings::SettingsStore;
-use ui::IntoElement;
 use util::RandomCharIter;
+use zed_actions::editor::{MoveDown, MoveUp};
 
 #[gpui::bench]
-fn editor_input_with_1000_cursors(bencher: &mut Bencher<'_>, cx: &mut BenchAppContext) {
+fn editor_input_with_1000_cursors(cx: &mut BenchAppContext) {
     init_context(cx);
 
     let text = String::from_iter(["line:\n"; 1000]);
     let buffer = cx.update(|cx| MultiBuffer::build_simple(&text, cx));
 
-    let mut cx = cx.add_empty_window();
-    let editor = cx.update(|window, cx| {
+    let mut window = cx.add_empty_window();
+    let editor = window.update(|window, cx| {
         let editor = cx.new(|cx| {
             let mut editor = Editor::new(EditorMode::full(), buffer, None, window, cx);
             editor.set_style(editor::EditorStyle::default(), window, cx);
@@ -35,8 +35,8 @@ fn editor_input_with_1000_cursors(bencher: &mut Bencher<'_>, cx: &mut BenchAppCo
         editor
     });
 
-    bencher.iter(|| {
-        cx.update(|window, cx| {
+    cx.bench_iter(|_| {
+        window.update(|window, cx| {
             editor.update(cx, |editor, cx| {
                 editor.handle_input("hello world", window, cx);
                 editor.delete_to_previous_word_start(
@@ -80,8 +80,10 @@ fn open_editor_with_one_long_line(bencher: &mut Bencher<'_>, args: &(String, Tes
     });
 }
 
-fn editor_render(bencher: &mut Bencher<'_>, cx: &TestAppContext) {
-    let mut cx = cx.clone();
+#[gpui::bench]
+fn editor_render(cx: &mut BenchAppContext) {
+    init_context(cx);
+
     let buffer = cx.update(|cx| {
         let mut rng = StdRng::seed_from_u64(1);
         let text_len = rng.random_range(10000..90000);
@@ -95,9 +97,9 @@ fn editor_render(bencher: &mut Bencher<'_>, cx: &TestAppContext) {
         }
     });
 
-    let cx = cx.add_empty_window();
-    let editor = cx.update(|window, cx| {
-        let editor = cx.new(|cx| {
+    let mut window = cx.add_empty_window();
+    let editor = window.update(|window, cx| {
+        let editor = window.replace_root(cx, |window, cx| {
             let mut editor = Editor::new(EditorMode::full(), buffer, None, window, cx);
             editor.set_style(editor::EditorStyle::default(), window, cx);
             editor
@@ -106,14 +108,15 @@ fn editor_render(bencher: &mut Bencher<'_>, cx: &TestAppContext) {
         editor
     });
 
-    bencher.iter(|| {
-        cx.update(|window, cx| {
-            let mut view = editor.clone().into_any_element();
-            let _ = view.request_layout(window, cx);
-            let _ = view.prepaint(window, cx);
-            view.paint(window, cx);
-        });
-    })
+    let mut move_down = true;
+    cx.bench_renderer(editor, move |editor, window, cx| {
+        if move_down {
+            editor.move_down(&MoveDown, window, cx);
+        } else {
+            editor.move_up(&MoveUp, window, cx);
+        }
+        move_down = !move_down;
+    });
 }
 
 fn init_context(cx: &mut BenchAppContext) {
@@ -141,16 +144,8 @@ fn criterion_benches(criterion: &mut criterion::Criterion) {
     let cx = gpui::TestAppContext::build(dispatcher, None);
     init_test_context(&cx);
 
-    let mut group = criterion.benchmark_group("Time to render");
-    group.bench_with_input(
-        BenchmarkId::new("editor_render", "TestAppContext"),
-        &cx,
-        editor_render,
-    );
-    group.finish();
-
     let text = String::from_iter(["char"; 1000]);
-    let input = (text, cx.clone());
+    let input = (text, cx);
     let mut group = criterion.benchmark_group("Build buffer with one long line");
     group.bench_with_input(
         BenchmarkId::new("editor_with_one_long_line", "(String, TestAppContext )"),
@@ -160,5 +155,10 @@ fn criterion_benches(criterion: &mut criterion::Criterion) {
     group.finish();
 }
 
-gpui::bench_group!(benches, editor_input_with_1000_cursors, criterion_benches);
+gpui::bench_group!(
+    benches,
+    editor_input_with_1000_cursors,
+    editor_render,
+    criterion_benches
+);
 gpui::bench_main!(benches);

crates/gpui/Cargo.toml

@@ -26,6 +26,7 @@ test-support = [
     "x11",
     "proptest",
 ]
+bench = ["test-support", "dep:criterion", "dep:hdrhistogram"]
 inspector = ["gpui_macros/inspector"]
 leak-detection = ["backtrace"]
 wayland = [
@@ -52,6 +53,7 @@ backtrace = { workspace = true, optional = true }
 bitflags = { workspace = true, optional = true }
 
 collections.workspace = true
+criterion = { workspace = true, optional = true }
 ctor.workspace = true
 derive_more.workspace = true
 etagere = "0.2"

crates/gpui/src/app.rs

@@ -23,8 +23,8 @@ use parking_lot::RwLock;
 use slotmap::SlotMap;
 
 pub use async_context::*;
-#[cfg(any(test, feature = "test-support"))]
-pub use bench_context::{BenchAppContext, BenchWindowContext};
+#[cfg(feature = "bench")]
+pub use bench_context::{BenchAppContext, BenchReport, BenchWindowContext};
 use collections::{FxHashMap, FxHashSet, HashMap, VecDeque};
 pub use context::*;
 pub use entity_map::*;
@@ -58,7 +58,7 @@ use crate::{
 };
 
 mod async_context;
-#[cfg(any(test, feature = "test-support"))]
+#[cfg(feature = "bench")]
 mod bench_context;
 mod context;
 mod entity_map;
@@ -1489,7 +1489,7 @@ impl App {
                     }
                 }
             } else {
-                #[cfg(any(test, feature = "test-support"))]
+                #[cfg(any(test, feature = "test-support", feature = "bench"))]
                 for window in self
                     .windows
                     .values()