harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

box the progress callback closure - fixes #134

Browse source
This commit is contained in:
Roman Steiner 2025-02-25 10:50:32 +01:00
parent a5d279c31b
commit 807bc242e9
No known key found for this signature in database
GPG key ID: 2D3EE6D67E86519E
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/whisper_params.rs
View file

@ -575,11 +575,13 @@ impl<'a, 'b> FullParams<'a, 'b> {
}
match closure.into() {
Some(mut closure) => {
self.fp.progress_callback = Some(trampoline::<F>);
self.fp.progress_callback_user_data = &mut closure as *mut F as *mut c_void;
// store the closure internally to make sure that the pointer above remains valid
self.progess_callback_safe = Some(Arc::new(Box::new(closure)));
Some(closure) => {
self.fp.progress_callback = Some(trampoline::<Box<dyn FnMut(i32)>>);
let boxed_closure = Box::new(closure) as Box<dyn FnMut(i32)>;
let boxed_closure = Box::new(boxed_closure);
let raw_ptr = Box::into_raw(boxed_closure);
self.fp.progress_callback_user_data = raw_ptr as *mut c_void;
self.progess_callback_safe = None;
}
None => {
self.fp.progress_callback = None;