Pattern mute and so on

src/state/playback.rs

@@ -1,5 +1,6 @@
 use crate::engine::PatternChange;
 use crate::model::{LaunchQuantization, SyncMode};
+use std::collections::HashSet;
 
 #[derive(Clone)]
 pub struct StagedChange {
@@ -8,10 +9,17 @@ pub struct StagedChange {
     pub sync_mode: SyncMode,
 }
 
+#[derive(Clone, Copy, PartialEq, Eq, Hash)]
+pub enum StagedMuteChange {
+    ToggleMute { bank: usize, pattern: usize },
+    ToggleSolo { bank: usize, pattern: usize },
+}
+
 pub struct PlaybackState {
     pub playing: bool,
     pub staged_changes: Vec<StagedChange>,
     pub queued_changes: Vec<StagedChange>,
+    pub staged_mute_changes: HashSet<StagedMuteChange>,
 }
 
 impl Default for PlaybackState {
@@ -20,6 +28,7 @@ impl Default for PlaybackState {
             playing: true,
             staged_changes: Vec::new(),
             queued_changes: Vec::new(),
+            staged_mute_changes: HashSet::new(),
         }
     }
 }
@@ -33,4 +42,38 @@ impl PlaybackState {
         self.staged_changes.clear();
         self.queued_changes.clear();
     }
+
+    pub fn stage_mute(&mut self, bank: usize, pattern: usize) {
+        let change = StagedMuteChange::ToggleMute { bank, pattern };
+        if self.staged_mute_changes.contains(&change) {
+            self.staged_mute_changes.remove(&change);
+        } else {
+            self.staged_mute_changes.insert(change);
+        }
+    }
+
+    pub fn stage_solo(&mut self, bank: usize, pattern: usize) {
+        let change = StagedMuteChange::ToggleSolo { bank, pattern };
+        if self.staged_mute_changes.contains(&change) {
+            self.staged_mute_changes.remove(&change);
+        } else {
+            self.staged_mute_changes.insert(change);
+        }
+    }
+
+    pub fn clear_staged_mutes(&mut self) {
+        self.staged_mute_changes.retain(|c| !matches!(c, StagedMuteChange::ToggleMute { .. }));
+    }
+
+    pub fn clear_staged_solos(&mut self) {
+        self.staged_mute_changes.retain(|c| !matches!(c, StagedMuteChange::ToggleSolo { .. }));
+    }
+
+    pub fn has_staged_mute(&self, bank: usize, pattern: usize) -> bool {
+        self.staged_mute_changes.contains(&StagedMuteChange::ToggleMute { bank, pattern })
+    }
+
+    pub fn has_staged_solo(&self, bank: usize, pattern: usize) -> bool {
+        self.staged_mute_changes.contains(&StagedMuteChange::ToggleSolo { bank, pattern })
+    }
 }