polarplotwidget.h

Go to the documentation of this file.

 
 
#ifndef POLARPLOTWIDGET_H
#define POLARPLOTWIDGET_H
 
#include "UI/PlotWidgets/QCustomPlot/qcustomplot.h"
#include "UI/CustomWidgets/CustomDoubleSpinBox.h"
#include <QCheckBox>
#include <QComboBox>
#include <QGridLayout>
#include <QLabel>
#include <QMap>
#include <QPen>
#include <QWidget>
#include <complex>
#include <limits>
 
 
class PolarPlotWidget : public QWidget {
  Q_OBJECT
 
public:
  struct Trace {
    QList<std::complex<double>> values;  
    QList<double> frequencies;           
    QPen pen;                            
    int displayMode;                     
  };
 
  struct Marker {
    QString id;          
    double frequency;    
    QPen pen;            
  };
 
  struct AxisSettings {
    double freqMax;         
    double freqMin;         
    QString freqUnit;       
    double radius_min;      
    double radius_max;      
    double radius_div;      
    QString marker_format;  
  };
 
  void setSettings(const AxisSettings& settings);
 
  PolarPlotWidget::AxisSettings getSettings() const;
 
  explicit PolarPlotWidget(QWidget* parent = nullptr);
 
  ~PolarPlotWidget(){
    clearGraphicsItems(); // Clean up any remaining graphics items
    }
 
  // Trace management functions
 
  void addTrace(const QString& name, const Trace& trace);
 
  void removeTrace(const QString& name);
 
  void clearTraces();
 
  QPen getTracePen(const QString& traceName) const;
 
  void setTracePen(const QString& traceName, const QPen& pen);
 
  QMap<QString, QPen> getTracesInfo() const;
 
  // Axis value access functions
 
  double getRmax() { return rAxisMax->value(); }
 
  double getRmin() { return rAxisMin->value(); }
 
  double getRdiv() { return rAxisDiv->value(); }
 
  int getDisplayMode() const {
    return displayModeCombo->currentIndex();
  }
 
 
  // Marker management functions
 
  bool addMarker(const QString& markerId, double frequency,
                 const QPen& pen = QPen(Qt::red, 2));
 
  bool removeMarker(const QString& markerId);
 
  bool updateMarkerFrequency(const QString& markerId, double newFrequency);
 
  void clearMarkers() {
    markers.clear();
    updatePlot();
  }
 
  QMap<QString, double> getMarkers() const;
 
  QCustomPlot* customPlot() const { return plot; }
 
private slots:
  void updateRAxis();
 
  void updateAngleAxis() {
    updatePlot();
  }
 
  void toggleDisplayMode() { updatePlot(); }
 
  void onFMinChanged(double value){
    fMin = value * getFrequencyMultiplier();
    updatePlot();
  }
 
  void onFMaxChanged(double value) {
    fMax = value * getFrequencyMultiplier();
    updatePlot();
  }
 
  void onFUnitChanged() {
    fMin = fMinSpinBox->value() * getFrequencyMultiplier();
    fMax = fMaxSpinBox->value() * getFrequencyMultiplier();
    updatePlot();
  }
 
private slots: // Slot to update axis settings widget based on the user zoomming and panning
 
  void checkAxisRanges();
 
  void checkAxisRanges(QMouseEvent* event) {
    // This overload handles mouse events - just call the main function
    Q_UNUSED(event)
    checkAxisRanges();
  }
 
  void onRadialRangeChanged(const QCPRange& newRange);
 
private:
  QCustomPlot* plot;
  QCPPolarAxisAngular* angularAxis;
  QCPPolarAxisRadial* radialAxis;
 
  CustomDoubleSpinBox* rAxisMin;
  CustomDoubleSpinBox* rAxisMax;
  CustomDoubleSpinBox* rAxisDiv;
  QComboBox* displayModeCombo;
 
  CustomDoubleSpinBox* fMinSpinBox;
  CustomDoubleSpinBox* fMaxSpinBox;
  QComboBox* fUnitComboBox;
  double fMin;
  double fMax;
  QStringList frequencyUnits;
 
  QMap<QString, Trace> traces;
  QMap<QString, Marker> markers;
  QMap<QString, QList<QCPPolarGraph*>>
      traceGraphs; // Each trace can have multiple graphs for phase wrapping
 
  // Marker items for drawing
  QList<QCPItemEllipse*> markerItems;
  QList<QCPItemText*> markerLabels;
 
  void updateFrequencyRange();
 
  QGridLayout* setupAxisSettings();
 
  void updatePlot();
 
  void clearGraphicsItems();
 
  void drawCustomMarkers();
 
  double getFrequencyMultiplier() const;
 
  std::complex<double> getComplexValueAtFrequency(const Trace& trace,
                                                  double frequency);
};
 
#endif // POLARPLOTWIDGET_H