/*

 * Copyright (C) 2016 foo86

 *

 * 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 <stdlib.h>

#include "dcadct.h"

#include "dcamath.h"

static void sum_a(const int *input, int *output, int len)

{

    int i;

    for (i = 0; i < len; i++)

        output[i] = input[2 * i] + input[2 * i + 1];

}

static void sum_b(const int *input, int *output, int len)

{

    int i;

    output[0] = input[0];

    for (i = 1; i < len; i++)

        output[i] = input[2 * i] + input[2 * i - 1];

}

static void sum_c(const int *input, int *output, int len)

{

    int i;

    for (i = 0; i < len; i++)

        output[i] = input[2 * i];

}

static void sum_d(const int *input, int *output, int len)

{

    int i;

    output[0] = input[1];

    for (i = 1; i < len; i++)

        output[i] = input[2 * i - 1] + input[2 * i + 1];

}

static void dct_a(const int *input, int *output)

{

    static const int cos_mod[8][8] = {

         { 8348215,  8027397,  7398092,  6484482,  5321677,  3954362,  2435084,   822227 },

         { 8027397,  5321677,   822227, -3954362, -7398092, -8348215, -6484482, -2435084 },

         { 7398092,   822227, -6484482, -8027397, -2435084,  5321677,  8348215,  3954362 },

         { 6484482, -3954362, -8027397,   822227,  8348215,  2435084, -7398092, -5321677 },

         { 5321677, -7398092, -2435084,  8348215,  -822227, -8027397,  3954362,  6484482 },

         { 3954362, -8348215,  5321677,  2435084, -8027397,  6484482,   822227, -7398092 },

         { 2435084, -6484482,  8348215, -7398092,  3954362,   822227, -5321677,  8027397 },

         {  822227, -2435084,  3954362, -5321677,  6484482, -7398092,  8027397, -8348215 }

    };

    int i, j;

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

        int64_t res = 0;

        for (j = 0; j < 8; j++)

            res += (int64_t)cos_mod[i][j] * input[j];

        output[i] = norm23(res);

    }

}

static void dct_b(const int *input, int *output)

{

    static const int cos_mod[8][7] = {

        {  8227423,  7750063,  6974873,  5931642,  4660461,  3210181,  1636536 },

        {  6974873,  3210181, -1636536, -5931642, -8227423, -7750063, -4660461 },

        {  4660461, -3210181, -8227423, -5931642,  1636536,  7750063,  6974873 },

        {  1636536, -7750063, -4660461,  5931642,  6974873, -3210181, -8227423 },

        { -1636536, -7750063,  4660461,  5931642, -6974873, -3210181,  8227423 },

        { -4660461, -3210181,  8227423, -5931642, -1636536,  7750063, -6974873 },

        { -6974873,  3210181,  1636536, -5931642,  8227423, -7750063,  4660461 },

        { -8227423,  7750063, -6974873,  5931642, -4660461,  3210181, -1636536 }

    };

    int i, j;

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

        int64_t res = input[0] * (INT64_C(1) << 23);

        for (j = 0; j < 7; j++)

            res += (int64_t)cos_mod[i][j] * input[1 + j];

        output[i] = norm23(res);

    }

}

static void mod_a(const int *input, int *output)

{

    static const int cos_mod[16] = {

          4199362,   4240198,   4323885,   4454708,

          4639772,   4890013,   5221943,   5660703,

         -6245623,  -7040975,  -8158494,  -9809974,

        -12450076, -17261920, -28585092, -85479984

    };

    int i, k;

    for (i = 0; i < 8; i++)

        output[i] = mul23(cos_mod[i], input[i] + input[8 + i]);

    for (i = 8, k = 7; i < 16; i++, k--)

        output[i] = mul23(cos_mod[i], input[k] - input[8 + k]);

}

static void mod_b(int *input, int *output)

{

    static const int cos_mod[8] = {

        4214598,  4383036,  4755871,  5425934,

        6611520,  8897610, 14448934, 42791536

    };

    int i, k;

    for (i = 0; i < 8; i++)

        input[8 + i] = mul23(cos_mod[i], input[8 + i]);

    for (i = 0; i < 8; i++)

        output[i] = input[i] + input[8 + i];

    for (i = 8, k = 7; i < 16; i++, k--)

        output[i] = input[k] - input[8 + k];

}

static void mod_c(const int *input, int *output)

{

    static const int cos_mod[32] = {

         1048892,  1051425,   1056522,   1064244,

         1074689,  1087987,   1104313,   1123884,

         1146975,  1173922,   1205139,   1241133,

         1282529,  1330095,   1384791,   1447815,

        -1520688, -1605358,  -1704360,  -1821051,

        -1959964, -2127368,  -2332183,  -2587535,

        -2913561, -3342802,  -3931480,  -4785806,

        -6133390, -8566050, -14253820, -42727120

    };

    int i, k;

    for (i = 0; i < 16; i++)

        output[i] = mul23(cos_mod[i], input[i] + input[16 + i]);

    for (i = 16, k = 15; i < 32; i++, k--)

        output[i] = mul23(cos_mod[i], input[k] - input[16 + k]);

}

static void clp_v(int *input, int len)

{

    int i;

    for (i = 0; i < len; i++)

        input[i] = clip23(input[i]);

}

static void imdct_half_32(int32_t *output, const int32_t *input)

{

    int buf_a[32], buf_b[32];

    int i, k, mag, shift, round;

    mag = 0;

    for (i = 0; i < 32; i++)

        mag += abs(input[i]);

    shift = mag > 0x400000 ? 2 : 0;

    round = shift > 0 ? 1 << (shift - 1) : 0;

    for (i = 0; i < 32; i++)

        buf_a[i] = (input[i] + round) >> shift;

    sum_a(buf_a, buf_b +  0, 16);

    sum_b(buf_a, buf_b + 16, 16);

    clp_v(buf_b, 32);

    sum_a(buf_b +  0, buf_a +  0, 8);

    sum_b(buf_b +  0, buf_a +  8, 8);

    sum_c(buf_b + 16, buf_a + 16, 8);

    sum_d(buf_b + 16, buf_a + 24, 8);

    clp_v(buf_a, 32);

    dct_a(buf_a +  0, buf_b +  0);

    dct_b(buf_a +  8, buf_b +  8);

    dct_b(buf_a + 16, buf_b + 16);

    dct_b(buf_a + 24, buf_b + 24);

    clp_v(buf_b, 32);

    mod_a(buf_b +  0, buf_a +  0);

    mod_b(buf_b + 16, buf_a + 16);

    clp_v(buf_a, 32);

    mod_c(buf_a, buf_b);

    for (i = 0; i < 32; i++)

        buf_b[i] = clip23(buf_b[i] * (1 << shift));

    for (i = 0, k = 31; i < 16; i++, k--) {

        output[     i] = clip23(buf_b[i] - buf_b[k]);

        output[16 + i] = clip23(buf_b[i] + buf_b[k]);

    }

}

static void mod64_a(const int *input, int *output)

{

    static const int cos_mod[32] = {

          4195568,   4205700,   4226086,    4256977,

          4298755,   4351949,   4417251,    4495537,

          4587901,   4695690,   4820557,    4964534,

          5130115,   5320382,   5539164,    5791261,

         -6082752,  -6421430,  -6817439,   -7284203,

         -7839855,  -8509474,  -9328732,  -10350140,

        -11654242, -13371208, -15725922,  -19143224,

        -24533560, -34264200, -57015280, -170908480

    };

    int i, k;

    for (i = 0; i < 16; i++)

        output[i] = mul23(cos_mod[i], input[i] + input[16 + i]);

    for (i = 16, k = 15; i < 32; i++, k--)

        output[i] = mul23(cos_mod[i], input[k] - input[16 + k]);

}

static void mod64_b(int *input, int *output)

{

    static const int cos_mod[16] = {

         4199362,  4240198,  4323885,  4454708,

         4639772,  4890013,  5221943,  5660703,

         6245623,  7040975,  8158494,  9809974,

        12450076, 17261920, 28585092, 85479984

    };

    int i, k;

    for (i = 0; i < 16; i++)

        input[16 + i] = mul23(cos_mod[i], input[16 + i]);

    for (i = 0; i < 16; i++)

        output[i] = input[i] + input[16 + i];

    for (i = 16, k = 15; i < 32; i++, k--)

        output[i] = input[k] - input[16 + k];

}

static void mod64_c(const int *input, int *output)

{

    static const int cos_mod[64] = {

          741511,    741958,    742853,    744199,

          746001,    748262,    750992,    754197,

          757888,    762077,    766777,    772003,

          777772,    784105,    791021,    798546,

          806707,    815532,    825054,    835311,

          846342,    858193,    870912,    884554,

          899181,    914860,    931667,    949686,

          969011,    989747,   1012012,   1035941,

        -1061684,  -1089412,  -1119320,  -1151629,

        -1186595,  -1224511,  -1265719,  -1310613,

        -1359657,  -1413400,  -1472490,  -1537703,

        -1609974,  -1690442,  -1780506,  -1881904,

        -1996824,  -2128058,  -2279225,  -2455101,

        -2662128,  -2909200,  -3208956,  -3579983,

        -4050785,  -4667404,  -5509372,  -6726913,

        -8641940, -12091426, -20144284, -60420720

    };

    int i, k;

    for (i = 0; i < 32; i++)

        output[i] = mul23(cos_mod[i], input[i] + input[32 + i]);

    for (i = 32, k = 31; i < 64; i++, k--)

        output[i] = mul23(cos_mod[i], input[k] - input[32 + k]);

}

static void imdct_half_64(int32_t *output, const int32_t *input)

{

    int buf_a[64], buf_b[64];

    int i, k, mag, shift, round;

    mag = 0;

    for (i = 0; i < 64; i++)

        mag += abs(input[i]);

    shift = mag > 0x400000 ? 2 : 0;

    round = shift > 0 ? 1 << (shift - 1) : 0;

    for (i = 0; i < 64; i++)

        buf_a[i] = (input[i] + round) >> shift;

    sum_a(buf_a, buf_b +  0, 32);

    sum_b(buf_a, buf_b + 32, 32);

    clp_v(buf_b, 64);

    sum_a(buf_b +  0, buf_a +  0, 16);

    sum_b(buf_b +  0, buf_a + 16, 16);

    sum_c(buf_b + 32, buf_a + 32, 16);

    sum_d(buf_b + 32, buf_a + 48, 16);

    clp_v(buf_a, 64);

    sum_a(buf_a +  0, buf_b +  0, 8);

    sum_b(buf_a +  0, buf_b +  8, 8);

    sum_c(buf_a + 16, buf_b + 16, 8);

    sum_d(buf_a + 16, buf_b + 24, 8);

    sum_c(buf_a + 32, buf_b + 32, 8);

    sum_d(buf_a + 32, buf_b + 40, 8);

    sum_c(buf_a + 48, buf_b + 48, 8);

    sum_d(buf_a + 48, buf_b + 56, 8);

    clp_v(buf_b, 64);

    dct_a(buf_b +  0, buf_a +  0);

    dct_b(buf_b +  8, buf_a +  8);

    dct_b(buf_b + 16, buf_a + 16);

    dct_b(buf_b + 24, buf_a + 24);

    dct_b(buf_b + 32, buf_a + 32);

    dct_b(buf_b + 40, buf_a + 40);

    dct_b(buf_b + 48, buf_a + 48);

    dct_b(buf_b + 56, buf_a + 56);

    clp_v(buf_a, 64);

    mod_a(buf_a +  0, buf_b +  0);

    mod_b(buf_a + 16, buf_b + 16);

    mod_b(buf_a + 32, buf_b + 32);

    mod_b(buf_a + 48, buf_b + 48);

    clp_v(buf_b, 64);

    mod64_a(buf_b +  0, buf_a +  0);

    mod64_b(buf_b + 32, buf_a + 32);

    clp_v(buf_a, 64);

    mod64_c(buf_a, buf_b);

    for (i = 0; i < 64; i++)

        buf_b[i] = clip23(buf_b[i] * (1 << shift));

    for (i = 0, k = 63; i < 32; i++, k--) {

        output[     i] = clip23(buf_b[i] - buf_b[k]);

        output[32 + i] = clip23(buf_b[i] + buf_b[k]);

    }

}

av_cold void ff_dcadct_init(DCADCTContext *c)

{

    c->imdct_half[0] = imdct_half_32;

    c->imdct_half[1] = imdct_half_64;

}