1 use nihav_core::muxers::*;
4 bw: &'a mut ByteWriter<'a>,
10 impl<'a> EAMuxer<'a> {
11 fn new(bw: &'a mut ByteWriter<'a>) -> Self {
21 impl<'a> MuxCore<'a> for EAMuxer<'a> {
22 #[allow(clippy::unreadable_literal)]
23 #[allow(clippy::cast_lossless)]
24 fn create(&mut self, strmgr: &StreamManager) -> MuxerResult<()> {
25 if strmgr.get_num_streams() == 0 {
26 return Err(MuxerError::InvalidArgument);
30 for stream in strmgr.iter() {
31 if stream.get_media_type() == StreamType::Video {
32 if stream.get_info().get_name() != "vp6" {
33 return Err(MuxerError::UnsupportedFormat);
37 return Err(MuxerError::UnsupportedFormat);
40 if nvideo == 0 || nvideo > 2 {
41 return Err(MuxerError::UnsupportedFormat);
43 self.has_alpha = nvideo == 2;
46 self.bw.write_buf(b"AVP6\x08\x00\x00\x00")?;
48 for (str_no, stream) in strmgr.iter().enumerate() {
49 if let NACodecTypeInfo::Video(ref vinfo) = stream.get_info().get_properties() {
50 let tag = if str_no == 0 { b"MVhd" } else { b"AVhd" };
52 self.bw.write_buf(tag)?;
53 self.bw.write_u32le(0x20)?;
54 self.bw.write_buf(b"vp60")?;
55 self.bw.write_u16le(vinfo.width as u16)?;
56 self.bw.write_u16le(vinfo.height as u16)?;
57 self.bw.write_u32le(0)?;
58 self.bw.write_u32le(0)?;
59 self.bw.write_u32le(stream.tb_den)?;
60 self.bw.write_u32le(stream.tb_num)?;
68 #[allow(clippy::collapsible_else_if)]
69 fn mux_frame(&mut self, _strmgr: &StreamManager, pkt: NAPacket) -> MuxerResult<()> {
70 let stream = pkt.get_stream();
71 let str_num = stream.get_num();
73 return Err(MuxerError::UnsupportedFormat);
76 let chunk_len = pkt.get_buffer().len() as u32;
78 let tag = if pkt.is_keyframe() {
79 if str_num == 0 { b"MV0K" } else { b"AV0K" }
81 if str_num == 0 { b"MV0F" } else { b"AV0F" }
83 self.max_size[str_num] = self.max_size[str_num].max(pkt.get_buffer().len() as u32);
84 self.bw.write_buf(tag)?;
85 self.bw.write_u32le(chunk_len + 8)?;
86 self.bw.write_buf(&pkt.get_buffer())?;
94 fn flush(&mut self) -> MuxerResult<()> {
97 fn end(&mut self) -> MuxerResult<()> {
99 self.bw.seek(SeekFrom::Start(0x10))?;
100 self.bw.write_u32le(self.nframes)?;
101 self.bw.write_u32le(self.max_size[0])?;
103 self.bw.seek(SeekFrom::Start(0x18))?;
104 self.bw.write_u32le(self.nframes)?;
105 self.bw.write_u32le(self.max_size[0])?;
106 self.bw.seek(SeekFrom::Start(0x38))?;
107 self.bw.write_u32le(self.nframes)?;
108 self.bw.write_u32le(self.max_size[1])?;
114 impl<'a> NAOptionHandler for EAMuxer<'a> {
115 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
116 fn set_options(&mut self, _options: &[NAOption]) { }
117 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
120 pub struct EAMuxerCreator {}
122 impl MuxerCreator for EAMuxerCreator {
123 fn new_muxer<'a>(&self, bw: &'a mut ByteWriter<'a>) -> Box<dyn MuxCore<'a> + 'a> {
124 Box::new(EAMuxer::new(bw))
126 fn get_name(&self) -> &'static str { "ea" }
127 fn get_capabilities(&self) -> MuxerCapabilities { MuxerCapabilities::OnlyVideo }
132 use nihav_core::codecs::*;
133 use nihav_core::demuxers::*;
134 use nihav_core::muxers::*;
135 use nihav_codec_support::test::enc_video::*;
140 let mut dmx_reg = RegisteredDemuxers::new();
141 nihav_commonfmt::generic_register_all_demuxers(&mut dmx_reg);
142 let dec_config = DecoderTestParams {
144 in_name: "assets/Duck/vp6_crash.avi",
146 stream_type: StreamType::None,
147 dmx_reg, dec_reg: RegisteredDecoders::new(),
149 let mut mux_reg = RegisteredMuxers::new();
150 game_register_all_muxers(&mut mux_reg);
151 /*let enc_config = EncoderTestParams {
154 out_name: "muxed.ea",
155 mux_reg, enc_reg: RegisteredEncoders::new(),
157 test_remuxing(&dec_config, &enc_config);*/
158 test_remuxing_md5(&dec_config, "ea", &mux_reg,
159 [0xc8c6484d, 0x863de1ae, 0x97a38a31, 0x59e2a7ef]);