format_hifi3.h

Go to the documentation of this file.

/* SPDX-License-Identifier: BSD-3-Clause
 *
 * Copyright(c) 2024 Intel Corporation. All rights reserved.
 *
 * Author: Shriram Shastry <malladi.sastry@linux.intel.com>
 */
 
#ifndef __SOF_AUDIO_FORMAT_HIFI3_H__
#define __SOF_AUDIO_FORMAT_HIFI3_H__
 
#include <stdint.h>
#include <xtensa/config/defs.h>
#include <xtensa/tie/xt_hifi3.h>
 
/* Saturation inline functions */

static inline int32_t sat_int32(int64_t x)
{
    /* Shift left by 32 bits with saturation */
    ae_f64 shifted = AE_SLAI64S(x, 32);
 
    /* Shift right by 32 bits */
    return (int32_t)AE_MOVINT32_FROMINT64(AE_SRAI64(shifted, 32));
}

static inline ae_int32x2 vec_sat_int32x2(int64_t x, int64_t y)
{
    ae_f64 d0, d1;
 
    /* Convert 64-bit integers to ae_f64 format */
    d0 = (ae_f64)x;
    d1 = (ae_f64)y;
 
    /* Round and saturate both 64-bit values to 32-bit and pack them */
    return (ae_int32x2)AE_ROUND32X2F64SSYM(d0, d1);
}

static inline int32_t sat_int24(int32_t x)
{
    /* Move 32-bit value to ae_f32x2 type */
    ae_f32x2 d0 = AE_MOVDA32(x);
 
    /* Saturate to 24-bit */
    return (ae_int32)AE_SAT24S(d0);
}

static inline ae_int32x2 vec_sat_int24x2(int32_t x, int32_t y)
{
    /* Move two 32-bit values to ae_f32x2 type */
    ae_f32x2 d0 = AE_MOVDA32X2(x, y);
 
    /* Saturate both values to 24-bit and pack them */
    return (ae_int32x2)AE_SAT24S(d0);
}

static inline int16_t sat_int16(int32_t x)
{
    /* Move 32-bit value to ae_int32x2 type */
    ae_int32x2 d0 = AE_MOVDA32(x);
 
    /* Saturate both values to 16-bit and pack them */
    ae_int16x4 result = AE_SAT16X4(d0, d0);
 
    /* Use AE_MOVAD16_0 to extract the first 16-bit value */
    return AE_MOVAD16_0(result);
}

static inline ae_int16x4 vec_sat_int16x4(int32_t x, int32_t y, int32_t z, int32_t q)
{
    /* Move four 32-bit values to ae_int32x2 type */
    ae_int32x2 d0 = AE_MOVDA32X2(x, y);
    ae_int32x2 d1 = AE_MOVDA32X2(z, q);
 
    /* Saturate all values to 16-bit and pack them */
    return AE_SAT16X4(d0, d1);
}

static inline int8_t sat_int8(int32_t x)
{
    /* Shift left by 24 bits */
    ae_f32x2 a_i = AE_SLAI32S(x, 24);
 
    /* Shift right by 24 bits to sign-extend the 8-bit value */
    a_i = AE_SRAI32(a_i, 24);
 
    /* Extract the lower 8 bits as an int8_t */
    return (int8_t)AE_MOVAD32_L(a_i);
}

static inline ae_int32x2 vec_sat_int8x2(int32_t x, int32_t y)
{
    /* Move two 32-bit values to ae_f32x2 type */
    ae_f32x2 d0 = AE_MOVDA32X2(x, y);
 
    /* Shift left by 24 bits and then right by 24 bits to sign-extend and saturate */
    return AE_F32X2_SRAI(AE_F32X2_SLAIS(d0, 24), 24);
}
 
#endif /* __SOF_AUDIO_FORMAT_HIFI3_H__ */