diff options
author | Kostya Shishkov <kostya.shishkov@gmail.com> | 2019-03-22 18:56:55 +0100 |
---|---|---|
committer | Kostya Shishkov <kostya.shishkov@gmail.com> | 2019-03-22 18:56:55 +0100 |
commit | 3daa5fbb2446bdb3a9578e5b6628b8308ca3fe5f (patch) | |
tree | 3fb9baf9e7cee496f960630741c9aae8a29613c6 | |
parent | ffe695773d38f16f1a6f04086f47c19c1c7b4a92 (diff) | |
download | nihav-tool-3daa5fbb2446bdb3a9578e5b6628b8308ca3fe5f.tar.gz |
output alpha part of the PGMYUV too
-rw-r--r-- | src/frmwriter.rs | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/src/frmwriter.rs b/src/frmwriter.rs index 1a94fab..c743fc0 100644 --- a/src/frmwriter.rs +++ b/src/frmwriter.rs @@ -12,7 +12,13 @@ pub fn write_pgmyuv(pfx: &str, strno: usize, num: u64, frm: Ref<NAFrame>) { let buf = frm.get_buffer().get_vbuf().unwrap(); let (w, h) = buf.get_dimensions(0); let (w2, h2) = buf.get_dimensions(1); - let tot_h = h + h2; + let has_alpha = buf.get_info().get_format().has_alpha(); + let tot_h; + if has_alpha { + tot_h = h * 2 + h2; + } else { + tot_h = h + h2; + } let hdr = format!("P5\n{} {}\n255\n", w, tot_h); ofile.write_all(hdr.as_bytes()).unwrap(); let dta = buf.get_data(); @@ -45,6 +51,17 @@ pub fn write_pgmyuv(pfx: &str, strno: usize, num: u64, frm: Ref<NAFrame>) { base1 += stride1; base2 += stride2; } + if has_alpha { + let ls = buf.get_stride(3); + let mut idx = buf.get_offset(3); + let mut idx2 = idx + w; + for _ in 0..h { + let line = &dta[idx..idx2]; + ofile.write_all(line).unwrap(); + idx += ls; + idx2 += ls; + } + } } pub fn write_palppm(pfx: &str, strno: usize, num: u64, frm: Ref<NAFrame>) { |