Reset UI when loading a .puffin file

Author: emilk

puffin_egui/src/lib.rs

@@ -324,6 +324,14 @@ fn latest_frames() -> Frames {
 }
 
 impl ProfilerUi {
+    pub fn reset(&mut self) {
+        let options = self.options;
+        *self = Self {
+            options,
+            ..Default::default()
+        };
+    }
+
     /// Show an [`egui::Window`] with the profiler contents.
     ///
     /// If you want to control the window yourself, use [`Self::ui`] instead.

puffin_viewer/src/main.rs

@@ -110,6 +110,7 @@ fn main() {
             Ok(profiler) => {
                 *GlobalProfiler::lock() = profiler;
                 PuffinViewer {
+                    profiler_ui: Default::default(),
                     source: Source::File(path),
                     error: None,
                 }
@@ -121,6 +122,7 @@ fn main() {
         }
     } else {
         PuffinViewer {
+            profiler_ui: Default::default(),
             source: Source::Http(puffin_http::Client::new(opt.url)),
             error: None,
         }
@@ -153,6 +155,7 @@ impl Source {
 }
 
 pub struct PuffinViewer {
+    profiler_ui: puffin_egui::ProfilerUi,
     source: Source,
     error: Option<String>,
 }
@@ -166,14 +169,15 @@ impl epi::App for PuffinViewer {
         egui::TopBottomPanel::top("menu_bar").show(ctx, |ui| {
             egui::menu::bar(ui, |ui| {
                 egui::menu::menu(ui, "File", |ui| {
-                    if ui.button("Load…").clicked() {
+                    if ui.button("Open…").clicked() {
                         if let Some(path) = rfd::FileDialog::new()
                             .add_filter("puffin", &["puffin"])
                             .pick_file()
                         {
                             match GlobalProfiler::load_path(&path) {
                                 Ok(profiler) => {
                                     *GlobalProfiler::lock() = profiler;
+                                    self.profiler_ui.reset();
                                     self.source = Source::File(path);
                                     self.error = None;
                                 }
@@ -185,7 +189,7 @@ impl epi::App for PuffinViewer {
                         }
                     }
 
-                    if ui.button("Save…").clicked() {
+                    if ui.button("Save as…").clicked() {
                         if let Some(path) = rfd::FileDialog::new()
                             .add_filter("puffin", &["puffin"])
                             .save_file()
@@ -214,6 +218,8 @@ impl epi::App for PuffinViewer {
             self.source.ui(ui);
         });
 
-        egui::CentralPanel::default().show(ctx, puffin_egui::profiler_ui);
+        egui::CentralPanel::default().show(ctx, |ui| {
+            self.profiler_ui.ui(ui);
+        });
     }
 }