Add metric selection

Author: Kobzol

collector/src/compare/mod.rs

@@ -8,17 +8,14 @@ use ratatui::{
     prelude::*,
     widgets::Block,
 };
-use std::fmt::Display;
 use std::future::Future;
 use std::pin::Pin;
 use std::sync::Arc;
-use tabled::settings::object::{Column, Columns};
-use tabled::settings::Modify;
 use tabled::Tabled;
 
 static ALL_METRICS: &[Metric] = &[
     Metric::InstructionsUser,
-    Metric::Cycles,
+    Metric::CyclesUser,
     Metric::WallTime,
     Metric::MaxRSS,
     Metric::LinkedArtifactSize,
@@ -31,7 +28,7 @@ static ALL_METRICS: &[Metric] = &[
     Metric::CpuClock,
     Metric::CpuClockUser,
     Metric::CrateMetadataSize,
-    Metric::CyclesUser,
+    Metric::Cycles,
     Metric::DepGraphSize,
     Metric::DocByteSize,
     Metric::DwoFileSize,
@@ -301,7 +298,7 @@ impl CompareScreen {
         base: Commit,
         modified: Commit,
     ) -> anyhow::Result<Self> {
-        let pstats = load_data(
+        let data = load_data(
             metric,
             &db_state.index,
             db_state.db.as_mut(),
@@ -314,10 +311,22 @@ impl CompareScreen {
             modified,
             db_state,
             metric,
-            data: pstats,
+            data,
             table_state: TableState::default(),
         })
     }
+
+    async fn reload_data(&mut self) -> anyhow::Result<()> {
+        self.data = load_data(
+            self.metric,
+            &self.db_state.index,
+            self.db_state.db.as_mut(),
+            &self.base,
+            &self.modified,
+        )
+        .await?;
+        Ok(())
+    }
 }
 
 impl Screen for CompareScreen {
@@ -330,15 +339,20 @@ impl Screen for CompareScreen {
                 [
                     // +2 because of borders
                     Constraint::Min((summary_table.lines().count() + 2) as u16),
+                    Constraint::Length(2),
                     Constraint::Percentage(100),
                 ]
                 .as_ref(),
             )
             .split(frame.area());
+
         frame.render_widget(
             Paragraph::new(Text::raw(summary_table)).block(Block::bordered().title("Summary")),
             layout[0],
         );
+
+        render_metric(frame, self.metric, layout[1]);
+
         let header = Row::new(vec![
             Line::from("Benchmark"),
             Line::from("Profile"),
@@ -398,7 +412,7 @@ impl Screen for CompareScreen {
         .row_highlight_style(Style::new().bold());
 
         let table_layout =
-            Layout::new(Direction::Horizontal, [Constraint::Max(120)]).split(layout[1]);
+            Layout::new(Direction::Horizontal, [Constraint::Max(120)]).split(layout[2]);
         frame.render_stateful_widget(table, table_layout[0], &mut self.table_state);
     }
 
@@ -410,6 +424,14 @@ impl Screen for CompareScreen {
             match key {
                 KeyCode::Down => self.table_state.select_next(),
                 KeyCode::Up => self.table_state.select_previous(),
+                KeyCode::Char('a') => {
+                    self.metric = select_metric(self.metric, -1);
+                    self.reload_data().await?;
+                }
+                KeyCode::Char('s') => {
+                    self.metric = select_metric(self.metric, 1);
+                    self.reload_data().await?;
+                }
                 _ => {}
             }
 
@@ -418,6 +440,23 @@ impl Screen for CompareScreen {
     }
 }
 
+fn select_metric(current: Metric, direction: isize) -> Metric {
+    let index = ALL_METRICS.iter().position(|m| *m == current).unwrap_or(0) as isize;
+    let index = ((index + direction) + ALL_METRICS.len() as isize) % ALL_METRICS.len() as isize;
+    ALL_METRICS[index as usize]
+}
+
+fn render_metric(frame: &mut Frame, metric: Metric, area: Rect) {
+    frame.render_widget(
+        Line::from(vec![
+            "Metric: ".into(),
+            metric.as_str().bold(),
+            " (switch: A/S)".into(),
+        ]),
+        area,
+    )
+}
+
 async fn load_data(
     metric: Metric,
     index: &Index,