rav1e 0.4.0-alpha

The fastest and safest AV1 encoder
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

// Copyright (c) 2019, The rav1e contributors. All rights reserved
//
// This source code is subject to the terms of the BSD 2 Clause License and
// the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
// was not distributed with this source code in the LICENSE file, you can
// obtain it at www.aomedia.org/license/software. If the Alliance for Open
// Media Patent License 1.0 was not distributed with this source code in the
// PATENTS file, you can obtain it at www.aomedia.org/license/patent.

use crate::partition::BlockSize;
use crate::predict::PREDICTION_MODES;
use crate::serialize::{Deserialize, Serialize};
use crate::transform::TX_TYPES;
use arrayvec::ArrayVec;
use std::ops::{Add, AddAssign};

#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct EncoderStats {
  /// Stores count of pixels belonging to each block size in this frame
  pub block_size_counts: [usize; BlockSize::BLOCK_SIZES_ALL],
  /// Stores count of pixels belonging to skip blocks in this frame
  pub skip_block_count: usize,
  /// Stores count of pixels belonging to each transform type in this frame
  pub tx_type_counts: [usize; TX_TYPES],
  /// Stores count of pixels belonging to each luma prediction mode in this frame
  pub luma_pred_mode_counts: ArrayVec<[usize; PREDICTION_MODES]>,
  /// Stores count of pixels belonging to each chroma prediction mode in this frame
  pub chroma_pred_mode_counts: ArrayVec<[usize; PREDICTION_MODES]>,
}

impl Default for EncoderStats {
  fn default() -> Self {
    let mut luma_pred_mode_counts = ArrayVec::new();
    luma_pred_mode_counts
      .try_extend_from_slice(&[0; PREDICTION_MODES])
      .unwrap();
    let mut chroma_pred_mode_counts = ArrayVec::new();
    chroma_pred_mode_counts
      .try_extend_from_slice(&[0; PREDICTION_MODES])
      .unwrap();
    EncoderStats {
      block_size_counts: [0; BlockSize::BLOCK_SIZES_ALL],
      skip_block_count: 0,
      tx_type_counts: [0; TX_TYPES],
      luma_pred_mode_counts,
      chroma_pred_mode_counts,
    }
  }
}

impl Add<&Self> for EncoderStats {
  type Output = Self;

  fn add(self, rhs: &EncoderStats) -> Self::Output {
    let mut lhs = self;
    lhs += rhs;
    lhs
  }
}

impl AddAssign<&Self> for EncoderStats {
  fn add_assign(&mut self, rhs: &EncoderStats) {
    for (s, v) in
      self.block_size_counts.iter_mut().zip(rhs.block_size_counts.iter())
    {
      *s += v;
    }
    for (s, v) in self
      .chroma_pred_mode_counts
      .iter_mut()
      .zip(rhs.chroma_pred_mode_counts.iter())
    {
      *s += v;
    }
    for (s, v) in self
      .luma_pred_mode_counts
      .iter_mut()
      .zip(rhs.luma_pred_mode_counts.iter())
    {
      *s += v;
    }
    for (s, v) in self.tx_type_counts.iter_mut().zip(rhs.tx_type_counts.iter())
    {
      *s += v;
    }
    self.skip_block_count += rhs.skip_block_count;
  }
}