Source code
Revision control
Copy as Markdown
Other Tools
/*
* Loongson MMI optimizations for libjpeg-turbo
*
* Copyright (C) 2014, 2018-2019, D. R. Commander. All Rights Reserved.
* Copyright (C) 2016-2018, Loongson Technology Corporation Limited, BeiJing.
* All Rights Reserved.
* Authors: LiuQingfa <liuqingfa-hf@loongson.cn>
*
* Based on the x86 SIMD extension for IJG JPEG library
* Copyright (C) 1999-2006, MIYASAKA Masaru.
*
* This software is provided 'as-is', without any express or implied
* warranty. In no event will the authors be held liable for any damages
* arising from the use of this software.
*
* Permission is granted to anyone to use this software for any purpose,
* including commercial applications, and to alter it and redistribute it
* freely, subject to the following restrictions:
*
* 1. The origin of this software must not be misrepresented; you must not
* claim that you wrote the original software. If you use this software
* in a product, an acknowledgment in the product documentation would be
* appreciated but is not required.
* 2. Altered source versions must be plainly marked as such, and must not be
* misrepresented as being the original software.
* 3. This notice may not be removed or altered from any source distribution.
*/
/* FAST INTEGER FORWARD DCT */
#include "jsimd_mmi.h"
#define CONST_BITS 8
#define F_0_382 ((short)98) /* FIX(0.382683433) */
#define F_0_541 ((short)139) /* FIX(0.541196100) */
#define F_0_707 ((short)181) /* FIX(0.707106781) */
#define F_1_306 ((short)334) /* FIX(1.306562965) */
#define PRE_MULTIPLY_SCALE_BITS 2
#define CONST_SHIFT (16 - PRE_MULTIPLY_SCALE_BITS - CONST_BITS)
enum const_index {
index_PW_F0707,
index_PW_F0382,
index_PW_F0541,
index_PW_F1306
};
static uint64_t const_value[] = {
_uint64_set1_pi16(F_0_707),
_uint64_set1_pi16(F_0_382),
_uint64_set1_pi16(F_0_541),
_uint64_set1_pi16(F_1_306)
};
#define PW_F0707 get_const_value(index_PW_F0707)
#define PW_F0382 get_const_value(index_PW_F0382)
#define PW_F0541 get_const_value(index_PW_F0541)
#define PW_F1306 get_const_value(index_PW_F1306)
#define DO_FDCT_MULTIPLY(out, in, multiplier) { \
__m64 mulhi, mullo, mul12, mul34; \
\
mullo = _mm_mullo_pi16(in, multiplier); \
mulhi = _mm_mulhi_pi16(in, multiplier); \
mul12 = _mm_unpacklo_pi16(mullo, mulhi); \
mul34 = _mm_unpackhi_pi16(mullo, mulhi); \
mul12 = _mm_srai_pi32(mul12, CONST_BITS); \
mul34 = _mm_srai_pi32(mul34, CONST_BITS); \
out = _mm_packs_pi32(mul12, mul34); \
}
#define DO_FDCT_COMMON() { \
\
/* Even part */ \
\
tmp10 = _mm_add_pi16(tmp0, tmp3); \
tmp13 = _mm_sub_pi16(tmp0, tmp3); \
tmp11 = _mm_add_pi16(tmp1, tmp2); \
tmp12 = _mm_sub_pi16(tmp1, tmp2); \
\
out0 = _mm_add_pi16(tmp10, tmp11); \
out4 = _mm_sub_pi16(tmp10, tmp11); \
\
z1 = _mm_add_pi16(tmp12, tmp13); \
DO_FDCT_MULTIPLY(z1, z1, PW_F0707) \
\
out2 = _mm_add_pi16(tmp13, z1); \
out6 = _mm_sub_pi16(tmp13, z1); \
\
/* Odd part */ \
\
tmp10 = _mm_add_pi16(tmp4, tmp5); \
tmp11 = _mm_add_pi16(tmp5, tmp6); \
tmp12 = _mm_add_pi16(tmp6, tmp7); \
\
z5 = _mm_sub_pi16(tmp10, tmp12); \
DO_FDCT_MULTIPLY(z5, z5, PW_F0382) \
\
DO_FDCT_MULTIPLY(z2, tmp10, PW_F0541) \
z2 = _mm_add_pi16(z2, z5); \
\
DO_FDCT_MULTIPLY(z4, tmp12, PW_F1306) \
z4 = _mm_add_pi16(z4, z5); \
\
DO_FDCT_MULTIPLY(z3, tmp11, PW_F0707) \
\
z11 = _mm_add_pi16(tmp7, z3); \
z13 = _mm_sub_pi16(tmp7, z3); \
\
out5 = _mm_add_pi16(z13, z2); \
out3 = _mm_sub_pi16(z13, z2); \
out1 = _mm_add_pi16(z11, z4); \
out7 = _mm_sub_pi16(z11, z4); \
}
#define DO_FDCT_PASS1() { \
__m64 row0l, row0h, row1l, row1h, row2l, row2h, row3l, row3h; \
__m64 row01a, row01b, row01c, row01d, row23a, row23b, row23c, row23d; \
__m64 col0, col1, col2, col3, col4, col5, col6, col7; \
\
row0l = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 0]); /* (00 01 02 03) */ \
row0h = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 0 + 4]); /* (04 05 06 07) */ \
row1l = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 1]); /* (10 11 12 13) */ \
row1h = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 1 + 4]); /* (14 15 16 17) */ \
row2l = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 2]); /* (20 21 22 23) */ \
row2h = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 2 + 4]); /* (24 25 26 27) */ \
row3l = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 3]); /* (30 31 32 33) */ \
row3h = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 3 + 4]); /* (34 35 36 37) */ \
\
/* Transpose coefficients */ \
\
row23a = _mm_unpacklo_pi16(row2l, row3l); /* row23a=(20 30 21 31) */ \
row23b = _mm_unpackhi_pi16(row2l, row3l); /* row23b=(22 32 23 33) */ \
row23c = _mm_unpacklo_pi16(row2h, row3h); /* row23c=(24 34 25 35) */ \
row23d = _mm_unpackhi_pi16(row2h, row3h); /* row23d=(26 36 27 37) */ \
\
row01a = _mm_unpacklo_pi16(row0l, row1l); /* row01a=(00 10 01 11) */ \
row01b = _mm_unpackhi_pi16(row0l, row1l); /* row01b=(02 12 03 13) */ \
row01c = _mm_unpacklo_pi16(row0h, row1h); /* row01c=(04 14 05 15) */ \
row01d = _mm_unpackhi_pi16(row0h, row1h); /* row01d=(06 16 07 17) */ \
\
col0 = _mm_unpacklo_pi32(row01a, row23a); /* col0=(00 10 20 30) */ \
col1 = _mm_unpackhi_pi32(row01a, row23a); /* col1=(01 11 21 31) */ \
col6 = _mm_unpacklo_pi32(row01d, row23d); /* col6=(06 16 26 36) */ \
col7 = _mm_unpackhi_pi32(row01d, row23d); /* col7=(07 17 27 37) */ \
\
tmp6 = _mm_sub_pi16(col1, col6); /* tmp6=col1-col6 */ \
tmp7 = _mm_sub_pi16(col0, col7); /* tmp7=col0-col7 */ \
tmp1 = _mm_add_pi16(col1, col6); /* tmp1=col1+col6 */ \
tmp0 = _mm_add_pi16(col0, col7); /* tmp0=col0+col7 */ \
\
col2 = _mm_unpacklo_pi32(row01b, row23b); /* col2=(02 12 22 32) */ \
col3 = _mm_unpackhi_pi32(row01b, row23b); /* col3=(03 13 23 33) */ \
col4 = _mm_unpacklo_pi32(row01c, row23c); /* col4=(04 14 24 34) */ \
col5 = _mm_unpackhi_pi32(row01c, row23c); /* col5=(05 15 25 35) */ \
\
tmp3 = _mm_add_pi16(col3, col4); /* tmp3=col3+col4 */ \
tmp2 = _mm_add_pi16(col2, col5); /* tmp2=col2+col5 */ \
tmp4 = _mm_sub_pi16(col3, col4); /* tmp4=col3-col4 */ \
tmp5 = _mm_sub_pi16(col2, col5); /* tmp5=col2-col5 */ \
\
DO_FDCT_COMMON() \
\
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 0], out0); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 0 + 4], out4); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 1], out1); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 1 + 4], out5); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 2], out2); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 2 + 4], out6); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 3], out3); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 3 + 4], out7); \
}
#define DO_FDCT_PASS2() { \
__m64 col0l, col0h, col1l, col1h, col2l, col2h, col3l, col3h; \
__m64 col01a, col01b, col01c, col01d, col23a, col23b, col23c, col23d; \
__m64 row0, row1, row2, row3, row4, row5, row6, row7; \
\
col0l = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 0]); /* (00 10 20 30) */ \
col1l = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 1]); /* (01 11 21 31) */ \
col2l = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 2]); /* (02 12 22 32) */ \
col3l = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 3]); /* (03 13 23 33) */ \
col0h = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 4]); /* (40 50 60 70) */ \
col1h = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 5]); /* (41 51 61 71) */ \
col2h = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 6]); /* (42 52 62 72) */ \
col3h = _mm_load_si64((__m64 *)&dataptr[DCTSIZE * 7]); /* (43 53 63 73) */ \
\
/* Transpose coefficients */ \
\
col23a = _mm_unpacklo_pi16(col2l, col3l); /* col23a=(02 03 12 13) */ \
col23b = _mm_unpackhi_pi16(col2l, col3l); /* col23b=(22 23 32 33) */ \
col23c = _mm_unpacklo_pi16(col2h, col3h); /* col23c=(42 43 52 53) */ \
col23d = _mm_unpackhi_pi16(col2h, col3h); /* col23d=(62 63 72 73) */ \
\
col01a = _mm_unpacklo_pi16(col0l, col1l); /* col01a=(00 01 10 11) */ \
col01b = _mm_unpackhi_pi16(col0l, col1l); /* col01b=(20 21 30 31) */ \
col01c = _mm_unpacklo_pi16(col0h, col1h); /* col01c=(40 41 50 51) */ \
col01d = _mm_unpackhi_pi16(col0h, col1h); /* col01d=(60 61 70 71) */ \
\
row0 = _mm_unpacklo_pi32(col01a, col23a); /* row0=(00 01 02 03) */ \
row1 = _mm_unpackhi_pi32(col01a, col23a); /* row1=(10 11 12 13) */ \
row6 = _mm_unpacklo_pi32(col01d, col23d); /* row6=(60 61 62 63) */ \
row7 = _mm_unpackhi_pi32(col01d, col23d); /* row7=(70 71 72 73) */ \
\
tmp6 = _mm_sub_pi16(row1, row6); /* tmp6=row1-row6 */ \
tmp7 = _mm_sub_pi16(row0, row7); /* tmp7=row0-row7 */ \
tmp1 = _mm_add_pi16(row1, row6); /* tmp1=row1+row6 */ \
tmp0 = _mm_add_pi16(row0, row7); /* tmp0=row0+row7 */ \
\
row2 = _mm_unpacklo_pi32(col01b, col23b); /* row2=(20 21 22 23) */ \
row3 = _mm_unpackhi_pi32(col01b, col23b); /* row3=(30 31 32 33) */ \
row4 = _mm_unpacklo_pi32(col01c, col23c); /* row4=(40 41 42 43) */ \
row5 = _mm_unpackhi_pi32(col01c, col23c); /* row5=(50 51 52 53) */ \
\
tmp3 = _mm_add_pi16(row3, row4); /* tmp3=row3+row4 */ \
tmp2 = _mm_add_pi16(row2, row5); /* tmp2=row2+row5 */ \
tmp4 = _mm_sub_pi16(row3, row4); /* tmp4=row3-row4 */ \
tmp5 = _mm_sub_pi16(row2, row5); /* tmp5=row2-row5 */ \
\
DO_FDCT_COMMON() \
\
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 0], out0); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 1], out1); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 2], out2); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 3], out3); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 4], out4); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 5], out5); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 6], out6); \
_mm_store_si64((__m64 *)&dataptr[DCTSIZE * 7], out7); \
}
void jsimd_fdct_ifast_mmi(DCTELEM *data)
{
__m64 tmp0, tmp1, tmp2, tmp3, tmp4, tmp5, tmp6, tmp7;
__m64 out0, out1, out2, out3, out4, out5, out6, out7;
__m64 tmp10, tmp11, tmp12, tmp13, z1, z2, z3, z4, z5, z11, z13;
DCTELEM *dataptr = data;
/* Pass 1: process rows. */
DO_FDCT_PASS1()
dataptr += DCTSIZE * 4;
DO_FDCT_PASS1()
/* Pass 2: process columns. */
dataptr = data;
DO_FDCT_PASS2()
dataptr += 4;
DO_FDCT_PASS2()
}