/*

 * Copyright (c) 2007-2008 Ian Caulfield

 *               2009 Ramiro Polla

 *

 * This file is part of FFmpeg.

 *

 * FFmpeg is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public

 * License as published by the Free Software Foundation; either

 * version 2.1 of the License, or (at your option) any later version.

 *

 * FFmpeg is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with FFmpeg; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

 */

#include "config.h"

#include "libavutil/attributes.h"

#include "mlpdsp.h"

#include "mlp.h"

static void mlp_filter_channel(int32_t *state, const int32_t *coeff,

                               int firorder, int iirorder,

                               unsigned int filter_shift, int32_t mask,

                               int blocksize, int32_t *sample_buffer)

{

    int32_t *firbuf = state;

    int32_t *iirbuf = state + MAX_BLOCKSIZE + MAX_FIR_ORDER;

    const int32_t *fircoeff = coeff;

    const int32_t *iircoeff = coeff + MAX_FIR_ORDER;

    int i;

    for (i = 0; i < blocksize; i++) {

        int32_t residual = *sample_buffer;

        unsigned int order;

        int64_t accum = 0;

        int32_t result;

        for (order = 0; order < firorder; order++)

            accum += (int64_t) firbuf[order] * fircoeff[order];

        for (order = 0; order < iirorder; order++)

            accum += (int64_t) iirbuf[order] * iircoeff[order];

        accum  = accum >> filter_shift;

        result = (accum + residual) & mask;

        *--firbuf = result;

        *--iirbuf = result - accum;

        *sample_buffer = result;

        sample_buffer += MAX_CHANNELS;

    }

}

void ff_mlp_rematrix_channel(int32_t *samples,

                             const int32_t *coeffs,

                             const uint8_t *bypassed_lsbs,

                             const int8_t *noise_buffer,

                             int index,

                             unsigned int dest_ch,

                             uint16_t blockpos,

                             unsigned int maxchan,

                             int matrix_noise_shift,

                             int access_unit_size_pow2,

                             int32_t mask)

{

    unsigned int src_ch, i;

    int index2 = 2 * index + 1;

    for (i = 0; i < blockpos; i++) {

        int64_t accum = 0;

        for (src_ch = 0; src_ch <= maxchan; src_ch++)

            accum += (int64_t) samples[src_ch] * coeffs[src_ch];

        if (matrix_noise_shift) {

            index &= access_unit_size_pow2 - 1;

            accum += noise_buffer[index] * (1 << (matrix_noise_shift + 7));

            index += index2;

        }

        samples[dest_ch] = ((accum >> 14) & mask) + *bypassed_lsbs;

        bypassed_lsbs += MAX_CHANNELS;

        samples += MAX_CHANNELS;

    }

}

static int32_t (*mlp_select_pack_output(uint8_t *ch_assign,

                                        int8_t *output_shift,

                                        uint8_t max_matrix_channel,

                                        int is32))(int32_t, uint16_t, int32_t (*)[], void *, uint8_t*, int8_t *, uint8_t, int)

{

    return ff_mlp_pack_output;

}

int32_t ff_mlp_pack_output(int32_t lossless_check_data,

                           uint16_t blockpos,

                           int32_t (*sample_buffer)[MAX_CHANNELS],

                           void *data,

                           uint8_t *ch_assign,

                           int8_t *output_shift,

                           uint8_t max_matrix_channel,

                           int is32)

{

    unsigned int i, out_ch = 0;

    int32_t *data_32 = data;

    int16_t *data_16 = data;

    for (i = 0; i < blockpos; i++) {

        for (out_ch = 0; out_ch <= max_matrix_channel; out_ch++) {

            int mat_ch = ch_assign[out_ch];

            int32_t sample = sample_buffer[i][mat_ch] *

                          (1U << output_shift[mat_ch]);

            lossless_check_data ^= (sample & 0xffffff) << mat_ch;

            if (is32)

                *data_32++ = sample * 256U;

            else

                *data_16++ = sample >> 8;

        }

    }

    return lossless_check_data;

}

av_cold void ff_mlpdsp_init(MLPDSPContext *c)

{

    c->mlp_filter_channel = mlp_filter_channel;

    c->mlp_rematrix_channel = ff_mlp_rematrix_channel;

    c->mlp_select_pack_output = mlp_select_pack_output;

#if ARCH_ARM

    ff_mlpdsp_init_arm(c);

#elif ARCH_X86

    ff_mlpdsp_init_x86(c);

#endif

}