/* -*- c++ -*- */
/*
 * Copyright 2008-2011 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio
 *
 * GNU Radio is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3, or (at your option)
 * any later version.
 *
 * GNU Radio is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GNU Radio; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#ifndef FREQUENCY_DISPLAY_PLOT_C
#define FREQUENCY_DISPLAY_PLOT_C

#include <gnuradio/qtgui/FrequencyDisplayPlot.h>

#include <gnuradio/qtgui/qtgui_types.h>
#include <qwt_scale_draw.h>
#include <qwt_legend.h>
#include <QColor>
#include <iostream>

#if QWT_VERSION < 0x060100
#include <qwt_legend_item.h>
#else /* QWT_VERSION < 0x060100 */
#include <qwt_legend_data.h>
#include <qwt_legend_label.h>
#endif /* QWT_VERSION < 0x060100 */

/***********************************************************************
 * Widget to provide mouse pointer coordinate text
 **********************************************************************/
class FreqDisplayZoomer: public QwtPlotZoomer, public FreqOffsetAndPrecisionClass
{
public:
#if QWT_VERSION < 0x060100
  FreqDisplayZoomer(QwtPlotCanvas* canvas, const unsigned int freqPrecision)
#else /* QWT_VERSION < 0x060100 */
  FreqDisplayZoomer(QWidget* canvas, const unsigned int freqPrecision)
#endif /* QWT_VERSION < 0x060100 */
    : QwtPlotZoomer(canvas),
      FreqOffsetAndPrecisionClass(freqPrecision)
  {
    setTrackerMode(QwtPicker::AlwaysOn);
  }
  
  virtual void updateTrackerText()
  {
    updateDisplay();
  }

  void setUnitType(const std::string &type)
  {
    d_unitType = type;
  }

protected:
  using QwtPlotZoomer::trackerText;
  virtual QwtText trackerText(QPoint const &p) const
  {
    QwtDoublePoint dp = QwtPlotZoomer::invTransform(p);
    QwtText t(QString("%1 %2, %3 dB")
              .arg(dp.x(), 0, 'f', getFrequencyPrecision())
              .arg(d_unitType.c_str()).arg(dp.y(), 0, 'f', 2));
    return t;
  }

private:
  std::string d_unitType;
};


/***********************************************************************
 * Main frequency display plotter widget
 **********************************************************************/
FrequencyDisplayPlot::FrequencyDisplayPlot(int nplots, QWidget* parent)
  : DisplayPlot(nplots, parent)
{
  d_start_frequency = -1;
  d_stop_frequency = 1;

  d_numPoints = 1024;
  d_min_fft_data = new double[d_numPoints];
  d_max_fft_data = new double[d_numPoints];
  d_xdata = new double[d_numPoints];

  setAxisTitle(QwtPlot::xBottom, "Frequency (Hz)");
  setAxisScaleDraw(QwtPlot::xBottom, new FreqDisplayScaleDraw(0));

  d_ymin = -120;
  d_ymax = 10;
  setAxisScaleEngine(QwtPlot::yLeft, new QwtLinearScaleEngine);
  setAxisScale(QwtPlot::yLeft, d_ymin, d_ymax);
  setAxisTitle(QwtPlot::yLeft, "Power (dB)");

  QList<QColor> default_colors;
  default_colors << QColor(Qt::blue) << QColor(Qt::red) << QColor(Qt::green)
	 << QColor(Qt::black) << QColor(Qt::cyan) << QColor(Qt::magenta)
	 << QColor(Qt::yellow) << QColor(Qt::gray) << QColor(Qt::darkRed)
	 << QColor(Qt::darkGreen) << QColor(Qt::darkBlue) << QColor(Qt::darkGray);

  // Automatically deleted when parent is deleted
  for(int i = 0; i < d_nplots; i++) {
    d_ydata.push_back(new double[d_numPoints]);
    memset(d_ydata[i], 0x0, d_numPoints*sizeof(double));

    d_plot_curve.push_back(new QwtPlotCurve(QString("Data %1").arg(i)));
    d_plot_curve[i]->attach(this);

    QwtSymbol *symbol = new QwtSymbol(QwtSymbol::NoSymbol, QBrush(default_colors[i]),
				      QPen(default_colors[i]), QSize(7,7));

#if QWT_VERSION < 0x060000
    d_plot_curve[i]->setRawData(d_xdata, d_ydata[i], d_numPoints);
    d_plot_curve[i]->setSymbol(*symbol);
#else
    d_plot_curve[i]->setRawSamples(d_xdata, d_ydata[i], d_numPoints);
    d_plot_curve[i]->setSymbol(symbol);
#endif
    setLineColor(i, default_colors[i]);
  }
  
  d_min_fft_plot_curve = new QwtPlotCurve("Minimum Power");
  d_min_fft_plot_curve->attach(this);
  const QColor default_min_fft_color = Qt::magenta;
  setMinFFTColor(default_min_fft_color);
#if QWT_VERSION < 0x060000
  d_min_fft_plot_curve->setRawData(d_xdata, d_min_fft_data, d_numPoints);
#else
  d_min_fft_plot_curve->setRawSamples(d_xdata, d_min_fft_data, d_numPoints);
#endif
  d_min_fft_plot_curve->setVisible(false);
  
  d_max_fft_plot_curve = new QwtPlotCurve("Maximum Power");
  d_max_fft_plot_curve->attach(this);
  QColor default_max_fft_color = Qt::darkYellow;
  setMaxFFTColor(default_max_fft_color);
#if QWT_VERSION < 0x060000
  d_max_fft_plot_curve->setRawData(d_xdata, d_max_fft_data, d_numPoints);
#else
  d_max_fft_plot_curve->setRawSamples(d_xdata, d_max_fft_data, d_numPoints);
#endif
  d_max_fft_plot_curve->setVisible(false);
  
  d_lower_intensity_marker= new QwtPlotMarker();
  d_lower_intensity_marker->setLineStyle(QwtPlotMarker::HLine);
  QColor default_marker_lower_intensity_color = Qt::cyan;
  setMarkerLowerIntensityColor(default_marker_lower_intensity_color);
  d_lower_intensity_marker->attach(this);
  
  d_upper_intensity_marker = new QwtPlotMarker();
  d_upper_intensity_marker->setLineStyle(QwtPlotMarker::HLine);
  QColor default_marker_upper_intensity_color = Qt::green;
  setMarkerUpperIntensityColor(default_marker_upper_intensity_color);
  d_upper_intensity_marker->attach(this);

  memset(d_xdata, 0x0, d_numPoints*sizeof(double));

  for(int64_t number = 0; number < d_numPoints; number++){
    d_min_fft_data[number] = 200.0;
    d_max_fft_data[number] = -280.0;
  }

  d_marker_peak_amplitude = new QwtPlotMarker();
  QColor default_marker_peak_amplitude_color = Qt::yellow;
  setMarkerPeakAmplitudeColor(default_marker_peak_amplitude_color);
  /// THIS CAUSES A PROBLEM!
  //d_marker_peak_amplitude->attach(this);

  d_marker_noise_floor_amplitude = new QwtPlotMarker();
  d_marker_noise_floor_amplitude->setLineStyle(QwtPlotMarker::HLine);
  QColor d_default_marker_noise_floor_amplitude_color = Qt::darkRed;
  setMarkerNoiseFloorAmplitudeColor(d_default_marker_noise_floor_amplitude_color);
  d_marker_noise_floor_amplitude->attach(this);

  d_marker_cf= new QwtPlotMarker();
  d_marker_cf->setLineStyle(QwtPlotMarker::VLine);
  QColor d_default_marker_cf_color = Qt::lightGray;
  setMarkerCFColor(d_default_marker_cf_color);
  d_marker_cf->attach(this);
  d_marker_cf->hide();

  d_peak_frequency = 0;
  d_peak_amplitude = -HUGE_VAL;

  d_noise_floor_amplitude = -HUGE_VAL;

  d_zoomer = new FreqDisplayZoomer(canvas(), 0);

#if QWT_VERSION < 0x060000
  d_zoomer->setSelectionFlags(QwtPicker::RectSelection | QwtPicker::DragSelection);
#endif

  d_zoomer->setMousePattern(QwtEventPattern::MouseSelect2,
                            Qt::RightButton, Qt::ControlModifier);
  d_zoomer->setMousePattern(QwtEventPattern::MouseSelect3,
                            Qt::RightButton);


  const QColor default_zoomer_color(Qt::darkRed);
  setZoomerColor(default_zoomer_color);

  // Do this after the zoomer has been built
  _resetXAxisPoints();

  // Turn off min/max hold plots in legend
#if QWT_VERSION < 0x060100
  QWidget *w;
  QwtLegend* legendDisplay = legend();
  w = legendDisplay->find(d_min_fft_plot_curve);
  ((QwtLegendItem*)w)->setChecked(true);
  w = legendDisplay->find(d_max_fft_plot_curve);
  ((QwtLegendItem*)w)->setChecked(true);
#else /* QWT_VERSION < 0x060100 */
  QWidget *w;
  w = ((QwtLegend*)legend())->legendWidget(itemToInfo(d_min_fft_plot_curve));
  ((QwtLegendLabel*)w)->setChecked(true);
  w = ((QwtLegend*)legend())->legendWidget(itemToInfo(d_max_fft_plot_curve));
  ((QwtLegendLabel*)w)->setChecked(true);
#endif /* QWT_VERSION < 0x060100 */

  replot();
}

FrequencyDisplayPlot::~FrequencyDisplayPlot()
{
  for(int i = 0; i < d_nplots; i++)
    delete [] d_ydata[i];
  delete[] d_max_fft_data;
  delete[] d_min_fft_data;
  delete[] d_xdata;
}

void
FrequencyDisplayPlot::setYaxis(double min, double max)
{
  // Get the new max/min values for the plot
  d_ymin = min;
  d_ymax = max;

  // Set the axis max/min to the new values
  setAxisScale(QwtPlot::yLeft, d_ymin, d_ymax);

  // Reset the base zoom level to the new axis scale set here.
  // But don't do it if we set the axis due to auto scaling.
  if(!d_autoscale_state)
    d_zoomer->setZoomBase();
}

void
FrequencyDisplayPlot::setFrequencyRange(const double centerfreq,
					const double bandwidth,
					const double units, const std::string &strunits)
{
  double startFreq  = (centerfreq - bandwidth/2.0f) / units;
  double stopFreq   = (centerfreq + bandwidth/2.0f) / units;

  d_xdata_multiplier = units;

  bool reset = false;
  if((startFreq != d_start_frequency) || (stopFreq != d_stop_frequency))
    reset = true;

  if(stopFreq > startFreq) {
    d_start_frequency = startFreq;
    d_stop_frequency = stopFreq;

    if((axisScaleDraw(QwtPlot::xBottom) != NULL) && (d_zoomer != NULL)) {
      double display_units = ceil(log10(units)/2.0);
      setAxisScaleDraw(QwtPlot::xBottom, new FreqDisplayScaleDraw(display_units));
      setAxisTitle(QwtPlot::xBottom, QString("Frequency (%1)").arg(strunits.c_str()));

      if(reset) {
        _resetXAxisPoints();
        clearMaxData();
        clearMinData();
      }

      ((FreqDisplayZoomer*)d_zoomer)->setFrequencyPrecision(display_units);
      ((FreqDisplayZoomer*)d_zoomer)->setUnitType(strunits);
    }
  }
}


double
FrequencyDisplayPlot::getStartFrequency() const
{
  return d_start_frequency;
}

double
FrequencyDisplayPlot::getStopFrequency() const
{
  return d_stop_frequency;
}

void
FrequencyDisplayPlot::replot()
{
  d_marker_noise_floor_amplitude->setYValue(d_noise_floor_amplitude);
  d_marker_peak_amplitude->setXValue(d_peak_frequency + d_start_frequency);

  // Make sure to take into account the start frequency
//  if(d_useCenterFrequencyFlag){
//    d_marker_peak_amplitude->setXValue((d_peak_frequency/1000.0) + d_start_frequency);
//  }
//  else{
//    _markerPeakAmplitude->setXValue(d_peak_frequency + d_start_frequency);
//  }
  d_marker_peak_amplitude->setYValue(d_peak_amplitude);

  QwtPlot::replot();
}

void
FrequencyDisplayPlot::plotNewData(const std::vector<double*> dataPoints,
				  const int64_t numDataPoints,
				  const double noiseFloorAmplitude, const double peakFrequency,
				  const double peakAmplitude, const double timeInterval)
{
  if(!d_stop) {
    if(numDataPoints > 0) {
      if(numDataPoints != d_numPoints) {
        d_numPoints = numDataPoints;

        delete[] d_min_fft_data;
        delete[] d_max_fft_data;
        delete[] d_xdata;
        d_xdata = new double[d_numPoints];
        d_min_fft_data = new double[d_numPoints];
        d_max_fft_data = new double[d_numPoints];
        
        for(int i = 0; i < d_nplots; i++) {
          delete[] d_ydata[i];
          d_ydata[i] = new double[d_numPoints];
          
#if QWT_VERSION < 0x060000
          d_plot_curve[i]->setRawData(d_xdata, d_ydata[i], d_numPoints);
#else
          d_plot_curve[i]->setRawSamples(d_xdata, d_ydata[i], d_numPoints);
#endif
        }
#if QWT_VERSION < 0x060000
        d_min_fft_plot_curve->setRawData(d_xdata, d_min_fft_data, d_numPoints);
        d_max_fft_plot_curve->setRawData(d_xdata, d_max_fft_data, d_numPoints);
#else
        d_min_fft_plot_curve->setRawSamples(d_xdata, d_min_fft_data, d_numPoints);
        d_max_fft_plot_curve->setRawSamples(d_xdata, d_max_fft_data, d_numPoints);
#endif
        _resetXAxisPoints();
        clearMaxData();
        clearMinData();
      }
      
      double bottom=1e20, top=-1e20;
      for(int n = 0; n < d_nplots; n++) {

        memcpy(d_ydata[n], dataPoints[n], numDataPoints*sizeof(double));

	for(int64_t point = 0; point < numDataPoints; point++) {
	  if(dataPoints[n][point] < d_min_fft_data[point]) {
	    d_min_fft_data[point] = dataPoints[n][point];
	  }
	  if(dataPoints[n][point] > d_max_fft_data[point]) {
	    d_max_fft_data[point] = dataPoints[n][point];
	  }

	  // Find overall top and bottom values in plot.
	  // Used for autoscaling y-axis.
	  if(dataPoints[n][point] < bottom) {
	    bottom = dataPoints[n][point];
	  }
	  if(dataPoints[n][point] > top) {
	    top = dataPoints[n][point];
	  }
	}
      }
      
      if(d_autoscale_state)
	_autoScale(bottom, top);
      
      d_noise_floor_amplitude = noiseFloorAmplitude;
      d_peak_frequency = peakFrequency;
      d_peak_amplitude = peakAmplitude;
      
      setUpperIntensityLevel(d_peak_amplitude);

      replot();
    }
  }
}

void
FrequencyDisplayPlot::plotNewData(const double* dataPoints,
				  const int64_t numDataPoints,
				  const double noiseFloorAmplitude, const double peakFrequency,
				  const double peakAmplitude, const double timeInterval)
{
  std::vector<double*> vecDataPoints;
  vecDataPoints.push_back((double*)dataPoints);
  plotNewData(vecDataPoints, numDataPoints, noiseFloorAmplitude,
	      peakFrequency, peakAmplitude, timeInterval);
}
   
void
FrequencyDisplayPlot::clearMaxData()
{
  for(int64_t number = 0; number < d_numPoints; number++) {
    d_max_fft_data[number] = d_ymin;
  }
}

void
FrequencyDisplayPlot::clearMinData()
{
  for(int64_t number = 0; number < d_numPoints; number++) {
    d_min_fft_data[number] = d_ymax;
  }
}

void
FrequencyDisplayPlot::_autoScale(double bottom, double top)
{
  // Auto scale the y-axis with a margin of 10 dB on either side.
  setYaxis(bottom - 10, top + 10);
}

void
FrequencyDisplayPlot::setAutoScale(bool state)
{
  d_autoscale_state = state;
}

void
FrequencyDisplayPlot::setMaxFFTVisible(const bool visibleFlag)
{
  d_max_fft_visible = visibleFlag;
  d_max_fft_plot_curve->setVisible(visibleFlag);
}

const bool
FrequencyDisplayPlot::getMaxFFTVisible() const
{
  return d_max_fft_visible;
}

void
FrequencyDisplayPlot::setMinFFTVisible(const bool visibleFlag)
{
  d_min_fft_visible = visibleFlag;
  d_min_fft_plot_curve->setVisible(visibleFlag);
}

const bool
FrequencyDisplayPlot::getMinFFTVisible() const
{
  return d_min_fft_visible;
}

void
FrequencyDisplayPlot::_resetXAxisPoints()
{
  double fft_bin_size = (d_stop_frequency - d_start_frequency)
    / static_cast<double>(d_numPoints-1);
  double freqValue = d_start_frequency;
  for(int64_t loc = 0; loc < d_numPoints; loc++) {
    d_xdata[loc] = freqValue;
    freqValue += fft_bin_size;
  }

  setAxisScale(QwtPlot::xBottom, d_start_frequency, d_stop_frequency);

  // Set up zoomer base for maximum unzoom x-axis
  // and reset to maximum unzoom level
  QwtDoubleRect zbase = d_zoomer->zoomBase();
  d_zoomer->zoom(zbase);
  d_zoomer->setZoomBase(zbase);
  d_zoomer->setZoomBase(true);
  d_zoomer->zoom(0);
}

void
FrequencyDisplayPlot::setLowerIntensityLevel(const double lowerIntensityLevel)
{
  d_lower_intensity_marker->setYValue(lowerIntensityLevel);
}

void
FrequencyDisplayPlot::setUpperIntensityLevel(const double upperIntensityLevel)
{
  d_upper_intensity_marker->setYValue(upperIntensityLevel);
}

void
FrequencyDisplayPlot::setTraceColour(QColor c)
{
  d_plot_curve[0]->setPen(QPen(c));
}

void
FrequencyDisplayPlot::setBGColour(QColor c)
{
  QPalette palette;
  palette.setColor(canvas()->backgroundRole(), c);
  canvas()->setPalette(palette);
}

void
FrequencyDisplayPlot::showCFMarker(const bool show)
{
  if(show)
    d_marker_cf->show();
  else
    d_marker_cf->hide();
}

void
FrequencyDisplayPlot::onPickerPointSelected(const QwtDoublePoint & p)
{
  QPointF point = p;
  //fprintf(stderr,"onPickerPointSelected %f %f %d\n", point.x(), point.y(), d_xdata_multiplier);
  point.setX(point.x() * d_xdata_multiplier);
  emit plotPointSelected(point);
}

void
FrequencyDisplayPlot::onPickerPointSelected6(const QPointF & p)
{
  QPointF point = p;
  //fprintf(stderr,"onPickerPointSelected %f %f %d\n", point.x(), point.y(), d_xdata_multiplier);
  point.setX(point.x() * d_xdata_multiplier);
  emit plotPointSelected(point);
}

void
FrequencyDisplayPlot::setMinFFTColor (QColor c)
{
  d_min_fft_color = c;
  d_min_fft_plot_curve->setPen(QPen(c));  
}
const QColor
FrequencyDisplayPlot::getMinFFTColor() const
{
  return d_min_fft_color;
}

void
FrequencyDisplayPlot::setMaxFFTColor (QColor c)
{
  d_max_fft_color = c;
  d_max_fft_plot_curve->setPen(QPen(c));  
}

const QColor
FrequencyDisplayPlot::getMaxFFTColor() const
{
  return d_max_fft_color;
}

void
FrequencyDisplayPlot::setMarkerLowerIntensityColor (QColor c)
{
  d_marker_lower_intensity_color = c;
  d_lower_intensity_marker->setLinePen(QPen(c));
}
const QColor
FrequencyDisplayPlot::getMarkerLowerIntensityColor () const
{
  return d_marker_lower_intensity_color;
}

void
FrequencyDisplayPlot::setMarkerLowerIntensityVisible (bool visible)
{
  d_marker_lower_intensity_visible = visible;
  if(visible)
    d_lower_intensity_marker->setLineStyle(QwtPlotMarker::HLine);
  else
    d_lower_intensity_marker->setLineStyle(QwtPlotMarker::NoLine);
}
const bool
FrequencyDisplayPlot::getMarkerLowerIntensityVisible() const
{
  return d_marker_lower_intensity_visible;
}

void
FrequencyDisplayPlot::setMarkerUpperIntensityColor(QColor c)
{
  d_marker_upper_intensity_color = c;
  d_upper_intensity_marker->setLinePen(QPen(c, 0, Qt::DotLine));
}

const QColor
FrequencyDisplayPlot::getMarkerUpperIntensityColor() const
{
  return d_marker_upper_intensity_color;
}

void
FrequencyDisplayPlot::setMarkerUpperIntensityVisible(bool visible)
{
  d_marker_upper_intensity_visible = visible;
  if(visible)
    d_upper_intensity_marker->setLineStyle(QwtPlotMarker::HLine);
  else
    d_upper_intensity_marker->setLineStyle(QwtPlotMarker::NoLine);
}

const bool
FrequencyDisplayPlot::getMarkerUpperIntensityVisible() const
{
  return d_marker_upper_intensity_visible;
}

void
FrequencyDisplayPlot::setMarkerPeakAmplitudeColor(QColor c)
{
  d_marker_peak_amplitude_color = c;
  d_marker_peak_amplitude->setLinePen(QPen(c));
  QwtSymbol symbol;
  symbol.setStyle(QwtSymbol::Diamond);
  symbol.setSize(8);
  symbol.setPen(QPen(c));
  symbol.setBrush(QBrush(c));
#if QWT_VERSION < 0x060000
  d_marker_peak_amplitude->setSymbol(symbol);
#else
  d_marker_peak_amplitude->setSymbol(&symbol);
#endif
}
const QColor
FrequencyDisplayPlot::getMarkerPeakAmplitudeColor() const
{
  return d_marker_peak_amplitude_color;
}

void
FrequencyDisplayPlot::setMarkerNoiseFloorAmplitudeColor(QColor c)
{
  d_marker_noise_floor_amplitude_color = c;
  d_marker_noise_floor_amplitude->setLinePen(QPen(c, 0, Qt::DotLine));
}

const QColor
FrequencyDisplayPlot::getMarkerNoiseFloorAmplitudeColor() const
{
  return d_marker_noise_floor_amplitude_color;
}

void
FrequencyDisplayPlot::setMarkerNoiseFloorAmplitudeVisible(bool visible)
{
  d_marker_noise_floor_amplitude_visible = visible;
  if(visible)
    d_marker_noise_floor_amplitude->setLineStyle(QwtPlotMarker::HLine);
  else
    d_marker_noise_floor_amplitude->setLineStyle(QwtPlotMarker::NoLine);
}

const bool
FrequencyDisplayPlot::getMarkerNoiseFloorAmplitudeVisible() const
{
  return d_marker_noise_floor_amplitude_visible;
}

void
FrequencyDisplayPlot::setMarkerCFColor(QColor c)
{
  d_marker_cf_color = c;
  d_marker_cf->setLinePen(QPen(c, 0, Qt::DotLine));
}

const QColor
FrequencyDisplayPlot::getMarkerCFColor() const
{
  return d_marker_cf_color;
}

#endif /* FREQUENCY_DISPLAY_PLOT_C */