diff mbox series

[FFmpeg-devel] avcodec/vaapi_av1: fill the remaining VAFilmGrainStructAV1 fields

Message ID 20201130224258.5225-1-jamrial@gmail.com
State Accepted
Commit 0e62efad7cd6b1fa07dcb3badb6ab92957a576dc
Headers show
Series [FFmpeg-devel] avcodec/vaapi_av1: fill the remaining VAFilmGrainStructAV1 fields | expand

Checks

Context Check Description
andriy/x86_make success Make finished
andriy/x86_make_fate success Make fate finished
andriy/PPC64_make success Make finished
andriy/PPC64_make_fate success Make fate finished

Commit Message

James Almer Nov. 30, 2020, 10:42 p.m. UTC
Signed-off-by: James Almer <jamrial@gmail.com>
---
 libavcodec/vaapi_av1.c | 60 +++++++++++++++++++++++++++++++++++-------
 1 file changed, 51 insertions(+), 9 deletions(-)

Comments

James Almer Dec. 3, 2020, 4:46 p.m. UTC | #1
On 11/30/2020 7:42 PM, James Almer wrote:
> Signed-off-by: James Almer <jamrial@gmail.com>
> ---
>   libavcodec/vaapi_av1.c | 60 +++++++++++++++++++++++++++++++++++-------
>   1 file changed, 51 insertions(+), 9 deletions(-)
> 
> diff --git a/libavcodec/vaapi_av1.c b/libavcodec/vaapi_av1.c
> index 2763c1e866..b76e217daa 100644
> --- a/libavcodec/vaapi_av1.c
> +++ b/libavcodec/vaapi_av1.c
> @@ -115,15 +115,27 @@ static int vaapi_av1_start_frame(AVCodecContext *avctx,
>               .temporal_update = frame_header->segmentation_temporal_update,
>               .update_data     = frame_header->segmentation_update_data,
>           },
> -        .film_grain_info.film_grain_info_fields.bits = {
> -            .apply_grain              = film_grain->apply_grain,
> -            .chroma_scaling_from_luma = film_grain->chroma_scaling_from_luma,
> -            .grain_scaling_minus_8    = film_grain->grain_scaling_minus_8,
> -            .ar_coeff_lag             = film_grain->ar_coeff_lag,
> -            .ar_coeff_shift_minus_6   = film_grain->ar_coeff_shift_minus_6,
> -            .grain_scale_shift        = film_grain->grain_scale_shift,
> -            .overlap_flag             = film_grain->overlap_flag,
> -            .clip_to_restricted_range = film_grain->clip_to_restricted_range,
> +        .film_grain_info = {
> +            .film_grain_info_fields.bits = {
> +                .apply_grain              = film_grain->apply_grain,
> +                .chroma_scaling_from_luma = film_grain->chroma_scaling_from_luma,
> +                .grain_scaling_minus_8    = film_grain->grain_scaling_minus_8,
> +                .ar_coeff_lag             = film_grain->ar_coeff_lag,
> +                .ar_coeff_shift_minus_6   = film_grain->ar_coeff_shift_minus_6,
> +                .grain_scale_shift        = film_grain->grain_scale_shift,
> +                .overlap_flag             = film_grain->overlap_flag,
> +                .clip_to_restricted_range = film_grain->clip_to_restricted_range,
> +            },
> +            .grain_seed    = film_grain->grain_seed,
> +            .num_y_points  = film_grain->num_y_points,
> +            .num_cb_points = film_grain->num_cb_points,
> +            .num_cr_points = film_grain->num_cr_points,
> +            .cb_mult       = film_grain->cb_mult,
> +            .cb_luma_mult  = film_grain->cb_luma_mult,
> +            .cb_offset     = film_grain->cb_offset,
> +            .cr_mult       = film_grain->cr_mult,
> +            .cr_luma_mult  = film_grain->cr_luma_mult,
> +            .cr_offset     = film_grain->cr_offset,
>           },
>           .pic_info_fields.bits = {
>               .frame_type                   = frame_header->frame_type,
> @@ -203,6 +215,36 @@ static int vaapi_av1_start_frame(AVCodecContext *avctx,
>           for (int j = 0; j < 6; j++)
>               pic_param.wm[i - 1].wmmat[j] = s->cur_frame.gm_params[i][j];
>       }
> +    if (film_grain->apply_grain) {
> +        for (int i = 0; i < film_grain->num_y_points; i++) {
> +            pic_param.film_grain_info.point_y_value[i] =
> +                film_grain->point_y_value[i];
> +            pic_param.film_grain_info.point_y_scaling[i] =
> +                film_grain->point_y_scaling[i];
> +        }
> +        for (int i = 0; i < film_grain->num_cb_points; i++) {
> +            pic_param.film_grain_info.point_cb_value[i] =
> +                film_grain->point_cb_value[i];
> +            pic_param.film_grain_info.point_cb_scaling[i] =
> +                film_grain->point_cb_scaling[i];
> +        }
> +        for (int i = 0; i < film_grain->num_cr_points; i++) {
> +            pic_param.film_grain_info.point_cr_value[i] =
> +                film_grain->point_cr_value[i];
> +            pic_param.film_grain_info.point_cr_scaling[i] =
> +                film_grain->point_cr_scaling[i];
> +        }
> +        for (int i = 0; i < 24; i++) {
> +            pic_param.film_grain_info.ar_coeffs_y[i] =
> +                film_grain->ar_coeffs_y_plus_128[i] - 128;
> +        }
> +        for (int i = 0; i < 25; i++) {
> +            pic_param.film_grain_info.ar_coeffs_cb[i] =
> +                film_grain->ar_coeffs_cb_plus_128[i] - 128;
> +            pic_param.film_grain_info.ar_coeffs_cr[i] =
> +                film_grain->ar_coeffs_cr_plus_128[i] - 128;
> +        }
> +    }
>       err = ff_vaapi_decode_make_param_buffer(avctx, pic,
>                                               VAPictureParameterBufferType,
>                                               &pic_param, sizeof(pic_param));

Ping.
James Almer Dec. 9, 2020, 3:34 p.m. UTC | #2
On 12/3/2020 1:46 PM, James Almer wrote:
> On 11/30/2020 7:42 PM, James Almer wrote:
>> Signed-off-by: James Almer <jamrial@gmail.com>
>> ---
>>   libavcodec/vaapi_av1.c | 60 +++++++++++++++++++++++++++++++++++-------
>>   1 file changed, 51 insertions(+), 9 deletions(-)
>>
>> diff --git a/libavcodec/vaapi_av1.c b/libavcodec/vaapi_av1.c
>> index 2763c1e866..b76e217daa 100644
>> --- a/libavcodec/vaapi_av1.c
>> +++ b/libavcodec/vaapi_av1.c
>> @@ -115,15 +115,27 @@ static int vaapi_av1_start_frame(AVCodecContext 
>> *avctx,
>>               .temporal_update = 
>> frame_header->segmentation_temporal_update,
>>               .update_data     = frame_header->segmentation_update_data,
>>           },
>> -        .film_grain_info.film_grain_info_fields.bits = {
>> -            .apply_grain              = film_grain->apply_grain,
>> -            .chroma_scaling_from_luma = 
>> film_grain->chroma_scaling_from_luma,
>> -            .grain_scaling_minus_8    = 
>> film_grain->grain_scaling_minus_8,
>> -            .ar_coeff_lag             = film_grain->ar_coeff_lag,
>> -            .ar_coeff_shift_minus_6   = 
>> film_grain->ar_coeff_shift_minus_6,
>> -            .grain_scale_shift        = film_grain->grain_scale_shift,
>> -            .overlap_flag             = film_grain->overlap_flag,
>> -            .clip_to_restricted_range = 
>> film_grain->clip_to_restricted_range,
>> +        .film_grain_info = {
>> +            .film_grain_info_fields.bits = {
>> +                .apply_grain              = film_grain->apply_grain,
>> +                .chroma_scaling_from_luma = 
>> film_grain->chroma_scaling_from_luma,
>> +                .grain_scaling_minus_8    = 
>> film_grain->grain_scaling_minus_8,
>> +                .ar_coeff_lag             = film_grain->ar_coeff_lag,
>> +                .ar_coeff_shift_minus_6   = 
>> film_grain->ar_coeff_shift_minus_6,
>> +                .grain_scale_shift        = 
>> film_grain->grain_scale_shift,
>> +                .overlap_flag             = film_grain->overlap_flag,
>> +                .clip_to_restricted_range = 
>> film_grain->clip_to_restricted_range,
>> +            },
>> +            .grain_seed    = film_grain->grain_seed,
>> +            .num_y_points  = film_grain->num_y_points,
>> +            .num_cb_points = film_grain->num_cb_points,
>> +            .num_cr_points = film_grain->num_cr_points,
>> +            .cb_mult       = film_grain->cb_mult,
>> +            .cb_luma_mult  = film_grain->cb_luma_mult,
>> +            .cb_offset     = film_grain->cb_offset,
>> +            .cr_mult       = film_grain->cr_mult,
>> +            .cr_luma_mult  = film_grain->cr_luma_mult,
>> +            .cr_offset     = film_grain->cr_offset,
>>           },
>>           .pic_info_fields.bits = {
>>               .frame_type                   = frame_header->frame_type,
>> @@ -203,6 +215,36 @@ static int vaapi_av1_start_frame(AVCodecContext 
>> *avctx,
>>           for (int j = 0; j < 6; j++)
>>               pic_param.wm[i - 1].wmmat[j] = 
>> s->cur_frame.gm_params[i][j];
>>       }
>> +    if (film_grain->apply_grain) {
>> +        for (int i = 0; i < film_grain->num_y_points; i++) {
>> +            pic_param.film_grain_info.point_y_value[i] =
>> +                film_grain->point_y_value[i];
>> +            pic_param.film_grain_info.point_y_scaling[i] =
>> +                film_grain->point_y_scaling[i];
>> +        }
>> +        for (int i = 0; i < film_grain->num_cb_points; i++) {
>> +            pic_param.film_grain_info.point_cb_value[i] =
>> +                film_grain->point_cb_value[i];
>> +            pic_param.film_grain_info.point_cb_scaling[i] =
>> +                film_grain->point_cb_scaling[i];
>> +        }
>> +        for (int i = 0; i < film_grain->num_cr_points; i++) {
>> +            pic_param.film_grain_info.point_cr_value[i] =
>> +                film_grain->point_cr_value[i];
>> +            pic_param.film_grain_info.point_cr_scaling[i] =
>> +                film_grain->point_cr_scaling[i];
>> +        }
>> +        for (int i = 0; i < 24; i++) {
>> +            pic_param.film_grain_info.ar_coeffs_y[i] =
>> +                film_grain->ar_coeffs_y_plus_128[i] - 128;
>> +        }
>> +        for (int i = 0; i < 25; i++) {
>> +            pic_param.film_grain_info.ar_coeffs_cb[i] =
>> +                film_grain->ar_coeffs_cb_plus_128[i] - 128;
>> +            pic_param.film_grain_info.ar_coeffs_cr[i] =
>> +                film_grain->ar_coeffs_cr_plus_128[i] - 128;
>> +        }
>> +    }
>>       err = ff_vaapi_decode_make_param_buffer(avctx, pic,
>>                                               
>> VAPictureParameterBufferType,
>>                                               &pic_param, 
>> sizeof(pic_param));
> 
> Ping.

Will apply it soon.
diff mbox series

Patch

diff --git a/libavcodec/vaapi_av1.c b/libavcodec/vaapi_av1.c
index 2763c1e866..b76e217daa 100644
--- a/libavcodec/vaapi_av1.c
+++ b/libavcodec/vaapi_av1.c
@@ -115,15 +115,27 @@  static int vaapi_av1_start_frame(AVCodecContext *avctx,
             .temporal_update = frame_header->segmentation_temporal_update,
             .update_data     = frame_header->segmentation_update_data,
         },
-        .film_grain_info.film_grain_info_fields.bits = {
-            .apply_grain              = film_grain->apply_grain,
-            .chroma_scaling_from_luma = film_grain->chroma_scaling_from_luma,
-            .grain_scaling_minus_8    = film_grain->grain_scaling_minus_8,
-            .ar_coeff_lag             = film_grain->ar_coeff_lag,
-            .ar_coeff_shift_minus_6   = film_grain->ar_coeff_shift_minus_6,
-            .grain_scale_shift        = film_grain->grain_scale_shift,
-            .overlap_flag             = film_grain->overlap_flag,
-            .clip_to_restricted_range = film_grain->clip_to_restricted_range,
+        .film_grain_info = {
+            .film_grain_info_fields.bits = {
+                .apply_grain              = film_grain->apply_grain,
+                .chroma_scaling_from_luma = film_grain->chroma_scaling_from_luma,
+                .grain_scaling_minus_8    = film_grain->grain_scaling_minus_8,
+                .ar_coeff_lag             = film_grain->ar_coeff_lag,
+                .ar_coeff_shift_minus_6   = film_grain->ar_coeff_shift_minus_6,
+                .grain_scale_shift        = film_grain->grain_scale_shift,
+                .overlap_flag             = film_grain->overlap_flag,
+                .clip_to_restricted_range = film_grain->clip_to_restricted_range,
+            },
+            .grain_seed    = film_grain->grain_seed,
+            .num_y_points  = film_grain->num_y_points,
+            .num_cb_points = film_grain->num_cb_points,
+            .num_cr_points = film_grain->num_cr_points,
+            .cb_mult       = film_grain->cb_mult,
+            .cb_luma_mult  = film_grain->cb_luma_mult,
+            .cb_offset     = film_grain->cb_offset,
+            .cr_mult       = film_grain->cr_mult,
+            .cr_luma_mult  = film_grain->cr_luma_mult,
+            .cr_offset     = film_grain->cr_offset,
         },
         .pic_info_fields.bits = {
             .frame_type                   = frame_header->frame_type,
@@ -203,6 +215,36 @@  static int vaapi_av1_start_frame(AVCodecContext *avctx,
         for (int j = 0; j < 6; j++)
             pic_param.wm[i - 1].wmmat[j] = s->cur_frame.gm_params[i][j];
     }
+    if (film_grain->apply_grain) {
+        for (int i = 0; i < film_grain->num_y_points; i++) {
+            pic_param.film_grain_info.point_y_value[i] =
+                film_grain->point_y_value[i];
+            pic_param.film_grain_info.point_y_scaling[i] =
+                film_grain->point_y_scaling[i];
+        }
+        for (int i = 0; i < film_grain->num_cb_points; i++) {
+            pic_param.film_grain_info.point_cb_value[i] =
+                film_grain->point_cb_value[i];
+            pic_param.film_grain_info.point_cb_scaling[i] =
+                film_grain->point_cb_scaling[i];
+        }
+        for (int i = 0; i < film_grain->num_cr_points; i++) {
+            pic_param.film_grain_info.point_cr_value[i] =
+                film_grain->point_cr_value[i];
+            pic_param.film_grain_info.point_cr_scaling[i] =
+                film_grain->point_cr_scaling[i];
+        }
+        for (int i = 0; i < 24; i++) {
+            pic_param.film_grain_info.ar_coeffs_y[i] =
+                film_grain->ar_coeffs_y_plus_128[i] - 128;
+        }
+        for (int i = 0; i < 25; i++) {
+            pic_param.film_grain_info.ar_coeffs_cb[i] =
+                film_grain->ar_coeffs_cb_plus_128[i] - 128;
+            pic_param.film_grain_info.ar_coeffs_cr[i] =
+                film_grain->ar_coeffs_cr_plus_128[i] - 128;
+        }
+    }
     err = ff_vaapi_decode_make_param_buffer(avctx, pic,
                                             VAPictureParameterBufferType,
                                             &pic_param, sizeof(pic_param));