| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. If the |
| * Alliance for Open Media Patent License 1.0 was not distributed with this |
| * source code in the PATENTS file, you can obtain it at |
| * aomedia.org/license/patent-license/. |
| */ |
| |
| #ifndef AOM_AV1_ENCODER_GLOBAL_MOTION_H_ |
| #define AOM_AV1_ENCODER_GLOBAL_MOTION_H_ |
| |
| #include "aom/aom_integer.h" |
| #include "aom_dsp/flow_estimation/flow_estimation.h" |
| #include "aom_scale/yv12config.h" |
| #include "aom_util/aom_thread.h" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| #define RANSAC_NUM_MOTIONS 1 |
| #define GM_MAX_REFINEMENT_STEPS 5 |
| #define MAX_DIRECTIONS 2 |
| |
| // The structure holds a valid reference frame type and its temporal distance |
| // from the source frame. |
| typedef struct { |
| int distance; |
| MV_REFERENCE_FRAME frame; |
| } FrameDistPair; |
| |
| typedef struct { |
| // Array of structure which holds the global motion parameters for a given |
| // motion model. motion_models[i] holds the parameters for a given motion |
| // model for the ith ransac motion. |
| MotionModel motion_models[RANSAC_NUM_MOTIONS]; |
| |
| // Pointer to hold inliers from motion model. |
| uint8_t *segment_map; |
| } GlobalMotionThreadData; |
| |
| typedef struct { |
| // Holds the mapping of each thread to past/future direction. |
| // thread_id_to_dir[i] indicates the direction id (past - 0/future - 1) |
| // assigned to the ith thread. |
| int8_t thread_id_to_dir[MAX_NUM_THREADS]; |
| |
| // A flag which holds the early exit status based on the speed feature |
| // 'prune_ref_frame_for_gm_search'. early_exit[i] will be set if the speed |
| // feature based early exit happens in the direction 'i'. |
| int8_t early_exit[MAX_DIRECTIONS]; |
| |
| // Counter for the next reference frame to be processed. |
| // next_frame_to_process[i] will hold the count of next reference frame to be |
| // processed in the direction 'i'. |
| int8_t next_frame_to_process[MAX_DIRECTIONS]; |
| } JobInfo; |
| |
| typedef struct { |
| // Data related to assigning jobs for global motion multi-threading. |
| JobInfo job_info; |
| |
| // Data specific to each worker in global motion multi-threading. |
| // thread_data[i] stores the thread specific data for worker 'i'. |
| GlobalMotionThreadData *thread_data; |
| |
| #if CONFIG_MULTITHREAD |
| // Mutex lock used while dispatching jobs. |
| pthread_mutex_t *mutex_; |
| #endif |
| |
| // Width and height for which segment_map is allocated for each thread. |
| int allocated_width; |
| int allocated_height; |
| |
| // Number of workers for which thread_data is allocated. |
| int8_t allocated_workers; |
| } AV1GlobalMotionSync; |
| |
| void av1_convert_model_to_params(const double *params, |
| WarpedMotionParams *model); |
| |
| // TODO(sarahparker) These need to be retuned for speed 0 and 1 to |
| // maximize gains from segmented error metric |
| static const double erroradv_tr = 0.65; |
| static const double erroradv_prod_tr = 20000; |
| |
| int av1_is_enough_erroradvantage(double best_erroradvantage, int params_cost); |
| |
| void av1_compute_feature_segmentation_map(uint8_t *segment_map, int width, |
| int height, int *inliers, |
| int num_inliers); |
| |
| // Returns the error between the result of applying motion 'wm' to the frame |
| // described by 'ref' and the frame described by 'dst'. |
| int64_t av1_warp_error(WarpedMotionParams *wm, int bd, const uint16_t *ref, |
| int width, int height, int stride, uint16_t *dst, |
| int p_col, int p_row, int p_width, int p_height, |
| int p_stride, int subsampling_x, int subsampling_y, |
| int64_t best_error, uint8_t *segment_map, |
| int segment_map_stride); |
| |
| // Returns the av1_warp_error between "dst" and the result of applying the |
| // motion params that result from fine-tuning "wm" to "ref". Note that "wm" is |
| // modified in place. |
| int64_t av1_refine_integerized_param( |
| WarpedMotionParams *wm, TransformationType wmtype, int bd, uint16_t *ref, |
| int r_width, int r_height, int r_stride, uint16_t *dst, int d_width, |
| int d_height, int d_stride, int n_refinements, int64_t best_frame_error, |
| uint8_t *segment_map, int segment_map_stride, int64_t erroradv_threshold); |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| #endif // AOM_AV1_ENCODER_GLOBAL_MOTION_H_ |