Kieran/ffmpeg-the-third
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

util/frame/video: refactor data() and data_mut()

Browse Source
This commit is contained in:
meh
2015-10-26 14:44:06 +01:00
parent e9a9ad5720
commit 8298586d9f
1 changed files with 14 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/util/frame/video.rs
View File

@ -267,33 +267,27 @@ impl Video {
} }
} }
pub fn data(&self) -> Vec<&[u8]> { pub fn data(&self, index: usize) -> &[u8] {
let mut result = Vec::new(); if index >= self.planes() {
panic!("out of bounds");
}
unsafe { unsafe {
for (i, length) in (*self.as_ptr()).linesize.iter().take_while(|l| **l > 0).enumerate() { slice::from_raw_parts((*self.as_ptr()).data[index],
result.push(slice::from_raw_parts( (*self.as_ptr()).linesize[index] as usize * self.height() as usize)
(*self.as_ptr()).data[i],
*length as usize * self.height() as usize));
} }
} }
result pub fn data_mut(&mut self, index: usize) -> &mut [u8] {
if index >= self.planes() {
panic!("out of bounds");
} }
pub fn data_mut(&mut self) -> Vec<&mut [u8]> {
let mut result = Vec::new();
unsafe { unsafe {
for (i, length) in (*self.as_ptr()).linesize.iter().take_while(|l| **l > 0).enumerate() { slice::from_raw_parts_mut((*self.as_mut_ptr()).data[index],
result.push(slice::from_raw_parts_mut( (*self.as_ptr()).linesize[index] as usize * self.height() as usize)
(*self.as_mut_ptr()).data[i],
*length as usize * self.height() as usize));
} }
} }
result
}
} }
impl Deref for Video { impl Deref for Video {