fix show, enable and sound speed setting

shinolab · Jun 28, 2024 · 614f2f6 · 614f2f6
1 parent d62d5f4
commit 614f2f6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/simulator/src/view/imgui/mod.rs b/simulator/src/view/imgui/mod.rs
@@ -435,7 +435,7 @@ impl ImGuiViewer {
                                 std::f32::MAX / 2.,
                             ) {
                                 cpus.iter().for_each(|cpu| {
-                                    sources.drives_mut().skip(body_pointer[cpu.idx()]).for_each(
+                                    sources.drives_mut().skip(body_pointer[cpu.idx()]).take(cpu.num_transducers()).for_each(
                                         |s| {
                                             s.set_wave_number(
                                                 cpu.fpga().ultrasound_freq(),
@@ -463,6 +463,7 @@ impl ImGuiViewer {
                                     sources
                                         .visibilities_mut()
                                         .skip(body_pointer[i])
+                                        .take(cpu.num_transducers())
                                         .for_each(|s| *s = v);
                                 }
                                 ui.same_line();
@@ -472,6 +473,7 @@ impl ImGuiViewer {
                                     sources
                                         .drives_mut()
                                         .skip(body_pointer[i])
+                                        .take(cpu.num_transducers())
                                         .for_each(|s| s.enable = v);
                                 }
                                 ui.same_line();