AOMedia AV1 Codec
encoder.h
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2016, Alliance for Open Media. All rights reserved
3  *
4  * This source code is subject to the terms of the BSD 2 Clause License and
5  * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
6  * was not distributed with this source code in the LICENSE file, you can
7  * obtain it at www.aomedia.org/license/software. If the Alliance for Open
8  * Media Patent License 1.0 was not distributed with this source code in the
9  * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
10  */
11 
15 #ifndef AOM_AV1_ENCODER_ENCODER_H_
16 #define AOM_AV1_ENCODER_ENCODER_H_
17 
18 #include <stdbool.h>
19 #include <stdio.h>
20 
21 #include "config/aom_config.h"
22 
23 #include "aom/aomcx.h"
24 
25 #include "av1/common/alloccommon.h"
26 #include "av1/common/av1_common_int.h"
27 #include "av1/common/blockd.h"
28 #include "av1/common/entropymode.h"
29 #include "av1/common/enums.h"
30 #include "av1/common/reconintra.h"
31 #include "av1/common/resize.h"
32 #include "av1/common/thread_common.h"
33 #include "av1/common/timing.h"
34 
35 #include "av1/encoder/aq_cyclicrefresh.h"
36 #include "av1/encoder/av1_quantize.h"
37 #include "av1/encoder/block.h"
38 #include "av1/encoder/context_tree.h"
39 #include "av1/encoder/encodemb.h"
40 #include "av1/encoder/external_partition.h"
41 #include "av1/encoder/firstpass.h"
42 #include "av1/encoder/global_motion.h"
43 #include "av1/encoder/level.h"
44 #include "av1/encoder/lookahead.h"
45 #include "av1/encoder/mcomp.h"
46 #include "av1/encoder/pickcdef.h"
47 #include "av1/encoder/ratectrl.h"
48 #include "av1/encoder/rd.h"
50 #include "av1/encoder/svc_layercontext.h"
51 #include "av1/encoder/temporal_filter.h"
52 #include "av1/encoder/thirdpass.h"
53 #include "av1/encoder/tokenize.h"
54 #include "av1/encoder/tpl_model.h"
55 #include "av1/encoder/av1_noise_estimate.h"
56 #include "av1/encoder/bitstream.h"
57 
58 #if CONFIG_INTERNAL_STATS
59 #include "aom_dsp/ssim.h"
60 #endif
61 #include "aom_dsp/variance.h"
62 #if CONFIG_DENOISE
63 #include "aom_dsp/noise_model.h"
64 #endif
65 #if CONFIG_TUNE_VMAF
66 #include "av1/encoder/tune_vmaf.h"
67 #endif
68 #if CONFIG_AV1_TEMPORAL_DENOISING
69 #include "av1/encoder/av1_temporal_denoiser.h"
70 #endif
71 #if CONFIG_TUNE_BUTTERAUGLI
72 #include "av1/encoder/tune_butteraugli.h"
73 #endif
74 
75 #include "aom/internal/aom_codec_internal.h"
76 #include "aom_util/aom_thread.h"
77 
78 #ifdef __cplusplus
79 extern "C" {
80 #endif
81 
82 // TODO(yunqing, any): Added suppression tag to quiet Doxygen warnings. Need to
83 // adjust it while we work on documentation.
85 // Number of frames required to test for scene cut detection
86 #define SCENE_CUT_KEY_TEST_INTERVAL 16
87 
88 // Lookahead index threshold to enable temporal filtering for second arf.
89 #define TF_LOOKAHEAD_IDX_THR 7
90 
91 #define HDR_QP_LEVELS 10
92 #define CHROMA_CB_QP_SCALE 1.04
93 #define CHROMA_CR_QP_SCALE 1.04
94 #define CHROMA_QP_SCALE -0.46
95 #define CHROMA_QP_OFFSET 9.26
96 #define QP_SCALE_FACTOR 2.0
97 #define DISABLE_HDR_LUMA_DELTAQ 1
98 
99 // Rational number with an int64 numerator
100 // This structure holds a fractional value
101 typedef struct aom_rational64 {
102  int64_t num; // fraction numerator
103  int den; // fraction denominator
104 } aom_rational64_t; // alias for struct aom_rational
105 
106 enum {
107  // Good Quality Fast Encoding. The encoder balances quality with the amount of
108  // time it takes to encode the output. Speed setting controls how fast.
109  GOOD,
110  // Realtime Fast Encoding. Will force some restrictions on bitrate
111  // constraints.
112  REALTIME,
113  // All intra mode. All the frames are coded as intra frames.
114  ALLINTRA
115 } UENUM1BYTE(MODE);
116 
117 enum {
118  FRAMEFLAGS_KEY = 1 << 0,
119  FRAMEFLAGS_GOLDEN = 1 << 1,
120  FRAMEFLAGS_BWDREF = 1 << 2,
121  // TODO(zoeliu): To determine whether a frame flag is needed for ALTREF2_FRAME
122  FRAMEFLAGS_ALTREF = 1 << 3,
123  FRAMEFLAGS_INTRAONLY = 1 << 4,
124  FRAMEFLAGS_SWITCH = 1 << 5,
125  FRAMEFLAGS_ERROR_RESILIENT = 1 << 6,
126 } UENUM1BYTE(FRAMETYPE_FLAGS);
127 
128 #if CONFIG_FPMT_TEST
129 enum {
130  PARALLEL_ENCODE = 0,
131  PARALLEL_SIMULATION_ENCODE,
132  NUM_FPMT_TEST_ENCODES
133 } UENUM1BYTE(FPMT_TEST_ENC_CFG);
134 #endif // CONFIG_FPMT_TEST
135 // 0 level frames are sometimes used for rate control purposes, but for
136 // reference mapping purposes, the minimum level should be 1.
137 #define MIN_PYR_LEVEL 1
138 static INLINE int get_true_pyr_level(int frame_level, int frame_order,
139  int max_layer_depth) {
140  if (frame_order == 0) {
141  // Keyframe case
142  return MIN_PYR_LEVEL;
143  } else if (frame_level == MAX_ARF_LAYERS) {
144  // Leaves
145  return max_layer_depth;
146  } else if (frame_level == (MAX_ARF_LAYERS + 1)) {
147  // Altrefs
148  return MIN_PYR_LEVEL;
149  }
150  return AOMMAX(MIN_PYR_LEVEL, frame_level);
151 }
152 
153 enum {
154  NO_AQ = 0,
155  VARIANCE_AQ = 1,
156  COMPLEXITY_AQ = 2,
157  CYCLIC_REFRESH_AQ = 3,
158  AQ_MODE_COUNT // This should always be the last member of the enum
159 } UENUM1BYTE(AQ_MODE);
160 enum {
161  NO_DELTA_Q = 0,
162  DELTA_Q_OBJECTIVE = 1, // Modulation to improve objective quality
163  DELTA_Q_PERCEPTUAL = 2, // Modulation to improve video perceptual quality
164  DELTA_Q_PERCEPTUAL_AI = 3, // Perceptual quality opt for all intra mode
165  DELTA_Q_USER_RATING_BASED = 4, // User rating based delta q mode
166  DELTA_Q_HDR = 5, // QP adjustment based on HDR block pixel average
167  DELTA_Q_MODE_COUNT // This should always be the last member of the enum
168 } UENUM1BYTE(DELTAQ_MODE);
169 
170 enum {
171  RESIZE_NONE = 0, // No frame resizing allowed.
172  RESIZE_FIXED = 1, // All frames are coded at the specified scale.
173  RESIZE_RANDOM = 2, // All frames are coded at a random scale.
174  RESIZE_DYNAMIC = 3, // Frames coded at lower scale based on rate control.
175  RESIZE_MODES
176 } UENUM1BYTE(RESIZE_MODE);
177 
178 enum {
179  SS_CFG_SRC = 0,
180  SS_CFG_LOOKAHEAD = 1,
181  SS_CFG_FPF = 2,
182  SS_CFG_TOTAL = 3
183 } UENUM1BYTE(SS_CFG_OFFSET);
184 
185 enum {
186  DISABLE_SCENECUT, // For LAP, lag_in_frames < 19
187  ENABLE_SCENECUT_MODE_1, // For LAP, lag_in_frames >=19 and < 33
188  ENABLE_SCENECUT_MODE_2 // For twopass and LAP - lag_in_frames >=33
189 } UENUM1BYTE(SCENECUT_MODE);
190 
191 #define MAX_VBR_CORPUS_COMPLEXITY 10000
192 
193 typedef enum {
194  MOD_FP, // First pass
195  MOD_TF, // Temporal filtering
196  MOD_TPL, // TPL
197  MOD_GME, // Global motion estimation
198  MOD_ENC, // Encode stage
199  MOD_LPF, // Deblocking loop filter
200  MOD_CDEF_SEARCH, // CDEF search
201  MOD_CDEF, // CDEF frame
202  MOD_LR, // Loop restoration filtering
203  MOD_PACK_BS, // Pack bitstream
204  MOD_FRAME_ENC, // Frame Parallel encode
205  MOD_AI, // All intra
206  NUM_MT_MODULES
207 } MULTI_THREADED_MODULES;
208 
217 typedef enum {
224 
228 typedef enum {
233  3,
235 
240 typedef enum {
241  SKIP_APPLY_RESTORATION = 1 << 0,
242  SKIP_APPLY_SUPERRES = 1 << 1,
243  SKIP_APPLY_CDEF = 1 << 2,
244  SKIP_APPLY_LOOPFILTER = 1 << 3,
246 
250 typedef struct {
254  RESIZE_MODE resize_mode;
265 } ResizeCfg;
266 
270 typedef struct {
287  BLOCK_SIZE min_partition_size;
292  BLOCK_SIZE max_partition_size;
293 } PartitionCfg;
294 
298 typedef struct {
348 } IntraModeCfg;
349 
353 typedef struct {
391 
395 typedef struct {
422 
426 typedef struct {
457 } SuperResCfg;
458 
462 typedef struct {
467 
472 
477 
483 
490 
494  bool auto_key;
495 
500 
505 
510 
515 } KeyFrameCfg;
516 
520 typedef struct {
522  // BUFFERING PARAMETERS
540 
545 
555  unsigned int max_intra_bitrate_pct;
560  unsigned int max_inter_bitrate_pct;
564  unsigned int gf_cbr_boost_pct;
569  unsigned int min_cr;
599  int cq_level;
604  enum aom_rc_mode mode;
611  int vbrbias;
623 
625 typedef struct {
626  // Indicates the number of frames lag before encoding is started.
627  int lag_in_frames;
628  // Indicates the minimum gf/arf interval to be used.
629  int min_gf_interval;
630  // Indicates the maximum gf/arf interval to be used.
631  int max_gf_interval;
632  // Indicates the minimum height for GF group pyramid structure to be used.
633  int gf_min_pyr_height;
634  // Indicates the maximum height for GF group pyramid structure to be used.
635  int gf_max_pyr_height;
636  // Indicates if automatic set and use of altref frames should be enabled.
637  bool enable_auto_arf;
638  // Indicates if automatic set and use of (b)ackward (r)ef (f)rames should be
639  // enabled.
640  bool enable_auto_brf;
641 } GFConfig;
642 
643 typedef struct {
644  // Indicates the number of tile groups.
645  unsigned int num_tile_groups;
646  // Indicates the MTU size for a tile group. If mtu is non-zero,
647  // num_tile_groups is set to DEFAULT_MAX_NUM_TG.
648  unsigned int mtu;
649  // Indicates the number of tile columns in log2.
650  int tile_columns;
651  // Indicates the number of tile rows in log2.
652  int tile_rows;
653  // Indicates the number of widths in the tile_widths[] array.
654  int tile_width_count;
655  // Indicates the number of heights in the tile_heights[] array.
656  int tile_height_count;
657  // Indicates the tile widths, and may be empty.
658  int tile_widths[MAX_TILE_COLS];
659  // Indicates the tile heights, and may be empty.
660  int tile_heights[MAX_TILE_ROWS];
661  // Indicates if large scale tile coding should be used.
662  bool enable_large_scale_tile;
663  // Indicates if single tile decoding mode should be enabled.
664  bool enable_single_tile_decoding;
665  // Indicates if EXT_TILE_DEBUG should be enabled.
666  bool enable_ext_tile_debug;
667 } TileConfig;
668 
669 typedef struct {
670  // Indicates the width of the input frame.
671  int width;
672  // Indicates the height of the input frame.
673  int height;
674  // If forced_max_frame_width is non-zero then it is used to force the maximum
675  // frame width written in write_sequence_header().
676  int forced_max_frame_width;
677  // If forced_max_frame_width is non-zero then it is used to force the maximum
678  // frame height written in write_sequence_header().
679  int forced_max_frame_height;
680  // Indicates the frame width after applying both super-resolution and resize
681  // to the coded frame.
682  int render_width;
683  // Indicates the frame height after applying both super-resolution and resize
684  // to the coded frame.
685  int render_height;
686 } FrameDimensionCfg;
687 
688 typedef struct {
689  // Indicates if warped motion should be enabled.
690  bool enable_warped_motion;
691  // Indicates if warped motion should be evaluated or not.
692  bool allow_warped_motion;
693  // Indicates if OBMC motion should be enabled.
694  bool enable_obmc;
695 } MotionModeCfg;
696 
697 typedef struct {
698  // Timing info for each frame.
699  aom_timing_info_t timing_info;
700  // Indicates the number of time units of a decoding clock.
701  uint32_t num_units_in_decoding_tick;
702  // Indicates if decoder model information is present in the coded sequence
703  // header.
704  bool decoder_model_info_present_flag;
705  // Indicates if display model information is present in the coded sequence
706  // header.
707  bool display_model_info_present_flag;
708  // Indicates if timing info for each frame is present.
709  bool timing_info_present;
710 } DecoderModelCfg;
711 
712 typedef struct {
713  // Indicates the update frequency for coeff costs.
714  COST_UPDATE_TYPE coeff;
715  // Indicates the update frequency for mode costs.
716  COST_UPDATE_TYPE mode;
717  // Indicates the update frequency for mv costs.
718  COST_UPDATE_TYPE mv;
719  // Indicates the update frequency for dv costs.
720  COST_UPDATE_TYPE dv;
721 } CostUpdateFreq;
722 
723 typedef struct {
724  // Indicates the maximum number of reference frames allowed per frame.
725  unsigned int max_reference_frames;
726  // Indicates if the reduced set of references should be enabled.
727  bool enable_reduced_reference_set;
728  // Indicates if one-sided compound should be enabled.
729  bool enable_onesided_comp;
730 } RefFrameCfg;
731 
732 typedef struct {
733  // Indicates the color space that should be used.
734  aom_color_primaries_t color_primaries;
735  // Indicates the characteristics of transfer function to be used.
736  aom_transfer_characteristics_t transfer_characteristics;
737  // Indicates the matrix coefficients to be used for the transfer function.
738  aom_matrix_coefficients_t matrix_coefficients;
739  // Indicates the chroma 4:2:0 sample position info.
740  aom_chroma_sample_position_t chroma_sample_position;
741  // Indicates if a limited color range or full color range should be used.
742  aom_color_range_t color_range;
743 } ColorCfg;
744 
745 typedef struct {
746  // Indicates if extreme motion vector unit test should be enabled or not.
747  unsigned int motion_vector_unit_test;
748  // Indicates if superblock multipass unit test should be enabled or not.
749  unsigned int sb_multipass_unit_test;
750 } UnitTestCfg;
751 
752 typedef struct {
753  // Indicates the file path to the VMAF model.
754  const char *vmaf_model_path;
755  // Indicates the path to the film grain parameters.
756  const char *film_grain_table_filename;
757  // Indicates the visual tuning metric.
758  aom_tune_metric tuning;
759  // Indicates if the current content is screen or default type.
760  aom_tune_content content;
761  // Indicates the film grain parameters.
762  int film_grain_test_vector;
763  // Indicates the in-block distortion metric to use.
764  aom_dist_metric dist_metric;
765 } TuneCfg;
766 
767 typedef struct {
768  // Indicates the framerate of the input video.
769  double init_framerate;
770  // Indicates the bit-depth of the input video.
771  unsigned int input_bit_depth;
772  // Indicates the maximum number of frames to be encoded.
773  unsigned int limit;
774  // Indicates the chrome subsampling x value.
775  unsigned int chroma_subsampling_x;
776  // Indicates the chrome subsampling y value.
777  unsigned int chroma_subsampling_y;
778 } InputCfg;
779 
780 typedef struct {
781  // If true, encoder will use fixed QP offsets, that are either:
782  // - Given by the user, and stored in 'fixed_qp_offsets' array, OR
783  // - Picked automatically from cq_level.
784  int use_fixed_qp_offsets;
785  // Indicates the minimum flatness of the quantization matrix.
786  int qm_minlevel;
787  // Indicates the maximum flatness of the quantization matrix.
788  int qm_maxlevel;
789  // Indicates if adaptive quantize_b should be enabled.
790  int quant_b_adapt;
791  // Indicates the Adaptive Quantization mode to be used.
792  AQ_MODE aq_mode;
793  // Indicates the delta q mode to be used.
794  DELTAQ_MODE deltaq_mode;
795  // Indicates the delta q mode strength.
796  DELTAQ_MODE deltaq_strength;
797  // Indicates if delta quantization should be enabled in chroma planes.
798  bool enable_chroma_deltaq;
799  // Indicates if delta quantization should be enabled for hdr video
800  bool enable_hdr_deltaq;
801  // Indicates if encoding with quantization matrices should be enabled.
802  bool using_qm;
803 } QuantizationCfg;
804 
809 typedef struct {
817 
826 
831 
836 
844 
849 
855 
864 
870 } AlgoCfg;
873 typedef struct {
874  // Indicates the codec bit-depth.
875  aom_bit_depth_t bit_depth;
876  // Indicates the superblock size that should be used by the encoder.
877  aom_superblock_size_t superblock_size;
878  // Indicates if loopfilter modulation should be enabled.
879  bool enable_deltalf_mode;
880  // Indicates how CDEF should be applied.
881  CDEF_CONTROL cdef_control;
882  // Indicates if loop restoration filter should be enabled.
883  bool enable_restoration;
884  // When enabled, video mode should be used even for single frame input.
885  bool force_video_mode;
886  // Indicates if the error resiliency features should be enabled.
887  bool error_resilient_mode;
888  // Indicates if frame parallel decoding feature should be enabled.
889  bool frame_parallel_decoding_mode;
890  // Indicates if the input should be encoded as monochrome.
891  bool enable_monochrome;
892  // When enabled, the encoder will use a full header even for still pictures.
893  // When disabled, a reduced header is used for still pictures.
894  bool full_still_picture_hdr;
895  // Indicates if dual interpolation filters should be enabled.
896  bool enable_dual_filter;
897  // Indicates if frame order hint should be enabled or not.
898  bool enable_order_hint;
899  // Indicates if ref_frame_mvs should be enabled at the sequence level.
900  bool ref_frame_mvs_present;
901  // Indicates if ref_frame_mvs should be enabled at the frame level.
902  bool enable_ref_frame_mvs;
903  // Indicates if interintra compound mode is enabled.
904  bool enable_interintra_comp;
905  // Indicates if global motion should be enabled.
906  bool enable_global_motion;
907  // Indicates if palette should be enabled.
908  bool enable_palette;
909 } ToolCfg;
910 
915 typedef struct AV1EncoderConfig {
917  // Configuration related to the input video.
918  InputCfg input_cfg;
919 
920  // Configuration related to frame-dimensions.
921  FrameDimensionCfg frm_dim_cfg;
922 
928 
933 
940  // Configuration related to Quantization.
941  QuantizationCfg q_cfg;
942 
943  // Internal frame size scaling.
944  ResizeCfg resize_cfg;
945 
946  // Frame Super-Resolution size scaling.
947  SuperResCfg superres_cfg;
948 
957  // Configuration related to encoder toolsets.
958  ToolCfg tool_cfg;
959 
960  // Configuration related to Group of frames.
961  GFConfig gf_cfg;
962 
963  // Tile related configuration parameters.
964  TileConfig tile_cfg;
965 
966  // Configuration related to Tune.
967  TuneCfg tune_cfg;
968 
969  // Configuration related to color.
970  ColorCfg color_cfg;
971 
972  // Configuration related to decoder model.
973  DecoderModelCfg dec_model_cfg;
974 
975  // Configuration related to reference frames.
976  RefFrameCfg ref_frm_cfg;
977 
978  // Configuration related to unit tests.
979  UnitTestCfg unit_test_cfg;
980 
981  // Flags related to motion mode.
982  MotionModeCfg motion_mode_cfg;
983 
984  // Flags related to intra mode search.
985  IntraModeCfg intra_mode_cfg;
986 
987  // Flags related to transform size/type.
988  TxfmSizeTypeCfg txfm_cfg;
989 
990  // Flags related to compound type.
991  CompoundTypeCfg comp_type_cfg;
992 
993  // Partition related information.
994  PartitionCfg part_cfg;
995 
996  // Configuration related to frequency of cost update.
997  CostUpdateFreq cost_upd_freq;
998 
999 #if CONFIG_DENOISE
1000  // Indicates the noise level.
1001  float noise_level;
1002  // Indicates the the denoisers block size.
1003  int noise_block_size;
1004  // Indicates whether to apply denoising to the frame to be encoded
1005  int enable_dnl_denoising;
1006 #endif
1007 
1008 #if CONFIG_AV1_TEMPORAL_DENOISING
1009  // Noise sensitivity.
1010  int noise_sensitivity;
1011 #endif
1012  // Bit mask to specify which tier each of the 32 possible operating points
1013  // conforms to.
1014  unsigned int tier_mask;
1015 
1016  // Indicates the number of pixels off the edge of a reference frame we're
1017  // allowed to go when forming an inter prediction.
1018  int border_in_pixels;
1019 
1020  // Indicates the maximum number of threads that may be used by the encoder.
1021  int max_threads;
1022 
1023  // Indicates the speed preset to be used.
1024  int speed;
1025 
1026  // Indicates the target sequence level index for each operating point(OP).
1027  AV1_LEVEL target_seq_level_idx[MAX_NUM_OPERATING_POINTS];
1028 
1029  // Indicates the bitstream profile to be used.
1030  BITSTREAM_PROFILE profile;
1031 
1043  // Total number of encoding passes.
1044  int passes;
1045 
1046  // the name of the second pass output file when passes > 2
1047  const char *two_pass_output;
1048 
1049  // the name of the second pass log file when passes > 2
1050  const char *second_pass_log;
1051 
1052  // Indicates if the encoding is GOOD or REALTIME.
1053  MODE mode;
1054 
1055  // Indicates if row-based multi-threading should be enabled or not.
1056  bool row_mt;
1057 
1058  // Indicates if frame parallel multi-threading should be enabled or not.
1059  bool fp_mt;
1060 
1061  // Indicates if 16bit frame buffers are to be used i.e., the content is >
1062  // 8-bit.
1063  bool use_highbitdepth;
1064 
1065  // Indicates the bitstream syntax mode. 0 indicates bitstream is saved as
1066  // Section 5 bitstream, while 1 indicates the bitstream is saved in Annex - B
1067  // format.
1068  bool save_as_annexb;
1069 
1070  // The path for partition stats reading and writing, used in the experiment
1071  // CONFIG_PARTITION_SEARCH_ORDER.
1072  const char *partition_info_path;
1073 
1074  // Exit the encoder when it fails to encode to a given level.
1075  int strict_level_conformance;
1076 
1077  // Max depth for the GOP after a key frame
1078  int kf_max_pyr_height;
1079 
1080  // A flag to control if we enable the superblock qp sweep for a given lambda
1081  int sb_qp_sweep;
1084 
1086 static INLINE int is_lossless_requested(const RateControlCfg *const rc_cfg) {
1087  return rc_cfg->best_allowed_q == 0 && rc_cfg->worst_allowed_q == 0;
1088 }
1094 typedef struct {
1100  int obmc_probs[FRAME_UPDATE_TYPES][BLOCK_SIZES_ALL];
1101 
1107  int warped_probs[FRAME_UPDATE_TYPES];
1108 
1115  int tx_type_probs[FRAME_UPDATE_TYPES][TX_SIZES_ALL][TX_TYPES];
1116 
1123  int switchable_interp_probs[FRAME_UPDATE_TYPES][SWITCHABLE_FILTER_CONTEXTS]
1124  [SWITCHABLE_FILTERS];
1125 } FrameProbInfo;
1126 
1129 typedef struct FRAME_COUNTS {
1130 // Note: This structure should only contain 'unsigned int' fields, or
1131 // aggregates built solely from 'unsigned int' fields/elements
1132 #if CONFIG_ENTROPY_STATS
1133  unsigned int kf_y_mode[KF_MODE_CONTEXTS][KF_MODE_CONTEXTS][INTRA_MODES];
1134  unsigned int angle_delta[DIRECTIONAL_MODES][2 * MAX_ANGLE_DELTA + 1];
1135  unsigned int y_mode[BLOCK_SIZE_GROUPS][INTRA_MODES];
1136  unsigned int uv_mode[CFL_ALLOWED_TYPES][INTRA_MODES][UV_INTRA_MODES];
1137  unsigned int cfl_sign[CFL_JOINT_SIGNS];
1138  unsigned int cfl_alpha[CFL_ALPHA_CONTEXTS][CFL_ALPHABET_SIZE];
1139  unsigned int palette_y_mode[PALATTE_BSIZE_CTXS][PALETTE_Y_MODE_CONTEXTS][2];
1140  unsigned int palette_uv_mode[PALETTE_UV_MODE_CONTEXTS][2];
1141  unsigned int palette_y_size[PALATTE_BSIZE_CTXS][PALETTE_SIZES];
1142  unsigned int palette_uv_size[PALATTE_BSIZE_CTXS][PALETTE_SIZES];
1143  unsigned int palette_y_color_index[PALETTE_SIZES]
1144  [PALETTE_COLOR_INDEX_CONTEXTS]
1145  [PALETTE_COLORS];
1146  unsigned int palette_uv_color_index[PALETTE_SIZES]
1147  [PALETTE_COLOR_INDEX_CONTEXTS]
1148  [PALETTE_COLORS];
1149  unsigned int partition[PARTITION_CONTEXTS][EXT_PARTITION_TYPES];
1150  unsigned int txb_skip[TOKEN_CDF_Q_CTXS][TX_SIZES][TXB_SKIP_CONTEXTS][2];
1151  unsigned int eob_extra[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES]
1152  [EOB_COEF_CONTEXTS][2];
1153  unsigned int dc_sign[PLANE_TYPES][DC_SIGN_CONTEXTS][2];
1154  unsigned int coeff_lps[TX_SIZES][PLANE_TYPES][BR_CDF_SIZE - 1][LEVEL_CONTEXTS]
1155  [2];
1156  unsigned int eob_flag[TX_SIZES][PLANE_TYPES][EOB_COEF_CONTEXTS][2];
1157  unsigned int eob_multi16[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][5];
1158  unsigned int eob_multi32[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][6];
1159  unsigned int eob_multi64[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][7];
1160  unsigned int eob_multi128[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][8];
1161  unsigned int eob_multi256[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][9];
1162  unsigned int eob_multi512[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][10];
1163  unsigned int eob_multi1024[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][11];
1164  unsigned int coeff_lps_multi[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES]
1165  [LEVEL_CONTEXTS][BR_CDF_SIZE];
1166  unsigned int coeff_base_multi[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES]
1167  [SIG_COEF_CONTEXTS][NUM_BASE_LEVELS + 2];
1168  unsigned int coeff_base_eob_multi[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES]
1169  [SIG_COEF_CONTEXTS_EOB][NUM_BASE_LEVELS + 1];
1170  unsigned int newmv_mode[NEWMV_MODE_CONTEXTS][2];
1171  unsigned int zeromv_mode[GLOBALMV_MODE_CONTEXTS][2];
1172  unsigned int refmv_mode[REFMV_MODE_CONTEXTS][2];
1173  unsigned int drl_mode[DRL_MODE_CONTEXTS][2];
1174  unsigned int inter_compound_mode[INTER_MODE_CONTEXTS][INTER_COMPOUND_MODES];
1175  unsigned int wedge_idx[BLOCK_SIZES_ALL][16];
1176  unsigned int interintra[BLOCK_SIZE_GROUPS][2];
1177  unsigned int interintra_mode[BLOCK_SIZE_GROUPS][INTERINTRA_MODES];
1178  unsigned int wedge_interintra[BLOCK_SIZES_ALL][2];
1179  unsigned int compound_type[BLOCK_SIZES_ALL][MASKED_COMPOUND_TYPES];
1180  unsigned int motion_mode[BLOCK_SIZES_ALL][MOTION_MODES];
1181  unsigned int obmc[BLOCK_SIZES_ALL][2];
1182  unsigned int intra_inter[INTRA_INTER_CONTEXTS][2];
1183  unsigned int comp_inter[COMP_INTER_CONTEXTS][2];
1184  unsigned int comp_ref_type[COMP_REF_TYPE_CONTEXTS][2];
1185  unsigned int uni_comp_ref[UNI_COMP_REF_CONTEXTS][UNIDIR_COMP_REFS - 1][2];
1186  unsigned int single_ref[REF_CONTEXTS][SINGLE_REFS - 1][2];
1187  unsigned int comp_ref[REF_CONTEXTS][FWD_REFS - 1][2];
1188  unsigned int comp_bwdref[REF_CONTEXTS][BWD_REFS - 1][2];
1189  unsigned int intrabc[2];
1190 
1191  unsigned int txfm_partition[TXFM_PARTITION_CONTEXTS][2];
1192  unsigned int intra_tx_size[MAX_TX_CATS][TX_SIZE_CONTEXTS][MAX_TX_DEPTH + 1];
1193  unsigned int skip_mode[SKIP_MODE_CONTEXTS][2];
1194  unsigned int skip_txfm[SKIP_CONTEXTS][2];
1195  unsigned int compound_index[COMP_INDEX_CONTEXTS][2];
1196  unsigned int comp_group_idx[COMP_GROUP_IDX_CONTEXTS][2];
1197  unsigned int delta_q[DELTA_Q_PROBS][2];
1198  unsigned int delta_lf_multi[FRAME_LF_COUNT][DELTA_LF_PROBS][2];
1199  unsigned int delta_lf[DELTA_LF_PROBS][2];
1200 
1201  unsigned int inter_ext_tx[EXT_TX_SETS_INTER][EXT_TX_SIZES][TX_TYPES];
1202  unsigned int intra_ext_tx[EXT_TX_SETS_INTRA][EXT_TX_SIZES][INTRA_MODES]
1203  [TX_TYPES];
1204  unsigned int filter_intra_mode[FILTER_INTRA_MODES];
1205  unsigned int filter_intra[BLOCK_SIZES_ALL][2];
1206  unsigned int switchable_restore[RESTORE_SWITCHABLE_TYPES];
1207  unsigned int wiener_restore[2];
1208  unsigned int sgrproj_restore[2];
1209 #endif // CONFIG_ENTROPY_STATS
1210 
1211  unsigned int switchable_interp[SWITCHABLE_FILTER_CONTEXTS]
1212  [SWITCHABLE_FILTERS];
1213 } FRAME_COUNTS;
1214 
1215 #define INTER_MODE_RD_DATA_OVERALL_SIZE 6400
1216 
1217 typedef struct {
1218  int ready;
1219  double a;
1220  double b;
1221  double dist_mean;
1222  double ld_mean;
1223  double sse_mean;
1224  double sse_sse_mean;
1225  double sse_ld_mean;
1226  int num;
1227  double dist_sum;
1228  double ld_sum;
1229  double sse_sum;
1230  double sse_sse_sum;
1231  double sse_ld_sum;
1232 } InterModeRdModel;
1233 
1234 typedef struct {
1235  int idx;
1236  int64_t rd;
1237 } RdIdxPair;
1238 // TODO(angiebird): This is an estimated size. We still need to figure what is
1239 // the maximum number of modes.
1240 #define MAX_INTER_MODES 1024
1241 // TODO(any): rename this struct to something else. There is already another
1242 // struct called inter_mode_info, which makes this terribly confusing.
1250 typedef struct inter_modes_info {
1255  int num;
1259  MB_MODE_INFO mbmi_arr[MAX_INTER_MODES];
1263  int mode_rate_arr[MAX_INTER_MODES];
1267  int64_t sse_arr[MAX_INTER_MODES];
1271  int64_t est_rd_arr[MAX_INTER_MODES];
1275  RdIdxPair rd_idx_pair_arr[MAX_INTER_MODES];
1279  RD_STATS rd_cost_arr[MAX_INTER_MODES];
1283  RD_STATS rd_cost_y_arr[MAX_INTER_MODES];
1287  RD_STATS rd_cost_uv_arr[MAX_INTER_MODES];
1288 } InterModesInfo;
1289 
1291 typedef struct {
1292  // TODO(kyslov): consider changing to 64bit
1293 
1294  // This struct is used for computing variance in choose_partitioning(), where
1295  // the max number of samples within a superblock is 32x32 (with 4x4 avg).
1296  // With 8bit bitdepth, uint32_t is enough for sum_square_error (2^8 * 2^8 * 32
1297  // * 32 = 2^26). For high bitdepth we need to consider changing this to 64 bit
1298  uint32_t sum_square_error;
1299  int32_t sum_error;
1300  int log2_count;
1301  int variance;
1302 } VPartVar;
1303 
1304 typedef struct {
1305  VPartVar none;
1306  VPartVar horz[2];
1307  VPartVar vert[2];
1308 } VPVariance;
1309 
1310 typedef struct {
1311  VPVariance part_variances;
1312  VPartVar split[4];
1313 } VP4x4;
1314 
1315 typedef struct {
1316  VPVariance part_variances;
1317  VP4x4 split[4];
1318 } VP8x8;
1319 
1320 typedef struct {
1321  VPVariance part_variances;
1322  VP8x8 split[4];
1323 } VP16x16;
1324 
1325 typedef struct {
1326  VPVariance part_variances;
1327  VP16x16 split[4];
1328 } VP32x32;
1329 
1330 typedef struct {
1331  VPVariance part_variances;
1332  VP32x32 split[4];
1333 } VP64x64;
1334 
1335 typedef struct {
1336  VPVariance part_variances;
1337  VP64x64 *split;
1338 } VP128x128;
1339 
1345 typedef struct {
1354  int64_t thresholds[5];
1355 
1362 
1366 typedef struct {
1367 #if CONFIG_MULTITHREAD
1368 
1372  pthread_mutex_t *mutex_;
1373  pthread_cond_t *cond_;
1375 #endif // CONFIG_MULTITHREAD
1376 
1399  int rows;
1409 
1412 // TODO(jingning) All spatially adaptive variables should go to TileDataEnc.
1413 typedef struct TileDataEnc {
1414  TileInfo tile_info;
1415  DECLARE_ALIGNED(16, FRAME_CONTEXT, tctx);
1416  FRAME_CONTEXT *row_ctx;
1417  uint64_t abs_sum_level;
1418  uint8_t allow_update_cdf;
1419  InterModeRdModel inter_mode_rd_models[BLOCK_SIZES_ALL];
1420  AV1EncRowMultiThreadSync row_mt_sync;
1421  MV firstpass_top_mv;
1422 } TileDataEnc;
1423 
1424 typedef struct RD_COUNTS {
1425  int compound_ref_used_flag;
1426  int skip_mode_used_flag;
1427  int tx_type_used[TX_SIZES_ALL][TX_TYPES];
1428  int obmc_used[BLOCK_SIZES_ALL][2];
1429  int warped_used[2];
1430  int newmv_or_intra_blocks;
1431  uint64_t seg_tmp_pred_cost[2];
1432 } RD_COUNTS;
1433 
1434 typedef struct ThreadData {
1435  MACROBLOCK mb;
1436  RD_COUNTS rd_counts;
1437  FRAME_COUNTS *counts;
1438  PC_TREE_SHARED_BUFFERS shared_coeff_buf;
1439  SIMPLE_MOTION_DATA_TREE *sms_tree;
1440  SIMPLE_MOTION_DATA_TREE *sms_root;
1441  uint32_t *hash_value_buffer[2][2];
1442  OBMCBuffer obmc_buffer;
1443  PALETTE_BUFFER *palette_buffer;
1444  CompoundTypeRdBuffers comp_rd_buffer;
1445  CONV_BUF_TYPE *tmp_conv_dst;
1446  uint64_t abs_sum_level;
1447  uint8_t *tmp_pred_bufs[2];
1448  int intrabc_used;
1449  int deltaq_used;
1450  int coefficient_size;
1451  int max_mv_magnitude;
1452  int interp_filter_selected[SWITCHABLE];
1453  FRAME_CONTEXT *tctx;
1454  VP64x64 *vt64x64;
1455  int32_t num_64x64_blocks;
1456  PICK_MODE_CONTEXT *firstpass_ctx;
1457  TemporalFilterData tf_data;
1458  TplTxfmStats tpl_txfm_stats;
1459  // Pointer to the array of structures to store gradient information of each
1460  // pixel in a superblock. The buffer constitutes of MAX_SB_SQUARE pixel level
1461  // structures for each of the plane types (PLANE_TYPE_Y and PLANE_TYPE_UV).
1462  PixelLevelGradientInfo *pixel_gradient_info;
1463  // Pointer to the array of structures to store source variance information of
1464  // each 4x4 sub-block in a superblock. Block4x4VarInfo structure is used to
1465  // store source variance and log of source variance of each 4x4 sub-block
1466  // for subsequent retrieval.
1467  Block4x4VarInfo *src_var_info_of_4x4_sub_blocks;
1468  // The pc tree root for RTC non-rd case.
1469  PC_TREE *rt_pc_root;
1470 } ThreadData;
1471 
1472 struct EncWorkerData;
1473 
1479 typedef struct {
1502 
1506  int thread_id_to_tile_id[MAX_NUM_THREADS];
1507 
1513 
1519 
1520 #if CONFIG_MULTITHREAD
1521 
1524  pthread_mutex_t *mutex_;
1528  pthread_cond_t *cond_;
1529 #endif
1530 
1538  void (*sync_read_ptr)(AV1EncRowMultiThreadSync *const, int, int);
1542  void (*sync_write_ptr)(AV1EncRowMultiThreadSync *const, int, int, int);
1545 
1549 #define NUM_RECODES_PER_FRAME 10
1550 
1554 #define MAX_PARALLEL_FRAMES 4
1555 
1560 typedef struct RestoreStateBuffers {
1564  uint16_t *cdef_srcbuf;
1565 
1569  uint16_t *cdef_colbuf[MAX_MB_PLANE];
1570 
1574  int32_t *rst_tmpbuf;
1575 
1579  RestorationLineBuffers *rlbs;
1581 
1585 typedef struct PrimaryMultiThreadInfo {
1590 
1594  int num_mod_workers[NUM_MT_MODULES];
1595 
1599  AVxWorker *workers;
1600 
1605  struct EncWorkerData *tile_thr_data;
1606 
1610  AV1CdefWorkerData *cdef_worker;
1611 
1617 
1623 
1627 typedef struct MultiThreadInfo {
1632 
1636  int num_mod_workers[NUM_MT_MODULES];
1637 
1641  AVxWorker *workers;
1642 
1647  struct EncWorkerData *tile_thr_data;
1648 
1654 
1659 
1664 
1668  AV1TplRowMultiThreadInfo tpl_row_mt;
1669 
1673  AV1LfSync lf_row_sync;
1674 
1678  AV1LrSync lr_row_sync;
1679 
1683  AV1EncPackBSSync pack_bs_sync;
1684 
1688  AV1GlobalMotionSync gm_sync;
1689 
1693  AV1TemporalFilterSync tf_sync;
1694 
1698  AV1CdefSync cdef_sync;
1699 
1703  AV1CdefWorkerData *cdef_worker;
1704 
1709 
1715 } MultiThreadInfo;
1716 
1719 typedef struct ActiveMap {
1720  int enabled;
1721  int update;
1722  unsigned char *map;
1723 } ActiveMap;
1724 
1730 typedef struct {
1735  double cs_rate_array[32];
1745 
1748 #if CONFIG_INTERNAL_STATS
1749 // types of stats
1750 enum {
1751  STAT_Y,
1752  STAT_U,
1753  STAT_V,
1754  STAT_ALL,
1755  NUM_STAT_TYPES // This should always be the last member of the enum
1756 } UENUM1BYTE(StatType);
1757 
1758 typedef struct IMAGE_STAT {
1759  double stat[NUM_STAT_TYPES];
1760  double worst;
1761 } ImageStat;
1762 #endif // CONFIG_INTERNAL_STATS
1763 
1764 typedef struct {
1765  int ref_count;
1766  YV12_BUFFER_CONFIG buf;
1767 } EncRefCntBuffer;
1768 
1776 typedef struct {
1789  int stride;
1791 
1794 #if CONFIG_COLLECT_PARTITION_STATS
1795 typedef struct FramePartitionTimingStats {
1796  int partition_decisions[6][EXT_PARTITION_TYPES];
1797  int partition_attempts[6][EXT_PARTITION_TYPES];
1798  int64_t partition_times[6][EXT_PARTITION_TYPES];
1799 
1800  int partition_redo;
1801 } FramePartitionTimingStats;
1802 #endif // CONFIG_COLLECT_PARTITION_STATS
1803 
1804 #if CONFIG_COLLECT_COMPONENT_TIMING
1805 #include "aom_ports/aom_timer.h"
1806 // Adjust the following to add new components.
1807 enum {
1808  av1_encode_strategy_time,
1809  av1_get_one_pass_rt_params_time,
1810  av1_get_second_pass_params_time,
1811  denoise_and_encode_time,
1812  apply_filtering_time,
1813  av1_tpl_setup_stats_time,
1814  encode_frame_to_data_rate_time,
1815  encode_with_or_without_recode_time,
1816  loop_filter_time,
1817  cdef_time,
1818  loop_restoration_time,
1819  av1_pack_bitstream_final_time,
1820  av1_encode_frame_time,
1821  av1_compute_global_motion_time,
1822  av1_setup_motion_field_time,
1823  encode_sb_row_time,
1824 
1825  rd_pick_partition_time,
1826  rd_use_partition_time,
1827  choose_var_based_partitioning_time,
1828  av1_prune_partitions_time,
1829  none_partition_search_time,
1830  split_partition_search_time,
1831  rectangular_partition_search_time,
1832  ab_partitions_search_time,
1833  rd_pick_4partition_time,
1834  encode_sb_time,
1835 
1836  rd_pick_sb_modes_time,
1837  av1_rd_pick_intra_mode_sb_time,
1838  av1_rd_pick_inter_mode_sb_time,
1839  set_params_rd_pick_inter_mode_time,
1840  skip_inter_mode_time,
1841  handle_inter_mode_time,
1842  evaluate_motion_mode_for_winner_candidates_time,
1843  do_tx_search_time,
1844  handle_intra_mode_time,
1845  refine_winner_mode_tx_time,
1846  av1_search_palette_mode_time,
1847  handle_newmv_time,
1848  compound_type_rd_time,
1849  interpolation_filter_search_time,
1850  motion_mode_rd_time,
1851 
1852  nonrd_use_partition_time,
1853  pick_sb_modes_nonrd_time,
1854  hybrid_intra_mode_search_time,
1855  nonrd_pick_inter_mode_sb_time,
1856  encode_b_nonrd_time,
1857 
1858  kTimingComponents,
1859 } UENUM1BYTE(TIMING_COMPONENT);
1860 
1861 static INLINE char const *get_component_name(int index) {
1862  switch (index) {
1863  case av1_encode_strategy_time: return "av1_encode_strategy_time";
1864  case av1_get_one_pass_rt_params_time:
1865  return "av1_get_one_pass_rt_params_time";
1866  case av1_get_second_pass_params_time:
1867  return "av1_get_second_pass_params_time";
1868  case denoise_and_encode_time: return "denoise_and_encode_time";
1869  case apply_filtering_time: return "apply_filtering_time";
1870  case av1_tpl_setup_stats_time: return "av1_tpl_setup_stats_time";
1871  case encode_frame_to_data_rate_time:
1872  return "encode_frame_to_data_rate_time";
1873  case encode_with_or_without_recode_time:
1874  return "encode_with_or_without_recode_time";
1875  case loop_filter_time: return "loop_filter_time";
1876  case cdef_time: return "cdef_time";
1877  case loop_restoration_time: return "loop_restoration_time";
1878  case av1_pack_bitstream_final_time: return "av1_pack_bitstream_final_time";
1879  case av1_encode_frame_time: return "av1_encode_frame_time";
1880  case av1_compute_global_motion_time:
1881  return "av1_compute_global_motion_time";
1882  case av1_setup_motion_field_time: return "av1_setup_motion_field_time";
1883  case encode_sb_row_time: return "encode_sb_row_time";
1884 
1885  case rd_pick_partition_time: return "rd_pick_partition_time";
1886  case rd_use_partition_time: return "rd_use_partition_time";
1887  case choose_var_based_partitioning_time:
1888  return "choose_var_based_partitioning_time";
1889  case av1_prune_partitions_time: return "av1_prune_partitions_time";
1890  case none_partition_search_time: return "none_partition_search_time";
1891  case split_partition_search_time: return "split_partition_search_time";
1892  case rectangular_partition_search_time:
1893  return "rectangular_partition_search_time";
1894  case ab_partitions_search_time: return "ab_partitions_search_time";
1895  case rd_pick_4partition_time: return "rd_pick_4partition_time";
1896  case encode_sb_time: return "encode_sb_time";
1897 
1898  case rd_pick_sb_modes_time: return "rd_pick_sb_modes_time";
1899  case av1_rd_pick_intra_mode_sb_time:
1900  return "av1_rd_pick_intra_mode_sb_time";
1901  case av1_rd_pick_inter_mode_sb_time:
1902  return "av1_rd_pick_inter_mode_sb_time";
1903  case set_params_rd_pick_inter_mode_time:
1904  return "set_params_rd_pick_inter_mode_time";
1905  case skip_inter_mode_time: return "skip_inter_mode_time";
1906  case handle_inter_mode_time: return "handle_inter_mode_time";
1907  case evaluate_motion_mode_for_winner_candidates_time:
1908  return "evaluate_motion_mode_for_winner_candidates_time";
1909  case do_tx_search_time: return "do_tx_search_time";
1910  case handle_intra_mode_time: return "handle_intra_mode_time";
1911  case refine_winner_mode_tx_time: return "refine_winner_mode_tx_time";
1912  case av1_search_palette_mode_time: return "av1_search_palette_mode_time";
1913  case handle_newmv_time: return "handle_newmv_time";
1914  case compound_type_rd_time: return "compound_type_rd_time";
1915  case interpolation_filter_search_time:
1916  return "interpolation_filter_search_time";
1917  case motion_mode_rd_time: return "motion_mode_rd_time";
1918 
1919  case nonrd_use_partition_time: return "nonrd_use_partition_time";
1920  case pick_sb_modes_nonrd_time: return "pick_sb_modes_nonrd_time";
1921  case hybrid_intra_mode_search_time: return "hybrid_intra_mode_search_time";
1922  case nonrd_pick_inter_mode_sb_time: return "nonrd_pick_inter_mode_sb_time";
1923  case encode_b_nonrd_time: return "encode_b_nonrd_time";
1924 
1925  default: assert(0);
1926  }
1927  return "error";
1928 }
1929 #endif
1930 
1931 // The maximum number of internal ARFs except ALTREF_FRAME
1932 #define MAX_INTERNAL_ARFS (REF_FRAMES - BWDREF_FRAME - 1)
1933 
1939 typedef struct {
1944 
1950  YV12_BUFFER_CONFIG *ref_buf[REF_FRAMES];
1951 
1955  unsigned char *src_buffer;
1956 
1962  int num_ref_frames[MAX_DIRECTIONS];
1963 
1970  FrameDistPair reference_frames[MAX_DIRECTIONS][REF_FRAMES - 1];
1971 
1984 
1990  int src_corners[2 * MAX_CORNERS];
1992 
2002 typedef struct {
2003  int width;
2004  int height;
2006 
2010 typedef struct {
2021 
2025 typedef struct {
2045  fractional_mv_step_fp *find_fractional_mv_step;
2052  search_site_config search_site_cfg[SS_CFG_TOTAL][NUM_DISTINCT_SEARCH_METHODS];
2054 
2063 typedef struct {
2068 
2076 typedef struct {
2077  int width;
2078  int height;
2080 
2084 typedef struct {
2088  int ref_relative_dist[INTER_REFS_PER_FRAME];
2098 
2114 typedef struct {
2122  unsigned int coeff_opt_thresholds[MODE_EVAL_TYPES][2];
2123 
2128  TX_SIZE_SEARCH_METHOD tx_size_search_methods[MODE_EVAL_TYPES];
2129 
2136  unsigned int use_transform_domain_distortion[MODE_EVAL_TYPES];
2137 
2143  unsigned int tx_domain_dist_threshold[MODE_EVAL_TYPES];
2144 
2150  unsigned int skip_txfm_level[MODE_EVAL_TYPES];
2151 
2157  unsigned int predict_dc_level[MODE_EVAL_TYPES];
2159 
2167 typedef struct {
2168  bool last_frame;
2178 
2182 typedef struct {
2187 
2192 
2197 
2203 
2208 
2213 
2218 
2224 } ExternalFlags;
2225 
2228 typedef struct {
2229  // Some misc info
2230  int high_prec;
2231  int q;
2232  int order;
2233 
2234  // MV counters
2235  int inter_count;
2236  int intra_count;
2237  int default_mvs;
2238  int mv_joint_count[4];
2239  int last_bit_zero;
2240  int last_bit_nonzero;
2241 
2242  // Keep track of the rates
2243  int total_mv_rate;
2244  int hp_total_mv_rate;
2245  int lp_total_mv_rate;
2246 
2247  // Texture info
2248  int horz_text;
2249  int vert_text;
2250  int diag_text;
2251 
2252  // Whether the current struct contains valid data
2253  int valid;
2254 } MV_STATS;
2255 
2256 typedef struct WeberStats {
2257  int64_t mb_wiener_variance;
2258  int64_t src_variance;
2259  int64_t rec_variance;
2260  int16_t src_pix_max;
2261  int16_t rec_pix_max;
2262  int64_t distortion;
2263  int64_t satd;
2264  double max_scale;
2265 } WeberStats;
2266 
2267 typedef struct {
2268  struct loopfilter lf;
2269  CdefInfo cdef_info;
2270  YV12_BUFFER_CONFIG copy_buffer;
2271  RATE_CONTROL rc;
2272  MV_STATS mv_stats;
2273 } CODING_CONTEXT;
2274 
2275 typedef struct {
2276  int frame_width;
2277  int frame_height;
2278  int mi_rows;
2279  int mi_cols;
2280  int mb_rows;
2281  int mb_cols;
2282  int num_mbs;
2283  aom_bit_depth_t bit_depth;
2284  int subsampling_x;
2285  int subsampling_y;
2286 } FRAME_INFO;
2287 
2291 typedef struct {
2292  int show_frame_count;
2293 } FRAME_INDEX_SET;
2294 
2300 typedef struct {
2306  uint8_t *map;
2314 
2318 typedef struct {
2322  int64_t prev_ts_start;
2326  int64_t prev_ts_end;
2331 } TimeStamps;
2332 
2337 typedef struct {
2341  tran_low_t *tcoeff;
2345  uint16_t *eobs;
2349  uint8_t *entropy_ctx;
2350 } CoeffBufferPool;
2351 
2352 #if !CONFIG_REALTIME_ONLY
2353 
2354 // DUCKY_ENCODE_FRAME_MODE is c version of EncodeFrameMode
2355 enum {
2356  DUCKY_ENCODE_FRAME_MODE_NONE, // Let native AV1 determine q index and rdmult
2357  DUCKY_ENCODE_FRAME_MODE_QINDEX, // DuckyEncode determines q index and AV1
2358  // determines rdmult
2359  DUCKY_ENCODE_FRAME_MODE_QINDEX_RDMULT, // DuckyEncode determines q index and
2360  // rdmult
2361 } UENUM1BYTE(DUCKY_ENCODE_FRAME_MODE);
2362 
2363 enum {
2364  DUCKY_ENCODE_GOP_MODE_NONE, // native AV1 decides GOP
2365  DUCKY_ENCODE_GOP_MODE_RCL, // rate control lib decides GOP
2366 } UENUM1BYTE(DUCKY_ENCODE_GOP_MODE);
2367 
2368 typedef struct DuckyEncodeFrameInfo {
2369  DUCKY_ENCODE_FRAME_MODE qp_mode;
2370  DUCKY_ENCODE_GOP_MODE gop_mode;
2371  int q_index;
2372  int rdmult;
2373  // These two arrays are equivalent to std::vector<SuperblockEncodeParameters>
2374  int *superblock_encode_qindex;
2375  int *superblock_encode_rdmult;
2376  int delta_q_enabled;
2377 } DuckyEncodeFrameInfo;
2378 
2379 typedef struct DuckyEncodeFrameResult {
2380  int global_order_idx;
2381  int q_index;
2382  int rdmult;
2383  int rate;
2384  int64_t dist;
2385  double psnr;
2386 } DuckyEncodeFrameResult;
2387 
2388 typedef struct DuckyEncodeInfo {
2389  DuckyEncodeFrameInfo frame_info;
2390  DuckyEncodeFrameResult frame_result;
2391 } DuckyEncodeInfo;
2393 #endif
2394 
2396 typedef struct RTC_REF {
2401  int reference[INTER_REFS_PER_FRAME];
2402  int ref_idx[INTER_REFS_PER_FRAME];
2403  int refresh[REF_FRAMES];
2404  int set_ref_frame_config;
2405  int non_reference_frame;
2406  int ref_frame_comp[3];
2407  int gld_idx_1layer;
2408 } RTC_REF;
2414 typedef struct AV1_COMP_DATA {
2418  unsigned char *cx_data;
2419 
2423  size_t cx_data_sz;
2424 
2428  size_t frame_size;
2429 
2433  unsigned int lib_flags;
2434 
2439 
2443  int64_t ts_frame_end;
2444 
2448  int flush;
2449 
2453  const aom_rational64_t *timestamp_ratio;
2454 
2459 
2464 } AV1_COMP_DATA;
2465 
2469 typedef struct AV1_PRIMARY {
2474 
2480 #if CONFIG_FPMT_TEST
2481 
2486  FPMT_TEST_ENC_CFG fpmt_unit_test_cfg;
2487 
2491  FrameProbInfo temp_frame_probs;
2492 
2498  FrameProbInfo temp_frame_probs_simulation;
2499 
2504  int temp_valid_gm_model_found[FRAME_UPDATE_TYPES];
2505 #endif // CONFIG_FPMT_TEST
2506 
2511  RefCntBuffer *ref_frame_map_copy[REF_FRAMES];
2512 
2517 
2522 
2527 
2532 
2537 
2542 
2547  struct AV1_COMP *cpi;
2548 
2553 
2557  struct lookahead_ctx *lookahead;
2558 
2565 
2570  struct aom_codec_pkt_list *output_pkt_list;
2571 
2576 
2581 
2586 
2590  GF_STATE gf_state;
2591 
2596 
2600  AV1LevelParams level_params;
2601 
2606 
2611 
2616 
2621 
2630  SequenceHeader seq_params;
2631 
2635  int use_svc;
2636 
2641 
2646 
2651 
2655  struct aom_internal_error_info error;
2656 
2662  aom_variance_fn_ptr_t fn_ptr[BLOCK_SIZES_ALL];
2663 
2669 
2674 
2678  MV_STATS mv_stats;
2679 
2680 #if CONFIG_INTERNAL_STATS
2681 
2682  uint64_t total_time_receive_data;
2683  uint64_t total_time_compress_data;
2684 
2685  unsigned int total_mode_chosen_counts[MAX_MODES];
2686 
2687  int count[2];
2688  uint64_t total_sq_error[2];
2689  uint64_t total_samples[2];
2690  ImageStat psnr[2];
2691 
2692  double total_blockiness;
2693  double worst_blockiness;
2694 
2695  int total_bytes;
2696  double summed_quality;
2697  double summed_weights;
2698  double summed_quality_hbd;
2699  double summed_weights_hbd;
2700  unsigned int total_recode_hits;
2701  double worst_ssim;
2702  double worst_ssim_hbd;
2703 
2704  ImageStat fastssim;
2705  ImageStat psnrhvs;
2706 
2707  int b_calculate_blockiness;
2708  int b_calculate_consistency;
2709 
2710  double total_inconsistency;
2711  double worst_consistency;
2712  Ssimv *ssim_vars;
2713  Metrics metrics;
2715 #endif
2716 
2717 #if CONFIG_ENTROPY_STATS
2718 
2721  FRAME_COUNTS aggregate_fc;
2722 #endif // CONFIG_ENTROPY_STATS
2723 
2730  int fb_of_context_type[REF_FRAMES];
2731 
2736 
2741 
2748  int valid_gm_model_found[FRAME_UPDATE_TYPES];
2749 
2753  RTC_REF rtc_ref;
2754 } AV1_PRIMARY;
2755 
2759 typedef struct AV1_COMP {
2764 
2769  EncQuantDequantParams enc_quant_dequant_params;
2770 
2774  ThreadData td;
2775 
2779  FRAME_COUNTS counts;
2780 
2785 
2792 
2798 
2803 
2808 
2813  TRELLIS_OPT_TYPE optimize_seg_arr[MAX_SEGMENTS];
2814 
2821 
2830 
2836 
2841 
2846 
2851 
2857 
2863 
2868 
2878 
2883 
2888 
2893  RefCntBuffer *scaled_ref_buf[INTER_REFS_PER_FRAME];
2894 
2898  RefCntBuffer *last_show_frame_buf;
2899 
2904 
2909 
2914 
2920 
2926 
2930  int64_t ambient_err;
2931 
2935  RD_OPT rd;
2936 
2941  CODING_CONTEXT coding_context;
2942 
2947 
2952 
2957 
2962 
2966  double framerate;
2967 
2972 
2976  int speed;
2977 
2982 
2987 
2993 
2998 
3007  ActiveMap active_map;
3008 
3012  unsigned char gf_frame_index;
3013 
3014 #if CONFIG_INTERNAL_STATS
3015 
3016  uint64_t time_compress_data;
3017 
3018  unsigned int mode_chosen_counts[MAX_MODES];
3019  int bytes;
3020  unsigned int frame_recode_hits;
3022 #endif
3023 
3024 #if CONFIG_SPEED_STATS
3025 
3028  unsigned int tx_search_count;
3029 #endif // CONFIG_SPEED_STATS
3030 
3036 
3040  FRAME_INFO frame_info;
3041 
3045  FRAME_INDEX_SET frame_index_set;
3046 
3051 
3059 
3064 
3069  TileDataEnc *tile_data;
3074 
3078  TokenInfo token_info;
3079 
3084 
3089 
3094 
3099 
3104 
3109 
3114 
3119 
3120 #if CONFIG_FPMT_TEST
3121 
3125  double temp_framerate;
3126 #endif
3127 
3133 
3138 
3143 
3150 
3155 
3160 
3164  AV1LrStruct lr_ctxt;
3165 
3169  aom_film_grain_table_t *film_grain_table;
3170 
3171 #if CONFIG_DENOISE
3172 
3176  struct aom_denoise_and_model_t *denoise_and_model;
3177 #endif
3178 
3183 
3192 
3200 
3201 #if CONFIG_COLLECT_PARTITION_STATS
3202 
3205  FramePartitionTimingStats partition_stats;
3206 #endif // CONFIG_COLLECT_PARTITION_STATS
3207 
3208 #if CONFIG_COLLECT_COMPONENT_TIMING
3209 
3212  uint64_t component_time[kTimingComponents];
3217  struct aom_usec_timer component_timer[kTimingComponents];
3221  uint64_t frame_component_time[kTimingComponents];
3222 #endif
3223 
3228 
3233 
3238 
3245 
3246 #if CONFIG_TUNE_VMAF
3247 
3250  TuneVMAFInfo vmaf_info;
3251 #endif
3252 
3253 #if CONFIG_TUNE_BUTTERAUGLI
3254 
3257  TuneButteraugliInfo butteraugli_info;
3258 #endif
3259 
3264 
3268  COMPRESSOR_STAGE compressor_stage;
3269 
3274  FRAME_TYPE last_frame_type;
3275 
3279  int num_tg;
3280 
3287 
3291  FirstPassData firstpass_data;
3292 
3296  NOISE_ESTIMATE noise_estimate;
3297 
3298 #if CONFIG_AV1_TEMPORAL_DENOISING
3299 
3302  AV1_DENOISER denoiser;
3303 #endif
3304 
3309  uint8_t *consec_zero_mv;
3310 
3314  BLOCK_SIZE fp_block_size;
3315 
3321 
3326 
3331  ExtPartController ext_part_controller;
3332 
3337  MV_STATS mv_stats;
3342 
3348 
3355 #if CONFIG_FPMT_TEST
3356 
3362  int wanted_fb;
3363 #endif // CONFIG_FPMT_TEST
3364 
3371 
3372 #if CONFIG_RD_COMMAND
3373 
3376  RD_COMMAND rd_command;
3377 #endif // CONFIG_RD_COMMAND
3378 
3382  WeberStats *mb_weber_stats;
3383 
3387  BLOCK_SIZE weber_bsize;
3388 
3393 
3398 
3403 
3404 #if CONFIG_BITRATE_ACCURACY
3405 
3408  VBR_RATECTRL_INFO vbr_rc_info;
3409 #endif
3410 
3411 #if CONFIG_RATECTRL_LOG
3412 
3415  RATECTRL_LOG rc_log;
3416 #endif // CONFIG_RATECTRL_LOG
3417 
3422 
3426  THIRD_PASS_DEC_CTX *third_pass_ctx;
3427 
3432 
3437 
3441  uint64_t rec_sse;
3442 
3448 
3449 #if !CONFIG_REALTIME_ONLY
3450 
3453  DuckyEncodeInfo ducky_encode_info;
3454 #endif // CONFIG_REALTIME_ONLY
3455  //
3460 
3464  unsigned int zeromv_skip_thresh_exit_part[BLOCK_SIZES_ALL];
3465 } AV1_COMP;
3466 
3470 typedef struct EncodeFrameInput {
3472  YV12_BUFFER_CONFIG *source;
3473  YV12_BUFFER_CONFIG *last_source;
3474  int64_t ts_duration;
3477 
3482 typedef struct EncodeFrameParams {
3490  FRAME_TYPE frame_type;
3491 
3493  int primary_ref_frame;
3494  int order_offset;
3495 
3501 
3503  int refresh_frame_flags;
3504 
3505  int show_existing_frame;
3506  int existing_fb_idx_to_show;
3507 
3513 
3517  int remapped_ref_idx[REF_FRAMES];
3518 
3524 
3528  int speed;
3530 
3533 // EncodeFrameResults contains information about the result of encoding a
3534 // single frame
3535 typedef struct {
3536  size_t size; // Size of resulting bitstream
3537 } EncodeFrameResults;
3538 
3539 void av1_initialize_enc(unsigned int usage, enum aom_rc_mode end_usage);
3540 
3541 struct AV1_COMP *av1_create_compressor(AV1_PRIMARY *ppi,
3542  const AV1EncoderConfig *oxcf,
3543  BufferPool *const pool,
3544  COMPRESSOR_STAGE stage,
3545  int lap_lag_in_frames);
3546 
3547 struct AV1_PRIMARY *av1_create_primary_compressor(
3548  struct aom_codec_pkt_list *pkt_list_head, int num_lap_buffers,
3549  const AV1EncoderConfig *oxcf);
3550 
3551 void av1_remove_compressor(AV1_COMP *cpi);
3552 
3553 void av1_remove_primary_compressor(AV1_PRIMARY *ppi);
3554 
3555 #if CONFIG_ENTROPY_STATS
3556 void print_entropy_stats(AV1_PRIMARY *const ppi);
3557 #endif
3558 #if CONFIG_INTERNAL_STATS
3559 void print_internal_stats(AV1_PRIMARY *ppi);
3560 #endif
3561 
3562 void av1_change_config_seq(AV1_PRIMARY *ppi, const AV1EncoderConfig *oxcf,
3563  bool *sb_size_changed);
3564 
3565 void av1_change_config(AV1_COMP *cpi, const AV1EncoderConfig *oxcf,
3566  bool sb_size_changed);
3567 
3568 void av1_check_initial_width(AV1_COMP *cpi, int use_highbitdepth,
3569  int subsampling_x, int subsampling_y);
3570 
3571 void av1_init_seq_coding_tools(AV1_PRIMARY *const ppi,
3572  const AV1EncoderConfig *oxcf, int use_svc);
3573 
3574 void av1_post_encode_updates(AV1_COMP *const cpi,
3575  const AV1_COMP_DATA *const cpi_data);
3576 
3577 void av1_scale_references_fpmt(AV1_COMP *cpi, int *ref_buffers_used_map);
3578 
3579 void av1_increment_scaled_ref_counts_fpmt(BufferPool *buffer_pool,
3580  int ref_buffers_used_map);
3581 
3582 void av1_release_scaled_references_fpmt(AV1_COMP *cpi);
3583 
3584 void av1_decrement_ref_counts_fpmt(BufferPool *buffer_pool,
3585  int ref_buffers_used_map);
3586 
3587 void av1_init_sc_decisions(AV1_PRIMARY *const ppi);
3588 
3589 AV1_COMP *av1_get_parallel_frame_enc_data(AV1_PRIMARY *const ppi,
3590  AV1_COMP_DATA *const first_cpi_data);
3591 
3592 int av1_init_parallel_frame_context(const AV1_COMP_DATA *const first_cpi_data,
3593  AV1_PRIMARY *const ppi,
3594  int *ref_buffers_used_map);
3614  YV12_BUFFER_CONFIG *sd, int64_t time_stamp,
3615  int64_t end_time_stamp);
3616 
3636 int av1_get_compressed_data(AV1_COMP *cpi, AV1_COMP_DATA *const cpi_data);
3637 
3644 int av1_encode(AV1_COMP *const cpi, uint8_t *const dest,
3645  const EncodeFrameInput *const frame_input,
3646  const EncodeFrameParams *const frame_params,
3647  EncodeFrameResults *const frame_results);
3648 
3650 int av1_get_preview_raw_frame(AV1_COMP *cpi, YV12_BUFFER_CONFIG *dest);
3651 
3652 int av1_get_last_show_frame(AV1_COMP *cpi, YV12_BUFFER_CONFIG *frame);
3653 
3654 aom_codec_err_t av1_copy_new_frame_enc(AV1_COMMON *cm,
3655  YV12_BUFFER_CONFIG *new_frame,
3656  YV12_BUFFER_CONFIG *sd);
3657 
3658 int av1_use_as_reference(int *ext_ref_frame_flags, int ref_frame_flags);
3659 
3660 int av1_copy_reference_enc(AV1_COMP *cpi, int idx, YV12_BUFFER_CONFIG *sd);
3661 
3662 int av1_set_reference_enc(AV1_COMP *cpi, int idx, YV12_BUFFER_CONFIG *sd);
3663 
3664 int av1_set_size_literal(AV1_COMP *cpi, int width, int height);
3665 
3666 void av1_set_frame_size(AV1_COMP *cpi, int width, int height);
3667 
3668 void av1_set_mv_search_params(AV1_COMP *cpi);
3669 
3670 int av1_set_active_map(AV1_COMP *cpi, unsigned char *map, int rows, int cols);
3671 
3672 int av1_get_active_map(AV1_COMP *cpi, unsigned char *map, int rows, int cols);
3673 
3674 int av1_set_internal_size(AV1EncoderConfig *const oxcf,
3675  ResizePendingParams *resize_pending_params,
3676  AOM_SCALING_MODE horiz_mode,
3677  AOM_SCALING_MODE vert_mode);
3678 
3679 int av1_get_quantizer(struct AV1_COMP *cpi);
3680 
3681 int av1_convert_sect5obus_to_annexb(uint8_t *buffer, size_t *input_size);
3682 
3683 // Set screen content options.
3684 // This function estimates whether to use screen content tools, by counting
3685 // the portion of blocks that have few luma colors.
3686 // Modifies:
3687 // cpi->commom.features.allow_screen_content_tools
3688 // cpi->common.features.allow_intrabc
3689 // cpi->use_screen_content_tools
3690 // cpi->is_screen_content_type
3691 // However, the estimation is not accurate and may misclassify videos.
3692 // A slower but more accurate approach that determines whether to use screen
3693 // content tools is employed later. See av1_determine_sc_tools_with_encoding().
3694 void av1_set_screen_content_options(struct AV1_COMP *cpi,
3695  FeatureFlags *features);
3696 
3697 void av1_update_frame_size(AV1_COMP *cpi);
3698 
3699 typedef struct {
3700  int pyr_level;
3701  int disp_order;
3702 } RefFrameMapPair;
3703 
3704 static INLINE void init_ref_map_pair(
3705  AV1_COMP *cpi, RefFrameMapPair ref_frame_map_pairs[REF_FRAMES]) {
3706  if (cpi->ppi->gf_group.update_type[cpi->gf_frame_index] == KF_UPDATE) {
3707  memset(ref_frame_map_pairs, -1, sizeof(*ref_frame_map_pairs) * REF_FRAMES);
3708  return;
3709  }
3710  memset(ref_frame_map_pairs, 0, sizeof(*ref_frame_map_pairs) * REF_FRAMES);
3711  for (int map_idx = 0; map_idx < REF_FRAMES; map_idx++) {
3712  // Get reference frame buffer.
3713  const RefCntBuffer *const buf = cpi->common.ref_frame_map[map_idx];
3714  if (ref_frame_map_pairs[map_idx].disp_order == -1) continue;
3715  if (buf == NULL) {
3716  ref_frame_map_pairs[map_idx].disp_order = -1;
3717  ref_frame_map_pairs[map_idx].pyr_level = -1;
3718  continue;
3719  } else if (buf->ref_count > 1) {
3720  // Once the keyframe is coded, the slots in ref_frame_map will all
3721  // point to the same frame. In that case, all subsequent pointers
3722  // matching the current are considered "free" slots. This will find
3723  // the next occurrence of the current pointer if ref_count indicates
3724  // there are multiple instances of it and mark it as free.
3725  for (int idx2 = map_idx + 1; idx2 < REF_FRAMES; ++idx2) {
3726  const RefCntBuffer *const buf2 = cpi->common.ref_frame_map[idx2];
3727  if (buf2 == buf) {
3728  ref_frame_map_pairs[idx2].disp_order = -1;
3729  ref_frame_map_pairs[idx2].pyr_level = -1;
3730  }
3731  }
3732  }
3733  ref_frame_map_pairs[map_idx].disp_order = (int)buf->display_order_hint;
3734  ref_frame_map_pairs[map_idx].pyr_level = buf->pyramid_level;
3735  }
3736 }
3737 
3738 #if CONFIG_FPMT_TEST
3739 static AOM_INLINE void calc_frame_data_update_flag(
3740  GF_GROUP *const gf_group, int gf_frame_index,
3741  bool *const do_frame_data_update) {
3742  *do_frame_data_update = true;
3743  // Set the flag to false for all frames in a given parallel encode set except
3744  // the last frame in the set with frame_parallel_level = 2.
3745  if (gf_group->frame_parallel_level[gf_frame_index] == 1) {
3746  *do_frame_data_update = false;
3747  } else if (gf_group->frame_parallel_level[gf_frame_index] == 2) {
3748  // Check if this is the last frame in the set with frame_parallel_level = 2.
3749  for (int i = gf_frame_index + 1; i < gf_group->size; i++) {
3750  if ((gf_group->frame_parallel_level[i] == 0 &&
3751  (gf_group->update_type[i] == ARF_UPDATE ||
3752  gf_group->update_type[i] == INTNL_ARF_UPDATE)) ||
3753  gf_group->frame_parallel_level[i] == 1) {
3754  break;
3755  } else if (gf_group->frame_parallel_level[i] == 2) {
3756  *do_frame_data_update = false;
3757  break;
3758  }
3759  }
3760  }
3761 }
3762 #endif
3763 
3764 // av1 uses 10,000,000 ticks/second as time stamp
3765 #define TICKS_PER_SEC 10000000LL
3766 
3767 static INLINE int64_t
3768 timebase_units_to_ticks(const aom_rational64_t *timestamp_ratio, int64_t n) {
3769  return n * timestamp_ratio->num / timestamp_ratio->den;
3770 }
3771 
3772 static INLINE int64_t
3773 ticks_to_timebase_units(const aom_rational64_t *timestamp_ratio, int64_t n) {
3774  int64_t round = timestamp_ratio->num / 2;
3775  if (round > 0) --round;
3776  return (n * timestamp_ratio->den + round) / timestamp_ratio->num;
3777 }
3778 
3779 static INLINE int frame_is_kf_gf_arf(const AV1_COMP *cpi) {
3780  const GF_GROUP *const gf_group = &cpi->ppi->gf_group;
3781  const FRAME_UPDATE_TYPE update_type =
3782  gf_group->update_type[cpi->gf_frame_index];
3783 
3784  return frame_is_intra_only(&cpi->common) || update_type == ARF_UPDATE ||
3785  update_type == GF_UPDATE;
3786 }
3787 
3788 // TODO(huisu@google.com, youzhou@microsoft.com): enable hash-me for HBD.
3789 static INLINE int av1_use_hash_me(const AV1_COMP *const cpi) {
3791  cpi->common.features.allow_intrabc &&
3792  frame_is_intra_only(&cpi->common));
3793 }
3794 
3795 static INLINE const YV12_BUFFER_CONFIG *get_ref_frame_yv12_buf(
3796  const AV1_COMMON *const cm, MV_REFERENCE_FRAME ref_frame) {
3797  const RefCntBuffer *const buf = get_ref_frame_buf(cm, ref_frame);
3798  return buf != NULL ? &buf->buf : NULL;
3799 }
3800 
3801 static INLINE void alloc_frame_mvs(AV1_COMMON *const cm, RefCntBuffer *buf) {
3802  assert(buf != NULL);
3803  ensure_mv_buffer(buf, cm);
3804  buf->width = cm->width;
3805  buf->height = cm->height;
3806 }
3807 
3808 // Get the allocated token size for a tile. It does the same calculation as in
3809 // the frame token allocation.
3810 static INLINE unsigned int allocated_tokens(const TileInfo *tile,
3811  int sb_size_log2, int num_planes) {
3812  int tile_mb_rows =
3813  ROUND_POWER_OF_TWO(tile->mi_row_end - tile->mi_row_start, 2);
3814  int tile_mb_cols =
3815  ROUND_POWER_OF_TWO(tile->mi_col_end - tile->mi_col_start, 2);
3816 
3817  return get_token_alloc(tile_mb_rows, tile_mb_cols, sb_size_log2, num_planes);
3818 }
3819 
3820 static INLINE void get_start_tok(AV1_COMP *cpi, int tile_row, int tile_col,
3821  int mi_row, TokenExtra **tok, int sb_size_log2,
3822  int num_planes) {
3823  AV1_COMMON *const cm = &cpi->common;
3824  const int tile_cols = cm->tiles.cols;
3825  TileDataEnc *this_tile = &cpi->tile_data[tile_row * tile_cols + tile_col];
3826  const TileInfo *const tile_info = &this_tile->tile_info;
3827 
3828  const int tile_mb_cols =
3829  (tile_info->mi_col_end - tile_info->mi_col_start + 2) >> 2;
3830  const int tile_mb_row = (mi_row - tile_info->mi_row_start + 2) >> 2;
3831 
3832  *tok = cpi->token_info.tile_tok[tile_row][tile_col] +
3833  get_token_alloc(tile_mb_row, tile_mb_cols, sb_size_log2, num_planes);
3834 }
3835 
3836 void av1_apply_encoding_flags(AV1_COMP *cpi, aom_enc_frame_flags_t flags);
3837 
3838 #define ALT_MIN_LAG 3
3839 static INLINE int is_altref_enabled(int lag_in_frames, bool enable_auto_arf) {
3840  return lag_in_frames >= ALT_MIN_LAG && enable_auto_arf;
3841 }
3842 
3843 static AOM_INLINE int can_disable_altref(const GFConfig *gf_cfg) {
3844  return is_altref_enabled(gf_cfg->lag_in_frames, gf_cfg->enable_auto_arf) &&
3845  (gf_cfg->gf_min_pyr_height == 0);
3846 }
3847 
3848 // Helper function to compute number of blocks on either side of the frame.
3849 static INLINE int get_num_blocks(const int frame_length, const int mb_length) {
3850  return (frame_length + mb_length - 1) / mb_length;
3851 }
3852 
3853 // Check if statistics generation stage
3854 static INLINE int is_stat_generation_stage(const AV1_COMP *const cpi) {
3855  assert(IMPLIES(cpi->compressor_stage == LAP_STAGE,
3856  cpi->oxcf.pass == AOM_RC_ONE_PASS && cpi->ppi->lap_enabled));
3857  return (cpi->oxcf.pass == AOM_RC_FIRST_PASS ||
3858  (cpi->compressor_stage == LAP_STAGE));
3859 }
3860 // Check if statistics consumption stage
3861 static INLINE int is_stat_consumption_stage_twopass(const AV1_COMP *const cpi) {
3862  return (cpi->oxcf.pass >= AOM_RC_SECOND_PASS);
3863 }
3864 
3865 // Check if statistics consumption stage
3866 static INLINE int is_stat_consumption_stage(const AV1_COMP *const cpi) {
3867  return (is_stat_consumption_stage_twopass(cpi) ||
3868  (cpi->oxcf.pass == AOM_RC_ONE_PASS &&
3869  (cpi->compressor_stage == ENCODE_STAGE) && cpi->ppi->lap_enabled));
3870 }
3871 
3872 // Decide whether 'dv_costs' need to be allocated/stored during the encoding.
3873 static AOM_INLINE bool av1_need_dv_costs(const AV1_COMP *const cpi) {
3874  return !cpi->sf.rt_sf.use_nonrd_pick_mode &&
3875  av1_allow_intrabc(&cpi->common) && !is_stat_generation_stage(cpi);
3876 }
3877 
3887 static INLINE int has_no_stats_stage(const AV1_COMP *const cpi) {
3888  assert(
3889  IMPLIES(!cpi->ppi->lap_enabled, cpi->compressor_stage == ENCODE_STAGE));
3890  return (cpi->oxcf.pass == AOM_RC_ONE_PASS && !cpi->ppi->lap_enabled);
3891 }
3892 
3895 static INLINE int is_one_pass_rt_params(const AV1_COMP *cpi) {
3896  return has_no_stats_stage(cpi) && cpi->oxcf.mode == REALTIME &&
3897  cpi->oxcf.gf_cfg.lag_in_frames == 0;
3898 }
3899 
3900 // Use default/internal reference structure for single-layer RTC.
3901 static INLINE int use_rtc_reference_structure_one_layer(const AV1_COMP *cpi) {
3902  return is_one_pass_rt_params(cpi) && cpi->ppi->number_spatial_layers == 1 &&
3903  cpi->ppi->number_temporal_layers == 1 &&
3904  !cpi->ppi->rtc_ref.set_ref_frame_config;
3905 }
3906 
3907 // Function return size of frame stats buffer
3908 static INLINE int get_stats_buf_size(int num_lap_buffer, int num_lag_buffer) {
3909  /* if lookahead is enabled return num_lap_buffers else num_lag_buffers */
3910  return (num_lap_buffer > 0 ? num_lap_buffer + 1 : num_lag_buffer);
3911 }
3912 
3913 // TODO(zoeliu): To set up cpi->oxcf.gf_cfg.enable_auto_brf
3914 
3915 static INLINE void set_ref_ptrs(const AV1_COMMON *cm, MACROBLOCKD *xd,
3916  MV_REFERENCE_FRAME ref0,
3917  MV_REFERENCE_FRAME ref1) {
3918  xd->block_ref_scale_factors[0] =
3919  get_ref_scale_factors_const(cm, ref0 >= LAST_FRAME ? ref0 : 1);
3920  xd->block_ref_scale_factors[1] =
3921  get_ref_scale_factors_const(cm, ref1 >= LAST_FRAME ? ref1 : 1);
3922 }
3923 
3924 static INLINE int get_chessboard_index(int frame_index) {
3925  return frame_index & 0x1;
3926 }
3927 
3928 static INLINE const int *cond_cost_list_const(const struct AV1_COMP *cpi,
3929  const int *cost_list) {
3930  const int use_cost_list = cpi->sf.mv_sf.subpel_search_method != SUBPEL_TREE &&
3931  cpi->sf.mv_sf.use_fullpel_costlist;
3932  return use_cost_list ? cost_list : NULL;
3933 }
3934 
3935 static INLINE int *cond_cost_list(const struct AV1_COMP *cpi, int *cost_list) {
3936  const int use_cost_list = cpi->sf.mv_sf.subpel_search_method != SUBPEL_TREE &&
3937  cpi->sf.mv_sf.use_fullpel_costlist;
3938  return use_cost_list ? cost_list : NULL;
3939 }
3940 
3941 // Compression ratio of current frame.
3942 double av1_get_compression_ratio(const AV1_COMMON *const cm,
3943  size_t encoded_frame_size);
3944 
3945 void av1_new_framerate(AV1_COMP *cpi, double framerate);
3946 
3947 void av1_setup_frame_size(AV1_COMP *cpi);
3948 
3949 #define LAYER_IDS_TO_IDX(sl, tl, num_tl) ((sl) * (num_tl) + (tl))
3950 
3951 // Returns 1 if a frame is scaled and 0 otherwise.
3952 static INLINE int av1_resize_scaled(const AV1_COMMON *cm) {
3953  return cm->superres_upscaled_width != cm->render_width ||
3955 }
3956 
3957 static INLINE int av1_frame_scaled(const AV1_COMMON *cm) {
3958  return av1_superres_scaled(cm) || av1_resize_scaled(cm);
3959 }
3960 
3961 // Don't allow a show_existing_frame to coincide with an error resilient
3962 // frame. An exception can be made for a forward keyframe since it has no
3963 // previous dependencies.
3964 static INLINE int encode_show_existing_frame(const AV1_COMMON *cm) {
3965  return cm->show_existing_frame && (!cm->features.error_resilient_mode ||
3966  cm->current_frame.frame_type == KEY_FRAME);
3967 }
3968 
3969 // Get index into the 'cpi->mbmi_ext_info.frame_base' array for the given
3970 // 'mi_row' and 'mi_col'.
3971 static INLINE int get_mi_ext_idx(const int mi_row, const int mi_col,
3972  const BLOCK_SIZE mi_alloc_bsize,
3973  const int mbmi_ext_stride) {
3974  const int mi_ext_size_1d = mi_size_wide[mi_alloc_bsize];
3975  const int mi_ext_row = mi_row / mi_ext_size_1d;
3976  const int mi_ext_col = mi_col / mi_ext_size_1d;
3977  return mi_ext_row * mbmi_ext_stride + mi_ext_col;
3978 }
3979 
3980 // Lighter version of set_offsets that only sets the mode info
3981 // pointers.
3982 static INLINE void set_mode_info_offsets(
3983  const CommonModeInfoParams *const mi_params,
3984  const MBMIExtFrameBufferInfo *const mbmi_ext_info, MACROBLOCK *const x,
3985  MACROBLOCKD *const xd, int mi_row, int mi_col) {
3986  set_mi_offsets(mi_params, xd, mi_row, mi_col);
3987  const int ext_idx = get_mi_ext_idx(mi_row, mi_col, mi_params->mi_alloc_bsize,
3988  mbmi_ext_info->stride);
3989  x->mbmi_ext_frame = mbmi_ext_info->frame_base + ext_idx;
3990 }
3991 
3992 // Check to see if the given partition size is allowed for a specified number
3993 // of mi block rows and columns remaining in the image.
3994 // If not then return the largest allowed partition size
3995 static INLINE BLOCK_SIZE find_partition_size(BLOCK_SIZE bsize, int rows_left,
3996  int cols_left, int *bh, int *bw) {
3997  int int_size = (int)bsize;
3998  if (rows_left <= 0 || cols_left <= 0) {
3999  return AOMMIN(bsize, BLOCK_8X8);
4000  } else {
4001  for (; int_size > 0; int_size -= 3) {
4002  *bh = mi_size_high[int_size];
4003  *bw = mi_size_wide[int_size];
4004  if ((*bh <= rows_left) && (*bw <= cols_left)) {
4005  break;
4006  }
4007  }
4008  }
4009  return (BLOCK_SIZE)int_size;
4010 }
4011 
4012 static const uint8_t av1_ref_frame_flag_list[REF_FRAMES] = { 0,
4013  AOM_LAST_FLAG,
4014  AOM_LAST2_FLAG,
4015  AOM_LAST3_FLAG,
4016  AOM_GOLD_FLAG,
4017  AOM_BWD_FLAG,
4018  AOM_ALT2_FLAG,
4019  AOM_ALT_FLAG };
4020 
4021 // When more than 'max_allowed_refs' are available, we reduce the number of
4022 // reference frames one at a time based on this order.
4023 static const MV_REFERENCE_FRAME disable_order[] = {
4024  LAST3_FRAME,
4025  LAST2_FRAME,
4026  ALTREF2_FRAME,
4027  BWDREF_FRAME,
4028 };
4029 
4030 static const MV_REFERENCE_FRAME
4031  ref_frame_priority_order[INTER_REFS_PER_FRAME] = {
4032  LAST_FRAME, ALTREF_FRAME, BWDREF_FRAME, GOLDEN_FRAME,
4033  ALTREF2_FRAME, LAST2_FRAME, LAST3_FRAME,
4034  };
4035 
4036 static INLINE int get_ref_frame_flags(const SPEED_FEATURES *const sf,
4037  const int use_one_pass_rt_params,
4038  const YV12_BUFFER_CONFIG **ref_frames,
4039  const int ext_ref_frame_flags) {
4040  // cpi->ext_flags.ref_frame_flags allows certain reference types to be
4041  // disabled by the external interface. These are set by
4042  // av1_apply_encoding_flags(). Start with what the external interface allows,
4043  // then suppress any reference types which we have found to be duplicates.
4044  int flags = ext_ref_frame_flags;
4045 
4046  for (int i = 1; i < INTER_REFS_PER_FRAME; ++i) {
4047  const YV12_BUFFER_CONFIG *const this_ref = ref_frames[i];
4048  // If this_ref has appeared before, mark the corresponding ref frame as
4049  // invalid. For one_pass_rt mode, only disable GOLDEN_FRAME if it's the
4050  // same as LAST_FRAME or ALTREF_FRAME (if ALTREF is being used in nonrd).
4051  int index =
4052  (use_one_pass_rt_params && ref_frame_priority_order[i] == GOLDEN_FRAME)
4053  ? (1 + sf->rt_sf.use_nonrd_altref_frame)
4054  : i;
4055  for (int j = 0; j < index; ++j) {
4056  // If this_ref has appeared before (same as the reference corresponding
4057  // to lower index j), remove it as a reference only if that reference
4058  // (for index j) is actually used as a reference.
4059  if (this_ref == ref_frames[j] &&
4060  (flags & (1 << (ref_frame_priority_order[j] - 1)))) {
4061  flags &= ~(1 << (ref_frame_priority_order[i] - 1));
4062  break;
4063  }
4064  }
4065  }
4066  return flags;
4067 }
4068 
4069 // Returns a Sequence Header OBU stored in an aom_fixed_buf_t, or NULL upon
4070 // failure. When a non-NULL aom_fixed_buf_t pointer is returned by this
4071 // function, the memory must be freed by the caller. Both the buf member of the
4072 // aom_fixed_buf_t, and the aom_fixed_buf_t pointer itself must be freed. Memory
4073 // returned must be freed via call to free().
4074 //
4075 // Note: The OBU returned is in Low Overhead Bitstream Format. Specifically,
4076 // the obu_has_size_field bit is set, and the buffer contains the obu_size
4077 // field.
4078 aom_fixed_buf_t *av1_get_global_headers(AV1_PRIMARY *ppi);
4079 
4080 #define MAX_GFUBOOST_FACTOR 10.0
4081 #define MIN_GFUBOOST_FACTOR 4.0
4082 
4083 static INLINE int is_frame_tpl_eligible(const GF_GROUP *const gf_group,
4084  uint8_t index) {
4085  const FRAME_UPDATE_TYPE update_type = gf_group->update_type[index];
4086  return update_type == ARF_UPDATE || update_type == GF_UPDATE ||
4087  update_type == KF_UPDATE;
4088 }
4089 
4090 static INLINE int is_frame_eligible_for_ref_pruning(const GF_GROUP *gf_group,
4091  int selective_ref_frame,
4092  int prune_ref_frames,
4093  int gf_index) {
4094  return (selective_ref_frame > 0) && (prune_ref_frames > 0) &&
4095  !is_frame_tpl_eligible(gf_group, gf_index);
4096 }
4097 
4098 // Get update type of the current frame.
4099 static INLINE FRAME_UPDATE_TYPE get_frame_update_type(const GF_GROUP *gf_group,
4100  int gf_frame_index) {
4101  return gf_group->update_type[gf_frame_index];
4102 }
4103 
4104 static INLINE int av1_pixels_to_mi(int pixels) {
4105  return ALIGN_POWER_OF_TWO(pixels, 3) >> MI_SIZE_LOG2;
4106 }
4107 
4108 static AOM_INLINE int is_psnr_calc_enabled(const AV1_COMP *cpi) {
4109  const AV1_COMMON *const cm = &cpi->common;
4110 
4111  return cpi->ppi->b_calculate_psnr && !is_stat_generation_stage(cpi) &&
4112  cm->show_frame;
4113 }
4114 
4115 static INLINE int is_frame_resize_pending(const AV1_COMP *const cpi) {
4116  const ResizePendingParams *const resize_pending_params =
4117  &cpi->resize_pending_params;
4118  return (resize_pending_params->width && resize_pending_params->height &&
4119  (cpi->common.width != resize_pending_params->width ||
4120  cpi->common.height != resize_pending_params->height));
4121 }
4122 
4123 // Check if loop filter is used.
4124 static INLINE int is_loopfilter_used(const AV1_COMMON *const cm) {
4125  return !cm->features.coded_lossless && !cm->tiles.large_scale;
4126 }
4127 
4128 // Check if CDEF is used.
4129 static INLINE int is_cdef_used(const AV1_COMMON *const cm) {
4130  return cm->seq_params->enable_cdef && !cm->features.coded_lossless &&
4131  !cm->tiles.large_scale;
4132 }
4133 
4134 // Check if loop restoration filter is used.
4135 static INLINE int is_restoration_used(const AV1_COMMON *const cm) {
4136  return cm->seq_params->enable_restoration && !cm->features.all_lossless &&
4137  !cm->tiles.large_scale;
4138 }
4139 
4140 // Checks if post-processing filters need to be applied.
4141 // NOTE: This function decides if the application of different post-processing
4142 // filters on the reconstructed frame can be skipped at the encoder side.
4143 // However the computation of different filter parameters that are signaled in
4144 // the bitstream is still required.
4145 static INLINE unsigned int derive_skip_apply_postproc_filters(
4146  const AV1_COMP *cpi, int use_loopfilter, int use_cdef, int use_superres,
4147  int use_restoration) {
4148  // Though CDEF parameter selection should be dependent on
4149  // deblocked/loop-filtered pixels for cdef_pick_method <=
4150  // CDEF_FAST_SEARCH_LVL5, CDEF strength values are calculated based on the
4151  // pixel values that are not loop-filtered in svc real-time encoding mode.
4152  // Hence this case is handled separately using the condition below.
4153  if (cpi->ppi->rtc_ref.non_reference_frame)
4154  return (SKIP_APPLY_LOOPFILTER | SKIP_APPLY_CDEF);
4155 
4157  return 0;
4158  assert(cpi->oxcf.mode == ALLINTRA);
4159 
4160  // The post-processing filters are applied one after the other in the
4161  // following order: deblocking->cdef->superres->restoration. In case of
4162  // ALLINTRA encoding, the reconstructed frame is not used as a reference
4163  // frame. Hence, the application of these filters can be skipped when
4164  // 1. filter parameters of the subsequent stages are not dependent on the
4165  // filtered output of the current stage or
4166  // 2. subsequent filtering stages are disabled
4167  if (use_restoration) return SKIP_APPLY_RESTORATION;
4168  if (use_superres) return SKIP_APPLY_SUPERRES;
4169  if (use_cdef) {
4170  // CDEF parameter selection is not dependent on the deblocked frame if
4171  // cdef_pick_method is CDEF_PICK_FROM_Q. Hence the application of deblocking
4172  // filters and cdef filters can be skipped in this case.
4173  return (cpi->sf.lpf_sf.cdef_pick_method == CDEF_PICK_FROM_Q &&
4174  use_loopfilter)
4175  ? (SKIP_APPLY_LOOPFILTER | SKIP_APPLY_CDEF)
4176  : SKIP_APPLY_CDEF;
4177  }
4178  if (use_loopfilter) return SKIP_APPLY_LOOPFILTER;
4179 
4180  return 0; // All post-processing stages disabled.
4181 }
4182 
4183 static INLINE void set_postproc_filter_default_params(AV1_COMMON *cm) {
4184  struct loopfilter *const lf = &cm->lf;
4185  CdefInfo *const cdef_info = &cm->cdef_info;
4186  RestorationInfo *const rst_info = cm->rst_info;
4187 
4188  lf->filter_level[0] = 0;
4189  lf->filter_level[1] = 0;
4190  cdef_info->cdef_bits = 0;
4191  cdef_info->cdef_strengths[0] = 0;
4192  cdef_info->nb_cdef_strengths = 1;
4193  cdef_info->cdef_uv_strengths[0] = 0;
4194  rst_info[0].frame_restoration_type = RESTORE_NONE;
4195  rst_info[1].frame_restoration_type = RESTORE_NONE;
4196  rst_info[2].frame_restoration_type = RESTORE_NONE;
4197 }
4198 
4199 static INLINE int is_inter_tx_size_search_level_one(
4200  const TX_SPEED_FEATURES *tx_sf) {
4201  return (tx_sf->inter_tx_size_search_init_depth_rect >= 1 &&
4202  tx_sf->inter_tx_size_search_init_depth_sqr >= 1);
4203 }
4204 
4205 static INLINE int get_lpf_opt_level(const SPEED_FEATURES *sf) {
4206  int lpf_opt_level = 0;
4207  if (is_inter_tx_size_search_level_one(&sf->tx_sf))
4208  lpf_opt_level = (sf->lpf_sf.lpf_pick == LPF_PICK_FROM_Q) ? 2 : 1;
4209  return lpf_opt_level;
4210 }
4211 
4212 // Enable switchable motion mode only if warp and OBMC tools are allowed
4213 static INLINE bool is_switchable_motion_mode_allowed(bool allow_warped_motion,
4214  bool enable_obmc) {
4215  return (allow_warped_motion || enable_obmc);
4216 }
4217 
4218 #if CONFIG_AV1_TEMPORAL_DENOISING
4219 static INLINE int denoise_svc(const struct AV1_COMP *const cpi) {
4220  return (!cpi->ppi->use_svc ||
4221  (cpi->ppi->use_svc &&
4222  cpi->svc.spatial_layer_id >= cpi->svc.first_layer_denoise));
4223 }
4224 #endif
4225 
4226 #if CONFIG_COLLECT_PARTITION_STATS == 2
4227 static INLINE void av1_print_fr_partition_timing_stats(
4228  const FramePartitionTimingStats *part_stats, const char *filename) {
4229  FILE *f = fopen(filename, "w");
4230  if (!f) {
4231  return;
4232  }
4233 
4234  fprintf(f, "bsize,redo,");
4235  for (int part = 0; part < EXT_PARTITION_TYPES; part++) {
4236  fprintf(f, "decision_%d,", part);
4237  }
4238  for (int part = 0; part < EXT_PARTITION_TYPES; part++) {
4239  fprintf(f, "attempt_%d,", part);
4240  }
4241  for (int part = 0; part < EXT_PARTITION_TYPES; part++) {
4242  fprintf(f, "time_%d,", part);
4243  }
4244  fprintf(f, "\n");
4245 
4246  static const int bsizes[6] = { 128, 64, 32, 16, 8, 4 };
4247 
4248  for (int bsize_idx = 0; bsize_idx < 6; bsize_idx++) {
4249  fprintf(f, "%d,%d,", bsizes[bsize_idx], part_stats->partition_redo);
4250  for (int part = 0; part < EXT_PARTITION_TYPES; part++) {
4251  fprintf(f, "%d,", part_stats->partition_decisions[bsize_idx][part]);
4252  }
4253  for (int part = 0; part < EXT_PARTITION_TYPES; part++) {
4254  fprintf(f, "%d,", part_stats->partition_attempts[bsize_idx][part]);
4255  }
4256  for (int part = 0; part < EXT_PARTITION_TYPES; part++) {
4257  fprintf(f, "%ld,", part_stats->partition_times[bsize_idx][part]);
4258  }
4259  fprintf(f, "\n");
4260  }
4261  fclose(f);
4262 }
4263 #endif // CONFIG_COLLECT_PARTITION_STATS == 2
4264 
4265 #if CONFIG_COLLECT_PARTITION_STATS
4266 static INLINE int av1_get_bsize_idx_for_part_stats(BLOCK_SIZE bsize) {
4267  assert(bsize == BLOCK_128X128 || bsize == BLOCK_64X64 ||
4268  bsize == BLOCK_32X32 || bsize == BLOCK_16X16 || bsize == BLOCK_8X8 ||
4269  bsize == BLOCK_4X4);
4270  switch (bsize) {
4271  case BLOCK_128X128: return 0;
4272  case BLOCK_64X64: return 1;
4273  case BLOCK_32X32: return 2;
4274  case BLOCK_16X16: return 3;
4275  case BLOCK_8X8: return 4;
4276  case BLOCK_4X4: return 5;
4277  default: assert(0 && "Invalid bsize for partition_stats."); return -1;
4278  }
4279 }
4280 #endif // CONFIG_COLLECT_PARTITION_STATS
4281 
4282 #if CONFIG_COLLECT_COMPONENT_TIMING
4283 static INLINE void start_timing(AV1_COMP *cpi, int component) {
4284  aom_usec_timer_start(&cpi->component_timer[component]);
4285 }
4286 static INLINE void end_timing(AV1_COMP *cpi, int component) {
4287  aom_usec_timer_mark(&cpi->component_timer[component]);
4288  cpi->frame_component_time[component] +=
4289  aom_usec_timer_elapsed(&cpi->component_timer[component]);
4290 }
4291 static INLINE char const *get_frame_type_enum(int type) {
4292  switch (type) {
4293  case 0: return "KEY_FRAME";
4294  case 1: return "INTER_FRAME";
4295  case 2: return "INTRA_ONLY_FRAME";
4296  case 3: return "S_FRAME";
4297  default: assert(0);
4298  }
4299  return "error";
4300 }
4301 #endif
4302 
4305 #ifdef __cplusplus
4306 } // extern "C"
4307 #endif
4308 
4309 #endif // AOM_AV1_ENCODER_ENCODER_H_
int seq_params_locked
Definition: encoder.h:2564
bool use_ref_frame_mvs
Definition: encoder.h:2207
int ref_idx_to_skip
Definition: encoder.h:3354
int sb_counter
Definition: encoder.h:3320
bool enable_masked_comp
Definition: encoder.h:404
RATE_CONTROL rc
Definition: encoder.h:2961
int64_t ts_start_last_show_frame
Definition: encoder.h:2516
CB_COEFF_BUFFER * coeff_buffer_base
Definition: encoder.h:2791
Encoder flags for intra prediction.
Definition: encoder.h:298
int key_freq_max
Definition: encoder.h:471
WinnerModeParams winner_mode_params
Definition: encoder.h:2951
CdefInfo cdef_info
Definition: av1_common_int.h:955
int do_update_frame_probs_interpfilter[10]
Definition: encoder.h:3118
int over_shoot_pct
Definition: encoder.h:585
RTC_REF rtc_ref
Definition: encoder.h:2753
Definition: encoder.h:220
bool refresh_idx_available
Definition: encoder.h:3347
int all_one_sided_refs
Definition: encoder.h:2992
bool enable_interintra_wedge
Definition: encoder.h:420
pthread_mutex_t * mutex_
Definition: encoder.h:1372
AV1_COMMON common
Definition: encoder.h:2802
MultiThreadInfo mt_info
Definition: encoder.h:3142
bool reduced_tx_type_set
Definition: encoder.h:370
int show_frame
Definition: av1_common_int.h:885
int allocated_cols
Definition: encoder.h:1501
bool enable_superres
Definition: encoder.h:456
int segment_map_w
Definition: encoder.h:1976
int * num_finished_cols
Definition: encoder.h:1381
uint8_t * entropy_ctx
Definition: encoder.h:2349
int speed
Definition: encoder.h:3528
int num_threads_working
Definition: encoder.h:1407
Primary Encoder parameters related to multi-threading.
Definition: encoder.h:1585
bool enable_smooth_intra
Definition: encoder.h:311
int do_update_frame_probs_warp[10]
Definition: encoder.h:3113
int cq_level
Definition: encoder.h:599
RdIdxPair rd_idx_pair_arr[MAX_INTER_MODES]
Definition: encoder.h:1275
int num_frame_recode
Definition: encoder.h:3093
int * num_tile_cols_done
Definition: encoder.h:1512
RateControlCfg rc_cfg
Definition: encoder.h:937
Rate Control parameters and status.
Definition: ratectrl.h:134
bool alt_ref_frame
Definition: encoder.h:2066
uint16_t * eobs
Definition: encoder.h:2345
unsigned int vbr_corpus_complexity_lap
Definition: encoder.h:550
struct loopfilter lf
Definition: av1_common_int.h:939
bool skip_postproc_filtering
Definition: encoder.h:869
bool enable_tx_size_search
Definition: encoder.h:389
int disable_trellis_quant
Definition: encoder.h:825
int superres_kf_qthresh
Definition: encoder.h:436
FRAME_TYPE frame_type
Definition: encoder.h:3490
InitialDimensions initial_dimensions
Definition: encoder.h:3050
Parameters related to temporal filtering.
Definition: temporal_filter.h:93
int use_ducky_encode
Definition: encoder.h:3447
AV1LevelParams level_params
Definition: encoder.h:2600
MV_STATS mv_stats
Definition: encoder.h:3337
double framerate
Definition: encoder.h:2966
TplParams tpl_data
Definition: encoder.h:2673
FeatureFlags features
Definition: av1_common_int.h:905
CYCLIC_REFRESH * cyclic_refresh
Definition: encoder.h:3002
BLOCK_SIZE weber_bsize
Definition: encoder.h:3387
aom_rc_mode
Rate control mode.
Definition: aom_encoder.h:183
double * tpl_rdmult_scaling_factors
Definition: encoder.h:2877
Frame refresh flags set by the external interface.
Definition: encoder.h:2167
static int has_no_stats_stage(const AV1_COMP *const cpi)
Check if the current stage has statistics.
Definition: encoder.h:3887
int allocated_sb_rows
Definition: encoder.h:1518
MV_SPEED_FEATURES mv_sf
Definition: speed_features.h:1715
Definition: encoder.h:221
Encoder rate control configuration parameters.
Definition: encoder.h:520
int filter_level[2]
Definition: encoder.h:2531
PRIMARY_RATE_CONTROL p_rc
Definition: encoder.h:2620
TEMPORAL_FILTER_INFO tf_info
Definition: encoder.h:2625
AV1TemporalFilterSync tf_sync
Definition: encoder.h:1693
bool is_dropped_frame
Definition: encoder.h:3402
Contains color maps used in palette mode.
Definition: block.h:325
bool update_pending
Definition: encoder.h:2176
int valid_gm_model_found[FRAME_UPDATE_TYPES]
Definition: encoder.h:2748
int sharpness
Definition: encoder.h:816
pthread_cond_t * cond_
Definition: encoder.h:1528
AV1GlobalMotionSync gm_sync
Definition: encoder.h:1688
size_t available_bs_size
Definition: encoder.h:3325
CODING_CONTEXT coding_context
Definition: encoder.h:2941
Definition: encoder.h:229
fractional_mv_step_fp * find_fractional_mv_step
Definition: encoder.h:2045
int show_existing_alt_ref
Definition: encoder.h:2580
TokenInfo token_info
Definition: encoder.h:3078
Encoder flags for compound prediction modes.
Definition: encoder.h:395
struct EncWorkerData * tile_thr_data
Definition: encoder.h:1605
int pipeline_lpf_mt_with_enc
Definition: encoder.h:1714
int cdef_bits
Number of CDEF strength values in bits.
Definition: av1_common_int.h:218
bool enable_angle_delta
Definition: encoder.h:333
enum aom_color_primaries aom_color_primaries_t
List of supported color primaries.
Provides definitions for using AOM or AV1 encoder algorithm within the aom Codec Interface.
bool alt_ref_frame
Definition: encoder.h:2172
MotionVectorSearchParams mv_search_params
Definition: encoder.h:2986
Encoder config related to the coding of key frames.
Definition: encoder.h:462
struct aom_codec_pkt_list * output_pkt_list
Definition: encoder.h:2570
int rows
Definition: encoder.h:1399
int num_mod_workers[NUM_MT_MODULES]
Definition: encoder.h:1636
int64_t ts_end_last_show_frame
Definition: encoder.h:2521
Definition: encoder.h:232
bool enable_interinter_wedge
Definition: encoder.h:416
enum aom_scaling_mode_1d AOM_SCALING_MODE
aom 1-D scaling mode
enum aom_matrix_coefficients aom_matrix_coefficients_t
List of supported matrix coefficients.
struct aom_internal_error_info error
Definition: encoder.h:2655
Stores the transforms coefficients for the whole superblock.
Definition: block.h:177
struct AV1_COMP * cpi
Definition: encoder.h:2547
unsigned char * cx_data
Definition: encoder.h:2418
bool enable_1to4_partitions
Definition: encoder.h:282
Definition: encoder.h:2337
AV1EncoderConfig oxcf
Definition: encoder.h:2807
RD_STATS rd_cost_uv_arr[MAX_INTER_MODES]
Definition: encoder.h:1287
struct EncodeFrameInput EncodeFrameInput
Input frames and last input frame.
int allocated_tiles
Definition: encoder.h:3073
Buffers to be backed up during parallel encode set to be restored later.
Definition: encoder.h:1560
int cdef_uv_strengths[16]
CDEF strength values for chroma.
Definition: av1_common_int.h:216
int do_update_frame_probs_obmc[10]
Definition: encoder.h:3108
bool enable_paeth_intra
Definition: encoder.h:315
MB_MODE_INFO_EXT_FRAME * frame_base
Definition: encoder.h:1781
TemporalFilterCtx tf_ctx
Definition: encoder.h:2882
bool use_intra_dct_only
Definition: encoder.h:375
FRAME_INDEX_SET frame_index_set
Definition: encoder.h:3045
int superres_upscaled_height
Definition: av1_common_int.h:800
aom_film_grain_table_t * film_grain_table
Definition: encoder.h:3169
int num
Definition: encoder.h:1255
int frames_left
Definition: encoder.h:2610
Encoder flags for transform sizes and types.
Definition: encoder.h:353
#define MAX_PARALLEL_FRAMES
Max number of frames that can be encoded in a parallel encode set.
Definition: encoder.h:1554
int rt_reduce_num_ref_buffers
Definition: encoder.h:2908
aom_fixed_buf_t twopass_stats_in
Definition: encoder.h:954
AVxWorker * workers
Definition: encoder.h:1599
PrimaryMultiThreadInfo p_mt_info
Definition: encoder.h:2735
SKIP_APPLY_POSTPROC_FILTER
This enum controls the application of post-processing filters on a reconstructed frame.
Definition: encoder.h:240
int use_screen_content_tools
Definition: encoder.h:3191
bool enable_dist_wtd_comp
Definition: encoder.h:399
struct aom_denoise_and_model_t * denoise_and_model
Definition: encoder.h:3176
bool all_lossless
Definition: av1_common_int.h:389
pthread_cond_t * cond_
Definition: encoder.h:1373
Contains buffers used by av1_compound_type_rd()
Definition: block.h:337
WeberStats * mb_weber_stats
Definition: encoder.h:3382
Parameters related to global motion search.
Definition: encoder.h:1939
YV12_BUFFER_CONFIG * unscaled_last_source
Definition: encoder.h:2845
Generic fixed size buffer structure.
Definition: aom_encoder.h:86
int64_t sse_arr[MAX_INTER_MODES]
Definition: encoder.h:1267
uint16_t * cdef_colbuf[3]
Definition: encoder.h:1569
YV12_BUFFER_CONFIG * source
Definition: encoder.h:2820
int sync_range
Definition: encoder.h:1388
Frame time stamps.
Definition: encoder.h:2318
Data related to the current GF/ARF group and the individual frames within the group.
Definition: firstpass.h:344
Encoder info used for decision on forcing integer motion vectors.
Definition: encoder.h:1730
FRAME_INFO frame_info
Definition: encoder.h:3040
Definition: encoder.h:230
int ref_refresh_index
Definition: encoder.h:3341
int num_mod_workers[NUM_MT_MODULES]
Definition: encoder.h:1594
GlobalMotionInfo gm_info
Definition: encoder.h:2946
MB_MODE_INFO_EXT_FRAME * mbmi_ext_frame
Finalized mbmi_ext for the whole frame.
Definition: block.h:881
int p_num_workers
Definition: encoder.h:1621
aom_superres_mode superres_mode
Definition: encoder.h:452
size_t frame_size
Definition: encoder.h:2428
int stride
Definition: encoder.h:1789
int av1_get_compressed_data(AV1_COMP *cpi, AV1_COMP_DATA *const cpi_data)
Encode a frame.
Definition: encoder.c:4588
int8_t nearest_future_ref
Definition: encoder.h:2096
AV1LrStruct lr_ctxt
Definition: encoder.h:3164
int alloc_size
Definition: encoder.h:1785
int nb_cdef_strengths
Number of CDEF strength values.
Definition: av1_common_int.h:212
Definition: encoder.h:218
int num_tg
Definition: encoder.h:3279
Params related to temporal dependency model.
Definition: tpl_model.h:146
DuckyEncodeInfo ducky_encode_info
Definition: encoder.h:3453
struct AV1_COMP * parallel_cpi[4]
Definition: encoder.h:2473
AV1LfSync lf_row_sync
Definition: encoder.h:1673
NOISE_ESTIMATE noise_estimate
Definition: encoder.h:3296
int num_workers
Definition: encoder.h:1631
bool allow_intrabc
Definition: av1_common_int.h:376
TileDataEnc * tile_data
Definition: encoder.h:3069
double new_framerate
Definition: encoder.h:3132
bool golden_frame
Definition: encoder.h:2169
LOOPFILTER_CONTROL loopfilter_control
Definition: encoder.h:863
uint64_t * src_sad_blk_64x64
Definition: encoder.h:3436
struct EncWorkerData * tile_thr_data
Definition: encoder.h:1647
SVC svc
Definition: encoder.h:3263
RestorationInfo rst_info[3]
Definition: av1_common_int.h:946
int internal_altref_allowed
Definition: encoder.h:2575
int cdef_strengths[16]
CDEF strength values for luma.
Definition: av1_common_int.h:214
int64_t starting_buffer_level_ms
Definition: encoder.h:529
int vbrmax_section
Definition: encoder.h:621
bool enable_ab_partitions
Definition: encoder.h:278
Structure to hold data corresponding to an encoded frame.
Definition: encoder.h:2414
double * tpl_sb_rdmult_scaling_factors
Definition: encoder.h:2668
int key_freq_min
Definition: encoder.h:466
int under_shoot_pct
Definition: encoder.h:579
RestoreStateBuffers restore_state_buf
Definition: encoder.h:1708
Encoder config related to resize.
Definition: encoder.h:250
Definition: enums.h:608
Struct used to hold inter mode data for fast tx search.
Definition: encoder.h:1250
KeyFrameCfg kf_cfg
Definition: encoder.h:932
MV_STATS mv_stats
Definition: encoder.h:2678
uint8_t resize_scale_denominator
Definition: encoder.h:259
bool allow_screen_content_tools
Definition: av1_common_int.h:375
Top level primary encoder structure.
Definition: encoder.h:2469
int vbrbias
Definition: encoder.h:611
int64_t norm_wiener_variance
Definition: encoder.h:3392
TimeStamps time_stamps
Definition: encoder.h:2956
struct PrimaryMultiThreadInfo PrimaryMultiThreadInfo
Primary Encoder parameters related to multi-threading.
uint8_t superres_scale_denominator
Definition: encoder.h:442
TWO_PASS twopass
Definition: encoder.h:2615
int64_t prev_ts_end
Definition: encoder.h:2326
Two pass status and control data.
Definition: firstpass.h:421
int mode_rate_arr[MAX_INTER_MODES]
Definition: encoder.h:1263
bool enable_rect_tx
Definition: encoder.h:365
Encoder-side probabilities for pruning of various AV1 tools.
Definition: encoder.h:1094
int skip_tpl_setup_stats
Definition: encoder.h:2867
ExtPartController ext_part_controller
Definition: encoder.h:3331
AV1EncPackBSSync pack_bs_sync
Definition: encoder.h:1683
COST_UPDATE_TYPE
This enum controls how often the entropy costs should be updated.
Definition: encoder.h:217
Encoder config for coding block partitioning.
Definition: encoder.h:270
BLOCK_SIZE max_partition_size
Definition: encoder.h:292
uint16_t interp_filter_search_mask
Definition: encoder.h:2019
enum aom_superblock_size aom_superblock_size_t
Superblock size selection.
bool enable_overlay
Definition: encoder.h:854
SPEED_FEATURES sf
Definition: encoder.h:2981
int sframe_dist
Definition: encoder.h:482
int best_allowed_q
Definition: encoder.h:595
bool golden_frame
Definition: encoder.h:2064
int ref_frame_flags
Definition: encoder.h:3512
GF_STATE gf_state
Definition: encoder.h:2590
Flags related to interpolation filter search.
Definition: encoder.h:2010
int width
Definition: av1_common_int.h:775
AV1CdefWorkerData * cdef_worker
Definition: encoder.h:1703
int64_t ambient_err
Definition: encoder.h:2930
int allocated_tile_rows
Definition: encoder.h:1483
int enable_keyframe_filtering
Definition: encoder.h:476
RefCntBuffer * ref_frame_map[REF_FRAMES]
Definition: av1_common_int.h:878
int num_src_corners
Definition: encoder.h:1983
bool use_inter_dct_only
Definition: encoder.h:380
ThreadData td
Definition: encoder.h:2774
bool pack_bs_mt_enabled
Definition: encoder.h:1658
Definition: encoder.h:222
aom_dist_metric
Distortion metric to use for RD optimization.
Definition: aomcx.h:1592
int worst_allowed_q
Definition: encoder.h:590
Thresholds for variance based partitioning.
Definition: encoder.h:1345
ResizePendingParams resize_pending_params
Definition: encoder.h:3063
GF_GROUP gf_group
Definition: encoder.h:2585
int allocated_rows
Definition: encoder.h:1494
bool fwd_kf_enabled
Definition: encoder.h:504
contains per-frame encoding parameters decided upon by av1_encode_strategy() and passed down to av1_e...
Definition: encoder.h:3482
Params related to MB_MODE_INFO arrays and related info.
Definition: av1_common_int.h:501
int render_height
Definition: av1_common_int.h:787
Definition: aom_encoder.h:175
SequenceHeader seq_params
Definition: encoder.h:2630
BLOCK_SIZE min_partition_size
Definition: encoder.h:287
int fwd_kf_dist
Definition: encoder.h:499
Top level speed vs quality trade off data struture.
Definition: speed_features.h:1686
int allocated_tile_cols
Definition: encoder.h:1487
RefreshFrameInfo refresh_frame
Definition: encoder.h:2903
bool coded_lossless
Definition: av1_common_int.h:385
int rate_size
Definition: encoder.h:1743
int vaq_refresh
Definition: encoder.h:3083
TX_SPEED_FEATURES tx_sf
Definition: speed_features.h:1735
bool enable_filter_intra
Definition: encoder.h:307
int lap_enabled
Definition: encoder.h:2595
struct RestoreStateBuffers RestoreStateBuffers
Buffers to be backed up during parallel encode set to be restored later.
Contains buffers used to speed up rdopt for obmc.
Definition: block.h:300
Parameters for motion vector search process.
Definition: encoder.h:2025
bool do_frame_data_update
Definition: encoder.h:3370
BLOCK_SIZE fp_block_size
Definition: encoder.h:3314
Frame level features.
Definition: av1_common_int.h:358
const struct scale_factors * block_ref_scale_factors[2]
Definition: blockd.h:691
Definition: speed_features.h:164
bool last_frame
Definition: encoder.h:2168
THIRD_PASS_DEC_CTX * third_pass_ctx
Definition: encoder.h:3426
SequenceHeader * seq_params
Definition: av1_common_int.h:976
struct AV1_COMP * cpi_lap
Definition: encoder.h:2552
LOOP_FILTER_SPEED_FEATURES lpf_sf
Definition: speed_features.h:1750
RefCntBuffer * scaled_ref_buf[INTER_REFS_PER_FRAME]
Definition: encoder.h:2893
AV1_PRIMARY * ppi
Definition: encoder.h:2763
bool enable_tpl_model
Definition: encoder.h:848
Parameters related to Restoration Info.
Definition: restoration.h:256
AlgoCfg algo_cfg
Definition: encoder.h:927
int frames_since_last_update
Definition: encoder.h:3459
unsigned int max_intra_bitrate_pct
Definition: encoder.h:555
int show_frame
Definition: encoder.h:3500
struct AV1_COMP AV1_COMP
Top level encoder structure.
double * ssim_rdmult_scaling_factors
Definition: encoder.h:3244
struct lookahead_ctx * lookahead
Definition: encoder.h:2557
bool enable_intrabc
Definition: encoder.h:514
AV1EncRowMultiThreadInfo enc_row_mt
Definition: encoder.h:1663
struct AV1EncoderConfig AV1EncoderConfig
Main encoder configuration data structure.
FirstPassData firstpass_data
Definition: encoder.h:3291
bool use_intra_default_tx_only
Definition: encoder.h:385
int existing_fb_idx_to_show
Definition: encoder.h:3149
EncQuantDequantParams enc_quant_dequant_params
Definition: encoder.h:2769
RESIZE_MODE resize_mode
Definition: encoder.h:254
aom_enc_pass
Multi-pass Encoding Pass.
Definition: aom_encoder.h:174
uint8_t resize_kf_scale_denominator
Definition: encoder.h:264
unsigned int gf_cbr_boost_pct
Definition: encoder.h:564
CommonTileParams tiles
Definition: av1_common_int.h:992
int superres_qthresh
Definition: encoder.h:431
bool auto_intra_tools_off
Definition: encoder.h:347
int use_svc
Definition: encoder.h:2635
CurrentFrame current_frame
Definition: av1_common_int.h:754
int render_width
Definition: av1_common_int.h:786
Describes look ahead buffer operations.
struct inter_modes_info InterModesInfo
Struct used to hold inter mode data for fast tx search.
InterpSearchFlags interp_search_flags
Definition: encoder.h:3182
int do_update_frame_probs_txtype[10]
Definition: encoder.h:3103
int height
Definition: encoder.h:2004
RefCntBuffer * ref_frame_map_copy[REF_FRAMES]
Definition: encoder.h:2511
struct AV1_COMP_DATA parallel_frames_data[4 - 1]
Definition: encoder.h:2479
unsigned int number_spatial_layers
Definition: encoder.h:2650
int remapped_ref_idx[REF_FRAMES]
Definition: encoder.h:3517
YV12 frame buffer data structure.
Definition: yv12config.h:39
YV12_BUFFER_CONFIG * unscaled_source
Definition: encoder.h:2835
Desired dimensions for an externally triggered resize.
Definition: encoder.h:2076
RefFrameDistanceInfo ref_frame_dist_info
Definition: encoder.h:3237
bool bwd_ref_frame
Definition: encoder.h:2170
Input frames and last input frame.
Definition: encoder.h:3470
FRAME_TYPE last_frame_type
Definition: encoder.h:3274
AVxWorker * workers
Definition: encoder.h:1641
int64_t threshold_minmax
Definition: encoder.h:1360
uint8_t superres_kf_scale_denominator
Definition: encoder.h:448
AV1CdefSync cdef_sync
Definition: encoder.h:1698
int error_resilient_mode
Definition: encoder.h:3486
int arnr_max_frames
Definition: encoder.h:830
bool enable_sframe
Definition: encoder.h:509
Definition: encoder.h:231
int flush
Definition: encoder.h:2448
Parameters related to CDEF.
Definition: av1_common_int.h:196
TWO_PASS_FRAME twopass_frame
Definition: encoder.h:3421
int mv_step_param
Definition: encoder.h:2036
aom_tune_content
Definition: aomcx.h:1554
YV12_BUFFER_CONFIG trial_frame_rst
Definition: encoder.h:2925
uint8_t * consec_zero_mv
Definition: encoder.h:3309
bool enable_intra_edge_filter
Definition: encoder.h:302
bool search_done
Definition: encoder.h:1943
int num_fp_contexts
Definition: encoder.h:2526
int av1_receive_raw_frame(AV1_COMP *cpi, aom_enc_frame_flags_t frame_flags, YV12_BUFFER_CONFIG *sd, int64_t time_stamp, int64_t end_time_stamp)
Obtain the raw frame data.
Definition: encoder.c:3997
int32_t * rst_tmpbuf
Definition: encoder.h:1574
unsigned int min_cr
Definition: encoder.h:569
int deltaq_used
Definition: encoder.h:3232
bool refresh_frame_context
Definition: encoder.h:2196
aom_variance_fn_ptr_t fn_ptr[BLOCK_SIZES_ALL]
Definition: encoder.h:2662
Encoder data related to row-based multi-threading.
Definition: encoder.h:1479
int64_t maximum_buffer_size_ms
Definition: encoder.h:539
int max_mv_magnitude
Definition: encoder.h:2031
int height
Definition: encoder.h:2078
Variables related to current coding block.
Definition: blockd.h:574
TRELLIS_OPT_TYPE optimize_seg_arr[8]
Definition: encoder.h:2813
int64_t prev_ts_start
Definition: encoder.h:2322
Parameters used for winner mode processing.
Definition: encoder.h:2114
bool refresh_frame_context_pending
Definition: encoder.h:2202
Segmentation related information for the current frame.
Definition: encoder.h:2300
int droppable
Definition: encoder.h:3035
int64_t target_bandwidth
Definition: encoder.h:544
enum aom_enc_pass pass
Definition: encoder.h:1040
CoeffBufferPool coeff_buffer_pool
Definition: encoder.h:2797
int frame_display_order_hint
Definition: encoder.h:2463
uint8_t cdf_update_mode
Definition: encoder.h:843
LOOPFILTER_CONTROL
This enum controls to which frames loopfilter is applied.
Definition: encoder.h:228
bool enable_smooth_interintra
Definition: encoder.h:408
Top level encoder structure.
Definition: encoder.h:2759
int intrabc_extra_top_right_sb_delay
Definition: encoder.h:1395
int drop_frames_water_mark
Definition: encoder.h:573
int intrabc_used
Definition: encoder.h:3154
bool enable_flip_idtx
Definition: encoder.h:361
int initial_mbs
Definition: encoder.h:3058
int vbrmin_section
Definition: encoder.h:616
unsigned int number_temporal_layers
Definition: encoder.h:2645
enum aom_color_range aom_color_range_t
List of supported color range.
int cols
Definition: av1_common_int.h:428
uint64_t rec_sse
Definition: encoder.h:3441
unsigned int zeromv_skip_thresh_exit_part[BLOCK_SIZES_ALL]
Definition: encoder.h:3464
enum aom_bit_depth aom_bit_depth_t
Bit depth for codecThis enumeration determines the bit depth of the codec.
struct MultiThreadInfo MultiThreadInfo
Encoder parameters related to multi-threading.
VarBasedPartitionInfo vbp_info
Definition: encoder.h:3088
int filter_level_u
Definition: encoder.h:2536
Definition: enums.h:604
bool has_lossless_segment
Definition: encoder.h:2312
int pop_lookahead
Definition: encoder.h:2458
int8_t nearest_past_ref
Definition: encoder.h:2092
FRAME_COUNTS counts
Definition: encoder.h:2779
int av1_encode(AV1_COMP *const cpi, uint8_t *const dest, const EncodeFrameInput *const frame_input, const EncodeFrameParams *const frame_params, EncodeFrameResults *const frame_results)
Run 1-pass/2-pass encoding.
Definition: encoder.c:3899
ActiveMap active_map
Definition: encoder.h:3007
aom_codec_err_t
Algorithm return codes.
Definition: aom_codec.h:155
REAL_TIME_SPEED_FEATURES rt_sf
Definition: speed_features.h:1755
const aom_rational64_t * timestamp_ratio
Definition: encoder.h:2453
EncSegmentationInfo enc_seg
Definition: encoder.h:2997
RefreshFrameInfo refresh_frame
Definition: encoder.h:3523
YV12_BUFFER_CONFIG last_frame_uf
Definition: encoder.h:2919
Top level common structure used by both encoder and decoder.
Definition: av1_common_int.h:750
int ref_frame_flags
Definition: encoder.h:2186
Algorithm configuration parameters.
Definition: encoder.h:809
aom_superres_mode superres_mode
Definition: encoder.h:3286
int segment_map_h
Definition: encoder.h:1977
Encoder config related to frame super-resolution.
Definition: encoder.h:426
enum aom_chroma_sample_position aom_chroma_sample_position_t
List of chroma sample positions.
Temporal filter info for a gop.
Definition: temporal_filter.h:156
Stores the prediction/txfm mode of the current coding block.
Definition: blockd.h:222
Refresh frame flags for different type of frames.
Definition: encoder.h:2063
unsigned char * src_buffer
Definition: encoder.h:1955
RestorationLineBuffers * rlbs
Definition: encoder.h:1579
unsigned int large_scale
Definition: av1_common_int.h:488
AV1CdefWorkerData * cdef_worker
Definition: encoder.h:1610
bool enable_diagonal_intra
Definition: encoder.h:328
bool alt2_ref_frame
Definition: encoder.h:2171
int filter_level_v
Definition: encoder.h:2541
int superres_upscaled_width
Definition: av1_common_int.h:799
Initial frame dimensions.
Definition: encoder.h:2002
bool enable_diff_wtd_comp
Definition: encoder.h:412
ExternalFlags ext_flags
Definition: encoder.h:2913
Refrence frame distance related variables.
Definition: encoder.h:2084
YV12_BUFFER_CONFIG scaled_last_source
Definition: encoder.h:2850
Encoder parameters related to multi-threading.
Definition: encoder.h:1627
uint8_t * map
Definition: encoder.h:2306
BLOCK_SIZE mi_alloc_bsize
Definition: av1_common_int.h:550
ExtRefreshFrameFlagsInfo refresh_frame
Definition: encoder.h:2191
Buffer to store mode information at mi_alloc_bsize (4x4 or 8x8) level.
Definition: encoder.h:1776
bool enable_rect_partitions
Definition: encoder.h:274
bool enable_directional_intra
Definition: encoder.h:323
int speed
Definition: encoder.h:2976
int is_screen_content_type
Definition: encoder.h:3199
int fb_of_context_type[REF_FRAMES]
Definition: encoder.h:2730
RD_STATS rd_cost_arr[MAX_INTER_MODES]
Definition: encoder.h:1279
ForceIntegerMVInfo force_intpel_info
Definition: encoder.h:2887
bool use_s_frame
Definition: encoder.h:2217
FrameProbInfo frame_new_probs[10]
Definition: encoder.h:3098
#define NUM_RECODES_PER_FRAME
Max number of recodes used to track the frame probabilities.
Definition: encoder.h:1549
bool use_primary_ref_none
Definition: encoder.h:2223
RD_OPT rd
Definition: encoder.h:2935
int rate_index
Definition: encoder.h:1739
FrameProbInfo frame_probs
Definition: encoder.h:2740
bool enable_tx64
Definition: encoder.h:357
MB_MODE_INFO mbmi_arr[MAX_INTER_MODES]
Definition: encoder.h:1259
AVxWorker * p_workers[4]
Definition: encoder.h:1616
aom_superres_mode
Frame super-resolution mode.
Definition: aom_encoder.h:205
int64_t est_rd_arr[MAX_INTER_MODES]
Definition: encoder.h:1271
unsigned int max_inter_bitrate_pct
Definition: encoder.h:560
int64_t ts_frame_end
Definition: encoder.h:2443
int b_calculate_psnr
Definition: encoder.h:2605
size_t cx_data_sz
Definition: encoder.h:2423
struct AV1_COMP_DATA AV1_COMP_DATA
Structure to hold data corresponding to an encoded frame.
Frame level Two pass status and control data.
Definition: firstpass.h:464
enum aom_transfer_characteristics aom_transfer_characteristics_t
List of supported transfer functions.
The stucture of CYCLIC_REFRESH.
Definition: aq_cyclicrefresh.h:36
int64_t optimal_buffer_level_ms
Definition: encoder.h:534
Encoder parameters for synchronization of row based multi-threading.
Definition: encoder.h:1366
long aom_enc_frame_flags_t
Encoded Frame Flags.
Definition: aom_encoder.h:375
YV12_BUFFER_CONFIG orig_source
Definition: encoder.h:2862
Main encoder configuration data structure.
Definition: encoder.h:915
AV1TplRowMultiThreadInfo tpl_row_mt
Definition: encoder.h:1668
bool bwd_ref_frame
Definition: encoder.h:2065
bool use_error_resilient
Definition: encoder.h:2212
RefCntBuffer * last_show_frame_buf
Definition: encoder.h:2898
bool buffer_removal_time_present
Definition: encoder.h:2640
Encoder&#39;s parameters related to the current coding block.
Definition: block.h:849
RD_STATS rd_cost_y_arr[MAX_INTER_MODES]
Definition: encoder.h:1283
int height
Definition: av1_common_int.h:776
MBMIExtFrameBufferInfo mbmi_ext_info
Definition: encoder.h:2784
bool auto_key
Definition: encoder.h:494
struct EncodeFrameParams EncodeFrameParams
contains per-frame encoding parameters decided upon by av1_encode_strategy() and passed down to av1_e...
bool error_resilient_mode
Definition: av1_common_int.h:400
int arnr_strength
Definition: encoder.h:835
aom_tune_metric
Model tuning parameters.
Definition: aomcx.h:1573
COMPRESSOR_STAGE compressor_stage
Definition: encoder.h:3268
Flags signalled by the external interface at frame level.
Definition: encoder.h:2182
Primary Rate Control parameters and status.
Definition: ratectrl.h:270
struct AV1_PRIMARY AV1_PRIMARY
Top level primary encoder structure.
int default_interp_skip_flags
Definition: encoder.h:2015
int64_t first_ts_start
Definition: encoder.h:2330
uint16_t * cdef_srcbuf
Definition: encoder.h:1564
FILE * second_pass_log_stream
Definition: encoder.h:3431
tran_low_t * tcoeff
Definition: encoder.h:2341
int width
Definition: encoder.h:2003
unsigned char gf_frame_index
Definition: encoder.h:3012
RestorationType frame_restoration_type
Definition: restoration.h:260
bool row_mt_enabled
Definition: encoder.h:1653
Definition: aom_encoder.h:176
int next_mi_row
Definition: encoder.h:1403
int width
Definition: encoder.h:2077
bool enable_cfl_intra
Definition: encoder.h:319
int sframe_mode
Definition: encoder.h:489
pthread_mutex_t * mutex_
Definition: encoder.h:1524
int * mb_delta_q
Definition: encoder.h:3397
int ref_frame_flags
Definition: encoder.h:2971
int num_workers
Definition: encoder.h:1589
int do_update_vbr_bits_off_target_fast
Definition: encoder.h:3137
AV1LrSync lr_row_sync
Definition: encoder.h:1678
YV12_BUFFER_CONFIG * last_source
Definition: encoder.h:2829
unsigned int lib_flags
Definition: encoder.h:2433
The stucture of SVC.
Definition: svc_layercontext.h:88
Definition: aom_encoder.h:177
YV12_BUFFER_CONFIG * unfiltered_source
Definition: encoder.h:2856
YV12_BUFFER_CONFIG scaled_source
Definition: encoder.h:2840
Stores best extended mode information at frame level.
Definition: block.h:213
int prune_ref_frame_mask
Definition: encoder.h:3159
int show_existing_frame
Definition: av1_common_int.h:900
int64_t ts_frame_start
Definition: encoder.h:2438
int frame_header_count
Definition: encoder.h:3227
Definition: encoder.h:219