/* * Copyright (c) 2016, Alliance for Open Media. All rights reserved. * * This source code is subject to the terms of the BSD 2 Clause License and * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License * was not distributed with this source code in the LICENSE file, you can * obtain it at www.aomedia.org/license/software. If the Alliance for Open * Media Patent License 1.0 was not distributed with this source code in the * PATENTS file, you can obtain it at www.aomedia.org/license/patent. */ #ifndef AOM_AV1_COMMON_RESTORATION_H_ #define AOM_AV1_COMMON_RESTORATION_H_ #include "aom_ports/mem.h" #include "config/aom_config.h" #include "av1/common/blockd.h" #include "av1/common/enums.h" #ifdef __cplusplus extern "C" { #endif /*! @file */ /*!\cond */ // Border for Loop restoration buffer #define AOM_RESTORATION_FRAME_BORDER … #define CLIP(x, lo, hi) … #define RINT(x) … #define RESTORATION_PROC_UNIT_SIZE … // Filter stripe grid offset upwards compared to the superblock grid #define RESTORATION_UNIT_OFFSET … #define SGRPROJ_BORDER_VERT … #define SGRPROJ_BORDER_HORZ … #define WIENER_BORDER_VERT … #define WIENER_HALFWIN … #define WIENER_BORDER_HORZ … // RESTORATION_BORDER_VERT determines line buffer requirement for LR. // Should be set at the max of SGRPROJ_BORDER_VERT and WIENER_BORDER_VERT. // Note the line buffer needed is twice the value of this macro. #if SGRPROJ_BORDER_VERT >= WIENER_BORDER_VERT #define RESTORATION_BORDER_VERT … #else #define RESTORATION_BORDER_VERT … #endif // SGRPROJ_BORDER_VERT >= WIENER_BORDER_VERT #if SGRPROJ_BORDER_HORZ >= WIENER_BORDER_HORZ #define RESTORATION_BORDER_HORZ … #else #define RESTORATION_BORDER_HORZ … #endif // SGRPROJ_BORDER_VERT >= WIENER_BORDER_VERT // How many border pixels do we need for each processing unit? #define RESTORATION_BORDER … // How many rows of deblocked pixels do we save above/below each processing // stripe? #define RESTORATION_CTX_VERT … // Additional pixels to the left and right in above/below buffers // It is RESTORATION_BORDER_HORZ rounded up to get nicer buffer alignment #define RESTORATION_EXTRA_HORZ … // Pad up to 20 more (may be much less is needed) #define RESTORATION_PADDING … #define RESTORATION_PROC_UNIT_PELS … #define RESTORATION_UNITSIZE_MAX … #define RESTORATION_UNITPELS_HORZ_MAX … #define RESTORATION_UNITPELS_VERT_MAX … #define RESTORATION_UNITPELS_MAX … // Two 32-bit buffers needed for the restored versions from two filters // TODO(debargha, rupert): Refactor to not need the large tilesize to be stored // on the decoder side. #define SGRPROJ_TMPBUF_SIZE … #define SGRPROJ_EXTBUF_SIZE … #define SGRPROJ_PARAMS_BITS … #define SGRPROJ_PARAMS … // Precision bits for projection #define SGRPROJ_PRJ_BITS … // Restoration precision bits generated higher than source before projection #define SGRPROJ_RST_BITS … // Internal precision bits for core selfguided_restoration #define SGRPROJ_SGR_BITS … #define SGRPROJ_SGR … #define SGRPROJ_PRJ_MIN0 … #define SGRPROJ_PRJ_MAX0 … #define SGRPROJ_PRJ_MIN1 … #define SGRPROJ_PRJ_MAX1 … #define SGRPROJ_PRJ_SUBEXP_K … #define SGRPROJ_BITS … #define MAX_RADIUS … #define MAX_NELEM … #define SGRPROJ_MTABLE_BITS … #define SGRPROJ_RECIP_BITS … #define WIENER_HALFWIN1 … #define WIENER_WIN … #define WIENER_WIN2 … #define WIENER_TMPBUF_SIZE … #define WIENER_EXTBUF_SIZE … // If WIENER_WIN_CHROMA == WIENER_WIN - 2, that implies 5x5 filters are used for // chroma. To use 7x7 for chroma set WIENER_WIN_CHROMA to WIENER_WIN. #define WIENER_WIN_CHROMA … #define WIENER_WIN_REDUCED … #define WIENER_WIN2_CHROMA … #define WIENER_STATS_DOWNSAMPLE_FACTOR … #define WIENER_FILT_PREC_BITS … #define WIENER_FILT_STEP … // Central values for the taps #define WIENER_FILT_TAP0_MIDV … #define WIENER_FILT_TAP1_MIDV … #define WIENER_FILT_TAP2_MIDV … #define WIENER_FILT_TAP3_MIDV … #define WIENER_FILT_TAP0_BITS … #define WIENER_FILT_TAP1_BITS … #define WIENER_FILT_TAP2_BITS … #define WIENER_FILT_BITS … #define WIENER_FILT_TAP0_MINV … #define WIENER_FILT_TAP1_MINV … #define WIENER_FILT_TAP2_MINV … #define WIENER_FILT_TAP0_MAXV … #define WIENER_FILT_TAP1_MAXV … #define WIENER_FILT_TAP2_MAXV … #define WIENER_FILT_TAP0_SUBEXP_K … #define WIENER_FILT_TAP1_SUBEXP_K … #define WIENER_FILT_TAP2_SUBEXP_K … // Max of SGRPROJ_TMPBUF_SIZE, DOMAINTXFMRF_TMPBUF_SIZE, WIENER_TMPBUF_SIZE #define RESTORATION_TMPBUF_SIZE … // Max of SGRPROJ_EXTBUF_SIZE, WIENER_EXTBUF_SIZE #define RESTORATION_EXTBUF_SIZE … // Check the assumptions of the existing code #if SUBPEL_TAPS != WIENER_WIN + 1 #error "Wiener filter currently only works if SUBPEL_TAPS == WIENER_WIN + 1" #endif #if WIENER_FILT_PREC_BITS != 7 #error "Wiener filter currently only works if WIENER_FILT_PREC_BITS == 7" #endif sgr_params_type; /*!\endcond */ /*!\brief Parameters related to Restoration Unit Info */ RestorationUnitInfo; /*!\cond */ // A restoration line buffer needs space for two lines plus a horizontal filter // margin of RESTORATION_EXTRA_HORZ on each side. #define RESTORATION_LINEBUFFER_WIDTH … RestorationLineBuffers; /*!\endcond */ /*!\brief Parameters related to Restoration Stripe boundaries */ RestorationStripeBoundaries; /*!\brief Parameters related to Restoration Info */ RestorationInfo; /*!\cond */ static inline void set_default_sgrproj(SgrprojInfo *sgrproj_info) { … } static inline void set_default_wiener(WienerInfo *wiener_info) { … } RestorationTileLimits; rest_unit_visitor_t; FilterFrameCtxt; AV1LrStruct; extern const sgr_params_type av1_sgr_params[SGRPROJ_PARAMS]; extern int sgrproj_mtable[SGRPROJ_PARAMS][2]; extern const int32_t av1_x_by_xplus1[256]; extern const int32_t av1_one_by_x[MAX_NELEM]; void av1_alloc_restoration_struct(struct AV1Common *cm, RestorationInfo *rsi, int is_uv); void av1_free_restoration_struct(RestorationInfo *rst_info); void av1_extend_frame(uint8_t *data, int width, int height, int stride, int border_horz, int border_vert, int highbd); void av1_decode_xq(const int *xqd, int *xq, const sgr_params_type *params); /*!\endcond */ /*!\brief Function for applying loop restoration filter to a single unit. * * \ingroup in_loop_restoration * This function applies the loop restoration filter to a single * loop restoration unit. * * \param[in] limits Limits of the unit * \param[in] rui The parameters to use for this unit and its * coefficients * \param[in] rsb Deblocked pixels to use for stripe boundaries * \param[in] rlbs Space to use as a scratch buffer * \param[in] ss_x Horizontal subsampling for plane * \param[in] ss_y Vertical subsampling for plane * \param[in] plane_w Width of the current plane * \param[in] plane_h Height of the current plane * \param[in] highbd Whether high bitdepth pipeline is used * \param[in] bit_depth Bit-depth of the video * \param[in] data8 Frame data (pointing at the top-left corner of * the frame, not the restoration unit). * \param[in] stride Stride of \c data8 * \param[out] dst8 Buffer where the results will be written. Like * \c data8, \c dst8 should point at the top-left * corner of the frame * \param[in] dst_stride Stride of \c dst8 * \param[in] tmpbuf Scratch buffer used by the sgrproj filter * which should be at least SGRPROJ_TMPBUF_SIZE * big. * \param[in] optimized_lr Whether to use fast optimized Loop Restoration * \param[in,out] error_info Error info for reporting errors * * \remark Nothing is returned. Instead, the filtered unit is output in * \c dst8 at the proper restoration unit offset. */ void av1_loop_restoration_filter_unit( const RestorationTileLimits *limits, const RestorationUnitInfo *rui, const RestorationStripeBoundaries *rsb, RestorationLineBuffers *rlbs, int plane_w, int plane_h, int ss_x, int ss_y, int highbd, int bit_depth, uint8_t *data8, int stride, uint8_t *dst8, int dst_stride, int32_t *tmpbuf, int optimized_lr, struct aom_internal_error_info *error_info); /*!\brief Function for applying loop restoration filter to a frame * * \ingroup in_loop_restoration * This function applies the loop restoration filter to a frame. * * \param[in,out] frame Compressed frame buffer * \param[in,out] cm Pointer to top level common structure * \param[in] optimized_lr Whether to use fast optimized Loop Restoration * \param[in] lr_ctxt Loop restoration context * * \remark Nothing is returned. Instead, the filtered frame is output in * \c frame. */ void av1_loop_restoration_filter_frame(YV12_BUFFER_CONFIG *frame, struct AV1Common *cm, int optimized_lr, void *lr_ctxt); /*!\cond */ void av1_loop_restoration_precal(void); struct AV1LrSyncData; sync_read_fn_t; sync_write_fn_t; // Return 1 iff the block at mi_row, mi_col with size bsize is a // top-level superblock containing the top-left corner of at least one // loop restoration unit. // // If the block is a top-level superblock, the function writes to // *rcol0, *rcol1, *rrow0, *rrow1. This means that the parameters for all // restoration units in the rectangle [*rcol0, *rcol1) x [*rrow0, *rrow1) // are signaled in this superblock. int av1_loop_restoration_corners_in_sb(const struct AV1Common *cm, int plane, int mi_row, int mi_col, BLOCK_SIZE bsize, int *rcol0, int *rcol1, int *rrow0, int *rrow1); void av1_loop_restoration_save_boundary_lines(const YV12_BUFFER_CONFIG *frame, struct AV1Common *cm, int after_cdef); void av1_loop_restoration_filter_frame_init(AV1LrStruct *lr_ctxt, YV12_BUFFER_CONFIG *frame, struct AV1Common *cm, int optimized_lr, int num_planes); void av1_loop_restoration_copy_planes(AV1LrStruct *loop_rest_ctxt, struct AV1Common *cm, int num_planes); void av1_foreach_rest_unit_in_row( RestorationTileLimits *limits, int plane_w, rest_unit_visitor_t on_rest_unit, int row_number, int unit_size, int hnum_rest_units, int vnum_rest_units, int plane, void *priv, int32_t *tmpbuf, RestorationLineBuffers *rlbs, sync_read_fn_t on_sync_read, sync_write_fn_t on_sync_write, struct AV1LrSyncData *const lr_sync, struct aom_internal_error_info *error_info); void av1_get_upsampled_plane_size(const struct AV1Common *cm, int is_uv, int *plane_w, int *plane_h); int av1_lr_count_units(int unit_size, int plane_size); void av1_lr_sync_read_dummy(void *const lr_sync, int r, int c, int plane); void av1_lr_sync_write_dummy(void *const lr_sync, int r, int c, const int sb_cols, int plane); /*!\endcond */ #ifdef __cplusplus } // extern "C" #endif #endif // AOM_AV1_COMMON_RESTORATION_H_
Codebase Browser
chromium