main.cpp

//******************************************************************************
// ïÂÏÚÎÁÞÅÎÉÅ:			main.c 				       
// îÁÚÎÁÞÅÎÉÅ:			
// òÁÚÒÁÂÏÔÞÉË:     	óÅÒÇÅÅ×Á å.é.
// äÁÔÁ ÐÏÓÌÅÄÎÅÊ ËÏÒÒÅËÔÉÒÏ×ËÉ: 
//******************************************************************************

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <math.h>

#include "def_types.h"

//#include "f_fft.h"
#include "bpf.h"

#define N_FFT 256

/*
æÕÎËÃÉÑ ÏÂÒÁÂÏÔËÉ ÐÏÒÃÉÉ ÄÁÎÎÙÈ:

- âðæ --- ÐÏËÁ ÞÔÏ ×ÏÚØÍÅÍ "ÏÂÅÒÔËÕ" ÄÌÑ Æ-ÉÉ bpf ÉÚ ÆÁÊÌÁ bpf.h
          NB ÍÏÖÎÏ ÚÁÍÅÎÉÔØ ÎÁ Æ-ÉÉ ÉÚ FFTW ÉÌÉ ÄÒÕÇÉÅ ÓÐÅÃ. ÂÉÂÌÉÏÔÅËÉ     
- ×ÙÞÉÓÌÅÎÉÅ Ë×. ÍÏÄÕÌÅÊ

÷ÈÏÄ:
    float s_in[N_FFT] --- ÐÏÒÃÉÑ ×ÈÏÄÎÏÇÏ ÓÉÇÎÁÌÁ 
÷ÙÈÏÄ:
    float A_out[N_FFT/2] --- Ë×ÁÄÒÁÔÙ ÍÏÄÕÌÅÊ ÓÐÅËÔÒÏ×
 */
void f_processing(float *s_in, float *A_out){

  //	cfloat fft_out[N_FFT];
  //	memset((float*) fft_out, 0.0, 2*N_FFT*sizeof(float));
  //	
  //	rfft_256(fft_out, s_in);

  
	float sig_in_re[N_FFT];
	float sig_in_im[N_FFT];

	int i;

	memcpy(sig_in_re, s_in, N_FFT*sizeof(float));
	memset(sig_in_im, 0.0, N_FFT*sizeof(float));


	bpf(256, 1, sig_in_re, sig_in_im);

	//	for(int i = 0; i<N_FFT/2; i++){
	//	A_out[i] = fft_out[i].r*fft_out[i].r + fft_out[i].i*fft_out[i].i;
	//}

	for(int i = 0; i<N_FFT/2; i++){
		A_out[i] = sig_in_re[i]*sig_in_re[i] + sig_in_im[i]*sig_in_im[i];
	}

	return;
}

/*
ðÕÔØ Ë wav-ÆÁÊÌÕ ÍÏÖÎÏ ÚÁÄÁÔØ ÉÚ ËÏÍÁÎÄÎÏÊ ÓÔÒÏËÉ,
ÐÏ ÕÍÏÌÞÁÎÉÀ ÏÔËÒÙ×ÁÅÔÓÑ data/ton_signal.wav --- ÓÇÅÎÅÒÉÒÏ×ÁÎ × íÁÔÌÁÂÅ (ÓÍ. ÐÁÒÁÍÅÔÒÙ ÄÌÑ ÐÒÏ×ÅÒËÉ)
 */
int main(int argc, char** argv) {

	char* fin_name = "data/ton_signal.wav";;
	if(argc > 1){
		fin_name = argv[1];
	}

	FILE* f_in = fopen(fin_name, "rb");
	if(f_in == NULL){
	  printf("ERROR in OPEN FILE [%s] \n", fin_name);
	  return 1;
	} else {
	  printf("OPEN FILE [%s] is OK \n", fin_name);
	}

//ÞÉÔÁÅÍ ÚÁÇÏÌÏ×ÏË
	t_wavhdr header;		

	size_t readed = fread(&header, sizeof(header), 1, f_in);

//ÐÅÞÁÔÁÅÍ ÐÁÒÁÍÅÔÒÙ wav-ÆÁÊÌÁ
	printf("***** Parametrs of WAV-file [%s] *****\n", fin_name);
	
	printf("chunkID =  %s\n", header.chunkID);
	printf("chunkSize =  %i\n", header.chunkSize);
	printf("format =  %s\n", header.format);
	printf("subchunk1ID =  %s\n", header.subchunk1ID);
	printf("subchunk1Size =  %i\n", header.subchunk1Size);
	printf("audioFormat =  %i\n", header.audioFormat);
	printf("numChannels = %i\n", header.numChannels);
	printf("sampleRate =  %i\n", header.sampleRate);
	printf("blockAlign = %i\n", header.blockAlign);
	printf("bitsPerSample = %i\n", header.bitsPerSample);
	
	printf("subchunk2ID =  %s\n", header.subchunk2ID);
	printf("subchunk2Size =  %i\n", header.subchunk2Size);
	
	printf("*********************************************************\n");

	//×ÓÅÇÏ ÏÔÓÞÅÔÏ× × ÆÁÊÌÅ	
	int N_samples = header.subchunk2Size * 8 / header.bitsPerSample;
	int sample_size = header.bitsPerSample/8; //[BYTES]
	printf("Samples count = %i\n", N_samples);

//×ÓÅÇÏ ÃÉËÌÏ× âðæ (ÐÏÒÃÉÊ ÏÂÒÁÂÏÔËÉ)
	int N_portions = N_samples / N_FFT; //ÃÅÌÙÈ
//+ È×ÏÓÔÉË
	int tail = N_samples - N_FFT*N_portions; 
	printf("N_portions for FFT = %i\n", N_portions);
	printf("tail = %i\n", tail);


	short s_in[N_FFT];//short --- × ÐÒÅÄÐÏÌÏÖÅÎÉÉ, ÞÔÏ sample_size = 2 ÂÁÊÔÁ É ËÁÎÁÌ 1
	memset(s_in, 0, N_FFT*sizeof(int));
	
	float fs_in[N_FFT];
	memset(fs_in, 0.0, N_FFT*sizeof(float));
	float A_out[N_FFT/2];
	memset(A_out, 0.0, (N_FFT/2)*sizeof(float));

// ðÏËÁ ÞÔÏ ÒÅÚÕÌØÔÁÔ × ÆÁÊÌ ÂÕÄÅÍ ÚÁÐÉÓÙ×ÁÔØ
	FILE* f_out = fopen("data/A_out.dat", "w");
	if(f_out == NULL){
		printf("ERROR in OPEN FILE [A_out.dat] \n");
	}else{
		printf("CREATING FILE [A_out.dat] is OK \n");
	}

//ïÂÒÁÂÏÔËÁ ÐÏ ×ÓÅÍ ÃÅÌÙÍ ÐÏÒÃÉÑÍ
	for(int j = 0; j < N_portions; j++){
	//þÉÔÁÅÍ ÓÁÍÉ ÄÁÎÎÙÅ ÉÚ ÆÁÊÌÁ ÐÏÒÃÉÑÍÉ ÐÏ N_FFT
		for (int i = 0; i < N_FFT; i++){
			fread(&s_in[i], sample_size, 1, f_in);
//"ÒÁÓÐÁËÏ×ËÁ" ËÁÖÄÏÇÏ ÏÔÓÞÅÔÁ
			fs_in[i] = (float)s_in[i];
		}

//ïÂÒÁÂÏÔËÁ ÎÏ×ÏÊ ÐÏÒÃÉÉ ÄÁÎÎÙÈ
		f_processing(fs_in, A_out);

//ðÏËÁ ÞÔÏ ÚÁÐÉÛÅÍ ÒÅÚÕÌØÔÁÔ × ÆÁÊÌ
		for (int i = 0; i < N_FFT/2; i++){
			fprintf(f_out, "%f\t", A_out[i]);
		} fprintf(f_out, "\n");
			
	}
	
//ïÂÒÁÂÏÔËÁ "È×ÏÓÔÉËÁ"
	//þÉÔÁÅÍ ÓÁÍÉ ÄÁÎÎÙÅ ÉÚ ÆÁÊÌÁ
//	for (int i = 0; i < N_FFT; i++){
//		fread(&s_in[i], sample_size, 1, f_in);
//"ÒÁÓÐÁËÏ×ËÁ" ËÁÖÄÏÇÏ ÏÔÓÞÅÔÁ
		//cs_in[i].r = (float)s_in[i];
		//cs_in[i].i = 0.0;
		//printf("[%i]%i\t(%f, %f)\n", i, s_in[i], cs_in[i].r, cs_in[i].i);
//	}
//		f_processing();
	
//ðÏËÁ ÞÔÏ ÚÁÐÉÛÅÍ ÒÅÚÕÌØÔÁÔ × ÆÁÊÌ

///////////////////////////////////////
// wav ÆÁÊÌ ÐÒÏÞÉÔÁÎ
	fclose(f_in);
	printf("FILE [%s] is READED \n", fin_name);
///////////////////////////////////////
//ÒÅÚÕÌØÔÁÔ ÔÏÖÅ ÚÁÐÉÓÁÎ
	fclose(f_out);
	printf("FILE [A_out.dat] is WRITED \n");

  return 0;
}