blob: 817e3620b00fa2e9df07c862c4296c6045e8c42c [file] [log] [blame]
/*
* Copyright (c) 2019, Alliance for Open Media. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#ifndef AOM_AV1_ENCODER_SVC_LAYERCONTEXT_H_
#define AOM_AV1_ENCODER_SVC_LAYERCONTEXT_H_
#include "av1/encoder/aq_cyclicrefresh.h"
#include "av1/encoder/encoder.h"
#include "av1/encoder/ratectrl.h"
#ifdef __cplusplus
extern "C" {
#endif
/*!
* \brief The stucture of quantities related to each spatial and temporal layer.
* \ingroup SVC
*/
typedef struct {
/*!\cond */
RATE_CONTROL rc;
PRIMARY_RATE_CONTROL p_rc;
int framerate_factor;
int64_t layer_target_bitrate;
int scaling_factor_num;
int scaling_factor_den;
int64_t target_bandwidth;
int64_t spatial_layer_target_bandwidth;
double framerate;
int avg_frame_size;
int max_q;
int min_q;
int frames_from_key_frame;
/*!\endcond */
/*!
* Cyclic refresh parameters (aq-mode=3), that need to be updated per-frame.
*/
int sb_index;
/*!
* Segmentation map
*/
int8_t *map;
/*!
* Segmentation map for last coded quantization paramters.
*/
uint8_t *last_coded_q_map;
/*!
* Number of blocks on segment 1
*/
int actual_num_seg1_blocks;
/*!
* Number of blocks on segment 2
*/
int actual_num_seg2_blocks;
/*!
* Counter used to detect scene change.
*/
int counter_encode_maxq_scene_change;
/*!
* Speed settings for each layer.
*/
uint8_t speed;
/*!
* GF group index.
*/
unsigned char group_index;
/*!
* If current layer is key frame.
*/
int is_key_frame;
/*!
* Maximum motion magnitude of previous encoded layer.
*/
int max_mv_magnitude;
} LAYER_CONTEXT;
/*!
* \brief The stucture of SVC.
* \ingroup SVC
*/
typedef struct SVC {
/*!\cond */
int spatial_layer_id;
int temporal_layer_id;
int number_spatial_layers;
int number_temporal_layers;
int set_ref_frame_config;
int non_reference_frame;
int use_flexible_mode;
int ksvc_fixed_mode;
/*!\endcond */
/*!
* LAST_FRAME (0), LAST2_FRAME(1), LAST3_FRAME(2), GOLDEN_FRAME(3),
* BWDREF_FRAME(4), ALTREF2_FRAME(5), ALTREF_FRAME(6).
*/
int reference[INTER_REFS_PER_FRAME];
/*!\cond */
int ref_idx[INTER_REFS_PER_FRAME];
int refresh[REF_FRAMES];
double base_framerate;
unsigned int current_superframe;
unsigned int buffer_time_index[REF_FRAMES];
unsigned char buffer_spatial_layer[REF_FRAMES];
int skip_mvsearch_last;
int skip_mvsearch_gf;
int spatial_layer_fb[REF_FRAMES];
int temporal_layer_fb[REF_FRAMES];
int num_encoded_top_layer;
int first_layer_denoise;
/*!\endcond */
/*!
* Layer context used for rate control in CBR mode.
*/
LAYER_CONTEXT layer_context[AOM_MAX_LAYERS];
/*!
* EIGHTTAP_SMOOTH or BILINEAR
*/
InterpFilter downsample_filter_type[AOM_MAX_SS_LAYERS];
/*!
* Downsample_filter_phase: = 0 will do sub-sampling (no weighted average),
* = 8 will center the target pixel and get a symmetric averaging filter.
*/
int downsample_filter_phase[AOM_MAX_SS_LAYERS];
/*!
* Force zero-mv in mode search for the spatial/inter-layer reference.
*/
int force_zero_mode_spatial_ref;
} SVC;
struct AV1_COMP;
/*!\brief Initialize layer context data from init_config().
*
* \ingroup SVC
* \callgraph
* \callergraph
*
* \param[in] cpi Top level encoder structure
*
* \return Nothing returned. Set cpi->svc.
*/
void av1_init_layer_context(struct AV1_COMP *const cpi);
/*!\brief Update the layer context from a change_config() call.
*
* \ingroup SVC
* \callgraph
* \callergraph
*
* \param[in] cpi Top level encoder structure
* \param[in] target_bandwidth Total target bandwidth
*
* \return Nothing returned. Buffer level for each layer is set.
*/
void av1_update_layer_context_change_config(struct AV1_COMP *const cpi,
const int64_t target_bandwidth);
/*!\brief Prior to encoding the frame, update framerate-related quantities
for the current temporal layer.
*
* \ingroup SVC
* \callgraph
* \callergraph
*
* \param[in] cpi Top level encoder structure
*
* \return Nothing returned. Frame related quantities for current temporal
layer are updated.
*/
void av1_update_temporal_layer_framerate(struct AV1_COMP *const cpi);
/*!\brief Prior to encoding the frame, set the layer context, for the current
layer to be encoded, to the cpi struct.
*
* \ingroup SVC
* \callgraph
* \callergraph
*
* \param[in] cpi Top level encoder structure
*
* \return Nothing returned. Layer context for current layer is set.
*/
void av1_restore_layer_context(struct AV1_COMP *const cpi);
/*!\brief Save the layer context after encoding the frame.
*
* \ingroup SVC
* \callgraph
* \callergraph
*
* \param[in] cpi Top level encoder structure
*
* \return Nothing returned.
*/
void av1_save_layer_context(struct AV1_COMP *const cpi);
/*!\brief Free the memory used for cyclic refresh in layer context.
*
* \ingroup SVC
* \callgraph
* \callergraph
*
* \param[in] cpi Top level encoder structure
*
* \return Nothing returned.
*/
void av1_free_svc_cyclic_refresh(struct AV1_COMP *const cpi);
/*!\brief Reset on key frame: reset counters, references and buffer updates.
*
* \ingroup SVC
* \callgraph
* \callergraph
*
* \param[in] cpi Top level encoder structure
* \param[in] is_key Whether current layer is key frame
*
* \return Nothing returned.
*/
void av1_svc_reset_temporal_layers(struct AV1_COMP *const cpi, int is_key);
/*!\brief Before encoding, set resolutions and allocate compressor data.
*
* \ingroup SVC
* \callgraph
* \callergraph
*
* \param[in] cpi Top level encoder structure
*
* \return Nothing returned.
*/
void av1_one_pass_cbr_svc_start_layer(struct AV1_COMP *const cpi);
/*!\brief Get primary reference frame for current layer
*
* \ingroup SVC
* \callgraph
* \callergraph
*
* \param[in] cpi Top level encoder structure
*
* \return The primary reference frame for current layer.
*/
int av1_svc_primary_ref_frame(const struct AV1_COMP *const cpi);
/*!\brief Get resolution for current layer.
*
* \ingroup SVC
* \param[in] width_org Original width, unscaled
* \param[in] height_org Original height, unscaled
* \param[in] num Numerator for the scale ratio
* \param[in] den Denominator for the scale ratio
* \param[in] width_out Output width, scaled for current layer
* \param[in] height_out Output height, scaled for current layer
*
* \return Nothing is returned. Instead the scaled width and height are set.
*/
void av1_get_layer_resolution(const int width_org, const int height_org,
const int num, const int den, int *width_out,
int *height_out);
void av1_set_svc_fixed_mode(struct AV1_COMP *const cpi);
void av1_svc_check_reset_layer_rc_flag(struct AV1_COMP *const cpi);
#ifdef __cplusplus
} // extern "C"
#endif
#endif // AOM_AV1_ENCODER_SVC_LAYERCONTEXT_H_