avcodec/cbs_av1: add missing frame restoration type enum values

Signed-off-by: Fei Wang <fei.w.wang@intel.com>
Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
Fei Wang 2020-09-02 15:31:28 +08:00 committed by James Almer
parent 2a19232c19
commit bf6db5732c
2 changed files with 9 additions and 1 deletions

View File

@ -160,4 +160,12 @@ enum {
AV1_SCALABILITY_L4T7_KEY_SHIFT = 28,
};
// Frame Restoration types (section 6.10.15)
enum {
AV1_RESTORE_NONE = 0,
AV1_RESTORE_WIENER = 1,
AV1_RESTORE_SGRPROJ = 2,
AV1_RESTORE_SWITCHABLE = 3,
};
#endif /* AVCODEC_AV1_H */

View File

@ -940,7 +940,7 @@ static int FUNC(lr_params)(CodedBitstreamContext *ctx, RWContext *rw,
for (i = 0; i < priv->num_planes; i++) {
fbs(2, lr_type[i], 1, i);
if (current->lr_type[i] != 0) {
if (current->lr_type[i] != AV1_RESTORE_NONE) {
uses_lr = 1;
if (i > 0)
uses_chroma_lr = 1;