[feat][nmsis] add nmsis component and nn,dsp demo

This commit is contained in:
jzlv 2021-09-26 13:38:51 +08:00
parent b2aada479b
commit 5d1126d0f0
989 changed files with 286224 additions and 0 deletions

View file

@ -0,0 +1,3 @@
#include "mat_helper.c"
#include "ref_helper.c"

View file

@ -0,0 +1,200 @@
#include "ref.h"
float riscv_snr_f32(float *pRef, float *pTest, uint32_t buffSize)
{
float EnergySignal = 0.0, EnergyError = 0.0;
uint32_t i;
float SNR;
int temp;
int *test;
for (i = 0; i < buffSize; i++) {
/* Checking for a NAN value in pRef array */
test = (int *)(&pRef[i]);
temp = *test;
if (temp == 0x7FC00000) {
return (0);
}
/* Checking for a NAN value in pTest array */
test = (int *)(&pTest[i]);
temp = *test;
if (temp == 0x7FC00000) {
return (0);
}
EnergySignal += pRef[i] * pRef[i];
EnergyError += (pRef[i] - pTest[i]) * (pRef[i] - pTest[i]);
}
/* Checking for a NAN value in EnergyError */
test = (int *)(&EnergyError);
temp = *test;
if (temp == 0x7FC00000) {
return (0);
}
SNR = 10 * log10(EnergySignal / EnergyError);
return (SNR);
}
float32_t ref_detrm(float32_t *pSrc, float32_t *temp, uint32_t size)
{
float32_t s = 1, det = 0;
int i, j, m, n, c;
if (size == 1) {
return (pSrc[0]);
} else {
det = 0;
for (c = 0; c < size; c++) {
m = 0;
n = 0;
for (i = 0; i < size; i++) {
for (j = 0; j < size; j++) {
temp[i * size + j] = 0;
if (i != 0 && j != c) {
temp[m * (size - 1) + n] = pSrc[i * size + j];
if (n < (size - 2)) {
n++;
} else {
n = 0;
m++;
}
}
}
}
det +=
s * (pSrc[c] * ref_detrm(temp, temp + size * size, size - 1));
s = -s;
}
}
return (det);
}
void ref_cofact(float32_t *pSrc, float32_t *pDst, float32_t *temp,
uint32_t size)
{
int p, q, m, n, i, j;
if (size == 1) {
pDst[0] = 1;
return;
}
for (q = 0; q < size; q++) {
for (p = 0; p < size; p++) {
m = 0;
n = 0;
for (i = 0; i < size; i++) {
for (j = 0; j < size; j++) {
temp[i * size + j] = 0;
if (i != q && j != p) {
temp[m * (size - 1) + n] = pSrc[i * size + j];
if (n < (size - 2)) {
n++;
} else {
n = 0;
m++;
}
}
}
}
pDst[q * size + p] =
ref_pow(-1, q + p) *
ref_detrm(temp, temp + (size - 1) * (size - 1), size - 1);
}
}
}
float64_t ref_detrm64(float64_t *pSrc, float64_t *temp, uint32_t size)
{
float64_t s = 1, det = 0;
int i, j, m, n, c;
if (size == 1) {
return (pSrc[0]);
} else {
det = 0;
for (c = 0; c < size; c++) {
m = 0;
n = 0;
for (i = 0; i < size; i++) {
for (j = 0; j < size; j++) {
temp[i * size + j] = 0;
if (i != 0 && j != c) {
temp[m * (size - 1) + n] = pSrc[i * size + j];
if (n < (size - 2)) {
n++;
} else {
n = 0;
m++;
}
}
}
}
det +=
s * (pSrc[c] * ref_detrm64(temp, temp + size * size, size - 1));
s = -s;
}
}
return (det);
}
void ref_cofact64(float64_t *pSrc, float64_t *pDst, float64_t *temp,
uint32_t size)
{
int p, q, m, n, i, j;
if (size == 1) {
pDst[0] = 1;
return;
}
for (q = 0; q < size; q++) {
for (p = 0; p < size; p++) {
m = 0;
n = 0;
for (i = 0; i < size; i++) {
for (j = 0; j < size; j++) {
temp[i * size + j] = 0;
if (i != q && j != p) {
temp[m * (size - 1) + n] = pSrc[i * size + j];
if (n < (size - 2)) {
n++;
} else {
n = 0;
m++;
}
}
}
}
pDst[q * size + p] =
ref_pow(-1, q + p) *
ref_detrm64(temp, temp + (size - 1) * (size - 1), size - 1);
}
}
}

View file

@ -0,0 +1,80 @@
#include "ref.h"
float32_t scratchArray[8192 * 2];
riscv_cfft_instance_f32 ref_cfft_sR_f32_len8192 = {8192, 0, 0, 0};
q31_t ref_sat_n(q31_t num, uint32_t bits)
{
int32_t posMax, negMin;
uint32_t i;
posMax = 1;
for (i = 0; i < (bits - 1); i++) {
posMax = posMax * 2;
}
if (num > 0) {
posMax = (posMax - 1);
if (num > posMax) {
num = posMax;
}
} else {
negMin = -posMax;
if (num < negMin) {
num = negMin;
}
}
return (num);
}
q31_t ref_sat_q31(q63_t num)
{
if (num > (q63_t)INT_MAX) {
return INT_MAX;
} else if (num < (q63_t)0xffffffff80000000ll) {
return INT_MIN;
} else {
return (q31_t)num;
}
}
q15_t ref_sat_q15(q31_t num)
{
if (num > (q31_t)SHRT_MAX) {
return SHRT_MAX;
} else if (num < (q31_t)0xffff8000) {
return SHRT_MIN;
} else {
return (q15_t)num;
}
}
q7_t ref_sat_q7(q15_t num)
{
if (num > (q15_t)SCHAR_MAX) {
return SCHAR_MAX;
} else if (num < (q15_t)0xff80) {
return SCHAR_MIN;
} else {
return (q7_t)num;
}
}
float32_t ref_pow(float32_t a, uint32_t b)
{
uint32_t i;
float32_t r = a;
for (i = 1; i < b; i++) {
r *= a;
}
if (b == 0) {
return 1;
}
return r;
}