mirror of
https://github.com/Fishwaldo/bl_mcu_sdk.git
synced 2025-07-22 20:59:03 +00:00
[feat][nmsis] add nmsis component and nn,dsp demo
This commit is contained in:
parent
b2aada479b
commit
5d1126d0f0
989 changed files with 286224 additions and 0 deletions
3
examples/dsp/HelperFunctions/HelperFunctions.c
Normal file
3
examples/dsp/HelperFunctions/HelperFunctions.c
Normal file
|
@ -0,0 +1,3 @@
|
|||
|
||||
#include "mat_helper.c"
|
||||
#include "ref_helper.c"
|
200
examples/dsp/HelperFunctions/math_helper.c
Normal file
200
examples/dsp/HelperFunctions/math_helper.c
Normal 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);
|
||||
}
|
||||
}
|
||||
}
|
80
examples/dsp/HelperFunctions/ref_helper.c
Normal file
80
examples/dsp/HelperFunctions/ref_helper.c
Normal 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;
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue