You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

143 lines
4.7 KiB
C++

/***************************************************************************
* Copyright (C) gempa GmbH *
* All rights reserved. *
* Contact: gempa GmbH (seiscomp-dev@gempa.de) *
* *
* GNU Affero General Public License Usage *
* This file may be used under the terms of the GNU Affero *
* Public License version 3.0 as published by the Free Software Foundation *
* and appearing in the file LICENSE included in the packaging of this *
* file. Please review the following information to ensure the GNU Affero *
* Public License version 3.0 requirements will be met: *
* https://www.gnu.org/licenses/agpl-3.0.html. *
* *
* Other Usage *
* Alternatively, this file may be used in accordance with the terms and *
* conditions contained in a signed written agreement between you and *
* gempa GmbH. *
***************************************************************************/
#ifndef SEISCOMP_GUI_CORE_SPECTRUMWIDGET_H
#define SEISCOMP_GUI_CORE_SPECTRUMWIDGET_H
#include <QWidget>
#include <seiscomp/gui/plot/datay.h>
#include <seiscomp/gui/plot/graph.h>
#ifndef Q_MOC_RUN
#include <seiscomp/math/fft.h>
#include <seiscomp/processing/response.h>
#endif
namespace Seiscomp {
namespace Gui {
class SC_GUI_API SpectrumWidget : public QWidget {
Q_OBJECT
// ----------------------------------------------------------------------
// X'truction
// ----------------------------------------------------------------------
public:
SpectrumWidget(QWidget *parent = 0, Qt::WindowFlags f = 0);
// ----------------------------------------------------------------------
// Public interface
// ----------------------------------------------------------------------
public:
void setMargin(int m) { _margin = qMax(0, m); }
/**
* @brief Sets the spectrum values and the Nyquist frequency as the
* frequency of the last spectrum value.
* @param freqNyquist The last spectrum value is measured for that frequency
* @param spec The spectrum values from 0Hz to [fNyquist]Hz
*/
void setSpectrum(double freqNyquist, const Math::ComplexArray &spec,
Processing::Response *resp = nullptr,
QString exportBasename = QString());
//! Access the x-axis
Axis &xAxis() { return _xAxis; }
//! Access the y-axis
Axis &yAxis() { return _yAxis; }
// ----------------------------------------------------------------------
// Public slots
// ----------------------------------------------------------------------
public slots:
void setAmplitudeSpectrum();
void setPhaseSpectrum();
void setPowerSpectrum();
void setLogScaleX(bool logScale);
void setLogScaleY(bool logScale);
void setShowSpectrum(bool show);
void setShowCorrected(bool show);
void setShowResponse(bool show);
/**
* @brief Exports all visible spectra into a simple ASCII file
*/
void exportSpectra();
// ----------------------------------------------------------------------
// QWidget interface
// ----------------------------------------------------------------------
protected:
void resizeEvent(QResizeEvent *e);
void paintEvent(QPaintEvent *e);
// ----------------------------------------------------------------------
// Private methods
// ----------------------------------------------------------------------
private:
void updateData();
void draw(QPainter &, const Graph *);
void updateRanges();
void updateAxisLabels();
// ----------------------------------------------------------------------
// Private members
// ----------------------------------------------------------------------
private:
enum Mode {
Amplitude,
Power,
Phase
};
QString _exportBasename;
double _freqNyquist;
Math::ComplexArray _spec;
Processing::ResponsePtr _resp;
Mode _mode;
Graph _graphPowerSpectrum;
Graph _graphResponseCorrectedPowerSpectrum;
Graph _graphResponsePowerSpectrum;
DataY _powerSpectrum;
DataY _responseCorrectedPowerSpectrum;
DataY _responsePowerSpectrum;
int _margin;
Axis _xAxis;
Axis _yAxis;
Axis _yAxis2;
};
}
}
#endif