uhd/host/lib/convert/sse2_sc16_to_fc64.cpp
Martin Braun 75ad0c5516 convert: Add chdr converters for sc16 -> fc32 and vice versa
These differ from the item32 converters in that they don't IQ swap, and
also don't have a BE/LE version.
2019-11-26 11:49:11 -08:00

150 lines
8.2 KiB
C++

//
// Copyright 2011-2012 Ettus Research LLC
// Copyright 2018 Ettus Research, a National Instruments Company
//
// SPDX-License-Identifier: GPL-3.0-or-later
//
#include "convert_common.hpp"
#include <uhd/utils/byteswap.hpp>
#include <emmintrin.h>
using namespace uhd::convert;
DECLARE_CONVERTER(sc16_item32_le, 1, fc64, 1, PRIORITY_SIMD)
{
const item32_t* input = reinterpret_cast<const item32_t*>(inputs[0]);
fc64_t* output = reinterpret_cast<fc64_t*>(outputs[0]);
const __m128d scalar = _mm_set1_pd(scale_factor / (1 << 16));
const __m128i zeroi = _mm_setzero_si128();
#define convert_item32_1_to_fc64_1_nswap_guts(_al_) \
for (; i + 3 < nsamps; i += 4) { \
/* load from input */ \
__m128i tmpi = _mm_loadu_si128(reinterpret_cast<const __m128i*>(input + i)); \
\
/* unpack + swap 16-bit pairs */ \
tmpi = _mm_shufflelo_epi16(tmpi, _MM_SHUFFLE(2, 3, 0, 1)); \
tmpi = _mm_shufflehi_epi16(tmpi, _MM_SHUFFLE(2, 3, 0, 1)); \
__m128i tmpilo = _mm_unpacklo_epi16(zeroi, tmpi); /* value in upper 16 bits */ \
__m128i tmpihi = _mm_unpackhi_epi16(zeroi, tmpi); \
\
/* convert and scale */ \
__m128d tmp0 = _mm_mul_pd(_mm_cvtepi32_pd(tmpilo), scalar); \
tmpilo = _mm_unpackhi_epi64(tmpilo, zeroi); \
__m128d tmp1 = _mm_mul_pd(_mm_cvtepi32_pd(tmpilo), scalar); \
__m128d tmp2 = _mm_mul_pd(_mm_cvtepi32_pd(tmpihi), scalar); \
tmpihi = _mm_unpackhi_epi64(tmpihi, zeroi); \
__m128d tmp3 = _mm_mul_pd(_mm_cvtepi32_pd(tmpihi), scalar); \
\
/* store to output */ \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 0), tmp0); \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 1), tmp1); \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 2), tmp2); \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 3), tmp3); \
}
size_t i = 0;
// dispatch according to alignment
if ((size_t(output) & 0xf) == 0) {
convert_item32_1_to_fc64_1_nswap_guts(_)
} else {
convert_item32_1_to_fc64_1_nswap_guts(u_)
}
// convert remainder
item32_sc16_to_xx<uhd::htowx>(input + i, output + i, nsamps - i, scale_factor);
}
DECLARE_CONVERTER(sc16_item32_be, 1, fc64, 1, PRIORITY_SIMD)
{
const item32_t* input = reinterpret_cast<const item32_t*>(inputs[0]);
fc64_t* output = reinterpret_cast<fc64_t*>(outputs[0]);
const __m128d scalar = _mm_set1_pd(scale_factor / (1 << 16));
const __m128i zeroi = _mm_setzero_si128();
#define convert_item32_1_to_fc64_1_bswap_guts(_al_) \
for (; i + 3 < nsamps; i += 4) { \
/* load from input */ \
__m128i tmpi = _mm_loadu_si128(reinterpret_cast<const __m128i*>(input + i)); \
\
/* byteswap + unpack -> byteswap 16 bit words */ \
tmpi = _mm_or_si128(_mm_srli_epi16(tmpi, 8), _mm_slli_epi16(tmpi, 8)); \
__m128i tmpilo = _mm_unpacklo_epi16(zeroi, tmpi); /* value in upper 16 bits */ \
__m128i tmpihi = _mm_unpackhi_epi16(zeroi, tmpi); \
\
/* convert and scale */ \
__m128d tmp0 = _mm_mul_pd(_mm_cvtepi32_pd(tmpilo), scalar); \
tmpilo = _mm_unpackhi_epi64(tmpilo, zeroi); \
__m128d tmp1 = _mm_mul_pd(_mm_cvtepi32_pd(tmpilo), scalar); \
__m128d tmp2 = _mm_mul_pd(_mm_cvtepi32_pd(tmpihi), scalar); \
tmpihi = _mm_unpackhi_epi64(tmpihi, zeroi); \
__m128d tmp3 = _mm_mul_pd(_mm_cvtepi32_pd(tmpihi), scalar); \
\
/* store to output */ \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 0), tmp0); \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 1), tmp1); \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 2), tmp2); \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 3), tmp3); \
}
size_t i = 0;
// dispatch according to alignment
if ((size_t(output) & 0xf) == 0) {
convert_item32_1_to_fc64_1_bswap_guts(_)
} else {
convert_item32_1_to_fc64_1_bswap_guts(u_)
}
// convert remainder
item32_sc16_to_xx<uhd::htonx>(input + i, output + i, nsamps - i, scale_factor);
}
DECLARE_CONVERTER(sc16_chdr, 1, fc64, 1, PRIORITY_SIMD)
{
const sc16_t* input = reinterpret_cast<const sc16_t*>(inputs[0]);
fc64_t* output = reinterpret_cast<fc64_t*>(outputs[0]);
const __m128d scalar = _mm_set1_pd(scale_factor / (1 << 16));
const __m128i zeroi = _mm_setzero_si128();
#define convert_chdr_1_to_fc64_1_guts(_al_) \
for (; i + 3 < nsamps; i += 4) { \
/* load from input */ \
__m128i tmpi = _mm_loadu_si128(reinterpret_cast<const __m128i*>(input + i)); \
\
/* unpack 16-bit pairs */ \
__m128i tmpilo = _mm_unpacklo_epi16(zeroi, tmpi); /* value in upper 16 bits */ \
__m128i tmpihi = _mm_unpackhi_epi16(zeroi, tmpi); \
\
/* convert and scale */ \
__m128d tmp0 = _mm_mul_pd(_mm_cvtepi32_pd(tmpilo), scalar); \
tmpilo = _mm_unpackhi_epi64(tmpilo, zeroi); \
__m128d tmp1 = _mm_mul_pd(_mm_cvtepi32_pd(tmpilo), scalar); \
__m128d tmp2 = _mm_mul_pd(_mm_cvtepi32_pd(tmpihi), scalar); \
tmpihi = _mm_unpackhi_epi64(tmpihi, zeroi); \
__m128d tmp3 = _mm_mul_pd(_mm_cvtepi32_pd(tmpihi), scalar); \
\
/* store to output */ \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 0), tmp0); \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 1), tmp1); \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 2), tmp2); \
_mm_store##_al_##pd(reinterpret_cast<double*>(output + i + 3), tmp3); \
}
size_t i = 0;
// dispatch according to alignment
if ((size_t(output) & 0xf) == 0) {
convert_chdr_1_to_fc64_1_guts(_)
} else {
convert_chdr_1_to_fc64_1_guts(u_)
}
// convert remainder
chdr_sc16_to_xx(input + i, output + i, nsamps - i, scale_factor);
}