diff options
author | Ronald S. Bultje <rsbultje@gmail.com> | 2018-09-05 00:32:00 +0300 |
---|---|---|
committer | Ronald S. Bultje <rsbultje@gmail.com> | 2018-09-22 14:59:19 +0300 |
commit | e2892ffa2dd1e893d0229c5fcbe0bbbee8e11c20 (patch) | |
tree | 9ba42513c2916db89d47f8b34fa0141d6aeb4dde /src/cdf.h | |
parent | d32eb2d935a31288c34ccaa42b09bc60eacdfd68 (diff) |
Initial decoder implementation.
With minor contributions from:
- Jean-Baptiste Kempf <jb@videolan.org>
- Marvin Scholz <epirat07@gmail.com>
- Hugo Beauzée-Luyssen <hugo@videolan.org>
Diffstat (limited to 'src/cdf.h')
-rw-r--r-- | src/cdf.h | 148 |
1 files changed, 148 insertions, 0 deletions
diff --git a/src/cdf.h b/src/cdf.h new file mode 100644 index 0000000..014f36b --- /dev/null +++ b/src/cdf.h @@ -0,0 +1,148 @@ +/* + * Copyright © 2018, VideoLAN and dav1d authors + * Copyright © 2018, Two Orioles, LLC + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR + * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; + * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND + * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS + * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#ifndef __AV1_CDF_H__ +#define __AV1_CDF_H__ + +#include <stdint.h> + +#include "src/levels.h" +#include "src/ref.h" +#include "src/thread_data.h" + +typedef struct CdfModeContext { + uint16_t y_mode[4][N_INTRA_PRED_MODES + 1]; + uint16_t use_filter_intra[N_BS_SIZES][2]; + uint16_t filter_intra[5 + 1]; + uint16_t uv_mode[2][N_INTRA_PRED_MODES][N_UV_INTRA_PRED_MODES + 1]; + uint16_t angle_delta[8][8]; + uint16_t filter[2][8][N_SWITCHABLE_FILTERS + 1]; + uint16_t newmv_mode[6][2]; + uint16_t globalmv_mode[2][2]; + uint16_t refmv_mode[6][2]; + uint16_t drl_bit[3][2]; + uint16_t comp_inter_mode[8][N_COMP_INTER_PRED_MODES + 1]; + uint16_t intra[4][2]; + uint16_t comp[5][2]; + uint16_t comp_dir[5][2]; + uint16_t jnt_comp[6][2]; + uint16_t mask_comp[6][2]; + uint16_t wedge_comp[9][2]; + uint16_t wedge_idx[9][16 + 1]; + uint16_t interintra[7][2]; + uint16_t interintra_mode[4][5]; + uint16_t interintra_wedge[7][2]; + uint16_t ref[6][3][2]; + uint16_t comp_fwd_ref[3][3][2]; + uint16_t comp_bwd_ref[2][3][2]; + uint16_t comp_uni_ref[3][3][2]; + uint16_t txsz[N_TX_SIZES - 1][3][4]; + uint16_t txpart[7][3][2]; + uint16_t txtp_inter[4][N_TX_SIZES][N_TX_TYPES + 1]; + uint16_t txtp_intra[3][N_TX_SIZES][N_INTRA_PRED_MODES][N_TX_TYPES + 1]; + uint16_t skip[3][2]; + uint16_t skip_mode[3][2]; + uint16_t partition[N_BL_LEVELS][4][N_PARTITIONS + 1]; + uint16_t seg_pred[3][2]; + uint16_t seg_id[3][NUM_SEGMENTS + 1]; + uint16_t cfl_sign[8 + 1]; + uint16_t cfl_alpha[6][16 + 1]; + uint16_t restore_wiener[2]; + uint16_t restore_sgrproj[2]; + uint16_t restore_switchable[3 + 1]; + uint16_t delta_q[4 + 1]; + uint16_t delta_lf[5][4 + 1]; + uint16_t obmc[N_BS_SIZES][2]; + uint16_t motion_mode[N_BS_SIZES][3 + 1]; + uint16_t pal_y[7][3][2]; + uint16_t pal_uv[2][2]; + uint16_t pal_sz[2][7][7 + 1]; + uint16_t color_map[2][7][5][8 + 1]; + uint16_t intrabc[2]; +} CdfModeContext; + +typedef struct CdfCoefContext { + uint16_t skip[N_TX_SIZES][13][2]; + uint16_t eob_bin_16[2][2][6]; + uint16_t eob_bin_32[2][2][7]; + uint16_t eob_bin_64[2][2][8]; + uint16_t eob_bin_128[2][2][9]; + uint16_t eob_bin_256[2][2][10]; + uint16_t eob_bin_512[2][2][11]; + uint16_t eob_bin_1024[2][2][12]; + uint16_t eob_hi_bit[N_TX_SIZES][2][11 /*22*/][2]; + uint16_t eob_base_tok[N_TX_SIZES][2][4][4]; + uint16_t base_tok[N_TX_SIZES][2][41][5]; + uint16_t dc_sign[2][3][2]; + uint16_t br_tok[4 /*5*/][2][21][5]; +} CdfCoefContext; + +typedef struct CdfMvComponent { + uint16_t classes[11 + 1]; + uint16_t class0[2]; + uint16_t classN[10][2]; + uint16_t class0_fp[2][4 + 1]; + uint16_t classN_fp[4 + 1]; + uint16_t class0_hp[2]; + uint16_t classN_hp[2]; + uint16_t sign[2]; +} CdfMvComponent; + +typedef struct CdfMvContext { + CdfMvComponent comp[2]; + uint16_t joint[N_MV_JOINTS + 1]; +} CdfMvContext; + +typedef struct CdfContext { + CdfModeContext m; + uint16_t kfym[5][5][N_INTRA_PRED_MODES + 1]; + CdfCoefContext coef; + CdfMvContext mv, dmv; +} CdfContext; + +typedef struct CdfThreadContext { + CdfContext *cdf; + Dav1dRef *ref; ///< allocation origin + struct thread_data *t; + atomic_uint *progress; +} CdfThreadContext; + +void av1_init_states(CdfThreadContext *cdf, int qidx); +void av1_update_tile_cdf(const Av1FrameHeader *hdr, CdfContext *dst, + const CdfContext *src); + +void cdf_thread_alloc(CdfThreadContext *cdf, struct thread_data *t); +void cdf_thread_ref(CdfThreadContext *dst, CdfThreadContext *src); +void cdf_thread_unref(CdfThreadContext *cdf); + +/* + * These are binary signals (so a signal is either "done" or "not done"). + */ +void cdf_thread_wait(CdfThreadContext *cdf); +void cdf_thread_signal(CdfThreadContext *cdf); + +#endif /* __AV1_CDF_H__ */ |