From f24f71f592b7b8d6efcbc99b4c95e7df97e4ff69 Mon Sep 17 00:00:00 2001
From: Sébastien Crozet <developer@crozet.re>
Date: Mon, 24 Jan 2022 14:35:34 +0100
Subject: Testbed: add a constructor that calls a custom closure with bevy’s
 App
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src_testbed/testbed.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'src_testbed')

diff --git a/src_testbed/testbed.rs b/src_testbed/testbed.rs
index dba40fd..1a4181b 100644
--- a/src_testbed/testbed.rs
+++ b/src_testbed/testbed.rs
@@ -227,7 +227,11 @@ impl TestbedApp {
         self.builders = SceneBuilders(builders)
     }
 
-    pub fn run(mut self) {
+    pub fn run(self) {
+        self.run_with_init(|_| {})
+    }
+
+    pub fn run_with_init(mut self, mut init: impl FnMut(&mut App)) {
         let mut args = env::args();
         let mut benchmark_mode = false;
 
@@ -392,8 +396,9 @@ impl TestbedApp {
                 .insert_non_send_resource(self.plugins)
                 .add_stage_before(CoreStage::Update, "physics", SystemStage::single_threaded())
                 .add_system_to_stage("physics", update_testbed.system())
-                .add_system(egui_focus.system())
-                .run();
+                .add_system(egui_focus.system());
+            init(&mut app);
+            app.run();
         }
     }
 }
-- 
cgit