#include "stdafx.h"
#include "mmsystem.h"
#include "math.h"
#include "makewave.h"
#include "findperi.h"
#include "realfft.h"

static double freqvalue(double * fbuffer, double f)
{
	double s;
	s=0;
	for(int i=1; i<3; i++)
		s+=peakvalue(fbuffer, (int)(i*f));
	return s;
}

double searchfreq(double * fbuffer, double& f){
	f=pow(2,f/12);
	double smax,s;
	smax = 0;
	double f1,fmax;
	for(f1=f*0.8; f1<f*1.2; f1+=1){
		s=freqvalue(fbuffer, f1);
		if(s>smax){
			smax = s;
			fmax = f1;
		}
	}
	f=12*log(fmax)/log(2);
	return f;
}

static double freqerr(double f, double testf)
{
	double rmd;
	rmd = fmod(f,testf);
	if(rmd>testf/2)
		rmd = testf-rmd;
	rmd/=f;
	return rmd;
}

static int testfreq(int peaknumber, double * peaks, double testf)
{
	double err;
	err=0;
	for(int i=0; i<peaknumber-1; i++){
		err+=freqerr(peaks[i],testf);
	}
	if(err<0.03*(peaknumber-1))
		return 1;
	return 0;
}

double getfreq(int peaknumber, double * peaks)
{
	if(peaknumber == 0)
		return 0;
	double fmax;
	fmax = peaks[peaknumber-1];
	if(peaknumber == 1)
		return fmax;
	double f;
	int n=0;
	for(int i=1; i<30; i++){
		f=fmax/i;
		if(f<30)
			break;
		if(testfreq(peaknumber, peaks, f)) {
			n=1;
			break;
		}
	}
	if(n)
		return f;
	return 0;
}

static double sinc(double x)
{
	if(fabs(x)<0.0000000001)
		return 1;
	return sin(x)/x;
}

double peakvalue(double * pr, int n){
	if(n<15)return 0;
	double s,s1;
	s=0;
	s1=0;
	for(int i=-15; i<15; i++){
		double x;
		x=pr[n+i];
		s+=x*x;
		x*=1-sinc(i*3.14159265358979/8);
		s1+=x*x;
	}
	return s/s1;
}

static double nearpeakvalue(double * pr, int n)
{
	double s=0;
	double s1;
	for(int i=-4; i<=4; i++){
		s1=peakvalue(pr, n+i);
		if(s1>s)
			s=s1;
	}
	return s;
}

static double totalpeakvalue(double * pr, int fb, int f)
{
	double s=0;
	for(int i=0; i<5; i++){
		s+=nearpeakvalue(pr, fb+f*i);
	}
	return s;
}

static int istherepeak(double * pr, int n)
{
	if(n<20 || n>1000) return 0;
	double s=pr[n];
	for(int i=n-10; i<=n+10; i++){
		if(pr[i]>s)
			return 0;
	}
	if(peakvalue(pr,n)>=3)
		return 1;
	return 0;
}

int findpeaks(double * pr, double * peaks){
	double smax=0;
	int i;
//	for(i=0; i<1000; i++)
//		if(pr[i]>smax)
//			smax=pr[i];
	int pnumber=0;
	for(i=30; i<1000; i++){
		if(istherepeak(pr, i)){
//			if(pr[i]<smax/8)
//				continue;
			peaks[pnumber]=i;
			pnumber++;
			if(pnumber>=80)
				break;
		}
	}
	if(pnumber>1)
		for(i=0; i<pnumber-1; i++)
			if(peaks[i+1]-peaks[i]<=30) {
				return 0;
			}
	return pnumber;
}

static int findfreq(double * pr)
{
	double s;
	s=0;
	int freq,fmax;
	int i;
	for(i=30; i<500; i++){
		if(s<pr[i]){
			s=pr[i];
			fmax=i;
		}
	}
	s=0;
	freq=fmax;
//	if(peakvalue(pr,fmax)<3)
//		return 0;
	double peaks[100];
	int peaknumber;
	peaknumber = findpeaks(pr,peaks);
	if(peaknumber<=2)
		return 0;
	freq = (int)getfreq(peaknumber,peaks);
	return freq;
}

int cbhrecord::isnoise(unsigned char * buf, DWORD size)
{
	static double pr[4096];
	int n,i,ii;
	for(i=0; i<4096; i++)
		pr[i]=0.0;
	ii=4096;
	if(size<ii)
		ii=size;
	for(i=0; i<ii; i++)
	{
		pr[i]=(double)(buf[i]-128);
	}
	n=4096;
	realfftm(pr,n);

	double s,smax;
	
	smax=0;
	for(i=15; i<100; i++)
	{
		if(smax<pr[i])
			smax=pr[i];
	}
	s=0;
	for(i=2000;i<2048;i++)
		s+=pr[i];
	s/=48;
	if(s<=0.00000001)
		noiselevel = 0;
	else
	noiselevel = smax/s;
	if(noiselevel<150)
		return 1;
	return 0;
}

int cbhrecord::start(HWND hWnd)
{
//	if(state)
//		return 0;
	if(!hWnd)
		return 0;
	DWORD i;
	rsize = 0;
	i=waveInOpen(&hwavein,WAVE_MAPPER,&waveformat,(WORD)hWnd,0,CALLBACK_WINDOW);
	i=waveInPrepareHeader(hwavein,&pwh1,sizeof(pwh1));
	i=waveInAddBuffer(hwavein,&pwh1, sizeof(pwh1));
	i=waveInPrepareHeader(hwavein,&pwh2,sizeof(pwh2));
	i=waveInAddBuffer(hwavein,&pwh2, sizeof(pwh2));
	i=waveInStart(hwavein);
	state = 1;
	rsize = 0;
	return 1;
}

int cbhrecord::stop()
{
	if(!state) return 1;
	DWORD i;
	i=waveInStop(hwavein);
	i=waveInReset(hwavein);
	i=waveInUnprepareHeader(hwavein, &pwh1, sizeof(pwh1));
	i=waveInUnprepareHeader(hwavein, &pwh2, sizeof(pwh2));
	i=waveInClose(hwavein);
	state = 0;
	return 1;
}

int cbhrecord::resycle(DWORD p)
{
	if(!p) return 1;
	if(rsize == tbsize) return 1;
	LPWAVEHDR pwh = (LPWAVEHDR)p;
	DWORD i;
	i=waveInUnprepareHeader(hwavein, pwh, sizeof(WAVEHDR));
	if(rsize+(pwh->dwBytesRecorded)>tbsize)
	{
		stop();
		return 1;
	}
	int j;
	j = isnoise((unsigned char*)pwh->lpData,pwh->dwBytesRecorded);
	if(j && (state==1)) {
		i=waveInPrepareHeader(hwavein,pwh,sizeof(WAVEHDR));
		i=waveInAddBuffer(hwavein,pwh, sizeof(WAVEHDR));
		return 0;
	}
	if(state==1)state=2;
	memcpy(tbuffer+rsize,pwh->lpData,pwh->dwBytesRecorded);
	rsize += pwh->dwBytesRecorded;
	if(j)
		state++;
	else
		state=2;
	i=waveInPrepareHeader(hwavein,pwh,sizeof(WAVEHDR));
	i=waveInAddBuffer(hwavein,pwh, sizeof(WAVEHDR));
	if((rsize >= tbsize) || (state>7)){
		if(state>7) {
			state -=4;
			if((double)rsize-state*bsize>0)
				rsize-=state*bsize;
		}
		stop();
		return 1;
	}
	return 0;
}

int cbhanalysis::isnoise(unsigned char * buffer, int size, double& freq)
{
	int n,i;
	if(size>16384)return 1;
	for(i=0; i<16384; i++)
		pr[i]=0;
	for(i=0; i<size; i++)
		pr[i]=(double)(buffer[i]-128);
	n=16384;
	realfftm(pr,16384);
	double s,smax;
	
	smax=0;
	for(i=60; i<400; i++)
	{
		if(smax<pr[i])
			smax=pr[i];
	}
	s=0;
	for(i=8000;i<8048;i++)
		s+=pr[i];
	s/=48;
	if(s>0.00000001)
		if(smax/s<50)
			return 1;
/*	s=0;
	for(i=1;i<30;i++){
		if(pr[i]>s)
			s=pr[i];
	}
	if(s>0.3*smax)
		return 1;*/
	freq = findfreq(pr);
	if(freq<30)
		return 1;
	return 0;
}

void cbhanalysis::calculate(unsigned char * buffer, int size)
{
	if(size <= 0)
		return;
	effect = 1;
	int j,m;
	double k;
	m=0;
	tmsize = 0;
	for(int i=0; i<508; i++)
	{
		j=4096;
		if(i*1024+4096 > size)
			break;
		noise[i]=isnoise(buffer+(i*1024), j, k);
//		if(k<40)
//			noise[i]=1;
		if(!noise[i]) {
			fbuffer[tmsize]=12*log(k)/log(2);
		}
		tmsize++;
		if(!noise[i] && i>6 && k>37)
		{	if(m==0) {
				samplesize=2*int(double(16384)/k);
				if(samplesize > 1000) samplesize = 1000;
				for(int n=0;n<samplesize;n++)
					samplewave[n]=buffer[i*1024+n];
				m=1;
			}
		}
		if(i*1024+4096>=size)
			break;
	}
	for(i=0;i<tmsize;i++) {
		if(!noise[i]) {
			if(i==0)
				break;
			for(int n=0;n<i;n++)
				fbuffer[n]=fbuffer[i];
			break;
		}
	}
	for(i=1;i<tmsize;i++) {
		if(noise[i]){
			fbuffer[i] = fbuffer[i-1];
		}
	}
	for(i=2;i<tmsize-1;i++){
		if((fbuffer[i]>fbuffer[i-1]+3 && fbuffer[i]>fbuffer[i+1]+3)
				|| (fbuffer[i]<fbuffer[i-1]-3 && fbuffer[i]<fbuffer[i+1]-3))
				fbuffer[i]=fbuffer[i-1];
	}

	playfreq(tmsize, fbuffer);
}

void drawcurve(CDC &myDC, CRect &rect, double * buffer, unsigned int size, int pos, 
			   int pnum, double *pks)
{
	CBrush blackbrush(RGB(0,0,0));
	CPen bluepen(PS_SOLID,1,RGB(0,0,255));
	CPen redpen(PS_SOLID,1,RGB(255,0,0));
	CPen greenpen(PS_SOLID,1,RGB(0,100,0));
	myDC.FillRect(&rect, &blackbrush);
	rect.DeflateRect(2,2,2,2);
	CPen * pOldPen;
	pOldPen = myDC.SelectObject(&bluepen);
	double cmax,cmin;
	cmax = -1000000;
	cmin = 10000000;
	unsigned int i;
	for(i=0;i<size;i++){
		if(buffer[i]>cmax)
			cmax = buffer[i];
		if(buffer[i]<cmin)
			cmin = buffer[i];
	}
	if(cmax<=cmin) {
		cmax = cmin;
		cmin = 0;
		if(cmax <= 0)
			cmin = cmax-1;
	}
	CPoint * ps;
	ps = new CPoint[size];
	double x,y;
	int ix,iy;
	for(i=0;i<size;i++) {
		x=i;
		y=buffer[i];
		ix = (int)(rect.left+x*rect.Width()/size);
		iy = (int)(rect.bottom - ((buffer[i]-cmin)/(cmax-cmin))*rect.Height());
		ps[i] = CPoint(ix,iy);
	}
	myDC.Polyline(ps,size);
	myDC.SelectObject(&redpen);
	ix = rect.left+pos*rect.Width()/size;
	myDC.MoveTo(ix,rect.bottom);
	myDC.LineTo(ix,rect.top);
	if(pnum) {
		myDC.SelectObject(&greenpen);
		for(i=0;i<pnum;i++) {
			ix = rect.left+pks[i]*rect.Width()/size;
			myDC.MoveTo(ix,rect.bottom);
			myDC.LineTo(ix,rect.top);
		}
	}

	delete []ps;
	myDC.SelectObject(pOldPen);
}

void drawcurve(CDC &myDC, CRect &rect, unsigned char * buffer, unsigned int size)
{
	double * buf;
	buf = new double[size];
	for(unsigned int i=0; i<size; i++)
		buf[i] = buffer[i];
	drawcurve(myDC,rect,buf,size);
	delete []buf;
}

void playfreq(int tmsize, double * fbuffer)
{
	double f,a;
	unsigned char * buf= new unsigned char[tmsize*1024];
	for(int i=0; i<tmsize; i++){
		f=pow(2,fbuffer[i]/12)*2*3.1415926535/16384;
		for(int j=0; j<1024; j++) {
			a = 128*sin(f*(i*1024+j))+128;
			buf[i*1024+j]=(unsigned char)a;
		}
	}
	myplaysound(buf,tmsize*1024);
}