projects / nihav.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
fix clippy errors
[nihav.git] / nihav-duck / src / codecs / truemotion2.rs
CommitLineData
8bde0eae
KS		 1use nihav_core::codecs::*;
2use nihav_core::io::byteio::*;
3use nihav_core::io::bitreader::*;
4use nihav_core::io::codebook::*;
5
6#[repr(u8)]
7enum TM2StreamType {
8 CHigh = 0,
9 CLow,
10 LHigh,
11 LLow,
12 Update,
13 Motion,
14 BlockType,
15 Num
16}
17
18#[repr(u8)]
19#[derive(Debug,Clone,Copy)]
20enum TM2BlockType {
21 HiRes,
22 MedRes,
23 LowRes,
24 NullRes,
25 Update,
26 Still,
27 Motion
28}
29
30const TM2_BLOCK_TYPES: [TM2BlockType; 7] = [
31 TM2BlockType::HiRes, TM2BlockType::MedRes, TM2BlockType::LowRes, TM2BlockType::NullRes,
32 TM2BlockType::Update, TM2BlockType::Still, TM2BlockType::Motion
33];
34
35trait ReadLenEsc {
36 fn read_len_esc(&mut self) -> DecoderResult<usize>;
37}
38
39const TM2_ESCAPE: usize = 0x80000000;
40
41impl<'a> ReadLenEsc for ByteReader<'a> {
42 fn read_len_esc(&mut self) -> DecoderResult<usize> {
43 let len = self.read_u32le()? as usize;
44 if len == TM2_ESCAPE {
45 let len2 = self.read_u32le()? as usize;
46 Ok(len2)
47 } else {
48 Ok(len)
49 }
50 }
51}
52
53struct HuffDef {
54 val_bits: u8,
55 max_bits: u8,
56 nelems: usize,
57}
58
59impl HuffDef {
60 fn read(&mut self, br: &mut BitReader, codes: &mut Vec<FullCodebookDesc<u8>>, prefix: u32, len: u8) -> DecoderResult<()> {
61 validate!(len <= self.max_bits);
62 if !br.read_bool()? {
63 validate!(codes.len() < self.nelems);
64 let sym = br.read(self.val_bits)? as u8;
65 codes.push(FullCodebookDesc { code: prefix, bits: len, sym });
66 } else {
67 self.read(br, codes, (prefix << 1) | 0, len + 1)?;
68 self.read(br, codes, (prefix << 1) | 1, len + 1)?;
69 }
70 Ok(())
71 }
72}
73
74struct HuffTree {
75 cb: Option<Codebook<u8>>,
76 sym0: u8,
77}
78
79impl HuffTree {
80 fn new() -> Self {
81 Self { cb: None, sym0: 0 }
82 }
83}
84
85const TM2_MAX_DELTAS: usize = 64;
86
87struct TM2Stream {
88 tokens: Vec<u8>,
89 deltas: [i32; TM2_MAX_DELTAS],
90 pos: usize,
91}
92
93impl Default for TM2Stream {
94 fn default() -> Self {
95 Self {
96 tokens: Vec::new(),
97 deltas: [0; TM2_MAX_DELTAS],
98 pos: 0,
99 }
100 }
101}
102
103impl TM2Stream {
104 fn read_header(&mut self, src: &[u8], br: &mut ByteReader) -> DecoderResult<()> {
105 self.tokens.truncate(0);
106 self.pos = 0;
107
108 let len = br.read_u32le()? as usize;
109 let endpos = br.tell() + (len as u64) * 4;
110 if len == 0 {
111 return Ok(());
112 }
113 let ntoks = br.read_u32le()? as usize;
114 validate!(ntoks < (1 << 24));
115 if (ntoks & 1) != 0 {
116 let dlen = br.read_len_esc()?;
117 if (dlen as i32) > 0 {
118 let rest_size = (endpos - br.tell()) as usize;
119 let skip_size = self.read_deltas(&src[br.tell() as usize..][..rest_size])?;
120 validate!(skip_size == dlen * 4);
121 br.read_skip(skip_size)?;
122 }
123 }
124 let _len = br.read_len_esc()?;
125 let _algo = br.read_u32le()?;
126
127 let mut htree = HuffTree::new();
128 let rest_size = (endpos - br.tell()) as usize;
129 let skip_size = self.read_huff_tree(&src[br.tell() as usize..][..rest_size], &mut htree)?;
130 br.read_skip(skip_size)?;
131
132 let len = br.read_u32le()? as usize;
133 validate!(br.tell() + (len as u64) * 4 <= endpos);
134 if len > 0 {
135 self.tokens.reserve(ntoks >> 1);
136 let rest_size = (endpos - br.tell()) as usize;
137 let skip_size = self.read_tokens(&src[br.tell() as usize..][..rest_size], &htree, ntoks >> 1)?;
138 br.read_skip(skip_size)?;
139 } else {
140 self.tokens.resize(ntoks >> 1, htree.sym0);
141 }
142
143
144 let pos = br.tell();
145 validate!(pos <= endpos);
146 let toskip = endpos - pos;
147 br.read_skip(toskip as usize)?;
148
149 Ok(())
150 }
151 fn read_deltas(&mut self, src: &[u8]) -> DecoderResult<usize> {
152 let mut br = BitReader::new(src, src.len(), BitReaderMode::LE32MSB);
153 let coded_deltas = br.read(9)? as usize;
154 let bits = br.read(5)? as u8;
155 validate!((coded_deltas <= TM2_MAX_DELTAS) && (bits > 0));
156 let mask = 1 << (bits - 1);
157 let bias = 1 << bits;
158 self.deltas = [0; TM2_MAX_DELTAS];
159 for i in 0..coded_deltas {
160 let val = br.read(bits)?;
161 if (val & mask) != 0 {
162 self.deltas[i] = (val as i32) - bias;
163 } else {
164 self.deltas[i] = val as i32;
165 }
166 }
167
168 Ok(((br.tell() + 31) >> 5) << 2)
169 }
170 fn read_huff_tree(&mut self, src: &[u8], htree: &mut HuffTree) -> DecoderResult<usize> {
171 let mut br = BitReader::new(src, src.len(), BitReaderMode::LE32MSB);
172
173 let val_bits = br.read(5)? as u8;
174 let max_bits = br.read(5)? as u8;
175 let min_bits = br.read(5)? as u8;
176 let nelems = br.read(17)? as usize;
177 validate!(val_bits > 0 && val_bits <= 6);
178 validate!(nelems > 0);
179 validate!((max_bits < 25) && (min_bits <= max_bits));
180
181 let mut codes: Vec<FullCodebookDesc<u8>> = Vec::with_capacity(nelems);
182 let mut hdef = HuffDef { val_bits, max_bits, nelems };
183 hdef.read(&mut br, &mut codes, 0, 0)?;
184 htree.sym0 = codes[0].sym;
185 if nelems > 1 {
186 let mut cr = FullCodebookDescReader::new(codes);
187 htree.cb = Some(Codebook::new(&mut cr, CodebookMode::MSB)?);
188 }
189
190 Ok(((br.tell() + 31) >> 5) << 2)
191 }
192 fn read_tokens(&mut self, src: &[u8], htree: &HuffTree, ntoks: usize) -> DecoderResult<usize> {
193 let mut br = BitReader::new(src, src.len(), BitReaderMode::LE32MSB);
194
195 if let Some(ref cb) = htree.cb {
196 for _ in 0..ntoks {
197 let tok = br.read_cb(cb)?;
198 self.tokens.push(tok);
199 }
200 }
201
202 Ok(((br.tell() + 31) >> 5) << 2)
203 }
204
205 fn get_block_type(&mut self) -> DecoderResult<u8> {
206 validate!(self.pos < self.tokens.len());
207 let res = self.tokens[self.pos];
208 self.pos += 1;
209 Ok(res)
210 }
211 fn get_token(&mut self) -> DecoderResult<i32> {
212 validate!(self.pos < self.tokens.len());
213 let idx = self.tokens[self.pos] as usize;
214 validate!(idx < TM2_MAX_DELTAS);
215 self.pos += 1;
216 Ok(self.deltas[idx])
217 }
218}
219
220#[derive(Default)]
221struct DeltaState {
222 dy: [i32; 4],
223 dc: [[i32; 2]; 2],
224}
225
c2a4fa57 226#[allow(clippy::erasing_op)]
8bde0eae
KS		 227impl DeltaState {
228 fn apply_y(&mut self, dst: &mut [u8], mut yoff: usize, ystride: usize, ydeltas: &[i32; 16], last: &mut [i32]) {
229 for y in 0..4 {
230 let mut d = self.dy[y];
231 for x in 0..4 {
232 d += ydeltas[x + y * 4];
233 last[x] += d;
234 dst[yoff + x] = last[x].max(0).min(255) as u8;
235 }
236 self.dy[y] = d;
237 yoff += ystride;
238 }
239 }
240 fn apply_c(&mut self, dst: &mut [i16], mut coff: usize, cstride: usize, cdeltas: &[i32; 4], idx: usize, last: &mut [i32]) {
241 for y in 0..2 {
242 let mut d = self.dc[idx][y];
243 for x in 0..2 {
244 d += cdeltas[x + y * 2];
245 last[x] += d;
246 dst[coff + x] = last[x] as i16;
247 }
248 self.dc[idx][y] = d;
249 coff += cstride;
250 }
251 }
252 fn interpolate_y_low(&mut self, last: &mut [i32]) {
253 let dsum = self.dy[0] + self.dy[1] + self.dy[2] + self.dy[3];
254 last[1] = (last[0] - dsum + last[2]) >> 1;
255 last[3] = (last[2] + last[4]) >> 1;
256
257 let t0 = self.dy[0] + self.dy[1];
258 let t1 = self.dy[2] + self.dy[3];
259 self.dy[0] = t0 >> 1;
260 self.dy[1] = t0 - (t0 >> 1);
261 self.dy[2] = t1 >> 1;
262 self.dy[3] = t1 - (t1 >> 1);
263 }
264 fn interpolate_y_null(&mut self, last: &mut [i32]) {
265 let dsum = self.dy[0] + self.dy[1] + self.dy[2] + self.dy[3];
266 let left = last[0] - dsum;
267 let right = last[4];
268 let diff = right - left;
269 last[1] = left + (diff >> 2);
270 last[2] = left + (diff >> 1);
271 last[3] = right - (diff >> 2);
272
273 let mut sum = left;
274 self.dy[0] = (left + (dsum >> 2)) - sum;
275 sum += self.dy[0];
276 self.dy[1] = (left + (dsum >> 1)) - sum;
277 sum += self.dy[1];
278 self.dy[2] = (left + dsum - (dsum >> 2)) - sum;
279 sum += self.dy[2];
280 self.dy[3] = (left + dsum) - sum;
281 }
282 fn interpolate_c(&mut self, idx: usize, last: &mut [i32]) {
283 let dsum = self.dc[idx][0] + self.dc[idx][1];
284 let l = (last[0] + last[2] - dsum) >> 1;
285 self.dc[idx][0] = dsum >> 1;
286 self.dc[idx][1] = dsum - (dsum >> 1);
287 last[1] = l;
288 }
289 fn recalc_y(&mut self, dst: &[u8], yoff: usize, ystride: usize, last: &mut [i32]) {
290 let src = &dst[yoff+3..];
291 self.dy[0] = (src[ystride * 0] as i32) - last[3];
292 self.dy[1] = (src[ystride * 1] as i32) - (src[ystride * 0] as i32);
293 self.dy[2] = (src[ystride * 2] as i32) - (src[ystride * 1] as i32);
294 self.dy[3] = (src[ystride * 3] as i32) - (src[ystride * 2] as i32);
295 let src = &dst[yoff + 3 * ystride..];
296 for x in 0..4 {
297 last[x] = src[x] as i32;
298 }
299 }
300 fn recalc_c(&mut self, dst: &[i16], coff: usize, cstride: usize, idx: usize, last: &mut [i32]) {
301 self.dc[idx][0] = (dst[coff + 1] as i32) - last[1];
302 self.dc[idx][1] = (dst[coff + 1 + cstride] as i32) - (dst[coff + 1] as i32);
303 last[0] = dst[coff + cstride + 0] as i32;
304 last[1] = dst[coff + cstride + 1] as i32;
305 }
306}
307
308#[derive(Default)]
309struct TM2Frame {
310 ydata: Vec<u8>,
311 udata: Vec<i16>,
312 vdata: Vec<i16>,
313 ystride: usize,
314 cstride: usize,
315}
316
317impl TM2Frame {
318 fn alloc(width: usize, height: usize) -> Self {
319 let ystride = (width + 3) & !3;
320 let ysize = ystride * ((height + 3) & !3);
321 let mut ydata = Vec::with_capacity(ysize);
322 ydata.resize(ysize, 0);
323 let cstride = ystride >> 1;
324 let csize = cstride * (((height + 3) & !3) >> 1);
325 let mut udata = Vec::with_capacity(csize);
326 udata.resize(csize, 0);
327 let mut vdata = Vec::with_capacity(csize);
328 vdata.resize(csize, 0);
329 Self { ydata, udata, vdata, ystride, cstride }
330 }
331}
332
333#[derive(Default)]
334struct TM2Decoder {
2422d969 335 info: NACodecInfoRef,
8bde0eae
KS		 336 streams: [TM2Stream; TM2StreamType::Num as usize],
337 width: usize,
338 height: usize,
339 cur_frame: TM2Frame,
340 prev_frame: TM2Frame,
341}
342
343impl TM2Decoder {
344 fn new() -> Self { Self::default() }
345 fn decode_blocks(&mut self) -> DecoderResult<bool> {
346 let ydst = &mut self.cur_frame.ydata;
347 let udst = &mut self.cur_frame.udata;
348 let vdst = &mut self.cur_frame.vdata;
349 let ystride = self.cur_frame.ystride;
350 let cstride = self.cur_frame.cstride;
351 let mut offs: [usize; 2] = [0; 2];
352 let mut is_intra = true;
353
354 let bw = self.width >> 2;
355 let bh = self.height >> 2;
356 validate!(self.streams[TM2StreamType::BlockType as usize].tokens.len() == bw * bh);
357
358 let mut ydeltas: [i32; 16] = [0; 16];
359 let mut cdeltas: [[i32; 4]; 2] = [[0; 4]; 2];
360 let mut lasty: Vec<i32> = Vec::with_capacity(self.width + 1);
361 lasty.resize(self.width + 1, 0);
362 let mut lastu: Vec<i32> = Vec::with_capacity(self.width/2 + 1);
363 lastu.resize(self.width/2 + 1, 0);
364 let mut lastv: Vec<i32> = Vec::with_capacity(self.width/2 + 1);
365 lastv.resize(self.width/2 + 1, 0);
366 for by in 0..bh {
367 let mut dstate = DeltaState::default();
368 for bx in 0..bw {
369 let bidx = self.streams[TM2StreamType::BlockType as usize].get_block_type()? as usize;
370 validate!(bidx < TM2_BLOCK_TYPES.len());
371 let btype = TM2_BLOCK_TYPES[bidx];
372 match btype {
373 TM2BlockType::HiRes => {
374 for i in 0..4 {
375 cdeltas[0][i] = self.streams[TM2StreamType::CHigh as usize].get_token()?;
376 cdeltas[1][i] = self.streams[TM2StreamType::CHigh as usize].get_token()?;
377 }
378 dstate.apply_c(udst, offs[1] + bx * 2, cstride, &cdeltas[0], 0, &mut lastu[bx*2+1..]);
379 dstate.apply_c(vdst, offs[1] + bx * 2, cstride, &cdeltas[1], 1, &mut lastv[bx*2+1..]);
380 for i in 0..4*4 {
381 ydeltas[i] = self.streams[TM2StreamType::LHigh as usize].get_token()?;
382 }
383 dstate.apply_y(ydst, offs[0] + bx * 4, ystride, &ydeltas, &mut lasty[bx*4+1..]);
384 },
385 TM2BlockType::MedRes => {
386 cdeltas = [[0; 4]; 2];
387 cdeltas[0][0] = self.streams[TM2StreamType::CLow as usize].get_token()?;
388 cdeltas[1][0] = self.streams[TM2StreamType::CLow as usize].get_token()?;
389 dstate.interpolate_c(0, &mut lastu[bx*2..]);
390 dstate.apply_c(udst, offs[1] + bx * 2, cstride, &cdeltas[0], 0, &mut lastu[bx*2+1..]);
391 dstate.interpolate_c(1, &mut lastv[bx*2..]);
392 dstate.apply_c(vdst, offs[1] + bx * 2, cstride, &cdeltas[1], 1, &mut lastv[bx*2+1..]);
393 for i in 0..4*4 {
394 ydeltas[i] = self.streams[TM2StreamType::LHigh as usize].get_token()?;
395 }
396 dstate.apply_y(ydst, offs[0] + bx * 4, ystride, &ydeltas, &mut lasty[bx*4+1..]);
397 },
398 TM2BlockType::LowRes => {
399 cdeltas = [[0; 4]; 2];
400 cdeltas[0][0] = self.streams[TM2StreamType::CLow as usize].get_token()?;
401 cdeltas[1][0] = self.streams[TM2StreamType::CLow as usize].get_token()?;
402 dstate.interpolate_c(0, &mut lastu[bx*2..]);
403 dstate.apply_c(udst, offs[1] + bx * 2, cstride, &cdeltas[0], 0, &mut lastu[bx*2+1..]);
404 dstate.interpolate_c(1, &mut lastv[bx*2..]);
405 dstate.apply_c(vdst, offs[1] + bx * 2, cstride, &cdeltas[1], 1, &mut lastv[bx*2+1..]);
406 ydeltas = [0; 16];
407 ydeltas[ 0] = self.streams[TM2StreamType::LLow as usize].get_token()?;
408 ydeltas[ 2] = self.streams[TM2StreamType::LLow as usize].get_token()?;
409 ydeltas[ 8] = self.streams[TM2StreamType::LLow as usize].get_token()?;
410 ydeltas[10] = self.streams[TM2StreamType::LLow as usize].get_token()?;
411 dstate.interpolate_y_low(&mut lasty[bx*4..]);
412 dstate.apply_y(ydst, offs[0] + bx * 4, ystride, &ydeltas, &mut lasty[bx*4+1..]);
413 },
414 TM2BlockType::NullRes => {
415 cdeltas = [[0; 4]; 2];
416 dstate.interpolate_c(0, &mut lastu[bx*2..]);
417 dstate.apply_c(udst, offs[1] + bx * 2, cstride, &cdeltas[0], 0, &mut lastu[bx*2+1..]);
418 dstate.interpolate_c(1, &mut lastv[bx*2..]);
419 dstate.apply_c(vdst, offs[1] + bx * 2, cstride, &cdeltas[1], 1, &mut lastv[bx*2+1..]);
420 ydeltas = [0; 16];
421 dstate.interpolate_y_null(&mut lasty[bx*4..]);
422 dstate.apply_y(ydst, offs[0] + bx * 4, ystride, &ydeltas, &mut lasty[bx*4+1..]);
423 },
424 TM2BlockType::Update => {
425 is_intra = false;
426
427 let mut coff = offs[1] + bx * 2;
428 let usrc = &self.prev_frame.udata;
429 let vsrc = &self.prev_frame.vdata;
430 for _ in 0..2 {
431 for x in 0..2 {
432 let du = self.streams[TM2StreamType::Update as usize].get_token()?;
433 let dv = self.streams[TM2StreamType::Update as usize].get_token()?;
434 udst[coff + x] = usrc[coff + x] + (du as i16);
435 vdst[coff + x] = vsrc[coff + x] + (dv as i16);
436 }
437 coff += cstride;
438 }
439 dstate.recalc_c(udst, offs[1] + bx * 2, cstride, 0, &mut lastu[bx*2+1..]);
440 dstate.recalc_c(vdst, offs[1] + bx * 2, cstride, 1, &mut lastv[bx*2+1..]);
441 let mut yoff = offs[0] + bx * 4;
442 let ysrc = &self.prev_frame.ydata;
443 for _ in 0..4 {
444 for x in 0..4 {
445 let dy = self.streams[TM2StreamType::Update as usize].get_token()?;
446 ydst[yoff + x] = ((ysrc[yoff + x] as i32) + dy) as u8;
447 }
448 yoff += ystride;
449 }
450 dstate.recalc_y(ydst, offs[0] + bx * 4, ystride, &mut lasty[bx*4+1..]);
451 },
452 TM2BlockType::Still => {
453 is_intra = false;
454
455 let mut coff = offs[1] + bx * 2;
456 let usrc = &self.prev_frame.udata;
457 let vsrc = &self.prev_frame.vdata;
458 for _ in 0..2 {
459 for x in 0..2 {
460 udst[coff + x] = usrc[coff + x];
461 vdst[coff + x] = vsrc[coff + x];
462 }
463 coff += cstride;
464 }
465 dstate.recalc_c(udst, offs[1] + bx * 2, cstride, 0, &mut lastu[bx*2+1..]);
466 dstate.recalc_c(vdst, offs[1] + bx * 2, cstride, 1, &mut lastv[bx*2+1..]);
467 let mut yoff = offs[0] + bx * 4;
468 let ysrc = &self.prev_frame.ydata;
469 for _ in 0..4 {
470 for x in 0..4 {
471 ydst[yoff + x] = ysrc[yoff + x];
472 }
473 yoff += ystride;
474 }
475 dstate.recalc_y(ydst, offs[0] + bx * 4, ystride, &mut lasty[bx*4+1..]);
476 },
477 TM2BlockType::Motion => {
478 is_intra = false;
479
480 let mx = self.streams[TM2StreamType::Motion as usize].get_token()?;
481 let my = self.streams[TM2StreamType::Motion as usize].get_token()?;
482 let xpos = (((bx as i32) * 4) + mx).max(0).min((self.width - 4) as i32) as usize;
483 let ypos = (((by as i32) * 4) + my).max(0).min((self.height - 4) as i32) as usize;
484 let mut coff = offs[1] + bx * 2;
485 let mut csoff = (xpos >> 1) + (ypos >> 1) * cstride;
486 let usrc = &self.prev_frame.udata;
487 let vsrc = &self.prev_frame.vdata;
488 for _ in 0..2 {
489 for x in 0..2 {
490 udst[coff + x] = usrc[csoff + x];
491 vdst[coff + x] = vsrc[csoff + x];
492 }
493 coff += cstride;
494 csoff += cstride;
495 }
496 dstate.recalc_c(udst, offs[1] + bx * 2, cstride, 0, &mut lastu[bx*2+1..]);
497 dstate.recalc_c(vdst, offs[1] + bx * 2, cstride, 1, &mut lastv[bx*2+1..]);
498 let mut yoff = offs[0] + bx * 4;
499 let mut ysoff = xpos + ypos * ystride;
500 let ysrc = &self.prev_frame.ydata;
501 for _ in 0..4 {
502 for x in 0..4 {
503 ydst[yoff + x] = ysrc[ysoff + x];
504 }
505 yoff += ystride;
506 ysoff += ystride;
507 }
508 dstate.recalc_y(ydst, offs[0] + bx * 4, ystride, &mut lasty[bx*4+1..]);
509 },
510 };
511 }
512 offs[0] += ystride * 4;
513 offs[1] += cstride * 2;
514 }
515
516 Ok(is_intra)
517 }
518 fn output_frame(&mut self, buf: &mut NAVideoBuffer<u8>) {
519 let fmt = buf.get_info().get_format();
520 let offs = [fmt.get_chromaton(0).unwrap().get_offset() as usize,
521 fmt.get_chromaton(1).unwrap().get_offset() as usize,
522 fmt.get_chromaton(2).unwrap().get_offset() as usize];
523 let stride = buf.get_stride(0);
1a967e6b 524 let data = buf.get_data_mut().unwrap();
8bde0eae
KS		 525 let dst = data.as_mut_slice();
526
527 let mut off = 0;
528 let mut ysrc = 0;
529 let mut csrc = 0;
530 for y in 0..self.height {
531 let out = &mut dst[off..];
532 for (x, pic) in out.chunks_exact_mut(3).take(self.width).enumerate() {
533 let y = self.cur_frame.ydata[ysrc + x] as i16;
534 let u = self.cur_frame.udata[csrc + (x >> 1)];
535 let v = self.cur_frame.vdata[csrc + (x >> 1)];
536 pic[offs[0]] = (y + u).max(0).min(255) as u8;
537 pic[offs[1]] = y.max(0).min(255) as u8;
538 pic[offs[2]] = (y + v).max(0).min(255) as u8;
539 }
540 off += stride;
541 ysrc += self.cur_frame.ystride;
542 if (y & 1) != 0 {
543 csrc += self.cur_frame.cstride;
544 }
545 }
546 }
547}
548
549impl NADecoder for TM2Decoder {
01613464 550 fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
8bde0eae
KS		 551 if let NACodecTypeInfo::Video(vinfo) = info.get_properties() {
552 let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(vinfo.get_width(), vinfo.get_height(), false, YUV410_FORMAT));
553 self.width = vinfo.get_width();
554 self.height = vinfo.get_height();
555 self.cur_frame = TM2Frame::alloc(self.width, self.height);
556 self.prev_frame = TM2Frame::alloc(self.width, self.height);
2422d969 557 self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();
8bde0eae
KS		 558 Ok(())
559 } else {
560 Err(DecoderError::InvalidData)
561 }
562 }
01613464 563 fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
8bde0eae
KS		 564 let src = pkt.get_buffer();
565 validate!(src.len() >= 40 + (TM2StreamType::Num as usize) * 4 + 4);
566 let mut mr = MemoryReader::new_read(&src);
567 let mut br = ByteReader::new(&mut mr);
568
569 let magic = br.read_u32be()?;
570 validate!(magic == 0x100 || magic == 0x101);
571 br.read_skip(36)?;
572 for str in self.streams.iter_mut() {
573 str.read_header(&src, &mut br)?;
574 }
575
576 let myinfo = NAVideoInfo::new(self.width, self.height, false, RGB24_FORMAT);
577 let bufret = alloc_video_buffer(myinfo, 2);
578 if let Err(_) = bufret { return Err(DecoderError::InvalidData); }
b70cc006 579 let bufinfo = bufret.unwrap();
8bde0eae
KS		 580 let mut buf = bufinfo.get_vbuf().unwrap();
581
582 let is_intra = self.decode_blocks()?;
583 self.output_frame(&mut buf);
584 std::mem::swap(&mut self.cur_frame, &mut self.prev_frame);
585
586 let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo);
587 frm.set_keyframe(is_intra);
588 frm.set_frame_type(if is_intra { FrameType::I } else { FrameType::P });
171860fc 589 Ok(frm.into_ref())
8bde0eae
KS		 590 }
591}
592
6011e201 593pub fn get_decoder() -> Box<dyn NADecoder> {
8bde0eae
KS		 594 Box::new(TM2Decoder::new())
595}
596
597#[cfg(test)]
598mod test {
599 use nihav_core::codecs::RegisteredDecoders;
600 use nihav_core::demuxers::RegisteredDemuxers;
601 use nihav_core::test::dec_video::*;
602 use crate::codecs::duck_register_all_codecs;
603 use nihav_commonfmt::demuxers::generic_register_all_demuxers;
604 #[test]
605 fn test_tm2() {
606 let mut dmx_reg = RegisteredDemuxers::new();
607 generic_register_all_demuxers(&mut dmx_reg);
608 let mut dec_reg = RegisteredDecoders::new();
609 duck_register_all_codecs(&mut dec_reg);
610
611 test_file_decoding("avi", "assets/Duck/tm20.avi", Some(16), true, false, None/*Some("tm2")*/, &dmx_reg, &dec_reg);
612 }
613}
NihAV is a research multimedia framework