maolan_engine/plugins/
ipc.rs1use crate::audio::io::AudioIO;
4use crate::mutex::UnsafeMutex;
5use maolan_plugin_protocol::events::EventPair;
6use maolan_plugin_protocol::protocol::*;
7use maolan_plugin_protocol::shm::ShmMapping;
8use std::path::{Path, PathBuf};
9use std::process::{Child, Command, Stdio};
10use std::sync::Arc;
11use std::sync::atomic::{AtomicU64, Ordering};
12use std::time::{Duration, Instant};
13
14static NEXT_INSTANCE_ID: AtomicU64 = AtomicU64::new(0);
15
16pub fn unique_instance_id(format: &str) -> String {
18 let n = NEXT_INSTANCE_ID.fetch_add(1, Ordering::Relaxed);
19 format!("{}-{}-{}", format, std::process::id(), n)
20}
21
22pub struct HostSpawnArgs<'a> {
24 pub host_binary: &'a Path,
25 pub format: &'a str,
26 pub plugin_spec: &'a str,
27 pub instance_id: &'a str,
28 pub extra_args: &'a [&'a str],
29}
30
31pub fn spawn_host(args: HostSpawnArgs) -> Result<(Child, ShmMapping, EventPair, String), String> {
33 let pid = std::process::id();
34 let shm_name = format!("/maolan-{pid}-{}", args.instance_id);
35
36 let mapping = ShmMapping::create(&shm_name, SHM_SIZE)
37 .map_err(|e| format!("failed to create shared memory: {e}"))?;
38 unsafe {
39 init_shm_layout(mapping.as_ptr(), mapping.size());
40 }
41
42 let mut events = EventPair::new().map_err(|e| format!("failed to create event pipes: {e}"))?;
43
44 let mut cmd = Command::new(args.host_binary);
45 cmd.arg(args.format)
46 .arg(args.plugin_spec)
47 .arg(&shm_name)
48 .arg(args.instance_id)
49 .stdin(Stdio::null())
50 .stdout(Stdio::null())
51 .stderr(Stdio::inherit());
52
53 for arg in args.extra_args {
54 cmd.arg(arg);
55 }
56
57 #[cfg(unix)]
58 {
59 cmd.arg(events.host_read_fd().to_string())
60 .arg(events.host_write_fd().to_string());
61 }
62 #[cfg(windows)]
63 {
64 cmd.arg(events.daw_to_host_name())
65 .arg(events.host_to_daw_name());
66 }
67
68 let parent_args: Vec<String> = std::env::args().collect();
69 if let Some(pos) = parent_args.iter().position(|a| a == "--log-level")
70 && pos + 1 < parent_args.len()
71 {
72 cmd.arg("--log-level").arg(&parent_args[pos + 1]);
73 }
74
75 let child = cmd
76 .spawn()
77 .map_err(|e| format!("failed to spawn {} host: {e}", args.format))?;
78
79 events.close_daw_unused();
80
81 Ok((child, mapping, events, shm_name))
82}
83
84pub fn wait_for_ready(header: &ShmHeader, timeout: Duration) -> bool {
86 let start = Instant::now();
87 while start.elapsed() < timeout {
88 if header.ready.load(Ordering::Acquire) != 0 {
89 return true;
90 }
91 std::thread::sleep(Duration::from_millis(5));
92 }
93 false
94}
95
96pub fn bypass_copy_inputs_to_outputs(inputs: &[Arc<AudioIO>], outputs: &[Arc<AudioIO>]) {
98 for (input, output) in inputs.iter().zip(outputs.iter()) {
99 let src = input.buffer.lock();
100 let dst = output.buffer.lock();
101 dst.fill(0.0);
102 for (d, s) in dst.iter_mut().zip(src.iter()) {
103 *d = *s;
104 }
105 *output.finished.lock() = true;
106 }
107 for output in outputs.iter().skip(inputs.len()) {
108 let dst = output.buffer.lock();
109 dst.fill(0.0);
110 *output.finished.lock() = true;
111 }
112}
113
114pub fn drop_host(
116 mapping: &Option<ShmMapping>,
117 events: &Option<EventPair>,
118 child: &UnsafeMutex<Option<Child>>,
119 shm_name: &str,
120) {
121 if let Some(mapping) = mapping
122 && let Some(events) = events
123 {
124 let header = unsafe { header_mut(mapping.as_ptr()) };
125 header.shutdown_request.store(1, Ordering::Release);
126 let _ = events.signal_host();
127 }
128 let mut child_opt = child.lock().take();
129 if let Some(mut child) = child_opt.take() {
130 let start = Instant::now();
131 while start.elapsed() < Duration::from_secs(2) {
132 if child.try_wait().map(|s| s.is_some()).unwrap_or(true) {
133 break;
134 }
135 std::thread::sleep(Duration::from_millis(10));
136 }
137 if child.try_wait().map(|s| s.is_none()).unwrap_or(false) {
138 let _ = child.kill();
139 }
140 }
141 let _ = ShmMapping::unlink(shm_name);
142}
143
144pub fn find_plugin_host_binary() -> Option<PathBuf> {
151 let exe_dir = std::env::current_exe()
152 .ok()
153 .and_then(|p| p.parent().map(PathBuf::from));
154
155 if let Some(ref dir) = exe_dir {
157 let candidate = dir.join("maolan-plugin-host");
158 if candidate.exists() {
159 return Some(candidate);
160 }
161 }
162
163 if let Ok(manifest) = std::env::var("CARGO_MANIFEST_DIR") {
165 let engine_root = Path::new(&manifest);
166 for profile in ["debug", "release"] {
167 let candidate = engine_root
168 .parent()
169 .unwrap_or(Path::new(""))
170 .join("daw")
171 .join("target")
172 .join(profile)
173 .join("maolan-plugin-host");
174 if candidate.exists() {
175 return Some(candidate);
176 }
177 }
178 }
179
180 if let Ok(path_var) = std::env::var("PATH") {
182 for dir in path_var.split(':') {
183 let candidate = Path::new(dir).join("maolan-plugin-host");
184 if candidate.exists() {
185 return Some(candidate);
186 }
187 }
188 }
189
190 None
191}
192
193pub unsafe fn copy_inputs_to_shm(inputs: &[Arc<AudioIO>], ptr: *mut u8, frames: usize) {
198 for (ch, input) in inputs.iter().enumerate() {
199 let src = input.buffer.lock();
200 let dst = unsafe { audio_channel_ptr(ptr, ch, 0) };
201 let len = frames.min(src.len());
202 unsafe {
203 std::ptr::copy_nonoverlapping(src.as_ptr(), dst, len);
204 }
205 }
206}
207
208pub unsafe fn copy_outputs_from_shm(outputs: &[Arc<AudioIO>], ptr: *mut u8, frames: usize) {
213 for (ch, output) in outputs.iter().enumerate() {
214 let dst = output.buffer.lock();
215 let src = unsafe { audio_channel_ptr(ptr, ch, 1) };
216 let len = frames.min(dst.len());
217 unsafe {
218 std::ptr::copy_nonoverlapping(src, dst.as_mut_ptr(), len);
219 }
220 *output.finished.lock() = true;
221 }
222}
223
224pub unsafe fn configure_shm_header(ptr: *mut u8, frames: usize, num_in: usize, num_out: usize) {
229 unsafe {
230 let h = header_mut(ptr);
231 h.block_size.store(frames as u32, Ordering::Release);
232 h.num_input_channels.store(num_in as u32, Ordering::Release);
233 h.num_output_channels
234 .store(num_out as u32, Ordering::Release);
235 }
236}
237
238#[macro_export]
241macro_rules! impl_ipc_processor_wrapper {
242 ($processor:ty) => {
243 impl $crate::mutex::UnsafeMutex<$processor> {
244 pub fn setup_audio_ports(&self) {
245 self.lock().setup_audio_ports();
246 }
247
248 pub fn audio_inputs(&self) -> &[std::sync::Arc<$crate::audio::io::AudioIO>] {
249 self.lock().audio_inputs()
250 }
251
252 pub fn audio_outputs(&self) -> &[std::sync::Arc<$crate::audio::io::AudioIO>] {
253 self.lock().audio_outputs()
254 }
255
256 pub fn main_audio_input_count(&self) -> usize {
257 self.lock().main_audio_input_count()
258 }
259
260 pub fn main_audio_output_count(&self) -> usize {
261 self.lock().main_audio_output_count()
262 }
263
264 pub fn midi_input_count(&self) -> usize {
265 self.lock().midi_input_count()
266 }
267
268 pub fn midi_output_count(&self) -> usize {
269 self.lock().midi_output_count()
270 }
271
272 pub fn set_bypassed(&self, bypassed: bool) {
273 self.lock().set_bypassed(bypassed);
274 }
275
276 pub fn name(&self) -> String {
277 self.lock().name().to_string()
278 }
279
280 pub fn run_host_callbacks_main_thread(&self) {
281 self.lock().run_host_callbacks_main_thread();
282 }
283
284 pub fn reconfigure_ports_if_needed(&self) -> Result<bool, String> {
285 self.lock().reconfigure_ports_if_needed()
286 }
287 }
288 };
289}