Oliver/video-parser
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

abstracted frame parsing

Browse Source
This commit is contained in:
Oliver Atkinson 2023-06-29 14:45:50 -06:00
parent 29e6b0c11b
commit 61b6adb5bb
2 changed files with 12 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -1 +1,2 @@
/target /target
/.vscode

42
src/test.rs
View File

@ -2,29 +2,8 @@ use std::io::*;
use ffmpeg::format::Pixel; use ffmpeg::format::Pixel;
use rand::Rng; use rand::Rng;
pub fn small_matrix() {
let mut matrix = [0u8; 432]; // 144 * 3 = 432
let mut rng = rand::thread_rng();
let r = rng.gen_range(0..50);
let g = rng.gen_range(0..50);
let b = rng.gen_range(0..50);
matrix.iter_mut().enumerate().for_each(|f| {
let value = match f.0 % 3 {
0 => r,
1 => g,
_ => b,
};
*f.1 = value;
});
print_raw(&matrix);
}
#[inline] #[inline]
/// Asuumes a square ig /// Assumes a square ig
pub fn print_raw(buf: &[u8]) { pub fn print_raw(buf: &[u8]) {
let mut stdout = stdout(); let mut stdout = stdout();
let mut lock = stdout.lock(); let mut lock = stdout.lock();
@ -36,27 +15,26 @@ pub fn print_raw(buf: &[u8]) {
// pub fn print_square(buf: &[u8], width_hight: usize, line) { // pub fn print_square(buf: &[u8], width_hight: usize, line) {
pub fn print_square(frame: &ffmpeg::frame::Video) { pub fn print_square(frame: &ffmpeg::frame::Video) {
if frame.format() != Pixel::RGB24 { panic!("Must use Pixel::RGB24"); } if frame.format() != Pixel::RGB24 { panic!("Must use Pixel::RGB24"); }
if frame.width() != frame.height() { panic!("Must be square frames"); } if frame.width() != frame.height() { panic!("Must be square aspect ratio"); }
// Not sure if you can stack multiple frames on top of each other or what,
// but we just want the first one.
let z = 0;
let buf = frame.data(0); let buf = frame.data(z);
let size = frame.width() as usize; let size = frame.width() as usize;
// How many bytes per pixel RGBA // How many bytes per pixel RGBA
// frame.as_ptr(); let linesize = unsafe { std::ptr::addr_of!((*frame.as_ptr()).linesize) };
let step = size*4; let step = (unsafe { *linesize })[z] as usize;
// (Assuming square) Step thru buffer. // (Assuming square) Step thru buffer.
for i in 0..size*2 { for i in 0..size {
let j = i*step; let j = i*step;
// Slices non-inclusively, ie: Even though j(n)+step = j(n+1) the ranges don't overlap. It take from..until // Slices non-inclusively, ie: Even though j(n)+step = j(n+1) the ranges don't overlap. It take from..until
let pre_slice = &buf[j..j+step]; let pre_slice = &buf[j..j+step];
// Remove the last 12 bytes, alpha? // Only take the size*3 of bytes, ignoring alpha values which are stored after.
let slice = &pre_slice[0..size*3]; let slice = &pre_slice[0..size*3];
// let lol = &buf[j..j+step][0..size*3]; // let lol = &buf[j..j+step][0..size*3];
// Skip odd lines (they are filled with zeros)
if i & 1 != 1 {
// debug
println!("{:?} || {}-{}", slice, j, j+step); println!("{:?} || {}-{}", slice, j, j+step);
} }
} }
}