diff --git a/gstreamer/src/functions.rs b/gstreamer/src/functions.rs index 4ab358440..23046e71d 100644 --- a/gstreamer/src/functions.rs +++ b/gstreamer/src/functions.rs @@ -8,6 +8,7 @@ use glib::translate::*; use gst_sys; +use std::mem; use std::ptr; use Element; @@ -104,3 +105,64 @@ pub fn util_seqnum_next() -> ::Seqnum { v } } + +#[cfg(any(feature = "v1_12", feature = "dox"))] +pub fn calculate_linear_regression( + xy: &[(u64, u64)], + temp: Option<&mut [(u64, u64)]>, +) -> Option<(u64, u64, u64, u64, f64)> { + unsafe { + assert_eq!(mem::size_of::() * 2, mem::size_of::<(u64, u64)>()); + assert_eq!(mem::align_of::(), mem::align_of::<(u64, u64)>()); + assert!(temp.as_ref().map(|temp| temp.len()).unwrap_or(xy.len()) >= xy.len()); + + let mut m_num = mem::MaybeUninit::uninit(); + let mut m_denom = mem::MaybeUninit::uninit(); + let mut b = mem::MaybeUninit::uninit(); + let mut xbase = mem::MaybeUninit::uninit(); + let mut r_squared = mem::MaybeUninit::uninit(); + + let res = from_glib(gst_sys::gst_calculate_linear_regression( + xy.as_ptr() as *const u64, + temp.map(|temp| temp.as_mut_ptr() as *mut u64) + .unwrap_or(ptr::null_mut()), + xy.len() as u32, + m_num.as_mut_ptr(), + m_denom.as_mut_ptr(), + b.as_mut_ptr(), + xbase.as_mut_ptr(), + r_squared.as_mut_ptr(), + )); + if res { + Some(( + m_num.assume_init(), + m_denom.assume_init(), + b.assume_init(), + xbase.assume_init(), + r_squared.assume_init(), + )) + } else { + None + } + } +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn test_calculate_linear_regression() { + ::init().unwrap(); + + let values = [(0, 0), (1, 1), (2, 2), (3, 3)]; + + let (m_num, m_denom, b, xbase, _) = calculate_linear_regression(&values, None).unwrap(); + assert_eq!((m_num, m_denom, b, xbase), (10, 10, 3, 3)); + + let mut temp = [(0, 0); 4]; + let (m_num, m_denom, b, xbase, _) = + calculate_linear_regression(&values, Some(&mut temp)).unwrap(); + assert_eq!((m_num, m_denom, b, xbase), (10, 10, 3, 3)); + } +}