362 lines · 10.6 KB · C
RawDownload
/* * 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;}

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t