diff options
Diffstat (limited to 'vp9/common/vp9_invtrans.c')
-rw-r--r-- | vp9/common/vp9_invtrans.c | 120 |
1 files changed, 120 insertions, 0 deletions
diff --git a/vp9/common/vp9_invtrans.c b/vp9/common/vp9_invtrans.c new file mode 100644 index 000000000..76eb10307 --- /dev/null +++ b/vp9/common/vp9_invtrans.c @@ -0,0 +1,120 @@ +/* + * Copyright (c) 2010 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 "vp9_invtrans.h" +#include "./vp9_rtcd.h" + +static void recon_dcblock(MACROBLOCKD *xd) { + BLOCKD *b = &xd->block[24]; + int i; + + for (i = 0; i < 16; i++) { + xd->block[i].dqcoeff[0] = b->diff[i]; + } +} + +static void recon_dcblock_8x8(MACROBLOCKD *xd) { + BLOCKD *b = &xd->block[24]; // for coeff 0, 2, 8, 10 + + xd->block[0].dqcoeff[0] = b->diff[0]; + xd->block[4].dqcoeff[0] = b->diff[1]; + xd->block[8].dqcoeff[0] = b->diff[4]; + xd->block[12].dqcoeff[0] = b->diff[8]; +} + +void vp9_inverse_transform_b_4x4(MACROBLOCKD *xd, int block, int pitch) { + BLOCKD *b = &xd->block[block]; + if (b->eob <= 1) + xd->inv_xform4x4_1_x8(b->dqcoeff, b->diff, pitch); + else + xd->inv_xform4x4_x8(b->dqcoeff, b->diff, pitch); +} + +void vp9_inverse_transform_mby_4x4(MACROBLOCKD *xd) { + int i; + BLOCKD *blockd = xd->block; + + if (xd->mode_info_context->mbmi.mode != SPLITMV) { + /* do 2nd order transform on the dc block */ + vp9_short_inv_walsh4x4(blockd[24].dqcoeff, blockd[24].diff); + recon_dcblock(xd); + } + + for (i = 0; i < 16; i++) { + vp9_inverse_transform_b_4x4(xd, i, 32); + } +} + +void vp9_inverse_transform_mbuv_4x4(MACROBLOCKD *xd) { + int i; + for (i = 16; i < 24; i++) { + vp9_inverse_transform_b_4x4(xd, i, 16); + } +} + +void vp9_inverse_transform_mb_4x4(MACROBLOCKD *xd) { + vp9_inverse_transform_mby_4x4(xd); + vp9_inverse_transform_mbuv_4x4(xd); +} + +void vp9_inverse_transform_b_8x8(short *input_dqcoeff, short *output_coeff, + int pitch) { + vp9_short_idct8x8(input_dqcoeff, output_coeff, pitch); +} + +void vp9_inverse_transform_mby_8x8(MACROBLOCKD *xd) { + int i; + BLOCKD *blockd = xd->block; + + if (xd->mode_info_context->mbmi.mode != SPLITMV) { + // do 2nd order transform on the dc block + vp9_short_ihaar2x2(blockd[24].dqcoeff, blockd[24].diff, 8); + recon_dcblock_8x8(xd); // need to change for 8x8 + } + + for (i = 0; i < 9; i += 8) { + vp9_inverse_transform_b_8x8(&blockd[i].dqcoeff[0], + &blockd[i].diff[0], 32); + } + for (i = 2; i < 11; i += 8) { + vp9_inverse_transform_b_8x8(&blockd[i + 2].dqcoeff[0], + &blockd[i].diff[0], 32); + } +} + +void vp9_inverse_transform_mbuv_8x8(MACROBLOCKD *xd) { + int i; + BLOCKD *blockd = xd->block; + + for (i = 16; i < 24; i += 4) { + vp9_inverse_transform_b_8x8(&blockd[i].dqcoeff[0], + &blockd[i].diff[0], 16); + } +} + +void vp9_inverse_transform_mb_8x8(MACROBLOCKD *xd) { + vp9_inverse_transform_mby_8x8(xd); + vp9_inverse_transform_mbuv_8x8(xd); +} + +void vp9_inverse_transform_b_16x16(short *input_dqcoeff, + short *output_coeff, int pitch) { + vp9_short_idct16x16(input_dqcoeff, output_coeff, pitch); +} + +void vp9_inverse_transform_mby_16x16(MACROBLOCKD *xd) { + vp9_inverse_transform_b_16x16(&xd->block[0].dqcoeff[0], + &xd->block[0].diff[0], 32); +} + +void vp9_inverse_transform_mb_16x16(MACROBLOCKD *xd) { + vp9_inverse_transform_mby_16x16(xd); + vp9_inverse_transform_mbuv_8x8(xd); +} |