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

refactor(state): create_state from whisper context wrapper

Browse source
This commit is contained in:
jiahua 2024-04-25 15:33:06 +08:00
parent 74e83185bf
commit 9d978add9d
3 changed files with 40 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/whisper_ctx.rs
View file

@ -10,7 +10,7 @@ use std::ffi::{c_int, CStr, CString};
/// then run a full transcription with [WhisperState::full].
#[derive(Debug)]
pub struct WhisperContext {
ctx: *mut whisper_rs_sys::whisper_context,
pub(crate) ctx: *mut whisper_rs_sys::whisper_context,
}
impl WhisperContext {
@ -114,24 +114,6 @@ impl WhisperContext {
}
}
// we don't implement `whisper_init()` here since i have zero clue what `whisper_model_loader` does
/// Create a new state object, ready for use.
///
/// # Returns
/// Ok(WhisperState) on success, Err(WhisperError) on failure.
///
/// # C++ equivalent
/// `struct whisper_state * whisper_init_state(struct whisper_context * ctx);`
pub fn create_state(&self) -> Result<WhisperState, WhisperError> {
let state = unsafe { whisper_rs_sys::whisper_init_state(self.ctx) };
if state.is_null() {
Err(WhisperError::InitError)
} else {
// SAFETY: this is known to be a valid pointer to a `whisper_state` struct
Ok(WhisperState::new(self.ctx, state))
}
}
/// Convert the provided text into tokens.
///