AOMedia AV1 Codec: av1/encoder/encoder.h Source File

| 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"
49#include "av1/encoder/speed_features.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
79extern "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
101typedef struct aom_rational64 {
102 int64_t num; // fraction numerator
103 int den; // fraction denominator
104} aom_rational64_t; // alias for struct aom_rational
105
106enum {
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
117enum {
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
129enum {
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
138static 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
153enum {
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);
160enum {
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
170enum {
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
178enum {
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
185enum {
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
193typedef 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 {
218 COST_UPD_SB,
219 COST_UPD_SBROW,
220 COST_UPD_TILE,
221 COST_UPD_OFF,
222 NUM_COST_UPDATE_TYPES,
223} COST_UPDATE_TYPE;
224
228 typedef enum {
229 LOOPFILTER_NONE = 0,
230 LOOPFILTER_ALL = 1,
231 LOOPFILTER_REFERENCE = 2,
232 LOOPFILTER_SELECTIVELY =
233 3,
234} LOOPFILTER_CONTROL;
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,
245} SKIP_APPLY_POSTPROC_FILTER;
246
250 typedef struct {
254 RESIZE_MODE resize_mode;
259 uint8_t resize_scale_denominator;
264 uint8_t resize_kf_scale_denominator;
265} ResizeCfg;
266
270 typedef struct {
274 bool enable_rect_partitions;
278 bool enable_ab_partitions;
282 bool enable_1to4_partitions;
287 BLOCK_SIZE min_partition_size;
292 BLOCK_SIZE max_partition_size;
293} PartitionCfg;
294
298 typedef struct {
302 bool enable_intra_edge_filter;
307 bool enable_filter_intra;
311 bool enable_smooth_intra;
315 bool enable_paeth_intra;
319 bool enable_cfl_intra;
323 bool enable_directional_intra;
328 bool enable_diagonal_intra;
333 bool enable_angle_delta;
347 bool auto_intra_tools_off;
348} IntraModeCfg;
349
353 typedef struct {
357 bool enable_tx64;
361 bool enable_flip_idtx;
365 bool enable_rect_tx;
370 bool reduced_tx_type_set;
375 bool use_intra_dct_only;
380 bool use_inter_dct_only;
385 bool use_intra_default_tx_only;
389 bool enable_tx_size_search;
390} TxfmSizeTypeCfg;
391
395 typedef struct {
399 bool enable_dist_wtd_comp;
404 bool enable_masked_comp;
408 bool enable_smooth_interintra;
412 bool enable_diff_wtd_comp;
416 bool enable_interinter_wedge;
420 bool enable_interintra_wedge;
421} CompoundTypeCfg;
422
426 typedef struct {
431 int superres_qthresh;
436 int superres_kf_qthresh;
442 uint8_t superres_scale_denominator;
448 uint8_t superres_kf_scale_denominator;
452 aom_superres_mode superres_mode;
456 bool enable_superres;
457} SuperResCfg;
458
462 typedef struct {
466 int key_freq_min;
467
471 int key_freq_max;
472
476 int enable_keyframe_filtering;
477
482 int sframe_dist;
483
489 int sframe_mode;
490
494 bool auto_key;
495
499 int fwd_kf_dist;
500
504 bool fwd_kf_enabled;
505
509 bool enable_sframe;
510
514 bool enable_intrabc;
515} KeyFrameCfg;
516
520 typedef struct {
522 // BUFFERING PARAMETERS
529 int64_t starting_buffer_level_ms;
534 int64_t optimal_buffer_level_ms;
539 int64_t maximum_buffer_size_ms;
540
544 int64_t target_bandwidth;
545
550 unsigned int vbr_corpus_complexity_lap;
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;
573 int drop_frames_water_mark;
579 int under_shoot_pct;
585 int over_shoot_pct;
590 int worst_allowed_q;
595 int best_allowed_q;
599 int cq_level;
604 enum aom_rc_mode mode;
611 int vbrbias;
616 int vbrmin_section;
621 int vbrmax_section;
622} RateControlCfg;
623
625typedef 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
643typedef 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
669typedef 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
688typedef 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
697typedef 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
712typedef 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
723typedef 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
732typedef 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
745typedef 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
752typedef 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
767typedef 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
780typedef 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 {
816 int sharpness;
817
825 int disable_trellis_quant;
826
830 int arnr_max_frames;
831
835 int arnr_strength;
836
843 uint8_t cdf_update_mode;
844
848 bool enable_tpl_model;
849
854 bool enable_overlay;
855
863 LOOPFILTER_CONTROL loopfilter_control;
864
869 bool skip_postproc_filtering;
870} AlgoCfg;
873typedef 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
927 AlgoCfg algo_cfg;
928
932 KeyFrameCfg kf_cfg;
933
937 RateControlCfg rc_cfg;
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
954 aom_fixed_buf_t twopass_stats_in;
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
1040 enum aom_enc_pass pass;
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;
1083} AV1EncoderConfig;
1084
1086static 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
1129typedef 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
1217typedef 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
1234typedef 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
1291typedef 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
1304typedef struct {
1305 VPartVar none;
1306 VPartVar horz[2];
1307 VPartVar vert[2];
1308} VPVariance;
1309
1310typedef struct {
1311 VPVariance part_variances;
1312 VPartVar split[4];
1313} VP4x4;
1314
1315typedef struct {
1316 VPVariance part_variances;
1317 VP4x4 split[4];
1318} VP8x8;
1319
1320typedef struct {
1321 VPVariance part_variances;
1322 VP8x8 split[4];
1323} VP16x16;
1324
1325typedef struct {
1326 VPVariance part_variances;
1327 VP16x16 split[4];
1328} VP32x32;
1329
1330typedef struct {
1331 VPVariance part_variances;
1332 VP32x32 split[4];
1333} VP64x64;
1334
1335typedef struct {
1336 VPVariance part_variances;
1337 VP64x64 *split;
1338} VP128x128;
1339
1345 typedef struct {
1354 int64_t thresholds[5];
1355
1360 int64_t threshold_minmax;
1361} VarBasedPartitionInfo;
1362
1366 typedef struct {
1367#if CONFIG_MULTITHREAD
1372 pthread_mutex_t *mutex_;
1373 pthread_cond_t *cond_;
1375#endif // CONFIG_MULTITHREAD
1381 int *num_finished_cols;
1388 int sync_range;
1395 int intrabc_extra_top_right_sb_delay;
1399 int rows;
1403 int next_mi_row;
1407 int num_threads_working;
1408} AV1EncRowMultiThreadSync;
1409
1412// TODO(jingning) All spatially adaptive variables should go to TileDataEnc.
1413typedef 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
1424typedef 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
1434typedef 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
1472struct EncWorkerData;
1473
1479 typedef struct {
1483 int allocated_tile_rows;
1487 int allocated_tile_cols;
1494 int allocated_rows;
1501 int allocated_cols;
1502
1506 int thread_id_to_tile_id[MAX_NUM_THREADS];
1507
1512 int *num_tile_cols_done;
1513
1518 int allocated_sb_rows;
1519
1520#if CONFIG_MULTITHREAD
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);
1544} AV1EncRowMultiThreadInfo;
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;
1580} RestoreStateBuffers;
1581
1585 typedef struct PrimaryMultiThreadInfo {
1589 int num_workers;
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
1616 AVxWorker *p_workers[MAX_PARALLEL_FRAMES];
1617
1621 int p_num_workers;
1622} PrimaryMultiThreadInfo;
1623
1627 typedef struct MultiThreadInfo {
1631 int num_workers;
1632
1636 int num_mod_workers[NUM_MT_MODULES];
1637
1641 AVxWorker *workers;
1642
1647 struct EncWorkerData *tile_thr_data;
1648
1653 bool row_mt_enabled;
1654
1658 bool pack_bs_mt_enabled;
1659
1663 AV1EncRowMultiThreadInfo enc_row_mt;
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
1708 RestoreStateBuffers restore_state_buf;
1709
1714 int pipeline_lpf_mt_with_enc;
1715} MultiThreadInfo;
1716
1719typedef struct ActiveMap {
1720 int enabled;
1721 int update;
1722 unsigned char *map;
1723} ActiveMap;
1724
1730 typedef struct {
1735 double cs_rate_array[32];
1739 int rate_index;
1743 int rate_size;
1744} ForceIntegerMVInfo;
1745
1748#if CONFIG_INTERNAL_STATS
1749// types of stats
1750enum {
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
1758typedef struct IMAGE_STAT {
1759 double stat[NUM_STAT_TYPES];
1760 double worst;
1761} ImageStat;
1762#endif // CONFIG_INTERNAL_STATS
1763
1764typedef struct {
1765 int ref_count;
1766 YV12_BUFFER_CONFIG buf;
1767} EncRefCntBuffer;
1768
1776 typedef struct {
1781 MB_MODE_INFO_EXT_FRAME *frame_base;
1785 int alloc_size;
1789 int stride;
1790} MBMIExtFrameBufferInfo;
1791
1794#if CONFIG_COLLECT_PARTITION_STATS
1795typedef 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.
1807enum {
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
1861static 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 {
1943 bool search_done;
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
1976 int segment_map_w;
1977 int segment_map_h;
1983 int num_src_corners;
1984
1990 int src_corners[2 * MAX_CORNERS];
1991} GlobalMotionInfo;
1992
2002 typedef struct {
2003 int width;
2004 int height;
2005} InitialDimensions;
2006
2010 typedef struct {
2015 int default_interp_skip_flags;
2019 uint16_t interp_filter_search_mask;
2020} InterpSearchFlags;
2021
2025 typedef struct {
2031 int max_mv_magnitude;
2036 int mv_step_param;
2045 fractional_mv_step_fp *find_fractional_mv_step;
2052 search_site_config search_site_cfg[SS_CFG_TOTAL][NUM_DISTINCT_SEARCH_METHODS];
2053} MotionVectorSearchParams;
2054
2063 typedef struct {
2064 bool golden_frame;
2065 bool bwd_ref_frame;
2066 bool alt_ref_frame;
2067} RefreshFrameInfo;
2068
2076 typedef struct {
2077 int width;
2078 int height;
2079} ResizePendingParams;
2080
2084 typedef struct {
2088 int ref_relative_dist[INTER_REFS_PER_FRAME];
2092 int8_t nearest_past_ref;
2096 int8_t nearest_future_ref;
2097} RefFrameDistanceInfo;
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];
2158} WinnerModeParams;
2159
2167 typedef struct {
2168 bool last_frame;
2169 bool golden_frame;
2170 bool bwd_ref_frame;
2171 bool alt2_ref_frame;
2172 bool alt_ref_frame;
2176 bool update_pending;
2177} ExtRefreshFrameFlagsInfo;
2178
2182 typedef struct {
2186 int ref_frame_flags;
2187
2191 ExtRefreshFrameFlagsInfo refresh_frame;
2192
2196 bool refresh_frame_context;
2197
2202 bool refresh_frame_context_pending;
2203
2207 bool use_ref_frame_mvs;
2208
2212 bool use_error_resilient;
2213
2217 bool use_s_frame;
2218
2223 bool use_primary_ref_none;
2224} ExternalFlags;
2225
2228typedef 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
2256typedef 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
2267typedef 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
2275typedef 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
2291typedef struct {
2292 int show_frame_count;
2293} FRAME_INDEX_SET;
2294
2300 typedef struct {
2306 uint8_t *map;
2312 bool has_lossless_segment;
2313} EncSegmentationInfo;
2314
2318 typedef struct {
2322 int64_t prev_ts_start;
2326 int64_t prev_ts_end;
2330 int64_t first_ts_start;
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
2354// DUCKY_ENCODE_FRAME_MODE is c version of EncodeFrameMode
2355enum {
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
2363enum {
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
2368typedef 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
2379typedef 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
2388typedef struct DuckyEncodeInfo {
2389 DuckyEncodeFrameInfo frame_info;
2390 DuckyEncodeFrameResult frame_result;
2391} DuckyEncodeInfo;
2393#endif
2394
2396typedef 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
2438 int64_t ts_frame_start;
2439
2443 int64_t ts_frame_end;
2444
2448 int flush;
2449
2453 const aom_rational64_t *timestamp_ratio;
2454
2458 int pop_lookahead;
2459
2463 int frame_display_order_hint;
2464} AV1_COMP_DATA;
2465
2469 typedef struct AV1_PRIMARY {
2473 struct AV1_COMP *parallel_cpi[MAX_PARALLEL_FRAMES];
2474
2479 struct AV1_COMP_DATA parallel_frames_data[MAX_PARALLEL_FRAMES - 1];
2480#if CONFIG_FPMT_TEST
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
2511 RefCntBuffer *ref_frame_map_copy[REF_FRAMES];
2512
2516 int64_t ts_start_last_show_frame;
2517
2521 int64_t ts_end_last_show_frame;
2522
2526 int num_fp_contexts;
2527
2531 int filter_level[2];
2532
2536 int filter_level_u;
2537
2541 int filter_level_v;
2542
2547 struct AV1_COMP *cpi;
2548
2552 struct AV1_COMP *cpi_lap;
2553
2557 struct lookahead_ctx *lookahead;
2558
2564 int seq_params_locked;
2565
2570 struct aom_codec_pkt_list *output_pkt_list;
2571
2575 int internal_altref_allowed;
2576
2580 int show_existing_alt_ref;
2581
2585 GF_GROUP gf_group;
2586
2590 GF_STATE gf_state;
2591
2595 int lap_enabled;
2596
2600 AV1LevelParams level_params;
2601
2605 int b_calculate_psnr;
2606
2610 int frames_left;
2611
2615 TWO_PASS twopass;
2616
2620 PRIMARY_RATE_CONTROL p_rc;
2621
2625 TEMPORAL_FILTER_INFO tf_info;
2630 SequenceHeader seq_params;
2631
2635 int use_svc;
2636
2640 bool buffer_removal_time_present;
2641
2645 unsigned int number_temporal_layers;
2646
2650 unsigned int number_spatial_layers;
2651
2655 struct aom_internal_error_info error;
2656
2662 aom_variance_fn_ptr_t fn_ptr[BLOCK_SIZES_ALL];
2663
2668 double *tpl_sb_rdmult_scaling_factors;
2669
2673 TplParams tpl_data;
2674
2678 MV_STATS mv_stats;
2679
2680#if CONFIG_INTERNAL_STATS
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
2721 FRAME_COUNTS aggregate_fc;
2722#endif // CONFIG_ENTROPY_STATS
2723
2730 int fb_of_context_type[REF_FRAMES];
2731
2735 PrimaryMultiThreadInfo p_mt_info;
2736
2740 FrameProbInfo frame_probs;
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 {
2763 AV1_PRIMARY *ppi;
2764
2769 EncQuantDequantParams enc_quant_dequant_params;
2770
2774 ThreadData td;
2775
2779 FRAME_COUNTS counts;
2780
2784 MBMIExtFrameBufferInfo mbmi_ext_info;
2785
2791 CB_COEFF_BUFFER *coeff_buffer_base;
2792
2797 CoeffBufferPool coeff_buffer_pool;
2798
2802 AV1_COMMON common;
2803
2807 AV1EncoderConfig oxcf;
2808
2813 TRELLIS_OPT_TYPE optimize_seg_arr[MAX_SEGMENTS];
2814
2820 YV12_BUFFER_CONFIG *source;
2821
2829 YV12_BUFFER_CONFIG *last_source;
2830
2835 YV12_BUFFER_CONFIG *unscaled_source;
2836
2840 YV12_BUFFER_CONFIG scaled_source;
2841
2845 YV12_BUFFER_CONFIG *unscaled_last_source;
2846
2850 YV12_BUFFER_CONFIG scaled_last_source;
2851
2856 YV12_BUFFER_CONFIG *unfiltered_source;
2857
2862 YV12_BUFFER_CONFIG orig_source;
2863
2867 int skip_tpl_setup_stats;
2868
2877 double *tpl_rdmult_scaling_factors;
2878
2882 TemporalFilterCtx tf_ctx;
2883
2887 ForceIntegerMVInfo force_intpel_info;
2888
2893 RefCntBuffer *scaled_ref_buf[INTER_REFS_PER_FRAME];
2894
2898 RefCntBuffer *last_show_frame_buf;
2899
2903 RefreshFrameInfo refresh_frame;
2904
2908 int rt_reduce_num_ref_buffers;
2909
2913 ExternalFlags ext_flags;
2914
2919 YV12_BUFFER_CONFIG last_frame_uf;
2920
2925 YV12_BUFFER_CONFIG trial_frame_rst;
2926
2930 int64_t ambient_err;
2931
2935 RD_OPT rd;
2936
2941 CODING_CONTEXT coding_context;
2942
2946 GlobalMotionInfo gm_info;
2947
2951 WinnerModeParams winner_mode_params;
2952
2956 TimeStamps time_stamps;
2957
2961 RATE_CONTROL rc;
2962
2966 double framerate;
2967
2971 int ref_frame_flags;
2972
2976 int speed;
2977
2981 SPEED_FEATURES sf;
2982
2986 MotionVectorSearchParams mv_search_params;
2987
2992 int all_one_sided_refs;
2993
2997 EncSegmentationInfo enc_seg;
2998
3002 CYCLIC_REFRESH *cyclic_refresh;
3007 ActiveMap active_map;
3008
3012 unsigned char gf_frame_index;
3013
3014#if CONFIG_INTERNAL_STATS
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
3028 unsigned int tx_search_count;
3029#endif // CONFIG_SPEED_STATS
3030
3035 int droppable;
3036
3040 FRAME_INFO frame_info;
3041
3045 FRAME_INDEX_SET frame_index_set;
3046
3050 InitialDimensions initial_dimensions;
3051
3058 int initial_mbs;
3059
3063 ResizePendingParams resize_pending_params;
3064
3069 TileDataEnc *tile_data;
3073 int allocated_tiles;
3074
3078 TokenInfo token_info;
3079
3083 int vaq_refresh;
3084
3088 VarBasedPartitionInfo vbp_info;
3089
3093 int num_frame_recode;
3094
3098 FrameProbInfo frame_new_probs[NUM_RECODES_PER_FRAME];
3099
3103 int do_update_frame_probs_txtype[NUM_RECODES_PER_FRAME];
3104
3108 int do_update_frame_probs_obmc[NUM_RECODES_PER_FRAME];
3109
3113 int do_update_frame_probs_warp[NUM_RECODES_PER_FRAME];
3114
3118 int do_update_frame_probs_interpfilter[NUM_RECODES_PER_FRAME];
3119
3120#if CONFIG_FPMT_TEST
3125 double temp_framerate;
3126#endif
3132 double new_framerate;
3133
3137 int do_update_vbr_bits_off_target_fast;
3138
3142 MultiThreadInfo mt_info;
3143
3149 int existing_fb_idx_to_show;
3150
3154 int intrabc_used;
3155
3159 int prune_ref_frame_mask;
3160
3164 AV1LrStruct lr_ctxt;
3165
3169 aom_film_grain_table_t *film_grain_table;
3170
3171#if CONFIG_DENOISE
3176 struct aom_denoise_and_model_t *denoise_and_model;
3177#endif
3178
3182 InterpSearchFlags interp_search_flags;
3183
3191 int use_screen_content_tools;
3192
3199 int is_screen_content_type;
3200
3201#if CONFIG_COLLECT_PARTITION_STATS
3205 FramePartitionTimingStats partition_stats;
3206#endif // CONFIG_COLLECT_PARTITION_STATS
3207
3208#if CONFIG_COLLECT_COMPONENT_TIMING
3212 uint64_t component_time[kTimingComponents];
3217 struct aom_usec_timer component_timer[kTimingComponents];
3221 uint64_t frame_component_time[kTimingComponents];
3222#endif
3223
3227 int frame_header_count;
3228
3232 int deltaq_used;
3233
3237 RefFrameDistanceInfo ref_frame_dist_info;
3238
3244 double *ssim_rdmult_scaling_factors;
3245
3246#if CONFIG_TUNE_VMAF
3250 TuneVMAFInfo vmaf_info;
3251#endif
3252
3253#if CONFIG_TUNE_BUTTERAUGLI
3257 TuneButteraugliInfo butteraugli_info;
3258#endif
3259
3263 SVC svc;
3264
3268 COMPRESSOR_STAGE compressor_stage;
3269
3274 FRAME_TYPE last_frame_type;
3275
3279 int num_tg;
3280
3286 aom_superres_mode superres_mode;
3287
3291 FirstPassData firstpass_data;
3292
3296 NOISE_ESTIMATE noise_estimate;
3297
3298#if CONFIG_AV1_TEMPORAL_DENOISING
3302 AV1_DENOISER denoiser;
3303#endif
3304
3309 uint8_t *consec_zero_mv;
3310
3314 BLOCK_SIZE fp_block_size;
3315
3320 int sb_counter;
3321
3325 size_t available_bs_size;
3326
3331 ExtPartController ext_part_controller;
3332
3337 MV_STATS mv_stats;
3341 int ref_refresh_index;
3342
3347 bool refresh_idx_available;
3348
3354 int ref_idx_to_skip;
3355#if CONFIG_FPMT_TEST
3362 int wanted_fb;
3363#endif // CONFIG_FPMT_TEST
3364
3370 bool do_frame_data_update;
3371
3372#if CONFIG_RD_COMMAND
3376 RD_COMMAND rd_command;
3377#endif // CONFIG_RD_COMMAND
3378
3382 WeberStats *mb_weber_stats;
3383
3387 BLOCK_SIZE weber_bsize;
3388
3392 int64_t norm_wiener_variance;
3393
3397 int *mb_delta_q;
3398
3402 bool is_dropped_frame;
3403
3404#if CONFIG_BITRATE_ACCURACY
3408 VBR_RATECTRL_INFO vbr_rc_info;
3409#endif
3410
3411#if CONFIG_RATECTRL_LOG
3415 RATECTRL_LOG rc_log;
3416#endif // CONFIG_RATECTRL_LOG
3417
3421 TWO_PASS_FRAME twopass_frame;
3422
3426 THIRD_PASS_DEC_CTX *third_pass_ctx;
3427
3431 FILE *second_pass_log_stream;
3432
3436 uint64_t *src_sad_blk_64x64;
3437
3441 uint64_t rec_sse;
3442
3447 int use_ducky_encode;
3448
3449#if !CONFIG_REALTIME_ONLY
3453 DuckyEncodeInfo ducky_encode_info;
3454#endif // CONFIG_REALTIME_ONLY
3455 //
3459 int frames_since_last_update;
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;
3476} EncodeFrameInput;
3477
3482 typedef struct EncodeFrameParams {
3486 int error_resilient_mode;
3490 FRAME_TYPE frame_type;
3491
3493 int primary_ref_frame;
3494 int order_offset;
3495
3500 int show_frame;
3501
3503 int refresh_frame_flags;
3504
3505 int show_existing_frame;
3506 int existing_fb_idx_to_show;
3507
3512 int ref_frame_flags;
3513
3517 int remapped_ref_idx[REF_FRAMES];
3518
3523 RefreshFrameInfo refresh_frame;
3524
3528 int speed;
3529} EncodeFrameParams;
3530
3533// EncodeFrameResults contains information about the result of encoding a
3534// single frame
3535typedef struct {
3536 size_t size; // Size of resulting bitstream
3537} EncodeFrameResults;
3538
3539void av1_initialize_enc(unsigned int usage, enum aom_rc_mode end_usage);
3540
3541struct 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
3547struct AV1_PRIMARY *av1_create_primary_compressor(
3548 struct aom_codec_pkt_list *pkt_list_head, int num_lap_buffers,
3549 const AV1EncoderConfig *oxcf);
3550
3551void av1_remove_compressor(AV1_COMP *cpi);
3552
3553void av1_remove_primary_compressor(AV1_PRIMARY *ppi);
3554
3555#if CONFIG_ENTROPY_STATS
3556void print_entropy_stats(AV1_PRIMARY *const ppi);
3557#endif
3558#if CONFIG_INTERNAL_STATS
3559void print_internal_stats(AV1_PRIMARY *ppi);
3560#endif
3561
3562void av1_change_config_seq(AV1_PRIMARY *ppi, const AV1EncoderConfig *oxcf,
3563 bool *sb_size_changed);
3564
3565void av1_change_config(AV1_COMP *cpi, const AV1EncoderConfig *oxcf,
3566 bool sb_size_changed);
3567
3568void av1_check_initial_width(AV1_COMP *cpi, int use_highbitdepth,
3569 int subsampling_x, int subsampling_y);
3570
3571void av1_init_seq_coding_tools(AV1_PRIMARY *const ppi,
3572 const AV1EncoderConfig *oxcf, int use_svc);
3573
3574void av1_post_encode_updates(AV1_COMP *const cpi,
3575 const AV1_COMP_DATA *const cpi_data);
3576
3577void av1_scale_references_fpmt(AV1_COMP *cpi, int *ref_buffers_used_map);
3578
3579void av1_increment_scaled_ref_counts_fpmt(BufferPool *buffer_pool,
3580 int ref_buffers_used_map);
3581
3582void av1_release_scaled_references_fpmt(AV1_COMP *cpi);
3583
3584void av1_decrement_ref_counts_fpmt(BufferPool *buffer_pool,
3585 int ref_buffers_used_map);
3586
3587void 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
3592int av1_init_parallel_frame_context(const AV1_COMP_DATA *const first_cpi_data,
3593 AV1_PRIMARY *const ppi,
3594 int *ref_buffers_used_map);
3613int av1_receive_raw_frame(AV1_COMP *cpi, aom_enc_frame_flags_t frame_flags,
3614 YV12_BUFFER_CONFIG *sd, int64_t time_stamp,
3615 int64_t end_time_stamp);
3616
3636int av1_get_compressed_data(AV1_COMP *cpi, AV1_COMP_DATA *const cpi_data);
3637
3644int 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
3650int av1_get_preview_raw_frame(AV1_COMP *cpi, YV12_BUFFER_CONFIG *dest);
3651
3652int 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
3658int av1_use_as_reference(int *ext_ref_frame_flags, int ref_frame_flags);
3659
3660int av1_copy_reference_enc(AV1_COMP *cpi, int idx, YV12_BUFFER_CONFIG *sd);
3661
3662int av1_set_reference_enc(AV1_COMP *cpi, int idx, YV12_BUFFER_CONFIG *sd);
3663
3664int av1_set_size_literal(AV1_COMP *cpi, int width, int height);
3665
3666void av1_set_frame_size(AV1_COMP *cpi, int width, int height);
3667
3668void av1_set_mv_search_params(AV1_COMP *cpi);
3669
3670int av1_set_active_map(AV1_COMP *cpi, unsigned char *map, int rows, int cols);
3671
3672int av1_get_active_map(AV1_COMP *cpi, unsigned char *map, int rows, int cols);
3673
3674int 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
3679int av1_get_quantizer(struct AV1_COMP *cpi);
3680
3681int 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().
3694void av1_set_screen_content_options(struct AV1_COMP *cpi,
3695 FeatureFlags *features);
3696
3697void av1_update_frame_size(AV1_COMP *cpi);
3698
3699typedef struct {
3700 int pyr_level;
3701 int disp_order;
3702} RefFrameMapPair;
3703
3704static 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
3739static 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
3767static INLINE int64_t
3768timebase_units_to_ticks(const aom_rational64_t *timestamp_ratio, int64_t n) {
3769 return n * timestamp_ratio->num / timestamp_ratio->den;
3770}
3771
3772static INLINE int64_t
3773ticks_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
3779static 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.
3789static INLINE int av1_use_hash_me(const AV1_COMP *const cpi) {
3790 return (cpi->common.features.allow_screen_content_tools &&
3791 cpi->common.features.allow_intrabc &&
3792 frame_is_intra_only(&cpi->common));
3793}
3794
3795static 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
3801static 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.
3810static 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
3820static 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
3836void av1_apply_encoding_flags(AV1_COMP *cpi, aom_enc_frame_flags_t flags);
3837
3838#define ALT_MIN_LAG 3
3839static 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
3843static 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.
3849static 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
3854static 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
3861static 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
3866static 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.
3873static 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
3895static 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.
3901static 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
3908static 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
3915static 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
3924static INLINE int get_chessboard_index(int frame_index) {
3925 return frame_index & 0x1;
3926}
3927
3928static 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
3935static 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.
3942double av1_get_compression_ratio(const AV1_COMMON *const cm,
3943 size_t encoded_frame_size);
3944
3945void av1_new_framerate(AV1_COMP *cpi, double framerate);
3946
3947void 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.
3952static INLINE int av1_resize_scaled(const AV1_COMMON *cm) {
3953 return cm->superres_upscaled_width != cm->render_width ||
3954 cm->superres_upscaled_height != cm->render_height;
3955}
3956
3957static 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.
3964static 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'.
3971static 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.
3982static 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
3995static 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
4012static 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.
4023static const MV_REFERENCE_FRAME disable_order[] = {
4024 LAST3_FRAME,
4025 LAST2_FRAME,
4026 ALTREF2_FRAME,
4027 BWDREF_FRAME,
4028};
4029
4030static 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
4036static 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
4083static 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
4090static 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.
4099static 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
4104static INLINE int av1_pixels_to_mi(int pixels) {
4105 return ALIGN_POWER_OF_TWO(pixels, 3) >> MI_SIZE_LOG2;
4106}
4107
4108static 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
4115static 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.
4124static 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.
4129static 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.
4135static 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.
4145static 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
4156 if (!cpi->oxcf.algo_cfg.skip_postproc_filtering || cpi->ppi->b_calculate_psnr)
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
4183static 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
4199static 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
4205static 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
4213static 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
4219static 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
4227static 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
4266static 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
4283static INLINE void start_timing(AV1_COMP *cpi, int component) {
4284 aom_usec_timer_start(&cpi->component_timer[component]);
4285}
4286static 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}
4291static 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_
aom_chroma_sample_position_t
enum aom_chroma_sample_position aom_chroma_sample_position_t
List of chroma sample positions.
aom_transfer_characteristics_t
enum aom_transfer_characteristics aom_transfer_characteristics_t
List of supported transfer functions.
aom_color_range_t
enum aom_color_range aom_color_range_t
List of supported color range.
aom_color_primaries_t
enum aom_color_primaries aom_color_primaries_t
List of supported color primaries.
aom_matrix_coefficients_t
enum aom_matrix_coefficients aom_matrix_coefficients_t
List of supported matrix coefficients.
aomcx.h
Provides definitions for using AOM or AV1 encoder algorithm within the aom Codec Interface.
block.h
COST_UPDATE_TYPE
COST_UPDATE_TYPE
This enum controls how often the entropy costs should be updated.
Definition: encoder.h:217
COST_UPD_SBROW
@ COST_UPD_SBROW
Definition: encoder.h:219
COST_UPD_TILE
@ COST_UPD_TILE
Definition: encoder.h:220
COST_UPD_OFF
@ COST_UPD_OFF
Definition: encoder.h:221
NUM_COST_UPDATE_TYPES
@ NUM_COST_UPDATE_TYPES
Definition: encoder.h:222
COST_UPD_SB
@ COST_UPD_SB
Definition: encoder.h:218
LOOPFILTER_CONTROL
LOOPFILTER_CONTROL
This enum controls to which frames loopfilter is applied.
Definition: encoder.h:228
LOOPFILTER_ALL
@ LOOPFILTER_ALL
Definition: encoder.h:230
LOOPFILTER_SELECTIVELY
@ LOOPFILTER_SELECTIVELY
Definition: encoder.h:232
LOOPFILTER_REFERENCE
@ LOOPFILTER_REFERENCE
Definition: encoder.h:231
LOOPFILTER_NONE
@ LOOPFILTER_NONE
Definition: encoder.h:229
NUM_RECODES_PER_FRAME
#define NUM_RECODES_PER_FRAME
Max number of recodes used to track the frame probabilities.
Definition: encoder.h:1549
InterModesInfo
struct inter_modes_info InterModesInfo
Struct used to hold inter mode data for fast tx search.
SKIP_APPLY_POSTPROC_FILTER
SKIP_APPLY_POSTPROC_FILTER
This enum controls the application of post-processing filters on a reconstructed frame.
Definition: encoder.h:240
MAX_PARALLEL_FRAMES
#define MAX_PARALLEL_FRAMES
Max number of frames that can be encoded in a parallel encode set.
Definition: encoder.h:1554
enums.h
RESTORE_NONE
@ RESTORE_NONE
Definition: enums.h:604
RESTORE_SWITCHABLE_TYPES
@ RESTORE_SWITCHABLE_TYPES
Definition: enums.h:608
aom_dist_metric
aom_dist_metric
Distortion metric to use for RD optimization.
Definition: aomcx.h:1592
aom_tune_content
aom_tune_content
Definition: aomcx.h:1554
AOM_SCALING_MODE
enum aom_scaling_mode_1d AOM_SCALING_MODE
aom 1-D scaling mode
aom_tune_metric
aom_tune_metric
Model tuning parameters.
Definition: aomcx.h:1573
aom_bit_depth_t
enum aom_bit_depth aom_bit_depth_t
Bit depth for codecThis enumeration determines the bit depth of the codec.
aom_superblock_size_t
enum aom_superblock_size aom_superblock_size_t
Superblock size selection.
aom_codec_err_t
aom_codec_err_t
Algorithm return codes.
Definition: aom_codec.h:155
aom_superres_mode
aom_superres_mode
Frame super-resolution mode.
Definition: aom_encoder.h:205
aom_rc_mode
aom_rc_mode
Rate control mode.
Definition: aom_encoder.h:183
aom_enc_pass
aom_enc_pass
Multi-pass Encoding Pass.
Definition: aom_encoder.h:174
aom_enc_frame_flags_t
long aom_enc_frame_flags_t
Encoded Frame Flags.
Definition: aom_encoder.h:375
AOM_RC_ONE_PASS
@ AOM_RC_ONE_PASS
Definition: aom_encoder.h:175
AOM_RC_SECOND_PASS
@ AOM_RC_SECOND_PASS
Definition: aom_encoder.h:177
AOM_RC_FIRST_PASS
@ AOM_RC_FIRST_PASS
Definition: aom_encoder.h:176
av1_get_compressed_data
int av1_get_compressed_data(AV1_COMP *cpi, AV1_COMP_DATA *const cpi_data)
Encode a frame.
Definition: encoder.c:4588
av1_receive_raw_frame
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
av1_encode
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
has_no_stats_stage
static int has_no_stats_stage(const AV1_COMP *const cpi)
Check if the current stage has statistics.
Definition: encoder.h:3887
lookahead.h
Describes look ahead buffer operations.
speed_features.h
CDEF_PICK_FROM_Q
@ CDEF_PICK_FROM_Q
Definition: speed_features.h:164
AV1Common
Top level common structure used by both encoder and decoder.
Definition: av1_common_int.h:750
AV1Common::rst_info
RestorationInfo rst_info[3]
Definition: av1_common_int.h:946
AV1Common::superres_upscaled_width
int superres_upscaled_width
Definition: av1_common_int.h:799
AV1Common::superres_upscaled_height
int superres_upscaled_height
Definition: av1_common_int.h:800
AV1Common::seq_params
SequenceHeader * seq_params
Definition: av1_common_int.h:976
AV1Common::width
int width
Definition: av1_common_int.h:775
AV1Common::cdef_info
CdefInfo cdef_info
Definition: av1_common_int.h:955
AV1Common::current_frame
CurrentFrame current_frame
Definition: av1_common_int.h:754
AV1Common::show_existing_frame
int show_existing_frame
Definition: av1_common_int.h:900
AV1Common::lf
struct loopfilter lf
Definition: av1_common_int.h:939
AV1Common::features
FeatureFlags features
Definition: av1_common_int.h:905
AV1Common::show_frame
int show_frame
Definition: av1_common_int.h:885
AV1Common::ref_frame_map
RefCntBuffer * ref_frame_map[REF_FRAMES]
Definition: av1_common_int.h:878
AV1Common::tiles
CommonTileParams tiles
Definition: av1_common_int.h:992
AV1Common::height
int height
Definition: av1_common_int.h:776
AV1Common::render_width
int render_width
Definition: av1_common_int.h:786
AV1Common::render_height
int render_height
Definition: av1_common_int.h:787
AV1EncRowMultiThreadInfo
Encoder data related to row-based multi-threading.
Definition: encoder.h:1479
AV1EncRowMultiThreadInfo::allocated_sb_rows
int allocated_sb_rows
Definition: encoder.h:1518
AV1EncRowMultiThreadInfo::mutex_
pthread_mutex_t * mutex_
Definition: encoder.h:1524
AV1EncRowMultiThreadInfo::allocated_tile_cols
int allocated_tile_cols
Definition: encoder.h:1487
AV1EncRowMultiThreadInfo::allocated_cols
int allocated_cols
Definition: encoder.h:1501
AV1EncRowMultiThreadInfo::cond_
pthread_cond_t * cond_
Definition: encoder.h:1528
AV1EncRowMultiThreadInfo::allocated_tile_rows
int allocated_tile_rows
Definition: encoder.h:1483
AV1EncRowMultiThreadInfo::allocated_rows
int allocated_rows
Definition: encoder.h:1494
AV1EncRowMultiThreadInfo::num_tile_cols_done
int * num_tile_cols_done
Definition: encoder.h:1512
AV1EncRowMultiThreadSync
Encoder parameters for synchronization of row based multi-threading.
Definition: encoder.h:1366
AV1EncRowMultiThreadSync::rows
int rows
Definition: encoder.h:1399
AV1EncRowMultiThreadSync::sync_range
int sync_range
Definition: encoder.h:1388
AV1EncRowMultiThreadSync::intrabc_extra_top_right_sb_delay
int intrabc_extra_top_right_sb_delay
Definition: encoder.h:1395
AV1EncRowMultiThreadSync::next_mi_row
int next_mi_row
Definition: encoder.h:1403
AV1EncRowMultiThreadSync::mutex_
pthread_mutex_t * mutex_
Definition: encoder.h:1372
AV1EncRowMultiThreadSync::cond_
pthread_cond_t * cond_
Definition: encoder.h:1373
AV1EncRowMultiThreadSync::num_threads_working
int num_threads_working
Definition: encoder.h:1407
AV1EncoderConfig
Main encoder configuration data structure.
Definition: encoder.h:915
AV1EncoderConfig::rc_cfg
RateControlCfg rc_cfg
Definition: encoder.h:937
AV1EncoderConfig::kf_cfg
KeyFrameCfg kf_cfg
Definition: encoder.h:932
AV1EncoderConfig::pass
enum aom_enc_pass pass
Definition: encoder.h:1040
AV1EncoderConfig::algo_cfg
AlgoCfg algo_cfg
Definition: encoder.h:927
AV1EncoderConfig::twopass_stats_in
aom_fixed_buf_t twopass_stats_in
Definition: encoder.h:954
AV1_COMP_DATA
Structure to hold data corresponding to an encoded frame.
Definition: encoder.h:2414
AV1_COMP_DATA::ts_frame_end
int64_t ts_frame_end
Definition: encoder.h:2443
AV1_COMP_DATA::pop_lookahead
int pop_lookahead
Definition: encoder.h:2458
AV1_COMP_DATA::ts_frame_start
int64_t ts_frame_start
Definition: encoder.h:2438
AV1_COMP_DATA::cx_data
unsigned char * cx_data
Definition: encoder.h:2418
AV1_COMP_DATA::cx_data_sz
size_t cx_data_sz
Definition: encoder.h:2423
AV1_COMP_DATA::flush
int flush
Definition: encoder.h:2448
AV1_COMP_DATA::frame_display_order_hint
int frame_display_order_hint
Definition: encoder.h:2463
AV1_COMP_DATA::lib_flags
unsigned int lib_flags
Definition: encoder.h:2433
AV1_COMP_DATA::frame_size
size_t frame_size
Definition: encoder.h:2428
AV1_COMP_DATA::timestamp_ratio
const aom_rational64_t * timestamp_ratio
Definition: encoder.h:2453
AV1_COMP
Top level encoder structure.
Definition: encoder.h:2759
AV1_COMP::do_update_frame_probs_warp
int do_update_frame_probs_warp[10]
Definition: encoder.h:3113
AV1_COMP::consec_zero_mv
uint8_t * consec_zero_mv
Definition: encoder.h:3309
AV1_COMP::do_update_frame_probs_obmc
int do_update_frame_probs_obmc[10]
Definition: encoder.h:3108
AV1_COMP::denoise_and_model
struct aom_denoise_and_model_t * denoise_and_model
Definition: encoder.h:3176
AV1_COMP::skip_tpl_setup_stats
int skip_tpl_setup_stats
Definition: encoder.h:2867
AV1_COMP::frames_since_last_update
int frames_since_last_update
Definition: encoder.h:3459
AV1_COMP::mb_delta_q
int * mb_delta_q
Definition: encoder.h:3397
AV1_COMP::vaq_refresh
int vaq_refresh
Definition: encoder.h:3083
AV1_COMP::last_frame_type
FRAME_TYPE last_frame_type
Definition: encoder.h:3274
AV1_COMP::unscaled_source
YV12_BUFFER_CONFIG * unscaled_source
Definition: encoder.h:2835
AV1_COMP::cyclic_refresh
CYCLIC_REFRESH * cyclic_refresh
Definition: encoder.h:3002
AV1_COMP::initial_dimensions
InitialDimensions initial_dimensions
Definition: encoder.h:3050
AV1_COMP::rc
RATE_CONTROL rc
Definition: encoder.h:2961
AV1_COMP::deltaq_used
int deltaq_used
Definition: encoder.h:3232
AV1_COMP::active_map
ActiveMap active_map
Definition: encoder.h:3007
AV1_COMP::mb_weber_stats
WeberStats * mb_weber_stats
Definition: encoder.h:3382
AV1_COMP::refresh_idx_available
bool refresh_idx_available
Definition: encoder.h:3347
AV1_COMP::token_info
TokenInfo token_info
Definition: encoder.h:3078
AV1_COMP::ambient_err
int64_t ambient_err
Definition: encoder.h:2930
AV1_COMP::film_grain_table
aom_film_grain_table_t * film_grain_table
Definition: encoder.h:3169
AV1_COMP::ref_refresh_index
int ref_refresh_index
Definition: encoder.h:3341
AV1_COMP::available_bs_size
size_t available_bs_size
Definition: encoder.h:3325
AV1_COMP::sf
SPEED_FEATURES sf
Definition: encoder.h:2981
AV1_COMP::optimize_seg_arr
TRELLIS_OPT_TYPE optimize_seg_arr[8]
Definition: encoder.h:2813
AV1_COMP::ext_part_controller
ExtPartController ext_part_controller
Definition: encoder.h:3331
AV1_COMP::second_pass_log_stream
FILE * second_pass_log_stream
Definition: encoder.h:3431
AV1_COMP::ssim_rdmult_scaling_factors
double * ssim_rdmult_scaling_factors
Definition: encoder.h:3244
AV1_COMP::rd
RD_OPT rd
Definition: encoder.h:2935
AV1_COMP::num_tg
int num_tg
Definition: encoder.h:3279
AV1_COMP::winner_mode_params
WinnerModeParams winner_mode_params
Definition: encoder.h:2951
AV1_COMP::ext_flags
ExternalFlags ext_flags
Definition: encoder.h:2913
AV1_COMP::enc_seg
EncSegmentationInfo enc_seg
Definition: encoder.h:2997
AV1_COMP::mv_search_params
MotionVectorSearchParams mv_search_params
Definition: encoder.h:2986
AV1_COMP::use_screen_content_tools
int use_screen_content_tools
Definition: encoder.h:3191
AV1_COMP::do_update_frame_probs_interpfilter
int do_update_frame_probs_interpfilter[10]
Definition: encoder.h:3118
AV1_COMP::coding_context
CODING_CONTEXT coding_context
Definition: encoder.h:2941
AV1_COMP::tf_ctx
TemporalFilterCtx tf_ctx
Definition: encoder.h:2882
AV1_COMP::force_intpel_info
ForceIntegerMVInfo force_intpel_info
Definition: encoder.h:2887
AV1_COMP::gm_info
GlobalMotionInfo gm_info
Definition: encoder.h:2946
AV1_COMP::coeff_buffer_pool
CoeffBufferPool coeff_buffer_pool
Definition: encoder.h:2797
AV1_COMP::frame_index_set
FRAME_INDEX_SET frame_index_set
Definition: encoder.h:3045
AV1_COMP::ref_frame_flags
int ref_frame_flags
Definition: encoder.h:2971
AV1_COMP::scaled_ref_buf
RefCntBuffer * scaled_ref_buf[INTER_REFS_PER_FRAME]
Definition: encoder.h:2893
AV1_COMP::gf_frame_index
unsigned char gf_frame_index
Definition: encoder.h:3012
AV1_COMP::oxcf
AV1EncoderConfig oxcf
Definition: encoder.h:2807
AV1_COMP::common
AV1_COMMON common
Definition: encoder.h:2802
AV1_COMP::lr_ctxt
AV1LrStruct lr_ctxt
Definition: encoder.h:3164
AV1_COMP::do_frame_data_update
bool do_frame_data_update
Definition: encoder.h:3370
AV1_COMP::do_update_frame_probs_txtype
int do_update_frame_probs_txtype[10]
Definition: encoder.h:3103
AV1_COMP::counts
FRAME_COUNTS counts
Definition: encoder.h:2779
AV1_COMP::compressor_stage
COMPRESSOR_STAGE compressor_stage
Definition: encoder.h:3268
AV1_COMP::intrabc_used
int intrabc_used
Definition: encoder.h:3154
AV1_COMP::num_frame_recode
int num_frame_recode
Definition: encoder.h:3093
AV1_COMP::rt_reduce_num_ref_buffers
int rt_reduce_num_ref_buffers
Definition: encoder.h:2908
AV1_COMP::refresh_frame
RefreshFrameInfo refresh_frame
Definition: encoder.h:2903
AV1_COMP::prune_ref_frame_mask
int prune_ref_frame_mask
Definition: encoder.h:3159
AV1_COMP::unscaled_last_source
YV12_BUFFER_CONFIG * unscaled_last_source
Definition: encoder.h:2845
AV1_COMP::third_pass_ctx
THIRD_PASS_DEC_CTX * third_pass_ctx
Definition: encoder.h:3426
AV1_COMP::all_one_sided_refs
int all_one_sided_refs
Definition: encoder.h:2992
AV1_COMP::mt_info
MultiThreadInfo mt_info
Definition: encoder.h:3142
AV1_COMP::vbp_info
VarBasedPartitionInfo vbp_info
Definition: encoder.h:3088
AV1_COMP::last_source
YV12_BUFFER_CONFIG * last_source
Definition: encoder.h:2829
AV1_COMP::existing_fb_idx_to_show
int existing_fb_idx_to_show
Definition: encoder.h:3149
AV1_COMP::unfiltered_source
YV12_BUFFER_CONFIG * unfiltered_source
Definition: encoder.h:2856
AV1_COMP::zeromv_skip_thresh_exit_part
unsigned int zeromv_skip_thresh_exit_part[BLOCK_SIZES_ALL]
Definition: encoder.h:3464
AV1_COMP::frame_info
FRAME_INFO frame_info
Definition: encoder.h:3040
AV1_COMP::frame_header_count
int frame_header_count
Definition: encoder.h:3227
AV1_COMP::droppable
int droppable
Definition: encoder.h:3035
AV1_COMP::last_show_frame_buf
RefCntBuffer * last_show_frame_buf
Definition: encoder.h:2898
AV1_COMP::superres_mode
aom_superres_mode superres_mode
Definition: encoder.h:3286
AV1_COMP::mbmi_ext_info
MBMIExtFrameBufferInfo mbmi_ext_info
Definition: encoder.h:2784
AV1_COMP::new_framerate
double new_framerate
Definition: encoder.h:3132
AV1_COMP::ppi
AV1_PRIMARY * ppi
Definition: encoder.h:2763
AV1_COMP::src_sad_blk_64x64
uint64_t * src_sad_blk_64x64
Definition: encoder.h:3436
AV1_COMP::norm_wiener_variance
int64_t norm_wiener_variance
Definition: encoder.h:3392
AV1_COMP::tpl_rdmult_scaling_factors
double * tpl_rdmult_scaling_factors
Definition: encoder.h:2877
AV1_COMP::sb_counter
int sb_counter
Definition: encoder.h:3320
AV1_COMP::tile_data
TileDataEnc * tile_data
Definition: encoder.h:3069
AV1_COMP::is_screen_content_type
int is_screen_content_type
Definition: encoder.h:3199
AV1_COMP::weber_bsize
BLOCK_SIZE weber_bsize
Definition: encoder.h:3387
AV1_COMP::interp_search_flags
InterpSearchFlags interp_search_flags
Definition: encoder.h:3182
AV1_COMP::time_stamps
TimeStamps time_stamps
Definition: encoder.h:2956
AV1_COMP::ref_idx_to_skip
int ref_idx_to_skip
Definition: encoder.h:3354
AV1_COMP::orig_source
YV12_BUFFER_CONFIG orig_source
Definition: encoder.h:2862
AV1_COMP::firstpass_data
FirstPassData firstpass_data
Definition: encoder.h:3291
AV1_COMP::framerate
double framerate
Definition: encoder.h:2966
AV1_COMP::speed
int speed
Definition: encoder.h:2976
AV1_COMP::fp_block_size
BLOCK_SIZE fp_block_size
Definition: encoder.h:3314
AV1_COMP::use_ducky_encode
int use_ducky_encode
Definition: encoder.h:3447
AV1_COMP::trial_frame_rst
YV12_BUFFER_CONFIG trial_frame_rst
Definition: encoder.h:2925
AV1_COMP::is_dropped_frame
bool is_dropped_frame
Definition: encoder.h:3402
AV1_COMP::td
ThreadData td
Definition: encoder.h:2774
AV1_COMP::resize_pending_params
ResizePendingParams resize_pending_params
Definition: encoder.h:3063
AV1_COMP::scaled_source
YV12_BUFFER_CONFIG scaled_source
Definition: encoder.h:2840
AV1_COMP::do_update_vbr_bits_off_target_fast
int do_update_vbr_bits_off_target_fast
Definition: encoder.h:3137
AV1_COMP::last_frame_uf
YV12_BUFFER_CONFIG last_frame_uf
Definition: encoder.h:2919
AV1_COMP::enc_quant_dequant_params
EncQuantDequantParams enc_quant_dequant_params
Definition: encoder.h:2769
AV1_COMP::ref_frame_dist_info
RefFrameDistanceInfo ref_frame_dist_info
Definition: encoder.h:3237
AV1_COMP::ducky_encode_info
DuckyEncodeInfo ducky_encode_info
Definition: encoder.h:3453
AV1_COMP::initial_mbs
int initial_mbs
Definition: encoder.h:3058
AV1_COMP::rec_sse
uint64_t rec_sse
Definition: encoder.h:3441
AV1_COMP::scaled_last_source
YV12_BUFFER_CONFIG scaled_last_source
Definition: encoder.h:2850
AV1_COMP::mv_stats
MV_STATS mv_stats
Definition: encoder.h:3337
AV1_COMP::frame_new_probs
FrameProbInfo frame_new_probs[10]
Definition: encoder.h:3098
AV1_COMP::source
YV12_BUFFER_CONFIG * source
Definition: encoder.h:2820
AV1_COMP::allocated_tiles
int allocated_tiles
Definition: encoder.h:3073
AV1_COMP::svc
SVC svc
Definition: encoder.h:3263
AV1_COMP::coeff_buffer_base
CB_COEFF_BUFFER * coeff_buffer_base
Definition: encoder.h:2791
AV1_COMP::noise_estimate
NOISE_ESTIMATE noise_estimate
Definition: encoder.h:3296
AV1_COMP::twopass_frame
TWO_PASS_FRAME twopass_frame
Definition: encoder.h:3421
AV1_PRIMARY
Top level primary encoder structure.
Definition: encoder.h:2469
AV1_PRIMARY::num_fp_contexts
int num_fp_contexts
Definition: encoder.h:2526
AV1_PRIMARY::buffer_removal_time_present
bool buffer_removal_time_present
Definition: encoder.h:2640
AV1_PRIMARY::valid_gm_model_found
int valid_gm_model_found[FRAME_UPDATE_TYPES]
Definition: encoder.h:2748
AV1_PRIMARY::output_pkt_list
struct aom_codec_pkt_list * output_pkt_list
Definition: encoder.h:2570
AV1_PRIMARY::filter_level
int filter_level[2]
Definition: encoder.h:2531
AV1_PRIMARY::seq_params
SequenceHeader seq_params
Definition: encoder.h:2630
AV1_PRIMARY::mv_stats
MV_STATS mv_stats
Definition: encoder.h:2678
AV1_PRIMARY::cpi
struct AV1_COMP * cpi
Definition: encoder.h:2547
AV1_PRIMARY::level_params
AV1LevelParams level_params
Definition: encoder.h:2600
AV1_PRIMARY::internal_altref_allowed
int internal_altref_allowed
Definition: encoder.h:2575
AV1_PRIMARY::rtc_ref
RTC_REF rtc_ref
Definition: encoder.h:2753
AV1_PRIMARY::b_calculate_psnr
int b_calculate_psnr
Definition: encoder.h:2605
AV1_PRIMARY::p_mt_info
PrimaryMultiThreadInfo p_mt_info
Definition: encoder.h:2735
AV1_PRIMARY::tf_info
TEMPORAL_FILTER_INFO tf_info
Definition: encoder.h:2625
AV1_PRIMARY::twopass
TWO_PASS twopass
Definition: encoder.h:2615
AV1_PRIMARY::frames_left
int frames_left
Definition: encoder.h:2610
AV1_PRIMARY::ts_start_last_show_frame
int64_t ts_start_last_show_frame
Definition: encoder.h:2516
AV1_PRIMARY::p_rc
PRIMARY_RATE_CONTROL p_rc
Definition: encoder.h:2620
AV1_PRIMARY::lap_enabled
int lap_enabled
Definition: encoder.h:2595
AV1_PRIMARY::frame_probs
FrameProbInfo frame_probs
Definition: encoder.h:2740
AV1_PRIMARY::show_existing_alt_ref
int show_existing_alt_ref
Definition: encoder.h:2580
AV1_PRIMARY::fb_of_context_type
int fb_of_context_type[REF_FRAMES]
Definition: encoder.h:2730
AV1_PRIMARY::use_svc
int use_svc
Definition: encoder.h:2635
AV1_PRIMARY::tpl_sb_rdmult_scaling_factors
double * tpl_sb_rdmult_scaling_factors
Definition: encoder.h:2668
AV1_PRIMARY::filter_level_v
int filter_level_v
Definition: encoder.h:2541
AV1_PRIMARY::filter_level_u
int filter_level_u
Definition: encoder.h:2536
AV1_PRIMARY::cpi_lap
struct AV1_COMP * cpi_lap
Definition: encoder.h:2552
AV1_PRIMARY::parallel_cpi
struct AV1_COMP * parallel_cpi[4]
Definition: encoder.h:2473
AV1_PRIMARY::ts_end_last_show_frame
int64_t ts_end_last_show_frame
Definition: encoder.h:2521
AV1_PRIMARY::lookahead
struct lookahead_ctx * lookahead
Definition: encoder.h:2557
AV1_PRIMARY::gf_state
GF_STATE gf_state
Definition: encoder.h:2590
AV1_PRIMARY::fn_ptr
aom_variance_fn_ptr_t fn_ptr[BLOCK_SIZES_ALL]
Definition: encoder.h:2662
AV1_PRIMARY::gf_group
GF_GROUP gf_group
Definition: encoder.h:2585
AV1_PRIMARY::parallel_frames_data
struct AV1_COMP_DATA parallel_frames_data[4 - 1]
Definition: encoder.h:2479
AV1_PRIMARY::tpl_data
TplParams tpl_data
Definition: encoder.h:2673
AV1_PRIMARY::number_temporal_layers
unsigned int number_temporal_layers
Definition: encoder.h:2645
AV1_PRIMARY::number_spatial_layers
unsigned int number_spatial_layers
Definition: encoder.h:2650
AV1_PRIMARY::seq_params_locked
int seq_params_locked
Definition: encoder.h:2564
AV1_PRIMARY::error
struct aom_internal_error_info error
Definition: encoder.h:2655
AlgoCfg
Algorithm configuration parameters.
Definition: encoder.h:809
AlgoCfg::disable_trellis_quant
int disable_trellis_quant
Definition: encoder.h:825
AlgoCfg::sharpness
int sharpness
Definition: encoder.h:816
AlgoCfg::skip_postproc_filtering
bool skip_postproc_filtering
Definition: encoder.h:869
AlgoCfg::arnr_max_frames
int arnr_max_frames
Definition: encoder.h:830
AlgoCfg::enable_tpl_model
bool enable_tpl_model
Definition: encoder.h:848
AlgoCfg::loopfilter_control
LOOPFILTER_CONTROL loopfilter_control
Definition: encoder.h:863
AlgoCfg::cdf_update_mode
uint8_t cdf_update_mode
Definition: encoder.h:843
AlgoCfg::enable_overlay
bool enable_overlay
Definition: encoder.h:854
AlgoCfg::arnr_strength
int arnr_strength
Definition: encoder.h:835
CB_COEFF_BUFFER
Stores the transforms coefficients for the whole superblock.
Definition: block.h:177
CYCLIC_REFRESH
The stucture of CYCLIC_REFRESH.
Definition: aq_cyclicrefresh.h:36
CdefInfo
Parameters related to CDEF.
Definition: av1_common_int.h:196
CdefInfo::cdef_bits
int cdef_bits
Number of CDEF strength values in bits.
Definition: av1_common_int.h:218
CdefInfo::cdef_uv_strengths
int cdef_uv_strengths[16]
CDEF strength values for chroma.
Definition: av1_common_int.h:216
CdefInfo::cdef_strengths
int cdef_strengths[16]
CDEF strength values for luma.
Definition: av1_common_int.h:214
CdefInfo::nb_cdef_strengths
int nb_cdef_strengths
Number of CDEF strength values.
Definition: av1_common_int.h:212
CoeffBufferPool
Definition: encoder.h:2337
CoeffBufferPool::entropy_ctx
uint8_t * entropy_ctx
Definition: encoder.h:2349
CoeffBufferPool::tcoeff
tran_low_t * tcoeff
Definition: encoder.h:2341
CoeffBufferPool::eobs
uint16_t * eobs
Definition: encoder.h:2345
CommonModeInfoParams
Params related to MB_MODE_INFO arrays and related info.
Definition: av1_common_int.h:501
CommonModeInfoParams::mi_alloc_bsize
BLOCK_SIZE mi_alloc_bsize
Definition: av1_common_int.h:550
CommonTileParams::cols
int cols
Definition: av1_common_int.h:428
CommonTileParams::large_scale
unsigned int large_scale
Definition: av1_common_int.h:488
CompoundTypeCfg
Encoder flags for compound prediction modes.
Definition: encoder.h:395
CompoundTypeCfg::enable_masked_comp
bool enable_masked_comp
Definition: encoder.h:404
CompoundTypeCfg::enable_diff_wtd_comp
bool enable_diff_wtd_comp
Definition: encoder.h:412
CompoundTypeCfg::enable_smooth_interintra
bool enable_smooth_interintra
Definition: encoder.h:408
CompoundTypeCfg::enable_interintra_wedge
bool enable_interintra_wedge
Definition: encoder.h:420
CompoundTypeCfg::enable_interinter_wedge
bool enable_interinter_wedge
Definition: encoder.h:416
CompoundTypeCfg::enable_dist_wtd_comp
bool enable_dist_wtd_comp
Definition: encoder.h:399
CompoundTypeRdBuffers
Contains buffers used by av1_compound_type_rd()
Definition: block.h:337
EncSegmentationInfo
Segmentation related information for the current frame.
Definition: encoder.h:2300
EncSegmentationInfo::map
uint8_t * map
Definition: encoder.h:2306
EncSegmentationInfo::has_lossless_segment
bool has_lossless_segment
Definition: encoder.h:2312
EncodeFrameInput
Input frames and last input frame.
Definition: encoder.h:3470
EncodeFrameParams
contains per-frame encoding parameters decided upon by av1_encode_strategy() and passed down to av1_e...
Definition: encoder.h:3482
EncodeFrameParams::error_resilient_mode
int error_resilient_mode
Definition: encoder.h:3486
EncodeFrameParams::remapped_ref_idx
int remapped_ref_idx[REF_FRAMES]
Definition: encoder.h:3517
EncodeFrameParams::ref_frame_flags
int ref_frame_flags
Definition: encoder.h:3512
EncodeFrameParams::speed
int speed
Definition: encoder.h:3528
EncodeFrameParams::frame_type
FRAME_TYPE frame_type
Definition: encoder.h:3490
EncodeFrameParams::show_frame
int show_frame
Definition: encoder.h:3500
EncodeFrameParams::refresh_frame
RefreshFrameInfo refresh_frame
Definition: encoder.h:3523
ExtRefreshFrameFlagsInfo
Frame refresh flags set by the external interface.
Definition: encoder.h:2167
ExtRefreshFrameFlagsInfo::golden_frame
bool golden_frame
Definition: encoder.h:2169
ExtRefreshFrameFlagsInfo::bwd_ref_frame
bool bwd_ref_frame
Definition: encoder.h:2170
ExtRefreshFrameFlagsInfo::update_pending
bool update_pending
Definition: encoder.h:2176
ExtRefreshFrameFlagsInfo::last_frame
bool last_frame
Definition: encoder.h:2168
ExtRefreshFrameFlagsInfo::alt_ref_frame
bool alt_ref_frame
Definition: encoder.h:2172
ExtRefreshFrameFlagsInfo::alt2_ref_frame
bool alt2_ref_frame
Definition: encoder.h:2171
ExternalFlags
Flags signalled by the external interface at frame level.
Definition: encoder.h:2182
ExternalFlags::use_primary_ref_none
bool use_primary_ref_none
Definition: encoder.h:2223
ExternalFlags::use_ref_frame_mvs
bool use_ref_frame_mvs
Definition: encoder.h:2207
ExternalFlags::refresh_frame
ExtRefreshFrameFlagsInfo refresh_frame
Definition: encoder.h:2191
ExternalFlags::ref_frame_flags
int ref_frame_flags
Definition: encoder.h:2186
ExternalFlags::use_error_resilient
bool use_error_resilient
Definition: encoder.h:2212
ExternalFlags::use_s_frame
bool use_s_frame
Definition: encoder.h:2217
ExternalFlags::refresh_frame_context
bool refresh_frame_context
Definition: encoder.h:2196
ExternalFlags::refresh_frame_context_pending
bool refresh_frame_context_pending
Definition: encoder.h:2202
FeatureFlags
Frame level features.
Definition: av1_common_int.h:358
FeatureFlags::allow_screen_content_tools
bool allow_screen_content_tools
Definition: av1_common_int.h:375
FeatureFlags::allow_intrabc
bool allow_intrabc
Definition: av1_common_int.h:376
FeatureFlags::coded_lossless
bool coded_lossless
Definition: av1_common_int.h:385
FeatureFlags::error_resilient_mode
bool error_resilient_mode
Definition: av1_common_int.h:400
FeatureFlags::all_lossless
bool all_lossless
Definition: av1_common_int.h:389
ForceIntegerMVInfo
Encoder info used for decision on forcing integer motion vectors.
Definition: encoder.h:1730
ForceIntegerMVInfo::rate_size
int rate_size
Definition: encoder.h:1743
ForceIntegerMVInfo::rate_index
int rate_index
Definition: encoder.h:1739
FrameProbInfo
Encoder-side probabilities for pruning of various AV1 tools.
Definition: encoder.h:1094
GF_GROUP
Data related to the current GF/ARF group and the individual frames within the group.
Definition: firstpass.h:344
GlobalMotionInfo
Parameters related to global motion search.
Definition: encoder.h:1939
GlobalMotionInfo::search_done
bool search_done
Definition: encoder.h:1943
GlobalMotionInfo::segment_map_h
int segment_map_h
Definition: encoder.h:1977
GlobalMotionInfo::num_src_corners
int num_src_corners
Definition: encoder.h:1983
GlobalMotionInfo::src_buffer
unsigned char * src_buffer
Definition: encoder.h:1955
GlobalMotionInfo::segment_map_w
int segment_map_w
Definition: encoder.h:1976
InitialDimensions
Initial frame dimensions.
Definition: encoder.h:2002
InitialDimensions::height
int height
Definition: encoder.h:2004
InitialDimensions::width
int width
Definition: encoder.h:2003
InterpSearchFlags
Flags related to interpolation filter search.
Definition: encoder.h:2010
InterpSearchFlags::default_interp_skip_flags
int default_interp_skip_flags
Definition: encoder.h:2015
InterpSearchFlags::interp_filter_search_mask
uint16_t interp_filter_search_mask
Definition: encoder.h:2019
IntraModeCfg
Encoder flags for intra prediction.
Definition: encoder.h:298
IntraModeCfg::enable_diagonal_intra
bool enable_diagonal_intra
Definition: encoder.h:328
IntraModeCfg::enable_smooth_intra
bool enable_smooth_intra
Definition: encoder.h:311
IntraModeCfg::auto_intra_tools_off
bool auto_intra_tools_off
Definition: encoder.h:347
IntraModeCfg::enable_filter_intra
bool enable_filter_intra
Definition: encoder.h:307
IntraModeCfg::enable_directional_intra
bool enable_directional_intra
Definition: encoder.h:323
IntraModeCfg::enable_paeth_intra
bool enable_paeth_intra
Definition: encoder.h:315
IntraModeCfg::enable_intra_edge_filter
bool enable_intra_edge_filter
Definition: encoder.h:302
IntraModeCfg::enable_cfl_intra
bool enable_cfl_intra
Definition: encoder.h:319
IntraModeCfg::enable_angle_delta
bool enable_angle_delta
Definition: encoder.h:333
KeyFrameCfg
Encoder config related to the coding of key frames.
Definition: encoder.h:462
KeyFrameCfg::key_freq_max
int key_freq_max
Definition: encoder.h:471
KeyFrameCfg::sframe_mode
int sframe_mode
Definition: encoder.h:489
KeyFrameCfg::auto_key
bool auto_key
Definition: encoder.h:494
KeyFrameCfg::enable_intrabc
bool enable_intrabc
Definition: encoder.h:514
KeyFrameCfg::sframe_dist
int sframe_dist
Definition: encoder.h:482
KeyFrameCfg::enable_sframe
bool enable_sframe
Definition: encoder.h:509
KeyFrameCfg::enable_keyframe_filtering
int enable_keyframe_filtering
Definition: encoder.h:476
KeyFrameCfg::fwd_kf_dist
int fwd_kf_dist
Definition: encoder.h:499
KeyFrameCfg::key_freq_min
int key_freq_min
Definition: encoder.h:466
KeyFrameCfg::fwd_kf_enabled
bool fwd_kf_enabled
Definition: encoder.h:504
MBMIExtFrameBufferInfo
Buffer to store mode information at mi_alloc_bsize (4x4 or 8x8) level.
Definition: encoder.h:1776
MBMIExtFrameBufferInfo::alloc_size
int alloc_size
Definition: encoder.h:1785
MBMIExtFrameBufferInfo::stride
int stride
Definition: encoder.h:1789
MBMIExtFrameBufferInfo::frame_base
MB_MODE_INFO_EXT_FRAME * frame_base
Definition: encoder.h:1781
MB_MODE_INFO_EXT_FRAME
Stores best extended mode information at frame level.
Definition: block.h:213
MB_MODE_INFO
Stores the prediction/txfm mode of the current coding block.
Definition: blockd.h:222
MotionVectorSearchParams
Parameters for motion vector search process.
Definition: encoder.h:2025
MotionVectorSearchParams::max_mv_magnitude
int max_mv_magnitude
Definition: encoder.h:2031
MotionVectorSearchParams::find_fractional_mv_step
fractional_mv_step_fp * find_fractional_mv_step
Definition: encoder.h:2045
MotionVectorSearchParams::mv_step_param
int mv_step_param
Definition: encoder.h:2036
MultiThreadInfo
Encoder parameters related to multi-threading.
Definition: encoder.h:1627
MultiThreadInfo::restore_state_buf
RestoreStateBuffers restore_state_buf
Definition: encoder.h:1708
MultiThreadInfo::cdef_worker
AV1CdefWorkerData * cdef_worker
Definition: encoder.h:1703
MultiThreadInfo::lr_row_sync
AV1LrSync lr_row_sync
Definition: encoder.h:1678
MultiThreadInfo::tile_thr_data
struct EncWorkerData * tile_thr_data
Definition: encoder.h:1647
MultiThreadInfo::tpl_row_mt
AV1TplRowMultiThreadInfo tpl_row_mt
Definition: encoder.h:1668
MultiThreadInfo::pack_bs_sync
AV1EncPackBSSync pack_bs_sync
Definition: encoder.h:1683
MultiThreadInfo::enc_row_mt
AV1EncRowMultiThreadInfo enc_row_mt
Definition: encoder.h:1663
MultiThreadInfo::lf_row_sync
AV1LfSync lf_row_sync
Definition: encoder.h:1673
MultiThreadInfo::cdef_sync
AV1CdefSync cdef_sync
Definition: encoder.h:1698
MultiThreadInfo::num_mod_workers
int num_mod_workers[NUM_MT_MODULES]
Definition: encoder.h:1636
MultiThreadInfo::num_workers
int num_workers
Definition: encoder.h:1631
MultiThreadInfo::pipeline_lpf_mt_with_enc
int pipeline_lpf_mt_with_enc
Definition: encoder.h:1714
MultiThreadInfo::workers
AVxWorker * workers
Definition: encoder.h:1641
MultiThreadInfo::pack_bs_mt_enabled
bool pack_bs_mt_enabled
Definition: encoder.h:1658
MultiThreadInfo::row_mt_enabled
bool row_mt_enabled
Definition: encoder.h:1653
MultiThreadInfo::tf_sync
AV1TemporalFilterSync tf_sync
Definition: encoder.h:1693
MultiThreadInfo::gm_sync
AV1GlobalMotionSync gm_sync
Definition: encoder.h:1688
OBMCBuffer
Contains buffers used to speed up rdopt for obmc.
Definition: block.h:300
PALETTE_BUFFER
Contains color maps used in palette mode.
Definition: block.h:325
PRIMARY_RATE_CONTROL
Primary Rate Control parameters and status.
Definition: ratectrl.h:270
PartitionCfg
Encoder config for coding block partitioning.
Definition: encoder.h:270
PartitionCfg::enable_rect_partitions
bool enable_rect_partitions
Definition: encoder.h:274
PartitionCfg::enable_1to4_partitions
bool enable_1to4_partitions
Definition: encoder.h:282
PartitionCfg::max_partition_size
BLOCK_SIZE max_partition_size
Definition: encoder.h:292
PartitionCfg::enable_ab_partitions
bool enable_ab_partitions
Definition: encoder.h:278
PartitionCfg::min_partition_size
BLOCK_SIZE min_partition_size
Definition: encoder.h:287
PrimaryMultiThreadInfo
Primary Encoder parameters related to multi-threading.
Definition: encoder.h:1585
PrimaryMultiThreadInfo::tile_thr_data
struct EncWorkerData * tile_thr_data
Definition: encoder.h:1605
PrimaryMultiThreadInfo::cdef_worker
AV1CdefWorkerData * cdef_worker
Definition: encoder.h:1610
PrimaryMultiThreadInfo::num_workers
int num_workers
Definition: encoder.h:1589
PrimaryMultiThreadInfo::workers
AVxWorker * workers
Definition: encoder.h:1599
PrimaryMultiThreadInfo::p_workers
AVxWorker * p_workers[4]
Definition: encoder.h:1616
PrimaryMultiThreadInfo::p_num_workers
int p_num_workers
Definition: encoder.h:1621
PrimaryMultiThreadInfo::num_mod_workers
int num_mod_workers[NUM_MT_MODULES]
Definition: encoder.h:1594
RATE_CONTROL
Rate Control parameters and status.
Definition: ratectrl.h:134
RateControlCfg
Encoder rate control configuration parameters.
Definition: encoder.h:520
RateControlCfg::worst_allowed_q
int worst_allowed_q
Definition: encoder.h:590
RateControlCfg::over_shoot_pct
int over_shoot_pct
Definition: encoder.h:585
RateControlCfg::max_intra_bitrate_pct
unsigned int max_intra_bitrate_pct
Definition: encoder.h:555
RateControlCfg::drop_frames_water_mark
int drop_frames_water_mark
Definition: encoder.h:573
RateControlCfg::vbrmax_section
int vbrmax_section
Definition: encoder.h:621
RateControlCfg::maximum_buffer_size_ms
int64_t maximum_buffer_size_ms
Definition: encoder.h:539
RateControlCfg::vbr_corpus_complexity_lap
unsigned int vbr_corpus_complexity_lap
Definition: encoder.h:550
RateControlCfg::min_cr
unsigned int min_cr
Definition: encoder.h:569
RateControlCfg::vbrbias
int vbrbias
Definition: encoder.h:611
RateControlCfg::gf_cbr_boost_pct
unsigned int gf_cbr_boost_pct
Definition: encoder.h:564
RateControlCfg::vbrmin_section
int vbrmin_section
Definition: encoder.h:616
RateControlCfg::mode
enum aom_rc_mode mode
Definition: encoder.h:604
RateControlCfg::max_inter_bitrate_pct
unsigned int max_inter_bitrate_pct
Definition: encoder.h:560
RateControlCfg::starting_buffer_level_ms
int64_t starting_buffer_level_ms
Definition: encoder.h:529
RateControlCfg::best_allowed_q
int best_allowed_q
Definition: encoder.h:595
RateControlCfg::under_shoot_pct
int under_shoot_pct
Definition: encoder.h:579
RateControlCfg::target_bandwidth
int64_t target_bandwidth
Definition: encoder.h:544
RateControlCfg::optimal_buffer_level_ms
int64_t optimal_buffer_level_ms
Definition: encoder.h:534
RateControlCfg::cq_level
int cq_level
Definition: encoder.h:599
RefFrameDistanceInfo
Refrence frame distance related variables.
Definition: encoder.h:2084
RefFrameDistanceInfo::nearest_past_ref
int8_t nearest_past_ref
Definition: encoder.h:2092
RefFrameDistanceInfo::nearest_future_ref
int8_t nearest_future_ref
Definition: encoder.h:2096
RefreshFrameInfo
Refresh frame flags for different type of frames.
Definition: encoder.h:2063
RefreshFrameInfo::bwd_ref_frame
bool bwd_ref_frame
Definition: encoder.h:2065
RefreshFrameInfo::golden_frame
bool golden_frame
Definition: encoder.h:2064
RefreshFrameInfo::alt_ref_frame
bool alt_ref_frame
Definition: encoder.h:2066
ResizeCfg
Encoder config related to resize.
Definition: encoder.h:250
ResizeCfg::resize_scale_denominator
uint8_t resize_scale_denominator
Definition: encoder.h:259
ResizeCfg::resize_kf_scale_denominator
uint8_t resize_kf_scale_denominator
Definition: encoder.h:264
ResizeCfg::resize_mode
RESIZE_MODE resize_mode
Definition: encoder.h:254
ResizePendingParams
Desired dimensions for an externally triggered resize.
Definition: encoder.h:2076
ResizePendingParams::width
int width
Definition: encoder.h:2077
ResizePendingParams::height
int height
Definition: encoder.h:2078
RestorationInfo
Parameters related to Restoration Info.
Definition: restoration.h:256
RestorationInfo::frame_restoration_type
RestorationType frame_restoration_type
Definition: restoration.h:260
RestoreStateBuffers
Buffers to be backed up during parallel encode set to be restored later.
Definition: encoder.h:1560
RestoreStateBuffers::rst_tmpbuf
int32_t * rst_tmpbuf
Definition: encoder.h:1574
RestoreStateBuffers::rlbs
RestorationLineBuffers * rlbs
Definition: encoder.h:1579
RestoreStateBuffers::cdef_colbuf
uint16_t * cdef_colbuf[3]
Definition: encoder.h:1569
RestoreStateBuffers::cdef_srcbuf
uint16_t * cdef_srcbuf
Definition: encoder.h:1564
SPEED_FEATURES
Top level speed vs quality trade off data struture.
Definition: speed_features.h:1686
SPEED_FEATURES::mv_sf
MV_SPEED_FEATURES mv_sf
Definition: speed_features.h:1715
SPEED_FEATURES::lpf_sf
LOOP_FILTER_SPEED_FEATURES lpf_sf
Definition: speed_features.h:1750
SPEED_FEATURES::tx_sf
TX_SPEED_FEATURES tx_sf
Definition: speed_features.h:1735
SPEED_FEATURES::rt_sf
REAL_TIME_SPEED_FEATURES rt_sf
Definition: speed_features.h:1755
SVC
The stucture of SVC.
Definition: svc_layercontext.h:88
SuperResCfg
Encoder config related to frame super-resolution.
Definition: encoder.h:426
SuperResCfg::superres_kf_scale_denominator
uint8_t superres_kf_scale_denominator
Definition: encoder.h:448
SuperResCfg::superres_mode
aom_superres_mode superres_mode
Definition: encoder.h:452
SuperResCfg::superres_kf_qthresh
int superres_kf_qthresh
Definition: encoder.h:436
SuperResCfg::enable_superres
bool enable_superres
Definition: encoder.h:456
SuperResCfg::superres_scale_denominator
uint8_t superres_scale_denominator
Definition: encoder.h:442
SuperResCfg::superres_qthresh
int superres_qthresh
Definition: encoder.h:431
TEMPORAL_FILTER_INFO
Temporal filter info for a gop.
Definition: temporal_filter.h:156
TWO_PASS_FRAME
Frame level Two pass status and control data.
Definition: firstpass.h:464
TWO_PASS
Two pass status and control data.
Definition: firstpass.h:421
TemporalFilterCtx
Parameters related to temporal filtering.
Definition: temporal_filter.h:93
TimeStamps
Frame time stamps.
Definition: encoder.h:2318
TimeStamps::prev_ts_start
int64_t prev_ts_start
Definition: encoder.h:2322
TimeStamps::first_ts_start
int64_t first_ts_start
Definition: encoder.h:2330
TimeStamps::prev_ts_end
int64_t prev_ts_end
Definition: encoder.h:2326
TplParams
Params related to temporal dependency model.
Definition: tpl_model.h:146
TxfmSizeTypeCfg
Encoder flags for transform sizes and types.
Definition: encoder.h:353
TxfmSizeTypeCfg::enable_tx64
bool enable_tx64
Definition: encoder.h:357
TxfmSizeTypeCfg::use_inter_dct_only
bool use_inter_dct_only
Definition: encoder.h:380
TxfmSizeTypeCfg::enable_flip_idtx
bool enable_flip_idtx
Definition: encoder.h:361
TxfmSizeTypeCfg::use_intra_default_tx_only
bool use_intra_default_tx_only
Definition: encoder.h:385
TxfmSizeTypeCfg::use_intra_dct_only
bool use_intra_dct_only
Definition: encoder.h:375
TxfmSizeTypeCfg::enable_rect_tx
bool enable_rect_tx
Definition: encoder.h:365
TxfmSizeTypeCfg::reduced_tx_type_set
bool reduced_tx_type_set
Definition: encoder.h:370
TxfmSizeTypeCfg::enable_tx_size_search
bool enable_tx_size_search
Definition: encoder.h:389
VarBasedPartitionInfo
Thresholds for variance based partitioning.
Definition: encoder.h:1345
VarBasedPartitionInfo::threshold_minmax
int64_t threshold_minmax
Definition: encoder.h:1360
WinnerModeParams
Parameters used for winner mode processing.
Definition: encoder.h:2114
aom_fixed_buf
Generic fixed size buffer structure.
Definition: aom_encoder.h:86
inter_modes_info
Struct used to hold inter mode data for fast tx search.
Definition: encoder.h:1250
inter_modes_info::rd_cost_arr
RD_STATS rd_cost_arr[MAX_INTER_MODES]
Definition: encoder.h:1279
inter_modes_info::est_rd_arr
int64_t est_rd_arr[MAX_INTER_MODES]
Definition: encoder.h:1271
inter_modes_info::sse_arr
int64_t sse_arr[MAX_INTER_MODES]
Definition: encoder.h:1267
inter_modes_info::rd_cost_y_arr
RD_STATS rd_cost_y_arr[MAX_INTER_MODES]
Definition: encoder.h:1283
inter_modes_info::rd_cost_uv_arr
RD_STATS rd_cost_uv_arr[MAX_INTER_MODES]
Definition: encoder.h:1287
inter_modes_info::mbmi_arr
MB_MODE_INFO mbmi_arr[MAX_INTER_MODES]
Definition: encoder.h:1259
inter_modes_info::rd_idx_pair_arr
RdIdxPair rd_idx_pair_arr[MAX_INTER_MODES]
Definition: encoder.h:1275
inter_modes_info::mode_rate_arr
int mode_rate_arr[MAX_INTER_MODES]
Definition: encoder.h:1263
inter_modes_info::num
int num
Definition: encoder.h:1255
macroblock
Encoder's parameters related to the current coding block.
Definition: block.h:849
macroblock::mbmi_ext_frame
MB_MODE_INFO_EXT_FRAME * mbmi_ext_frame
Finalized mbmi_ext for the whole frame.
Definition: block.h:881
macroblockd
Variables related to current coding block.
Definition: blockd.h:574
macroblockd::block_ref_scale_factors
const struct scale_factors * block_ref_scale_factors[2]
Definition: blockd.h:691
yv12_buffer_config
YV12 frame buffer data structure.
Definition: yv12config.h:39