[FFmpeg-devel] [PATCH 2/2] swresample/rematrix: add support for custom order channel layouts

[James Almer]

On 2/2/2025 6:46 PM, James Almer wrote:
> Limited to the same channels as a native layout, but not constrained by channel ordering.
> 
> Signed-off-by: James Almer <jamrial at gmail.com>
> ---
> Untested, as i don't have a sample from which we export a custom layout (Like PCM in mp4).

Tried with one such sample and it appears to work.

> 
>   libswresample/rematrix.c | 28 +++++++++++++++++-----------
>   1 file changed, 17 insertions(+), 11 deletions(-)
> 
> diff --git a/libswresample/rematrix.c b/libswresample/rematrix.c
> index 8b6e8ae1c5..e3c2cc1211 100644
> --- a/libswresample/rematrix.c
> +++ b/libswresample/rematrix.c
> @@ -104,6 +104,14 @@ static int clean_layout(AVChannelLayout *out, const AVChannelLayout *in, void *s
>   }
>   
>   static int sane_layout(AVChannelLayout *ch_layout) {
> +    if(ch_layout->nb_channels >= SWR_CH_MAX)
> +        return 0;
> +    if(ch_layout->order == AV_CHANNEL_ORDER_CUSTOM)
> +        for (int i = 0; i < ch_layout->nb_channels; i++) {
> +            if (ch_layout->u.map[i].id >= 64)
> +                return 0;
> +        }
> +    else
>       if (ch_layout->order != AV_CHANNEL_ORDER_NATIVE)
>           return 0;
>       if(!av_channel_layout_subset(ch_layout, AV_CH_LAYOUT_SURROUND)) // at least 1 front speaker
> @@ -118,8 +126,6 @@ static int sane_layout(AVChannelLayout *ch_layout) {
>           return 0;
>       if(!even(av_channel_layout_subset(ch_layout, (AV_CH_TOP_FRONT_LEFT | AV_CH_TOP_FRONT_RIGHT))))
>           return 0;
> -    if(ch_layout->nb_channels >= SWR_CH_MAX)
> -        return 0;
>   
>       return 1;
>   }
> @@ -130,9 +136,10 @@ static void build_matrix(const AVChannelLayout *in_ch_layout, const AVChannelLay
>                            ptrdiff_t stride, enum AVMatrixEncoding matrix_encoding)
>   {
>       double matrix[NUM_NAMED_CHANNELS][NUM_NAMED_CHANNELS]={{0}};
> -    uint64_t unaccounted = in_ch_layout->u.mask & ~out_ch_layout->u.mask;
> +    uint64_t unaccounted =  av_channel_layout_subset(in_ch_layout,  UINT64_MAX) &
> +                           ~av_channel_layout_subset(out_ch_layout, UINT64_MAX);
>       double maxcoef=0;
> -    int i, j, out_i;
> +    int i, j;
>   
>       for(i=0; i<FF_ARRAY_ELEMS(matrix); i++){
>           if(   av_channel_layout_index_from_channel(in_ch_layout, i) >= 0
> @@ -305,14 +312,15 @@ static void build_matrix(const AVChannelLayout *in_ch_layout, const AVChannelLay
>       }
>   
>   
> -    for(out_i=i=0; i<64; i++){
> +    for(i=0; i<64; i++){
>           double sum=0;
> -        int in_i=0;
> -        if (av_channel_layout_index_from_channel(out_ch_layout, i) < 0)
> +        int out_i = av_channel_layout_index_from_channel(out_ch_layout, i);
> +        if (out_i < 0)
>               continue;
>           for(j=0; j<64; j++){
> -            if (av_channel_layout_index_from_channel(in_ch_layout, j) < 0)
> -               continue;
> +            int in_i = av_channel_layout_index_from_channel(in_ch_layout, j);
> +            if (in_i < 0)
> +                continue;
>               if (i < FF_ARRAY_ELEMS(matrix) && j < FF_ARRAY_ELEMS(matrix[0]))
>                   matrix_param[stride*out_i + in_i] = matrix[i][j];
>               else
> @@ -320,10 +328,8 @@ static void build_matrix(const AVChannelLayout *in_ch_layout, const AVChannelLay
>                   (   av_channel_layout_index_from_channel(in_ch_layout, i) >= 0
>                    && av_channel_layout_index_from_channel(out_ch_layout, i) >= 0);
>               sum += fabs(matrix_param[stride*out_i + in_i]);
> -            in_i++;
>           }
>           maxcoef= FFMAX(maxcoef, sum);
> -        out_i++;
>       }
>       if(rematrix_volume  < 0)
>           maxcoef = -rematrix_volume;

-------------- next part --------------
A non-text attachment was scrubbed...
Name: OpenPGP_signature.asc
Type: application/pgp-signature
Size: 495 bytes
Desc: OpenPGP digital signature
URL: <https://ffmpeg.org/pipermail/ffmpeg-devel/attachments/20250202/342c14af/attachment.sig>