127 строки
4.0 KiB
C
127 строки
4.0 KiB
C
/*
|
|
* Copyright (c) 2015 The WebM project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "vp10/common/vp10_txfm.h"
|
|
#include "vp10/common/vp10_fwd_txfm1d.h"
|
|
|
|
typedef void (*TxfmFunc)(const int32_t *input, int32_t *output,
|
|
const int8_t *cos_bit, const int8_t *stage_range);
|
|
|
|
static inline TxfmFunc fwd_txfm_type_to_func(TXFM_TYPE txfm_type) {
|
|
switch (txfm_type) {
|
|
case TXFM_TYPE_DCT4:
|
|
return vp10_fdct4_new;
|
|
break;
|
|
case TXFM_TYPE_DCT8:
|
|
return vp10_fdct8_new;
|
|
break;
|
|
case TXFM_TYPE_DCT16:
|
|
return vp10_fdct16_new;
|
|
break;
|
|
case TXFM_TYPE_DCT32:
|
|
return vp10_fdct32_new;
|
|
break;
|
|
case TXFM_TYPE_DCT64:
|
|
return vp10_fdct64_new;
|
|
break;
|
|
case TXFM_TYPE_ADST4:
|
|
return vp10_fadst4_new;
|
|
break;
|
|
case TXFM_TYPE_ADST8:
|
|
return vp10_fadst8_new;
|
|
break;
|
|
case TXFM_TYPE_ADST16:
|
|
return vp10_fadst16_new;
|
|
break;
|
|
case TXFM_TYPE_ADST32:
|
|
return vp10_fadst32_new;
|
|
break;
|
|
default:
|
|
assert(0);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
static inline void fwd_txfm2d_c(const int16_t *input, int32_t *output,
|
|
const int stride, const TXFM_2D_CFG *cfg,
|
|
int32_t *buf) {
|
|
int i, j;
|
|
const int txfm_size = cfg->txfm_size;
|
|
const int8_t *shift = cfg->shift;
|
|
const int8_t *stage_range_col = cfg->stage_range_col;
|
|
const int8_t *stage_range_row = cfg->stage_range_row;
|
|
const int8_t *cos_bit_col = cfg->cos_bit_col;
|
|
const int8_t *cos_bit_row = cfg->cos_bit_row;
|
|
const TxfmFunc txfm_func_col = fwd_txfm_type_to_func(cfg->txfm_type_col);
|
|
const TxfmFunc txfm_func_row = fwd_txfm_type_to_func(cfg->txfm_type_row);
|
|
|
|
// use output buffer as temp buffer
|
|
int32_t* temp_in = output;
|
|
int32_t* temp_out = output + txfm_size;
|
|
|
|
// Columns
|
|
for (i = 0; i < txfm_size; ++i) {
|
|
for (j = 0; j < txfm_size; ++j)
|
|
temp_in[j] = input[j * stride + i];
|
|
round_shift_array(temp_in, txfm_size, -shift[0]);
|
|
txfm_func_col(temp_in, temp_out, cos_bit_col, stage_range_col);
|
|
round_shift_array(temp_out, txfm_size, -shift[1]);
|
|
for (j = 0; j < txfm_size; ++j)
|
|
buf[j * txfm_size + i] = temp_out[j];
|
|
}
|
|
|
|
// Rows
|
|
for (i = 0; i < txfm_size; ++i) {
|
|
txfm_func_row(buf + i * txfm_size, output + i * txfm_size, cos_bit_row,
|
|
stage_range_row);
|
|
round_shift_array(output + i * txfm_size, txfm_size, -shift[2]);
|
|
}
|
|
}
|
|
|
|
void vp10_fwd_txfm2d_4x4_c(const int16_t *input, int32_t *output,
|
|
const int stride, const TXFM_2D_CFG *cfg,
|
|
const int bd) {
|
|
int32_t txfm_buf[4 * 4];
|
|
(void)bd;
|
|
fwd_txfm2d_c(input, output, stride, cfg, txfm_buf);
|
|
}
|
|
|
|
void vp10_fwd_txfm2d_8x8_c(const int16_t *input, int32_t *output,
|
|
const int stride, const TXFM_2D_CFG *cfg,
|
|
const int bd) {
|
|
int32_t txfm_buf[8 * 8];
|
|
(void)bd;
|
|
fwd_txfm2d_c(input, output, stride, cfg, txfm_buf);
|
|
}
|
|
|
|
void vp10_fwd_txfm2d_16x16_c(const int16_t *input, int32_t *output,
|
|
const int stride, const TXFM_2D_CFG *cfg,
|
|
const int bd) {
|
|
int32_t txfm_buf[16 * 16];
|
|
(void)bd;
|
|
fwd_txfm2d_c(input, output, stride, cfg, txfm_buf);
|
|
}
|
|
|
|
void vp10_fwd_txfm2d_32x32_c(const int16_t *input, int32_t *output,
|
|
const int stride, const TXFM_2D_CFG *cfg,
|
|
const int bd) {
|
|
int32_t txfm_buf[32 * 32];
|
|
(void)bd;
|
|
fwd_txfm2d_c(input, output, stride, cfg, txfm_buf);
|
|
}
|
|
|
|
void vp10_fwd_txfm2d_64x64_c(const int16_t *input, int32_t *output,
|
|
const int stride, const TXFM_2D_CFG *cfg,
|
|
const int bd) {
|
|
int32_t txfm_buf[64 * 64];
|
|
(void)bd;
|
|
fwd_txfm2d_c(input, output, stride, cfg, txfm_buf);
|
|
}
|