use std::rc::Rc;
use log::error;
use crate::bindings;
use crate::buffer::Buffer;
use crate::buffer::BufferType;
use crate::display::Display;
use crate::va_check;
use crate::Config;
use crate::EncCodedBuffer;
use crate::Surface;
use crate::SurfaceMemoryDescriptor;
use crate::VaError;
pub struct Context {
display: Rc<Display>,
id: bindings::VAContextID,
}
impl Context {
pub(crate) fn new<D: SurfaceMemoryDescriptor>(
display: Rc<Display>,
config: &Config,
coded_width: u32,
coded_height: u32,
surfaces: Option<&Vec<Surface<D>>>,
progressive: bool,
) -> Result<Rc<Self>, VaError> {
let mut context_id = 0;
let flags = if progressive {
bindings::constants::VA_PROGRESSIVE as i32
} else {
0
};
let mut render_targets = match surfaces {
Some(surfaces) => Surface::as_id_vec(surfaces),
None => Default::default(),
};
va_check(unsafe {
bindings::vaCreateContext(
display.handle(),
config.id(),
coded_width as i32,
coded_height as i32,
flags,
render_targets.as_mut_ptr(),
render_targets.len() as i32,
&mut context_id,
)
})?;
Ok(Rc::new(Self {
display,
id: context_id,
}))
}
pub fn display(&self) -> &Rc<Display> {
&self.display
}
pub(crate) fn id(&self) -> bindings::VAContextID {
self.id
}
pub fn create_buffer(self: &Rc<Self>, type_: BufferType) -> Result<Buffer, VaError> {
Buffer::new(Rc::clone(self), type_)
}
pub fn create_enc_coded(self: &Rc<Self>, size: usize) -> Result<EncCodedBuffer, VaError> {
EncCodedBuffer::new(Rc::clone(self), size)
}
}
impl Drop for Context {
fn drop(&mut self) {
let status =
va_check(unsafe { bindings::vaDestroyContext(self.display.handle(), self.id) });
if status.is_err() {
error!("vaDestroyContext failed: {}", status.unwrap_err());
}
}
}