454 lines
14 KiB
C++
454 lines
14 KiB
C++
#include "TwoDSpectralCompliance.h"
|
||
#include "ui_TwoDSpectralCompliance.h"
|
||
#include "CustomQwtPlot.h"
|
||
#include <QwtPlotCanvas>
|
||
#include <QwtText>
|
||
#include <QwtLegend>
|
||
#include <QwtPlotSpectrogram>
|
||
#include <QwtLinearColorMap>
|
||
#include <QwtMatrixRasterData>
|
||
#include <QHBoxLayout>
|
||
#include <QVBoxLayout>
|
||
#include <QFile>
|
||
#include <QTextStream>
|
||
#include <QDebug>
|
||
#include <cmath>
|
||
#include <QwtInterval>
|
||
#include <qwt_matrix_raster_data.h>
|
||
#include <algorithm>
|
||
#include <QFormLayout>
|
||
#include <QLabel>
|
||
#include <QPushButton>
|
||
#include <QLineEdit>
|
||
#include <QMouseEvent>
|
||
#include <QSet>
|
||
#include "BusyIndicator.h"
|
||
#include <QThread>
|
||
|
||
class HeatMapColorMap : public QwtLinearColorMap
|
||
{
|
||
public:
|
||
HeatMapColorMap() : QwtLinearColorMap(Qt::blue, Qt::red)
|
||
{
|
||
addColorStop(0.25, Qt::cyan);
|
||
addColorStop(0.5, Qt::green);
|
||
addColorStop(0.75, Qt::yellow);
|
||
addColorStop(0.90, Qt::darkRed);
|
||
}
|
||
};
|
||
|
||
TwoDSpectralCompliance::TwoDSpectralCompliance(QWidget *parent) :
|
||
MeasureAnalysisView(parent),
|
||
ui(new Ui::TwoDSpectralCompliance)
|
||
{
|
||
ui->setupUi(this);
|
||
QHBoxLayout* layout = new QHBoxLayout(this);
|
||
layout->setContentsMargins(0, 0, 0, 0);
|
||
this->_plot = new CustomQwtPlot(this);
|
||
layout->addWidget(this->_plot);
|
||
setupPlot();
|
||
createFloatingInfoWidget();
|
||
_busy_indicator = new BusyIndicator(this);
|
||
|
||
}
|
||
|
||
TwoDSpectralCompliance::~TwoDSpectralCompliance()
|
||
{
|
||
delete ui;
|
||
}
|
||
|
||
void TwoDSpectralCompliance::InitViewWorkspace(const QString &project_name)
|
||
{
|
||
Q_UNUSED(project_name);
|
||
}
|
||
|
||
void TwoDSpectralCompliance::SetAnalyzeDataFilename(const QMap<QString, QVariant> &data_files_set)
|
||
{
|
||
if (data_files_set.isEmpty())
|
||
return;
|
||
QString csvFile = data_files_set.first().toString();
|
||
if (csvFile.isEmpty())
|
||
return;
|
||
auto functionToRun = [this, csvFile](){
|
||
_busy_indicator->Start();
|
||
readCsv(csvFile);
|
||
generateSurfaceData();
|
||
updateSpectrogram();
|
||
_busy_indicator->Stop();
|
||
};
|
||
QThread* load_data_thread = QThread::create(functionToRun);
|
||
load_data_thread->start();
|
||
}
|
||
|
||
|
||
void TwoDSpectralCompliance::setupPlot()
|
||
{
|
||
_plot->setCanvasBackground(Qt::white);
|
||
QwtPlotCanvas* canvas = qobject_cast<QwtPlotCanvas*>(_plot->canvas());
|
||
if (canvas)
|
||
canvas->setFrameStyle(QFrame::NoFrame);
|
||
|
||
QFont font = this->font();
|
||
font.setBold(false);
|
||
QwtText energy_label = QStringLiteral(u"初级粒子能量");
|
||
energy_label.setFont(font);
|
||
QwtText count_label = QStringLiteral(u"次级粒子能量");
|
||
count_label.setFont(font);
|
||
_plot->setAxisTitle(QwtPlot::xBottom, energy_label);
|
||
_plot->setAxisTitle(QwtPlot::yLeft, count_label);
|
||
|
||
_plot->setAxisAutoScale(QwtPlot::xBottom, true);
|
||
_plot->setAxisAutoScale(QwtPlot::yLeft, true);
|
||
|
||
_plot->enableAxis(QwtPlot::xBottom);
|
||
_plot->enableAxis(QwtPlot::yLeft);
|
||
|
||
QwtLegend* legend = new QwtLegend();
|
||
legend->setDefaultItemMode(QwtLegendData::ReadOnly);
|
||
_plot->insertLegend(legend, QwtPlot::RightLegend);
|
||
|
||
_plot->SetAxisDragScale(QwtPlot::xBottom, true);
|
||
_plot->SetAxisDragScale(QwtPlot::yLeft, true);
|
||
|
||
_spectrogram = new QwtPlotSpectrogram();
|
||
_spectrogram->setColorMap(new HeatMapColorMap());
|
||
_spectrogram->attach(_plot);
|
||
}
|
||
|
||
void TwoDSpectralCompliance::readCsv(const QString &filename)
|
||
{
|
||
m_rawData.clear();
|
||
QFile file(filename);
|
||
if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
qWarning() << QStringLiteral(u"无法打开文件:") << filename;
|
||
return;
|
||
}
|
||
|
||
QTextStream stream(&file);
|
||
QString header = stream.readLine();
|
||
Q_UNUSED(header);
|
||
|
||
while (!stream.atEnd()) {
|
||
QString line = stream.readLine();
|
||
if (line.isEmpty())
|
||
continue;
|
||
|
||
QStringList fields = line.split(',');
|
||
if (fields.size() < 5)
|
||
continue;
|
||
|
||
EventData data;
|
||
data.eventId = fields[0].toInt();
|
||
data.board = fields[1].toInt();
|
||
data.channel = fields[2].toInt();
|
||
data.energy = fields[3].toDouble();
|
||
data.timeCounter = fields[4].toULongLong();
|
||
m_rawData.append(data);
|
||
}
|
||
file.close();
|
||
}
|
||
|
||
void TwoDSpectralCompliance::generateSurfaceData()
|
||
{
|
||
m_surfaceData.clear();
|
||
if (m_rawData.isEmpty())
|
||
return;
|
||
|
||
QMap<int, QVector<EventData>> eventMap;
|
||
for (const auto& ev : m_rawData) {
|
||
eventMap[ev.eventId].append(ev);
|
||
}
|
||
|
||
QMap<QPair<float, float>, int> countMap;
|
||
|
||
for (auto it = eventMap.begin(); it != eventMap.end(); ++it) {
|
||
const auto& events = it.value();
|
||
if (events.isEmpty())
|
||
continue;
|
||
|
||
float primaryEnergy = events[0].energy;
|
||
float secondarySum = 0.0f;
|
||
for (int i = 1; i < events.size(); ++i) {
|
||
secondarySum += events[i].energy;
|
||
}
|
||
|
||
auto key = qMakePair(primaryEnergy, secondarySum);
|
||
countMap[key]++;
|
||
}
|
||
|
||
for (auto it = countMap.begin(); it != countMap.end(); ++it) {
|
||
TwoSurfacePoint point;
|
||
point.primaryEnergy = it.key().first;
|
||
point.secondaryEnergySum = it.key().second;
|
||
point.count = it.value();
|
||
m_surfaceData.append(point);
|
||
}
|
||
|
||
}
|
||
|
||
void TwoDSpectralCompliance::updateSpectrogram()
|
||
{
|
||
if (m_surfaceData.isEmpty()) {
|
||
_spectrogram->setData(new QwtMatrixRasterData());
|
||
_plot->replot();
|
||
return;
|
||
}
|
||
|
||
float minPrim = 1e9f, maxPrim = -1e9f;
|
||
float minSec = 1e9f, maxSec = -1e9f;
|
||
int maxCount = 0;
|
||
|
||
for (const auto& pt : m_surfaceData) {
|
||
minPrim = std::min(minPrim, pt.primaryEnergy);
|
||
maxPrim = std::max(maxPrim, pt.primaryEnergy);
|
||
minSec = std::min(minSec, pt.secondaryEnergySum);
|
||
maxSec = std::max(maxSec, pt.secondaryEnergySum);
|
||
maxCount = std::max(maxCount, pt.count);
|
||
}
|
||
|
||
double primRange = maxPrim - minPrim;
|
||
double secRange = maxSec - minSec;
|
||
if (primRange < 1e-6) primRange = 1.0;
|
||
if (secRange < 1e-6) secRange = 1.0;
|
||
double primStart = minPrim - primRange * 0.05;
|
||
double primEnd = maxPrim + primRange * 0.05;
|
||
double secStart = minSec - secRange * 0.05;
|
||
double secEnd = maxSec + secRange * 0.05;
|
||
|
||
const int numCols = 200;
|
||
const int numRows = 200;
|
||
|
||
double stepX = (primEnd - primStart) / numCols;
|
||
double stepY = (secEnd - secStart) / numRows;
|
||
|
||
QVector<double> zValues(numRows * numCols, 0.0);
|
||
|
||
for (const auto& pt : m_surfaceData) {
|
||
int col = static_cast<int>((pt.primaryEnergy - primStart) / stepX);
|
||
int row = static_cast<int>((pt.secondaryEnergySum - secStart) / stepY);
|
||
col = qBound(0, col, numCols - 1);
|
||
row = qBound(0, row, numRows - 1);
|
||
zValues[row * numCols + col] += pt.count;
|
||
}
|
||
|
||
QwtMatrixRasterData* data = new QwtMatrixRasterData();
|
||
data->setInterval(Qt::XAxis, QwtInterval(primStart, primEnd));
|
||
data->setInterval(Qt::YAxis, QwtInterval(secStart, secEnd));
|
||
data->setInterval(Qt::ZAxis, QwtInterval(0, maxCount));
|
||
data->setValueMatrix(zValues, numCols);
|
||
|
||
_spectrogram->setData(data);
|
||
_plot->setAxisScale(QwtPlot::xBottom, primStart, primEnd);
|
||
_plot->setAxisScale(QwtPlot::yLeft, secStart, secEnd);
|
||
_plot->replot();
|
||
}
|
||
void TwoDSpectralCompliance::createFloatingInfoWidget()
|
||
{
|
||
// 悬浮窗口
|
||
m_floatingWidget = new QWidget(this);
|
||
m_floatingWidget->setObjectName("FloatingInfoWidget");
|
||
m_floatingWidget->setStyleSheet(
|
||
"QWidget#FloatingInfoWidget {"
|
||
" background-color: rgba(240, 240, 240, 240);"
|
||
" border: 1px solid gray;"
|
||
" border-radius: 5px;"
|
||
"}"
|
||
);
|
||
m_floatingWidget->setFixedSize(320, 220);
|
||
m_floatingWidget->setAttribute(Qt::WA_TranslucentBackground, false);
|
||
|
||
QVBoxLayout* mainLayout = new QVBoxLayout(m_floatingWidget);
|
||
mainLayout->setSpacing(5);
|
||
mainLayout->setContentsMargins(8, 8, 8, 8);
|
||
|
||
// 标题栏（可拖动，并包含隐藏按钮）
|
||
QWidget* titleBar = new QWidget();
|
||
QHBoxLayout* titleLayout = new QHBoxLayout(titleBar);
|
||
titleLayout->setContentsMargins(0, 0, 0, 0);
|
||
QLabel* titleLabel = new QLabel(QStringLiteral(u"数据统计信息"));
|
||
titleLabel->setStyleSheet("font-weight: bold;");
|
||
titleLayout->addWidget(titleLabel);
|
||
titleLayout->addStretch();
|
||
QPushButton* hideBtn = new QPushButton(QStringLiteral("−"));
|
||
hideBtn->setFixedSize(20, 20);
|
||
hideBtn->setText(QStringLiteral(u"X"));
|
||
hideBtn->setStyleSheet("border: none; font-weight: bold;");
|
||
titleLayout->addWidget(hideBtn);
|
||
mainLayout->addWidget(titleBar);
|
||
|
||
QWidget* contentWidget = new QWidget();
|
||
QFormLayout* formLayout = new QFormLayout(contentWidget);
|
||
formLayout->setSpacing(8);
|
||
formLayout->setLabelAlignment(Qt::AlignRight);
|
||
|
||
m_totalEventsEdit = new QLineEdit;
|
||
m_totalEventsEdit->setReadOnly(true);
|
||
m_totalEventsEdit->setFixedWidth(130);
|
||
|
||
m_totalPointsEdit = new QLineEdit;
|
||
m_totalPointsEdit->setReadOnly(true);
|
||
m_totalPointsEdit->setFixedWidth(130);
|
||
m_maxCountEdit = new QLineEdit;
|
||
m_maxCountEdit->setReadOnly(true);
|
||
m_maxCountEdit->setFixedWidth(130);
|
||
|
||
m_primRangeEdit = new QLineEdit;
|
||
m_primRangeEdit->setReadOnly(true);
|
||
m_primRangeEdit->setFixedWidth(130);
|
||
|
||
m_secRangeEdit = new QLineEdit;
|
||
m_secRangeEdit->setReadOnly(true);
|
||
m_secRangeEdit->setFixedWidth(130);
|
||
|
||
formLayout->addRow(QStringLiteral(u"总事件数:"), m_totalEventsEdit);
|
||
formLayout->addRow(QStringLiteral(u"有效数据点:"), m_totalPointsEdit);
|
||
formLayout->addRow(QStringLiteral(u"最大计数:"), m_maxCountEdit);
|
||
formLayout->addRow(QStringLiteral(u"初级能量范围:"), m_primRangeEdit);
|
||
formLayout->addRow(QStringLiteral(u"次级能量和范围:"), m_secRangeEdit);
|
||
|
||
mainLayout->addWidget(contentWidget);
|
||
mainLayout->addStretch();
|
||
|
||
connect(hideBtn, &QPushButton::clicked, this, &TwoDSpectralCompliance::hideFloatingWidget);
|
||
|
||
titleBar->setCursor(Qt::SizeAllCursor);
|
||
m_floatingWidget->installEventFilter(this);
|
||
|
||
m_showButton = new QPushButton(this);
|
||
m_showButton->setFixedSize(28, 28);
|
||
m_showButton->setToolTip(QStringLiteral(u"显示信息窗口"));
|
||
m_showButton->setStyleSheet(
|
||
"QPushButton {"
|
||
" background-color: rgba(200,200,200,200);"
|
||
" border: 1px solid #888;"
|
||
" border-radius: 14px;"
|
||
" font-weight: bold;"
|
||
"}"
|
||
"QPushButton:hover { background-color: rgba(150,150,150,200); }"
|
||
);
|
||
m_showButton->setText(QStringLiteral(u"⊕"));
|
||
connect(m_showButton, &QPushButton::clicked, this, &TwoDSpectralCompliance::showFloatingWidget);
|
||
|
||
m_floatingWidget->move(10, 10);
|
||
updateShowButtonPosition();
|
||
m_floatingWidget->hide();
|
||
m_showButton->show();
|
||
|
||
// m_floatingWidget->raise();
|
||
updateInfoContent();
|
||
}
|
||
|
||
void TwoDSpectralCompliance::hideFloatingWidget()
|
||
{
|
||
m_floatingWidget->hide();
|
||
m_showButton->show();
|
||
updateShowButtonPosition();
|
||
}
|
||
|
||
void TwoDSpectralCompliance::showFloatingWidget()
|
||
{
|
||
m_floatingWidget->show();
|
||
m_floatingWidget->raise();
|
||
m_showButton->hide();
|
||
updateInfoContent(); // 刷新内容
|
||
}
|
||
|
||
void TwoDSpectralCompliance::updateShowButtonPosition()
|
||
{
|
||
if (m_showButton && m_showButton->isVisible()) {
|
||
m_showButton->move(10, 10);
|
||
}
|
||
}
|
||
|
||
void TwoDSpectralCompliance::updateInfoContent()
|
||
{
|
||
if (!m_totalEventsEdit)
|
||
return;
|
||
|
||
if (m_surfaceData.isEmpty()) {
|
||
m_totalEventsEdit->setText(QStringLiteral(u"无数据"));
|
||
m_totalPointsEdit->setText(QStringLiteral(u"无数据"));
|
||
m_maxCountEdit->setText(QStringLiteral(u"无数据"));
|
||
m_primRangeEdit->setText(QStringLiteral(u"无数据"));
|
||
m_secRangeEdit->setText(QStringLiteral(u"无数据"));
|
||
return;
|
||
}
|
||
|
||
int totalPoints = m_surfaceData.size();
|
||
int maxCount = 0;
|
||
float minPrim = 1e9f, maxPrim = -1e9f;
|
||
float minSec = 1e9f, maxSec = -1e9f;
|
||
for (const auto& pt : m_surfaceData) {
|
||
minPrim = std::min(minPrim, pt.primaryEnergy);
|
||
maxPrim = std::max(maxPrim, pt.primaryEnergy);
|
||
minSec = std::min(minSec, pt.secondaryEnergySum);
|
||
maxSec = std::max(maxSec, pt.secondaryEnergySum);
|
||
maxCount = std::max(maxCount, pt.count);
|
||
}
|
||
|
||
QSet<int> eventIds;
|
||
for (const auto& ev : m_rawData) {
|
||
eventIds.insert(ev.eventId);
|
||
}
|
||
int totalEvents = eventIds.size();
|
||
|
||
m_totalEventsEdit->setText(QString::number(totalEvents));
|
||
m_totalPointsEdit->setText(QString::number(totalPoints));
|
||
m_maxCountEdit->setText(QString::number(maxCount));
|
||
m_primRangeEdit->setText(QString("[%1, %2]").arg(minPrim, 0, 'f', 2).arg(maxPrim, 0, 'f', 2));
|
||
m_secRangeEdit->setText(QString("[%1, %2]").arg(minSec, 0, 'f', 2).arg(maxSec, 0, 'f', 2));
|
||
}
|
||
|
||
void TwoDSpectralCompliance::resizeEvent(QResizeEvent *event)
|
||
{
|
||
QWidget::resizeEvent(event);
|
||
updateShowButtonPosition();
|
||
if (m_floatingWidget && m_floatingWidget->isVisible()) {
|
||
// 限制悬浮窗口不超出父窗口边界
|
||
QRect rect = m_floatingWidget->geometry();
|
||
int newX = qBound(0, rect.x(), width() - rect.width());
|
||
int newY = qBound(0, rect.y(), height() - rect.height());
|
||
if (newX != rect.x() || newY != rect.y())
|
||
m_floatingWidget->move(newX, newY);
|
||
m_floatingWidget->raise();
|
||
}
|
||
}
|
||
|
||
bool TwoDSpectralCompliance::eventFilter(QObject *obj, QEvent *event)
|
||
{
|
||
if (obj == m_floatingWidget) {
|
||
if (event->type() == QEvent::MouseButtonPress) {
|
||
QMouseEvent *mouseEvent = static_cast<QMouseEvent*>(event);
|
||
if (mouseEvent->button() == Qt::LeftButton) {
|
||
m_dragging = true;
|
||
m_dragPosition = mouseEvent->globalPos() - m_floatingWidget->frameGeometry().topLeft();
|
||
event->accept();
|
||
return true;
|
||
}
|
||
} else if (event->type() == QEvent::MouseMove) {
|
||
if (m_dragging) {
|
||
QMouseEvent *mouseEvent = static_cast<QMouseEvent*>(event);
|
||
m_floatingWidget->move(mouseEvent->globalPos() - m_dragPosition);
|
||
event->accept();
|
||
return true;
|
||
}
|
||
} else if (event->type() == QEvent::MouseButtonRelease) {
|
||
if (m_dragging) {
|
||
m_dragging = false;
|
||
event->accept();
|
||
return true;
|
||
}
|
||
}
|
||
}
|
||
return QWidget::eventFilter(obj, event);
|
||
}
|
||
|
||
void TwoDSpectralCompliance::showEvent(QShowEvent *e)
|
||
{
|
||
Q_UNUSED(e);
|
||
if (_busy_indicator) {
|
||
_busy_indicator->setGeometry(this->rect());
|
||
this->update();
|
||
}
|
||
}
|