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

#include <liquid/liquid.h>

#include "layer1/freq_est.h"

#include "utils.h"

#define RESULT_CHECK(stmt) { \
	result_t res = stmt; \
	if(res != OK) { \
		fprintf(stderr, "Error %d in %s:%d!\n", res, __FILE__, __LINE__); \
		exit(1); \
	} \
}


int main(void)
{
#define N 1024
	float complex s1[N];
	float complex s2[N];

	nco_crcf nco;

	// ** Initialize **

	nco = nco_crcf_create(LIQUID_NCO);

	nco_crcf_set_frequency(nco, 0.05f);
	nco_crcf_set_phase(nco, 1.37f);

	// ** Generate signal **

	for(size_t i = 0; i < N; i++) {
		s1[i] = 1.0f;
	}

	// ** Mix up **

	nco_crcf_mix_block_up(nco, s1, s2, N);

	dump_array_cf(s1, N, 1.0f, "/tmp/freq_est_source.cpx");
	dump_array_cf(s2, N, 1.0f, "/tmp/freq_est_mixed.cpx");


	// ** Estimate frequency **

	float final_phase;
	float est_freq = freq_est_data_free(s2, 32, &final_phase);

	fprintf(stderr, "Estimated frequency: %.6f rad/sample\n", est_freq);
	fprintf(stderr, "Final phase: %.6f rad\n", final_phase);

	// ** Cleanup **

	nco_crcf_destroy(nco);

	fprintf(stderr, "Done.\n");
}