Support muting sims

src/emulator.rs

@@ -12,8 +12,8 @@ use std::{
 use anyhow::Result;
 
 use crate::{audio::Audio, graphics::TextureSink};
-use shrooms_vb_core::Sim;
 pub use shrooms_vb_core::VBKey;
+use shrooms_vb_core::{Sim, EXPECTED_FRAME_SIZE};
 
 mod shrooms_vb_core;
 
@@ -23,6 +23,7 @@ pub struct EmulatorBuilder {
     sim_count: Arc<AtomicUsize>,
     running: Arc<[AtomicBool; 2]>,
     has_game: Arc<[AtomicBool; 2]>,
+    audio_on: Arc<[AtomicBool; 2]>,
     linked: Arc<AtomicBool>,
 }
 
@@ -52,6 +53,7 @@ impl EmulatorBuilder {
             sim_count: Arc::new(AtomicUsize::new(0)),
             running: Arc::new([AtomicBool::new(false), AtomicBool::new(false)]),
             has_game: Arc::new([AtomicBool::new(false), AtomicBool::new(false)]),
+            audio_on: Arc::new([AtomicBool::new(true), AtomicBool::new(true)]),
             linked: Arc::new(AtomicBool::new(false)),
         };
         let client = EmulatorClient {
@@ -59,6 +61,7 @@ impl EmulatorBuilder {
             sim_count: builder.sim_count.clone(),
             running: builder.running.clone(),
             has_game: builder.has_game.clone(),
+            audio_on: builder.audio_on.clone(),
             linked: builder.linked.clone(),
         };
         (builder, client)
@@ -77,6 +80,7 @@ impl EmulatorBuilder {
             self.sim_count,
             self.running,
             self.has_game,
+            self.audio_on,
             self.linked,
         )?;
         if let Some(path) = self.rom {
@@ -93,6 +97,7 @@ pub struct Emulator {
     sim_count: Arc<AtomicUsize>,
     running: Arc<[AtomicBool; 2]>,
     has_game: Arc<[AtomicBool; 2]>,
+    audio_on: Arc<[AtomicBool; 2]>,
     linked: Arc<AtomicBool>,
     renderers: HashMap<SimId, TextureSink>,
 }
@@ -103,6 +108,7 @@ impl Emulator {
         sim_count: Arc<AtomicUsize>,
         running: Arc<[AtomicBool; 2]>,
         has_game: Arc<[AtomicBool; 2]>,
+        audio_on: Arc<[AtomicBool; 2]>,
         linked: Arc<AtomicBool>,
     ) -> Result<Self> {
         Ok(Self {
@@ -112,6 +118,7 @@ impl Emulator {
             sim_count,
             running,
             has_game,
+            audio_on,
             linked,
             renderers: HashMap::new(),
         })
@@ -210,15 +217,28 @@ impl Emulator {
                     }
                 }
             }
-            let weight = 1.0 / self.sims.len() as f32;
-            for sim in self.sims.iter_mut() {
-                sim.read_samples(&mut audio_samples, weight);
+            let p1_audio =
+                p1_running && self.audio_on[SimId::Player1.to_index()].load(Ordering::Acquire);
+            let p2_audio =
+                p2_running && self.audio_on[SimId::Player2.to_index()].load(Ordering::Acquire);
+            let weight = if p1_audio && p2_audio { 0.5 } else { 1.0 };
+            if p1_audio {
+                if let Some(sim) = self.sims.get_mut(SimId::Player1.to_index()) {
+                    sim.read_samples(&mut audio_samples, weight);
+                }
             }
-            if !audio_samples.is_empty() {
+            if p2_audio {
+                if let Some(sim) = self.sims.get_mut(SimId::Player2.to_index()) {
+                    sim.read_samples(&mut audio_samples, weight);
+                }
+            }
+            if audio_samples.is_empty() {
+                audio_samples.resize(EXPECTED_FRAME_SIZE, 0.0);
+            } else {
                 idle = false;
-                self.audio.update(&audio_samples);
-                audio_samples.clear();
             }
+            self.audio.update(&audio_samples);
+            audio_samples.clear();
             if idle {
                 // The game is paused, and we have output all the video/audio we have.
                 // Block the thread until a new command comes in.
@@ -274,6 +294,10 @@ impl Emulator {
                     }
                 }
             }
+            EmulatorCommand::SetAudioEnabled(p1, p2) => {
+                self.audio_on[SimId::Player1.to_index()].store(p1, Ordering::Release);
+                self.audio_on[SimId::Player2.to_index()].store(p2, Ordering::Release);
+            }
             EmulatorCommand::Link => {
                 self.link_sims();
             }
@@ -302,6 +326,7 @@ pub enum EmulatorCommand {
     StopSecondSim,
     Pause,
     Resume,
+    SetAudioEnabled(bool, bool),
     Link,
     Unlink,
     Reset(SimId),
@@ -314,6 +339,7 @@ pub struct EmulatorClient {
     sim_count: Arc<AtomicUsize>,
     running: Arc<[AtomicBool; 2]>,
     has_game: Arc<[AtomicBool; 2]>,
+    audio_on: Arc<[AtomicBool; 2]>,
     linked: Arc<AtomicBool>,
 }
 
@@ -330,6 +356,9 @@ impl EmulatorClient {
     pub fn are_sims_linked(&self) -> bool {
         self.linked.load(Ordering::Acquire)
     }
+    pub fn is_audio_enabled(&self, sim_id: SimId) -> bool {
+        self.audio_on[sim_id.to_index()].load(Ordering::Acquire)
+    }
     pub fn send_command(&self, command: EmulatorCommand) {
         if let Err(err) = self.queue.send(command) {
             eprintln!(