first commit

Author: zhodj

arc_design_contest/2021/HLJU_Pathological_Voice_Diagnosis/README.md

@@ -0,0 +1 @@
+

arc_design_contest/2021/HLJU_Pathological_Voice_Diagnosis/images/WM8978.png

@@ -0,0 +1,11 @@
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+void cppsetup();
+
+void cpploop();
+
+#ifdef __cplusplus
+}
+#endif

arc_design_contest/2021/HLJU_Pathological_Voice_Diagnosis/images/arc_iot_dev_kit.png

@@ -0,0 +1,153 @@
+#include "stdlib.h"
+#include "math.h"
+
+//#define __EXTERN_W__
+
+typedef struct
+{
+	float real;
+	float imag;
+} Complex;
+
+#ifdef __EXTERN_W__
+//========Paste the definitions for W here========
+const Complex FFT_W[15] = 
+{
+	{0.98078528040323,-0.195090322016128},
+	{0.923879532511287,-0.38268343236509},
+	{0.831469612302545,-0.555570233019602},
+	{0.707106781186547,-0.707106781186547},
+	{0.555570233019602,-0.831469612302545},
+	{0.38268343236509,-0.923879532511287},
+	{0.195090322016128,-0.98078528040323},
+	{5.55111512312578E-17,-1},
+	{-0.195090322016128,-0.98078528040323},
+	{-0.38268343236509,-0.923879532511287},
+	{-0.555570233019602,-0.831469612302545},
+	{-0.707106781186547,-0.707106781186547},
+	{-0.831469612302545,-0.555570233019602},
+	{-0.923879532511286,-0.38268343236509},
+	{-0.98078528040323,-0.195090322016128}
+};
+const Complex *w = FFT_W;
+//================================================
+#endif
+
+unsigned char fftx(Complex *d,int m)
+{
+#ifndef __EXTERN_W__
+	static Complex *w;
+	static int mw = 0;
+	float arg, w_real, w_imag, wr_real, wr_imag, wtemp;
+#endif
+	static int n = 1;
+	Complex temp, tmp1, tmp2;
+	Complex *di, *dip, *dj, *wp;
+	int i, j, k, l, le, wi;
+#ifndef __EXTERN_W__
+	if(m != mw)
+	{
+		if(mw != 0)
+			free(w);
+		mw = m;
+		if(m == 0)
+			return 0;
+		n = 1 << m;
+		le = n >> 1;
+		w = (Complex*)calloc(le - 1,sizeof(Complex));
+		if(!w)
+			return 0;
+		arg = 4.0 * atan(1.0) / le;
+		wr_real = w_real = cos(arg);
+		wr_imag = w_imag = -sin(arg);
+		dj = w;
+		for(j = 1; j < le; j++)
+		{
+			dj->real = (float)wr_real;
+			dj->imag = (float)wr_imag;
+			dj++;
+			wtemp = wr_real * w_real - wr_imag * w_imag;
+			wr_imag = wr_real * w_imag + wr_imag * w_real;
+			wr_real = wtemp;
+		}
+	}
+#else
+	n = 1 << m;
+#endif
+	le = n;
+	wi = 1;
+	for(l = 0; l < m; l++)
+	{
+		le >>= 1;
+		for(i = 0; i < n; i += (le << 1))
+		{
+			di = d + i;
+			dip = di + le;
+			temp.real = (di->real + dip->real);
+			temp.imag = (di->imag + dip->imag);
+			dip->real = (di->real - dip->real);
+			dip->imag = (di->imag - dip->imag);
+			*di = temp;                       
+		}
+		wp = (Complex*)w + wi - 1;
+		for(j = 1; j < le; j++)
+		{
+			tmp1 = *wp;
+			for(i = j; i < n; i += (le << 1))
+			{
+				di = d + i;
+				dip = di + le;
+				temp.real = (di->real + dip->real);
+				temp.imag = (di->imag + dip->imag);
+				tmp2.real = (di->real - dip->real);
+				tmp2.imag = (di->imag - dip->imag);
+				dip->real = (tmp2.real * tmp1.real - tmp2.imag * tmp1.imag);
+				dip->imag = (tmp2.real * tmp1.imag + tmp2.imag * tmp1.real);
+				*di = temp;
+			}
+			wp += wi;
+		}
+		wi <<= 1;
+	}
+	
+	for(i = 0; i < n; i++)
+	{
+		j = 0;
+		for(k = 0; k < m; k++)
+			j = (j << 1) | ((i >> k) & 1);
+		if(i < j)
+		{
+			di = d + i;
+			dj = d + j;
+			temp = *dj;
+			*dj = *di;
+			*di = temp;
+		}
+	}
+	return 1;
+}
+
+void AmpSpectrum(Complex *pData,int m,float* pDCAmp,float* pDistortion)
+{
+   // Amp = Abs / (1.414213562 * N)
+	int i;
+	int N = 1 << (m - 1);
+	int N2 = 1 << m;
+	float BasePower;     //exactly BasePower*2*N*N;
+	float SyntPower = 0; //exactly SyntPower*2*N*N;
+	float ModelSquare;
+	*pDCAmp = sqrt(pData[0].real * pData[0].real + pData[0].imag * pData[0].imag) / (N << 1);
+	BasePower = pData[1].real * pData[1].real + pData[1].imag * pData[1].imag;
+	pData[N2 - 1].real = sqrt(BasePower);
+	pData[N2 - 1].imag = asin(pData[N2 - 1].imag / pData[N2 - 1].real);
+	pData[N2 - 1].real /= N;
+	for(i = 2; i < N; i++)
+	{
+		ModelSquare = pData[i].real * pData[i].real + pData[i].imag * pData[i].imag;
+		pData[N2 - i].real = sqrt(ModelSquare);
+		pData[N2 - i].imag = asin(pData[N2-i].imag/pData[N2-i].real);
+		pData[N2 - i].real /= N;
+		SyntPower += ModelSquare;
+	}
+	*pDistortion = sqrt(SyntPower / BasePower);
+}

arc_design_contest/2021/HLJU_Pathological_Voice_Diagnosis/images/oled.png

@@ -0,0 +1,10 @@
+#include "stdlib.h"
+#include "math.h"
+
+typedef struct
+{
+	float real;
+	float imag;
+} Complex;
+
+unsigned char fftx(Complex *x,int m);

arc_design_contest/2021/HLJU_Pathological_Voice_Diagnosis/src/CppAddWrapper.h

@@ -0,0 +1,114 @@
+#include "codec.h"
+#define AUDIO_BUF_NUM			640			/* !<the length of buffer>*/
+
+int16_t* audio_buf = NULL;
+int16_t* rx_buffer1 = NULL;
+DEV_BUFFER rx_buffer,tx_buffer;
+
+unsigned long frame_energy(int16_t *buf, uint16_t len)
+{
+	unsigned long sum = 0;
+	int16_t i, tmp;
+
+	for(i=0;i<len;i++)
+	{
+		tmp =  buf[2*i];
+		sum += tmp * tmp;
+	}
+
+	return sum;
+}
+
+uint16_t zc(int16_t *buf, uint16_t len)//zero cross times.
+{
+	uint16_t cnt=0,i;
+	int16_t tmp1,tmp2;
+
+	for(i=0;i<(len-1);i++)
+	{
+		tmp1 = buf[2*i];
+		tmp2 = buf[2*i+2];
+		if(((tmp1>0)&&(tmp2<0))||((tmp1<0)&&(tmp2>0)))cnt++;
+	}
+
+	return cnt;
+}
+
+volatile uint8_t bufcnt=0,word_state=0;
+volatile int16_t state_cnt=0;
+
+static void cs4344_rx_isr(void)
+{
+	uint16_t zc_cnt,i;
+	unsigned long eng;
+
+	zc_cnt = zc((rx_buffer1 + 640*bufcnt), 320);
+	eng = frame_energy((rx_buffer1 + 640*bufcnt), 320);
+	if((eng>200000000)&&((zc_cnt>4)&&(zc_cnt<55))&&(state_cnt<7))state_cnt++;//audio valid condition.
+	else if(state_cnt > 0)state_cnt--;
+
+	if(bufcnt<17)bufcnt++;
+	else bufcnt = 0;
+
+	if(state_cnt > 6)
+	{
+		rx_buffer.buf = audio_buf + 640*16;
+		rx_buffer.len = 16*1000*2 - 640*16;
+		cs4344_rx_isr_restart(&rx_buffer);
+		for(i=0;i<16;i++)
+		{
+			if((bufcnt + i + 2)>17)memcpy(audio_buf + i*640,rx_buffer1 + (bufcnt + i + 2 - 18)*640,1280);
+			else memcpy(audio_buf + i*640,rx_buffer1 + (bufcnt + i + 2)*640,1280);
+		}
+		state_cnt = 0;
+		word_state = 1;
+	}
+	else if(word_state == 1)
+	{
+		word_state = 2;
+		rx_buffer.len = 640;
+		rx_buffer.buf = rx_buffer1 + 640*bufcnt;
+	}
+	else
+	{
+		rx_buffer.len = 640;
+		rx_buffer.buf = rx_buffer1 + 640*bufcnt;
+		cs4344_rx_isr_restart(&rx_buffer);
+	}
+}
+
+void extract_features(MFCC_STR *mfcc_str, const int16_t *audio_data, int8_t *mfcc_buffer)
+{
+	uint16_t f;
+	int32_t mfcc_buffer_head = 0;
+	for (f = 0; f < NUM_FRAMES; f++)
+    {
+      mfcc_compute(mfcc_str, audio_data + (f * FRAME_SHIFT * 2),&mfcc_buffer[mfcc_buffer_head]);
+      mfcc_buffer_head += NUM_MFCC_COEFFS;
+    }
+}
+
+static void cs4344_tx_isr(void)
+{}
+
+void CODEC_init(int16_t* ao_buf, int16_t* rx_buf)
+{
+
+  rx_buffer1 = rx_buf;
+  audio_buf = ao_buf;
+
+	rx_buffer.buf = (int16_t *)rx_buffer1;
+	rx_buffer.len = AUDIO_BUF_NUM;
+	rx_buffer.ofs = 0;
+
+	tx_buffer.buf = (int16_t *)audio_buf;
+	tx_buffer.len = 16*1000*2;
+	tx_buffer.ofs = 0;
+
+	cs4344_rx_init(CS4344_SAMPLE_FREQ_16, CS4344_DATA_FORMAT_16, CS4344_MODE_ISR, &rx_buffer, cs4344_rx_isr);
+	cs4344_tx_init(CS4344_SAMPLE_FREQ_16, CS4344_DATA_FORMAT_16, CS4344_MODE_ISR, &tx_buffer, cs4344_tx_isr);
+
+	EMBARC_PRINTF("Please connet the mclk !\r\n");
+	board_delay_ms(5000, OSP_DELAY_OS_COMPAT_DISABLE);
+}
+

arc_design_contest/2021/HLJU_Pathological_Voice_Diagnosis/src/FFT.c

@@ -0,0 +1,15 @@
+#ifndef _CODEC_H_
+#define _CODEC_H_
+
+#include "embARC.h"
+#include "embARC_debug.h"
+#include "wm8978i2s.h"
+#include "mfcc.h"
+
+void CODEC_init(int16_t* ao_buf, int16_t* rx_buf);
+//void CODEC_init();
+
+void extract_features(MFCC_STR *mfcc_str, const int16_t *audio_data, int8_t *mfcc_buffer);
+
+#define audio_buffer_len 16 * 1000 * 2
+#endif