js8call/plotter.h

// -*- Mode: C++ -*-
///////////////////////////////////////////////////////////////////////////
// Some code in this file and accompanying files is based on work by
// Moe Wheatley, AE4Y, released under the "Simplified BSD License".
// For more details see the accompanying file LICENSE_WHEATLEY.TXT
///////////////////////////////////////////////////////////////////////////

#ifndef PLOTTER_H
#define PLOTTER_H

#include <array>
#include <QColor>
#include <QFrame>
#include <QPixmap>
#include <QSize>
#include <QString>
#include <QVector>
#include "WF.hpp"

extern QVector<QColor> g_ColorTbl;  // XXX

class CPlotter : public QFrame
{
  Q_OBJECT

public:

  using Spectrum = WF::Spectrum;

  explicit CPlotter(QWidget *parent = nullptr);
  ~CPlotter();

  // Sizing

  QSize minimumSizeHint() const override;
  QSize sizeHint()        const override;

  // Inline accessors

  int      binsPerPixel() const { return m_binsPerPixel; }
  int      freq()         const { return m_freq;         }
  float    fSpan()        const { return m_fSpan;        }
  int      plot2dGain()   const { return m_plot2dGain;   }
  int      plot2dZero()   const { return m_plot2dZero;   }
  int      plotGain()     const { return m_plotGain;     }
  int      plotWidth()    const { return m_w;            }
  int      plotZero()     const { return m_plotZero;     }
  bool     scaleOK()      const { return m_scaleOK;      }
  Spectrum spectrum()     const { return m_spectrum;     }
  int      startFreq()    const { return m_startFreq;    }

  int
  frequencyAt(int const x) const
  {
    return static_cast<int>(freqFromX(x));
  }

  // Inline manipulators

  void setFlatten     (bool     const flatten     ) { m_flatten      = flatten;      }
  void setPlot2dZero  (int      const plot2dZero  ) { m_plot2dZero   = plot2dZero;   }
  void setPlotGain    (int      const plotGain    ) { m_plotGain     = plotGain;     }
  void setPlotZero    (int      const plotZero    ) { m_plotZero     = plotZero;     }
  void setSpectrum    (Spectrum const spectrum    ) { m_spectrum     = spectrum;     }
  void setWaterfallAvg(int      const waterfallAvg) { m_waterfallAvg = waterfallAvg; }

  // Statics

  static QVector<QColor>  const & colors()                               { return g_ColorTbl;      }
  static void                     setColours(QVector<QColor> const & cl) {        g_ColorTbl = cl; }

  // Manipulators

  void draw(float swide[], bool bScroll);		//Update the waterfall
  void drawDecodeLine    (const QColor & color, int ia, int ib   );
  void drawHorizontalLine(const QColor & color, int x,  int width);
  void replot();
  void setFreq(int n);
  void setStartFreq(int f);
  void setPlot2dGain(int n);
  void setBinsPerPixel(int n);
  void setPeriod(int n);
  void setDialFreq(double d);
  void setBand(QString const & band);
  void setFilterCenter(int center);
  void setFilterWidth(int width);
  void setFilterEnabled(bool enabled);
  void setFilterOpacity(int alpha);
  void setPercent2DScreen(int percent);
  void setSubMode(int nSubMode);

signals:

  void changeFreq(int);

protected:

  // Event Handlers

  void paintEvent       (QPaintEvent  *) override;
  void resizeEvent      (QResizeEvent *) override;
  void leaveEvent       (QEvent       *) override;
  void wheelEvent       (QWheelEvent  *) override;
  void mouseMoveEvent   (QMouseEvent  *) override;
  void mouseReleaseEvent(QMouseEvent  *) override;

private:

  static constexpr std::size_t MaxScreenSize = 2048;

  // Accessors

  bool  in30MBand()        const;
  int   xFromFreq(float f) const;
  float freqFromX(int   x) const;

  // Manipulators

  void drawOverlay();
  void drawOverlayScale(double, int, float);
  void drawOverlayDial(int);
  void drawOverlayHover(int);
  void drawOverlayFilter();

  std::array<float,  MaxScreenSize> m_sum    = {};
  std::array<QPoint, MaxScreenSize> m_points = {};

  Spectrum  m_spectrum = Spectrum::Current;

  QPixmap m_FilterOverlayPixmap;
  QPixmap m_DialOverlayPixmap;
  QPixmap m_HoverOverlayPixmap;
  QPixmap m_WaterfallPixmap;
  QPixmap m_2DPixmap;
  QPixmap m_ScalePixmap;
  QPixmap m_OverlayPixmap;

  QSize   m_size;
  QString m_band;

  bool m_filterEnabled  = false;
  bool m_paintEventBusy = false;
  bool m_scaleOK        = false;
  bool m_replot         = false;
  bool m_flatten        = false;

  int m_nSubMode         =  0;
  int m_filterCenter     =  0;
  int m_filterWidth      =  0;
  int m_filterOpacity    =  127;
  int m_Percent2DScreen  =  0;
  int m_Percent2DScreen0 =  0;
  int m_plotZero         =  0;
  int m_plotGain         =  0;
  int m_plot2dGain       =  0;
  int m_plot2dZero       =  0;
  int m_binsPerPixel     =  2;
  int m_waterfallAvg     =  1;
  int m_lastMouseX       = -1;
  int m_line             =  0;
  int m_startFreq        =  0;
  int m_freq             =  0;
  int m_w                =  0;
  int m_h                =  0;
  int m_h1               =  0;
  int m_h2               =  0;
  int m_j                =  0;
  int m_TRperiod         =  15;

  float   m_fSpan    = 2000.0f;
  double  m_dialFreq = 0.0;
};

#endif // PLOTTER_H