use crate::{CandidateTranscript, OwnedCandidateTranscript};
#[repr(transparent)]
pub struct Metadata(*mut coqui_stt_sys::Metadata);
unsafe impl Send for Metadata {}
unsafe impl Sync for Metadata {}
impl Drop for Metadata {
#[inline]
fn drop(&mut self) {
unsafe { coqui_stt_sys::STT_FreeMetadata(self.0) }
}
}
impl Metadata {
pub(crate) fn new(ptr: *mut coqui_stt_sys::Metadata) -> Self {
if ptr.is_null() {
unreachable!("attempted to construct Metadata with a null pointer");
}
Self(ptr)
}
#[inline]
#[must_use]
pub fn transcripts(&self) -> &[CandidateTranscript] {
let data = unsafe { (*self.0).transcripts.cast() };
let len = self.num_transcripts() as usize;
unsafe { std::slice::from_raw_parts(data, len) }
}
#[inline]
#[must_use]
#[allow(clippy::missing_const_for_fn)]
pub fn num_transcripts(&self) -> u32 {
unsafe { (*self.0).num_transcripts }
}
#[inline]
#[must_use]
pub fn to_owned(&self) -> OwnedMetadata {
OwnedMetadata(
self.transcripts()
.iter()
.map(CandidateTranscript::to_owned)
.collect(),
)
}
}
pub struct OwnedMetadata(Vec<OwnedCandidateTranscript>);
impl OwnedMetadata {
#[inline]
#[must_use]
pub fn transcripts(&self) -> &[OwnedCandidateTranscript] {
self.0.as_slice()
}
#[inline]
#[must_use]
pub fn num_transcripts(&self) -> u32 {
self.0.len() as u32
}
#[inline]
#[must_use]
#[allow(clippy::missing_const_for_fn)]
pub fn into_inner(self) -> Vec<OwnedCandidateTranscript> {
self.0
}
}