221 lines
8.5 KiB
C++
221 lines
8.5 KiB
C++
#include "EnergyCountPlotView.h"
|
|
#include "CustomQwtPlot.h"
|
|
#include "csv.h"
|
|
#include <GlobalDefine.h>
|
|
#include <QHBoxLayout>
|
|
#include <QMenu>
|
|
#include <QwtLegend>
|
|
#include <QwtPlotCanvas>
|
|
#include <QwtPlotCurve>
|
|
#include <QwtPlotMarker>
|
|
#include <QwtScaleMap>
|
|
#include <QwtText>
|
|
#include <QDialog>
|
|
#include <QPushButton>
|
|
#include <QCheckBox>
|
|
|
|
|
|
EnergyCountPlotView::EnergyCountPlotView(QWidget *parent)
|
|
: MeasureAnalysisView { parent }
|
|
{
|
|
this->setViewType(PlotFrame);
|
|
|
|
QHBoxLayout* layout = new QHBoxLayout(this);
|
|
this->_plot = new CustomQwtPlot(this);
|
|
layout->addWidget(this->_plot);
|
|
setupPlot();
|
|
this->_menu = new QMenu(this);
|
|
setupMenu();
|
|
}
|
|
|
|
EnergyCountPlotView::~EnergyCountPlotView()
|
|
{
|
|
LOG_DEBUG(QStringLiteral(u"%1析构.").arg(this->GetViewName()));
|
|
}
|
|
|
|
void EnergyCountPlotView::InitViewWorkspace(const QString &project_name)
|
|
{
|
|
Q_UNUSED(project_name);
|
|
}
|
|
|
|
void EnergyCountPlotView::SetAnalyzeDataFilename(const QMap<QString, QVariant> &data_files_set)
|
|
{
|
|
QStringList ch_count_data_name = data_files_set.keys();
|
|
std::sort(ch_count_data_name.begin(), ch_count_data_name.end(), [](const QString& a, const QString& b) {
|
|
int num_a = ExtractNumberFromString(a);
|
|
int num_b = ExtractNumberFromString(b);
|
|
return num_a < num_b;
|
|
});
|
|
for (const QString& ch_count_data_name : ch_count_data_name) {
|
|
if (ch_count_data_name.contains(ch_count_data_name)) {
|
|
const QString ch_count_data_filename = data_files_set[ch_count_data_name].toString();
|
|
loadDataFromFile(ch_count_data_name, ch_count_data_filename);
|
|
}
|
|
}
|
|
}
|
|
|
|
void EnergyCountPlotView::setupPlot()
|
|
{
|
|
_plot->setTitle(QString(QStringLiteral(u"能量计数谱")));
|
|
|
|
_plot->setCanvasBackground(Qt::white);
|
|
QwtPlotCanvas* canvas = qobject_cast<QwtPlotCanvas*>(_plot->canvas());
|
|
canvas->setFrameStyle(QFrame::NoFrame);
|
|
|
|
_plot->setAxisTitle(QwtPlot::xBottom, QString(QStringLiteral(u"能量")));
|
|
_plot->setAxisTitle(QwtPlot::yLeft, QString(QStringLiteral(u"计数")));
|
|
// set axis auto scale
|
|
_plot->setAxisAutoScale(QwtPlot::xBottom, true);
|
|
_plot->setAxisAutoScale(QwtPlot::yLeft, true);
|
|
// 启用网格线
|
|
_plot->enableAxis(QwtPlot::xBottom);
|
|
_plot->enableAxis(QwtPlot::yLeft);
|
|
|
|
// 设置QWT图例
|
|
QwtLegend* legend = new QwtLegend();
|
|
legend->setDefaultItemMode(QwtLegendData::ReadOnly);
|
|
_plot->insertLegend(legend, QwtPlot::RightLegend);
|
|
|
|
_plot->SetXaxisDragScale(true);
|
|
_data_selector = new CustomQwtPlotXaxisSelector(_plot->canvas());
|
|
_data_selector->setEnabled(false);
|
|
}
|
|
|
|
void EnergyCountPlotView::setupMenu()
|
|
{
|
|
this->setContextMenuPolicy(Qt::CustomContextMenu);
|
|
connect(this, &EnergyCountPlotView::customContextMenuRequested, [this](const QPoint& pos) {
|
|
this->_menu->exec(this->mapToGlobal(pos));
|
|
});
|
|
QAction* action_plot_reset = this->_menu->addAction(QStringLiteral(u"还原"));
|
|
action_plot_reset->setObjectName("plot_reset");
|
|
connect(action_plot_reset, &QAction::triggered, [this]() {
|
|
this->_plot->ResetPlot();
|
|
});
|
|
this->_menu->addSeparator();
|
|
QAction* action_set_curve_show = this->_menu->addAction(QStringLiteral(u"选择曲线"));
|
|
action_set_curve_show->setObjectName("curve_show_setting");
|
|
connect(action_set_curve_show, &QAction::triggered, this, &EnergyCountPlotView::onActionCurveShowSetting);
|
|
QAction* action_plot_config = this->_menu->addAction(QStringLiteral(u"图表配置"));
|
|
action_plot_config->setObjectName("plot_config");
|
|
connect(action_plot_config, &QAction::triggered, this, &EnergyCountPlotView::onActionPlotConfigure);
|
|
}
|
|
|
|
void EnergyCountPlotView::loadDataFromFile(const QString &data_name, const QString &filename)
|
|
{
|
|
std::string address_str = QString(QStringLiteral(u"能量(KeV)")).toStdString();
|
|
std::string count_str = QString(QStringLiteral(u"计数")).toStdString();
|
|
|
|
io::CSVReader<
|
|
2,
|
|
io::trim_chars<' ', '\t'>,
|
|
io::double_quote_escape<',', '"'>,
|
|
io::throw_on_overflow,
|
|
io::empty_line_comment>
|
|
reader(QStrToSysPath(filename));
|
|
reader.read_header(io::ignore_extra_column, address_str, count_str);
|
|
|
|
double energy;
|
|
unsigned long long energy_count;
|
|
QVector<double> x, y;
|
|
while (reader.read_row(energy, energy_count)) {
|
|
x.push_back(energy);
|
|
y.push_back(energy_count);
|
|
}
|
|
// 绘制曲线
|
|
QwtPlotCurve* curve = new QwtPlotCurve(data_name);
|
|
curve->setSamples(x, y);
|
|
_plot->AddCurve(curve);
|
|
}
|
|
|
|
void EnergyCountPlotView::onActionCurveShowSetting()
|
|
{
|
|
if (!_curve_show_setting_dlg) {
|
|
_curve_show_setting_dlg = new QDialog(this, Qt::Dialog | Qt::WindowCloseButtonHint);
|
|
_curve_show_setting_dlg->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);
|
|
_curve_show_setting_dlg->setSizeGripEnabled(false);
|
|
_curve_show_setting_dlg->setWindowTitle(QString(QStringLiteral(u"选择%1曲线显示").arg(this->_plot->title().text())));
|
|
_curve_show_setting_dlg->setWindowModality(Qt::WindowModal);
|
|
_curve_show_setting_dlg->setModal(false);
|
|
QVBoxLayout* layout = new QVBoxLayout(_curve_show_setting_dlg);
|
|
|
|
const QString& all_ch_text = QStringLiteral(u"全通道");
|
|
QStringList list_ch_names;
|
|
for (QwtPlotCurve* curve : this->_plot->GetCurveList()) {
|
|
const QString& every_ch_text = curve->title().text();
|
|
if ( all_ch_text != every_ch_text ) {
|
|
list_ch_names.append(every_ch_text);
|
|
}
|
|
}
|
|
std::sort(list_ch_names.begin(), list_ch_names.end(), [](const QString& a, const QString& b) {
|
|
int num_a = ExtractNumberFromString(a);
|
|
int num_b = ExtractNumberFromString(b);
|
|
return num_a < num_b;
|
|
});
|
|
// 自动计算多列排布
|
|
int num_columns = std::sqrt(this->_plot->GetCurveList().size());
|
|
if (num_columns == 0)
|
|
num_columns = 1;
|
|
QVBoxLayout* checkbox_layout = new QVBoxLayout();
|
|
QHBoxLayout* checkbox_column_layout = new QHBoxLayout();
|
|
QMap<QwtPlotCurve*, QCheckBox*> curve_checkbox_map;
|
|
auto addCurveToLayout = [&](const QString& curve_name){
|
|
QwtPlotCurve* curve = this->_plot->GetCurve(curve_name);
|
|
QCheckBox* check_box = new QCheckBox(curve->title().text());
|
|
check_box->setChecked(curve->isVisible());
|
|
checkbox_column_layout->addWidget(check_box);
|
|
connect(check_box, &QCheckBox::stateChanged, [curve](int state) {
|
|
curve->setVisible(state == Qt::Checked);
|
|
curve->plot()->replot();
|
|
});
|
|
curve_checkbox_map[curve] = check_box;
|
|
if (checkbox_column_layout->count() >= num_columns) {
|
|
checkbox_layout->addLayout(checkbox_column_layout);
|
|
checkbox_column_layout = new QHBoxLayout();
|
|
}
|
|
};
|
|
addCurveToLayout(all_ch_text);
|
|
for (const QString& ch_name : list_ch_names) {
|
|
addCurveToLayout(ch_name);
|
|
}
|
|
if (checkbox_column_layout->count() < num_columns) {
|
|
checkbox_column_layout->addStretch();
|
|
}
|
|
checkbox_layout->addLayout(checkbox_column_layout);
|
|
// 全选和反选
|
|
auto curveCheckboxUpdate = [this, curve_checkbox_map]() {
|
|
for (QwtPlotCurve* curve : this->_plot->GetCurveList()) {
|
|
curve_checkbox_map[curve]->setChecked(curve->isVisible());
|
|
}
|
|
};
|
|
QHBoxLayout* button_layout = new QHBoxLayout();
|
|
QPushButton* btn_all_select = new QPushButton(QString(QStringLiteral(u"全选")));
|
|
connect(btn_all_select, &QPushButton::clicked, [this, curveCheckboxUpdate]() {
|
|
for (QwtPlotCurve* curve : this->_plot->GetCurveList()) {
|
|
curve->setVisible(true);
|
|
}
|
|
curveCheckboxUpdate();
|
|
this->_plot->replot();
|
|
});
|
|
button_layout->addWidget(btn_all_select);
|
|
QPushButton* btn_reserve_select = new QPushButton(QString(QStringLiteral(u"反选")));
|
|
connect(btn_reserve_select, &QPushButton::clicked, [this, curveCheckboxUpdate]() {
|
|
for (QwtPlotCurve* curve : this->_plot->GetCurveList()) {
|
|
curve->setVisible(!curve->isVisible());
|
|
}
|
|
curveCheckboxUpdate();
|
|
this->_plot->replot();
|
|
});
|
|
button_layout->addWidget(btn_reserve_select);
|
|
|
|
layout->addLayout(button_layout);
|
|
layout->addLayout(checkbox_layout);
|
|
}
|
|
_curve_show_setting_dlg->show();
|
|
}
|
|
|
|
void EnergyCountPlotView::onActionPlotConfigure()
|
|
{
|
|
|
|
}
|