Coverage Report

Created: 2025-07-23 06:32

/src/aom/av1/decoder/decodeframe.c
Line
Count
Source (jump to first uncovered line)
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
12
#include <assert.h>
13
#include <stdbool.h>
14
#include <stddef.h>
15
16
#include "config/aom_config.h"
17
#include "config/aom_scale_rtcd.h"
18
19
#include "aom/aom_codec.h"
20
#include "aom/aom_image.h"
21
#include "aom/internal/aom_codec_internal.h"
22
#include "aom_dsp/aom_dsp_common.h"
23
#include "aom_dsp/binary_codes_reader.h"
24
#include "aom_dsp/bitreader.h"
25
#include "aom_dsp/bitreader_buffer.h"
26
#include "aom_dsp/txfm_common.h"
27
#include "aom_mem/aom_mem.h"
28
#include "aom_ports/aom_timer.h"
29
#include "aom_ports/mem.h"
30
#include "aom_ports/mem_ops.h"
31
#include "aom_scale/yv12config.h"
32
#include "aom_util/aom_pthread.h"
33
#include "aom_util/aom_thread.h"
34
35
#if CONFIG_BITSTREAM_DEBUG || CONFIG_MISMATCH_DEBUG
36
#include "aom_util/debug_util.h"
37
#endif  // CONFIG_BITSTREAM_DEBUG || CONFIG_MISMATCH_DEBUG
38
39
#include "av1/common/alloccommon.h"
40
#include "av1/common/av1_common_int.h"
41
#include "av1/common/blockd.h"
42
#include "av1/common/cdef.h"
43
#include "av1/common/cfl.h"
44
#include "av1/common/common_data.h"
45
#include "av1/common/common.h"
46
#include "av1/common/entropy.h"
47
#include "av1/common/entropymode.h"
48
#include "av1/common/entropymv.h"
49
#include "av1/common/enums.h"
50
#include "av1/common/frame_buffers.h"
51
#include "av1/common/idct.h"
52
#include "av1/common/mv.h"
53
#include "av1/common/mvref_common.h"
54
#include "av1/common/obmc.h"
55
#include "av1/common/pred_common.h"
56
#include "av1/common/quant_common.h"
57
#include "av1/common/reconinter.h"
58
#include "av1/common/reconintra.h"
59
#include "av1/common/resize.h"
60
#include "av1/common/restoration.h"
61
#include "av1/common/scale.h"
62
#include "av1/common/seg_common.h"
63
#include "av1/common/thread_common.h"
64
#include "av1/common/tile_common.h"
65
#include "av1/common/warped_motion.h"
66
67
#include "av1/decoder/decodeframe.h"
68
#include "av1/decoder/decodemv.h"
69
#include "av1/decoder/decoder.h"
70
#include "av1/decoder/decodetxb.h"
71
#include "av1/decoder/detokenize.h"
72
#if CONFIG_INSPECTION
73
#include "av1/decoder/inspection.h"
74
#endif
75
76
#define ACCT_STR __func__
77
78
86.8k
#define AOM_MIN_THREADS_PER_TILE 1
79
126k
#define AOM_MAX_THREADS_PER_TILE 2
80
81
// This is needed by ext_tile related unit tests.
82
#define EXT_TILE_DEBUG 1
83
#define MC_TEMP_BUF_PELS                       \
84
215k
  (((MAX_SB_SIZE)*2 + (AOM_INTERP_EXTEND)*2) * \
85
215k
   ((MAX_SB_SIZE)*2 + (AOM_INTERP_EXTEND)*2))
86
87
// Checks that the remaining bits start with a 1 and ends with 0s.
88
// It consumes an additional byte, if already byte aligned before the check.
89
79.5k
int av1_check_trailing_bits(AV1Decoder *pbi, struct aom_read_bit_buffer *rb) {
90
  // bit_offset is set to 0 (mod 8) when the reader is already byte aligned
91
79.5k
  int bits_before_alignment = 8 - rb->bit_offset % 8;
92
79.5k
  int trailing = aom_rb_read_literal(rb, bits_before_alignment);
93
79.5k
  if (trailing != (1 << (bits_before_alignment - 1))) {
94
5.46k
    pbi->error.error_code = AOM_CODEC_CORRUPT_FRAME;
95
5.46k
    return -1;
96
5.46k
  }
97
74.1k
  return 0;
98
79.5k
}
99
100
// Use only_chroma = 1 to only set the chroma planes
101
static inline void set_planes_to_neutral_grey(
102
    const SequenceHeader *const seq_params, const YV12_BUFFER_CONFIG *const buf,
103
246k
    int only_chroma) {
104
246k
  if (seq_params->use_highbitdepth) {
105
116k
    const int val = 1 << (seq_params->bit_depth - 1);
106
454k
    for (int plane = only_chroma; plane < MAX_MB_PLANE; plane++) {
107
338k
      const int is_uv = plane > 0;
108
338k
      uint16_t *const base = CONVERT_TO_SHORTPTR(buf->buffers[plane]);
109
      // Set the first row to neutral grey. Then copy the first row to all
110
      // subsequent rows.
111
338k
      if (buf->crop_heights[is_uv] > 0) {
112
338k
        aom_memset16(base, val, buf->crop_widths[is_uv]);
113
107M
        for (int row_idx = 1; row_idx < buf->crop_heights[is_uv]; row_idx++) {
114
106M
          memcpy(&base[row_idx * buf->strides[is_uv]], base,
115
106M
                 sizeof(*base) * buf->crop_widths[is_uv]);
116
106M
        }
117
338k
      }
118
338k
    }
119
130k
  } else {
120
521k
    for (int plane = only_chroma; plane < MAX_MB_PLANE; plane++) {
121
391k
      const int is_uv = plane > 0;
122
52.0M
      for (int row_idx = 0; row_idx < buf->crop_heights[is_uv]; row_idx++) {
123
51.6M
        memset(&buf->buffers[plane][row_idx * buf->strides[is_uv]], 1 << 7,
124
51.6M
               buf->crop_widths[is_uv]);
125
51.6M
      }
126
391k
    }
127
130k
  }
128
246k
}
129
130
static inline void loop_restoration_read_sb_coeffs(const AV1_COMMON *const cm,
131
                                                   MACROBLOCKD *xd,
132
                                                   aom_reader *const r,
133
                                                   int plane, int runit_idx);
134
135
201k
static int read_is_valid(const uint8_t *start, size_t len, const uint8_t *end) {
136
201k
  return len != 0 && len <= (size_t)(end - start);
137
201k
}
138
139
static TX_MODE read_tx_mode(struct aom_read_bit_buffer *rb,
140
168k
                            int coded_lossless) {
141
168k
  if (coded_lossless) return ONLY_4X4;
142
139k
  return aom_rb_read_bit(rb) ? TX_MODE_SELECT : TX_MODE_LARGEST;
143
168k
}
144
145
static REFERENCE_MODE read_frame_reference_mode(
146
168k
    const AV1_COMMON *cm, struct aom_read_bit_buffer *rb) {
147
168k
  if (frame_is_intra_only(cm)) {
148
124k
    return SINGLE_REFERENCE;
149
124k
  } else {
150
44.0k
    return aom_rb_read_bit(rb) ? REFERENCE_MODE_SELECT : SINGLE_REFERENCE;
151
44.0k
  }
152
168k
}
153
154
static inline void inverse_transform_block(DecoderCodingBlock *dcb, int plane,
155
                                           const TX_TYPE tx_type,
156
                                           const TX_SIZE tx_size, uint8_t *dst,
157
15.7M
                                           int stride, int reduced_tx_set) {
158
15.7M
  tran_low_t *const dqcoeff = dcb->dqcoeff_block[plane] + dcb->cb_offset[plane];
159
15.7M
  eob_info *eob_data = dcb->eob_data[plane] + dcb->txb_offset[plane];
160
15.7M
  uint16_t scan_line = eob_data->max_scan_line;
161
15.7M
  uint16_t eob = eob_data->eob;
162
15.7M
  av1_inverse_transform_block(&dcb->xd, dqcoeff, plane, tx_type, tx_size, dst,
163
15.7M
                              stride, eob, reduced_tx_set);
164
15.7M
  memset(dqcoeff, 0, (scan_line + 1) * sizeof(dqcoeff[0]));
165
15.7M
}
166
167
static inline void read_coeffs_tx_intra_block(
168
    const AV1_COMMON *const cm, DecoderCodingBlock *dcb, aom_reader *const r,
169
103M
    const int plane, const int row, const int col, const TX_SIZE tx_size) {
170
103M
  MB_MODE_INFO *mbmi = dcb->xd.mi[0];
171
103M
  if (!mbmi->skip_txfm) {
172
#if TXCOEFF_TIMER
173
    struct aom_usec_timer timer;
174
    aom_usec_timer_start(&timer);
175
#endif
176
23.8M
    av1_read_coeffs_txb(cm, dcb, r, plane, row, col, tx_size);
177
#if TXCOEFF_TIMER
178
    aom_usec_timer_mark(&timer);
179
    const int64_t elapsed_time = aom_usec_timer_elapsed(&timer);
180
    cm->txcoeff_timer += elapsed_time;
181
    ++cm->txb_count;
182
#endif
183
23.8M
  }
184
103M
}
185
186
static inline void decode_block_void(const AV1_COMMON *const cm,
187
                                     DecoderCodingBlock *dcb,
188
                                     aom_reader *const r, const int plane,
189
                                     const int row, const int col,
190
140M
                                     const TX_SIZE tx_size) {
191
140M
  (void)cm;
192
140M
  (void)dcb;
193
140M
  (void)r;
194
140M
  (void)plane;
195
140M
  (void)row;
196
140M
  (void)col;
197
140M
  (void)tx_size;
198
140M
}
199
200
static inline void predict_inter_block_void(AV1_COMMON *const cm,
201
                                            DecoderCodingBlock *dcb,
202
2.93M
                                            BLOCK_SIZE bsize) {
203
2.93M
  (void)cm;
204
2.93M
  (void)dcb;
205
2.93M
  (void)bsize;
206
2.93M
}
207
208
static inline void cfl_store_inter_block_void(AV1_COMMON *const cm,
209
2.93M
                                              MACROBLOCKD *const xd) {
210
2.93M
  (void)cm;
211
2.93M
  (void)xd;
212
2.93M
}
213
214
static inline void predict_and_reconstruct_intra_block(
215
    const AV1_COMMON *const cm, DecoderCodingBlock *dcb, aom_reader *const r,
216
95.9M
    const int plane, const int row, const int col, const TX_SIZE tx_size) {
217
95.9M
  (void)r;
218
95.9M
  MACROBLOCKD *const xd = &dcb->xd;
219
95.9M
  MB_MODE_INFO *mbmi = xd->mi[0];
220
95.9M
  PLANE_TYPE plane_type = get_plane_type(plane);
221
222
95.9M
  av1_predict_intra_block_facade(cm, xd, plane, col, row, tx_size);
223
224
95.9M
  if (!mbmi->skip_txfm) {
225
17.3M
    eob_info *eob_data = dcb->eob_data[plane] + dcb->txb_offset[plane];
226
17.3M
    if (eob_data->eob) {
227
8.59M
      const bool reduced_tx_set_used = cm->features.reduced_tx_set_used;
228
      // tx_type was read out in av1_read_coeffs_txb.
229
8.59M
      const TX_TYPE tx_type = av1_get_tx_type(xd, plane_type, row, col, tx_size,
230
8.59M
                                              reduced_tx_set_used);
231
8.59M
      struct macroblockd_plane *const pd = &xd->plane[plane];
232
8.59M
      uint8_t *dst = &pd->dst.buf[(row * pd->dst.stride + col) << MI_SIZE_LOG2];
233
8.59M
      inverse_transform_block(dcb, plane, tx_type, tx_size, dst, pd->dst.stride,
234
8.59M
                              reduced_tx_set_used);
235
8.59M
    }
236
17.3M
  }
237
95.9M
  if (plane == AOM_PLANE_Y && store_cfl_required(cm, xd)) {
238
2.06M
    cfl_store_tx(xd, row, col, tx_size, mbmi->bsize);
239
2.06M
  }
240
95.9M
}
241
242
static inline void inverse_transform_inter_block(
243
    const AV1_COMMON *const cm, DecoderCodingBlock *dcb, aom_reader *const r,
244
    const int plane, const int blk_row, const int blk_col,
245
7.15M
    const TX_SIZE tx_size) {
246
7.15M
  (void)r;
247
7.15M
  MACROBLOCKD *const xd = &dcb->xd;
248
7.15M
  PLANE_TYPE plane_type = get_plane_type(plane);
249
7.15M
  const struct macroblockd_plane *const pd = &xd->plane[plane];
250
7.15M
  const bool reduced_tx_set_used = cm->features.reduced_tx_set_used;
251
  // tx_type was read out in av1_read_coeffs_txb.
252
7.15M
  const TX_TYPE tx_type = av1_get_tx_type(xd, plane_type, blk_row, blk_col,
253
7.15M
                                          tx_size, reduced_tx_set_used);
254
255
7.15M
  uint8_t *dst =
256
7.15M
      &pd->dst.buf[(blk_row * pd->dst.stride + blk_col) << MI_SIZE_LOG2];
257
7.15M
  inverse_transform_block(dcb, plane, tx_type, tx_size, dst, pd->dst.stride,
258
7.15M
                          reduced_tx_set_used);
259
#if CONFIG_MISMATCH_DEBUG
260
  int pixel_c, pixel_r;
261
  BLOCK_SIZE bsize = txsize_to_bsize[tx_size];
262
  int blk_w = block_size_wide[bsize];
263
  int blk_h = block_size_high[bsize];
264
  const int mi_row = -xd->mb_to_top_edge >> (3 + MI_SIZE_LOG2);
265
  const int mi_col = -xd->mb_to_left_edge >> (3 + MI_SIZE_LOG2);
266
  mi_to_pixel_loc(&pixel_c, &pixel_r, mi_col, mi_row, blk_col, blk_row,
267
                  pd->subsampling_x, pd->subsampling_y);
268
  mismatch_check_block_tx(dst, pd->dst.stride, cm->current_frame.order_hint,
269
                          plane, pixel_c, pixel_r, blk_w, blk_h,
270
                          xd->cur_buf->flags & YV12_FLAG_HIGHBITDEPTH);
271
#endif
272
7.15M
}
273
274
static inline void set_cb_buffer_offsets(DecoderCodingBlock *dcb,
275
170M
                                         TX_SIZE tx_size, int plane) {
276
170M
  dcb->cb_offset[plane] += tx_size_wide[tx_size] * tx_size_high[tx_size];
277
170M
  dcb->txb_offset[plane] =
278
170M
      dcb->cb_offset[plane] / (TX_SIZE_W_MIN * TX_SIZE_H_MIN);
279
170M
}
280
281
static inline void decode_reconstruct_tx(AV1_COMMON *cm, ThreadData *const td,
282
                                         aom_reader *r,
283
                                         MB_MODE_INFO *const mbmi, int plane,
284
                                         BLOCK_SIZE plane_bsize, int blk_row,
285
                                         int blk_col, int block,
286
14.5M
                                         TX_SIZE tx_size, int *eob_total) {
287
14.5M
  DecoderCodingBlock *const dcb = &td->dcb;
288
14.5M
  MACROBLOCKD *const xd = &dcb->xd;
289
14.5M
  const struct macroblockd_plane *const pd = &xd->plane[plane];
290
14.5M
  const TX_SIZE plane_tx_size =
291
14.5M
      plane ? av1_get_max_uv_txsize(mbmi->bsize, pd->subsampling_x,
292
8.22M
                                    pd->subsampling_y)
293
14.5M
            : mbmi->inter_tx_size[av1_get_txb_size_index(plane_bsize, blk_row,
294
6.29M
                                                         blk_col)];
295
  // Scale to match transform block unit.
296
14.5M
  const int max_blocks_high = max_block_high(xd, plane_bsize, plane);
297
14.5M
  const int max_blocks_wide = max_block_wide(xd, plane_bsize, plane);
298
299
14.5M
  if (blk_row >= max_blocks_high || blk_col >= max_blocks_wide) return;
300
301
14.5M
  if (tx_size == plane_tx_size || plane) {
302
14.1M
    td->read_coeffs_tx_inter_block_visit(cm, dcb, r, plane, blk_row, blk_col,
303
14.1M
                                         tx_size);
304
305
14.1M
    td->inverse_tx_inter_block_visit(cm, dcb, r, plane, blk_row, blk_col,
306
14.1M
                                     tx_size);
307
14.1M
    eob_info *eob_data = dcb->eob_data[plane] + dcb->txb_offset[plane];
308
14.1M
    *eob_total += eob_data->eob;
309
14.1M
    set_cb_buffer_offsets(dcb, tx_size, plane);
310
14.1M
  } else {
311
404k
    const TX_SIZE sub_txs = sub_tx_size_map[tx_size];
312
404k
    assert(IMPLIES(tx_size <= TX_4X4, sub_txs == tx_size));
313
440k
    assert(IMPLIES(tx_size > TX_4X4, sub_txs < tx_size));
314
440k
    const int bsw = tx_size_wide_unit[sub_txs];
315
440k
    const int bsh = tx_size_high_unit[sub_txs];
316
440k
    const int sub_step = bsw * bsh;
317
440k
    const int row_end =
318
440k
        AOMMIN(tx_size_high_unit[tx_size], max_blocks_high - blk_row);
319
440k
    const int col_end =
320
440k
        AOMMIN(tx_size_wide_unit[tx_size], max_blocks_wide - blk_col);
321
322
440k
    assert(bsw > 0 && bsh > 0);
323
324
1.20M
    for (int row = 0; row < row_end; row += bsh) {
325
766k
      const int offsetr = blk_row + row;
326
2.10M
      for (int col = 0; col < col_end; col += bsw) {
327
1.34M
        const int offsetc = blk_col + col;
328
329
1.34M
        decode_reconstruct_tx(cm, td, r, mbmi, plane, plane_bsize, offsetr,
330
1.34M
                              offsetc, block, sub_txs, eob_total);
331
1.34M
        block += sub_step;
332
1.34M
      }
333
766k
    }
334
440k
  }
335
14.5M
}
336
337
static inline void set_offsets(AV1_COMMON *const cm, MACROBLOCKD *const xd,
338
                               BLOCK_SIZE bsize, int mi_row, int mi_col, int bw,
339
13.5M
                               int bh, int x_mis, int y_mis) {
340
13.5M
  const int num_planes = av1_num_planes(cm);
341
13.5M
  const CommonModeInfoParams *const mi_params = &cm->mi_params;
342
13.5M
  const TileInfo *const tile = &xd->tile;
343
344
13.5M
  set_mi_offsets(mi_params, xd, mi_row, mi_col);
345
13.5M
  xd->mi[0]->bsize = bsize;
346
#if CONFIG_RD_DEBUG
347
  xd->mi[0]->mi_row = mi_row;
348
  xd->mi[0]->mi_col = mi_col;
349
#endif
350
351
13.5M
  assert(x_mis && y_mis);
352
67.9M
  for (int x = 1; x < x_mis; ++x) xd->mi[x] = xd->mi[0];
353
13.5M
  int idx = mi_params->mi_stride;
354
66.7M
  for (int y = 1; y < y_mis; ++y) {
355
53.1M
    memcpy(&xd->mi[idx], &xd->mi[0], x_mis * sizeof(xd->mi[0]));
356
53.1M
    idx += mi_params->mi_stride;
357
53.1M
  }
358
359
13.5M
  set_plane_n4(xd, bw, bh, num_planes);
360
13.5M
  set_entropy_context(xd, mi_row, mi_col, num_planes);
361
362
  // Distance of Mb to the various image edges. These are specified to 8th pel
363
  // as they are always compared to values that are in 1/8th pel units
364
13.5M
  set_mi_row_col(xd, tile, mi_row, bh, mi_col, bw, mi_params->mi_rows,
365
13.5M
                 mi_params->mi_cols);
366
367
13.5M
  av1_setup_dst_planes(xd->plane, bsize, &cm->cur_frame->buf, mi_row, mi_col, 0,
368
13.5M
                       num_planes);
369
13.5M
}
370
371
static inline void decode_mbmi_block(AV1Decoder *const pbi,
372
                                     DecoderCodingBlock *dcb, int mi_row,
373
                                     int mi_col, aom_reader *r,
374
                                     PARTITION_TYPE partition,
375
13.5M
                                     BLOCK_SIZE bsize) {
376
13.5M
  AV1_COMMON *const cm = &pbi->common;
377
13.5M
  const SequenceHeader *const seq_params = cm->seq_params;
378
13.5M
  const int bw = mi_size_wide[bsize];
379
13.5M
  const int bh = mi_size_high[bsize];
380
13.5M
  const int x_mis = AOMMIN(bw, cm->mi_params.mi_cols - mi_col);
381
13.5M
  const int y_mis = AOMMIN(bh, cm->mi_params.mi_rows - mi_row);
382
13.5M
  MACROBLOCKD *const xd = &dcb->xd;
383
384
#if CONFIG_ACCOUNTING
385
  aom_accounting_set_context(&pbi->accounting, mi_col, mi_row);
386
#endif
387
13.5M
  set_offsets(cm, xd, bsize, mi_row, mi_col, bw, bh, x_mis, y_mis);
388
13.5M
  xd->mi[0]->partition = partition;
389
13.5M
  av1_read_mode_info(pbi, dcb, r, x_mis, y_mis);
390
13.5M
  if (bsize >= BLOCK_8X8 &&
391
13.5M
      (seq_params->subsampling_x || seq_params->subsampling_y)) {
392
7.33M
    const BLOCK_SIZE uv_subsize =
393
7.33M
        av1_ss_size_lookup[bsize][seq_params->subsampling_x]
394
7.33M
                          [seq_params->subsampling_y];
395
7.33M
    if (uv_subsize == BLOCK_INVALID)
396
0
      aom_internal_error(xd->error_info, AOM_CODEC_CORRUPT_FRAME,
397
0
                         "Invalid block size.");
398
7.33M
  }
399
13.5M
}
400
401
typedef struct PadBlock {
402
  int x0;
403
  int x1;
404
  int y0;
405
  int y1;
406
} PadBlock;
407
408
#if CONFIG_AV1_HIGHBITDEPTH
409
static inline void highbd_build_mc_border(const uint8_t *src8, int src_stride,
410
                                          uint8_t *dst8, int dst_stride, int x,
411
                                          int y, int b_w, int b_h, int w,
412
1.08M
                                          int h) {
413
  // Get a pointer to the start of the real data for this row.
414
1.08M
  const uint16_t *src = CONVERT_TO_SHORTPTR(src8);
415
1.08M
  uint16_t *dst = CONVERT_TO_SHORTPTR(dst8);
416
1.08M
  const uint16_t *ref_row = src - x - y * src_stride;
417
418
1.08M
  if (y >= h)
419
109k
    ref_row += (h - 1) * src_stride;
420
978k
  else if (y > 0)
421
620k
    ref_row += y * src_stride;
422
423
22.5M
  do {
424
22.5M
    int right = 0, copy;
425
22.5M
    int left = x < 0 ? -x : 0;
426
427
22.5M
    if (left > b_w) left = b_w;
428
429
22.5M
    if (x + b_w > w) right = x + b_w - w;
430
431
22.5M
    if (right > b_w) right = b_w;
432
433
22.5M
    copy = b_w - left - right;
434
435
22.5M
    if (left) aom_memset16(dst, ref_row[0], left);
436
437
22.5M
    if (copy) memcpy(dst + left, ref_row + x + left, copy * sizeof(uint16_t));
438
439
22.5M
    if (right) aom_memset16(dst + left + copy, ref_row[w - 1], right);
440
441
22.5M
    dst += dst_stride;
442
22.5M
    ++y;
443
444
22.5M
    if (y > 0 && y < h) ref_row += src_stride;
445
22.5M
  } while (--b_h);
446
1.08M
}
447
#endif  // CONFIG_AV1_HIGHBITDEPTH
448
449
static inline void build_mc_border(const uint8_t *src, int src_stride,
450
                                   uint8_t *dst, int dst_stride, int x, int y,
451
1.23M
                                   int b_w, int b_h, int w, int h) {
452
  // Get a pointer to the start of the real data for this row.
453
1.23M
  const uint8_t *ref_row = src - x - y * src_stride;
454
455
1.23M
  if (y >= h)
456
271k
    ref_row += (h - 1) * src_stride;
457
959k
  else if (y > 0)
458
554k
    ref_row += y * src_stride;
459
460
22.2M
  do {
461
22.2M
    int right = 0, copy;
462
22.2M
    int left = x < 0 ? -x : 0;
463
464
22.2M
    if (left > b_w) left = b_w;
465
466
22.2M
    if (x + b_w > w) right = x + b_w - w;
467
468
22.2M
    if (right > b_w) right = b_w;
469
470
22.2M
    copy = b_w - left - right;
471
472
22.2M
    if (left) memset(dst, ref_row[0], left);
473
474
22.2M
    if (copy) memcpy(dst + left, ref_row + x + left, copy);
475
476
22.2M
    if (right) memset(dst + left + copy, ref_row[w - 1], right);
477
478
22.2M
    dst += dst_stride;
479
22.2M
    ++y;
480
481
22.2M
    if (y > 0 && y < h) ref_row += src_stride;
482
22.2M
  } while (--b_h);
483
1.23M
}
484
485
static inline int update_extend_mc_border_params(
486
    const struct scale_factors *const sf, struct buf_2d *const pre_buf,
487
    MV32 scaled_mv, PadBlock *block, int subpel_x_mv, int subpel_y_mv,
488
14.0M
    int do_warp, int is_intrabc, int *x_pad, int *y_pad) {
489
14.0M
  const int is_scaled = av1_is_scaled(sf);
490
  // Get reference width and height.
491
14.0M
  int frame_width = pre_buf->width;
492
14.0M
  int frame_height = pre_buf->height;
493
494
  // Do border extension if there is motion or
495
  // width/height is not a multiple of 8 pixels.
496
14.0M
  if ((!is_intrabc) && (!do_warp) &&
497
14.0M
      (is_scaled || scaled_mv.col || scaled_mv.row || (frame_width & 0x7) ||
498
13.4M
       (frame_height & 0x7))) {
499
11.2M
    if (subpel_x_mv || (sf->x_step_q4 != SUBPEL_SHIFTS)) {
500
11.2M
      block->x0 -= AOM_INTERP_EXTEND - 1;
501
11.2M
      block->x1 += AOM_INTERP_EXTEND;
502
11.2M
      *x_pad = 1;
503
11.2M
    }
504
505
11.2M
    if (subpel_y_mv || (sf->y_step_q4 != SUBPEL_SHIFTS)) {
506
11.2M
      block->y0 -= AOM_INTERP_EXTEND - 1;
507
11.2M
      block->y1 += AOM_INTERP_EXTEND;
508
11.2M
      *y_pad = 1;
509
11.2M
    }
510
511
    // Skip border extension if block is inside the frame.
512
11.2M
    if (block->x0 < 0 || block->x1 > frame_width - 1 || block->y0 < 0 ||
513
11.2M
        block->y1 > frame_height - 1) {
514
2.31M
      return 1;
515
2.31M
    }
516
11.2M
  }
517
11.7M
  return 0;
518
14.0M
}
519
520
static inline void extend_mc_border(const struct scale_factors *const sf,
521
                                    struct buf_2d *const pre_buf,
522
                                    MV32 scaled_mv, PadBlock block,
523
                                    int subpel_x_mv, int subpel_y_mv,
524
                                    int do_warp, int is_intrabc, int highbd,
525
                                    uint8_t *mc_buf, uint8_t **pre,
526
14.0M
                                    int *src_stride) {
527
14.0M
  int x_pad = 0, y_pad = 0;
528
14.0M
  if (update_extend_mc_border_params(sf, pre_buf, scaled_mv, &block,
529
14.0M
                                     subpel_x_mv, subpel_y_mv, do_warp,
530
14.0M
                                     is_intrabc, &x_pad, &y_pad)) {
531
    // Get reference block pointer.
532
2.31M
    const uint8_t *const buf_ptr =
533
2.31M
        pre_buf->buf0 + block.y0 * pre_buf->stride + block.x0;
534
2.31M
    int buf_stride = pre_buf->stride;
535
2.31M
    const int b_w = block.x1 - block.x0;
536
2.31M
    const int b_h = block.y1 - block.y0;
537
538
2.31M
#if CONFIG_AV1_HIGHBITDEPTH
539
    // Extend the border.
540
2.31M
    if (highbd) {
541
1.08M
      highbd_build_mc_border(buf_ptr, buf_stride, mc_buf, b_w, block.x0,
542
1.08M
                             block.y0, b_w, b_h, pre_buf->width,
543
1.08M
                             pre_buf->height);
544
1.23M
    } else {
545
1.23M
      build_mc_border(buf_ptr, buf_stride, mc_buf, b_w, block.x0, block.y0, b_w,
546
1.23M
                      b_h, pre_buf->width, pre_buf->height);
547
1.23M
    }
548
#else
549
    (void)highbd;
550
    build_mc_border(buf_ptr, buf_stride, mc_buf, b_w, block.x0, block.y0, b_w,
551
                    b_h, pre_buf->width, pre_buf->height);
552
#endif
553
2.31M
    *src_stride = b_w;
554
2.31M
    *pre = mc_buf + y_pad * (AOM_INTERP_EXTEND - 1) * b_w +
555
2.31M
           x_pad * (AOM_INTERP_EXTEND - 1);
556
2.31M
  }
557
14.0M
}
558
559
static inline void dec_calc_subpel_params(
560
    const MV *const src_mv, InterPredParams *const inter_pred_params,
561
    const MACROBLOCKD *const xd, int mi_x, int mi_y, uint8_t **pre,
562
    SubpelParams *subpel_params, int *src_stride, PadBlock *block,
563
14.0M
    MV32 *scaled_mv, int *subpel_x_mv, int *subpel_y_mv) {
564
14.0M
  const struct scale_factors *sf = inter_pred_params->scale_factors;
565
14.0M
  struct buf_2d *pre_buf = &inter_pred_params->ref_frame_buf;
566
14.0M
  const int bw = inter_pred_params->block_width;
567
14.0M
  const int bh = inter_pred_params->block_height;
568
14.0M
  const int is_scaled = av1_is_scaled(sf);
569
14.0M
  if (is_scaled) {
570
2.56M
    int ssx = inter_pred_params->subsampling_x;
571
2.56M
    int ssy = inter_pred_params->subsampling_y;
572
2.56M
    int orig_pos_y = inter_pred_params->pix_row << SUBPEL_BITS;
573
2.56M
    orig_pos_y += src_mv->row * (1 << (1 - ssy));
574
2.56M
    int orig_pos_x = inter_pred_params->pix_col << SUBPEL_BITS;
575
2.56M
    orig_pos_x += src_mv->col * (1 << (1 - ssx));
576
2.56M
    int pos_y = av1_scaled_y(orig_pos_y, sf);
577
2.56M
    int pos_x = av1_scaled_x(orig_pos_x, sf);
578
2.56M
    pos_x += SCALE_EXTRA_OFF;
579
2.56M
    pos_y += SCALE_EXTRA_OFF;
580
581
2.56M
    const int top = -AOM_LEFT_TOP_MARGIN_SCALED(ssy);
582
2.56M
    const int left = -AOM_LEFT_TOP_MARGIN_SCALED(ssx);
583
2.56M
    const int bottom = (pre_buf->height + AOM_INTERP_EXTEND)
584
2.56M
                       << SCALE_SUBPEL_BITS;
585
2.56M
    const int right = (pre_buf->width + AOM_INTERP_EXTEND) << SCALE_SUBPEL_BITS;
586
2.56M
    pos_y = clamp(pos_y, top, bottom);
587
2.56M
    pos_x = clamp(pos_x, left, right);
588
589
2.56M
    subpel_params->subpel_x = pos_x & SCALE_SUBPEL_MASK;
590
2.56M
    subpel_params->subpel_y = pos_y & SCALE_SUBPEL_MASK;
591
2.56M
    subpel_params->xs = sf->x_step_q4;
592
2.56M
    subpel_params->ys = sf->y_step_q4;
593
594
    // Get reference block top left coordinate.
595
2.56M
    block->x0 = pos_x >> SCALE_SUBPEL_BITS;
596
2.56M
    block->y0 = pos_y >> SCALE_SUBPEL_BITS;
597
598
    // Get reference block bottom right coordinate.
599
2.56M
    block->x1 =
600
2.56M
        ((pos_x + (bw - 1) * subpel_params->xs) >> SCALE_SUBPEL_BITS) + 1;
601
2.56M
    block->y1 =
602
2.56M
        ((pos_y + (bh - 1) * subpel_params->ys) >> SCALE_SUBPEL_BITS) + 1;
603
604
2.56M
    MV temp_mv;
605
2.56M
    temp_mv = clamp_mv_to_umv_border_sb(xd, src_mv, bw, bh,
606
2.56M
                                        inter_pred_params->subsampling_x,
607
2.56M
                                        inter_pred_params->subsampling_y);
608
2.56M
    *scaled_mv = av1_scale_mv(&temp_mv, mi_x, mi_y, sf);
609
2.56M
    scaled_mv->row += SCALE_EXTRA_OFF;
610
2.56M
    scaled_mv->col += SCALE_EXTRA_OFF;
611
612
2.56M
    *subpel_x_mv = scaled_mv->col & SCALE_SUBPEL_MASK;
613
2.56M
    *subpel_y_mv = scaled_mv->row & SCALE_SUBPEL_MASK;
614
11.4M
  } else {
615
    // Get block position in current frame.
616
11.4M
    int pos_x = inter_pred_params->pix_col << SUBPEL_BITS;
617
11.4M
    int pos_y = inter_pred_params->pix_row << SUBPEL_BITS;
618
619
11.4M
    const MV mv_q4 = clamp_mv_to_umv_border_sb(
620
11.4M
        xd, src_mv, bw, bh, inter_pred_params->subsampling_x,
621
11.4M
        inter_pred_params->subsampling_y);
622
11.4M
    subpel_params->xs = subpel_params->ys = SCALE_SUBPEL_SHIFTS;
623
11.4M
    subpel_params->subpel_x = (mv_q4.col & SUBPEL_MASK) << SCALE_EXTRA_BITS;
624
11.4M
    subpel_params->subpel_y = (mv_q4.row & SUBPEL_MASK) << SCALE_EXTRA_BITS;
625
626
    // Get reference block top left coordinate.
627
11.4M
    pos_x += mv_q4.col;
628
11.4M
    pos_y += mv_q4.row;
629
11.4M
    block->x0 = pos_x >> SUBPEL_BITS;
630
11.4M
    block->y0 = pos_y >> SUBPEL_BITS;
631
632
    // Get reference block bottom right coordinate.
633
11.4M
    block->x1 = (pos_x >> SUBPEL_BITS) + (bw - 1) + 1;
634
11.4M
    block->y1 = (pos_y >> SUBPEL_BITS) + (bh - 1) + 1;
635
636
11.4M
    scaled_mv->row = mv_q4.row;
637
11.4M
    scaled_mv->col = mv_q4.col;
638
11.4M
    *subpel_x_mv = scaled_mv->col & SUBPEL_MASK;
639
11.4M
    *subpel_y_mv = scaled_mv->row & SUBPEL_MASK;
640
11.4M
  }
641
14.0M
  *pre = pre_buf->buf0 + block->y0 * pre_buf->stride + block->x0;
642
14.0M
  *src_stride = pre_buf->stride;
643
14.0M
}
644
645
static inline void dec_calc_subpel_params_and_extend(
646
    const MV *const src_mv, InterPredParams *const inter_pred_params,
647
    MACROBLOCKD *const xd, int mi_x, int mi_y, int ref, uint8_t **mc_buf,
648
14.0M
    uint8_t **pre, SubpelParams *subpel_params, int *src_stride) {
649
14.0M
  PadBlock block;
650
14.0M
  MV32 scaled_mv;
651
14.0M
  int subpel_x_mv, subpel_y_mv;
652
14.0M
  dec_calc_subpel_params(src_mv, inter_pred_params, xd, mi_x, mi_y, pre,
653
14.0M
                         subpel_params, src_stride, &block, &scaled_mv,
654
14.0M
                         &subpel_x_mv, &subpel_y_mv);
655
14.0M
  extend_mc_border(
656
14.0M
      inter_pred_params->scale_factors, &inter_pred_params->ref_frame_buf,
657
14.0M
      scaled_mv, block, subpel_x_mv, subpel_y_mv,
658
14.0M
      inter_pred_params->mode == WARP_PRED, inter_pred_params->is_intrabc,
659
14.0M
      inter_pred_params->use_hbd_buf, mc_buf[ref], pre, src_stride);
660
14.0M
}
661
662
#define IS_DEC 1
663
#include "av1/common/reconinter_template.inc"
664
#undef IS_DEC
665
666
static void dec_build_inter_predictors(const AV1_COMMON *cm,
667
                                       DecoderCodingBlock *dcb, int plane,
668
                                       const MB_MODE_INFO *mi,
669
                                       int build_for_obmc, int bw, int bh,
670
11.3M
                                       int mi_x, int mi_y) {
671
11.3M
  build_inter_predictors(cm, &dcb->xd, plane, mi, build_for_obmc, bw, bh, mi_x,
672
11.3M
                         mi_y, dcb->mc_buf);
673
11.3M
}
674
675
static inline void dec_build_inter_predictor(const AV1_COMMON *cm,
676
                                             DecoderCodingBlock *dcb,
677
                                             int mi_row, int mi_col,
678
3.49M
                                             BLOCK_SIZE bsize) {
679
3.49M
  MACROBLOCKD *const xd = &dcb->xd;
680
3.49M
  const int num_planes = av1_num_planes(cm);
681
12.6M
  for (int plane = 0; plane < num_planes; ++plane) {
682
9.80M
    if (plane && !xd->is_chroma_ref) break;
683
9.17M
    const int mi_x = mi_col * MI_SIZE;
684
9.17M
    const int mi_y = mi_row * MI_SIZE;
685
9.17M
    dec_build_inter_predictors(cm, dcb, plane, xd->mi[0], 0,
686
9.17M
                               xd->plane[plane].width, xd->plane[plane].height,
687
9.17M
                               mi_x, mi_y);
688
9.17M
    if (is_interintra_pred(xd->mi[0])) {
689
769k
      BUFFER_SET ctx = { { xd->plane[0].dst.buf, xd->plane[1].dst.buf,
690
769k
                           xd->plane[2].dst.buf },
691
769k
                         { xd->plane[0].dst.stride, xd->plane[1].dst.stride,
692
769k
                           xd->plane[2].dst.stride } };
693
769k
      av1_build_interintra_predictor(cm, xd, xd->plane[plane].dst.buf,
694
769k
                                     xd->plane[plane].dst.stride, &ctx, plane,
695
769k
                                     bsize);
696
769k
    }
697
9.17M
  }
698
3.49M
}
699
700
static inline void dec_build_prediction_by_above_pred(
701
    MACROBLOCKD *const xd, int rel_mi_row, int rel_mi_col, uint8_t op_mi_size,
702
479k
    int dir, MB_MODE_INFO *above_mbmi, void *fun_ctxt, const int num_planes) {
703
479k
  struct build_prediction_ctxt *ctxt = (struct build_prediction_ctxt *)fun_ctxt;
704
479k
  const int above_mi_col = xd->mi_col + rel_mi_col;
705
479k
  int mi_x, mi_y;
706
479k
  MB_MODE_INFO backup_mbmi = *above_mbmi;
707
708
479k
  (void)rel_mi_row;
709
479k
  (void)dir;
710
711
479k
  av1_setup_build_prediction_by_above_pred(xd, rel_mi_col, op_mi_size,
712
479k
                                           &backup_mbmi, ctxt, num_planes);
713
479k
  mi_x = above_mi_col << MI_SIZE_LOG2;
714
479k
  mi_y = xd->mi_row << MI_SIZE_LOG2;
715
716
479k
  const BLOCK_SIZE bsize = xd->mi[0]->bsize;
717
718
1.91M
  for (int j = 0; j < num_planes; ++j) {
719
1.43M
    const struct macroblockd_plane *pd = &xd->plane[j];
720
1.43M
    int bw = (op_mi_size * MI_SIZE) >> pd->subsampling_x;
721
1.43M
    int bh = clamp(block_size_high[bsize] >> (pd->subsampling_y + 1), 4,
722
1.43M
                   block_size_high[BLOCK_64X64] >> (pd->subsampling_y + 1));
723
724
1.43M
    if (av1_skip_u4x4_pred_in_obmc(bsize, pd, 0)) continue;
725
778k
    dec_build_inter_predictors(ctxt->cm, (DecoderCodingBlock *)ctxt->dcb, j,
726
778k
                               &backup_mbmi, 1, bw, bh, mi_x, mi_y);
727
778k
  }
728
479k
}
729
730
static inline void dec_build_prediction_by_above_preds(
731
    const AV1_COMMON *cm, DecoderCodingBlock *dcb,
732
    uint8_t *tmp_buf[MAX_MB_PLANE], int tmp_width[MAX_MB_PLANE],
733
480k
    int tmp_height[MAX_MB_PLANE], int tmp_stride[MAX_MB_PLANE]) {
734
480k
  MACROBLOCKD *const xd = &dcb->xd;
735
480k
  if (!xd->up_available) return;
736
737
  // Adjust mb_to_bottom_edge to have the correct value for the OBMC
738
  // prediction block. This is half the height of the original block,
739
  // except for 128-wide blocks, where we only use a height of 32.
740
472k
  const int this_height = xd->height * MI_SIZE;
741
472k
  const int pred_height = AOMMIN(this_height / 2, 32);
742
472k
  xd->mb_to_bottom_edge += GET_MV_SUBPEL(this_height - pred_height);
743
472k
  struct build_prediction_ctxt ctxt = {
744
472k
    cm, tmp_buf, tmp_width, tmp_height, tmp_stride, xd->mb_to_right_edge, dcb
745
472k
  };
746
472k
  const BLOCK_SIZE bsize = xd->mi[0]->bsize;
747
472k
  foreach_overlappable_nb_above(cm, xd,
748
472k
                                max_neighbor_obmc[mi_size_wide_log2[bsize]],
749
472k
                                dec_build_prediction_by_above_pred, &ctxt);
750
751
472k
  xd->mb_to_left_edge = -GET_MV_SUBPEL(xd->mi_col * MI_SIZE);
752
472k
  xd->mb_to_right_edge = ctxt.mb_to_far_edge;
753
472k
  xd->mb_to_bottom_edge -= GET_MV_SUBPEL(this_height - pred_height);
754
472k
}
755
756
static inline void dec_build_prediction_by_left_pred(
757
    MACROBLOCKD *const xd, int rel_mi_row, int rel_mi_col, uint8_t op_mi_size,
758
477k
    int dir, MB_MODE_INFO *left_mbmi, void *fun_ctxt, const int num_planes) {
759
477k
  struct build_prediction_ctxt *ctxt = (struct build_prediction_ctxt *)fun_ctxt;
760
477k
  const int left_mi_row = xd->mi_row + rel_mi_row;
761
477k
  int mi_x, mi_y;
762
477k
  MB_MODE_INFO backup_mbmi = *left_mbmi;
763
764
477k
  (void)rel_mi_col;
765
477k
  (void)dir;
766
767
477k
  av1_setup_build_prediction_by_left_pred(xd, rel_mi_row, op_mi_size,
768
477k
                                          &backup_mbmi, ctxt, num_planes);
769
477k
  mi_x = xd->mi_col << MI_SIZE_LOG2;
770
477k
  mi_y = left_mi_row << MI_SIZE_LOG2;
771
477k
  const BLOCK_SIZE bsize = xd->mi[0]->bsize;
772
773
1.90M
  for (int j = 0; j < num_planes; ++j) {
774
1.42M
    const struct macroblockd_plane *pd = &xd->plane[j];
775
1.42M
    int bw = clamp(block_size_wide[bsize] >> (pd->subsampling_x + 1), 4,
776
1.42M
                   block_size_wide[BLOCK_64X64] >> (pd->subsampling_x + 1));
777
1.42M
    int bh = (op_mi_size << MI_SIZE_LOG2) >> pd->subsampling_y;
778
779
1.42M
    if (av1_skip_u4x4_pred_in_obmc(bsize, pd, 1)) continue;
780
1.42M
    dec_build_inter_predictors(ctxt->cm, (DecoderCodingBlock *)ctxt->dcb, j,
781
1.42M
                               &backup_mbmi, 1, bw, bh, mi_x, mi_y);
782
1.42M
  }
783
477k
}
784
785
static inline void dec_build_prediction_by_left_preds(
786
    const AV1_COMMON *cm, DecoderCodingBlock *dcb,
787
    uint8_t *tmp_buf[MAX_MB_PLANE], int tmp_width[MAX_MB_PLANE],
788
480k
    int tmp_height[MAX_MB_PLANE], int tmp_stride[MAX_MB_PLANE]) {
789
480k
  MACROBLOCKD *const xd = &dcb->xd;
790
480k
  if (!xd->left_available) return;
791
792
  // Adjust mb_to_right_edge to have the correct value for the OBMC
793
  // prediction block. This is half the width of the original block,
794
  // except for 128-wide blocks, where we only use a width of 32.
795
475k
  const int this_width = xd->width * MI_SIZE;
796
475k
  const int pred_width = AOMMIN(this_width / 2, 32);
797
475k
  xd->mb_to_right_edge += GET_MV_SUBPEL(this_width - pred_width);
798
799
475k
  struct build_prediction_ctxt ctxt = {
800
475k
    cm, tmp_buf, tmp_width, tmp_height, tmp_stride, xd->mb_to_bottom_edge, dcb
801
475k
  };
802
475k
  const BLOCK_SIZE bsize = xd->mi[0]->bsize;
803
475k
  foreach_overlappable_nb_left(cm, xd,
804
475k
                               max_neighbor_obmc[mi_size_high_log2[bsize]],
805
475k
                               dec_build_prediction_by_left_pred, &ctxt);
806
807
475k
  xd->mb_to_top_edge = -GET_MV_SUBPEL(xd->mi_row * MI_SIZE);
808
475k
  xd->mb_to_right_edge -= GET_MV_SUBPEL(this_width - pred_width);
809
475k
  xd->mb_to_bottom_edge = ctxt.mb_to_far_edge;
810
475k
}
811
812
static inline void dec_build_obmc_inter_predictors_sb(const AV1_COMMON *cm,
813
480k
                                                      DecoderCodingBlock *dcb) {
814
480k
  const int num_planes = av1_num_planes(cm);
815
480k
  uint8_t *dst_buf1[MAX_MB_PLANE], *dst_buf2[MAX_MB_PLANE];
816
480k
  int dst_stride1[MAX_MB_PLANE] = { MAX_SB_SIZE, MAX_SB_SIZE, MAX_SB_SIZE };
817
480k
  int dst_stride2[MAX_MB_PLANE] = { MAX_SB_SIZE, MAX_SB_SIZE, MAX_SB_SIZE };
818
480k
  int dst_width1[MAX_MB_PLANE] = { MAX_SB_SIZE, MAX_SB_SIZE, MAX_SB_SIZE };
819
480k
  int dst_width2[MAX_MB_PLANE] = { MAX_SB_SIZE, MAX_SB_SIZE, MAX_SB_SIZE };
820
480k
  int dst_height1[MAX_MB_PLANE] = { MAX_SB_SIZE, MAX_SB_SIZE, MAX_SB_SIZE };
821
480k
  int dst_height2[MAX_MB_PLANE] = { MAX_SB_SIZE, MAX_SB_SIZE, MAX_SB_SIZE };
822
823
480k
  MACROBLOCKD *const xd = &dcb->xd;
824
480k
  av1_setup_obmc_dst_bufs(xd, dst_buf1, dst_buf2);
825
826
480k
  dec_build_prediction_by_above_preds(cm, dcb, dst_buf1, dst_width1,
827
480k
                                      dst_height1, dst_stride1);
828
480k
  dec_build_prediction_by_left_preds(cm, dcb, dst_buf2, dst_width2, dst_height2,
829
480k
                                     dst_stride2);
830
480k
  const int mi_row = xd->mi_row;
831
480k
  const int mi_col = xd->mi_col;
832
480k
  av1_setup_dst_planes(xd->plane, xd->mi[0]->bsize, &cm->cur_frame->buf, mi_row,
833
480k
                       mi_col, 0, num_planes);
834
480k
  av1_build_obmc_inter_prediction(cm, xd, dst_buf1, dst_stride1, dst_buf2,
835
480k
                                  dst_stride2);
836
480k
}
837
838
static inline void cfl_store_inter_block(AV1_COMMON *const cm,
839
3.49M
                                         MACROBLOCKD *const xd) {
840
3.49M
  MB_MODE_INFO *mbmi = xd->mi[0];
841
3.49M
  if (store_cfl_required(cm, xd)) {
842
628k
    cfl_store_block(xd, mbmi->bsize, mbmi->tx_size);
843
628k
  }
844
3.49M
}
845
846
static inline void predict_inter_block(AV1_COMMON *const cm,
847
                                       DecoderCodingBlock *dcb,
848
3.49M
                                       BLOCK_SIZE bsize) {
849
3.49M
  MACROBLOCKD *const xd = &dcb->xd;
850
3.49M
  MB_MODE_INFO *mbmi = xd->mi[0];
851
3.49M
  const int num_planes = av1_num_planes(cm);
852
3.49M
  const int mi_row = xd->mi_row;
853
3.49M
  const int mi_col = xd->mi_col;
854
7.52M
  for (int ref = 0; ref < 1 + has_second_ref(mbmi); ++ref) {
855
4.03M
    const MV_REFERENCE_FRAME frame = mbmi->ref_frame[ref];
856
4.03M
    if (frame < LAST_FRAME) {
857
52.9k
      assert(is_intrabc_block(mbmi));
858
52.9k
      assert(frame == INTRA_FRAME);
859
52.9k
      assert(ref == 0);
860
3.97M
    } else {
861
3.97M
      const RefCntBuffer *ref_buf = get_ref_frame_buf(cm, frame);
862
3.97M
      const struct scale_factors *ref_scale_factors =
863
3.97M
          get_ref_scale_factors_const(cm, frame);
864
865
3.97M
      xd->block_ref_scale_factors[ref] = ref_scale_factors;
866
3.97M
      av1_setup_pre_planes(xd, ref, &ref_buf->buf, mi_row, mi_col,
867
3.97M
                           ref_scale_factors, num_planes);
868
3.97M
    }
869
4.03M
  }
870
871
3.49M
  dec_build_inter_predictor(cm, dcb, mi_row, mi_col, bsize);
872
3.49M
  if (mbmi->motion_mode == OBMC_CAUSAL) {
873
480k
    dec_build_obmc_inter_predictors_sb(cm, dcb);
874
480k
  }
875
#if CONFIG_MISMATCH_DEBUG
876
  for (int plane = 0; plane < num_planes; ++plane) {
877
    const struct macroblockd_plane *pd = &xd->plane[plane];
878
    int pixel_c, pixel_r;
879
    mi_to_pixel_loc(&pixel_c, &pixel_r, mi_col, mi_row, 0, 0, pd->subsampling_x,
880
                    pd->subsampling_y);
881
    if (!is_chroma_reference(mi_row, mi_col, bsize, pd->subsampling_x,
882
                             pd->subsampling_y))
883
      continue;
884
    mismatch_check_block_pre(pd->dst.buf, pd->dst.stride,
885
                             cm->current_frame.order_hint, plane, pixel_c,
886
                             pixel_r, pd->width, pd->height,
887
                             xd->cur_buf->flags & YV12_FLAG_HIGHBITDEPTH);
888
  }
889
#endif
890
3.49M
}
891
892
static inline void set_color_index_map_offset(MACROBLOCKD *const xd, int plane,
893
133k
                                              aom_reader *r) {
894
133k
  (void)r;
895
133k
  Av1ColorMapParam params;
896
133k
  const MB_MODE_INFO *const mbmi = xd->mi[0];
897
133k
  av1_get_block_dimensions(mbmi->bsize, plane, xd, &params.plane_width,
898
133k
                           &params.plane_height, NULL, NULL);
899
133k
  xd->color_index_map_offset[plane] += params.plane_width * params.plane_height;
900
133k
}
901
902
static inline void decode_token_recon_block(AV1Decoder *const pbi,
903
                                            ThreadData *const td, aom_reader *r,
904
19.1M
                                            BLOCK_SIZE bsize) {
905
19.1M
  AV1_COMMON *const cm = &pbi->common;
906
19.1M
  DecoderCodingBlock *const dcb = &td->dcb;
907
19.1M
  MACROBLOCKD *const xd = &dcb->xd;
908
19.1M
  const int num_planes = av1_num_planes(cm);
909
19.1M
  MB_MODE_INFO *mbmi = xd->mi[0];
910
911
19.1M
  if (!is_inter_block(mbmi)) {
912
12.6M
    int row, col;
913
12.6M
    assert(bsize == get_plane_block_size(bsize, xd->plane[0].subsampling_x,
914
12.6M
                                         xd->plane[0].subsampling_y));
915
12.6M
    const int max_blocks_wide = max_block_wide(xd, bsize, 0);
916
12.6M
    const int max_blocks_high = max_block_high(xd, bsize, 0);
917
12.6M
    const BLOCK_SIZE max_unit_bsize = BLOCK_64X64;
918
12.6M
    int mu_blocks_wide = mi_size_wide[max_unit_bsize];
919
12.6M
    int mu_blocks_high = mi_size_high[max_unit_bsize];
920
12.6M
    mu_blocks_wide = AOMMIN(max_blocks_wide, mu_blocks_wide);
921
12.6M
    mu_blocks_high = AOMMIN(max_blocks_high, mu_blocks_high);
922
923
25.5M
    for (row = 0; row < max_blocks_high; row += mu_blocks_high) {
924
26.3M
      for (col = 0; col < max_blocks_wide; col += mu_blocks_wide) {
925
50.3M
        for (int plane = 0; plane < num_planes; ++plane) {
926
37.5M
          if (plane && !xd->is_chroma_ref) break;
927
36.9M
          const struct macroblockd_plane *const pd = &xd->plane[plane];
928
36.9M
          const TX_SIZE tx_size = av1_get_tx_size(plane, xd);
929
36.9M
          const int stepr = tx_size_high_unit[tx_size];
930
36.9M
          const int stepc = tx_size_wide_unit[tx_size];
931
932
36.9M
          const int unit_height = ROUND_POWER_OF_TWO(
933
36.9M
              AOMMIN(mu_blocks_high + row, max_blocks_high), pd->subsampling_y);
934
36.9M
          const int unit_width = ROUND_POWER_OF_TWO(
935
36.9M
              AOMMIN(mu_blocks_wide + col, max_blocks_wide), pd->subsampling_x);
936
937
85.5M
          for (int blk_row = row >> pd->subsampling_y; blk_row < unit_height;
938
48.6M
               blk_row += stepr) {
939
204M
            for (int blk_col = col >> pd->subsampling_x; blk_col < unit_width;
940
156M
                 blk_col += stepc) {
941
156M
              td->read_coeffs_tx_intra_block_visit(cm, dcb, r, plane, blk_row,
942
156M
                                                   blk_col, tx_size);
943
156M
              td->predict_and_recon_intra_block_visit(
944
156M
                  cm, dcb, r, plane, blk_row, blk_col, tx_size);
945
156M
              set_cb_buffer_offsets(dcb, tx_size, plane);
946
156M
            }
947
48.6M
          }
948
36.9M
        }
949
13.4M
      }
950
12.8M
    }
951
12.6M
  } else {
952
6.42M
    td->predict_inter_block_visit(cm, dcb, bsize);
953
    // Reconstruction
954
6.42M
    if (!mbmi->skip_txfm) {
955
4.56M
      int eobtotal = 0;
956
957
4.56M
      const int max_blocks_wide = max_block_wide(xd, bsize, 0);
958
4.56M
      const int max_blocks_high = max_block_high(xd, bsize, 0);
959
4.56M
      int row, col;
960
961
4.56M
      const BLOCK_SIZE max_unit_bsize = BLOCK_64X64;
962
4.56M
      assert(max_unit_bsize ==
963
4.56M
             get_plane_block_size(BLOCK_64X64, xd->plane[0].subsampling_x,
964
4.56M
                                  xd->plane[0].subsampling_y));
965
4.55M
      int mu_blocks_wide = mi_size_wide[max_unit_bsize];
966
4.55M
      int mu_blocks_high = mi_size_high[max_unit_bsize];
967
968
4.55M
      mu_blocks_wide = AOMMIN(max_blocks_wide, mu_blocks_wide);
969
4.55M
      mu_blocks_high = AOMMIN(max_blocks_high, mu_blocks_high);
970
971
9.13M
      for (row = 0; row < max_blocks_high; row += mu_blocks_high) {
972
9.21M
        for (col = 0; col < max_blocks_wide; col += mu_blocks_wide) {
973
16.7M
          for (int plane = 0; plane < num_planes; ++plane) {
974
12.9M
            if (plane && !xd->is_chroma_ref) break;
975
12.1M
            const struct macroblockd_plane *const pd = &xd->plane[plane];
976
12.1M
            const int ss_x = pd->subsampling_x;
977
12.1M
            const int ss_y = pd->subsampling_y;
978
12.1M
            const BLOCK_SIZE plane_bsize =
979
12.1M
                get_plane_block_size(bsize, ss_x, ss_y);
980
12.1M
            const TX_SIZE max_tx_size =
981
12.1M
                get_vartx_max_txsize(xd, plane_bsize, plane);
982
12.1M
            const int bh_var_tx = tx_size_high_unit[max_tx_size];
983
12.1M
            const int bw_var_tx = tx_size_wide_unit[max_tx_size];
984
12.1M
            int block = 0;
985
12.1M
            int step =
986
12.1M
                tx_size_wide_unit[max_tx_size] * tx_size_high_unit[max_tx_size];
987
12.1M
            int blk_row, blk_col;
988
12.1M
            const int unit_height = ROUND_POWER_OF_TWO(
989
12.1M
                AOMMIN(mu_blocks_high + row, max_blocks_high), ss_y);
990
12.1M
            const int unit_width = ROUND_POWER_OF_TWO(
991
12.1M
                AOMMIN(mu_blocks_wide + col, max_blocks_wide), ss_x);
992
993
24.3M
            for (blk_row = row >> ss_y; blk_row < unit_height;
994
12.2M
                 blk_row += bh_var_tx) {
995
25.4M
              for (blk_col = col >> ss_x; blk_col < unit_width;
996
13.2M
                   blk_col += bw_var_tx) {
997
13.2M
                decode_reconstruct_tx(cm, td, r, mbmi, plane, plane_bsize,
998
13.2M
                                      blk_row, blk_col, block, max_tx_size,
999
13.2M
                                      &eobtotal);
1000
13.2M
                block += step;
1001
13.2M
              }
1002
12.2M
            }
1003
12.1M
          }
1004
4.63M
        }
1005
4.58M
      }
1006
4.55M
    }
1007
6.42M
    td->cfl_store_inter_block_visit(cm, xd);
1008
6.42M
  }
1009
1010
19.1M
  av1_visit_palette(pbi, xd, r, set_color_index_map_offset);
1011
19.1M
}
1012
1013
static inline void set_inter_tx_size(MB_MODE_INFO *mbmi, int stride_log2,
1014
                                     int tx_w_log2, int tx_h_log2, int min_txs,
1015
                                     int split_size, int txs, int blk_row,
1016
1.17M
                                     int blk_col) {
1017
2.83M
  for (int idy = 0; idy < tx_size_high_unit[split_size];
1018
1.65M
       idy += tx_size_high_unit[min_txs]) {
1019
4.19M
    for (int idx = 0; idx < tx_size_wide_unit[split_size];
1020
2.54M
         idx += tx_size_wide_unit[min_txs]) {
1021
2.54M
      const int index = (((blk_row + idy) >> tx_h_log2) << stride_log2) +
1022
2.54M
                        ((blk_col + idx) >> tx_w_log2);
1023
2.54M
      mbmi->inter_tx_size[index] = txs;
1024
2.54M
    }
1025
1.65M
  }
1026
1.17M
}
1027
1028
static inline void read_tx_size_vartx(MACROBLOCKD *xd, MB_MODE_INFO *mbmi,
1029
                                      TX_SIZE tx_size, int depth, int blk_row,
1030
1.36M
                                      int blk_col, aom_reader *r) {
1031
1.36M
  FRAME_CONTEXT *ec_ctx = xd->tile_ctx;
1032
1.36M
  int is_split = 0;
1033
1.36M
  const BLOCK_SIZE bsize = mbmi->bsize;
1034
1.36M
  const int max_blocks_high = max_block_high(xd, bsize, 0);
1035
1.36M
  const int max_blocks_wide = max_block_wide(xd, bsize, 0);
1036
1.36M
  if (blk_row >= max_blocks_high || blk_col >= max_blocks_wide) return;
1037
1.36M
  assert(tx_size > TX_4X4);
1038
1.36M
  TX_SIZE txs = max_txsize_rect_lookup[bsize];
1039
2.72M
  for (int level = 0; level < MAX_VARTX_DEPTH - 1; ++level)
1040
1.36M
    txs = sub_tx_size_map[txs];
1041
1.36M
  const int tx_w_log2 = tx_size_wide_log2[txs] - MI_SIZE_LOG2;
1042
1.36M
  const int tx_h_log2 = tx_size_high_log2[txs] - MI_SIZE_LOG2;
1043
1.36M
  const int bw_log2 = mi_size_wide_log2[bsize];
1044
1.36M
  const int stride_log2 = bw_log2 - tx_w_log2;
1045
1046
1.36M
  if (depth == MAX_VARTX_DEPTH) {
1047
142k
    set_inter_tx_size(mbmi, stride_log2, tx_w_log2, tx_h_log2, txs, tx_size,
1048
142k
                      tx_size, blk_row, blk_col);
1049
142k
    mbmi->tx_size = tx_size;
1050
142k
    txfm_partition_update(xd->above_txfm_context + blk_col,
1051
142k
                          xd->left_txfm_context + blk_row, tx_size, tx_size);
1052
142k
    return;
1053
142k
  }
1054
1055
1.22M
  const int ctx = txfm_partition_context(xd->above_txfm_context + blk_col,
1056
1.22M
                                         xd->left_txfm_context + blk_row,
1057
1.22M
                                         mbmi->bsize, tx_size);
1058
1.22M
  is_split = aom_read_symbol(r, ec_ctx->txfm_partition_cdf[ctx], 2, ACCT_STR);
1059
1060
1.22M
  if (is_split) {
1061
308k
    const TX_SIZE sub_txs = sub_tx_size_map[tx_size];
1062
308k
    const int bsw = tx_size_wide_unit[sub_txs];
1063
308k
    const int bsh = tx_size_high_unit[sub_txs];
1064
1065
308k
    if (sub_txs == TX_4X4) {
1066
121k
      set_inter_tx_size(mbmi, stride_log2, tx_w_log2, tx_h_log2, txs, tx_size,
1067
121k
                        sub_txs, blk_row, blk_col);
1068
121k
      mbmi->tx_size = sub_txs;
1069
121k
      txfm_partition_update(xd->above_txfm_context + blk_col,
1070
121k
                            xd->left_txfm_context + blk_row, sub_txs, tx_size);
1071
121k
      return;
1072
121k
    }
1073
1074
186k
    assert(bsw > 0 && bsh > 0);
1075
500k
    for (int row = 0; row < tx_size_high_unit[tx_size]; row += bsh) {
1076
828k
      for (int col = 0; col < tx_size_wide_unit[tx_size]; col += bsw) {
1077
514k
        int offsetr = blk_row + row;
1078
514k
        int offsetc = blk_col + col;
1079
514k
        read_tx_size_vartx(xd, mbmi, sub_txs, depth + 1, offsetr, offsetc, r);
1080
514k
      }
1081
313k
    }
1082
913k
  } else {
1083
913k
    set_inter_tx_size(mbmi, stride_log2, tx_w_log2, tx_h_log2, txs, tx_size,
1084
913k
                      tx_size, blk_row, blk_col);
1085
913k
    mbmi->tx_size = tx_size;
1086
913k
    txfm_partition_update(xd->above_txfm_context + blk_col,
1087
913k
                          xd->left_txfm_context + blk_row, tx_size, tx_size);
1088
913k
  }
1089
1.22M
}
1090
1091
static TX_SIZE read_selected_tx_size(const MACROBLOCKD *const xd,
1092
1.48M
                                     aom_reader *r) {
1093
  // TODO(debargha): Clean up the logic here. This function should only
1094
  // be called for intra.
1095
1.48M
  const BLOCK_SIZE bsize = xd->mi[0]->bsize;
1096
1.48M
  const int32_t tx_size_cat = bsize_to_tx_size_cat(bsize);
1097
1.48M
  const int max_depths = bsize_to_max_depth(bsize);
1098
1.48M
  const int ctx = get_tx_size_context(xd);
1099
1.48M
  FRAME_CONTEXT *ec_ctx = xd->tile_ctx;
1100
1.48M
  const int depth = aom_read_symbol(r, ec_ctx->tx_size_cdf[tx_size_cat][ctx],
1101
1.48M
                                    max_depths + 1, ACCT_STR);
1102
1.48M
  assert(depth >= 0 && depth <= max_depths);
1103
1.48M
  const TX_SIZE tx_size = depth_to_tx_size(depth, bsize);
1104
1.48M
  return tx_size;
1105
1.48M
}
1106
1107
static TX_SIZE read_tx_size(const MACROBLOCKD *const xd, TX_MODE tx_mode,
1108
                            int is_inter, int allow_select_inter,
1109
12.7M
                            aom_reader *r) {
1110
12.7M
  const BLOCK_SIZE bsize = xd->mi[0]->bsize;
1111
12.7M
  if (xd->lossless[xd->mi[0]->segment_id]) return TX_4X4;
1112
1113
12.6M
  if (block_signals_txsize(bsize)) {
1114
11.9M
    if ((!is_inter || allow_select_inter) && tx_mode == TX_MODE_SELECT) {
1115
1.48M
      const TX_SIZE coded_tx_size = read_selected_tx_size(xd, r);
1116
1.48M
      return coded_tx_size;
1117
10.4M
    } else {
1118
10.4M
      return tx_size_from_tx_mode(bsize, tx_mode);
1119
10.4M
    }
1120
11.9M
  } else {
1121
694k
    assert(IMPLIES(tx_mode == ONLY_4X4, bsize == BLOCK_4X4));
1122
695k
    return max_txsize_rect_lookup[bsize];
1123
694k
  }
1124
12.6M
}
1125
1126
static inline void parse_decode_block(AV1Decoder *const pbi,
1127
                                      ThreadData *const td, int mi_row,
1128
                                      int mi_col, aom_reader *r,
1129
                                      PARTITION_TYPE partition,
1130
13.5M
                                      BLOCK_SIZE bsize) {
1131
13.5M
  DecoderCodingBlock *const dcb = &td->dcb;
1132
13.5M
  MACROBLOCKD *const xd = &dcb->xd;
1133
13.5M
  decode_mbmi_block(pbi, dcb, mi_row, mi_col, r, partition, bsize);
1134
1135
13.5M
  av1_visit_palette(pbi, xd, r, av1_decode_palette_tokens);
1136
1137
13.5M
  AV1_COMMON *cm = &pbi->common;
1138
13.5M
  const int num_planes = av1_num_planes(cm);
1139
13.5M
  MB_MODE_INFO *mbmi = xd->mi[0];
1140
13.5M
  int inter_block_tx = is_inter_block(mbmi) || is_intrabc_block(mbmi);
1141
13.5M
  if (cm->features.tx_mode == TX_MODE_SELECT && block_signals_txsize(bsize) &&
1142
13.5M
      !mbmi->skip_txfm && inter_block_tx && !xd->lossless[mbmi->segment_id]) {
1143
819k
    const TX_SIZE max_tx_size = max_txsize_rect_lookup[bsize];
1144
819k
    const int bh = tx_size_high_unit[max_tx_size];
1145
819k
    const int bw = tx_size_wide_unit[max_tx_size];
1146
819k
    const int width = mi_size_wide[bsize];
1147
819k
    const int height = mi_size_high[bsize];
1148
1149
1.64M
    for (int idy = 0; idy < height; idy += bh)
1150
1.68M
      for (int idx = 0; idx < width; idx += bw)
1151
851k
        read_tx_size_vartx(xd, mbmi, max_tx_size, 0, idy, idx, r);
1152
12.7M
  } else {
1153
12.7M
    mbmi->tx_size = read_tx_size(xd, cm->features.tx_mode, inter_block_tx,
1154
12.7M
                                 !mbmi->skip_txfm, r);
1155
12.7M
    if (inter_block_tx)
1156
3.05M
      memset(mbmi->inter_tx_size, mbmi->tx_size, sizeof(mbmi->inter_tx_size));
1157
12.7M
    set_txfm_ctxs(mbmi->tx_size, xd->width, xd->height,
1158
12.7M
                  mbmi->skip_txfm && is_inter_block(mbmi), xd);
1159
12.7M
  }
1160
1161
13.5M
  if (cm->delta_q_info.delta_q_present_flag) {
1162
22.0M
    for (int i = 0; i < MAX_SEGMENTS; i++) {
1163
19.5M
      const int current_qindex =
1164
19.5M
          av1_get_qindex(&cm->seg, i, xd->current_base_qindex);
1165
19.5M
      const CommonQuantParams *const quant_params = &cm->quant_params;
1166
73.1M
      for (int j = 0; j < num_planes; ++j) {
1167
53.5M
        const int dc_delta_q = j == 0 ? quant_params->y_dc_delta_q
1168
53.5M
                                      : (j == 1 ? quant_params->u_dc_delta_q
1169
34.0M
                                                : quant_params->v_dc_delta_q);
1170
53.5M
        const int ac_delta_q = j == 0 ? 0
1171
53.5M
                                      : (j == 1 ? quant_params->u_ac_delta_q
1172
34.0M
                                                : quant_params->v_ac_delta_q);
1173
53.5M
        xd->plane[j].seg_dequant_QTX[i][0] = av1_dc_quant_QTX(
1174
53.5M
            current_qindex, dc_delta_q, cm->seq_params->bit_depth);
1175
53.5M
        xd->plane[j].seg_dequant_QTX[i][1] = av1_ac_quant_QTX(
1176
53.5M
            current_qindex, ac_delta_q, cm->seq_params->bit_depth);
1177
53.5M
      }
1178
19.5M
    }
1179
2.44M
  }
1180
13.5M
  if (mbmi->skip_txfm) av1_reset_entropy_context(xd, bsize, num_planes);
1181
1182
13.5M
  decode_token_recon_block(pbi, td, r, bsize);
1183
13.5M
}
1184
1185
static inline void set_offsets_for_pred_and_recon(AV1Decoder *const pbi,
1186
                                                  ThreadData *const td,
1187
                                                  int mi_row, int mi_col,
1188
5.55M
                                                  BLOCK_SIZE bsize) {
1189
5.55M
  AV1_COMMON *const cm = &pbi->common;
1190
5.55M
  const CommonModeInfoParams *const mi_params = &cm->mi_params;
1191
5.55M
  DecoderCodingBlock *const dcb = &td->dcb;
1192
5.55M
  MACROBLOCKD *const xd = &dcb->xd;
1193
5.55M
  const int bw = mi_size_wide[bsize];
1194
5.55M
  const int bh = mi_size_high[bsize];
1195
5.55M
  const int num_planes = av1_num_planes(cm);
1196
1197
5.55M
  const int offset = mi_row * mi_params->mi_stride + mi_col;
1198
5.55M
  const TileInfo *const tile = &xd->tile;
1199
1200
5.55M
  xd->mi = mi_params->mi_grid_base + offset;
1201
5.55M
  xd->tx_type_map =
1202
5.55M
      &mi_params->tx_type_map[mi_row * mi_params->mi_stride + mi_col];
1203
5.55M
  xd->tx_type_map_stride = mi_params->mi_stride;
1204
1205
5.55M
  set_plane_n4(xd, bw, bh, num_planes);
1206
1207
  // Distance of Mb to the various image edges. These are specified to 8th pel
1208
  // as they are always compared to values that are in 1/8th pel units
1209
5.55M
  set_mi_row_col(xd, tile, mi_row, bh, mi_col, bw, mi_params->mi_rows,
1210
5.55M
                 mi_params->mi_cols);
1211
1212
5.55M
  av1_setup_dst_planes(xd->plane, bsize, &cm->cur_frame->buf, mi_row, mi_col, 0,
1213
5.55M
                       num_planes);
1214
5.55M
}
1215
1216
static inline void decode_block(AV1Decoder *const pbi, ThreadData *const td,
1217
                                int mi_row, int mi_col, aom_reader *r,
1218
5.55M
                                PARTITION_TYPE partition, BLOCK_SIZE bsize) {
1219
5.55M
  (void)partition;
1220
5.55M
  set_offsets_for_pred_and_recon(pbi, td, mi_row, mi_col, bsize);
1221
5.55M
  decode_token_recon_block(pbi, td, r, bsize);
1222
5.55M
}
1223
1224
static PARTITION_TYPE read_partition(MACROBLOCKD *xd, int mi_row, int mi_col,
1225
                                     aom_reader *r, int has_rows, int has_cols,
1226
9.15M
                                     BLOCK_SIZE bsize) {
1227
9.15M
  const int ctx = partition_plane_context(xd, mi_row, mi_col, bsize);
1228
9.15M
  FRAME_CONTEXT *ec_ctx = xd->tile_ctx;
1229
1230
9.15M
  if (!has_rows && !has_cols) return PARTITION_SPLIT;
1231
1232
9.05M
  assert(ctx >= 0);
1233
9.05M
  aom_cdf_prob *partition_cdf = ec_ctx->partition_cdf[ctx];
1234
9.05M
  if (has_rows && has_cols) {
1235
8.73M
    return (PARTITION_TYPE)aom_read_symbol(
1236
8.73M
        r, partition_cdf, partition_cdf_length(bsize), ACCT_STR);
1237
8.73M
  } else if (!has_rows && has_cols) {
1238
127k
    assert(bsize > BLOCK_8X8);
1239
127k
    aom_cdf_prob cdf[2];
1240
127k
    partition_gather_vert_alike(cdf, partition_cdf, bsize);
1241
127k
    assert(cdf[1] == AOM_ICDF(CDF_PROB_TOP));
1242
127k
    return aom_read_cdf(r, cdf, 2, ACCT_STR) ? PARTITION_SPLIT : PARTITION_HORZ;
1243
194k
  } else {
1244
194k
    assert(has_rows && !has_cols);
1245
195k
    assert(bsize > BLOCK_8X8);
1246
195k
    aom_cdf_prob cdf[2];
1247
195k
    partition_gather_horz_alike(cdf, partition_cdf, bsize);
1248
195k
    assert(cdf[1] == AOM_ICDF(CDF_PROB_TOP));
1249
195k
    return aom_read_cdf(r, cdf, 2, ACCT_STR) ? PARTITION_SPLIT : PARTITION_VERT;
1250
195k
  }
1251
9.05M
}
1252
1253
// TODO(slavarnway): eliminate bsize and subsize in future commits
1254
static inline void decode_partition(AV1Decoder *const pbi, ThreadData *const td,
1255
                                    int mi_row, int mi_col, aom_reader *reader,
1256
14.6M
                                    BLOCK_SIZE bsize, int parse_decode_flag) {
1257
14.6M
  assert(bsize < BLOCK_SIZES_ALL);
1258
14.6M
  AV1_COMMON *const cm = &pbi->common;
1259
14.6M
  DecoderCodingBlock *const dcb = &td->dcb;
1260
14.6M
  MACROBLOCKD *const xd = &dcb->xd;
1261
14.6M
  const int bw = mi_size_wide[bsize];
1262
14.6M
  const int hbs = bw >> 1;
1263
14.6M
  PARTITION_TYPE partition;
1264
14.6M
  BLOCK_SIZE subsize;
1265
14.6M
  const int quarter_step = bw / 4;
1266
14.6M
  BLOCK_SIZE bsize2 = get_partition_subsize(bsize, PARTITION_SPLIT);
1267
14.6M
  const int has_rows = (mi_row + hbs) < cm->mi_params.mi_rows;
1268
14.6M
  const int has_cols = (mi_col + hbs) < cm->mi_params.mi_cols;
1269
1270
14.6M
  if (mi_row >= cm->mi_params.mi_rows || mi_col >= cm->mi_params.mi_cols)
1271
722k
    return;
1272
1273
  // parse_decode_flag takes the following values :
1274
  // 01 - do parse only
1275
  // 10 - do decode only
1276
  // 11 - do parse and decode
1277
13.8M
  static const block_visitor_fn_t block_visit[4] = { NULL, parse_decode_block,
1278
13.8M
                                                     decode_block,
1279
13.8M
                                                     parse_decode_block };
1280
1281
13.8M
  if (parse_decode_flag & 1) {
1282
9.85M
    const int num_planes = av1_num_planes(cm);
1283
38.4M
    for (int plane = 0; plane < num_planes; ++plane) {
1284
28.5M
      int rcol0, rcol1, rrow0, rrow1;
1285
1286
      // Skip some unnecessary work if loop restoration is disabled
1287
28.5M
      if (cm->rst_info[plane].frame_restoration_type == RESTORE_NONE) continue;
1288
1289
13.7M
      if (av1_loop_restoration_corners_in_sb(cm, plane, mi_row, mi_col, bsize,
1290
13.7M
                                             &rcol0, &rcol1, &rrow0, &rrow1)) {
1291
251k
        const int rstride = cm->rst_info[plane].horz_units;
1292
502k
        for (int rrow = rrow0; rrow < rrow1; ++rrow) {
1293
509k
          for (int rcol = rcol0; rcol < rcol1; ++rcol) {
1294
258k
            const int runit_idx = rcol + rrow * rstride;
1295
258k
            loop_restoration_read_sb_coeffs(cm, xd, reader, plane, runit_idx);
1296
258k
          }
1297
251k
        }
1298
251k
      }
1299
13.7M
    }
1300
1301
9.85M
    partition = (bsize < BLOCK_8X8) ? PARTITION_NONE
1302
9.85M
                                    : read_partition(xd, mi_row, mi_col, reader,
1303
9.15M
                                                     has_rows, has_cols, bsize);
1304
9.85M
  } else {
1305
4.02M
    partition = get_partition(cm, mi_row, mi_col, bsize);
1306
4.02M
  }
1307
13.8M
  subsize = get_partition_subsize(bsize, partition);
1308
13.8M
  if (subsize == BLOCK_INVALID) {
1309
    // When an internal error occurs ensure that xd->mi_row is set appropriately
1310
    // w.r.t. current tile, which is used to signal processing of current row is
1311
    // done.
1312
0
    xd->mi_row = mi_row;
1313
0
    aom_internal_error(xd->error_info, AOM_CODEC_CORRUPT_FRAME,
1314
0
                       "Partition is invalid for block size %dx%d",
1315
0
                       block_size_wide[bsize], block_size_high[bsize]);
1316
0
  }
1317
  // Check the bitstream is conformant: if there is subsampling on the
1318
  // chroma planes, subsize must subsample to a valid block size.
1319
13.8M
  const struct macroblockd_plane *const pd_u = &xd->plane[1];
1320
13.8M
  if (get_plane_block_size(subsize, pd_u->subsampling_x, pd_u->subsampling_y) ==
1321
13.8M
      BLOCK_INVALID) {
1322
    // When an internal error occurs ensure that xd->mi_row is set appropriately
1323
    // w.r.t. current tile, which is used to signal processing of current row is
1324
    // done.
1325
889
    xd->mi_row = mi_row;
1326
889
    aom_internal_error(xd->error_info, AOM_CODEC_CORRUPT_FRAME,
1327
889
                       "Block size %dx%d invalid with this subsampling mode",
1328
889
                       block_size_wide[subsize], block_size_high[subsize]);
1329
889
  }
1330
1331
13.8M
#define DEC_BLOCK_STX_ARG
1332
19.1M
#define DEC_BLOCK_EPT_ARG partition,
1333
13.8M
#define DEC_BLOCK(db_r, db_c, db_subsize)                                  \
1334
19.1M
  block_visit[parse_decode_flag](pbi, td, DEC_BLOCK_STX_ARG(db_r), (db_c), \
1335
19.1M
                                 reader, DEC_BLOCK_EPT_ARG(db_subsize))
1336
13.8M
#define DEC_PARTITION(db_r, db_c, db_subsize)                        \
1337
13.8M
  decode_partition(pbi, td, DEC_BLOCK_STX_ARG(db_r), (db_c), reader, \
1338
13.1M
                   (db_subsize), parse_decode_flag)
1339
1340
13.8M
  switch (partition) {
1341
5.62M
    case PARTITION_NONE: DEC_BLOCK(mi_row, mi_col, subsize); break;
1342
1.55M
    case PARTITION_HORZ:
1343
1.55M
      DEC_BLOCK(mi_row, mi_col, subsize);
1344
1.55M
      if (has_rows) DEC_BLOCK(mi_row + hbs, mi_col, subsize);
1345
1.55M
      break;
1346
1.03M
    case PARTITION_VERT:
1347
1.03M
      DEC_BLOCK(mi_row, mi_col, subsize);
1348
1.03M
      if (has_cols) DEC_BLOCK(mi_row, mi_col + hbs, subsize);
1349
1.03M
      break;
1350
3.28M
    case PARTITION_SPLIT:
1351
3.28M
      DEC_PARTITION(mi_row, mi_col, subsize);
1352
3.28M
      DEC_PARTITION(mi_row, mi_col + hbs, subsize);
1353
3.28M
      DEC_PARTITION(mi_row + hbs, mi_col, subsize);
1354
3.28M
      DEC_PARTITION(mi_row + hbs, mi_col + hbs, subsize);
1355
3.28M
      break;
1356
283k
    case PARTITION_HORZ_A:
1357
283k
      DEC_BLOCK(mi_row, mi_col, bsize2);
1358
283k
      DEC_BLOCK(mi_row, mi_col + hbs, bsize2);
1359
283k
      DEC_BLOCK(mi_row + hbs, mi_col, subsize);
1360
283k
      break;
1361
285k
    case PARTITION_HORZ_B:
1362
285k
      DEC_BLOCK(mi_row, mi_col, subsize);
1363
285k
      DEC_BLOCK(mi_row + hbs, mi_col, bsize2);
1364
285k
      DEC_BLOCK(mi_row + hbs, mi_col + hbs, bsize2);
1365
285k
      break;
1366
193k
    case PARTITION_VERT_A:
1367
193k
      DEC_BLOCK(mi_row, mi_col, bsize2);
1368
193k
      DEC_BLOCK(mi_row + hbs, mi_col, bsize2);
1369
193k
      DEC_BLOCK(mi_row, mi_col + hbs, subsize);
1370
193k
      break;
1371
208k
    case PARTITION_VERT_B:
1372
208k
      DEC_BLOCK(mi_row, mi_col, subsize);
1373
208k
      DEC_BLOCK(mi_row, mi_col + hbs, bsize2);
1374
208k
      DEC_BLOCK(mi_row + hbs, mi_col + hbs, bsize2);
1375
208k
      break;
1376
864k
    case PARTITION_HORZ_4:
1377
4.31M
      for (int i = 0; i < 4; ++i) {
1378
3.45M
        int this_mi_row = mi_row + i * quarter_step;
1379
3.45M
        if (i > 0 && this_mi_row >= cm->mi_params.mi_rows) break;
1380
3.44M
        DEC_BLOCK(this_mi_row, mi_col, subsize);
1381
3.44M
      }
1382
864k
      break;
1383
564k
    case PARTITION_VERT_4:
1384
2.81M
      for (int i = 0; i < 4; ++i) {
1385
2.25M
        int this_mi_col = mi_col + i * quarter_step;
1386
2.25M
        if (i > 0 && this_mi_col >= cm->mi_params.mi_cols) break;
1387
2.25M
        DEC_BLOCK(mi_row, this_mi_col, subsize);
1388
2.25M
      }
1389
564k
      break;
1390
0
    default: assert(0 && "Invalid partition type");
1391
13.8M
  }
1392
1393
13.8M
#undef DEC_PARTITION
1394
13.8M
#undef DEC_BLOCK
1395
13.8M
#undef DEC_BLOCK_EPT_ARG
1396
13.8M
#undef DEC_BLOCK_STX_ARG
1397
1398
13.8M
  if (parse_decode_flag & 1)
1399
9.82M
    update_ext_partition_context(xd, mi_row, mi_col, subsize, bsize, partition);
1400
13.8M
}
1401
1402
static inline void setup_bool_decoder(
1403
    MACROBLOCKD *const xd, const uint8_t *data, const uint8_t *data_end,
1404
    const size_t read_size, struct aom_internal_error_info *error_info,
1405
165k
    aom_reader *r, uint8_t allow_update_cdf) {
1406
  // Validate the calculated partition length. If the buffer
1407
  // described by the partition can't be fully read, then restrict
1408
  // it to the portion that can be (for EC mode) or throw an error.
1409
165k
  if (!read_is_valid(data, read_size, data_end)) {
1410
    // When internal error occurs ensure that xd->mi_row is set appropriately
1411
    // w.r.t. current tile, which is used to signal processing of current row is
1412
    // done in row-mt decoding.
1413
438
    xd->mi_row = xd->tile.mi_row_start;
1414
1415
438
    aom_internal_error(error_info, AOM_CODEC_CORRUPT_FRAME,
1416
438
                       "Truncated packet or corrupt tile length");
1417
438
  }
1418
165k
  if (aom_reader_init(r, data, read_size)) {
1419
    // When internal error occurs ensure that xd->mi_row is set appropriately
1420
    // w.r.t. current tile, which is used to signal processing of current row is
1421
    // done in row-mt decoding.
1422
0
    xd->mi_row = xd->tile.mi_row_start;
1423
1424
0
    aom_internal_error(error_info, AOM_CODEC_MEM_ERROR,
1425
0
                       "Failed to allocate bool decoder %d", 1);
1426
0
  }
1427
1428
165k
  r->allow_update_cdf = allow_update_cdf;
1429
165k
}
1430
1431
static inline void setup_segmentation(AV1_COMMON *const cm,
1432
172k
                                      struct aom_read_bit_buffer *rb) {
1433
172k
  struct segmentation *const seg = &cm->seg;
1434
1435
172k
  seg->update_map = 0;
1436
172k
  seg->update_data = 0;
1437
172k
  seg->temporal_update = 0;
1438
1439
172k
  seg->enabled = aom_rb_read_bit(rb);
1440
172k
  if (!seg->enabled) {
1441
146k
    if (cm->cur_frame->seg_map) {
1442
146k
      memset(cm->cur_frame->seg_map, 0,
1443
146k
             (cm->cur_frame->mi_rows * cm->cur_frame->mi_cols));
1444
146k
    }
1445
1446
146k
    memset(seg, 0, sizeof(*seg));
1447
146k
    segfeatures_copy(&cm->cur_frame->seg, seg);
1448
146k
    return;
1449
146k
  }
1450
26.7k
  if (cm->seg.enabled && cm->prev_frame &&
1451
26.7k
      (cm->mi_params.mi_rows == cm->prev_frame->mi_rows) &&
1452
26.7k
      (cm->mi_params.mi_cols == cm->prev_frame->mi_cols)) {
1453
9.86k
    cm->last_frame_seg_map = cm->prev_frame->seg_map;
1454
16.8k
  } else {
1455
16.8k
    cm->last_frame_seg_map = NULL;
1456
16.8k
  }
1457
  // Read update flags
1458
26.7k
  if (cm->features.primary_ref_frame == PRIMARY_REF_NONE) {
1459
    // These frames can't use previous frames, so must signal map + features
1460
14.8k
    seg->update_map = 1;
1461
14.8k
    seg->temporal_update = 0;
1462
14.8k
    seg->update_data = 1;
1463
14.8k
  } else {
1464
11.9k
    seg->update_map = aom_rb_read_bit(rb);
1465
11.9k
    if (seg->update_map) {
1466
7.17k
      seg->temporal_update = aom_rb_read_bit(rb);
1467
7.17k
    } else {
1468
4.75k
      seg->temporal_update = 0;
1469
4.75k
    }
1470
11.9k
    seg->update_data = aom_rb_read_bit(rb);
1471
11.9k
  }
1472
1473
  // Segmentation data update
1474
26.7k
  if (seg->update_data) {
1475
19.0k
    av1_clearall_segfeatures(seg);
1476
1477
167k
    for (int i = 0; i < MAX_SEGMENTS; i++) {
1478
1.33M
      for (int j = 0; j < SEG_LVL_MAX; j++) {
1479
1.18M
        int data = 0;
1480
1.18M
        const int feature_enabled = aom_rb_read_bit(rb);
1481
1.18M
        if (feature_enabled) {
1482
307k
          av1_enable_segfeature(seg, i, j);
1483
1484
307k
          const int data_max = av1_seg_feature_data_max(j);
1485
307k
          const int data_min = -data_max;
1486
307k
          const int ubits = get_unsigned_bits(data_max);
1487
1488
307k
          if (av1_is_segfeature_signed(j)) {
1489
183k
            data = aom_rb_read_inv_signed_literal(rb, ubits);
1490
183k
          } else {
1491
123k
            data = aom_rb_read_literal(rb, ubits);
1492
123k
          }
1493
1494
307k
          data = clamp(data, data_min, data_max);
1495
307k
        }
1496
1.18M
        av1_set_segdata(seg, i, j, data);
1497
1.18M
      }
1498
148k
    }
1499
19.0k
    av1_calculate_segdata(seg);
1500
19.0k
  } else if (cm->prev_frame) {
1501
7.72k
    segfeatures_copy(seg, &cm->prev_frame->seg);
1502
7.72k
  }
1503
26.7k
  segfeatures_copy(&cm->cur_frame->seg, seg);
1504
26.7k
}
1505
1506
static inline void decode_restoration_mode(AV1_COMMON *cm,
1507
83.3k
                                           struct aom_read_bit_buffer *rb) {
1508
83.3k
  assert(!cm->features.all_lossless);
1509
83.3k
  const int num_planes = av1_num_planes(cm);
1510
83.3k
  if (cm->features.allow_intrabc) return;
1511
74.5k
  int all_none = 1, chroma_none = 1;
1512
283k
  for (int p = 0; p < num_planes; ++p) {
1513
208k
    RestorationInfo *rsi = &cm->rst_info[p];
1514
208k
    if (aom_rb_read_bit(rb)) {
1515
58.2k
      rsi->frame_restoration_type =
1516
58.2k
          aom_rb_read_bit(rb) ? RESTORE_SGRPROJ : RESTORE_WIENER;
1517
150k
    } else {
1518
150k
      rsi->frame_restoration_type =
1519
150k
          aom_rb_read_bit(rb) ? RESTORE_SWITCHABLE : RESTORE_NONE;
1520
150k
    }
1521
208k
    if (rsi->frame_restoration_type != RESTORE_NONE) {
1522
71.3k
      all_none = 0;
1523
71.3k
      chroma_none &= p == 0;
1524
71.3k
    }
1525
208k
  }
1526
74.5k
  if (!all_none) {
1527
36.2k
    assert(cm->seq_params->sb_size == BLOCK_64X64 ||
1528
36.2k
           cm->seq_params->sb_size == BLOCK_128X128);
1529
36.2k
    const int sb_size = cm->seq_params->sb_size == BLOCK_128X128 ? 128 : 64;
1530
1531
134k
    for (int p = 0; p < num_planes; ++p)
1532
97.7k
      cm->rst_info[p].restoration_unit_size = sb_size;
1533
1534
36.2k
    RestorationInfo *rsi = &cm->rst_info[0];
1535
1536
36.2k
    if (sb_size == 64) {
1537
6.45k
      rsi->restoration_unit_size <<= aom_rb_read_bit(rb);
1538
6.45k
    }
1539
36.2k
    if (rsi->restoration_unit_size > 64) {
1540
32.8k
      rsi->restoration_unit_size <<= aom_rb_read_bit(rb);
1541
32.8k
    }
1542
38.3k
  } else {
1543
38.3k
    const int size = RESTORATION_UNITSIZE_MAX;
1544
148k
    for (int p = 0; p < num_planes; ++p)
1545
110k
      cm->rst_info[p].restoration_unit_size = size;
1546
38.3k
  }
1547
1548
74.5k
  if (num_planes > 1) {
1549
66.8k
    int s =
1550
66.8k
        AOMMIN(cm->seq_params->subsampling_x, cm->seq_params->subsampling_y);
1551
66.8k
    if (s && !chroma_none) {
1552
20.4k
      cm->rst_info[1].restoration_unit_size =
1553
20.4k
          cm->rst_info[0].restoration_unit_size >> (aom_rb_read_bit(rb) * s);
1554
46.3k
    } else {
1555
46.3k
      cm->rst_info[1].restoration_unit_size =
1556
46.3k
          cm->rst_info[0].restoration_unit_size;
1557
46.3k
    }
1558
66.8k
    cm->rst_info[2].restoration_unit_size =
1559
66.8k
        cm->rst_info[1].restoration_unit_size;
1560
66.8k
  }
1561
74.5k
}
1562
1563
static inline void read_wiener_filter(int wiener_win, WienerInfo *wiener_info,
1564
                                      WienerInfo *ref_wiener_info,
1565
42.3k
                                      aom_reader *rb) {
1566
42.3k
  memset(wiener_info->vfilter, 0, sizeof(wiener_info->vfilter));
1567
42.3k
  memset(wiener_info->hfilter, 0, sizeof(wiener_info->hfilter));
1568
1569
42.3k
  if (wiener_win == WIENER_WIN)
1570
21.2k
    wiener_info->vfilter[0] = wiener_info->vfilter[WIENER_WIN - 1] =
1571
21.2k
        aom_read_primitive_refsubexpfin(
1572
21.2k
            rb, WIENER_FILT_TAP0_MAXV - WIENER_FILT_TAP0_MINV + 1,
1573
21.2k
            WIENER_FILT_TAP0_SUBEXP_K,
1574
21.2k
            ref_wiener_info->vfilter[0] - WIENER_FILT_TAP0_MINV, ACCT_STR) +
1575
21.2k
        WIENER_FILT_TAP0_MINV;
1576
21.1k
  else
1577
21.1k
    wiener_info->vfilter[0] = wiener_info->vfilter[WIENER_WIN - 1] = 0;
1578
42.3k
  wiener_info->vfilter[1] = wiener_info->vfilter[WIENER_WIN - 2] =
1579
42.3k
      aom_read_primitive_refsubexpfin(
1580
42.3k
          rb, WIENER_FILT_TAP1_MAXV - WIENER_FILT_TAP1_MINV + 1,
1581
42.3k
          WIENER_FILT_TAP1_SUBEXP_K,
1582
42.3k
          ref_wiener_info->vfilter[1] - WIENER_FILT_TAP1_MINV, ACCT_STR) +
1583
42.3k
      WIENER_FILT_TAP1_MINV;
1584
42.3k
  wiener_info->vfilter[2] = wiener_info->vfilter[WIENER_WIN - 3] =
1585
42.3k
      aom_read_primitive_refsubexpfin(
1586
42.3k
          rb, WIENER_FILT_TAP2_MAXV - WIENER_FILT_TAP2_MINV + 1,
1587
42.3k
          WIENER_FILT_TAP2_SUBEXP_K,
1588
42.3k
          ref_wiener_info->vfilter[2] - WIENER_FILT_TAP2_MINV, ACCT_STR) +
1589
42.3k
      WIENER_FILT_TAP2_MINV;
1590
  // The central element has an implicit +WIENER_FILT_STEP
1591
42.3k
  wiener_info->vfilter[WIENER_HALFWIN] =
1592
42.3k
      -2 * (wiener_info->vfilter[0] + wiener_info->vfilter[1] +
1593
42.3k
            wiener_info->vfilter[2]);
1594
1595
42.3k
  if (wiener_win == WIENER_WIN)
1596
21.2k
    wiener_info->hfilter[0] = wiener_info->hfilter[WIENER_WIN - 1] =
1597
21.2k
        aom_read_primitive_refsubexpfin(
1598
21.2k
            rb, WIENER_FILT_TAP0_MAXV - WIENER_FILT_TAP0_MINV + 1,
1599
21.2k
            WIENER_FILT_TAP0_SUBEXP_K,
1600
21.2k
            ref_wiener_info->hfilter[0] - WIENER_FILT_TAP0_MINV, ACCT_STR) +
1601
21.2k
        WIENER_FILT_TAP0_MINV;
1602
21.1k
  else
1603
21.1k
    wiener_info->hfilter[0] = wiener_info->hfilter[WIENER_WIN - 1] = 0;
1604
42.3k
  wiener_info->hfilter[1] = wiener_info->hfilter[WIENER_WIN - 2] =
1605
42.3k
      aom_read_primitive_refsubexpfin(
1606
42.3k
          rb, WIENER_FILT_TAP1_MAXV - WIENER_FILT_TAP1_MINV + 1,
1607
42.3k
          WIENER_FILT_TAP1_SUBEXP_K,
1608
42.3k
          ref_wiener_info->hfilter[1] - WIENER_FILT_TAP1_MINV, ACCT_STR) +
1609
42.3k
      WIENER_FILT_TAP1_MINV;
1610
42.3k
  wiener_info->hfilter[2] = wiener_info->hfilter[WIENER_WIN - 3] =
1611
42.3k
      aom_read_primitive_refsubexpfin(
1612
42.3k
          rb, WIENER_FILT_TAP2_MAXV - WIENER_FILT_TAP2_MINV + 1,
1613
42.3k
          WIENER_FILT_TAP2_SUBEXP_K,
1614
42.3k
          ref_wiener_info->hfilter[2] - WIENER_FILT_TAP2_MINV, ACCT_STR) +
1615
42.3k
      WIENER_FILT_TAP2_MINV;
1616
  // The central element has an implicit +WIENER_FILT_STEP
1617
42.3k
  wiener_info->hfilter[WIENER_HALFWIN] =
1618
42.3k
      -2 * (wiener_info->hfilter[0] + wiener_info->hfilter[1] +
1619
42.3k
            wiener_info->hfilter[2]);
1620
42.3k
  *ref_wiener_info = *wiener_info;
1621
42.3k
}
1622
1623
static inline void read_sgrproj_filter(SgrprojInfo *sgrproj_info,
1624
                                       SgrprojInfo *ref_sgrproj_info,
1625
103k
                                       aom_reader *rb) {
1626
103k
  sgrproj_info->ep = aom_read_literal(rb, SGRPROJ_PARAMS_BITS, ACCT_STR);
1627
103k
  const sgr_params_type *params = &av1_sgr_params[sgrproj_info->ep];
1628
1629
103k
  if (params->r[0] == 0) {
1630
20.8k
    sgrproj_info->xqd[0] = 0;
1631
20.8k
    sgrproj_info->xqd[1] =
1632
20.8k
        aom_read_primitive_refsubexpfin(
1633
20.8k
            rb, SGRPROJ_PRJ_MAX1 - SGRPROJ_PRJ_MIN1 + 1, SGRPROJ_PRJ_SUBEXP_K,
1634
20.8k
            ref_sgrproj_info->xqd[1] - SGRPROJ_PRJ_MIN1, ACCT_STR) +
1635
20.8k
        SGRPROJ_PRJ_MIN1;
1636
83.0k
  } else if (params->r[1] == 0) {
1637
24.5k
    sgrproj_info->xqd[0] =
1638
24.5k
        aom_read_primitive_refsubexpfin(
1639
24.5k
            rb, SGRPROJ_PRJ_MAX0 - SGRPROJ_PRJ_MIN0 + 1, SGRPROJ_PRJ_SUBEXP_K,
1640
24.5k
            ref_sgrproj_info->xqd[0] - SGRPROJ_PRJ_MIN0, ACCT_STR) +
1641
24.5k
        SGRPROJ_PRJ_MIN0;
1642
24.5k
    sgrproj_info->xqd[1] = clamp((1 << SGRPROJ_PRJ_BITS) - sgrproj_info->xqd[0],
1643
24.5k
                                 SGRPROJ_PRJ_MIN1, SGRPROJ_PRJ_MAX1);
1644
58.4k
  } else {
1645
58.4k
    sgrproj_info->xqd[0] =
1646
58.4k
        aom_read_primitive_refsubexpfin(
1647
58.4k
            rb, SGRPROJ_PRJ_MAX0 - SGRPROJ_PRJ_MIN0 + 1, SGRPROJ_PRJ_SUBEXP_K,
1648
58.4k
            ref_sgrproj_info->xqd[0] - SGRPROJ_PRJ_MIN0, ACCT_STR) +
1649
58.4k
        SGRPROJ_PRJ_MIN0;
1650
58.4k
    sgrproj_info->xqd[1] =
1651
58.4k
        aom_read_primitive_refsubexpfin(
1652
58.4k
            rb, SGRPROJ_PRJ_MAX1 - SGRPROJ_PRJ_MIN1 + 1, SGRPROJ_PRJ_SUBEXP_K,
1653
58.4k
            ref_sgrproj_info->xqd[1] - SGRPROJ_PRJ_MIN1, ACCT_STR) +
1654
58.4k
        SGRPROJ_PRJ_MIN1;
1655
58.4k
  }
1656
1657
103k
  *ref_sgrproj_info = *sgrproj_info;
1658
103k
}
1659
1660
static inline void loop_restoration_read_sb_coeffs(const AV1_COMMON *const cm,
1661
                                                   MACROBLOCKD *xd,
1662
                                                   aom_reader *const r,
1663
258k
                                                   int plane, int runit_idx) {
1664
258k
  const RestorationInfo *rsi = &cm->rst_info[plane];
1665
258k
  RestorationUnitInfo *rui = &rsi->unit_info[runit_idx];
1666
258k
  assert(rsi->frame_restoration_type != RESTORE_NONE);
1667
1668
258k
  assert(!cm->features.all_lossless);
1669
1670
258k
  const int wiener_win = (plane > 0) ? WIENER_WIN_CHROMA : WIENER_WIN;
1671
258k
  WienerInfo *wiener_info = xd->wiener_info + plane;
1672
258k
  SgrprojInfo *sgrproj_info = xd->sgrproj_info + plane;
1673
1674
258k
  if (rsi->frame_restoration_type == RESTORE_SWITCHABLE) {
1675
69.6k
    rui->restoration_type =
1676
69.6k
        aom_read_symbol(r, xd->tile_ctx->switchable_restore_cdf,
1677
69.6k
                        RESTORE_SWITCHABLE_TYPES, ACCT_STR);
1678
69.6k
    switch (rui->restoration_type) {
1679
20.2k
      case RESTORE_WIENER:
1680
20.2k
        read_wiener_filter(wiener_win, &rui->wiener_info, wiener_info, r);
1681
20.2k
        break;
1682
23.5k
      case RESTORE_SGRPROJ:
1683
23.5k
        read_sgrproj_filter(&rui->sgrproj_info, sgrproj_info, r);
1684
23.5k
        break;
1685
25.7k
      default: assert(rui->restoration_type == RESTORE_NONE); break;
1686
69.6k
    }
1687
188k
  } else if (rsi->frame_restoration_type == RESTORE_WIENER) {
1688
47.1k
    if (aom_read_symbol(r, xd->tile_ctx->wiener_restore_cdf, 2, ACCT_STR)) {
1689
22.0k
      rui->restoration_type = RESTORE_WIENER;
1690
22.0k
      read_wiener_filter(wiener_win, &rui->wiener_info, wiener_info, r);
1691
25.0k
    } else {
1692
25.0k
      rui->restoration_type = RESTORE_NONE;
1693
25.0k
    }
1694
141k
  } else if (rsi->frame_restoration_type == RESTORE_SGRPROJ) {
1695
141k
    if (aom_read_symbol(r, xd->tile_ctx->sgrproj_restore_cdf, 2, ACCT_STR)) {
1696
80.4k
      rui->restoration_type = RESTORE_SGRPROJ;
1697
80.4k
      read_sgrproj_filter(&rui->sgrproj_info, sgrproj_info, r);
1698
80.4k
    } else {
1699
61.0k
      rui->restoration_type = RESTORE_NONE;
1700
61.0k
    }
1701
141k
  }
1702
258k
}
1703
1704
static inline void setup_loopfilter(AV1_COMMON *cm,
1705
170k
                                    struct aom_read_bit_buffer *rb) {
1706
170k
  const int num_planes = av1_num_planes(cm);
1707
170k
  struct loopfilter *lf = &cm->lf;
1708
1709
170k
  if (cm->features.allow_intrabc || cm->features.coded_lossless) {
1710
    // write default deltas to frame buffer
1711
70.9k
    av1_set_default_ref_deltas(cm->cur_frame->ref_deltas);
1712
70.9k
    av1_set_default_mode_deltas(cm->cur_frame->mode_deltas);
1713
70.9k
    return;
1714
70.9k
  }
1715
100k
  assert(!cm->features.coded_lossless);
1716
100k
  if (cm->prev_frame) {
1717
    // write deltas to frame buffer
1718
33.7k
    memcpy(lf->ref_deltas, cm->prev_frame->ref_deltas, REF_FRAMES);
1719
33.7k
    memcpy(lf->mode_deltas, cm->prev_frame->mode_deltas, MAX_MODE_LF_DELTAS);
1720
66.2k
  } else {
1721
66.2k
    av1_set_default_ref_deltas(lf->ref_deltas);
1722
66.2k
    av1_set_default_mode_deltas(lf->mode_deltas);
1723
66.2k
  }
1724
100k
  lf->filter_level[0] = aom_rb_read_literal(rb, 6);
1725
100k
  lf->filter_level[1] = aom_rb_read_literal(rb, 6);
1726
100k
  if (num_planes > 1) {
1727
84.8k
    if (lf->filter_level[0] || lf->filter_level[1]) {
1728
51.8k
      lf->filter_level_u = aom_rb_read_literal(rb, 6);
1729
51.8k
      lf->filter_level_v = aom_rb_read_literal(rb, 6);
1730
51.8k
    }
1731
84.8k
  }
1732
100k
  lf->sharpness_level = aom_rb_read_literal(rb, 3);
1733
1734
  // Read in loop filter deltas applied at the MB level based on mode or ref
1735
  // frame.
1736
100k
  lf->mode_ref_delta_update = 0;
1737
1738
100k
  lf->mode_ref_delta_enabled = aom_rb_read_bit(rb);
1739
100k
  if (lf->mode_ref_delta_enabled) {
1740
30.9k
    lf->mode_ref_delta_update = aom_rb_read_bit(rb);
1741
30.9k
    if (lf->mode_ref_delta_update) {
1742
141k
      for (int i = 0; i < REF_FRAMES; i++)
1743
126k
        if (aom_rb_read_bit(rb))
1744
53.5k
          lf->ref_deltas[i] = aom_rb_read_inv_signed_literal(rb, 6);
1745
1746
46.8k
      for (int i = 0; i < MAX_MODE_LF_DELTAS; i++)
1747
31.1k
        if (aom_rb_read_bit(rb))
1748
6.09k
          lf->mode_deltas[i] = aom_rb_read_inv_signed_literal(rb, 6);
1749
15.7k
    }
1750
30.9k
  }
1751
1752
  // write deltas to frame buffer
1753
100k
  memcpy(cm->cur_frame->ref_deltas, lf->ref_deltas, REF_FRAMES);
1754
100k
  memcpy(cm->cur_frame->mode_deltas, lf->mode_deltas, MAX_MODE_LF_DELTAS);
1755
100k
}
1756
1757
64.0k
static inline void setup_cdef(AV1_COMMON *cm, struct aom_read_bit_buffer *rb) {
1758
64.0k
  const int num_planes = av1_num_planes(cm);
1759
64.0k
  CdefInfo *const cdef_info = &cm->cdef_info;
1760
1761
64.0k
  if (cm->features.allow_intrabc) return;
1762
56.1k
  cdef_info->cdef_damping = aom_rb_read_literal(rb, 2) + 3;
1763
56.1k
  cdef_info->cdef_bits = aom_rb_read_literal(rb, 2);
1764
56.1k
  cdef_info->nb_cdef_strengths = 1 << cdef_info->cdef_bits;
1765
155k
  for (int i = 0; i < cdef_info->nb_cdef_strengths; i++) {
1766
99.4k
    cdef_info->cdef_strengths[i] = aom_rb_read_literal(rb, CDEF_STRENGTH_BITS);
1767
99.4k
    cdef_info->cdef_uv_strengths[i] =
1768
99.4k
        num_planes > 1 ? aom_rb_read_literal(rb, CDEF_STRENGTH_BITS) : 0;
1769
99.4k
  }
1770
56.1k
}
1771
1772
465k
static inline int read_delta_q(struct aom_read_bit_buffer *rb) {
1773
465k
  return aom_rb_read_bit(rb) ? aom_rb_read_inv_signed_literal(rb, 6) : 0;
1774
465k
}
1775
1776
static inline void setup_quantization(CommonQuantParams *quant_params,
1777
                                      int num_planes, bool separate_uv_delta_q,
1778
172k
                                      struct aom_read_bit_buffer *rb) {
1779
172k
  quant_params->base_qindex = aom_rb_read_literal(rb, QINDEX_BITS);
1780
172k
  quant_params->y_dc_delta_q = read_delta_q(rb);
1781
172k
  if (num_planes > 1) {
1782
143k
    int diff_uv_delta = 0;
1783
143k
    if (separate_uv_delta_q) diff_uv_delta = aom_rb_read_bit(rb);
1784
143k
    quant_params->u_dc_delta_q = read_delta_q(rb);
1785
143k
    quant_params->u_ac_delta_q = read_delta_q(rb);
1786
143k
    if (diff_uv_delta) {
1787
2.49k
      quant_params->v_dc_delta_q = read_delta_q(rb);
1788
2.49k
      quant_params->v_ac_delta_q = read_delta_q(rb);
1789
141k
    } else {
1790
141k
      quant_params->v_dc_delta_q = quant_params->u_dc_delta_q;
1791
141k
      quant_params->v_ac_delta_q = quant_params->u_ac_delta_q;
1792
141k
    }
1793
143k
  } else {
1794
29.0k
    quant_params->u_dc_delta_q = 0;
1795
29.0k
    quant_params->u_ac_delta_q = 0;
1796
29.0k
    quant_params->v_dc_delta_q = 0;
1797
29.0k
    quant_params->v_ac_delta_q = 0;
1798
29.0k
  }
1799
172k
  quant_params->using_qmatrix = aom_rb_read_bit(rb);
1800
172k
  if (quant_params->using_qmatrix) {
1801
37.6k
    quant_params->qmatrix_level_y = aom_rb_read_literal(rb, QM_LEVEL_BITS);
1802
37.6k
    quant_params->qmatrix_level_u = aom_rb_read_literal(rb, QM_LEVEL_BITS);
1803
37.6k
    if (!separate_uv_delta_q)
1804
36.2k
      quant_params->qmatrix_level_v = quant_params->qmatrix_level_u;
1805
1.44k
    else
1806
1.44k
      quant_params->qmatrix_level_v = aom_rb_read_literal(rb, QM_LEVEL_BITS);
1807
135k
  } else {
1808
135k
    quant_params->qmatrix_level_y = 0;
1809
135k
    quant_params->qmatrix_level_u = 0;
1810
135k
    quant_params->qmatrix_level_v = 0;
1811
135k
  }
1812
172k
}
1813
1814
// Get global dequant matrix.
1815
static const qm_val_t *get_iqmatrix(const CommonQuantParams *quant_params,
1816
19.7M
                                    int qmlevel, int plane, TX_SIZE tx_size) {
1817
19.7M
  assert(quant_params->giqmatrix[qmlevel][plane][tx_size] != NULL ||
1818
19.7M
         qmlevel == NUM_QM_LEVELS - 1);
1819
19.7M
  return quant_params->giqmatrix[qmlevel][plane][tx_size];
1820
19.7M
}
1821
1822
// Build y/uv dequant values based on segmentation.
1823
static inline void setup_segmentation_dequant(AV1_COMMON *const cm,
1824
170k
                                              MACROBLOCKD *const xd) {
1825
170k
  const int bit_depth = cm->seq_params->bit_depth;
1826
  // When segmentation is disabled, only the first value is used.  The
1827
  // remaining are don't cares.
1828
170k
  const int max_segments = cm->seg.enabled ? MAX_SEGMENTS : 1;
1829
170k
  CommonQuantParams *const quant_params = &cm->quant_params;
1830
517k
  for (int i = 0; i < max_segments; ++i) {
1831
346k
    const int qindex = xd->qindex[i];
1832
346k
    quant_params->y_dequant_QTX[i][0] =
1833
346k
        av1_dc_quant_QTX(qindex, quant_params->y_dc_delta_q, bit_depth);
1834
346k
    quant_params->y_dequant_QTX[i][1] = av1_ac_quant_QTX(qindex, 0, bit_depth);
1835
346k
    quant_params->u_dequant_QTX[i][0] =
1836
346k
        av1_dc_quant_QTX(qindex, quant_params->u_dc_delta_q, bit_depth);
1837
346k
    quant_params->u_dequant_QTX[i][1] =
1838
346k
        av1_ac_quant_QTX(qindex, quant_params->u_ac_delta_q, bit_depth);
1839
346k
    quant_params->v_dequant_QTX[i][0] =
1840
346k
        av1_dc_quant_QTX(qindex, quant_params->v_dc_delta_q, bit_depth);
1841
346k
    quant_params->v_dequant_QTX[i][1] =
1842
346k
        av1_ac_quant_QTX(qindex, quant_params->v_ac_delta_q, bit_depth);
1843
346k
    const int use_qmatrix = av1_use_qmatrix(quant_params, xd, i);
1844
    // NB: depends on base index so there is only 1 set per frame
1845
    // No quant weighting when lossless or signalled not using QM
1846
346k
    const int qmlevel_y =
1847
346k
        use_qmatrix ? quant_params->qmatrix_level_y : NUM_QM_LEVELS - 1;
1848
6.93M
    for (int j = 0; j < TX_SIZES_ALL; ++j) {
1849
6.59M
      quant_params->y_iqmatrix[i][j] =
1850
6.59M
          get_iqmatrix(quant_params, qmlevel_y, AOM_PLANE_Y, j);
1851
6.59M
    }
1852
346k
    const int qmlevel_u =
1853
346k
        use_qmatrix ? quant_params->qmatrix_level_u : NUM_QM_LEVELS - 1;
1854
6.93M
    for (int j = 0; j < TX_SIZES_ALL; ++j) {
1855
6.59M
      quant_params->u_iqmatrix[i][j] =
1856
6.59M
          get_iqmatrix(quant_params, qmlevel_u, AOM_PLANE_U, j);
1857
6.59M
    }
1858
346k
    const int qmlevel_v =
1859
346k
        use_qmatrix ? quant_params->qmatrix_level_v : NUM_QM_LEVELS - 1;
1860
6.93M
    for (int j = 0; j < TX_SIZES_ALL; ++j) {
1861
6.59M
      quant_params->v_iqmatrix[i][j] =
1862
6.59M
          get_iqmatrix(quant_params, qmlevel_v, AOM_PLANE_V, j);
1863
6.59M
    }
1864
346k
  }
1865
170k
}
1866
1867
48.8k
static InterpFilter read_frame_interp_filter(struct aom_read_bit_buffer *rb) {
1868
48.8k
  return aom_rb_read_bit(rb) ? SWITCHABLE
1869
48.8k
                             : aom_rb_read_literal(rb, LOG_SWITCHABLE_FILTERS);
1870
48.8k
}
1871
1872
static void read_frame_size(struct aom_read_bit_buffer *rb, int num_bits_width,
1873
96.8k
                            int num_bits_height, int *width, int *height) {
1874
96.8k
  *width = aom_rb_read_literal(rb, num_bits_width) + 1;
1875
96.8k
  *height = aom_rb_read_literal(rb, num_bits_height) + 1;
1876
96.8k
}
1877
1878
static inline void setup_render_size(AV1_COMMON *cm,
1879
153k
                                     struct aom_read_bit_buffer *rb) {
1880
153k
  cm->render_width = cm->superres_upscaled_width;
1881
153k
  cm->render_height = cm->superres_upscaled_height;
1882
153k
  if (aom_rb_read_bit(rb))
1883
30.9k
    read_frame_size(rb, 16, 16, &cm->render_width, &cm->render_height);
1884
153k
}
1885
1886
// TODO(afergs): make "struct aom_read_bit_buffer *const rb"?
1887
static inline void setup_superres(AV1_COMMON *const cm,
1888
                                  struct aom_read_bit_buffer *rb, int *width,
1889
186k
                                  int *height) {
1890
186k
  cm->superres_upscaled_width = *width;
1891
186k
  cm->superres_upscaled_height = *height;
1892
1893
186k
  const SequenceHeader *const seq_params = cm->seq_params;
1894
186k
  if (!seq_params->enable_superres) return;
1895
1896
76.1k
  if (aom_rb_read_bit(rb)) {
1897
33.1k
    cm->superres_scale_denominator =
1898
33.1k
        (uint8_t)aom_rb_read_literal(rb, SUPERRES_SCALE_BITS);
1899
33.1k
    cm->superres_scale_denominator += SUPERRES_SCALE_DENOMINATOR_MIN;
1900
    // Don't edit cm->width or cm->height directly, or the buffers won't get
1901
    // resized correctly
1902
33.1k
    av1_calculate_scaled_superres_size(width, height,
1903
33.1k
                                       cm->superres_scale_denominator);
1904
42.9k
  } else {
1905
    // 1:1 scaling - ie. no scaling, scale not provided
1906
42.9k
    cm->superres_scale_denominator = SCALE_NUMERATOR;
1907
42.9k
  }
1908
76.1k
}
1909
1910
static inline void resize_context_buffers(AV1_COMMON *cm, int width,
1911
186k
                                          int height) {
1912
186k
#if CONFIG_SIZE_LIMIT
1913
186k
  if (width > DECODE_WIDTH_LIMIT || height > DECODE_HEIGHT_LIMIT)
1914
281
    aom_internal_error(cm->error, AOM_CODEC_CORRUPT_FRAME,
1915
281
                       "Dimensions of %dx%d beyond allowed size of %dx%d.",
1916
281
                       width, height, DECODE_WIDTH_LIMIT, DECODE_HEIGHT_LIMIT);
1917
186k
#endif
1918
186k
  if (cm->width != width || cm->height != height) {
1919
87.4k
    const int new_mi_rows = CEIL_POWER_OF_TWO(height, MI_SIZE_LOG2);
1920
87.4k
    const int new_mi_cols = CEIL_POWER_OF_TWO(width, MI_SIZE_LOG2);
1921
1922
    // Allocations in av1_alloc_context_buffers() depend on individual
1923
    // dimensions as well as the overall size.
1924
87.4k
    if (new_mi_cols > cm->mi_params.mi_cols ||
1925
87.4k
        new_mi_rows > cm->mi_params.mi_rows) {
1926
53.0k
      if (av1_alloc_context_buffers(cm, width, height, BLOCK_4X4)) {
1927
        // The cm->mi_* values have been cleared and any existing context
1928
        // buffers have been freed. Clear cm->width and cm->height to be
1929
        // consistent and to force a realloc next time.
1930
302
        cm->width = 0;
1931
302
        cm->height = 0;
1932
302
        aom_internal_error(cm->error, AOM_CODEC_MEM_ERROR,
1933
302
                           "Failed to allocate context buffers");
1934
302
      }
1935
53.0k
    } else {
1936
34.4k
      cm->mi_params.set_mb_mi(&cm->mi_params, width, height, BLOCK_4X4);
1937
34.4k
    }
1938
87.4k
    av1_init_mi_buffers(&cm->mi_params);
1939
87.4k
    cm->width = width;
1940
87.4k
    cm->height = height;
1941
87.4k
  }
1942
1943
186k
  ensure_mv_buffer(cm->cur_frame, cm);
1944
186k
  cm->cur_frame->width = cm->width;
1945
186k
  cm->cur_frame->height = cm->height;
1946
186k
}
1947
1948
178k
static inline void setup_buffer_pool(AV1_COMMON *cm) {
1949
178k
  BufferPool *const pool = cm->buffer_pool;
1950
178k
  const SequenceHeader *const seq_params = cm->seq_params;
1951
1952
178k
  lock_buffer_pool(pool);
1953
178k
  if (aom_realloc_frame_buffer(
1954
178k
          &cm->cur_frame->buf, cm->width, cm->height, seq_params->subsampling_x,
1955
178k
          seq_params->subsampling_y, seq_params->use_highbitdepth,
1956
178k
          AOM_DEC_BORDER_IN_PIXELS, cm->features.byte_alignment,
1957
178k
          &cm->cur_frame->raw_frame_buffer, pool->get_fb_cb, pool->cb_priv,
1958
178k
          false, 0)) {
1959
142
    unlock_buffer_pool(pool);
1960
142
    aom_internal_error(cm->error, AOM_CODEC_MEM_ERROR,
1961
142
                       "Failed to allocate frame buffer");
1962
142
  }
1963
178k
  unlock_buffer_pool(pool);
1964
1965
178k
  cm->cur_frame->buf.bit_depth = (unsigned int)seq_params->bit_depth;
1966
178k
  cm->cur_frame->buf.color_primaries = seq_params->color_primaries;
1967
178k
  cm->cur_frame->buf.transfer_characteristics =
1968
178k
      seq_params->transfer_characteristics;
1969
178k
  cm->cur_frame->buf.matrix_coefficients = seq_params->matrix_coefficients;
1970
178k
  cm->cur_frame->buf.monochrome = seq_params->monochrome;
1971
178k
  cm->cur_frame->buf.chroma_sample_position =
1972
178k
      seq_params->chroma_sample_position;
1973
178k
  cm->cur_frame->buf.color_range = seq_params->color_range;
1974
178k
  cm->cur_frame->buf.render_width = cm->render_width;
1975
178k
  cm->cur_frame->buf.render_height = cm->render_height;
1976
178k
}
1977
1978
static inline void setup_frame_size(AV1_COMMON *cm,
1979
                                    int frame_size_override_flag,
1980
150k
                                    struct aom_read_bit_buffer *rb) {
1981
150k
  const SequenceHeader *const seq_params = cm->seq_params;
1982
150k
  int width, height;
1983
1984
150k
  if (frame_size_override_flag) {
1985
60.5k
    int num_bits_width = seq_params->num_bits_width;
1986
60.5k
    int num_bits_height = seq_params->num_bits_height;
1987
60.5k
    read_frame_size(rb, num_bits_width, num_bits_height, &width, &height);
1988
60.5k
    if (width > seq_params->max_frame_width ||
1989
60.5k
        height > seq_params->max_frame_height) {
1990
1.86k
      aom_internal_error(cm->error, AOM_CODEC_CORRUPT_FRAME,
1991
1.86k
                         "Frame dimensions are larger than the maximum values");
1992
1.86k
    }
1993
90.1k
  } else {
1994
90.1k
    width = seq_params->max_frame_width;
1995
90.1k
    height = seq_params->max_frame_height;
1996
90.1k
  }
1997
1998
150k
  setup_superres(cm, rb, &width, &height);
1999
150k
  resize_context_buffers(cm, width, height);
2000
150k
  setup_render_size(cm, rb);
2001
150k
  setup_buffer_pool(cm);
2002
150k
}
2003
2004
static inline void setup_sb_size(SequenceHeader *seq_params,
2005
76.8k
                                 struct aom_read_bit_buffer *rb) {
2006
76.8k
  set_sb_size(seq_params, aom_rb_read_bit(rb) ? BLOCK_128X128 : BLOCK_64X64);
2007
76.8k
}
2008
2009
static inline int valid_ref_frame_img_fmt(aom_bit_depth_t ref_bit_depth,
2010
                                          int ref_xss, int ref_yss,
2011
                                          aom_bit_depth_t this_bit_depth,
2012
226k
                                          int this_xss, int this_yss) {
2013
226k
  return ref_bit_depth == this_bit_depth && ref_xss == this_xss &&
2014
226k
         ref_yss == this_yss;
2015
226k
}
2016
2017
static inline void setup_frame_size_with_refs(AV1_COMMON *cm,
2018
37.9k
                                              struct aom_read_bit_buffer *rb) {
2019
37.9k
  int width, height;
2020
37.9k
  int found = 0;
2021
37.9k
  int has_valid_ref_frame = 0;
2022
135k
  for (int i = LAST_FRAME; i <= ALTREF_FRAME; ++i) {
2023
129k
    if (aom_rb_read_bit(rb)) {
2024
32.5k
      const RefCntBuffer *const ref_buf = get_ref_frame_buf(cm, i);
2025
      // This will never be NULL in a normal stream, as streams are required to
2026
      // have a shown keyframe before any inter frames, which would refresh all
2027
      // the reference buffers. However, it might be null if we're starting in
2028
      // the middle of a stream, and static analysis will error if we don't do
2029
      // a null check here.
2030
32.5k
      if (ref_buf == NULL) {
2031
0
        aom_internal_error(cm->error, AOM_CODEC_CORRUPT_FRAME,
2032
0
                           "Invalid condition: invalid reference buffer");
2033
32.5k
      } else {
2034
32.5k
        const YV12_BUFFER_CONFIG *const buf = &ref_buf->buf;
2035
32.5k
        width = buf->y_crop_width;
2036
32.5k
        height = buf->y_crop_height;
2037
32.5k
        cm->render_width = buf->render_width;
2038
32.5k
        cm->render_height = buf->render_height;
2039
32.5k
        setup_superres(cm, rb, &width, &height);
2040
32.5k
        resize_context_buffers(cm, width, height);
2041
32.5k
        found = 1;
2042
32.5k
        break;
2043
32.5k
      }
2044
32.5k
    }
2045
129k
  }
2046
2047
37.9k
  const SequenceHeader *const seq_params = cm->seq_params;
2048
37.9k
  if (!found) {
2049
5.38k
    int num_bits_width = seq_params->num_bits_width;
2050
5.38k
    int num_bits_height = seq_params->num_bits_height;
2051
2052
5.38k
    read_frame_size(rb, num_bits_width, num_bits_height, &width, &height);
2053
5.38k
    setup_superres(cm, rb, &width, &height);
2054
5.38k
    resize_context_buffers(cm, width, height);
2055
5.38k
    setup_render_size(cm, rb);
2056
5.38k
  }
2057
2058
37.9k
  if (width <= 0 || height <= 0)
2059
0
    aom_internal_error(cm->error, AOM_CODEC_CORRUPT_FRAME,
2060
0
                       "Invalid frame size");
2061
2062
  // Check to make sure at least one of frames that this frame references
2063
  // has valid dimensions.
2064
303k
  for (int i = LAST_FRAME; i <= ALTREF_FRAME; ++i) {
2065
265k
    const RefCntBuffer *const ref_frame = get_ref_frame_buf(cm, i);
2066
265k
    has_valid_ref_frame |=
2067
265k
        valid_ref_frame_size(ref_frame->buf.y_crop_width,
2068
265k
                             ref_frame->buf.y_crop_height, width, height);
2069
265k
  }
2070
37.9k
  if (!has_valid_ref_frame)
2071
2.15k
    aom_internal_error(cm->error, AOM_CODEC_CORRUPT_FRAME,
2072
2.15k
                       "Referenced frame has invalid size");
2073
264k
  for (int i = LAST_FRAME; i <= ALTREF_FRAME; ++i) {
2074
226k
    const RefCntBuffer *const ref_frame = get_ref_frame_buf(cm, i);
2075
226k
    if (!valid_ref_frame_img_fmt(
2076
226k
            ref_frame->buf.bit_depth, ref_frame->buf.subsampling_x,
2077
226k
            ref_frame->buf.subsampling_y, seq_params->bit_depth,
2078
226k
            seq_params->subsampling_x, seq_params->subsampling_y))
2079
5.24k
      aom_internal_error(cm->error, AOM_CODEC_CORRUPT_FRAME,
2080
5.24k
                         "Referenced frame has incompatible color format");
2081
226k
  }
2082
37.9k
  setup_buffer_pool(cm);
2083
37.9k
}
2084
2085
// Same function as av1_read_uniform but reading from uncompresses header wb
2086
207k
static int rb_read_uniform(struct aom_read_bit_buffer *const rb, int n) {
2087
207k
  const int l = get_unsigned_bits(n);
2088
207k
  const int m = (1 << l) - n;
2089
207k
  const int v = aom_rb_read_literal(rb, l - 1);
2090
207k
  assert(l != 0);
2091
207k
  if (v < m)
2092
185k
    return v;
2093
21.3k
  else
2094
21.3k
    return (v << 1) - m + aom_rb_read_bit(rb);
2095
207k
}
2096
2097
static inline void read_tile_info_max_tile(
2098
173k
    AV1_COMMON *const cm, struct aom_read_bit_buffer *const rb) {
2099
173k
  const SequenceHeader *const seq_params = cm->seq_params;
2100
173k
  CommonTileParams *const tiles = &cm->tiles;
2101
173k
  int width_sb =
2102
173k
      CEIL_POWER_OF_TWO(cm->mi_params.mi_cols, seq_params->mib_size_log2);
2103
173k
  int height_sb =
2104
173k
      CEIL_POWER_OF_TWO(cm->mi_params.mi_rows, seq_params->mib_size_log2);
2105
2106
173k
  av1_get_tile_limits(cm);
2107
173k
  tiles->uniform_spacing = aom_rb_read_bit(rb);
2108
2109
  // Read tile columns
2110
173k
  if (tiles->uniform_spacing) {
2111
98.7k
    tiles->log2_cols = tiles->min_log2_cols;
2112
103k
    while (tiles->log2_cols < tiles->max_log2_cols) {
2113
64.5k
      if (!aom_rb_read_bit(rb)) {
2114
59.3k
        break;
2115
59.3k
      }
2116
5.20k
      tiles->log2_cols++;
2117
5.20k
    }
2118
98.7k
  } else {
2119
75.2k
    int i;
2120
75.2k
    int start_sb;
2121
178k
    for (i = 0, start_sb = 0; width_sb > 0 && i < MAX_TILE_COLS; i++) {
2122
103k
      const int size_sb =
2123
103k
          1 + rb_read_uniform(rb, AOMMIN(width_sb, tiles->max_width_sb));
2124
103k
      tiles->col_start_sb[i] = start_sb;
2125
103k
      start_sb += size_sb;
2126
103k
      width_sb -= size_sb;
2127
103k
    }
2128
75.2k
    tiles->cols = i;
2129
75.2k
    tiles->col_start_sb[i] = start_sb + width_sb;
2130
75.2k
  }
2131
173k
  av1_calculate_tile_cols(seq_params, cm->mi_params.mi_rows,
2132
173k
                          cm->mi_params.mi_cols, tiles);
2133
2134
  // Read tile rows
2135
173k
  if (tiles->uniform_spacing) {
2136
98.7k
    tiles->log2_rows = tiles->min_log2_rows;
2137
106k
    while (tiles->log2_rows < tiles->max_log2_rows) {
2138
64.4k
      if (!aom_rb_read_bit(rb)) {
2139
56.6k
        break;
2140
56.6k
      }
2141
7.81k
      tiles->log2_rows++;
2142
7.81k
    }
2143
98.7k
  } else {
2144
75.2k
    int i;
2145
75.2k
    int start_sb;
2146
179k
    for (i = 0, start_sb = 0; height_sb > 0 && i < MAX_TILE_ROWS; i++) {
2147
103k
      const int size_sb =
2148
103k
          1 + rb_read_uniform(rb, AOMMIN(height_sb, tiles->max_height_sb));
2149
103k
      tiles->row_start_sb[i] = start_sb;
2150
103k
      start_sb += size_sb;
2151
103k
      height_sb -= size_sb;
2152
103k
    }
2153
75.2k
    tiles->rows = i;
2154
75.2k
    tiles->row_start_sb[i] = start_sb + height_sb;
2155
75.2k
  }
2156
173k
  av1_calculate_tile_rows(seq_params, cm->mi_params.mi_rows, tiles);
2157
173k
}
2158
2159
21.3k
void av1_set_single_tile_decoding_mode(AV1_COMMON *const cm) {
2160
21.3k
  cm->tiles.single_tile_decoding = 0;
2161
21.3k
  if (cm->tiles.large_scale) {
2162
21.3k
    struct loopfilter *lf = &cm->lf;
2163
21.3k
    RestorationInfo *const rst_info = cm->rst_info;
2164
21.3k
    const CdefInfo *const cdef_info = &cm->cdef_info;
2165
2166
    // Figure out single_tile_decoding by loopfilter_level.
2167
21.3k
    const int no_loopfilter = !(lf->filter_level[0] || lf->filter_level[1]);
2168
21.3k
    const int no_cdef = cdef_info->cdef_bits == 0 &&
2169
21.3k
                        cdef_info->cdef_strengths[0] == 0 &&
2170
21.3k
                        cdef_info->cdef_uv_strengths[0] == 0;
2171
21.3k
    const int no_restoration =
2172
21.3k
        rst_info[0].frame_restoration_type == RESTORE_NONE &&
2173
21.3k
        rst_info[1].frame_restoration_type == RESTORE_NONE &&
2174
21.3k
        rst_info[2].frame_restoration_type == RESTORE_NONE;
2175
21.3k
    assert(IMPLIES(cm->features.coded_lossless, no_loopfilter && no_cdef));
2176
21.3k
    assert(IMPLIES(cm->features.all_lossless, no_restoration));
2177
21.3k
    cm->tiles.single_tile_decoding = no_loopfilter && no_cdef && no_restoration;
2178
21.3k
  }
2179
21.3k
}
2180
2181
static inline void read_tile_info(AV1Decoder *const pbi,
2182
173k
                                  struct aom_read_bit_buffer *const rb) {
2183
173k
  AV1_COMMON *const cm = &pbi->common;
2184
2185
173k
  read_tile_info_max_tile(cm, rb);
2186
2187
173k
  pbi->context_update_tile_id = 0;
2188
173k
  if (cm->tiles.rows * cm->tiles.cols > 1) {
2189
    // tile to use for cdf update
2190
18.9k
    pbi->context_update_tile_id =
2191
18.9k
        aom_rb_read_literal(rb, cm->tiles.log2_rows + cm->tiles.log2_cols);
2192
18.9k
    if (pbi->context_update_tile_id >= cm->tiles.rows * cm->tiles.cols) {
2193
731
      aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
2194
731
                         "Invalid context_update_tile_id");
2195
731
    }
2196
    // tile size magnitude
2197
18.9k
    pbi->tile_size_bytes = aom_rb_read_literal(rb, 2) + 1;
2198
18.9k
  }
2199
173k
}
2200
2201
#if EXT_TILE_DEBUG
2202
static inline void read_ext_tile_info(AV1Decoder *const pbi,
2203
19.9k
                                      struct aom_read_bit_buffer *const rb) {
2204
19.9k
  AV1_COMMON *const cm = &pbi->common;
2205
2206
  // This information is stored as a separate byte.
2207
19.9k
  int mod = rb->bit_offset % CHAR_BIT;
2208
19.9k
  if (mod > 0) aom_rb_read_literal(rb, CHAR_BIT - mod);
2209
19.9k
  assert(rb->bit_offset % CHAR_BIT == 0);
2210
2211
19.9k
  if (cm->tiles.cols * cm->tiles.rows > 1) {
2212
    // Read the number of bytes used to store tile size
2213
4.54k
    pbi->tile_col_size_bytes = aom_rb_read_literal(rb, 2) + 1;
2214
4.54k
    pbi->tile_size_bytes = aom_rb_read_literal(rb, 2) + 1;
2215
4.54k
  }
2216
19.9k
}
2217
#endif  // EXT_TILE_DEBUG
2218
2219
36.8k
static size_t mem_get_varsize(const uint8_t *src, int sz) {
2220
36.8k
  switch (sz) {
2221
30.5k
    case 1: return src[0];
2222
4.83k
    case 2: return mem_get_le16(src);
2223
676
    case 3: return mem_get_le24(src);
2224
820
    case 4: return mem_get_le32(src);
2225
0
    default: assert(0 && "Invalid size"); return -1;
2226
36.8k
  }
2227
36.8k
}
2228
2229
#if EXT_TILE_DEBUG
2230
// Reads the next tile returning its size and adjusting '*data' accordingly
2231
// based on 'is_last'. On return, '*data' is updated to point to the end of the
2232
// raw tile buffer in the bit stream.
2233
static inline void get_ls_tile_buffer(
2234
    const uint8_t *const data_end, struct aom_internal_error_info *error_info,
2235
    const uint8_t **data, TileBufferDec (*const tile_buffers)[MAX_TILE_COLS],
2236
21.4k
    int tile_size_bytes, int col, int row, int tile_copy_mode) {
2237
21.4k
  size_t size;
2238
2239
21.4k
  size_t copy_size = 0;
2240
21.4k
  const uint8_t *copy_data = NULL;
2241
2242
21.4k
  if (!read_is_valid(*data, tile_size_bytes, data_end))
2243
644
    aom_internal_error(error_info, AOM_CODEC_CORRUPT_FRAME,
2244
644
                       "Truncated packet or corrupt tile length");
2245
21.4k
  size = mem_get_varsize(*data, tile_size_bytes);
2246
2247
  // If tile_copy_mode = 1, then the top bit of the tile header indicates copy
2248
  // mode.
2249
21.4k
  if (tile_copy_mode && (size >> (tile_size_bytes * 8 - 1)) == 1) {
2250
    // The remaining bits in the top byte signal the row offset
2251
14.5k
    int offset = (size >> (tile_size_bytes - 1) * 8) & 0x7f;
2252
14.5k
    if (offset > row) {
2253
717
      aom_internal_error(
2254
717
          error_info, AOM_CODEC_CORRUPT_FRAME,
2255
717
          "Invalid row offset in tile copy mode: row=%d offset=%d", row,
2256
717
          offset);
2257
717
    }
2258
2259
    // Currently, only use tiles in same column as reference tiles.
2260
14.5k
    copy_data = tile_buffers[row - offset][col].data;
2261
14.5k
    copy_size = tile_buffers[row - offset][col].size;
2262
14.5k
    size = 0;
2263
14.5k
  } else {
2264
6.85k
    size += AV1_MIN_TILE_SIZE_BYTES;
2265
6.85k
  }
2266
2267
21.4k
  *data += tile_size_bytes;
2268
2269
21.4k
  if (size > (size_t)(data_end - *data))
2270
344
    aom_internal_error(error_info, AOM_CODEC_CORRUPT_FRAME,
2271
344
                       "Truncated packet or corrupt tile size");
2272
2273
21.4k
  if (size > 0) {
2274
5.86k
    tile_buffers[row][col].data = *data;
2275
5.86k
    tile_buffers[row][col].size = size;
2276
15.5k
  } else {
2277
15.5k
    tile_buffers[row][col].data = copy_data;
2278
15.5k
    tile_buffers[row][col].size = copy_size;
2279
15.5k
  }
2280
2281
21.4k
  *data += size;
2282
21.4k
}
2283
2284
// Returns the end of the last tile buffer
2285
// (tile_buffers[cm->tiles.rows - 1][cm->tiles.cols - 1]).
2286
static const uint8_t *get_ls_tile_buffers(
2287
    AV1Decoder *pbi, const uint8_t *data, const uint8_t *data_end,
2288
18.9k
    TileBufferDec (*const tile_buffers)[MAX_TILE_COLS]) {
2289
18.9k
  AV1_COMMON *const cm = &pbi->common;
2290
18.9k
  const int tile_cols = cm->tiles.cols;
2291
18.9k
  const int tile_rows = cm->tiles.rows;
2292
18.9k
  const int have_tiles = tile_cols * tile_rows > 1;
2293
18.9k
  const uint8_t *raw_data_end;  // The end of the last tile buffer
2294
2295
18.9k
  if (!have_tiles) {
2296
15.3k
    const size_t tile_size = data_end - data;
2297
15.3k
    tile_buffers[0][0].data = data;
2298
15.3k
    tile_buffers[0][0].size = tile_size;
2299
15.3k
    raw_data_end = NULL;
2300
15.3k
  } else {
2301
    // We locate only the tile buffers that are required, which are the ones
2302
    // specified by pbi->dec_tile_col and pbi->dec_tile_row. Also, we always
2303
    // need the last (bottom right) tile buffer, as we need to know where the
2304
    // end of the compressed frame buffer is for proper superframe decoding.
2305
2306
3.62k
    const uint8_t *tile_col_data_end[MAX_TILE_COLS] = { NULL };
2307
3.62k
    const uint8_t *const data_start = data;
2308
2309
3.62k
    const int dec_tile_row = AOMMIN(pbi->dec_tile_row, tile_rows);
2310
3.62k
    const int single_row = pbi->dec_tile_row >= 0;
2311
3.62k
    const int tile_rows_start = single_row ? dec_tile_row : 0;
2312
3.62k
    const int tile_rows_end = single_row ? tile_rows_start + 1 : tile_rows;
2313
3.62k
    const int dec_tile_col = AOMMIN(pbi->dec_tile_col, tile_cols);
2314
3.62k
    const int single_col = pbi->dec_tile_col >= 0;
2315
3.62k
    const int tile_cols_start = single_col ? dec_tile_col : 0;
2316
3.62k
    const int tile_cols_end = single_col ? tile_cols_start + 1 : tile_cols;
2317
2318
3.62k
    const int tile_col_size_bytes = pbi->tile_col_size_bytes;
2319
3.62k
    const int tile_size_bytes = pbi->tile_size_bytes;
2320
3.62k
    int tile_width, tile_height;
2321
3.62k
    if (!av1_get_uniform_tile_size(cm, &tile_width, &tile_height)) {
2322
725
      aom_internal_error(
2323
725
          &pbi->error, AOM_CODEC_CORRUPT_FRAME,
2324
725
          "Not all the tiles in the tile list have the same size.");
2325
725
    }
2326
3.62k
    const int tile_copy_mode =
2327
3.62k
        ((AOMMAX(tile_width, tile_height) << MI_SIZE_LOG2) <= 256) ? 1 : 0;
2328
    // Read tile column sizes for all columns (we need the last tile buffer)
2329
7.55k
    for (int c = 0; c < tile_cols; ++c) {
2330
3.93k
      const int is_last = c == tile_cols - 1;
2331
3.93k
      size_t tile_col_size;
2332
2333
3.93k
      if (!is_last) {
2334
1.42k
        if (tile_col_size_bytes > data_end - data) {
2335
33
          aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
2336
33
                             "Not enough data to read tile_col_size");
2337
33
        }
2338
1.42k
        tile_col_size = mem_get_varsize(data, tile_col_size_bytes);
2339
1.42k
        data += tile_col_size_bytes;
2340
1.42k
        if (tile_col_size > (size_t)(data_end - data)) {
2341
356
          aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
2342
356
                             "tile_col_data_end[%d] is out of bound", c);
2343
356
        }
2344
1.42k
        tile_col_data_end[c] = data + tile_col_size;
2345
2.50k
      } else {
2346
2.50k
        tile_col_size = data_end - data;
2347
2.50k
        tile_col_data_end[c] = data_end;
2348
2.50k
      }
2349
3.93k
      data += tile_col_size;
2350
3.93k
    }
2351
2352
3.62k
    data = data_start;
2353
2354
    // Read the required tile sizes.
2355
6.17k
    for (int c = tile_cols_start; c < tile_cols_end; ++c) {
2356
2.55k
      const int is_last = c == tile_cols - 1;
2357
2358
2.55k
      if (c > 0) data = tile_col_data_end[c - 1];
2359
2360
2.55k
      if (!is_last) data += tile_col_size_bytes;
2361
2362
      // Get the whole of the last column, otherwise stop at the required tile.
2363
23.9k
      for (int r = 0; r < (is_last ? tile_rows : tile_rows_end); ++r) {
2364
21.4k
        get_ls_tile_buffer(tile_col_data_end[c], &pbi->error, &data,
2365
21.4k
                           tile_buffers, tile_size_bytes, c, r, tile_copy_mode);
2366
21.4k
      }
2367
2.55k
    }
2368
2369
    // If we have not read the last column, then read it to get the last tile.
2370
3.62k
    if (tile_cols_end != tile_cols) {
2371
0
      const int c = tile_cols - 1;
2372
2373
0
      data = tile_col_data_end[c - 1];
2374
2375
0
      for (int r = 0; r < tile_rows; ++r) {
2376
0
        get_ls_tile_buffer(tile_col_data_end[c], &pbi->error, &data,
2377
0
                           tile_buffers, tile_size_bytes, c, r, tile_copy_mode);
2378
0
      }
2379
0
    }
2380
3.62k
    raw_data_end = data;
2381
3.62k
  }
2382
18.9k
  return raw_data_end;
2383
18.9k
}
2384
#endif  // EXT_TILE_DEBUG
2385
2386
static const uint8_t *get_ls_single_tile_buffer(
2387
    AV1Decoder *pbi, const uint8_t *data,
2388
0
    TileBufferDec (*const tile_buffers)[MAX_TILE_COLS]) {
2389
0
  assert(pbi->dec_tile_row >= 0 && pbi->dec_tile_col >= 0);
2390
0
  tile_buffers[pbi->dec_tile_row][pbi->dec_tile_col].data = data;
2391
0
  tile_buffers[pbi->dec_tile_row][pbi->dec_tile_col].size =
2392
0
      (size_t)pbi->coded_tile_data_size;
2393
0
  return data + pbi->coded_tile_data_size;
2394
0
}
2395
2396
// Reads the next tile returning its size and adjusting '*data' accordingly
2397
// based on 'is_last'.
2398
static inline void get_tile_buffer(const uint8_t *const data_end,
2399
                                   const int tile_size_bytes, int is_last,
2400
                                   struct aom_internal_error_info *error_info,
2401
                                   const uint8_t **data,
2402
141k
                                   TileBufferDec *const buf) {
2403
141k
  size_t size;
2404
2405
141k
  if (!is_last) {
2406
14.8k
    if (!read_is_valid(*data, tile_size_bytes, data_end))
2407
124
      aom_internal_error(error_info, AOM_CODEC_CORRUPT_FRAME,
2408
124
                         "Not enough data to read tile size");
2409
2410
14.8k
    size = mem_get_varsize(*data, tile_size_bytes) + AV1_MIN_TILE_SIZE_BYTES;
2411
14.8k
    *data += tile_size_bytes;
2412
2413
14.8k
    if (size > (size_t)(data_end - *data))
2414
1.97k
      aom_internal_error(error_info, AOM_CODEC_CORRUPT_FRAME,
2415
1.97k
                         "Truncated packet or corrupt tile size");
2416
126k
  } else {
2417
126k
    size = data_end - *data;
2418
126k
  }
2419
2420
141k
  buf->data = *data;
2421
141k
  buf->size = size;
2422
2423
141k
  *data += size;
2424
141k
}
2425
2426
static inline void get_tile_buffers(
2427
    AV1Decoder *pbi, const uint8_t *data, const uint8_t *data_end,
2428
    TileBufferDec (*const tile_buffers)[MAX_TILE_COLS], int start_tile,
2429
128k
    int end_tile) {
2430
128k
  AV1_COMMON *const cm = &pbi->common;
2431
128k
  const int tile_cols = cm->tiles.cols;
2432
128k
  const int tile_rows = cm->tiles.rows;
2433
128k
  int tc = 0;
2434
2435
260k
  for (int r = 0; r < tile_rows; ++r) {
2436
273k
    for (int c = 0; c < tile_cols; ++c, ++tc) {
2437
141k
      TileBufferDec *const buf = &tile_buffers[r][c];
2438
2439
141k
      const int is_last = (tc == end_tile);
2440
141k
      const size_t hdr_offset = 0;
2441
2442
141k
      if (tc < start_tile || tc > end_tile) continue;
2443
2444
141k
      if (data + hdr_offset >= data_end)
2445
181
        aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
2446
181
                           "Data ended before all tiles were read.");
2447
141k
      data += hdr_offset;
2448
141k
      get_tile_buffer(data_end, pbi->tile_size_bytes, is_last, &pbi->error,
2449
141k
                      &data, buf);
2450
141k
    }
2451
132k
  }
2452
128k
}
2453
2454
static inline void set_cb_buffer(AV1Decoder *pbi, DecoderCodingBlock *dcb,
2455
                                 CB_BUFFER *cb_buffer_base,
2456
1.51M
                                 const int num_planes, int mi_row, int mi_col) {
2457
1.51M
  AV1_COMMON *const cm = &pbi->common;
2458
1.51M
  int mib_size_log2 = cm->seq_params->mib_size_log2;
2459
1.51M
  int stride = (cm->mi_params.mi_cols >> mib_size_log2) + 1;
2460
1.51M
  int offset = (mi_row >> mib_size_log2) * stride + (mi_col >> mib_size_log2);
2461
1.51M
  CB_BUFFER *cb_buffer = cb_buffer_base + offset;
2462
2463
5.79M
  for (int plane = 0; plane < num_planes; ++plane) {
2464
4.27M
    dcb->dqcoeff_block[plane] = cb_buffer->dqcoeff[plane];
2465
4.27M
    dcb->eob_data[plane] = cb_buffer->eob_data[plane];
2466
4.27M
    dcb->cb_offset[plane] = 0;
2467
4.27M
    dcb->txb_offset[plane] = 0;
2468
4.27M
  }
2469
1.51M
  MACROBLOCKD *const xd = &dcb->xd;
2470
1.51M
  xd->plane[0].color_index_map = cb_buffer->color_index_map[0];
2471
1.51M
  xd->plane[1].color_index_map = cb_buffer->color_index_map[1];
2472
1.51M
  xd->color_index_map_offset[0] = 0;
2473
1.51M
  xd->color_index_map_offset[1] = 0;
2474
1.51M
}
2475
2476
16.5k
static inline void decoder_alloc_tile_data(AV1Decoder *pbi, const int n_tiles) {
2477
16.5k
  AV1_COMMON *const cm = &pbi->common;
2478
16.5k
  aom_free(pbi->tile_data);
2479
16.5k
  pbi->allocated_tiles = 0;
2480
16.5k
  CHECK_MEM_ERROR(cm, pbi->tile_data,
2481
16.5k
                  aom_memalign(32, n_tiles * sizeof(*pbi->tile_data)));
2482
16.5k
  pbi->allocated_tiles = n_tiles;
2483
57.3k
  for (int i = 0; i < n_tiles; i++) {
2484
40.8k
    TileDataDec *const tile_data = pbi->tile_data + i;
2485
40.8k
    av1_zero(tile_data->dec_row_mt_sync);
2486
40.8k
  }
2487
16.5k
  pbi->allocated_row_mt_sync_rows = 0;
2488
16.5k
}
2489
2490
// Set up nsync by width.
2491
31.9k
static inline int get_sync_range(int width) {
2492
// nsync numbers are picked by testing.
2493
#if 0
2494
  if (width < 640)
2495
    return 1;
2496
  else if (width <= 1280)
2497
    return 2;
2498
  else if (width <= 4096)
2499
    return 4;
2500
  else
2501
    return 8;
2502
#else
2503
31.9k
  (void)width;
2504
31.9k
#endif
2505
31.9k
  return 1;
2506
31.9k
}
2507
2508
// Allocate memory for decoder row synchronization
2509
static inline void dec_row_mt_alloc(AV1DecRowMTSync *dec_row_mt_sync,
2510
31.9k
                                    AV1_COMMON *cm, int rows) {
2511
31.9k
  dec_row_mt_sync->allocated_sb_rows = rows;
2512
31.9k
#if CONFIG_MULTITHREAD
2513
31.9k
  {
2514
31.9k
    int i;
2515
2516
31.9k
    CHECK_MEM_ERROR(cm, dec_row_mt_sync->mutex_,
2517
31.9k
                    aom_malloc(sizeof(*(dec_row_mt_sync->mutex_)) * rows));
2518
31.9k
    if (dec_row_mt_sync->mutex_) {
2519
97.5k
      for (i = 0; i < rows; ++i) {
2520
65.6k
        pthread_mutex_init(&dec_row_mt_sync->mutex_[i], NULL);
2521
65.6k
      }
2522
31.9k
    }
2523
2524
31.9k
    CHECK_MEM_ERROR(cm, dec_row_mt_sync->cond_,
2525
31.9k
                    aom_malloc(sizeof(*(dec_row_mt_sync->cond_)) * rows));
2526
31.9k
    if (dec_row_mt_sync->cond_) {
2527
97.5k
      for (i = 0; i < rows; ++i) {
2528
65.6k
        pthread_cond_init(&dec_row_mt_sync->cond_[i], NULL);
2529
65.6k
      }
2530
31.9k
    }
2531
31.9k
  }
2532
31.9k
#endif  // CONFIG_MULTITHREAD
2533
2534
31.9k
  CHECK_MEM_ERROR(cm, dec_row_mt_sync->cur_sb_col,
2535
31.9k
                  aom_malloc(sizeof(*(dec_row_mt_sync->cur_sb_col)) * rows));
2536
2537
  // Set up nsync.
2538
31.9k
  dec_row_mt_sync->sync_range = get_sync_range(cm->width);
2539
31.9k
}
2540
2541
// Deallocate decoder row synchronization related mutex and data
2542
71.2k
void av1_dec_row_mt_dealloc(AV1DecRowMTSync *dec_row_mt_sync) {
2543
71.2k
  if (dec_row_mt_sync != NULL) {
2544
71.2k
#if CONFIG_MULTITHREAD
2545
71.2k
    int i;
2546
71.2k
    if (dec_row_mt_sync->mutex_ != NULL) {
2547
97.5k
      for (i = 0; i < dec_row_mt_sync->allocated_sb_rows; ++i) {
2548
65.6k
        pthread_mutex_destroy(&dec_row_mt_sync->mutex_[i]);
2549
65.6k
      }
2550
31.9k
      aom_free(dec_row_mt_sync->mutex_);
2551
31.9k
    }
2552
71.2k
    if (dec_row_mt_sync->cond_ != NULL) {
2553
97.5k
      for (i = 0; i < dec_row_mt_sync->allocated_sb_rows; ++i) {
2554
65.6k
        pthread_cond_destroy(&dec_row_mt_sync->cond_[i]);
2555
65.6k
      }
2556
31.9k
      aom_free(dec_row_mt_sync->cond_);
2557
31.9k
    }
2558
71.2k
#endif  // CONFIG_MULTITHREAD
2559
71.2k
    aom_free(dec_row_mt_sync->cur_sb_col);
2560
2561
    // clear the structure as the source of this call may be a resize in which
2562
    // case this call will be followed by an _alloc() which may fail.
2563
71.2k
    av1_zero(*dec_row_mt_sync);
2564
71.2k
  }
2565
71.2k
}
2566
2567
static inline void sync_read(AV1DecRowMTSync *const dec_row_mt_sync, int r,
2568
410k
                             int c) {
2569
410k
#if CONFIG_MULTITHREAD
2570
410k
  const int nsync = dec_row_mt_sync->sync_range;
2571
2572
410k
  if (r && !(c & (nsync - 1))) {
2573
266k
    pthread_mutex_t *const mutex = &dec_row_mt_sync->mutex_[r - 1];
2574
266k
    pthread_mutex_lock(mutex);
2575
2576
343k
    while (c > dec_row_mt_sync->cur_sb_col[r - 1] - nsync -
2577
343k
                   dec_row_mt_sync->intrabc_extra_top_right_sb_delay) {
2578
77.2k
      pthread_cond_wait(&dec_row_mt_sync->cond_[r - 1], mutex);
2579
77.2k
    }
2580
266k
    pthread_mutex_unlock(mutex);
2581
266k
  }
2582
#else
2583
  (void)dec_row_mt_sync;
2584
  (void)r;
2585
  (void)c;
2586
#endif  // CONFIG_MULTITHREAD
2587
410k
}
2588
2589
static inline void sync_write(AV1DecRowMTSync *const dec_row_mt_sync, int r,
2590
413k
                              int c, const int sb_cols) {
2591
413k
#if CONFIG_MULTITHREAD
2592
413k
  const int nsync = dec_row_mt_sync->sync_range;
2593
413k
  int cur;
2594
413k
  int sig = 1;
2595
2596
413k
  if (c < sb_cols - 1) {
2597
290k
    cur = c;
2598
290k
    if (c % nsync) sig = 0;
2599
290k
  } else {
2600
122k
    cur = sb_cols + nsync + dec_row_mt_sync->intrabc_extra_top_right_sb_delay;
2601
122k
  }
2602
2603
413k
  if (sig) {
2604
413k
    pthread_mutex_lock(&dec_row_mt_sync->mutex_[r]);
2605
2606
413k
    dec_row_mt_sync->cur_sb_col[r] = cur;
2607
2608
413k
    pthread_cond_signal(&dec_row_mt_sync->cond_[r]);
2609
413k
    pthread_mutex_unlock(&dec_row_mt_sync->mutex_[r]);
2610
413k
  }
2611
#else
2612
  (void)dec_row_mt_sync;
2613
  (void)r;
2614
  (void)c;
2615
  (void)sb_cols;
2616
#endif  // CONFIG_MULTITHREAD
2617
413k
}
2618
2619
static inline void signal_decoding_done_for_erroneous_row(
2620
2.11k
    AV1Decoder *const pbi, const MACROBLOCKD *const xd) {
2621
2.11k
  AV1_COMMON *const cm = &pbi->common;
2622
2.11k
  const TileInfo *const tile = &xd->tile;
2623
2.11k
  const int sb_row_in_tile =
2624
2.11k
      ((xd->mi_row - tile->mi_row_start) >> cm->seq_params->mib_size_log2);
2625
2.11k
  const int sb_cols_in_tile = av1_get_sb_cols_in_tile(cm, tile);
2626
2.11k
  TileDataDec *const tile_data =
2627
2.11k
      pbi->tile_data + tile->tile_row * cm->tiles.cols + tile->tile_col;
2628
2.11k
  AV1DecRowMTSync *dec_row_mt_sync = &tile_data->dec_row_mt_sync;
2629
2630
2.11k
  sync_write(dec_row_mt_sync, sb_row_in_tile, sb_cols_in_tile - 1,
2631
2.11k
             sb_cols_in_tile);
2632
2.11k
}
2633
2634
static inline void decode_tile_sb_row(AV1Decoder *pbi, ThreadData *const td,
2635
                                      const TileInfo *tile_info,
2636
120k
                                      const int mi_row) {
2637
120k
  AV1_COMMON *const cm = &pbi->common;
2638
120k
  const int num_planes = av1_num_planes(cm);
2639
120k
  TileDataDec *const tile_data = pbi->tile_data +
2640
120k
                                 tile_info->tile_row * cm->tiles.cols +
2641
120k
                                 tile_info->tile_col;
2642
120k
  const int sb_cols_in_tile = av1_get_sb_cols_in_tile(cm, tile_info);
2643
120k
  const int sb_row_in_tile =
2644
120k
      (mi_row - tile_info->mi_row_start) >> cm->seq_params->mib_size_log2;
2645
120k
  int sb_col_in_tile = 0;
2646
120k
  int row_mt_exit = 0;
2647
2648
530k
  for (int mi_col = tile_info->mi_col_start; mi_col < tile_info->mi_col_end;
2649
410k
       mi_col += cm->seq_params->mib_size, sb_col_in_tile++) {
2650
410k
    set_cb_buffer(pbi, &td->dcb, pbi->cb_buffer_base, num_planes, mi_row,
2651
410k
                  mi_col);
2652
2653
410k
    sync_read(&tile_data->dec_row_mt_sync, sb_row_in_tile, sb_col_in_tile);
2654
2655
410k
#if CONFIG_MULTITHREAD
2656
410k
    pthread_mutex_lock(pbi->row_mt_mutex_);
2657
410k
#endif
2658
410k
    row_mt_exit = pbi->frame_row_mt_info.row_mt_exit;
2659
410k
#if CONFIG_MULTITHREAD
2660
410k
    pthread_mutex_unlock(pbi->row_mt_mutex_);
2661
410k
#endif
2662
2663
410k
    if (!row_mt_exit) {
2664
      // Decoding of the super-block
2665
408k
      decode_partition(pbi, td, mi_row, mi_col, td->bit_reader,
2666
408k
                       cm->seq_params->sb_size, 0x2);
2667
408k
    }
2668
2669
410k
    sync_write(&tile_data->dec_row_mt_sync, sb_row_in_tile, sb_col_in_tile,
2670
410k
               sb_cols_in_tile);
2671
410k
  }
2672
120k
}
2673
2674
105k
static int check_trailing_bits_after_symbol_coder(aom_reader *r) {
2675
105k
  if (aom_reader_has_overflowed(r)) return -1;
2676
2677
105k
  uint32_t nb_bits = aom_reader_tell(r);
2678
105k
  uint32_t nb_bytes = (nb_bits + 7) >> 3;
2679
105k
  const uint8_t *p = aom_reader_find_begin(r) + nb_bytes;
2680
2681
  // aom_reader_tell() returns 1 for a newly initialized decoder, and the
2682
  // return value only increases as values are decoded. So nb_bits > 0, and
2683
  // thus p > p_begin. Therefore accessing p[-1] is safe.
2684
105k
  uint8_t last_byte = p[-1];
2685
105k
  uint8_t pattern = 128 >> ((nb_bits - 1) & 7);
2686
105k
  if ((last_byte & (2 * pattern - 1)) != pattern) return -1;
2687
2688
  // Make sure that all padding bytes are zero as required by the spec.
2689
97.2k
  const uint8_t *p_end = aom_reader_find_end(r);
2690
155k
  while (p < p_end) {
2691
59.1k
    if (*p != 0) return -1;
2692
58.3k
    p++;
2693
58.3k
  }
2694
96.4k
  return 0;
2695
97.2k
}
2696
2697
static inline void set_decode_func_pointers(ThreadData *td,
2698
297k
                                            int parse_decode_flag) {
2699
297k
  td->read_coeffs_tx_intra_block_visit = decode_block_void;
2700
297k
  td->predict_and_recon_intra_block_visit = decode_block_void;
2701
297k
  td->read_coeffs_tx_inter_block_visit = decode_block_void;
2702
297k
  td->inverse_tx_inter_block_visit = decode_block_void;
2703
297k
  td->predict_inter_block_visit = predict_inter_block_void;
2704
297k
  td->cfl_store_inter_block_visit = cfl_store_inter_block_void;
2705
2706
297k
  if (parse_decode_flag & 0x1) {
2707
200k
    td->read_coeffs_tx_intra_block_visit = read_coeffs_tx_intra_block;
2708
200k
    td->read_coeffs_tx_inter_block_visit = av1_read_coeffs_txb;
2709
200k
  }
2710
297k
  if (parse_decode_flag & 0x2) {
2711
174k
    td->predict_and_recon_intra_block_visit =
2712
174k
        predict_and_reconstruct_intra_block;
2713
174k
    td->inverse_tx_inter_block_visit = inverse_transform_inter_block;
2714
174k
    td->predict_inter_block_visit = predict_inter_block;
2715
174k
    td->cfl_store_inter_block_visit = cfl_store_inter_block;
2716
174k
  }
2717
297k
}
2718
2719
static inline void decode_tile(AV1Decoder *pbi, ThreadData *const td,
2720
77.0k
                               int tile_row, int tile_col) {
2721
77.0k
  TileInfo tile_info;
2722
2723
77.0k
  AV1_COMMON *const cm = &pbi->common;
2724
77.0k
  const int num_planes = av1_num_planes(cm);
2725
2726
77.0k
  av1_tile_set_row(&tile_info, cm, tile_row);
2727
77.0k
  av1_tile_set_col(&tile_info, cm, tile_col);
2728
77.0k
  DecoderCodingBlock *const dcb = &td->dcb;
2729
77.0k
  MACROBLOCKD *const xd = &dcb->xd;
2730
2731
77.0k
  av1_zero_above_context(cm, xd, tile_info.mi_col_start, tile_info.mi_col_end,
2732
77.0k
                         tile_row);
2733
77.0k
  av1_reset_loop_filter_delta(xd, num_planes);
2734
77.0k
  av1_reset_loop_restoration(xd, num_planes);
2735
2736
183k
  for (int mi_row = tile_info.mi_row_start; mi_row < tile_info.mi_row_end;
2737
140k
       mi_row += cm->seq_params->mib_size) {
2738
140k
    av1_zero_left_context(xd);
2739
2740
762k
    for (int mi_col = tile_info.mi_col_start; mi_col < tile_info.mi_col_end;
2741
655k
         mi_col += cm->seq_params->mib_size) {
2742
655k
      set_cb_buffer(pbi, dcb, &td->cb_buffer_base, num_planes, 0, 0);
2743
2744
      // Bit-stream parsing and decoding of the superblock
2745
655k
      decode_partition(pbi, td, mi_row, mi_col, td->bit_reader,
2746
655k
                       cm->seq_params->sb_size, 0x3);
2747
2748
655k
      if (aom_reader_has_overflowed(td->bit_reader)) {
2749
33.8k
        aom_merge_corrupted_flag(&dcb->corrupted, 1);
2750
33.8k
        return;
2751
33.8k
      }
2752
655k
    }
2753
140k
  }
2754
2755
43.1k
  int corrupted =
2756
43.1k
      (check_trailing_bits_after_symbol_coder(td->bit_reader)) ? 1 : 0;
2757
43.1k
  aom_merge_corrupted_flag(&dcb->corrupted, corrupted);
2758
43.1k
}
2759
2760
static const uint8_t *decode_tiles(AV1Decoder *pbi, const uint8_t *data,
2761
                                   const uint8_t *data_end, int start_tile,
2762
80.5k
                                   int end_tile) {
2763
80.5k
  AV1_COMMON *const cm = &pbi->common;
2764
80.5k
  ThreadData *const td = &pbi->td;
2765
80.5k
  CommonTileParams *const tiles = &cm->tiles;
2766
80.5k
  const int tile_cols = tiles->cols;
2767
80.5k
  const int tile_rows = tiles->rows;
2768
80.5k
  const int n_tiles = tile_cols * tile_rows;
2769
80.5k
  TileBufferDec(*const tile_buffers)[MAX_TILE_COLS] = pbi->tile_buffers;
2770
80.5k
  const int dec_tile_row = AOMMIN(pbi->dec_tile_row, tile_rows);
2771
80.5k
  const int single_row = pbi->dec_tile_row >= 0;
2772
80.5k
  const int dec_tile_col = AOMMIN(pbi->dec_tile_col, tile_cols);
2773
80.5k
  const int single_col = pbi->dec_tile_col >= 0;
2774
80.5k
  int tile_rows_start;
2775
80.5k
  int tile_rows_end;
2776
80.5k
  int tile_cols_start;
2777
80.5k
  int tile_cols_end;
2778
80.5k
  int inv_col_order;
2779
80.5k
  int inv_row_order;
2780
80.5k
  int tile_row, tile_col;
2781
80.5k
  uint8_t allow_update_cdf;
2782
80.5k
  const uint8_t *raw_data_end = NULL;
2783
2784
80.5k
  if (tiles->large_scale) {
2785
14.8k
    tile_rows_start = single_row ? dec_tile_row : 0;
2786
14.8k
    tile_rows_end = single_row ? dec_tile_row + 1 : tile_rows;
2787
14.8k
    tile_cols_start = single_col ? dec_tile_col : 0;
2788
14.8k
    tile_cols_end = single_col ? tile_cols_start + 1 : tile_cols;
2789
14.8k
    inv_col_order = pbi->inv_tile_order && !single_col;
2790
14.8k
    inv_row_order = pbi->inv_tile_order && !single_row;
2791
14.8k
    allow_update_cdf = 0;
2792
65.6k
  } else {
2793
65.6k
    tile_rows_start = 0;
2794
65.6k
    tile_rows_end = tile_rows;
2795
65.6k
    tile_cols_start = 0;
2796
65.6k
    tile_cols_end = tile_cols;
2797
65.6k
    inv_col_order = pbi->inv_tile_order;
2798
65.6k
    inv_row_order = pbi->inv_tile_order;
2799
65.6k
    allow_update_cdf = 1;
2800
65.6k
  }
2801
2802
  // No tiles to decode.
2803
80.5k
  if (tile_rows_end <= tile_rows_start || tile_cols_end <= tile_cols_start ||
2804
      // First tile is larger than end_tile.
2805
80.5k
      tile_rows_start * tiles->cols + tile_cols_start > end_tile ||
2806
      // Last tile is smaller than start_tile.
2807
80.5k
      (tile_rows_end - 1) * tiles->cols + tile_cols_end - 1 < start_tile)
2808
0
    return data;
2809
2810
80.5k
  allow_update_cdf = allow_update_cdf && !cm->features.disable_cdf_update;
2811
2812
80.5k
  assert(tile_rows <= MAX_TILE_ROWS);
2813
80.5k
  assert(tile_cols <= MAX_TILE_COLS);
2814
2815
80.5k
#if EXT_TILE_DEBUG
2816
80.5k
  if (tiles->large_scale && !pbi->ext_tile_debug)
2817
0
    raw_data_end = get_ls_single_tile_buffer(pbi, data, tile_buffers);
2818
80.5k
  else if (tiles->large_scale && pbi->ext_tile_debug)
2819
14.8k
    raw_data_end = get_ls_tile_buffers(pbi, data, data_end, tile_buffers);
2820
65.6k
  else
2821
65.6k
#endif  // EXT_TILE_DEBUG
2822
65.6k
    get_tile_buffers(pbi, data, data_end, tile_buffers, start_tile, end_tile);
2823
2824
80.5k
  if (pbi->tile_data == NULL || n_tiles != pbi->allocated_tiles) {
2825
11.7k
    decoder_alloc_tile_data(pbi, n_tiles);
2826
11.7k
  }
2827
80.5k
  if (pbi->dcb.xd.seg_mask == NULL)
2828
80.5k
    CHECK_MEM_ERROR(cm, pbi->dcb.xd.seg_mask,
2829
80.5k
                    (uint8_t *)aom_memalign(
2830
80.5k
                        16, 2 * MAX_SB_SQUARE * sizeof(*pbi->dcb.xd.seg_mask)));
2831
#if CONFIG_ACCOUNTING
2832
  if (pbi->acct_enabled) {
2833
    aom_accounting_reset(&pbi->accounting);
2834
  }
2835
#endif
2836
2837
80.5k
  set_decode_func_pointers(&pbi->td, 0x3);
2838
2839
  // Load all tile information into thread_data.
2840
80.5k
  td->dcb = pbi->dcb;
2841
2842
80.5k
  td->dcb.corrupted = 0;
2843
80.5k
  td->dcb.mc_buf[0] = td->mc_buf[0];
2844
80.5k
  td->dcb.mc_buf[1] = td->mc_buf[1];
2845
80.5k
  td->dcb.xd.tmp_conv_dst = td->tmp_conv_dst;
2846
234k
  for (int j = 0; j < 2; ++j) {
2847
153k
    td->dcb.xd.tmp_obmc_bufs[j] = td->tmp_obmc_bufs[j];
2848
153k
  }
2849
2850
157k
  for (tile_row = tile_rows_start; tile_row < tile_rows_end; ++tile_row) {
2851
76.9k
    const int row = inv_row_order ? tile_rows - 1 - tile_row : tile_row;
2852
2853
154k
    for (tile_col = tile_cols_start; tile_col < tile_cols_end; ++tile_col) {
2854
77.3k
      const int col = inv_col_order ? tile_cols - 1 - tile_col : tile_col;
2855
77.3k
      TileDataDec *const tile_data = pbi->tile_data + row * tiles->cols + col;
2856
77.3k
      const TileBufferDec *const tile_bs_buf = &tile_buffers[row][col];
2857
2858
77.3k
      if (row * tiles->cols + col < start_tile ||
2859
77.3k
          row * tiles->cols + col > end_tile)
2860
33
        continue;
2861
2862
77.3k
      td->bit_reader = &tile_data->bit_reader;
2863
77.3k
      av1_zero(td->cb_buffer_base.dqcoeff);
2864
77.3k
      av1_tile_init(&td->dcb.xd.tile, cm, row, col);
2865
77.3k
      td->dcb.xd.current_base_qindex = cm->quant_params.base_qindex;
2866
77.3k
      setup_bool_decoder(&td->dcb.xd, tile_bs_buf->data, data_end,
2867
77.3k
                         tile_bs_buf->size, &pbi->error, td->bit_reader,
2868
77.3k
                         allow_update_cdf);
2869
#if CONFIG_ACCOUNTING
2870
      if (pbi->acct_enabled) {
2871
        td->bit_reader->accounting = &pbi->accounting;
2872
        td->bit_reader->accounting->last_tell_frac =
2873
            aom_reader_tell_frac(td->bit_reader);
2874
      } else {
2875
        td->bit_reader->accounting = NULL;
2876
      }
2877
#endif
2878
77.3k
      av1_init_macroblockd(cm, &td->dcb.xd);
2879
77.3k
      av1_init_above_context(&cm->above_contexts, av1_num_planes(cm), row,
2880
77.3k
                             &td->dcb.xd);
2881
2882
      // Initialise the tile context from the frame context
2883
77.3k
      tile_data->tctx = *cm->fc;
2884
77.3k
      td->dcb.xd.tile_ctx = &tile_data->tctx;
2885
2886
      // decode tile
2887
77.3k
      decode_tile(pbi, td, row, col);
2888
77.3k
      aom_merge_corrupted_flag(&pbi->dcb.corrupted, td->dcb.corrupted);
2889
77.3k
      if (pbi->dcb.corrupted)
2890
36.8k
        aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
2891
36.8k
                           "Failed to decode tile data");
2892
77.3k
    }
2893
76.9k
  }
2894
2895
80.5k
  if (tiles->large_scale) {
2896
7.32k
    if (n_tiles == 1) {
2897
      // Find the end of the single tile buffer
2898
7.31k
      return aom_reader_find_end(&pbi->tile_data->bit_reader);
2899
7.31k
    }
2900
    // Return the end of the last tile buffer
2901
1
    return raw_data_end;
2902
7.32k
  }
2903
73.2k
  TileDataDec *const tile_data = pbi->tile_data + end_tile;
2904
2905
73.2k
  return aom_reader_find_end(&tile_data->bit_reader);
2906
80.5k
}
2907
2908
185k
static TileJobsDec *get_dec_job_info(AV1DecTileMT *tile_mt_info) {
2909
185k
  TileJobsDec *cur_job_info = NULL;
2910
185k
#if CONFIG_MULTITHREAD
2911
185k
  pthread_mutex_lock(tile_mt_info->job_mutex);
2912
2913
185k
  if (tile_mt_info->jobs_dequeued < tile_mt_info->jobs_enqueued) {
2914
88.1k
    cur_job_info = tile_mt_info->job_queue + tile_mt_info->jobs_dequeued;
2915
88.1k
    tile_mt_info->jobs_dequeued++;
2916
88.1k
  }
2917
2918
185k
  pthread_mutex_unlock(tile_mt_info->job_mutex);
2919
#else
2920
  (void)tile_mt_info;
2921
#endif
2922
185k
  return cur_job_info;
2923
185k
}
2924
2925
static inline void tile_worker_hook_init(AV1Decoder *const pbi,
2926
                                         DecWorkerData *const thread_data,
2927
                                         const TileBufferDec *const tile_buffer,
2928
                                         TileDataDec *const tile_data,
2929
88.1k
                                         uint8_t allow_update_cdf) {
2930
88.1k
  AV1_COMMON *cm = &pbi->common;
2931
88.1k
  ThreadData *const td = thread_data->td;
2932
88.1k
  int tile_row = tile_data->tile_info.tile_row;
2933
88.1k
  int tile_col = tile_data->tile_info.tile_col;
2934
2935
88.1k
  td->bit_reader = &tile_data->bit_reader;
2936
88.1k
  av1_zero(td->cb_buffer_base.dqcoeff);
2937
2938
88.1k
  MACROBLOCKD *const xd = &td->dcb.xd;
2939
88.1k
  av1_tile_init(&xd->tile, cm, tile_row, tile_col);
2940
88.1k
  xd->current_base_qindex = cm->quant_params.base_qindex;
2941
2942
88.1k
  setup_bool_decoder(xd, tile_buffer->data, thread_data->data_end,
2943
88.1k
                     tile_buffer->size, &thread_data->error_info,
2944
88.1k
                     td->bit_reader, allow_update_cdf);
2945
#if CONFIG_ACCOUNTING
2946
  if (pbi->acct_enabled) {
2947
    td->bit_reader->accounting = &pbi->accounting;
2948
    td->bit_reader->accounting->last_tell_frac =
2949
        aom_reader_tell_frac(td->bit_reader);
2950
  } else {
2951
    td->bit_reader->accounting = NULL;
2952
  }
2953
#endif
2954
88.1k
  av1_init_macroblockd(cm, xd);
2955
88.1k
  xd->error_info = &thread_data->error_info;
2956
88.1k
  av1_init_above_context(&cm->above_contexts, av1_num_planes(cm), tile_row, xd);
2957
2958
  // Initialise the tile context from the frame context
2959
88.1k
  tile_data->tctx = *cm->fc;
2960
88.1k
  xd->tile_ctx = &tile_data->tctx;
2961
#if CONFIG_ACCOUNTING
2962
  if (pbi->acct_enabled) {
2963
    tile_data->bit_reader.accounting->last_tell_frac =
2964
        aom_reader_tell_frac(&tile_data->bit_reader);
2965
  }
2966
#endif
2967
88.1k
}
2968
2969
0
static int tile_worker_hook(void *arg1, void *arg2) {
2970
0
  DecWorkerData *const thread_data = (DecWorkerData *)arg1;
2971
0
  AV1Decoder *const pbi = (AV1Decoder *)arg2;
2972
0
  AV1_COMMON *cm = &pbi->common;
2973
0
  ThreadData *const td = thread_data->td;
2974
0
  uint8_t allow_update_cdf;
2975
2976
  // The jmp_buf is valid only for the duration of the function that calls
2977
  // setjmp(). Therefore, this function must reset the 'setjmp' field to 0
2978
  // before it returns.
2979
0
  if (setjmp(thread_data->error_info.jmp)) {
2980
0
    thread_data->error_info.setjmp = 0;
2981
0
    thread_data->td->dcb.corrupted = 1;
2982
0
    return 0;
2983
0
  }
2984
0
  thread_data->error_info.setjmp = 1;
2985
2986
0
  allow_update_cdf = cm->tiles.large_scale ? 0 : 1;
2987
0
  allow_update_cdf = allow_update_cdf && !cm->features.disable_cdf_update;
2988
2989
0
  set_decode_func_pointers(td, 0x3);
2990
2991
0
  assert(cm->tiles.cols > 0);
2992
0
  while (!td->dcb.corrupted) {
2993
0
    TileJobsDec *cur_job_info = get_dec_job_info(&pbi->tile_mt_info);
2994
2995
0
    if (cur_job_info != NULL) {
2996
0
      const TileBufferDec *const tile_buffer = cur_job_info->tile_buffer;
2997
0
      TileDataDec *const tile_data = cur_job_info->tile_data;
2998
0
      tile_worker_hook_init(pbi, thread_data, tile_buffer, tile_data,
2999
0
                            allow_update_cdf);
3000
      // decode tile
3001
0
      int tile_row = tile_data->tile_info.tile_row;
3002
0
      int tile_col = tile_data->tile_info.tile_col;
3003
0
      decode_tile(pbi, td, tile_row, tile_col);
3004
0
    } else {
3005
0
      break;
3006
0
    }
3007
0
  }
3008
0
  thread_data->error_info.setjmp = 0;
3009
0
  return !td->dcb.corrupted;
3010
0
}
3011
3012
static inline int get_max_row_mt_workers_per_tile(AV1_COMMON *cm,
3013
213k
                                                  const TileInfo *tile) {
3014
  // NOTE: Currently value of max workers is calculated based
3015
  // on the parse and decode time. As per the theoretical estimate
3016
  // when percentage of parse time is equal to percentage of decode
3017
  // time, number of workers needed to parse + decode a tile can not
3018
  // exceed more than 2.
3019
  // TODO(any): Modify this value if parsing is optimized in future.
3020
213k
  int sb_rows = av1_get_sb_rows_in_tile(cm, tile);
3021
213k
  int max_workers =
3022
213k
      sb_rows == 1 ? AOM_MIN_THREADS_PER_TILE : AOM_MAX_THREADS_PER_TILE;
3023
213k
  return max_workers;
3024
213k
}
3025
3026
// The caller must hold pbi->row_mt_mutex_ when calling this function.
3027
// Returns 1 if either the next job is stored in *next_job_info or 1 is stored
3028
// in *end_of_frame.
3029
// NOTE: The caller waits on pbi->row_mt_cond_ if this function returns 0.
3030
// The return value of this function depends on the following variables:
3031
// - frame_row_mt_info->mi_rows_parse_done
3032
// - frame_row_mt_info->mi_rows_decode_started
3033
// - frame_row_mt_info->row_mt_exit
3034
// Therefore we may need to signal or broadcast pbi->row_mt_cond_ if any of
3035
// these variables is modified.
3036
static int get_next_job_info(AV1Decoder *const pbi,
3037
                             AV1DecRowMTJobInfo *next_job_info,
3038
253k
                             int *end_of_frame) {
3039
253k
  AV1_COMMON *cm = &pbi->common;
3040
253k
  TileDataDec *tile_data;
3041
253k
  AV1DecRowMTSync *dec_row_mt_sync;
3042
253k
  AV1DecRowMTInfo *frame_row_mt_info = &pbi->frame_row_mt_info;
3043
253k
  const int tile_rows_start = frame_row_mt_info->tile_rows_start;
3044
253k
  const int tile_rows_end = frame_row_mt_info->tile_rows_end;
3045
253k
  const int tile_cols_start = frame_row_mt_info->tile_cols_start;
3046
253k
  const int tile_cols_end = frame_row_mt_info->tile_cols_end;
3047
253k
  const int start_tile = frame_row_mt_info->start_tile;
3048
253k
  const int end_tile = frame_row_mt_info->end_tile;
3049
253k
  const int sb_mi_size = mi_size_wide[cm->seq_params->sb_size];
3050
253k
  int num_mis_to_decode, num_threads_working;
3051
253k
  int num_mis_waiting_for_decode;
3052
253k
  int min_threads_working = INT_MAX;
3053
253k
  int max_mis_to_decode = 0;
3054
253k
  int tile_row_idx, tile_col_idx;
3055
253k
  int tile_row = -1;
3056
253k
  int tile_col = -1;
3057
3058
253k
  memset(next_job_info, 0, sizeof(*next_job_info));
3059
3060
  // Frame decode is completed or error is encountered.
3061
253k
  *end_of_frame = (frame_row_mt_info->mi_rows_decode_started ==
3062
253k
                   frame_row_mt_info->mi_rows_to_decode) ||
3063
253k
                  (frame_row_mt_info->row_mt_exit == 1);
3064
253k
  if (*end_of_frame) {
3065
97.8k
    return 1;
3066
97.8k
  }
3067
3068
  // Decoding cannot start as bit-stream parsing is not complete.
3069
155k
  assert(frame_row_mt_info->mi_rows_parse_done >=
3070
155k
         frame_row_mt_info->mi_rows_decode_started);
3071
155k
  if (frame_row_mt_info->mi_rows_parse_done ==
3072
155k
      frame_row_mt_info->mi_rows_decode_started)
3073
34.9k
    return 0;
3074
3075
  // Choose the tile to decode.
3076
249k
  for (tile_row_idx = tile_rows_start; tile_row_idx < tile_rows_end;
3077
128k
       ++tile_row_idx) {
3078
292k
    for (tile_col_idx = tile_cols_start; tile_col_idx < tile_cols_end;
3079
164k
         ++tile_col_idx) {
3080
164k
      if (tile_row_idx * cm->tiles.cols + tile_col_idx < start_tile ||
3081
164k
          tile_row_idx * cm->tiles.cols + tile_col_idx > end_tile)
3082
256
        continue;
3083
3084
163k
      tile_data = pbi->tile_data + tile_row_idx * cm->tiles.cols + tile_col_idx;
3085
163k
      dec_row_mt_sync = &tile_data->dec_row_mt_sync;
3086
3087
163k
      num_threads_working = dec_row_mt_sync->num_threads_working;
3088
163k
      num_mis_waiting_for_decode = (dec_row_mt_sync->mi_rows_parse_done -
3089
163k
                                    dec_row_mt_sync->mi_rows_decode_started) *
3090
163k
                                   dec_row_mt_sync->mi_cols;
3091
163k
      num_mis_to_decode =
3092
163k
          (dec_row_mt_sync->mi_rows - dec_row_mt_sync->mi_rows_decode_started) *
3093
163k
          dec_row_mt_sync->mi_cols;
3094
3095
163k
      assert(num_mis_to_decode >= num_mis_waiting_for_decode);
3096
3097
      // Pick the tile which has minimum number of threads working on it.
3098
163k
      if (num_mis_waiting_for_decode > 0) {
3099
123k
        if (num_threads_working < min_threads_working) {
3100
121k
          min_threads_working = num_threads_working;
3101
121k
          max_mis_to_decode = 0;
3102
121k
        }
3103
123k
        if (num_threads_working == min_threads_working &&
3104
123k
            num_mis_to_decode > max_mis_to_decode &&
3105
123k
            num_threads_working <
3106
121k
                get_max_row_mt_workers_per_tile(cm, &tile_data->tile_info)) {
3107
120k
          max_mis_to_decode = num_mis_to_decode;
3108
120k
          tile_row = tile_row_idx;
3109
120k
          tile_col = tile_col_idx;
3110
120k
        }
3111
123k
      }
3112
163k
    }
3113
128k
  }
3114
  // No job found to process
3115
120k
  if (tile_row == -1 || tile_col == -1) return 0;
3116
3117
120k
  tile_data = pbi->tile_data + tile_row * cm->tiles.cols + tile_col;
3118
120k
  dec_row_mt_sync = &tile_data->dec_row_mt_sync;
3119
3120
120k
  next_job_info->tile_row = tile_row;
3121
120k
  next_job_info->tile_col = tile_col;
3122
120k
  next_job_info->mi_row = dec_row_mt_sync->mi_rows_decode_started +
3123
120k
                          tile_data->tile_info.mi_row_start;
3124
3125
120k
  dec_row_mt_sync->num_threads_working++;
3126
120k
  dec_row_mt_sync->mi_rows_decode_started += sb_mi_size;
3127
120k
  frame_row_mt_info->mi_rows_decode_started += sb_mi_size;
3128
120k
  assert(frame_row_mt_info->mi_rows_parse_done >=
3129
120k
         frame_row_mt_info->mi_rows_decode_started);
3130
120k
#if CONFIG_MULTITHREAD
3131
120k
  if (frame_row_mt_info->mi_rows_decode_started ==
3132
120k
      frame_row_mt_info->mi_rows_to_decode) {
3133
59.1k
    pthread_cond_broadcast(pbi->row_mt_cond_);
3134
59.1k
  }
3135
120k
#endif
3136
3137
120k
  return 1;
3138
120k
}
3139
3140
static inline void signal_parse_sb_row_done(AV1Decoder *const pbi,
3141
                                            TileDataDec *const tile_data,
3142
128k
                                            const int sb_mi_size) {
3143
128k
  AV1DecRowMTInfo *frame_row_mt_info = &pbi->frame_row_mt_info;
3144
128k
#if CONFIG_MULTITHREAD
3145
128k
  pthread_mutex_lock(pbi->row_mt_mutex_);
3146
128k
#endif
3147
128k
  assert(frame_row_mt_info->mi_rows_parse_done >=
3148
128k
         frame_row_mt_info->mi_rows_decode_started);
3149
128k
  tile_data->dec_row_mt_sync.mi_rows_parse_done += sb_mi_size;
3150
128k
  frame_row_mt_info->mi_rows_parse_done += sb_mi_size;
3151
128k
#if CONFIG_MULTITHREAD
3152
  // A new decode job is available. Wake up one worker thread to handle the
3153
  // new decode job.
3154
  // NOTE: This assumes we bump mi_rows_parse_done and mi_rows_decode_started
3155
  // by the same increment (sb_mi_size).
3156
128k
  pthread_cond_signal(pbi->row_mt_cond_);
3157
128k
  pthread_mutex_unlock(pbi->row_mt_mutex_);
3158
128k
#endif
3159
128k
}
3160
3161
// This function is very similar to decode_tile(). It would be good to figure
3162
// out how to share code.
3163
static inline void parse_tile_row_mt(AV1Decoder *pbi, ThreadData *const td,
3164
88.0k
                                     TileDataDec *const tile_data) {
3165
88.0k
  AV1_COMMON *const cm = &pbi->common;
3166
88.0k
  const int sb_mi_size = mi_size_wide[cm->seq_params->sb_size];
3167
88.0k
  const int num_planes = av1_num_planes(cm);
3168
88.0k
  const TileInfo *const tile_info = &tile_data->tile_info;
3169
88.0k
  int tile_row = tile_info->tile_row;
3170
88.0k
  DecoderCodingBlock *const dcb = &td->dcb;
3171
88.0k
  MACROBLOCKD *const xd = &dcb->xd;
3172
3173
88.0k
  av1_zero_above_context(cm, xd, tile_info->mi_col_start, tile_info->mi_col_end,
3174
88.0k
                         tile_row);
3175
88.0k
  av1_reset_loop_filter_delta(xd, num_planes);
3176
88.0k
  av1_reset_loop_restoration(xd, num_planes);
3177
3178
218k
  for (int mi_row = tile_info->mi_row_start; mi_row < tile_info->mi_row_end;
3179
148k
       mi_row += cm->seq_params->mib_size) {
3180
148k
    av1_zero_left_context(xd);
3181
3182
582k
    for (int mi_col = tile_info->mi_col_start; mi_col < tile_info->mi_col_end;
3183
452k
         mi_col += cm->seq_params->mib_size) {
3184
452k
      set_cb_buffer(pbi, dcb, pbi->cb_buffer_base, num_planes, mi_row, mi_col);
3185
3186
      // Bit-stream parsing of the superblock
3187
452k
      decode_partition(pbi, td, mi_row, mi_col, td->bit_reader,
3188
452k
                       cm->seq_params->sb_size, 0x1);
3189
3190
452k
      if (aom_reader_has_overflowed(td->bit_reader)) {
3191
17.8k
        aom_merge_corrupted_flag(&dcb->corrupted, 1);
3192
17.8k
        return;
3193
17.8k
      }
3194
452k
    }
3195
130k
    signal_parse_sb_row_done(pbi, tile_data, sb_mi_size);
3196
130k
  }
3197
3198
70.1k
  int corrupted =
3199
70.1k
      (check_trailing_bits_after_symbol_coder(td->bit_reader)) ? 1 : 0;
3200
70.1k
  aom_merge_corrupted_flag(&dcb->corrupted, corrupted);
3201
70.1k
}
3202
3203
123k
static int row_mt_worker_hook(void *arg1, void *arg2) {
3204
123k
  DecWorkerData *const thread_data = (DecWorkerData *)arg1;
3205
123k
  AV1Decoder *const pbi = (AV1Decoder *)arg2;
3206
123k
  ThreadData *const td = thread_data->td;
3207
123k
  uint8_t allow_update_cdf;
3208
123k
  AV1DecRowMTInfo *frame_row_mt_info = &pbi->frame_row_mt_info;
3209
123k
  td->dcb.corrupted = 0;
3210
3211
  // The jmp_buf is valid only for the duration of the function that calls
3212
  // setjmp(). Therefore, this function must reset the 'setjmp' field to 0
3213
  // before it returns.
3214
123k
  if (setjmp(thread_data->error_info.jmp)) {
3215
2.11k
    thread_data->error_info.setjmp = 0;
3216
2.11k
    thread_data->td->dcb.corrupted = 1;
3217
2.11k
#if CONFIG_MULTITHREAD
3218
2.11k
    pthread_mutex_lock(pbi->row_mt_mutex_);
3219
2.11k
#endif
3220
2.11k
    frame_row_mt_info->row_mt_exit = 1;
3221
2.11k
#if CONFIG_MULTITHREAD
3222
2.11k
    pthread_cond_broadcast(pbi->row_mt_cond_);
3223
2.11k
    pthread_mutex_unlock(pbi->row_mt_mutex_);
3224
2.11k
#endif
3225
    // If any SB row (erroneous row) processed by a thread encounters an
3226
    // internal error, there is a need to indicate other threads that decoding
3227
    // of the erroneous row is complete. This ensures that other threads which
3228
    // wait upon the completion of SB's present in erroneous row are not waiting
3229
    // indefinitely.
3230
2.11k
    signal_decoding_done_for_erroneous_row(pbi, &thread_data->td->dcb.xd);
3231
2.11k
    return 0;
3232
2.11k
  }
3233
121k
  thread_data->error_info.setjmp = 1;
3234
3235
121k
  AV1_COMMON *cm = &pbi->common;
3236
121k
  allow_update_cdf = cm->tiles.large_scale ? 0 : 1;
3237
121k
  allow_update_cdf = allow_update_cdf && !cm->features.disable_cdf_update;
3238
3239
121k
  set_decode_func_pointers(td, 0x1);
3240
3241
121k
  assert(cm->tiles.cols > 0);
3242
211k
  while (!td->dcb.corrupted) {
3243
185k
    TileJobsDec *cur_job_info = get_dec_job_info(&pbi->tile_mt_info);
3244
3245
185k
    if (cur_job_info != NULL) {
3246
88.1k
      const TileBufferDec *const tile_buffer = cur_job_info->tile_buffer;
3247
88.1k
      TileDataDec *const tile_data = cur_job_info->tile_data;
3248
88.1k
      tile_worker_hook_init(pbi, thread_data, tile_buffer, tile_data,
3249
88.1k
                            allow_update_cdf);
3250
88.1k
#if CONFIG_MULTITHREAD
3251
88.1k
      pthread_mutex_lock(pbi->row_mt_mutex_);
3252
88.1k
#endif
3253
88.1k
      tile_data->dec_row_mt_sync.num_threads_working++;
3254
88.1k
#if CONFIG_MULTITHREAD
3255
88.1k
      pthread_mutex_unlock(pbi->row_mt_mutex_);
3256
88.1k
#endif
3257
      // decode tile
3258
88.1k
      parse_tile_row_mt(pbi, td, tile_data);
3259
88.1k
#if CONFIG_MULTITHREAD
3260
88.1k
      pthread_mutex_lock(pbi->row_mt_mutex_);
3261
88.1k
#endif
3262
88.1k
      tile_data->dec_row_mt_sync.num_threads_working--;
3263
88.1k
#if CONFIG_MULTITHREAD
3264
88.1k
      pthread_mutex_unlock(pbi->row_mt_mutex_);
3265
88.1k
#endif
3266
97.5k
    } else {
3267
97.5k
      break;
3268
97.5k
    }
3269
185k
  }
3270
3271
123k
  if (td->dcb.corrupted) {
3272
23.6k
    thread_data->error_info.setjmp = 0;
3273
23.6k
#if CONFIG_MULTITHREAD
3274
23.6k
    pthread_mutex_lock(pbi->row_mt_mutex_);
3275
23.6k
#endif
3276
23.6k
    frame_row_mt_info->row_mt_exit = 1;
3277
23.6k
#if CONFIG_MULTITHREAD
3278
23.6k
    pthread_cond_broadcast(pbi->row_mt_cond_);
3279
23.6k
    pthread_mutex_unlock(pbi->row_mt_mutex_);
3280
23.6k
#endif
3281
23.6k
    return 0;
3282
23.6k
  }
3283
3284
99.5k
  set_decode_func_pointers(td, 0x2);
3285
3286
220k
  while (1) {
3287
218k
    AV1DecRowMTJobInfo next_job_info;
3288
218k
    int end_of_frame = 0;
3289
3290
218k
#if CONFIG_MULTITHREAD
3291
218k
    pthread_mutex_lock(pbi->row_mt_mutex_);
3292
218k
#endif
3293
253k
    while (!get_next_job_info(pbi, &next_job_info, &end_of_frame)) {
3294
35.1k
#if CONFIG_MULTITHREAD
3295
35.1k
      pthread_cond_wait(pbi->row_mt_cond_, pbi->row_mt_mutex_);
3296
35.1k
#endif
3297
35.1k
    }
3298
218k
#if CONFIG_MULTITHREAD
3299
218k
    pthread_mutex_unlock(pbi->row_mt_mutex_);
3300
218k
#endif
3301
3302
218k
    if (end_of_frame) break;
3303
3304
120k
    int tile_row = next_job_info.tile_row;
3305
120k
    int tile_col = next_job_info.tile_col;
3306
120k
    int mi_row = next_job_info.mi_row;
3307
3308
120k
    TileDataDec *tile_data =
3309
120k
        pbi->tile_data + tile_row * cm->tiles.cols + tile_col;
3310
120k
    AV1DecRowMTSync *dec_row_mt_sync = &tile_data->dec_row_mt_sync;
3311
3312
120k
    av1_tile_init(&td->dcb.xd.tile, cm, tile_row, tile_col);
3313
120k
    av1_init_macroblockd(cm, &td->dcb.xd);
3314
120k
    td->dcb.xd.error_info = &thread_data->error_info;
3315
3316
120k
    decode_tile_sb_row(pbi, td, &tile_data->tile_info, mi_row);
3317
3318
120k
#if CONFIG_MULTITHREAD
3319
120k
    pthread_mutex_lock(pbi->row_mt_mutex_);
3320
120k
#endif
3321
120k
    dec_row_mt_sync->num_threads_working--;
3322
120k
#if CONFIG_MULTITHREAD
3323
120k
    pthread_mutex_unlock(pbi->row_mt_mutex_);
3324
120k
#endif
3325
120k
  }
3326
99.5k
  thread_data->error_info.setjmp = 0;
3327
99.5k
  return !td->dcb.corrupted;
3328
123k
}
3329
3330
// sorts in descending order
3331
86.3k
static int compare_tile_buffers(const void *a, const void *b) {
3332
86.3k
  const TileJobsDec *const buf1 = (const TileJobsDec *)a;
3333
86.3k
  const TileJobsDec *const buf2 = (const TileJobsDec *)b;
3334
86.3k
  return (((int)buf2->tile_buffer->size) - ((int)buf1->tile_buffer->size));
3335
86.3k
}
3336
3337
static inline void enqueue_tile_jobs(AV1Decoder *pbi, AV1_COMMON *cm,
3338
                                     int tile_rows_start, int tile_rows_end,
3339
                                     int tile_cols_start, int tile_cols_end,
3340
65.9k
                                     int start_tile, int end_tile) {
3341
65.9k
  AV1DecTileMT *tile_mt_info = &pbi->tile_mt_info;
3342
65.9k
  TileJobsDec *tile_job_queue = tile_mt_info->job_queue;
3343
65.9k
  tile_mt_info->jobs_enqueued = 0;
3344
65.9k
  tile_mt_info->jobs_dequeued = 0;
3345
3346
151k
  for (int row = tile_rows_start; row < tile_rows_end; row++) {
3347
177k
    for (int col = tile_cols_start; col < tile_cols_end; col++) {
3348
92.3k
      if (row * cm->tiles.cols + col < start_tile ||
3349
92.3k
          row * cm->tiles.cols + col > end_tile)
3350
154
        continue;
3351
92.1k
      tile_job_queue->tile_buffer = &pbi->tile_buffers[row][col];
3352
92.1k
      tile_job_queue->tile_data = pbi->tile_data + row * cm->tiles.cols + col;
3353
92.1k
      tile_job_queue++;
3354
92.1k
      tile_mt_info->jobs_enqueued++;
3355
92.1k
    }
3356
85.3k
  }
3357
65.9k
}
3358
3359
static inline void alloc_dec_jobs(AV1DecTileMT *tile_mt_info, AV1_COMMON *cm,
3360
4.80k
                                  int tile_rows, int tile_cols) {
3361
4.80k
  tile_mt_info->alloc_tile_rows = tile_rows;
3362
4.80k
  tile_mt_info->alloc_tile_cols = tile_cols;
3363
4.80k
  int num_tiles = tile_rows * tile_cols;
3364
4.80k
#if CONFIG_MULTITHREAD
3365
4.80k
  {
3366
4.80k
    CHECK_MEM_ERROR(cm, tile_mt_info->job_mutex,
3367
4.80k
                    aom_malloc(sizeof(*tile_mt_info->job_mutex) * num_tiles));
3368
3369
32.5k
    for (int i = 0; i < num_tiles; i++) {
3370
27.7k
      pthread_mutex_init(&tile_mt_info->job_mutex[i], NULL);
3371
27.7k
    }
3372
4.80k
  }
3373
4.80k
#endif
3374
4.80k
  CHECK_MEM_ERROR(cm, tile_mt_info->job_queue,
3375
4.80k
                  aom_malloc(sizeof(*tile_mt_info->job_queue) * num_tiles));
3376
4.80k
}
3377
3378
306k
void av1_free_mc_tmp_buf(ThreadData *thread_data) {
3379
306k
  int ref;
3380
919k
  for (ref = 0; ref < 2; ref++) {
3381
613k
    if (thread_data->mc_buf_use_highbd)
3382
160k
      aom_free(CONVERT_TO_SHORTPTR(thread_data->mc_buf[ref]));
3383
452k
    else
3384
452k
      aom_free(thread_data->mc_buf[ref]);
3385
613k
    thread_data->mc_buf[ref] = NULL;
3386
613k
  }
3387
306k
  thread_data->mc_buf_size = 0;
3388
306k
  thread_data->mc_buf_use_highbd = 0;
3389
3390
306k
  aom_free(thread_data->tmp_conv_dst);
3391
306k
  thread_data->tmp_conv_dst = NULL;
3392
306k
  aom_free(thread_data->seg_mask);
3393
306k
  thread_data->seg_mask = NULL;
3394
919k
  for (int i = 0; i < 2; ++i) {
3395
613k
    aom_free(thread_data->tmp_obmc_bufs[i]);
3396
613k
    thread_data->tmp_obmc_bufs[i] = NULL;
3397
613k
  }
3398
306k
}
3399
3400
static inline void allocate_mc_tmp_buf(AV1_COMMON *const cm,
3401
                                       ThreadData *thread_data, int buf_size,
3402
164k
                                       int use_highbd) {
3403
492k
  for (int ref = 0; ref < 2; ref++) {
3404
    // The mc_buf/hbd_mc_buf must be zeroed to fix a intermittent valgrind error
3405
    // 'Conditional jump or move depends on uninitialised value' from the loop
3406
    // filter. Uninitialized reads in convolve function (e.g. horiz_4tap path in
3407
    // av1_convolve_2d_sr_avx2()) from mc_buf/hbd_mc_buf are seen to be the
3408
    // potential reason for this issue.
3409
328k
    if (use_highbd) {
3410
160k
      uint16_t *hbd_mc_buf;
3411
160k
      CHECK_MEM_ERROR(cm, hbd_mc_buf, (uint16_t *)aom_memalign(16, buf_size));
3412
160k
      memset(hbd_mc_buf, 0, buf_size);
3413
160k
      thread_data->mc_buf[ref] = CONVERT_TO_BYTEPTR(hbd_mc_buf);
3414
167k
    } else {
3415
167k
      CHECK_MEM_ERROR(cm, thread_data->mc_buf[ref],
3416
167k
                      (uint8_t *)aom_memalign(16, buf_size));
3417
167k
      memset(thread_data->mc_buf[ref], 0, buf_size);
3418
167k
    }
3419
328k
  }
3420
164k
  thread_data->mc_buf_size = buf_size;
3421
164k
  thread_data->mc_buf_use_highbd = use_highbd;
3422
3423
164k
  CHECK_MEM_ERROR(cm, thread_data->tmp_conv_dst,
3424
164k
                  aom_memalign(32, MAX_SB_SIZE * MAX_SB_SIZE *
3425
164k
                                       sizeof(*thread_data->tmp_conv_dst)));
3426
164k
  CHECK_MEM_ERROR(cm, thread_data->seg_mask,
3427
164k
                  (uint8_t *)aom_memalign(
3428
164k
                      16, 2 * MAX_SB_SQUARE * sizeof(*thread_data->seg_mask)));
3429
3430
492k
  for (int i = 0; i < 2; ++i) {
3431
328k
    CHECK_MEM_ERROR(
3432
328k
        cm, thread_data->tmp_obmc_bufs[i],
3433
328k
        aom_memalign(16, 2 * MAX_MB_PLANE * MAX_SB_SQUARE *
3434
328k
                             sizeof(*thread_data->tmp_obmc_bufs[i])));
3435
328k
  }
3436
164k
}
3437
3438
static inline void reset_dec_workers(AV1Decoder *pbi, AVxWorkerHook worker_hook,
3439
65.9k
                                     int num_workers) {
3440
65.9k
  const AVxWorkerInterface *const winterface = aom_get_worker_interface();
3441
3442
  // Reset tile decoding hook
3443
189k
  for (int worker_idx = 0; worker_idx < num_workers; ++worker_idx) {
3444
123k
    AVxWorker *const worker = &pbi->tile_workers[worker_idx];
3445
123k
    DecWorkerData *const thread_data = pbi->thread_data + worker_idx;
3446
123k
    thread_data->td->dcb = pbi->dcb;
3447
123k
    thread_data->td->dcb.corrupted = 0;
3448
123k
    thread_data->td->dcb.mc_buf[0] = thread_data->td->mc_buf[0];
3449
123k
    thread_data->td->dcb.mc_buf[1] = thread_data->td->mc_buf[1];
3450
123k
    thread_data->td->dcb.xd.tmp_conv_dst = thread_data->td->tmp_conv_dst;
3451
123k
    if (worker_idx)
3452
57.6k
      thread_data->td->dcb.xd.seg_mask = thread_data->td->seg_mask;
3453
370k
    for (int j = 0; j < 2; ++j) {
3454
247k
      thread_data->td->dcb.xd.tmp_obmc_bufs[j] =
3455
247k
          thread_data->td->tmp_obmc_bufs[j];
3456
247k
    }
3457
123k
    winterface->sync(worker);
3458
3459
123k
    worker->hook = worker_hook;
3460
123k
    worker->data1 = thread_data;
3461
123k
    worker->data2 = pbi;
3462
123k
  }
3463
#if CONFIG_ACCOUNTING
3464
  if (pbi->acct_enabled) {
3465
    aom_accounting_reset(&pbi->accounting);
3466
  }
3467
#endif
3468
65.9k
}
3469
3470
static inline void launch_dec_workers(AV1Decoder *pbi, const uint8_t *data_end,
3471
65.9k
                                      int num_workers) {
3472
65.9k
  const AVxWorkerInterface *const winterface = aom_get_worker_interface();
3473
3474
189k
  for (int worker_idx = num_workers - 1; worker_idx >= 0; --worker_idx) {
3475
123k
    AVxWorker *const worker = &pbi->tile_workers[worker_idx];
3476
123k
    DecWorkerData *const thread_data = (DecWorkerData *)worker->data1;
3477
3478
123k
    thread_data->data_end = data_end;
3479
3480
123k
    worker->had_error = 0;
3481
123k
    if (worker_idx == 0) {
3482
65.9k
      winterface->execute(worker);
3483
65.9k
    } else {
3484
57.6k
      winterface->launch(worker);
3485
57.6k
    }
3486
123k
  }
3487
65.9k
}
3488
3489
65.9k
static inline void sync_dec_workers(AV1Decoder *pbi, int num_workers) {
3490
65.9k
  const AVxWorkerInterface *const winterface = aom_get_worker_interface();
3491
65.9k
  int corrupted = 0;
3492
3493
189k
  for (int worker_idx = num_workers; worker_idx > 0; --worker_idx) {
3494
123k
    AVxWorker *const worker = &pbi->tile_workers[worker_idx - 1];
3495
123k
    aom_merge_corrupted_flag(&corrupted, !winterface->sync(worker));
3496
123k
  }
3497
3498
65.9k
  pbi->dcb.corrupted = corrupted;
3499
65.9k
}
3500
3501
67.3k
static inline void decode_mt_init(AV1Decoder *pbi) {
3502
67.3k
  AV1_COMMON *const cm = &pbi->common;
3503
67.3k
  const AVxWorkerInterface *const winterface = aom_get_worker_interface();
3504
67.3k
  int worker_idx;
3505
3506
  // Create workers and thread_data
3507
67.3k
  if (pbi->num_workers == 0) {
3508
3.67k
    const int num_threads = pbi->max_threads;
3509
3.67k
    CHECK_MEM_ERROR(cm, pbi->tile_workers,
3510
3.67k
                    aom_malloc(num_threads * sizeof(*pbi->tile_workers)));
3511
3.67k
    CHECK_MEM_ERROR(cm, pbi->thread_data,
3512
3.67k
                    aom_calloc(num_threads, sizeof(*pbi->thread_data)));
3513
3514
132k
    for (worker_idx = 0; worker_idx < num_threads; ++worker_idx) {
3515
128k
      AVxWorker *const worker = &pbi->tile_workers[worker_idx];
3516
128k
      DecWorkerData *const thread_data = pbi->thread_data + worker_idx;
3517
3518
128k
      winterface->init(worker);
3519
128k
      worker->thread_name = "aom tile worker";
3520
128k
      if (worker_idx != 0 && !winterface->reset(worker)) {
3521
0
        aom_internal_error(&pbi->error, AOM_CODEC_ERROR,
3522
0
                           "Tile decoder thread creation failed");
3523
0
      }
3524
128k
      ++pbi->num_workers;
3525
3526
128k
      if (worker_idx != 0) {
3527
        // Allocate thread data.
3528
125k
        CHECK_MEM_ERROR(cm, thread_data->td,
3529
125k
                        aom_memalign(32, sizeof(*thread_data->td)));
3530
125k
        av1_zero(*thread_data->td);
3531
125k
      } else {
3532
        // Main thread acts as a worker and uses the thread data in pbi
3533
3.67k
        thread_data->td = &pbi->td;
3534
3.67k
      }
3535
128k
      thread_data->error_info.error_code = AOM_CODEC_OK;
3536
128k
      thread_data->error_info.setjmp = 0;
3537
128k
    }
3538
3.67k
  }
3539
67.3k
  const int use_highbd = cm->seq_params->use_highbitdepth;
3540
67.3k
  const int buf_size = MC_TEMP_BUF_PELS << use_highbd;
3541
2.65M
  for (worker_idx = 1; worker_idx < pbi->max_threads; ++worker_idx) {
3542
2.58M
    DecWorkerData *const thread_data = pbi->thread_data + worker_idx;
3543
2.58M
    if (thread_data->td->mc_buf_size != buf_size) {
3544
145k
      av1_free_mc_tmp_buf(thread_data->td);
3545
145k
      allocate_mc_tmp_buf(cm, thread_data->td, buf_size, use_highbd);
3546
145k
    }
3547
2.58M
  }
3548
67.3k
}
3549
3550
static inline void tile_mt_queue(AV1Decoder *pbi, int tile_cols, int tile_rows,
3551
                                 int tile_rows_start, int tile_rows_end,
3552
                                 int tile_cols_start, int tile_cols_end,
3553
65.9k
                                 int start_tile, int end_tile) {
3554
65.9k
  AV1_COMMON *const cm = &pbi->common;
3555
65.9k
  if (pbi->tile_mt_info.alloc_tile_cols != tile_cols ||
3556
65.9k
      pbi->tile_mt_info.alloc_tile_rows != tile_rows) {
3557
4.80k
    av1_dealloc_dec_jobs(&pbi->tile_mt_info);
3558
4.80k
    alloc_dec_jobs(&pbi->tile_mt_info, cm, tile_rows, tile_cols);
3559
4.80k
  }
3560
65.9k
  enqueue_tile_jobs(pbi, cm, tile_rows_start, tile_rows_end, tile_cols_start,
3561
65.9k
                    tile_cols_end, start_tile, end_tile);
3562
65.9k
  qsort(pbi->tile_mt_info.job_queue, pbi->tile_mt_info.jobs_enqueued,
3563
65.9k
        sizeof(pbi->tile_mt_info.job_queue[0]), compare_tile_buffers);
3564
65.9k
}
3565
3566
static const uint8_t *decode_tiles_mt(AV1Decoder *pbi, const uint8_t *data,
3567
                                      const uint8_t *data_end, int start_tile,
3568
0
                                      int end_tile) {
3569
0
  AV1_COMMON *const cm = &pbi->common;
3570
0
  CommonTileParams *const tiles = &cm->tiles;
3571
0
  const int tile_cols = tiles->cols;
3572
0
  const int tile_rows = tiles->rows;
3573
0
  const int n_tiles = tile_cols * tile_rows;
3574
0
  TileBufferDec(*const tile_buffers)[MAX_TILE_COLS] = pbi->tile_buffers;
3575
0
  const int dec_tile_row = AOMMIN(pbi->dec_tile_row, tile_rows);
3576
0
  const int single_row = pbi->dec_tile_row >= 0;
3577
0
  const int dec_tile_col = AOMMIN(pbi->dec_tile_col, tile_cols);
3578
0
  const int single_col = pbi->dec_tile_col >= 0;
3579
0
  int tile_rows_start;
3580
0
  int tile_rows_end;
3581
0
  int tile_cols_start;
3582
0
  int tile_cols_end;
3583
0
  int tile_count_tg;
3584
0
  int num_workers;
3585
0
  const uint8_t *raw_data_end = NULL;
3586
3587
0
  if (tiles->large_scale) {
3588
0
    tile_rows_start = single_row ? dec_tile_row : 0;
3589
0
    tile_rows_end = single_row ? dec_tile_row + 1 : tile_rows;
3590
0
    tile_cols_start = single_col ? dec_tile_col : 0;
3591
0
    tile_cols_end = single_col ? tile_cols_start + 1 : tile_cols;
3592
0
  } else {
3593
0
    tile_rows_start = 0;
3594
0
    tile_rows_end = tile_rows;
3595
0
    tile_cols_start = 0;
3596
0
    tile_cols_end = tile_cols;
3597
0
  }
3598
0
  tile_count_tg = end_tile - start_tile + 1;
3599
0
  num_workers = AOMMIN(pbi->max_threads, tile_count_tg);
3600
3601
  // No tiles to decode.
3602
0
  if (tile_rows_end <= tile_rows_start || tile_cols_end <= tile_cols_start ||
3603
      // First tile is larger than end_tile.
3604
0
      tile_rows_start * tile_cols + tile_cols_start > end_tile ||
3605
      // Last tile is smaller than start_tile.
3606
0
      (tile_rows_end - 1) * tile_cols + tile_cols_end - 1 < start_tile)
3607
0
    return data;
3608
3609
0
  assert(tile_rows <= MAX_TILE_ROWS);
3610
0
  assert(tile_cols <= MAX_TILE_COLS);
3611
0
  assert(tile_count_tg > 0);
3612
0
  assert(num_workers > 0);
3613
0
  assert(start_tile <= end_tile);
3614
0
  assert(start_tile >= 0 && end_tile < n_tiles);
3615
3616
0
  decode_mt_init(pbi);
3617
3618
  // get tile size in tile group
3619
0
#if EXT_TILE_DEBUG
3620
0
  if (tiles->large_scale) assert(pbi->ext_tile_debug == 1);
3621
0
  if (tiles->large_scale)
3622
0
    raw_data_end = get_ls_tile_buffers(pbi, data, data_end, tile_buffers);
3623
0
  else
3624
0
#endif  // EXT_TILE_DEBUG
3625
0
    get_tile_buffers(pbi, data, data_end, tile_buffers, start_tile, end_tile);
3626
3627
0
  if (pbi->tile_data == NULL || n_tiles != pbi->allocated_tiles) {
3628
0
    decoder_alloc_tile_data(pbi, n_tiles);
3629
0
  }
3630
0
  if (pbi->dcb.xd.seg_mask == NULL)
3631
0
    CHECK_MEM_ERROR(cm, pbi->dcb.xd.seg_mask,
3632
0
                    (uint8_t *)aom_memalign(
3633
0
                        16, 2 * MAX_SB_SQUARE * sizeof(*pbi->dcb.xd.seg_mask)));
3634
3635
0
  for (int row = 0; row < tile_rows; row++) {
3636
0
    for (int col = 0; col < tile_cols; col++) {
3637
0
      TileDataDec *tile_data = pbi->tile_data + row * tiles->cols + col;
3638
0
      av1_tile_init(&tile_data->tile_info, cm, row, col);
3639
0
    }
3640
0
  }
3641
3642
0
  tile_mt_queue(pbi, tile_cols, tile_rows, tile_rows_start, tile_rows_end,
3643
0
                tile_cols_start, tile_cols_end, start_tile, end_tile);
3644
3645
0
  reset_dec_workers(pbi, tile_worker_hook, num_workers);
3646
0
  launch_dec_workers(pbi, data_end, num_workers);
3647
0
  sync_dec_workers(pbi, num_workers);
3648
3649
0
  if (pbi->dcb.corrupted)
3650
0
    aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
3651
0
                       "Failed to decode tile data");
3652
3653
0
  if (tiles->large_scale) {
3654
0
    if (n_tiles == 1) {
3655
      // Find the end of the single tile buffer
3656
0
      return aom_reader_find_end(&pbi->tile_data->bit_reader);
3657
0
    }
3658
    // Return the end of the last tile buffer
3659
0
    return raw_data_end;
3660
0
  }
3661
0
  TileDataDec *const tile_data = pbi->tile_data + end_tile;
3662
3663
0
  return aom_reader_find_end(&tile_data->bit_reader);
3664
0
}
3665
3666
65.9k
static inline void dec_alloc_cb_buf(AV1Decoder *pbi) {
3667
65.9k
  AV1_COMMON *const cm = &pbi->common;
3668
65.9k
  int size = ((cm->mi_params.mi_rows >> cm->seq_params->mib_size_log2) + 1) *
3669
65.9k
             ((cm->mi_params.mi_cols >> cm->seq_params->mib_size_log2) + 1);
3670
3671
65.9k
  if (pbi->cb_buffer_alloc_size < size) {
3672
4.42k
    av1_dec_free_cb_buf(pbi);
3673
4.42k
    CHECK_MEM_ERROR(cm, pbi->cb_buffer_base,
3674
4.42k
                    aom_memalign(32, sizeof(*pbi->cb_buffer_base) * size));
3675
4.42k
    memset(pbi->cb_buffer_base, 0, sizeof(*pbi->cb_buffer_base) * size);
3676
4.42k
    pbi->cb_buffer_alloc_size = size;
3677
4.42k
  }
3678
65.9k
}
3679
3680
static inline void row_mt_frame_init(AV1Decoder *pbi, int tile_rows_start,
3681
                                     int tile_rows_end, int tile_cols_start,
3682
                                     int tile_cols_end, int start_tile,
3683
65.9k
                                     int end_tile, int max_sb_rows) {
3684
65.9k
  AV1_COMMON *const cm = &pbi->common;
3685
65.9k
  AV1DecRowMTInfo *frame_row_mt_info = &pbi->frame_row_mt_info;
3686
3687
65.9k
  frame_row_mt_info->tile_rows_start = tile_rows_start;
3688
65.9k
  frame_row_mt_info->tile_rows_end = tile_rows_end;
3689
65.9k
  frame_row_mt_info->tile_cols_start = tile_cols_start;
3690
65.9k
  frame_row_mt_info->tile_cols_end = tile_cols_end;
3691
65.9k
  frame_row_mt_info->start_tile = start_tile;
3692
65.9k
  frame_row_mt_info->end_tile = end_tile;
3693
65.9k
  frame_row_mt_info->mi_rows_to_decode = 0;
3694
65.9k
  frame_row_mt_info->mi_rows_parse_done = 0;
3695
65.9k
  frame_row_mt_info->mi_rows_decode_started = 0;
3696
65.9k
  frame_row_mt_info->row_mt_exit = 0;
3697
3698
151k
  for (int tile_row = tile_rows_start; tile_row < tile_rows_end; ++tile_row) {
3699
177k
    for (int tile_col = tile_cols_start; tile_col < tile_cols_end; ++tile_col) {
3700
92.3k
      if (tile_row * cm->tiles.cols + tile_col < start_tile ||
3701
92.3k
          tile_row * cm->tiles.cols + tile_col > end_tile)
3702
154
        continue;
3703
3704
92.1k
      TileDataDec *const tile_data =
3705
92.1k
          pbi->tile_data + tile_row * cm->tiles.cols + tile_col;
3706
92.1k
      const TileInfo *const tile_info = &tile_data->tile_info;
3707
3708
92.1k
      tile_data->dec_row_mt_sync.mi_rows_parse_done = 0;
3709
92.1k
      tile_data->dec_row_mt_sync.mi_rows_decode_started = 0;
3710
92.1k
      tile_data->dec_row_mt_sync.num_threads_working = 0;
3711
92.1k
      tile_data->dec_row_mt_sync.mi_rows =
3712
92.1k
          ALIGN_POWER_OF_TWO(tile_info->mi_row_end - tile_info->mi_row_start,
3713
92.1k
                             cm->seq_params->mib_size_log2);
3714
92.1k
      tile_data->dec_row_mt_sync.mi_cols =
3715
92.1k
          ALIGN_POWER_OF_TWO(tile_info->mi_col_end - tile_info->mi_col_start,
3716
92.1k
                             cm->seq_params->mib_size_log2);
3717
92.1k
      tile_data->dec_row_mt_sync.intrabc_extra_top_right_sb_delay =
3718
92.1k
          av1_get_intrabc_extra_top_right_sb_delay(cm);
3719
3720
92.1k
      frame_row_mt_info->mi_rows_to_decode +=
3721
92.1k
          tile_data->dec_row_mt_sync.mi_rows;
3722
3723
      // Initialize cur_sb_col to -1 for all SB rows.
3724
92.1k
      memset(tile_data->dec_row_mt_sync.cur_sb_col, -1,
3725
92.1k
             sizeof(*tile_data->dec_row_mt_sync.cur_sb_col) * max_sb_rows);
3726
92.1k
    }
3727
85.2k
  }
3728
3729
65.9k
#if CONFIG_MULTITHREAD
3730
65.9k
  if (pbi->row_mt_mutex_ == NULL) {
3731
3.45k
    CHECK_MEM_ERROR(cm, pbi->row_mt_mutex_,
3732
3.45k
                    aom_malloc(sizeof(*(pbi->row_mt_mutex_))));
3733
3.45k
    if (pbi->row_mt_mutex_) {
3734
3.45k
      pthread_mutex_init(pbi->row_mt_mutex_, NULL);
3735
3.45k
    }
3736
3.45k
  }
3737
3738
65.9k
  if (pbi->row_mt_cond_ == NULL) {
3739
3.45k
    CHECK_MEM_ERROR(cm, pbi->row_mt_cond_,
3740
3.45k
                    aom_malloc(sizeof(*(pbi->row_mt_cond_))));
3741
3.45k
    if (pbi->row_mt_cond_) {
3742
3.45k
      pthread_cond_init(pbi->row_mt_cond_, NULL);
3743
3.45k
    }
3744
3.45k
  }
3745
65.9k
#endif
3746
65.9k
}
3747
3748
static const uint8_t *decode_tiles_row_mt(AV1Decoder *pbi, const uint8_t *data,
3749
                                          const uint8_t *data_end,
3750
67.3k
                                          int start_tile, int end_tile) {
3751
67.3k
  AV1_COMMON *const cm = &pbi->common;
3752
67.3k
  CommonTileParams *const tiles = &cm->tiles;
3753
67.3k
  const int tile_cols = tiles->cols;
3754
67.3k
  const int tile_rows = tiles->rows;
3755
67.3k
  const int n_tiles = tile_cols * tile_rows;
3756
67.3k
  TileBufferDec(*const tile_buffers)[MAX_TILE_COLS] = pbi->tile_buffers;
3757
67.3k
  const int dec_tile_row = AOMMIN(pbi->dec_tile_row, tile_rows);
3758
67.3k
  const int single_row = pbi->dec_tile_row >= 0;
3759
67.3k
  const int dec_tile_col = AOMMIN(pbi->dec_tile_col, tile_cols);
3760
67.3k
  const int single_col = pbi->dec_tile_col >= 0;
3761
67.3k
  int tile_rows_start;
3762
67.3k
  int tile_rows_end;
3763
67.3k
  int tile_cols_start;
3764
67.3k
  int tile_cols_end;
3765
67.3k
  int tile_count_tg;
3766
67.3k
  int num_workers = 0;
3767
67.3k
  int max_threads;
3768
67.3k
  const uint8_t *raw_data_end = NULL;
3769
67.3k
  int max_sb_rows = 0;
3770
3771
67.3k
  if (tiles->large_scale) {
3772
4.10k
    tile_rows_start = single_row ? dec_tile_row : 0;
3773
4.10k
    tile_rows_end = single_row ? dec_tile_row + 1 : tile_rows;
3774
4.10k
    tile_cols_start = single_col ? dec_tile_col : 0;
3775
4.10k
    tile_cols_end = single_col ? tile_cols_start + 1 : tile_cols;
3776
63.2k
  } else {
3777
63.2k
    tile_rows_start = 0;
3778
63.2k
    tile_rows_end = tile_rows;
3779
63.2k
    tile_cols_start = 0;
3780
63.2k
    tile_cols_end = tile_cols;
3781
63.2k
  }
3782
67.3k
  tile_count_tg = end_tile - start_tile + 1;
3783
67.3k
  max_threads = pbi->max_threads;
3784
3785
  // No tiles to decode.
3786
67.3k
  if (tile_rows_end <= tile_rows_start || tile_cols_end <= tile_cols_start ||
3787
      // First tile is larger than end_tile.
3788
67.3k
      tile_rows_start * tile_cols + tile_cols_start > end_tile ||
3789
      // Last tile is smaller than start_tile.
3790
67.3k
      (tile_rows_end - 1) * tile_cols + tile_cols_end - 1 < start_tile)
3791
0
    return data;
3792
3793
67.3k
  assert(tile_rows <= MAX_TILE_ROWS);
3794
67.3k
  assert(tile_cols <= MAX_TILE_COLS);
3795
67.3k
  assert(tile_count_tg > 0);
3796
67.3k
  assert(max_threads > 0);
3797
67.3k
  assert(start_tile <= end_tile);
3798
67.3k
  assert(start_tile >= 0 && end_tile < n_tiles);
3799
3800
67.3k
  (void)tile_count_tg;
3801
3802
67.3k
  decode_mt_init(pbi);
3803
3804
  // get tile size in tile group
3805
67.3k
#if EXT_TILE_DEBUG
3806
67.3k
  if (tiles->large_scale) assert(pbi->ext_tile_debug == 1);
3807
67.3k
  if (tiles->large_scale)
3808
4.10k
    raw_data_end = get_ls_tile_buffers(pbi, data, data_end, tile_buffers);
3809
63.2k
  else
3810
63.2k
#endif  // EXT_TILE_DEBUG
3811
63.2k
    get_tile_buffers(pbi, data, data_end, tile_buffers, start_tile, end_tile);
3812
3813
67.3k
  if (pbi->tile_data == NULL || n_tiles != pbi->allocated_tiles) {
3814
4.80k
    if (pbi->tile_data != NULL) {
3815
6.90k
      for (int i = 0; i < pbi->allocated_tiles; i++) {
3816
5.55k
        TileDataDec *const tile_data = pbi->tile_data + i;
3817
5.55k
        av1_dec_row_mt_dealloc(&tile_data->dec_row_mt_sync);
3818
5.55k
      }
3819
1.34k
    }
3820
4.80k
    decoder_alloc_tile_data(pbi, n_tiles);
3821
4.80k
  }
3822
67.3k
  if (pbi->dcb.xd.seg_mask == NULL)
3823
67.3k
    CHECK_MEM_ERROR(cm, pbi->dcb.xd.seg_mask,
3824
67.3k
                    (uint8_t *)aom_memalign(
3825
67.3k
                        16, 2 * MAX_SB_SQUARE * sizeof(*pbi->dcb.xd.seg_mask)));
3826
3827
152k
  for (int row = 0; row < tile_rows; row++) {
3828
177k
    for (int col = 0; col < tile_cols; col++) {
3829
92.3k
      TileDataDec *tile_data = pbi->tile_data + row * tiles->cols + col;
3830
92.3k
      av1_tile_init(&tile_data->tile_info, cm, row, col);
3831
3832
92.3k
      max_sb_rows = AOMMAX(max_sb_rows,
3833
92.3k
                           av1_get_sb_rows_in_tile(cm, &tile_data->tile_info));
3834
92.3k
      num_workers += get_max_row_mt_workers_per_tile(cm, &tile_data->tile_info);
3835
92.3k
    }
3836
85.3k
  }
3837
67.3k
  num_workers = AOMMIN(num_workers, max_threads);
3838
3839
67.3k
  if (pbi->allocated_row_mt_sync_rows != max_sb_rows) {
3840
40.8k
    for (int i = 0; i < n_tiles; ++i) {
3841
31.9k
      TileDataDec *const tile_data = pbi->tile_data + i;
3842
31.9k
      av1_dec_row_mt_dealloc(&tile_data->dec_row_mt_sync);
3843
31.9k
      dec_row_mt_alloc(&tile_data->dec_row_mt_sync, cm, max_sb_rows);
3844
31.9k
    }
3845
8.97k
    pbi->allocated_row_mt_sync_rows = max_sb_rows;
3846
8.97k
  }
3847
3848
67.3k
  tile_mt_queue(pbi, tile_cols, tile_rows, tile_rows_start, tile_rows_end,
3849
67.3k
                tile_cols_start, tile_cols_end, start_tile, end_tile);
3850
3851
67.3k
  dec_alloc_cb_buf(pbi);
3852
3853
67.3k
  row_mt_frame_init(pbi, tile_rows_start, tile_rows_end, tile_cols_start,
3854
67.3k
                    tile_cols_end, start_tile, end_tile, max_sb_rows);
3855
3856
67.3k
  reset_dec_workers(pbi, row_mt_worker_hook, num_workers);
3857
67.3k
  launch_dec_workers(pbi, data_end, num_workers);
3858
67.3k
  sync_dec_workers(pbi, num_workers);
3859
3860
67.3k
  if (pbi->dcb.corrupted)
3861
6.76k
    aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
3862
6.76k
                       "Failed to decode tile data");
3863
3864
67.3k
  if (tiles->large_scale) {
3865
2.69k
    if (n_tiles == 1) {
3866
      // Find the end of the single tile buffer
3867
2.68k
      return aom_reader_find_end(&pbi->tile_data->bit_reader);
3868
2.68k
    }
3869
    // Return the end of the last tile buffer
3870
5
    return raw_data_end;
3871
2.69k
  }
3872
64.6k
  TileDataDec *const tile_data = pbi->tile_data + end_tile;
3873
3874
64.6k
  return aom_reader_find_end(&tile_data->bit_reader);
3875
67.3k
}
3876
3877
12.2k
static inline void error_handler(void *data) {
3878
12.2k
  AV1_COMMON *const cm = (AV1_COMMON *)data;
3879
12.2k
  aom_internal_error(cm->error, AOM_CODEC_CORRUPT_FRAME, "Truncated packet");
3880
12.2k
}
3881
3882
// Reads the high_bitdepth and twelve_bit fields in color_config() and sets
3883
// seq_params->bit_depth based on the values of those fields and
3884
// seq_params->profile. Reports errors by calling rb->error_handler() or
3885
// aom_internal_error().
3886
static inline void read_bitdepth(struct aom_read_bit_buffer *rb,
3887
                                 SequenceHeader *seq_params,
3888
76.7k
                                 struct aom_internal_error_info *error_info) {
3889
76.7k
  const int high_bitdepth = aom_rb_read_bit(rb);
3890
76.7k
  if (seq_params->profile == PROFILE_2 && high_bitdepth) {
3891
29.1k
    const int twelve_bit = aom_rb_read_bit(rb);
3892
29.1k
    seq_params->bit_depth = twelve_bit ? AOM_BITS_12 : AOM_BITS_10;
3893
47.6k
  } else if (seq_params->profile <= PROFILE_2) {
3894
47.5k
    seq_params->bit_depth = high_bitdepth ? AOM_BITS_10 : AOM_BITS_8;
3895
47.5k
  } else {
3896
57
    aom_internal_error(error_info, AOM_CODEC_UNSUP_BITSTREAM,
3897
57
                       "Unsupported profile/bit-depth combination");
3898
57
  }
3899
#if !CONFIG_AV1_HIGHBITDEPTH
3900
  if (seq_params->bit_depth > AOM_BITS_8) {
3901
    aom_internal_error(error_info, AOM_CODEC_UNSUP_BITSTREAM,
3902
                       "Bit-depth %d not supported", seq_params->bit_depth);
3903
  }
3904
#endif
3905
76.7k
}
3906
3907
static void read_film_grain_params(AV1_COMMON *cm,
3908
64.9k
                                   struct aom_read_bit_buffer *rb) {
3909
64.9k
  aom_film_grain_t *pars = &cm->film_grain_params;
3910
64.9k
  const SequenceHeader *const seq_params = cm->seq_params;
3911
3912
64.9k
  pars->apply_grain = aom_rb_read_bit(rb);
3913
64.9k
  if (!pars->apply_grain) {
3914
39.1k
    memset(pars, 0, sizeof(*pars));
3915
39.1k
    return;
3916
39.1k
  }
3917
3918
25.8k
  pars->random_seed = aom_rb_read_literal(rb, 16);
3919
25.8k
  if (cm->current_frame.frame_type == INTER_FRAME)
3920
930
    pars->update_parameters = aom_rb_read_bit(rb);
3921
24.9k
  else
3922
24.9k
    pars->update_parameters = 1;
3923
3924
25.8k
  pars->bit_depth = seq_params->bit_depth;
3925
3926
25.8k
  if (!pars->update_parameters) {
3927
    // inherit parameters from a previous reference frame
3928
893
    int film_grain_params_ref_idx = aom_rb_read_literal(rb, 3);
3929
    // Section 6.8.20: It is a requirement of bitstream conformance that
3930
    // film_grain_params_ref_idx is equal to ref_frame_idx[ j ] for some value
3931
    // of j in the range 0 to REFS_PER_FRAME - 1.
3932
893
    int found = 0;
3933
4.44k
    for (int i = 0; i < INTER_REFS_PER_FRAME; ++i) {
3934
4.38k
      if (film_grain_params_ref_idx == cm->remapped_ref_idx[i]) {
3935
832
        found = 1;
3936
832
        break;
3937
832
      }
3938
4.38k
    }
3939
893
    if (!found) {
3940
61
      aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
3941
61
                         "Invalid film grain reference idx %d. ref_frame_idx = "
3942
61
                         "{%d, %d, %d, %d, %d, %d, %d}",
3943
61
                         film_grain_params_ref_idx, cm->remapped_ref_idx[0],
3944
61
                         cm->remapped_ref_idx[1], cm->remapped_ref_idx[2],
3945
61
                         cm->remapped_ref_idx[3], cm->remapped_ref_idx[4],
3946
61
                         cm->remapped_ref_idx[5], cm->remapped_ref_idx[6]);
3947
61
    }
3948
893
    RefCntBuffer *const buf = cm->ref_frame_map[film_grain_params_ref_idx];
3949
893
    if (buf == NULL) {
3950
0
      aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
3951
0
                         "Invalid Film grain reference idx");
3952
0
    }
3953
893
    if (!buf->film_grain_params_present) {
3954
216
      aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
3955
216
                         "Film grain reference parameters not available");
3956
216
    }
3957
893
    uint16_t random_seed = pars->random_seed;
3958
893
    *pars = buf->film_grain_params;   // inherit paramaters
3959
893
    pars->random_seed = random_seed;  // with new random seed
3960
893
    return;
3961
893
  }
3962
3963
  // Scaling functions parameters
3964
24.9k
  pars->num_y_points = aom_rb_read_literal(rb, 4);  // max 14
3965
24.9k
  if (pars->num_y_points > 14)
3966
157
    aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
3967
157
                       "Number of points for film grain luma scaling function "
3968
157
                       "exceeds the maximum value.");
3969
61.8k
  for (int i = 0; i < pars->num_y_points; i++) {
3970
36.8k
    pars->scaling_points_y[i][0] = aom_rb_read_literal(rb, 8);
3971
36.8k
    if (i && pars->scaling_points_y[i - 1][0] >= pars->scaling_points_y[i][0])
3972
710
      aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
3973
710
                         "First coordinate of the scaling function points "
3974
710
                         "shall be increasing.");
3975
36.8k
    pars->scaling_points_y[i][1] = aom_rb_read_literal(rb, 8);
3976
36.8k
  }
3977
3978
24.9k
  if (!seq_params->monochrome)
3979
23.4k
    pars->chroma_scaling_from_luma = aom_rb_read_bit(rb);
3980
1.57k
  else
3981
1.57k
    pars->chroma_scaling_from_luma = 0;
3982
3983
24.9k
  if (seq_params->monochrome || pars->chroma_scaling_from_luma ||
3984
24.9k
      ((seq_params->subsampling_x == 1) && (seq_params->subsampling_y == 1) &&
3985
13.2k
       (pars->num_y_points == 0))) {
3986
13.2k
    pars->num_cb_points = 0;
3987
13.2k
    pars->num_cr_points = 0;
3988
13.2k
  } else {
3989
11.7k
    pars->num_cb_points = aom_rb_read_literal(rb, 4);  // max 10
3990
11.7k
    if (pars->num_cb_points > 10)
3991
166
      aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
3992
166
                         "Number of points for film grain cb scaling function "
3993
166
                         "exceeds the maximum value.");
3994
17.5k
    for (int i = 0; i < pars->num_cb_points; i++) {
3995
5.79k
      pars->scaling_points_cb[i][0] = aom_rb_read_literal(rb, 8);
3996
5.79k
      if (i &&
3997
5.79k
          pars->scaling_points_cb[i - 1][0] >= pars->scaling_points_cb[i][0])
3998
324
        aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
3999
324
                           "First coordinate of the scaling function points "
4000
324
                           "shall be increasing.");
4001
5.79k
      pars->scaling_points_cb[i][1] = aom_rb_read_literal(rb, 8);
4002
5.79k
    }
4003
4004
11.7k
    pars->num_cr_points = aom_rb_read_literal(rb, 4);  // max 10
4005
11.7k
    if (pars->num_cr_points > 10)
4006
74
      aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
4007
74
                         "Number of points for film grain cr scaling function "
4008
74
                         "exceeds the maximum value.");
4009
28.4k
    for (int i = 0; i < pars->num_cr_points; i++) {
4010
16.7k
      pars->scaling_points_cr[i][0] = aom_rb_read_literal(rb, 8);
4011
16.7k
      if (i &&
4012
16.7k
          pars->scaling_points_cr[i - 1][0] >= pars->scaling_points_cr[i][0])
4013
573
        aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
4014
573
                           "First coordinate of the scaling function points "
4015
573
                           "shall be increasing.");
4016
16.7k
      pars->scaling_points_cr[i][1] = aom_rb_read_literal(rb, 8);
4017
16.7k
    }
4018
4019
11.7k
    if ((seq_params->subsampling_x == 1) && (seq_params->subsampling_y == 1) &&
4020
11.7k
        (((pars->num_cb_points == 0) && (pars->num_cr_points != 0)) ||
4021
2.45k
         ((pars->num_cb_points != 0) && (pars->num_cr_points == 0))))
4022
15
      aom_internal_error(cm->error, AOM_CODEC_UNSUP_BITSTREAM,
4023
15
                         "In YCbCr 4:2:0, film grain shall be applied "
4024
15
                         "to both chroma components or neither.");
4025
11.7k
  }
4026
4027
24.9k
  pars->scaling_shift = aom_rb_read_literal(rb, 2) + 8;  // 8 + value
4028
4029
  // AR coefficients
4030
  // Only sent if the corresponsing scaling function has
4031
  // more than 0 points
4032
4033
24.9k
  pars->ar_coeff_lag = aom_rb_read_literal(rb, 2);
4034
4035
24.9k
  int num_pos_luma = 2 * pars->ar_coeff_lag * (pars->ar_coeff_lag + 1);
4036
24.9k
  int num_pos_chroma = num_pos_luma;
4037
24.9k
  if (pars->num_y_points > 0) ++num_pos_chroma;
4038
4039
24.9k
  if (pars->num_y_points)
4040
59.3k
    for (int i = 0; i < num_pos_luma; i++)
4041
40.9k
      pars->ar_coeffs_y[i] = aom_rb_read_literal(rb, 8) - 128;
4042
4043
24.9k
  if (pars->num_cb_points || pars->chroma_scaling_from_luma)
4044
66.2k
    for (int i = 0; i < num_pos_chroma; i++)
4045
52.5k
      pars->ar_coeffs_cb[i] = aom_rb_read_literal(rb, 8) - 128;
4046
4047
24.9k
  if (pars->num_cr_points || pars->chroma_scaling_from_luma)
4048
62.6k
    for (int i = 0; i < num_pos_chroma; i++)
4049
47.5k
      pars->ar_coeffs_cr[i] = aom_rb_read_literal(rb, 8) - 128;
4050
4051
24.9k
  pars->ar_coeff_shift = aom_rb_read_literal(rb, 2) + 6;  // 6 + value
4052
4053
24.9k
  pars->grain_scale_shift = aom_rb_read_literal(rb, 2);
4054
4055
24.9k
  if (pars->num_cb_points) {
4056
2.90k
    pars->cb_mult = aom_rb_read_literal(rb, 8);
4057
2.90k
    pars->cb_luma_mult = aom_rb_read_literal(rb, 8);
4058
2.90k
    pars->cb_offset = aom_rb_read_literal(rb, 9);
4059
2.90k
  }
4060
4061
24.9k
  if (pars->num_cr_points) {
4062
5.51k
    pars->cr_mult = aom_rb_read_literal(rb, 8);
4063
5.51k
    pars->cr_luma_mult = aom_rb_read_literal(rb, 8);
4064
5.51k
    pars->cr_offset = aom_rb_read_literal(rb, 9);
4065
5.51k
  }
4066
4067
24.9k
  pars->overlap_flag = aom_rb_read_bit(rb);
4068
4069
24.9k
  pars->clip_to_restricted_range = aom_rb_read_bit(rb);
4070
24.9k
}
4071
4072
static inline void read_film_grain(AV1_COMMON *cm,
4073
167k
                                   struct aom_read_bit_buffer *rb) {
4074
167k
  if (cm->seq_params->film_grain_params_present &&
4075
167k
      (cm->show_frame || cm->showable_frame)) {
4076
64.9k
    read_film_grain_params(cm, rb);
4077
102k
  } else {
4078
102k
    memset(&cm->film_grain_params, 0, sizeof(cm->film_grain_params));
4079
102k
  }
4080
167k
  cm->film_grain_params.bit_depth = cm->seq_params->bit_depth;
4081
167k
  cm->cur_frame->film_grain_params = cm->film_grain_params;
4082
167k
}
4083
4084
void av1_read_color_config(struct aom_read_bit_buffer *rb,
4085
                           int allow_lowbitdepth, SequenceHeader *seq_params,
4086
76.7k
                           struct aom_internal_error_info *error_info) {
4087
76.7k
  read_bitdepth(rb, seq_params, error_info);
4088
4089
76.7k
  seq_params->use_highbitdepth =
4090
76.7k
      seq_params->bit_depth > AOM_BITS_8 || !allow_lowbitdepth;
4091
  // monochrome bit (not needed for PROFILE_1)
4092
76.7k
  const int is_monochrome =
4093
76.7k
      seq_params->profile != PROFILE_1 ? aom_rb_read_bit(rb) : 0;
4094
76.7k
  seq_params->monochrome = is_monochrome;
4095
76.7k
  int color_description_present_flag = aom_rb_read_bit(rb);
4096
76.7k
  if (color_description_present_flag) {
4097
3.55k
    seq_params->color_primaries = aom_rb_read_literal(rb, 8);
4098
3.55k
    seq_params->transfer_characteristics = aom_rb_read_literal(rb, 8);
4099
3.55k
    seq_params->matrix_coefficients = aom_rb_read_literal(rb, 8);
4100
73.2k
  } else {
4101
73.2k
    seq_params->color_primaries = AOM_CICP_CP_UNSPECIFIED;
4102
73.2k
    seq_params->transfer_characteristics = AOM_CICP_TC_UNSPECIFIED;
4103
73.2k
    seq_params->matrix_coefficients = AOM_CICP_MC_UNSPECIFIED;
4104
73.2k
  }
4105
76.7k
  if (is_monochrome) {
4106
    // [16,235] (including xvycc) vs [0,255] range
4107
18.0k
    seq_params->color_range = aom_rb_read_bit(rb);
4108
18.0k
    seq_params->subsampling_y = seq_params->subsampling_x = 1;
4109
18.0k
    seq_params->chroma_sample_position = AOM_CSP_UNKNOWN;
4110
18.0k
    seq_params->separate_uv_delta_q = 0;
4111
18.0k
    return;
4112
18.0k
  }
4113
58.7k
  if (seq_params->color_primaries == AOM_CICP_CP_BT_709 &&
4114
58.7k
      seq_params->transfer_characteristics == AOM_CICP_TC_SRGB &&
4115
58.7k
      seq_params->matrix_coefficients == AOM_CICP_MC_IDENTITY) {
4116
2
    seq_params->subsampling_y = seq_params->subsampling_x = 0;
4117
2
    seq_params->color_range = 1;  // assume full color-range
4118
2
    if (!(seq_params->profile == PROFILE_1 ||
4119
2
          (seq_params->profile == PROFILE_2 &&
4120
2
           seq_params->bit_depth == AOM_BITS_12))) {
4121
2
      aom_internal_error(
4122
2
          error_info, AOM_CODEC_UNSUP_BITSTREAM,
4123
2
          "sRGB colorspace not compatible with specified profile");
4124
2
    }
4125
58.7k
  } else {
4126
    // [16,235] (including xvycc) vs [0,255] range
4127
58.7k
    seq_params->color_range = aom_rb_read_bit(rb);
4128
58.7k
    if (seq_params->profile == PROFILE_0) {
4129
      // 420 only
4130
28.1k
      seq_params->subsampling_x = seq_params->subsampling_y = 1;
4131
30.6k
    } else if (seq_params->profile == PROFILE_1) {
4132
      // 444 only
4133
9.88k
      seq_params->subsampling_x = seq_params->subsampling_y = 0;
4134
20.7k
    } else {
4135
20.7k
      assert(seq_params->profile == PROFILE_2);
4136
19.5k
      if (seq_params->bit_depth == AOM_BITS_12) {
4137
17.0k
        seq_params->subsampling_x = aom_rb_read_bit(rb);
4138
17.0k
        if (seq_params->subsampling_x)
4139
1.27k
          seq_params->subsampling_y = aom_rb_read_bit(rb);  // 422 or 420
4140
15.7k
        else
4141
15.7k
          seq_params->subsampling_y = 0;  // 444
4142
17.0k
      } else {
4143
        // 422
4144
2.50k
        seq_params->subsampling_x = 1;
4145
2.50k
        seq_params->subsampling_y = 0;
4146
2.50k
      }
4147
19.5k
    }
4148
57.5k
    if (seq_params->matrix_coefficients == AOM_CICP_MC_IDENTITY &&
4149
57.5k
        (seq_params->subsampling_x || seq_params->subsampling_y)) {
4150
10
      aom_internal_error(
4151
10
          error_info, AOM_CODEC_UNSUP_BITSTREAM,
4152
10
          "Identity CICP Matrix incompatible with non 4:4:4 color sampling");
4153
10
    }
4154
57.5k
    if (seq_params->subsampling_x && seq_params->subsampling_y) {
4155
28.4k
      seq_params->chroma_sample_position = aom_rb_read_literal(rb, 2);
4156
28.4k
    }
4157
57.5k
  }
4158
57.5k
  seq_params->separate_uv_delta_q = aom_rb_read_bit(rb);
4159
57.5k
}
4160
4161
void av1_read_timing_info_header(aom_timing_info_t *timing_info,
4162
                                 struct aom_internal_error_info *error,
4163
5.50k
                                 struct aom_read_bit_buffer *rb) {
4164
5.50k
  timing_info->num_units_in_display_tick =
4165
5.50k
      aom_rb_read_unsigned_literal(rb,
4166
5.50k
                                   32);  // Number of units in a display tick
4167
5.50k
  timing_info->time_scale = aom_rb_read_unsigned_literal(rb, 32);  // Time scale
4168
5.50k
  if (timing_info->num_units_in_display_tick == 0 ||
4169
5.50k
      timing_info->time_scale == 0) {
4170
288
    aom_internal_error(
4171
288
        error, AOM_CODEC_UNSUP_BITSTREAM,
4172
288
        "num_units_in_display_tick and time_scale must be greater than 0.");
4173
288
  }
4174
5.50k
  timing_info->equal_picture_interval =
4175
5.50k
      aom_rb_read_bit(rb);  // Equal picture interval bit
4176
5.50k
  if (timing_info->equal_picture_interval) {
4177
3.09k
    const uint32_t num_ticks_per_picture_minus_1 = aom_rb_read_uvlc(rb);
4178
3.09k
    if (num_ticks_per_picture_minus_1 == UINT32_MAX) {
4179
11
      aom_internal_error(
4180
11
          error, AOM_CODEC_UNSUP_BITSTREAM,
4181
11
          "num_ticks_per_picture_minus_1 cannot be (1 << 32) - 1.");
4182
11
    }
4183
3.09k
    timing_info->num_ticks_per_picture = num_ticks_per_picture_minus_1 + 1;
4184
3.09k
  }
4185
5.50k
}
4186
4187
void av1_read_decoder_model_info(aom_dec_model_info_t *decoder_model_info,
4188
1.15k
                                 struct aom_read_bit_buffer *rb) {
4189
1.15k
  decoder_model_info->encoder_decoder_buffer_delay_length =
4190
1.15k
      aom_rb_read_literal(rb, 5) + 1;
4191
1.15k
  decoder_model_info->num_units_in_decoding_tick =
4192
1.15k
      aom_rb_read_unsigned_literal(rb,
4193
1.15k
                                   32);  // Number of units in a decoding tick
4194
1.15k
  decoder_model_info->buffer_removal_time_length =
4195
1.15k
      aom_rb_read_literal(rb, 5) + 1;
4196
1.15k
  decoder_model_info->frame_presentation_time_length =
4197
1.15k
      aom_rb_read_literal(rb, 5) + 1;
4198
1.15k
}
4199
4200
void av1_read_op_parameters_info(aom_dec_model_op_parameters_t *op_params,
4201
                                 int buffer_delay_length,
4202
2.00k
                                 struct aom_read_bit_buffer *rb) {
4203
2.00k
  op_params->decoder_buffer_delay =
4204
2.00k
      aom_rb_read_unsigned_literal(rb, buffer_delay_length);
4205
2.00k
  op_params->encoder_buffer_delay =
4206
2.00k
      aom_rb_read_unsigned_literal(rb, buffer_delay_length);
4207
2.00k
  op_params->low_delay_mode_flag = aom_rb_read_bit(rb);
4208
2.00k
}
4209
4210
static inline void read_temporal_point_info(AV1_COMMON *const cm,
4211
1.33k
                                            struct aom_read_bit_buffer *rb) {
4212
1.33k
  cm->frame_presentation_time = aom_rb_read_unsigned_literal(
4213
1.33k
      rb, cm->seq_params->decoder_model_info.frame_presentation_time_length);
4214
1.33k
}
4215
4216
void av1_read_sequence_header(AV1_COMMON *cm, struct aom_read_bit_buffer *rb,
4217
77.4k
                              SequenceHeader *seq_params) {
4218
77.4k
  const int num_bits_width = aom_rb_read_literal(rb, 4) + 1;
4219
77.4k
  const int num_bits_height = aom_rb_read_literal(rb, 4) + 1;
4220
77.4k
  const int max_frame_width = aom_rb_read_literal(rb, num_bits_width) + 1;
4221
77.4k
  const int max_frame_height = aom_rb_read_literal(rb, num_bits_height) + 1;
4222
4223
77.4k
  seq_params->num_bits_width = num_bits_width;
4224
77.4k
  seq_params->num_bits_height = num_bits_height;
4225
77.4k
  seq_params->max_frame_width = max_frame_width;
4226
77.4k
  seq_params->max_frame_height = max_frame_height;
4227
4228
77.4k
  if (seq_params->reduced_still_picture_hdr) {
4229
16.6k
    seq_params->frame_id_numbers_present_flag = 0;
4230
60.7k
  } else {
4231
60.7k
    seq_params->frame_id_numbers_present_flag = aom_rb_read_bit(rb);
4232
60.7k
  }
4233
77.4k
  if (seq_params->frame_id_numbers_present_flag) {
4234
    // We must always have delta_frame_id_length < frame_id_length,
4235
    // in order for a frame to be referenced with a unique delta.
4236
    // Avoid wasting bits by using a coding that enforces this restriction.
4237
15.2k
    seq_params->delta_frame_id_length = aom_rb_read_literal(rb, 4) + 2;
4238
15.2k
    seq_params->frame_id_length =
4239
15.2k
        aom_rb_read_literal(rb, 3) + seq_params->delta_frame_id_length + 1;
4240
15.2k
    if (seq_params->frame_id_length > 16)
4241
440
      aom_internal_error(cm->error, AOM_CODEC_CORRUPT_FRAME,
4242
440
                         "Invalid frame_id_length");
4243
15.2k
  }
4244
4245
77.4k
  setup_sb_size(seq_params, rb);
4246
4247
77.4k
  seq_params->enable_filter_intra = aom_rb_read_bit(rb);
4248
77.4k
  seq_params->enable_intra_edge_filter = aom_rb_read_bit(rb);
4249
4250
77.4k
  if (seq_params->reduced_still_picture_hdr) {
4251
16.6k
    seq_params->enable_interintra_compound = 0;
4252
16.6k
    seq_params->enable_masked_compound = 0;
4253
16.6k
    seq_params->enable_warped_motion = 0;
4254
16.6k
    seq_params->enable_dual_filter = 0;
4255
16.6k
    seq_params->order_hint_info.enable_order_hint = 0;
4256
16.6k
    seq_params->order_hint_info.enable_dist_wtd_comp = 0;
4257
16.6k
    seq_params->order_hint_info.enable_ref_frame_mvs = 0;
4258
16.6k
    seq_params->force_screen_content_tools = 2;  // SELECT_SCREEN_CONTENT_TOOLS
4259
16.6k
    seq_params->force_integer_mv = 2;            // SELECT_INTEGER_MV
4260
16.6k
    seq_params->order_hint_info.order_hint_bits_minus_1 = -1;
4261
60.7k
  } else {
4262
60.7k
    seq_params->enable_interintra_compound = aom_rb_read_bit(rb);
4263
60.7k
    seq_params->enable_masked_compound = aom_rb_read_bit(rb);
4264
60.7k
    seq_params->enable_warped_motion = aom_rb_read_bit(rb);
4265
60.7k
    seq_params->enable_dual_filter = aom_rb_read_bit(rb);
4266
4267
60.7k
    seq_params->order_hint_info.enable_order_hint = aom_rb_read_bit(rb);
4268
60.7k
    seq_params->order_hint_info.enable_dist_wtd_comp =
4269
60.7k
        seq_params->order_hint_info.enable_order_hint ? aom_rb_read_bit(rb) : 0;
4270
60.7k
    seq_params->order_hint_info.enable_ref_frame_mvs =
4271
60.7k
        seq_params->order_hint_info.enable_order_hint ? aom_rb_read_bit(rb) : 0;
4272
4273
60.7k
    if (aom_rb_read_bit(rb)) {
4274
38.8k
      seq_params->force_screen_content_tools =
4275
38.8k
          2;  // SELECT_SCREEN_CONTENT_TOOLS
4276
38.8k
    } else {
4277
21.8k
      seq_params->force_screen_content_tools = aom_rb_read_bit(rb);
4278
21.8k
    }
4279
4280
60.7k
    if (seq_params->force_screen_content_tools > 0) {
4281
57.0k
      if (aom_rb_read_bit(rb)) {
4282
44.2k
        seq_params->force_integer_mv = 2;  // SELECT_INTEGER_MV
4283
44.2k
      } else {
4284
12.7k
        seq_params->force_integer_mv = aom_rb_read_bit(rb);
4285
12.7k
      }
4286
57.0k
    } else {
4287
3.74k
      seq_params->force_integer_mv = 2;  // SELECT_INTEGER_MV
4288
3.74k
    }
4289
60.7k
    seq_params->order_hint_info.order_hint_bits_minus_1 =
4290
60.7k
        seq_params->order_hint_info.enable_order_hint
4291
60.7k
            ? aom_rb_read_literal(rb, 3)
4292
60.7k
            : -1;
4293
60.7k
  }
4294
4295
77.4k
  seq_params->enable_superres = aom_rb_read_bit(rb);
4296
77.4k
  seq_params->enable_cdef = aom_rb_read_bit(rb);
4297
77.4k
  seq_params->enable_restoration = aom_rb_read_bit(rb);
4298
77.4k
}
4299
4300
static int read_global_motion_params(WarpedMotionParams *params,
4301
                                     const WarpedMotionParams *ref_params,
4302
                                     struct aom_read_bit_buffer *rb,
4303
305k
                                     int allow_hp) {
4304
305k
  TransformationType type = aom_rb_read_bit(rb);
4305
305k
  if (type != IDENTITY) {
4306
34.4k
    if (aom_rb_read_bit(rb))
4307
25.5k
      type = ROTZOOM;
4308
8.90k
    else
4309
8.90k
      type = aom_rb_read_bit(rb) ? TRANSLATION : AFFINE;
4310
34.4k
  }
4311
4312
305k
  *params = default_warp_params;
4313
305k
  params->wmtype = type;
4314
4315
305k
  if (type >= ROTZOOM) {
4316
31.6k
    params->wmmat[2] = aom_rb_read_signed_primitive_refsubexpfin(
4317
31.6k
                           rb, GM_ALPHA_MAX + 1, SUBEXPFIN_K,
4318
31.6k
                           (ref_params->wmmat[2] >> GM_ALPHA_PREC_DIFF) -
4319
31.6k
                               (1 << GM_ALPHA_PREC_BITS)) *
4320
31.6k
                           GM_ALPHA_DECODE_FACTOR +
4321
31.6k
                       (1 << WARPEDMODEL_PREC_BITS);
4322
31.6k
    params->wmmat[3] = aom_rb_read_signed_primitive_refsubexpfin(
4323
31.6k
                           rb, GM_ALPHA_MAX + 1, SUBEXPFIN_K,
4324
31.6k
                           (ref_params->wmmat[3] >> GM_ALPHA_PREC_DIFF)) *
4325
31.6k
                       GM_ALPHA_DECODE_FACTOR;
4326
31.6k
  }
4327
4328
305k
  if (type >= AFFINE) {
4329
6.08k
    params->wmmat[4] = aom_rb_read_signed_primitive_refsubexpfin(
4330
6.08k
                           rb, GM_ALPHA_MAX + 1, SUBEXPFIN_K,
4331
6.08k
                           (ref_params->wmmat[4] >> GM_ALPHA_PREC_DIFF)) *
4332
6.08k
                       GM_ALPHA_DECODE_FACTOR;
4333
6.08k
    params->wmmat[5] = aom_rb_read_signed_primitive_refsubexpfin(
4334
6.08k
                           rb, GM_ALPHA_MAX + 1, SUBEXPFIN_K,
4335
6.08k
                           (ref_params->wmmat[5] >> GM_ALPHA_PREC_DIFF) -
4336
6.08k
                               (1 << GM_ALPHA_PREC_BITS)) *
4337
6.08k
                           GM_ALPHA_DECODE_FACTOR +
4338
6.08k
                       (1 << WARPEDMODEL_PREC_BITS);
4339
299k
  } else {
4340
299k
    params->wmmat[4] = -params->wmmat[3];
4341
299k
    params->wmmat[5] = params->wmmat[2];
4342
299k
  }
4343
4344
305k
  if (type >= TRANSLATION) {
4345
34.2k
    const int trans_bits = (type == TRANSLATION)
4346
34.2k
                               ? GM_ABS_TRANS_ONLY_BITS - !allow_hp
4347
34.2k
                               : GM_ABS_TRANS_BITS;
4348
34.2k
    const int trans_dec_factor =
4349
34.2k
        (type == TRANSLATION) ? GM_TRANS_ONLY_DECODE_FACTOR * (1 << !allow_hp)
4350
34.2k
                              : GM_TRANS_DECODE_FACTOR;
4351
34.2k
    const int trans_prec_diff = (type == TRANSLATION)
4352
34.2k
                                    ? GM_TRANS_ONLY_PREC_DIFF + !allow_hp
4353
34.2k
                                    : GM_TRANS_PREC_DIFF;
4354
34.2k
    params->wmmat[0] = aom_rb_read_signed_primitive_refsubexpfin(
4355
34.2k
                           rb, (1 << trans_bits) + 1, SUBEXPFIN_K,
4356
34.2k
                           (ref_params->wmmat[0] >> trans_prec_diff)) *
4357
34.2k
                       trans_dec_factor;
4358
34.2k
    params->wmmat[1] = aom_rb_read_signed_primitive_refsubexpfin(
4359
34.2k
                           rb, (1 << trans_bits) + 1, SUBEXPFIN_K,
4360
34.2k
                           (ref_params->wmmat[1] >> trans_prec_diff)) *
4361
34.2k
                       trans_dec_factor;
4362
34.2k
  }
4363
4364
305k
  int good_shear_params = av1_get_shear_params(params);
4365
305k
  if (!good_shear_params) return 0;
4366
4367
290k
  return 1;
4368
305k
}
4369
4370
static inline void read_global_motion(AV1_COMMON *cm,
4371
43.9k
                                      struct aom_read_bit_buffer *rb) {
4372
349k
  for (int frame = LAST_FRAME; frame <= ALTREF_FRAME; ++frame) {
4373
305k
    const WarpedMotionParams *ref_params =
4374
305k
        cm->prev_frame ? &cm->prev_frame->global_motion[frame]
4375
305k
                       : &default_warp_params;
4376
305k
    int good_params =
4377
305k
        read_global_motion_params(&cm->global_motion[frame], ref_params, rb,
4378
305k
                                  cm->features.allow_high_precision_mv);
4379
305k
    if (!good_params) {
4380
#if WARPED_MOTION_DEBUG
4381
      printf("Warning: unexpected global motion shear params from aomenc\n");
4382
#endif
4383
14.6k
      cm->global_motion[frame].invalid = 1;
4384
14.6k
    }
4385
4386
    // TODO(sarahparker, debargha): The logic in the commented out code below
4387
    // does not work currently and causes mismatches when resize is on. Fix it
4388
    // before turning the optimization back on.
4389
    /*
4390
    YV12_BUFFER_CONFIG *ref_buf = get_ref_frame(cm, frame);
4391
    if (cm->width == ref_buf->y_crop_width &&
4392
        cm->height == ref_buf->y_crop_height) {
4393
      read_global_motion_params(&cm->global_motion[frame],
4394
                                &cm->prev_frame->global_motion[frame], rb,
4395
                                cm->features.allow_high_precision_mv);
4396
    } else {
4397
      cm->global_motion[frame] = default_warp_params;
4398
    }
4399
    */
4400
    /*
4401
    printf("Dec Ref %d [%d/%d]: %d %d %d %d\n",
4402
           frame, cm->current_frame.frame_number, cm->show_frame,
4403
           cm->global_motion[frame].wmmat[0],
4404
           cm->global_motion[frame].wmmat[1],
4405
           cm->global_motion[frame].wmmat[2],
4406
           cm->global_motion[frame].wmmat[3]);
4407
           */
4408
305k
  }
4409
43.9k
  memcpy(cm->cur_frame->global_motion, cm->global_motion,
4410
43.9k
         REF_FRAMES * sizeof(WarpedMotionParams));
4411
43.9k
}
4412
4413
// Release the references to the frame buffers in cm->ref_frame_map and reset
4414
// all elements of cm->ref_frame_map to NULL.
4415
125k
static inline void reset_ref_frame_map(AV1_COMMON *const cm) {
4416
125k
  BufferPool *const pool = cm->buffer_pool;
4417
4418
1.12M
  for (int i = 0; i < REF_FRAMES; i++) {
4419
1.00M
    decrease_ref_count(cm->ref_frame_map[i], pool);
4420
1.00M
    cm->ref_frame_map[i] = NULL;
4421
1.00M
  }
4422
125k
}
4423
4424
// If the refresh_frame_flags bitmask is set, update reference frame id values
4425
// and mark frames as valid for reference.
4426
184k
static inline void update_ref_frame_id(AV1Decoder *const pbi) {
4427
184k
  AV1_COMMON *const cm = &pbi->common;
4428
184k
  int refresh_frame_flags = cm->current_frame.refresh_frame_flags;
4429
1.66M
  for (int i = 0; i < REF_FRAMES; i++) {
4430
1.47M
    if ((refresh_frame_flags >> i) & 1) {
4431
780k
      cm->ref_frame_id[i] = cm->current_frame_id;
4432
780k
      pbi->valid_for_referencing[i] = 1;
4433
780k
    }
4434
1.47M
  }
4435
184k
}
4436
4437
static inline void show_existing_frame_reset(AV1Decoder *const pbi,
4438
49
                                             int existing_frame_idx) {
4439
49
  AV1_COMMON *const cm = &pbi->common;
4440
4441
49
  assert(cm->show_existing_frame);
4442
4443
49
  cm->current_frame.frame_type = KEY_FRAME;
4444
4445
49
  cm->current_frame.refresh_frame_flags = (1 << REF_FRAMES) - 1;
4446
4447
392
  for (int i = 0; i < INTER_REFS_PER_FRAME; ++i) {
4448
343
    cm->remapped_ref_idx[i] = INVALID_IDX;
4449
343
  }
4450
4451
49
  if (pbi->need_resync) {
4452
27
    reset_ref_frame_map(cm);
4453
27
    pbi->need_resync = 0;
4454
27
  }
4455
4456
  // Note that the displayed frame must be valid for referencing in order to
4457
  // have been selected.
4458
49
  cm->current_frame_id = cm->ref_frame_id[existing_frame_idx];
4459
49
  update_ref_frame_id(pbi);
4460
4461
49
  cm->features.refresh_frame_context = REFRESH_FRAME_CONTEXT_DISABLED;
4462
49
}
4463
4464
26.7k
static inline void reset_frame_buffers(AV1_COMMON *cm) {
4465
26.7k
  RefCntBuffer *const frame_bufs = cm->buffer_pool->frame_bufs;
4466
26.7k
  int i;
4467
4468
26.7k
  lock_buffer_pool(cm->buffer_pool);
4469
26.7k
  reset_ref_frame_map(cm);
4470
26.7k
  assert(cm->cur_frame->ref_count == 1);
4471
454k
  for (i = 0; i < cm->buffer_pool->num_frame_bufs; ++i) {
4472
    // Reset all unreferenced frame buffers. We can also reset cm->cur_frame
4473
    // because we are the sole owner of cm->cur_frame.
4474
427k
    if (frame_bufs[i].ref_count > 0 && &frame_bufs[i] != cm->cur_frame) {
4475
5.86k
      continue;
4476
5.86k
    }
4477
421k
    frame_bufs[i].order_hint = 0;
4478
421k
    av1_zero(frame_bufs[i].ref_order_hints);
4479
421k
  }
4480
26.7k
  av1_zero_unused_internal_frame_buffers(&cm->buffer_pool->int_frame_buffers);
4481
26.7k
  unlock_buffer_pool(cm->buffer_pool);
4482
26.7k
}
4483
4484
// On success, returns 0. On failure, calls aom_internal_error and does not
4485
// return.
4486
static int read_uncompressed_header(AV1Decoder *pbi,
4487
218k
                                    struct aom_read_bit_buffer *rb) {
4488
218k
  AV1_COMMON *const cm = &pbi->common;
4489
218k
  const SequenceHeader *const seq_params = cm->seq_params;
4490
218k
  CurrentFrame *const current_frame = &cm->current_frame;
4491
218k
  FeatureFlags *const features = &cm->features;
4492
218k
  MACROBLOCKD *const xd = &pbi->dcb.xd;
4493
218k
  BufferPool *const pool = cm->buffer_pool;
4494
218k
  RefCntBuffer *const frame_bufs = pool->frame_bufs;
4495
218k
  aom_s_frame_info *sframe_info = &pbi->sframe_info;
4496
218k
  sframe_info->is_s_frame = 0;
4497
218k
  sframe_info->is_s_frame_at_altref = 0;
4498
4499
218k
  if (!pbi->sequence_header_ready) {
4500
842
    aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4501
842
                       "No sequence header");
4502
842
  }
4503
4504
218k
  if (seq_params->reduced_still_picture_hdr) {
4505
32.6k
    cm->show_existing_frame = 0;
4506
32.6k
    cm->show_frame = 1;
4507
32.6k
    current_frame->frame_type = KEY_FRAME;
4508
32.6k
    if (pbi->sequence_header_changed) {
4509
      // This is the start of a new coded video sequence.
4510
6.76k
      pbi->sequence_header_changed = 0;
4511
6.76k
      pbi->decoding_first_frame = 1;
4512
6.76k
      reset_frame_buffers(cm);
4513
6.76k
    }
4514
32.6k
    features->error_resilient_mode = 1;
4515
185k
  } else {
4516
185k
    cm->show_existing_frame = aom_rb_read_bit(rb);
4517
185k
    pbi->reset_decoder_state = 0;
4518
4519
185k
    if (cm->show_existing_frame) {
4520
3.78k
      if (pbi->sequence_header_changed) {
4521
282
        aom_internal_error(
4522
282
            &pbi->error, AOM_CODEC_CORRUPT_FRAME,
4523
282
            "New sequence header starts with a show_existing_frame.");
4524
282
      }
4525
      // Show an existing frame directly.
4526
3.78k
      const int existing_frame_idx = aom_rb_read_literal(rb, 3);
4527
3.78k
      RefCntBuffer *const frame_to_show = cm->ref_frame_map[existing_frame_idx];
4528
3.78k
      if (frame_to_show == NULL) {
4529
1.63k
        aom_internal_error(&pbi->error, AOM_CODEC_UNSUP_BITSTREAM,
4530
1.63k
                           "Buffer does not contain a decoded frame");
4531
1.63k
      }
4532
3.78k
      if (seq_params->decoder_model_info_present_flag &&
4533
3.78k
          seq_params->timing_info.equal_picture_interval == 0) {
4534
61
        read_temporal_point_info(cm, rb);
4535
61
      }
4536
3.78k
      if (seq_params->frame_id_numbers_present_flag) {
4537
1.28k
        int frame_id_length = seq_params->frame_id_length;
4538
1.28k
        int display_frame_id = aom_rb_read_literal(rb, frame_id_length);
4539
        /* Compare display_frame_id with ref_frame_id and check valid for
4540
         * referencing */
4541
1.28k
        if (display_frame_id != cm->ref_frame_id[existing_frame_idx] ||
4542
1.28k
            pbi->valid_for_referencing[existing_frame_idx] == 0)
4543
1.03k
          aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4544
1.03k
                             "Reference buffer frame ID mismatch");
4545
1.28k
      }
4546
3.78k
      lock_buffer_pool(pool);
4547
3.78k
      assert(frame_to_show->ref_count > 0);
4548
      // cm->cur_frame should be the buffer referenced by the return value
4549
      // of the get_free_fb() call in assign_cur_frame_new_fb() (called by
4550
      // av1_receive_compressed_data()), so the ref_count should be 1.
4551
839
      assert(cm->cur_frame->ref_count == 1);
4552
      // assign_frame_buffer_p() decrements ref_count directly rather than
4553
      // call decrease_ref_count(). If cm->cur_frame->raw_frame_buffer has
4554
      // already been allocated, it will not be released by
4555
      // assign_frame_buffer_p()!
4556
839
      assert(!cm->cur_frame->raw_frame_buffer.data);
4557
839
      assign_frame_buffer_p(&cm->cur_frame, frame_to_show);
4558
839
      pbi->reset_decoder_state = frame_to_show->frame_type == KEY_FRAME;
4559
839
      unlock_buffer_pool(pool);
4560
4561
839
      cm->lf.filter_level[0] = 0;
4562
839
      cm->lf.filter_level[1] = 0;
4563
839
      cm->show_frame = 1;
4564
839
      current_frame->order_hint = frame_to_show->order_hint;
4565
4566
      // Section 6.8.2: It is a requirement of bitstream conformance that when
4567
      // show_existing_frame is used to show a previous frame, that the value
4568
      // of showable_frame for the previous frame was equal to 1.
4569
839
      if (!frame_to_show->showable_frame) {
4570
506
        aom_internal_error(&pbi->error, AOM_CODEC_UNSUP_BITSTREAM,
4571
506
                           "Buffer does not contain a showable frame");
4572
506
      }
4573
      // Section 6.8.2: It is a requirement of bitstream conformance that when
4574
      // show_existing_frame is used to show a previous frame with
4575
      // RefFrameType[ frame_to_show_map_idx ] equal to KEY_FRAME, that the
4576
      // frame is output via the show_existing_frame mechanism at most once.
4577
839
      if (pbi->reset_decoder_state) frame_to_show->showable_frame = 0;
4578
4579
839
      cm->film_grain_params = frame_to_show->film_grain_params;
4580
4581
839
      if (pbi->reset_decoder_state) {
4582
49
        show_existing_frame_reset(pbi, existing_frame_idx);
4583
790
      } else {
4584
790
        current_frame->refresh_frame_flags = 0;
4585
790
      }
4586
4587
839
      return 0;
4588
839
    }
4589
4590
181k
    current_frame->frame_type = (FRAME_TYPE)aom_rb_read_literal(rb, 2);
4591
181k
    if (pbi->sequence_header_changed) {
4592
22.5k
      if (current_frame->frame_type == KEY_FRAME) {
4593
        // This is the start of a new coded video sequence.
4594
19.9k
        pbi->sequence_header_changed = 0;
4595
19.9k
        pbi->decoding_first_frame = 1;
4596
19.9k
        reset_frame_buffers(cm);
4597
19.9k
      } else {
4598
2.58k
        aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4599
2.58k
                           "Sequence header has changed without a keyframe.");
4600
2.58k
      }
4601
22.5k
    }
4602
4603
181k
    cm->show_frame = aom_rb_read_bit(rb);
4604
181k
    if (cm->show_frame == 0) pbi->is_arf_frame_present = 1;
4605
181k
    if (cm->show_frame == 0 && cm->current_frame.frame_type == KEY_FRAME)
4606
48.6k
      pbi->is_fwd_kf_present = 1;
4607
181k
    if (cm->current_frame.frame_type == S_FRAME) {
4608
151
      sframe_info->is_s_frame = 1;
4609
151
      sframe_info->is_s_frame_at_altref = cm->show_frame ? 0 : 1;
4610
151
    }
4611
181k
    if (seq_params->still_picture &&
4612
181k
        (current_frame->frame_type != KEY_FRAME || !cm->show_frame)) {
4613
201
      aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4614
201
                         "Still pictures must be coded as shown keyframes");
4615
201
    }
4616
181k
    cm->showable_frame = current_frame->frame_type != KEY_FRAME;
4617
181k
    if (cm->show_frame) {
4618
104k
      if (seq_params->decoder_model_info_present_flag &&
4619
104k
          seq_params->timing_info.equal_picture_interval == 0)
4620
1.27k
        read_temporal_point_info(cm, rb);
4621
104k
    } else {
4622
      // See if this frame can be used as show_existing_frame in future
4623
77.7k
      cm->showable_frame = aom_rb_read_bit(rb);
4624
77.7k
    }
4625
181k
    cm->cur_frame->showable_frame = cm->showable_frame;
4626
181k
    features->error_resilient_mode =
4627
181k
        frame_is_sframe(cm) ||
4628
181k
                (current_frame->frame_type == KEY_FRAME && cm->show_frame)
4629
181k
            ? 1
4630
181k
            : aom_rb_read_bit(rb);
4631
181k
  }
4632
4633
214k
  if (current_frame->frame_type == KEY_FRAME && cm->show_frame) {
4634
    /* All frames need to be marked as not valid for referencing */
4635
541k
    for (int i = 0; i < REF_FRAMES; i++) {
4636
481k
      pbi->valid_for_referencing[i] = 0;
4637
481k
    }
4638
60.1k
  }
4639
214k
  features->disable_cdf_update = aom_rb_read_bit(rb);
4640
214k
  if (seq_params->force_screen_content_tools == 2) {
4641
168k
    features->allow_screen_content_tools = aom_rb_read_bit(rb);
4642
168k
  } else {
4643
46.1k
    features->allow_screen_content_tools =
4644
46.1k
        seq_params->force_screen_content_tools;
4645
46.1k
  }
4646
4647
214k
  if (features->allow_screen_content_tools) {
4648
102k
    if (seq_params->force_integer_mv == 2) {
4649
78.3k
      features->cur_frame_force_integer_mv = aom_rb_read_bit(rb);
4650
78.3k
    } else {
4651
23.7k
      features->cur_frame_force_integer_mv = seq_params->force_integer_mv;
4652
23.7k
    }
4653
112k
  } else {
4654
112k
    features->cur_frame_force_integer_mv = 0;
4655
112k
  }
4656
4657
214k
  int frame_size_override_flag = 0;
4658
214k
  features->allow_intrabc = 0;
4659
214k
  features->primary_ref_frame = PRIMARY_REF_NONE;
4660
4661
214k
  if (!seq_params->reduced_still_picture_hdr) {
4662
177k
    if (seq_params->frame_id_numbers_present_flag) {
4663
24.4k
      int frame_id_length = seq_params->frame_id_length;
4664
24.4k
      int diff_len = seq_params->delta_frame_id_length;
4665
24.4k
      int prev_frame_id = 0;
4666
24.4k
      int have_prev_frame_id =
4667
24.4k
          !pbi->decoding_first_frame &&
4668
24.4k
          !(current_frame->frame_type == KEY_FRAME && cm->show_frame);
4669
24.4k
      if (have_prev_frame_id) {
4670
2.82k
        prev_frame_id = cm->current_frame_id;
4671
2.82k
      }
4672
24.4k
      cm->current_frame_id = aom_rb_read_literal(rb, frame_id_length);
4673
4674
24.4k
      if (have_prev_frame_id) {
4675
2.82k
        int diff_frame_id;
4676
2.82k
        if (cm->current_frame_id > prev_frame_id) {
4677
885
          diff_frame_id = cm->current_frame_id - prev_frame_id;
4678
1.94k
        } else {
4679
1.94k
          diff_frame_id =
4680
1.94k
              (1 << frame_id_length) + cm->current_frame_id - prev_frame_id;
4681
1.94k
        }
4682
        /* Check current_frame_id for conformance */
4683
2.82k
        if (prev_frame_id == cm->current_frame_id ||
4684
2.82k
            diff_frame_id >= (1 << (frame_id_length - 1))) {
4685
1.96k
          aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4686
1.96k
                             "Invalid value of current_frame_id");
4687
1.96k
        }
4688
2.82k
      }
4689
      /* Check if some frames need to be marked as not valid for referencing */
4690
204k
      for (int i = 0; i < REF_FRAMES; i++) {
4691
179k
        if (cm->current_frame_id - (1 << diff_len) > 0) {
4692
58.0k
          if (cm->ref_frame_id[i] > cm->current_frame_id ||
4693
58.0k
              cm->ref_frame_id[i] < cm->current_frame_id - (1 << diff_len))
4694
41.2k
            pbi->valid_for_referencing[i] = 0;
4695
121k
        } else {
4696
121k
          if (cm->ref_frame_id[i] > cm->current_frame_id &&
4697
121k
              cm->ref_frame_id[i] < (1 << frame_id_length) +
4698
49.0k
                                        cm->current_frame_id - (1 << diff_len))
4699
39.0k
            pbi->valid_for_referencing[i] = 0;
4700
121k
        }
4701
179k
      }
4702
24.4k
    }
4703
4704
177k
    frame_size_override_flag = frame_is_sframe(cm) ? 1 : aom_rb_read_bit(rb);
4705
4706
177k
    current_frame->order_hint = aom_rb_read_literal(
4707
177k
        rb, seq_params->order_hint_info.order_hint_bits_minus_1 + 1);
4708
4709
177k
    if (seq_params->order_hint_info.enable_order_hint)
4710
154k
      current_frame->frame_number = current_frame->order_hint;
4711
4712
177k
    if (!features->error_resilient_mode && !frame_is_intra_only(cm)) {
4713
64.0k
      features->primary_ref_frame = aom_rb_read_literal(rb, PRIMARY_REF_BITS);
4714
64.0k
    }
4715
177k
  }
4716
4717
214k
  if (seq_params->decoder_model_info_present_flag) {
4718
10.0k
    pbi->buffer_removal_time_present = aom_rb_read_bit(rb);
4719
10.0k
    if (pbi->buffer_removal_time_present) {
4720
2.94k
      for (int op_num = 0;
4721
15.6k
           op_num < seq_params->operating_points_cnt_minus_1 + 1; op_num++) {
4722
12.6k
        if (seq_params->op_params[op_num].decoder_model_param_present_flag) {
4723
5.08k
          if (seq_params->operating_point_idc[op_num] == 0 ||
4724
5.08k
              (((seq_params->operating_point_idc[op_num] >>
4725
4.91k
                 cm->temporal_layer_id) &
4726
4.91k
                0x1) &&
4727
4.91k
               ((seq_params->operating_point_idc[op_num] >>
4728
2.71k
                 (cm->spatial_layer_id + 8)) &
4729
2.71k
                0x1))) {
4730
2.52k
            cm->buffer_removal_times[op_num] = aom_rb_read_unsigned_literal(
4731
2.52k
                rb, seq_params->decoder_model_info.buffer_removal_time_length);
4732
2.55k
          } else {
4733
2.55k
            cm->buffer_removal_times[op_num] = 0;
4734
2.55k
          }
4735
7.57k
        } else {
4736
7.57k
          cm->buffer_removal_times[op_num] = 0;
4737
7.57k
        }
4738
12.6k
      }
4739
2.94k
    }
4740
10.0k
  }
4741
214k
  if (current_frame->frame_type == KEY_FRAME) {
4742
108k
    if (!cm->show_frame) {  // unshown keyframe (forward keyframe)
4743
48.1k
      current_frame->refresh_frame_flags = aom_rb_read_literal(rb, REF_FRAMES);
4744
60.1k
    } else {  // shown keyframe
4745
60.1k
      current_frame->refresh_frame_flags = (1 << REF_FRAMES) - 1;
4746
60.1k
    }
4747
4748
865k
    for (int i = 0; i < INTER_REFS_PER_FRAME; ++i) {
4749
757k
      cm->remapped_ref_idx[i] = INVALID_IDX;
4750
757k
    }
4751
108k
    if (pbi->need_resync) {
4752
78.5k
      reset_ref_frame_map(cm);
4753
78.5k
      pbi->need_resync = 0;
4754
78.5k
    }
4755
108k
  } else {
4756
106k
    if (current_frame->frame_type == INTRA_ONLY_FRAME) {
4757
24.6k
      current_frame->refresh_frame_flags = aom_rb_read_literal(rb, REF_FRAMES);
4758
24.6k
      if (current_frame->refresh_frame_flags == 0xFF) {
4759
242
        aom_internal_error(&pbi->error, AOM_CODEC_UNSUP_BITSTREAM,
4760
242
                           "Intra only frames cannot have refresh flags 0xFF");
4761
242
      }
4762
24.6k
      if (pbi->need_resync) {
4763
19.7k
        reset_ref_frame_map(cm);
4764
19.7k
        pbi->need_resync = 0;
4765
19.7k
      }
4766
81.6k
    } else if (pbi->need_resync != 1) { /* Skip if need resync */
4767
57.3k
      current_frame->refresh_frame_flags =
4768
57.3k
          frame_is_sframe(cm) ? 0xFF : aom_rb_read_literal(rb, REF_FRAMES);
4769
57.3k
    }
4770
106k
  }
4771
4772
214k
  if (!frame_is_intra_only(cm) || current_frame->refresh_frame_flags != 0xFF) {
4773
    // Read all ref frame order hints if error_resilient_mode == 1
4774
147k
    if (features->error_resilient_mode &&
4775
147k
        seq_params->order_hint_info.enable_order_hint) {
4776
301k
      for (int ref_idx = 0; ref_idx < REF_FRAMES; ref_idx++) {
4777
        // Read order hint from bit stream
4778
268k
        unsigned int order_hint = aom_rb_read_literal(
4779
268k
            rb, seq_params->order_hint_info.order_hint_bits_minus_1 + 1);
4780
        // Get buffer
4781
268k
        RefCntBuffer *buf = cm->ref_frame_map[ref_idx];
4782
268k
        if (buf == NULL || order_hint != buf->order_hint) {
4783
235k
          if (buf != NULL) {
4784
78.4k
            lock_buffer_pool(pool);
4785
78.4k
            decrease_ref_count(buf, pool);
4786
78.4k
            unlock_buffer_pool(pool);
4787
78.4k
            cm->ref_frame_map[ref_idx] = NULL;
4788
78.4k
          }
4789
          // If no corresponding buffer exists, allocate a new buffer with all
4790
          // pixels set to neutral grey.
4791
235k
          int buf_idx = get_free_fb(cm);
4792
235k
          if (buf_idx == INVALID_IDX) {
4793
0
            aom_internal_error(&pbi->error, AOM_CODEC_MEM_ERROR,
4794
0
                               "Unable to find free frame buffer");
4795
0
          }
4796
235k
          buf = &frame_bufs[buf_idx];
4797
235k
          lock_buffer_pool(pool);
4798
235k
#if CONFIG_SIZE_LIMIT
4799
235k
          if (seq_params->max_frame_width > DECODE_WIDTH_LIMIT ||
4800
235k
              seq_params->max_frame_height > DECODE_HEIGHT_LIMIT) {
4801
158
            decrease_ref_count(buf, pool);
4802
158
            unlock_buffer_pool(pool);
4803
158
            aom_internal_error(
4804
158
                cm->error, AOM_CODEC_CORRUPT_FRAME,
4805
158
                "Dimensions of %dx%d beyond allowed size of %dx%d.",
4806
158
                seq_params->max_frame_width, seq_params->max_frame_height,
4807
158
                DECODE_WIDTH_LIMIT, DECODE_HEIGHT_LIMIT);
4808
158
          }
4809
235k
#endif
4810
235k
          if (aom_realloc_frame_buffer(
4811
235k
                  &buf->buf, seq_params->max_frame_width,
4812
235k
                  seq_params->max_frame_height, seq_params->subsampling_x,
4813
235k
                  seq_params->subsampling_y, seq_params->use_highbitdepth,
4814
235k
                  AOM_BORDER_IN_PIXELS, features->byte_alignment,
4815
235k
                  &buf->raw_frame_buffer, pool->get_fb_cb, pool->cb_priv, false,
4816
235k
                  0)) {
4817
72
            decrease_ref_count(buf, pool);
4818
72
            unlock_buffer_pool(pool);
4819
72
            aom_internal_error(&pbi->error, AOM_CODEC_MEM_ERROR,
4820
72
                               "Failed to allocate frame buffer");
4821
72
          }
4822
235k
          unlock_buffer_pool(pool);
4823
          // According to the specification, valid bitstreams are required to
4824
          // never use missing reference frames so the filling process for
4825
          // missing frames is not normatively defined and RefValid for missing
4826
          // frames is set to 0.
4827
4828
          // To make libaom more robust when the bitstream has been corrupted
4829
          // by the loss of some frames of data, this code adds a neutral grey
4830
          // buffer in place of missing frames, i.e.
4831
          //
4832
235k
          set_planes_to_neutral_grey(seq_params, &buf->buf, 0);
4833
          //
4834
          // and allows the frames to be used for referencing, i.e.
4835
          //
4836
235k
          pbi->valid_for_referencing[ref_idx] = 1;
4837
          //
4838
          // Please note such behavior is not normative and other decoders may
4839
          // use a different approach.
4840
235k
          cm->ref_frame_map[ref_idx] = buf;
4841
235k
          buf->order_hint = order_hint;
4842
235k
        }
4843
268k
      }
4844
33.7k
    }
4845
147k
  }
4846
4847
214k
  if (current_frame->frame_type == KEY_FRAME) {
4848
108k
    setup_frame_size(cm, frame_size_override_flag, rb);
4849
4850
108k
    if (features->allow_screen_content_tools && !av1_superres_scaled(cm))
4851
52.1k
      features->allow_intrabc = aom_rb_read_bit(rb);
4852
108k
    features->allow_ref_frame_mvs = 0;
4853
108k
    cm->prev_frame = NULL;
4854
108k
  } else {
4855
106k
    features->allow_ref_frame_mvs = 0;
4856
4857
106k
    if (current_frame->frame_type == INTRA_ONLY_FRAME) {
4858
24.1k
      cm->cur_frame->film_grain_params_present =
4859
24.1k
          seq_params->film_grain_params_present;
4860
24.1k
      setup_frame_size(cm, frame_size_override_flag, rb);
4861
24.1k
      if (features->allow_screen_content_tools && !av1_superres_scaled(cm))
4862
21.1k
        features->allow_intrabc = aom_rb_read_bit(rb);
4863
4864
82.1k
    } else if (pbi->need_resync != 1) { /* Skip if need resync */
4865
57.3k
      int frame_refs_short_signaling = 0;
4866
      // Frame refs short signaling is off when error resilient mode is on.
4867
57.3k
      if (seq_params->order_hint_info.enable_order_hint)
4868
57.0k
        frame_refs_short_signaling = aom_rb_read_bit(rb);
4869
4870
57.3k
      if (frame_refs_short_signaling) {
4871
        // == LAST_FRAME ==
4872
28.1k
        const int lst_ref = aom_rb_read_literal(rb, REF_FRAMES_LOG2);
4873
28.1k
        const RefCntBuffer *const lst_buf = cm->ref_frame_map[lst_ref];
4874
4875
        // == GOLDEN_FRAME ==
4876
28.1k
        const int gld_ref = aom_rb_read_literal(rb, REF_FRAMES_LOG2);
4877
28.1k
        const RefCntBuffer *const gld_buf = cm->ref_frame_map[gld_ref];
4878
4879
        // Most of the time, streams start with a keyframe. In that case,
4880
        // ref_frame_map will have been filled in at that point and will not
4881
        // contain any NULLs. However, streams are explicitly allowed to start
4882
        // with an intra-only frame, so long as they don't then signal a
4883
        // reference to a slot that hasn't been set yet. That's what we are
4884
        // checking here.
4885
28.1k
        if (lst_buf == NULL)
4886
96
          aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4887
96
                             "Inter frame requests nonexistent reference");
4888
28.1k
        if (gld_buf == NULL)
4889
13
          aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4890
13
                             "Inter frame requests nonexistent reference");
4891
4892
28.1k
        av1_set_frame_refs(cm, cm->remapped_ref_idx, lst_ref, gld_ref);
4893
28.1k
      }
4894
4895
451k
      for (int i = 0; i < INTER_REFS_PER_FRAME; ++i) {
4896
394k
        int ref = 0;
4897
394k
        if (!frame_refs_short_signaling) {
4898
201k
          ref = aom_rb_read_literal(rb, REF_FRAMES_LOG2);
4899
4900
          // Most of the time, streams start with a keyframe. In that case,
4901
          // ref_frame_map will have been filled in at that point and will not
4902
          // contain any NULLs. However, streams are explicitly allowed to start
4903
          // with an intra-only frame, so long as they don't then signal a
4904
          // reference to a slot that hasn't been set yet. That's what we are
4905
          // checking here.
4906
201k
          if (cm->ref_frame_map[ref] == NULL)
4907
375
            aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4908
375
                               "Inter frame requests nonexistent reference");
4909
201k
          cm->remapped_ref_idx[i] = ref;
4910
201k
        } else {
4911
193k
          ref = cm->remapped_ref_idx[i];
4912
193k
        }
4913
        // Check valid for referencing
4914
394k
        if (pbi->valid_for_referencing[ref] == 0)
4915
56
          aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4916
56
                             "Reference frame not valid for referencing");
4917
4918
394k
        cm->ref_frame_sign_bias[LAST_FRAME + i] = 0;
4919
4920
394k
        if (seq_params->frame_id_numbers_present_flag) {
4921
71
          int frame_id_length = seq_params->frame_id_length;
4922
71
          int diff_len = seq_params->delta_frame_id_length;
4923
71
          int delta_frame_id_minus_1 = aom_rb_read_literal(rb, diff_len);
4924
71
          int ref_frame_id =
4925
71
              ((cm->current_frame_id - (delta_frame_id_minus_1 + 1) +
4926
71
                (1 << frame_id_length)) %
4927
71
               (1 << frame_id_length));
4928
          // Compare values derived from delta_frame_id_minus_1 and
4929
          // refresh_frame_flags.
4930
71
          if (ref_frame_id != cm->ref_frame_id[ref])
4931
43
            aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4932
43
                               "Reference buffer frame ID mismatch");
4933
71
        }
4934
394k
      }
4935
4936
57.3k
      if (!features->error_resilient_mode && frame_size_override_flag) {
4937
37.9k
        setup_frame_size_with_refs(cm, rb);
4938
37.9k
      } else {
4939
19.4k
        setup_frame_size(cm, frame_size_override_flag, rb);
4940
19.4k
      }
4941
4942
57.3k
      if (features->cur_frame_force_integer_mv) {
4943
4.21k
        features->allow_high_precision_mv = 0;
4944
53.1k
      } else {
4945
53.1k
        features->allow_high_precision_mv = aom_rb_read_bit(rb);
4946
53.1k
      }
4947
57.3k
      features->interp_filter = read_frame_interp_filter(rb);
4948
57.3k
      features->switchable_motion_mode = aom_rb_read_bit(rb);
4949
57.3k
    }
4950
4951
106k
    cm->prev_frame = get_primary_ref_frame_buf(cm);
4952
106k
    if (features->primary_ref_frame != PRIMARY_REF_NONE &&
4953
106k
        get_primary_ref_frame_buf(cm) == NULL) {
4954
7.81k
      aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
4955
7.81k
                         "Reference frame containing this frame's initial "
4956
7.81k
                         "frame context is unavailable.");
4957
7.81k
    }
4958
4959
106k
    if (!(current_frame->frame_type == INTRA_ONLY_FRAME) &&
4960
106k
        pbi->need_resync != 1) {
4961
48.8k
      if (frame_might_allow_ref_frame_mvs(cm))
4962
39.3k
        features->allow_ref_frame_mvs = aom_rb_read_bit(rb);
4963
9.46k
      else
4964
9.46k
        features->allow_ref_frame_mvs = 0;
4965
4966
372k
      for (int i = LAST_FRAME; i <= ALTREF_FRAME; ++i) {
4967
323k
        const RefCntBuffer *const ref_buf = get_ref_frame_buf(cm, i);
4968
323k
        struct scale_factors *const ref_scale_factors =
4969
323k
            get_ref_scale_factors(cm, i);
4970
323k
        av1_setup_scale_factors_for_frame(
4971
323k
            ref_scale_factors, ref_buf->buf.y_crop_width,
4972
323k
            ref_buf->buf.y_crop_height, cm->width, cm->height);
4973
323k
        if ((!av1_is_valid_scale(ref_scale_factors)))
4974
4.60k
          aom_internal_error(&pbi->error, AOM_CODEC_UNSUP_BITSTREAM,
4975
4.60k
                             "Reference frame has invalid dimensions");
4976
323k
      }
4977
48.8k
    }
4978
106k
  }
4979
4980
214k
  av1_setup_frame_buf_refs(cm);
4981
4982
214k
  av1_setup_frame_sign_bias(cm);
4983
4984
214k
  cm->cur_frame->frame_type = current_frame->frame_type;
4985
4986
214k
  update_ref_frame_id(pbi);
4987
4988
214k
  const int might_bwd_adapt = !(seq_params->reduced_still_picture_hdr) &&
4989
214k
                              !(features->disable_cdf_update);
4990
214k
  if (might_bwd_adapt) {
4991
123k
    features->refresh_frame_context = aom_rb_read_bit(rb)
4992
123k
                                          ? REFRESH_FRAME_CONTEXT_DISABLED
4993
123k
                                          : REFRESH_FRAME_CONTEXT_BACKWARD;
4994
123k
  } else {
4995
91.3k
    features->refresh_frame_context = REFRESH_FRAME_CONTEXT_DISABLED;
4996
91.3k
  }
4997
4998
214k
  cm->cur_frame->buf.bit_depth = seq_params->bit_depth;
4999
214k
  cm->cur_frame->buf.color_primaries = seq_params->color_primaries;
5000
214k
  cm->cur_frame->buf.transfer_characteristics =
5001
214k
      seq_params->transfer_characteristics;
5002
214k
  cm->cur_frame->buf.matrix_coefficients = seq_params->matrix_coefficients;
5003
214k
  cm->cur_frame->buf.monochrome = seq_params->monochrome;
5004
214k
  cm->cur_frame->buf.chroma_sample_position =
5005
214k
      seq_params->chroma_sample_position;
5006
214k
  cm->cur_frame->buf.color_range = seq_params->color_range;
5007
214k
  cm->cur_frame->buf.render_width = cm->render_width;
5008
214k
  cm->cur_frame->buf.render_height = cm->render_height;
5009
5010
214k
  if (pbi->need_resync) {
5011
10.4k
    aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
5012
10.4k
                       "Keyframe / intra-only frame required to reset decoder"
5013
10.4k
                       " state");
5014
10.4k
  }
5015
5016
214k
  if (features->allow_intrabc) {
5017
    // Set parameters corresponding to no filtering.
5018
42.8k
    struct loopfilter *lf = &cm->lf;
5019
42.8k
    lf->filter_level[0] = 0;
5020
42.8k
    lf->filter_level[1] = 0;
5021
42.8k
    cm->cdef_info.cdef_bits = 0;
5022
42.8k
    cm->cdef_info.cdef_strengths[0] = 0;
5023
42.8k
    cm->cdef_info.nb_cdef_strengths = 1;
5024
42.8k
    cm->cdef_info.cdef_uv_strengths[0] = 0;
5025
42.8k
    cm->rst_info[0].frame_restoration_type = RESTORE_NONE;
5026
42.8k
    cm->rst_info[1].frame_restoration_type = RESTORE_NONE;
5027
42.8k
    cm->rst_info[2].frame_restoration_type = RESTORE_NONE;
5028
42.8k
  }
5029
5030
214k
  read_tile_info(pbi, rb);
5031
214k
  if (!av1_is_min_tile_width_satisfied(cm)) {
5032
250
    aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
5033
250
                       "Minimum tile width requirement not satisfied");
5034
250
  }
5035
5036
214k
  CommonQuantParams *const quant_params = &cm->quant_params;
5037
214k
  setup_quantization(quant_params, av1_num_planes(cm),
5038
214k
                     cm->seq_params->separate_uv_delta_q, rb);
5039
214k
  xd->bd = (int)seq_params->bit_depth;
5040
5041
214k
  CommonContexts *const above_contexts = &cm->above_contexts;
5042
214k
  if (above_contexts->num_planes < av1_num_planes(cm) ||
5043
214k
      above_contexts->num_mi_cols < cm->mi_params.mi_cols ||
5044
214k
      above_contexts->num_tile_rows < cm->tiles.rows) {
5045
20.3k
    av1_free_above_context_buffers(above_contexts);
5046
20.3k
    if (av1_alloc_above_context_buffers(above_contexts, cm->tiles.rows,
5047
20.3k
                                        cm->mi_params.mi_cols,
5048
20.3k
                                        av1_num_planes(cm))) {
5049
0
      aom_internal_error(&pbi->error, AOM_CODEC_MEM_ERROR,
5050
0
                         "Failed to allocate context buffers");
5051
0
    }
5052
20.3k
  }
5053
5054
214k
  if (features->primary_ref_frame == PRIMARY_REF_NONE) {
5055
137k
    av1_setup_past_independence(cm);
5056
137k
  }
5057
5058
214k
  setup_segmentation(cm, rb);
5059
5060
214k
  cm->delta_q_info.delta_q_res = 1;
5061
214k
  cm->delta_q_info.delta_lf_res = 1;
5062
214k
  cm->delta_q_info.delta_lf_present_flag = 0;
5063
214k
  cm->delta_q_info.delta_lf_multi = 0;
5064
214k
  cm->delta_q_info.delta_q_present_flag =
5065
214k
      quant_params->base_qindex > 0 ? aom_rb_read_bit(rb) : 0;
5066
214k
  if (cm->delta_q_info.delta_q_present_flag) {
5067
23.2k
    xd->current_base_qindex = quant_params->base_qindex;
5068
23.2k
    cm->delta_q_info.delta_q_res = 1 << aom_rb_read_literal(rb, 2);
5069
23.2k
    if (!features->allow_intrabc)
5070
19.0k
      cm->delta_q_info.delta_lf_present_flag = aom_rb_read_bit(rb);
5071
23.2k
    if (cm->delta_q_info.delta_lf_present_flag) {
5072
8.58k
      cm->delta_q_info.delta_lf_res = 1 << aom_rb_read_literal(rb, 2);
5073
8.58k
      cm->delta_q_info.delta_lf_multi = aom_rb_read_bit(rb);
5074
8.58k
      av1_reset_loop_filter_delta(xd, av1_num_planes(cm));
5075
8.58k
    }
5076
23.2k
  }
5077
5078
214k
  xd->cur_frame_force_integer_mv = features->cur_frame_force_integer_mv;
5079
5080
1.58M
  for (int i = 0; i < MAX_SEGMENTS; ++i) {
5081
1.36M
    const int qindex = av1_get_qindex(&cm->seg, i, quant_params->base_qindex);
5082
1.36M
    xd->lossless[i] =
5083
1.36M
        qindex == 0 && quant_params->y_dc_delta_q == 0 &&
5084
1.36M
        quant_params->u_dc_delta_q == 0 && quant_params->u_ac_delta_q == 0 &&
5085
1.36M
        quant_params->v_dc_delta_q == 0 && quant_params->v_ac_delta_q == 0;
5086
1.36M
    xd->qindex[i] = qindex;
5087
1.36M
  }
5088
214k
  features->coded_lossless = is_coded_lossless(cm, xd);
5089
214k
  features->all_lossless = features->coded_lossless && !av1_superres_scaled(cm);
5090
214k
  setup_segmentation_dequant(cm, xd);
5091
214k
  if (features->coded_lossless) {
5092
28.9k
    cm->lf.filter_level[0] = 0;
5093
28.9k
    cm->lf.filter_level[1] = 0;
5094
28.9k
  }
5095
214k
  if (features->coded_lossless || !seq_params->enable_cdef) {
5096
106k
    cm->cdef_info.cdef_bits = 0;
5097
106k
    cm->cdef_info.cdef_strengths[0] = 0;
5098
106k
    cm->cdef_info.cdef_uv_strengths[0] = 0;
5099
106k
  }
5100
214k
  if (features->all_lossless || !seq_params->enable_restoration) {
5101
85.9k
    cm->rst_info[0].frame_restoration_type = RESTORE_NONE;
5102
85.9k
    cm->rst_info[1].frame_restoration_type = RESTORE_NONE;
5103
85.9k
    cm->rst_info[2].frame_restoration_type = RESTORE_NONE;
5104
85.9k
  }
5105
214k
  setup_loopfilter(cm, rb);
5106
5107
214k
  if (!features->coded_lossless && seq_params->enable_cdef) {
5108
64.0k
    setup_cdef(cm, rb);
5109
64.0k
  }
5110
214k
  if (!features->all_lossless && seq_params->enable_restoration) {
5111
83.3k
    decode_restoration_mode(cm, rb);
5112
83.3k
  }
5113
5114
214k
  features->tx_mode = read_tx_mode(rb, features->coded_lossless);
5115
214k
  current_frame->reference_mode = read_frame_reference_mode(cm, rb);
5116
5117
214k
  av1_setup_skip_mode_allowed(cm);
5118
214k
  current_frame->skip_mode_info.skip_mode_flag =
5119
214k
      current_frame->skip_mode_info.skip_mode_allowed ? aom_rb_read_bit(rb) : 0;
5120
5121
214k
  if (frame_might_allow_warped_motion(cm))
5122
28.3k
    features->allow_warped_motion = aom_rb_read_bit(rb);
5123
186k
  else
5124
186k
    features->allow_warped_motion = 0;
5125
5126
214k
  features->reduced_tx_set_used = aom_rb_read_bit(rb);
5127
5128
214k
  if (features->allow_ref_frame_mvs && !frame_might_allow_ref_frame_mvs(cm)) {
5129
0
    aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
5130
0
                       "Frame wrongly requests reference frame MVs");
5131
0
  }
5132
5133
214k
  if (!frame_is_intra_only(cm)) read_global_motion(cm, rb);
5134
5135
214k
  cm->cur_frame->film_grain_params_present =
5136
214k
      seq_params->film_grain_params_present;
5137
214k
  read_film_grain(cm, rb);
5138
5139
214k
#if EXT_TILE_DEBUG
5140
214k
  if (pbi->ext_tile_debug && cm->tiles.large_scale) {
5141
19.9k
    read_ext_tile_info(pbi, rb);
5142
19.9k
    av1_set_single_tile_decoding_mode(cm);
5143
19.9k
  }
5144
214k
#endif  // EXT_TILE_DEBUG
5145
214k
  return 0;
5146
218k
}
5147
5148
struct aom_read_bit_buffer *av1_init_read_bit_buffer(
5149
    AV1Decoder *pbi, struct aom_read_bit_buffer *rb, const uint8_t *data,
5150
425k
    const uint8_t *data_end) {
5151
425k
  rb->bit_offset = 0;
5152
425k
  rb->error_handler = error_handler;
5153
425k
  rb->error_handler_data = &pbi->common;
5154
425k
  rb->bit_buffer = data;
5155
425k
  rb->bit_buffer_end = data_end;
5156
425k
  return rb;
5157
425k
}
5158
5159
203k
BITSTREAM_PROFILE av1_read_profile(struct aom_read_bit_buffer *rb) {
5160
203k
  int profile = aom_rb_read_literal(rb, PROFILE_BITS);
5161
203k
  return (BITSTREAM_PROFILE)profile;
5162
203k
}
5163
5164
30.7k
static inline void superres_post_decode(AV1Decoder *pbi) {
5165
30.7k
  AV1_COMMON *const cm = &pbi->common;
5166
30.7k
  BufferPool *const pool = cm->buffer_pool;
5167
5168
30.7k
  if (!av1_superres_scaled(cm)) return;
5169
11.4k
  assert(!cm->features.all_lossless);
5170
5171
11.4k
  av1_superres_upscale(cm, pool, 0);
5172
11.4k
}
5173
5174
uint32_t av1_decode_frame_headers_and_setup(AV1Decoder *pbi,
5175
                                            struct aom_read_bit_buffer *rb,
5176
218k
                                            int trailing_bits_present) {
5177
218k
  AV1_COMMON *const cm = &pbi->common;
5178
218k
  const int num_planes = av1_num_planes(cm);
5179
218k
  MACROBLOCKD *const xd = &pbi->dcb.xd;
5180
5181
#if CONFIG_BITSTREAM_DEBUG
5182
  if (cm->seq_params->order_hint_info.enable_order_hint) {
5183
    aom_bitstream_queue_set_frame_read(cm->current_frame.order_hint * 2 +
5184
                                       cm->show_frame);
5185
  } else {
5186
    // This is currently used in RTC encoding. cm->show_frame is always 1.
5187
    assert(cm->show_frame);
5188
    aom_bitstream_queue_set_frame_read(cm->current_frame.frame_number);
5189
  }
5190
#endif
5191
#if CONFIG_MISMATCH_DEBUG
5192
  mismatch_move_frame_idx_r();
5193
#endif
5194
5195
1.74M
  for (int i = LAST_FRAME; i <= ALTREF_FRAME; ++i) {
5196
1.52M
    cm->global_motion[i] = default_warp_params;
5197
1.52M
    cm->cur_frame->global_motion[i] = default_warp_params;
5198
1.52M
  }
5199
218k
  xd->global_motion = cm->global_motion;
5200
5201
218k
  read_uncompressed_header(pbi, rb);
5202
5203
218k
  if (trailing_bits_present) av1_check_trailing_bits(pbi, rb);
5204
5205
218k
  if (!cm->tiles.single_tile_decoding &&
5206
218k
      (pbi->dec_tile_row >= 0 || pbi->dec_tile_col >= 0)) {
5207
0
    pbi->dec_tile_row = -1;
5208
0
    pbi->dec_tile_col = -1;
5209
0
  }
5210
5211
218k
  const uint32_t uncomp_hdr_size =
5212
218k
      (uint32_t)aom_rb_bytes_read(rb);  // Size of the uncompressed header
5213
218k
  YV12_BUFFER_CONFIG *new_fb = &cm->cur_frame->buf;
5214
218k
  xd->cur_buf = new_fb;
5215
218k
  if (av1_allow_intrabc(cm)) {
5216
40.6k
    av1_setup_scale_factors_for_frame(
5217
40.6k
        &cm->sf_identity, xd->cur_buf->y_crop_width, xd->cur_buf->y_crop_height,
5218
40.6k
        xd->cur_buf->y_crop_width, xd->cur_buf->y_crop_height);
5219
40.6k
  }
5220
5221
  // Showing a frame directly.
5222
218k
  if (cm->show_existing_frame) {
5223
333
    if (pbi->reset_decoder_state) {
5224
      // Use the default frame context values.
5225
49
      *cm->fc = *cm->default_frame_context;
5226
49
      if (!cm->fc->initialized)
5227
0
        aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
5228
0
                           "Uninitialized entropy context.");
5229
49
    }
5230
333
    return uncomp_hdr_size;
5231
333
  }
5232
5233
217k
  cm->mi_params.setup_mi(&cm->mi_params);
5234
5235
217k
  av1_calculate_ref_frame_side(cm);
5236
217k
  if (cm->features.allow_ref_frame_mvs) av1_setup_motion_field(cm);
5237
5238
217k
  av1_setup_block_planes(xd, cm->seq_params->subsampling_x,
5239
217k
                         cm->seq_params->subsampling_y, num_planes);
5240
217k
  if (cm->features.primary_ref_frame == PRIMARY_REF_NONE) {
5241
    // use the default frame context values
5242
129k
    *cm->fc = *cm->default_frame_context;
5243
129k
  } else {
5244
88.8k
    *cm->fc = get_primary_ref_frame_buf(cm)->frame_context;
5245
88.8k
  }
5246
217k
  if (!cm->fc->initialized)
5247
37
    aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
5248
37
                       "Uninitialized entropy context.");
5249
5250
217k
  pbi->dcb.corrupted = 0;
5251
217k
  return uncomp_hdr_size;
5252
218k
}
5253
5254
// Once-per-frame initialization
5255
147k
static inline void setup_frame_info(AV1Decoder *pbi) {
5256
147k
  AV1_COMMON *const cm = &pbi->common;
5257
5258
147k
  if (cm->rst_info[0].frame_restoration_type != RESTORE_NONE ||
5259
147k
      cm->rst_info[1].frame_restoration_type != RESTORE_NONE ||
5260
147k
      cm->rst_info[2].frame_restoration_type != RESTORE_NONE) {
5261
32.1k
    av1_alloc_restoration_buffers(cm, /*is_sgr_enabled =*/true);
5262
118k
    for (int p = 0; p < av1_num_planes(cm); p++) {
5263
86.1k
      av1_alloc_restoration_struct(cm, &cm->rst_info[p], p > 0);
5264
86.1k
    }
5265
32.1k
  }
5266
5267
147k
  const int use_highbd = cm->seq_params->use_highbitdepth;
5268
147k
  const int buf_size = MC_TEMP_BUF_PELS << use_highbd;
5269
147k
  if (pbi->td.mc_buf_size != buf_size) {
5270
18.3k
    av1_free_mc_tmp_buf(&pbi->td);
5271
18.3k
    allocate_mc_tmp_buf(cm, &pbi->td, buf_size, use_highbd);
5272
18.3k
  }
5273
147k
}
5274
5275
void av1_decode_tg_tiles_and_wrapup(AV1Decoder *pbi, const uint8_t *data,
5276
                                    const uint8_t *data_end,
5277
                                    const uint8_t **p_data_end, int start_tile,
5278
147k
                                    int end_tile, int initialize_flag) {
5279
147k
  AV1_COMMON *const cm = &pbi->common;
5280
147k
  CommonTileParams *const tiles = &cm->tiles;
5281
147k
  MACROBLOCKD *const xd = &pbi->dcb.xd;
5282
147k
  const int tile_count_tg = end_tile - start_tile + 1;
5283
5284
147k
  xd->error_info = cm->error;
5285
147k
  if (initialize_flag) setup_frame_info(pbi);
5286
147k
  const int num_planes = av1_num_planes(cm);
5287
5288
147k
  if (pbi->max_threads > 1 && !(tiles->large_scale && !pbi->ext_tile_debug) &&
5289
147k
      pbi->row_mt)
5290
67.3k
    *p_data_end =
5291
67.3k
        decode_tiles_row_mt(pbi, data, data_end, start_tile, end_tile);
5292
80.5k
  else if (pbi->max_threads > 1 && tile_count_tg > 1 &&
5293
80.5k
           !(tiles->large_scale && !pbi->ext_tile_debug))
5294
0
    *p_data_end = decode_tiles_mt(pbi, data, data_end, start_tile, end_tile);
5295
80.5k
  else
5296
80.5k
    *p_data_end = decode_tiles(pbi, data, data_end, start_tile, end_tile);
5297
5298
  // If the bit stream is monochrome, set the U and V buffers to a constant.
5299
147k
  if (num_planes < 3) {
5300
11.7k
    set_planes_to_neutral_grey(cm->seq_params, xd->cur_buf, 1);
5301
11.7k
  }
5302
5303
147k
  if (end_tile != tiles->rows * tiles->cols - 1) {
5304
6
    return;
5305
6
  }
5306
5307
147k
  av1_alloc_cdef_buffers(cm, &pbi->cdef_worker, &pbi->cdef_sync,
5308
147k
                         pbi->num_workers, 1);
5309
147k
  av1_alloc_cdef_sync(cm, &pbi->cdef_sync, pbi->num_workers);
5310
5311
147k
  if (!cm->features.allow_intrabc && !tiles->single_tile_decoding) {
5312
77.5k
    if (cm->lf.filter_level[0] || cm->lf.filter_level[1]) {
5313
32.2k
      av1_loop_filter_frame_mt(&cm->cur_frame->buf, cm, &pbi->dcb.xd, 0,
5314
32.2k
                               num_planes, 0, pbi->tile_workers,
5315
32.2k
                               pbi->num_workers, &pbi->lf_row_sync, 0);
5316
32.2k
    }
5317
5318
77.5k
    const int do_cdef =
5319
77.5k
        !pbi->skip_loop_filter && !cm->features.coded_lossless &&
5320
77.5k
        (cm->cdef_info.cdef_bits || cm->cdef_info.cdef_strengths[0] ||
5321
60.4k
         cm->cdef_info.cdef_uv_strengths[0]);
5322
77.5k
    const int do_superres = av1_superres_scaled(cm);
5323
77.5k
    const int optimized_loop_restoration = !do_cdef && !do_superres;
5324
77.5k
    const int do_loop_restoration =
5325
77.5k
        cm->rst_info[0].frame_restoration_type != RESTORE_NONE ||
5326
77.5k
        cm->rst_info[1].frame_restoration_type != RESTORE_NONE ||
5327
77.5k
        cm->rst_info[2].frame_restoration_type != RESTORE_NONE;
5328
    // Frame border extension is not required in the decoder
5329
    // as it happens in extend_mc_border().
5330
77.5k
    int do_extend_border_mt = 0;
5331
77.5k
    if (!optimized_loop_restoration) {
5332
30.7k
      if (do_loop_restoration)
5333
20.2k
        av1_loop_restoration_save_boundary_lines(&pbi->common.cur_frame->buf,
5334
20.2k
                                                 cm, 0);
5335
5336
30.7k
      if (do_cdef) {
5337
23.8k
        if (pbi->num_workers > 1) {
5338
16.7k
          av1_cdef_frame_mt(cm, &pbi->dcb.xd, pbi->cdef_worker,
5339
16.7k
                            pbi->tile_workers, &pbi->cdef_sync,
5340
16.7k
                            pbi->num_workers, av1_cdef_init_fb_row_mt,
5341
16.7k
                            do_extend_border_mt);
5342
16.7k
        } else {
5343
7.08k
          av1_cdef_frame(&pbi->common.cur_frame->buf, cm, &pbi->dcb.xd,
5344
7.08k
                         av1_cdef_init_fb_row);
5345
7.08k
        }
5346
23.8k
      }
5347
5348
30.7k
      superres_post_decode(pbi);
5349
5350
30.7k
      if (do_loop_restoration) {
5351
20.2k
        av1_loop_restoration_save_boundary_lines(&pbi->common.cur_frame->buf,
5352
20.2k
                                                 cm, 1);
5353
20.2k
        if (pbi->num_workers > 1) {
5354
14.3k
          av1_loop_restoration_filter_frame_mt(
5355
14.3k
              (YV12_BUFFER_CONFIG *)xd->cur_buf, cm, optimized_loop_restoration,
5356
14.3k
              pbi->tile_workers, pbi->num_workers, &pbi->lr_row_sync,
5357
14.3k
              &pbi->lr_ctxt, do_extend_border_mt);
5358
14.3k
        } else {
5359
5.94k
          av1_loop_restoration_filter_frame((YV12_BUFFER_CONFIG *)xd->cur_buf,
5360
5.94k
                                            cm, optimized_loop_restoration,
5361
5.94k
                                            &pbi->lr_ctxt);
5362
5.94k
        }
5363
20.2k
      }
5364
46.7k
    } else {
5365
      // In no cdef and no superres case. Provide an optimized version of
5366
      // loop_restoration_filter.
5367
46.7k
      if (do_loop_restoration) {
5368
2.88k
        if (pbi->num_workers > 1) {
5369
872
          av1_loop_restoration_filter_frame_mt(
5370
872
              (YV12_BUFFER_CONFIG *)xd->cur_buf, cm, optimized_loop_restoration,
5371
872
              pbi->tile_workers, pbi->num_workers, &pbi->lr_row_sync,
5372
872
              &pbi->lr_ctxt, do_extend_border_mt);
5373
2.01k
        } else {
5374
2.01k
          av1_loop_restoration_filter_frame((YV12_BUFFER_CONFIG *)xd->cur_buf,
5375
2.01k
                                            cm, optimized_loop_restoration,
5376
2.01k
                                            &pbi->lr_ctxt);
5377
2.01k
        }
5378
2.88k
      }
5379
46.7k
    }
5380
77.5k
  }
5381
5382
147k
  if (!pbi->dcb.corrupted) {
5383
92.7k
    if (cm->features.refresh_frame_context == REFRESH_FRAME_CONTEXT_BACKWARD) {
5384
27.3k
      assert(pbi->context_update_tile_id < pbi->allocated_tiles);
5385
27.3k
      *cm->fc = pbi->tile_data[pbi->context_update_tile_id].tctx;
5386
27.3k
      av1_reset_cdf_symbol_counters(cm->fc);
5387
27.3k
    }
5388
92.7k
  } else {
5389
55.1k
    aom_internal_error(&pbi->error, AOM_CODEC_CORRUPT_FRAME,
5390
55.1k
                       "Decode failed. Frame data is corrupted.");
5391
55.1k
  }
5392
5393
#if CONFIG_INSPECTION
5394
  if (pbi->inspect_cb != NULL) {
5395
    (*pbi->inspect_cb)(pbi, pbi->inspect_ctx);
5396
  }
5397
#endif
5398
5399
  // Non frame parallel update frame context here.
5400
147k
  if (!tiles->large_scale) {
5401
82.6k
    cm->cur_frame->frame_context = *cm->fc;
5402
82.6k
  }
5403
5404
147k
  if (cm->show_frame && !cm->seq_params->order_hint_info.enable_order_hint) {
5405
12.1k
    ++cm->current_frame.frame_number;
5406
12.1k
  }
5407
147k
}