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

#include "uhd.h"
#include "utils/vector.h"

int uhd_rssi_scan(void *uhd, float *freqs, float *rssi, int nof_bands, double fs, int nsamp) {
	int i;
	int ret = -1;
	_Complex float *buffer;
	double f;

	buffer = calloc(nsamp, sizeof(_Complex float));
	if (!buffer) {
		goto free_and_exit;
	}

	uhd_set_rx_gain(uhd, 0.0);
	uhd_set_rx_srate(uhd, fs);

	for (i=0;i<nof_bands;i++) {
		f = (double) freqs[i];
		uhd_set_rx_freq(uhd, f);
		uhd_rx_wait_lo_locked(uhd);
		if (!i) {
			uhd_start_rx_stream(uhd);
			usleep(500000);
		}
		if (uhd_recv(uhd, buffer, nsamp, 1) != nsamp) {
			goto free_and_exit;
		}
		rssi[i] = vec_power(buffer, nsamp);
		/* FIXME: First sample has a bias of 30 dB!! */
		if (i == 0) {
			rssi[i] /= 1000;
		}
		printf("Freq %4.1f Mhz - RSSI: %3.2f dBm\r", f/1000000, 10*log10f(rssi[i]) + 30); fflush(stdout);
	}
	uhd_stop_rx_stream(uhd);

	ret = 0;
free_and_exit:
	free(buffer);
	return ret;
}