GitHub-Mirror/bl_mcu_sdk
2
0
Fork 0
mirror of https://github.com/Fishwaldo/bl_mcu_sdk.git synced 2025-07-23 13:18:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

200
examples/dsp/HelperFunctions/math_helper.c Normal file
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);
}
}
}