1 files changed, 157 insertions, 0 deletions
diff --git a/gr-qtgui/src/lib/fftdisplay.cc b/gr-qtgui/src/lib/fftdisplay.cc
new file mode 100644
index 0000000000..d701642801
--- /dev/null
+++ b/gr-qtgui/src/lib/fftdisplay.cc
@@ -0,0 +1,157 @@
+#ifndef FFT_DISPLAY_CC
+#define FFT_DISPLAY_CC
+
+#include <qwt_painter.h>
+#include <qwt_plot_canvas.h>
+#include <qwt_plot_curve.h>
+#include <qwt_scale_engine.h>
+#include <qapplication.h>
+#include <fftdisplay.h>
+
+const int fft_display_event::EVENT_TYPE_ID = QEvent::User+100;
+
+fft_display_event::fft_display_event( std::vector<gr_complex>* fft_data, const float start_frequency, const float stop_frequency):QCustomEvent(fft_display_event::EVENT_TYPE_ID){
+	d_fft_data.resize(fft_data->size());
+	for(unsigned int i = 0; i < fft_data->size(); i++){
+		d_fft_data[i] = fft_data->operator[](i);
+	}
+	d_start_frequency = start_frequency;
+	d_stop_frequency = stop_frequency;
+}
+
+fft_display_event::~fft_display_event(){
+
+}
+
+const std::vector<gr_complex>& fft_display_event::get_fft_data() const{
+	return d_fft_data;
+}
+
+float fft_display_event::get_start_frequency()const{
+	return d_start_frequency;
+}
+
+float fft_display_event::get_stop_frequency()const{
+	return d_stop_frequency;
+}
+
+fft_display::fft_display(const unsigned int fft_size, QWidget* parent):
+	QwtPlot(parent)
+{
+    	// Disable polygon clipping
+    	QwtPainter::setDeviceClipping(false);
+
+    	// We don't need the cache here
+    	canvas()->setPaintAttribute(QwtPlotCanvas::PaintCached, false);
+    	canvas()->setPaintAttribute(QwtPlotCanvas::PaintPacked, false);
+
+	canvas()->setPaletteBackgroundColor(QColor("white"));
+
+	d_fft_bin_size = fft_size;
+	if(d_fft_bin_size < 1){
+		d_fft_bin_size = 1;
+	}
+
+	d_start_frequency = 0.0;
+	d_stop_frequency = 4000.0;
+
+	d_fft_data = new std::vector<gr_complex>(d_fft_bin_size);
+	d_plot_data = new double[d_fft_bin_size];
+	d_x_data = new double[d_fft_bin_size];
+	for( unsigned int i = 0; i < d_fft_bin_size; i++){
+		d_fft_data->operator[](i) = gr_complex(static_cast<float>(i), 0.0);
+		d_x_data[i] = d_start_frequency + ((d_stop_frequency-d_start_frequency) / static_cast<float>(d_fft_bin_size)*static_cast<float>(i));
+		d_plot_data[i] = 1.0;
+	}
+	
+	// Set the Appropriate Axis Scale Engine
+#warning Pass the axis info as necessary...
+	if(true){
+		setAxisScaleEngine(QwtPlot::yLeft, new QwtLinearScaleEngine);
+	}
+	else{
+		setAxisScaleEngine(QwtPlot::yLeft, new QwtLog10ScaleEngine);
+	}
+
+	QwtPlotCurve* fft_plot_curve = new QwtPlotCurve("FFT Spectrum");
+	fft_plot_curve->attach(this);
+	fft_plot_curve->setPen(QPen(Qt::blue));
+	fft_plot_curve->setRawData(d_x_data, d_plot_data, d_fft_bin_size);
+
+	setTitle("Spectral Display");
+}
+
+fft_display::~fft_display(){
+	/* The Qwt objects are destroyed by Qt when their parent is destroyed */
+
+	delete[] d_plot_data;
+	delete[] d_x_data;
+	delete d_fft_data;
+}
+
+
+void fft_display::set_data(const std::vector<gr_complex>& input_data){
+	unsigned int min_points = d_fft_data->size();
+	if(min_points < input_data.size()){
+		min_points = input_data.size();
+	}
+	if(min_points > d_fft_bin_size){
+		min_points = d_fft_bin_size;
+	}
+	for(unsigned int point = 0; point < min_points; point++){
+		d_fft_data->operator[](point) = input_data[point];
+	}
+}
+
+void fft_display::update_display(){
+
+	// Tell the event loop to display the new data - the event loop handles deleting this object
+	qApp->postEvent(this, new fft_display_event(d_fft_data, d_start_frequency, d_stop_frequency));
+
+}
+
+void fft_display::customEvent(QCustomEvent* e){
+	if(e->type() == fft_display_event::EVENT_TYPE_ID){
+		fft_display_event* fft_display_event_ptr = (fft_display_event*)e;
+		// Write out the FFT data to the display here
+
+		gr_complex data_value;
+		for(unsigned int number = 0; number < fft_display_event_ptr->get_fft_data().size(); number++){
+			data_value = fft_display_event_ptr->get_fft_data()[number];
+			d_plot_data[number] = abs(data_value);
+#warning Add code here for handling magnitude, scaling, etc...
+
+			d_x_data[number] = d_start_frequency + ((d_stop_frequency-d_start_frequency) / static_cast<float>(d_fft_bin_size)*static_cast<float>(number));
+		}
+
+    		// Axis 
+    		setAxisTitle(QwtPlot::xBottom, "Frequency (Hz)");
+    		setAxisScale(QwtPlot::xBottom, fft_display_event_ptr->get_start_frequency(), fft_display_event_ptr->get_stop_frequency());
+
+    		setAxisTitle(QwtPlot::yLeft, "Values");
+		replot();
+	}
+}
+
+void fft_display::set_start_frequency(const float new_freq){
+	d_start_frequency = new_freq;
+}
+
+float fft_display::get_start_frequency()const{
+	return d_start_frequency;
+}
+
+void fft_display::set_stop_frequency(const float new_freq){
+	d_stop_frequency = new_freq;
+}
+
+float fft_display::get_stop_frequency()const{
+	return d_stop_frequency;
+}
+
+unsigned int fft_display::get_fft_bin_size()const{
+	return d_fft_bin_size;
+}
+
+
+#endif /* FFT_DISPLAY_CC */