xuhai_cpp/EnergySpectrumAnalyer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增道址计数谱视图显示，新增道址计数数据处理

Browse Source
This commit is contained in:
anxinglong 2026-06-15 18:00:16 +08:00
parent 2e6e78ada3
commit 9eeb09f18a
4 changed files with 313 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/EnergyCountPlotView/EnergyCountPlotView.cpp
View File

@ -143,6 +143,27 @@ void EnergyCountPlotView::loadDataFromFile(const QString &data_name, const QStri
_plot->AddCurve(curve);
}
void EnergyCountPlotView::updateView(const QMap<QString, QVariant>& data_files_set)
{
// 清除现有所有曲线
for (QwtPlotCurve* curve : this->_plot->GetCurveList()) {
curve->detach();
delete curve;
}
_plot->clearCurve();
// 重新加载数据
if (!data_files_set.isEmpty()) {
SetAnalyzeDataFilename(data_files_set);
}
// 清除标记并重绘
this->_plot->CleanMarkers();
this->_plot->CleanZoneItems();
this->_plot->ResetPlot();
this->_plot->replot();
}
void EnergyCountPlotView::onActionCurveShowSetting()
{
if (!_curve_show_setting_dlg) {

1
src/EnergyCountPlotView/EnergyCountPlotView.h
View File

@ -20,6 +20,7 @@ public:
virtual void InitViewWorkspace(const QString& project_name) override final;
virtual void SetAnalyzeDataFilename(const QMap<QString, QVariant>& data_files_set);
void updateView(const QMap<QString, QVariant>& data_files_set);
private:
void setupPlot();
void setupMenu();

290
src/MainWindow.cpp
View File

@ -34,6 +34,7 @@
#include "MeasureAnalysisDataTableView.h"
#include "DataCalcProcess/GaussPolyCoe.h"
#include "EnergyScaleDataModel.h"
#include "EnergyCountPlotView.h"
#include "ParticleCountPlotView.h"
#include <array>
#include "csv.h"
@ -104,7 +105,10 @@ MainWindow::MainWindow(QWidget* parent)
_measure_client = new MeasureClient;
m_AddressCountTimer = new QTimer(this);
m_AddressCountTimer->setInterval(10000);//10秒刷新一次
m_EnergyCountTimer = new QTimer(this);
m_EnergyCountTimer->setInterval(10000);//10秒刷新一次
connect(m_AddressCountTimer, &QTimer::timeout, this, &MainWindow::on_AddressCountTimer);
connect(m_EnergyCountTimer, &QTimer::timeout, this, &MainWindow::on_EnergyCountTimer);
connect(_measure_client, &MeasureClient::getDeviceListResult, this, &MainWindow::onGetDeviceListResult);
connect(_measure_client, &MeasureClient::startMeasureResult, this, &MainWindow::onStartMeasureResult);
@ -598,7 +602,7 @@ void MainWindow::onGetDeviceListResult(bool success, const QString &message, con
void MainWindow::onErrorOccurred(const QString &error_string)
{
LOG_INFO(QStringLiteral(u"错误: %1").arg(error_string));
// LOG_INFO(QStringLiteral(u"错误: %1").arg(error_string));
}
void MainWindow::onRunningInfo(const QString &run_info)
@ -619,6 +623,8 @@ void MainWindow::onGvfData(const QByteArray &data)
changeEnergyCountData(particles);
//处理道址计数谱
changeAddressCountView(particles);
//处理能量计数谱
changeEnergyCountView(particles);
}
//处理粒子数据
@ -690,11 +696,11 @@ void MainWindow::updataTable()
MeasureAnalysisDataTableView* table = dynamic_cast<MeasureAnalysisDataTableView*>(view);
table->AppendRow();
};break;
case AnalysisType::EnergyCountData:
{
MeasureAnalysisDataTableView* table = dynamic_cast<MeasureAnalysisDataTableView*>(view);
table->AppendRow();
};break;
// case AnalysisType::EnergyCountData:
// {
// MeasureAnalysisDataTableView* table = dynamic_cast<MeasureAnalysisDataTableView*>(view);
// table->AppendRow();
// };break;
}
}
}
@ -947,7 +953,216 @@ void MainWindow::changeParticleEnergyData(QList<ParticleData> &dataList)
//处理能量计数
void MainWindow::changeEnergyCountData(QList<ParticleData> &dataList)
{
if (dataList.isEmpty()) {
return;
}
MeasureAnalysisProjectModel* project_model = ProjectList::Instance()->GetCurrentProjectModel();
if (!project_model) {
LOG_ERROR(QStringLiteral(u"当前没有打开的测量项目，无法处理能量计数数据"));
return;
}
const QString& project_name = project_model->GetProjectName();
// 加载能量刻度数据
EnergyScaleDataModel energy_scale_data_model(project_model->GetEnergyScaleFilename());
if (!energy_scale_data_model.LoadData()) {
LOG_WARN(QStringLiteral(u"[%1]加载能量刻度文件失败，跳过本次能量计数处理").arg(project_name));
return;
}
if (!energy_scale_data_model.IsValid()) {
LOG_WARN(QStringLiteral(u"[%1]能量刻度数据无效，请检查刻度配置").arg(project_name));
return;
}
// 计算本次数据的增量计数
QHash<uint, QHash<double, unsigned long long>> deltaCounts; // 通道号 -> 能量值 -> 增量
QHash<double, unsigned long long> deltaAllChannelCounts; // 全通道增量
int totalParticles = 0;
int successCount = 0;
int skipCount = 0;
for (const auto &particle : dataList) {
int channel_num = (particle.boardId) * 4 + (particle.channelId + 1);
const QString& channel_name = QStringLiteral(u"通道%1").arg(channel_num);
// 获取该通道的能量刻度系数
std::vector<double> coeffs = energy_scale_data_model.GetEnergyFitResultCoeffs(channel_name);
if (coeffs.empty()) {
skipCount++;
continue; // 该通道未配置能量刻度，跳过
}
// 道址转能量（保留1位小数，作为分组键，避免浮点数精度问题）
double energy = GaussPolyCoe::Predict(coeffs, particle.address);
double energy_key = qRound(energy * 10.0) / 10.0; // 精确到0.1KeV
deltaCounts[channel_num][energy_key]++;
deltaAllChannelCounts[energy_key]++;
totalParticles++;
successCount++;
}
if (deltaCounts.isEmpty()) {
if (skipCount > 0) {
LOG_DEBUG(QStringLiteral(u"本次%1个粒子全部因无能量刻度配置而跳过").arg(skipCount));
}
return;
}
const QString& energy_count_dir = QDir(project_model->GetProjectDir()).filePath(QStringLiteral(u"通道能量计数"));
QDir energy_count_output_dir(energy_count_dir);
if (!energy_count_output_dir.exists()) {
if (!energy_count_output_dir.mkpath(energy_count_dir)) {
LOG_ERROR(QStringLiteral(u"无法创建通道能量计数目录: %1").arg(energy_count_dir));
return;
}
}
QMutexLocker locker(&m_energyCountMutex);
bool hasError = false;
QHash<uint, QString> newChannelFiles; // 本次新增的通道文件
QMap<QString, QMap<QString, QStandardItem *>> all_nodes = ProjectList::Instance()->getProjectNodeItems();
QMap<QString, QStandardItem *> project_nodes = all_nodes[project_name];
for (auto channelIt = deltaCounts.constBegin(); channelIt != deltaCounts.constEnd(); ++channelIt) {
uint channel_num = channelIt.key();
const auto& energyDeltas = channelIt.value();
auto& channelCounts = channel_energy_counts[channel_num];
for (auto energyIt = energyDeltas.constBegin(); energyIt != energyDeltas.constEnd(); ++energyIt) {
double energy_key = energyIt.key();
unsigned long long delta = energyIt.value();
channelCounts[energy_key] += delta;
}
QString count_data_filename;
if (energy_count_filename_list.contains(channel_num)) {
count_data_filename = energy_count_filename_list[channel_num];
} else {
count_data_filename = energy_count_output_dir.filePath(QStringLiteral(u"通道%1能量计数.csv").arg(channel_num));
project_model->SetChannelEnergyCountDataFilename(channel_num, count_data_filename);
energy_count_filename_list.insert(channel_num, count_data_filename);
newChannelFiles.insert(channel_num, count_data_filename);
}
QFile outFile(count_data_filename);
if (!outFile.open(QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate)) {
LOG_ERROR(QStringLiteral(u"无法打开通道%1的能量计数文件: %2，错误: %3")
.arg(channel_num)
.arg(count_data_filename)
.arg(outFile.errorString()));
hasError = true;
continue;
}
QTextStream out(&outFile);
out.setCodec("UTF-8");
// 写入表头
out << QStringLiteral(u"能量(KeV),计数\n");
// 按能量值排序后输出
QList<double> sortedEnergies = channelCounts.keys();
std::sort(sortedEnergies.begin(), sortedEnergies.end());
QString csvBuffer;
csvBuffer.reserve(sortedEnergies.size() * 40);
for (double energy : sortedEnergies) {
unsigned long long total = channelCounts[energy];
csvBuffer += QString("%1,%2\n").arg(energy, 0, 'f', 1).arg(total);
}
out << csvBuffer;
out.flush(); // 确保数据立即写入磁盘
outFile.close();
}
if (!deltaAllChannelCounts.isEmpty()) {
// 更新内存中的全通道总计数
for (auto energyIt = deltaAllChannelCounts.constBegin(); energyIt != deltaAllChannelCounts.constEnd(); ++energyIt) {
double bin_energy = energyIt.key();
unsigned long long delta = energyIt.value();
all_channel_energy_counts[bin_energy] += delta;
}
// 全通道输出文件路径
QString all_channel_energy_count_filename = energy_count_output_dir.filePath(QStringLiteral(u"全通道.csv"));
QFile allChannelFile(all_channel_energy_count_filename);
if (allChannelFile.open(QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate)) {
QTextStream outAll(&allChannelFile);
outAll.setCodec("UTF-8");
// 写入表头
outAll << QStringLiteral(u"能量(KeV),计数\n");
// 按能量值升序排序
QList<double> sortedAllEnergies = all_channel_energy_counts.keys();
std::sort(sortedAllEnergies.begin(), sortedAllEnergies.end());
QString allCsvBuffer;
allCsvBuffer.reserve(sortedAllEnergies.size() * 40);
for (double bin_energy : sortedAllEnergies) {
unsigned long long total = all_channel_energy_counts[bin_energy];
allCsvBuffer += QString("%1,%2\n").arg(bin_energy, 0, 'f', 3).arg(total);
}
outAll << allCsvBuffer;
outAll.flush();
allChannelFile.close();
// 保存全通道文件路径到项目模型
project_model->SetAllChannelEnergyTotalCountDataFilename(all_channel_energy_count_filename);
} else {
LOG_ERROR(QStringLiteral(u"无法打开全通道能量计数文件: %1，错误: %2")
.arg(all_channel_energy_count_filename)
.arg(allChannelFile.errorString()));
hasError = true;
}
}
// 更新项目树节点状态
if (!newChannelFiles.isEmpty() || !hasError) {
for (auto it = newChannelFiles.constBegin(); it != newChannelFiles.constEnd(); ++it) {
project_model->SetChannelEnergyCountDataFilename(it.key(), it.value());
}
const QString& energy_count_item_name = QStringLiteral(u"能量计数");
if (project_nodes.contains(energy_count_item_name)) {
auto energy_count_item = project_nodes[energy_count_item_name];
bool status_ok = !energy_count_filename_list.isEmpty();
QString status = status_ok ? QStringLiteral(u"有效") : QStringLiteral(u"无效");
ProjectList::Instance()->SetNodeStatus(energy_count_item, status, status_ok);
for (auto it = newChannelFiles.constBegin(); it != newChannelFiles.constEnd(); ++it) {
uint ch_num = it.key();
QString item_name = QStringLiteral(u"通道%1能量计数").arg(ch_num);
if (project_nodes.contains(item_name)) {
continue; // 节点已存在，跳过
}
project_model->SaveProjectModel();
const QVariant& analys_type = QVariant::fromValue(AnalysisType::EnergyCountData);
QStandardItem* node_item = ProjectList::Instance()->AddChildNode(
energy_count_item, item_name, status, analys_type, true, status_ok);
node_item->setData(project_name, Qt::UserRole + 2);
node_item->setData(ch_num, Qt::UserRole + 3);
project_nodes[item_name] = node_item;
}
}
}
// 更新数据表格视图
// updataTable();
if (hasError) {
LOG_WARN(QStringLiteral(u"部分通道的能量计数数据写入失败，请检查磁盘空间和文件权限"));
}
}
//处理道址计数谱
void MainWindow::changeAddressCountView(QList<ParticleData> &dataList)
@ -955,7 +1170,6 @@ void MainWindow::changeAddressCountView(QList<ParticleData> &dataList)
if (dataList.isEmpty()) {
return;
}
MeasureAnalysisProjectModel* pro_model = ProjectList::Instance()->GetCurrentProjectModel();
if (!pro_model) {
LOG_ERROR(QStringLiteral(u"当前没有打开的测量项目，无法处理能谱数据"));
@ -975,9 +1189,32 @@ void MainWindow::changeAddressCountView(QList<ParticleData> &dataList)
}
}
void MainWindow::changeEnergyCountView(QList<ParticleData> &dataList)
{
if (dataList.isEmpty()) {
return;
}
MeasureAnalysisProjectModel* pro_model = ProjectList::Instance()->GetCurrentProjectModel();
if (!pro_model) {
LOG_ERROR(QStringLiteral(u"当前没有打开的测量项目，无法处理能谱数据"));
return;
}
bool status_ok = !pro_model->GetChannelAddressCountDataFilenameList().isEmpty();
QString status = status_ok ? QStringLiteral(u"有效") : QStringLiteral(u"无效");
QString item_name = QStringLiteral(u"能量计数谱");
QStandardItem * particleData = nodeMap[item_name];
//获取能量计数谱状态
bool bStatus = ProjectList::Instance()->GetNodeStatus(particleData);
if(!bStatus)
{
ProjectList::Instance()->SetNodeStatus(particleData,status,true);
m_EnergyCountTimer->start();
}
}
void MainWindow::on_AddressCountTimer()
{
QMap<QString, QVariant> data_files_set;
//获取道址计数谱
auto dockList = _dock_manager->dockWidgetsMap().values();
for(auto dock : dockList)
@ -986,6 +1223,7 @@ void MainWindow::on_AddressCountTimer()
if(!view) continue;
if(view->GetAnalyzeType() == AnalysisType::AddressCountSpectrumView)
{
QMap<QString, QVariant> data_files_set;
MeasureAnalysisProjectModel* project_model = ProjectList::Instance()->GetCurrentProjectModel();
if (project_model) {
auto file_name_list = project_model->GetChannelAddressCountDataFilenameList();
@ -993,7 +1231,7 @@ void MainWindow::on_AddressCountTimer()
auto ch_num_list = file_name_list.keys();
for(auto ch_num : ch_num_list) {
auto file_name = file_name_list[ch_num];
if ( !file_name.isEmpty() ) {
if ( !file_name.isEmpty()) {
data_files_set[QStringLiteral(u"通道%1").arg(ch_num)] = file_name;
}
}
@ -1004,6 +1242,40 @@ void MainWindow::on_AddressCountTimer()
}
}
void MainWindow::on_EnergyCountTimer()
{
//获取道址计数谱
auto dockList = _dock_manager->dockWidgetsMap().values();
for(auto dock : dockList)
{
EnergyCountPlotView* view = dynamic_cast<EnergyCountPlotView*>(dock->widget());
if(!view) continue;
if(view->GetAnalyzeType() == AnalysisType::EnergyCountSpectrumView)
{
QMap<QString, QVariant> data_files_set;
MeasureAnalysisProjectModel* project_model = ProjectList::Instance()->GetCurrentProjectModel();
if (project_model) {
auto all_ch_energy_count_file_name = project_model->GetAllChannelEnergyTotalCountDataFilename();
if (!all_ch_energy_count_file_name.isEmpty()) {
data_files_set[QStringLiteral(u"全通道")] = all_ch_energy_count_file_name;
}
auto file_name_list = project_model->GetChannelEnergyCountDataFilenameList();
if ( !file_name_list.isEmpty() ) {
auto ch_num_list = file_name_list.keys();
for(auto ch_num : ch_num_list) {
auto file_name = file_name_list[ch_num];
if ( !file_name.isEmpty()) {
data_files_set[QStringLiteral(u"通道%1").arg(ch_num)] = file_name;
}
}
}
}
view->updateView(data_files_set);
}
}
}
void MainWindow::on_action_stop_measure_triggered()
{
const QString& device_guid = deviceList.at(0);

13
src/MainWindow.h
View File

@ -68,6 +68,8 @@ private:
void changeEnergyCountData(QList<ParticleData> &dataList);
//处理道址计数谱
void changeAddressCountView(QList<ParticleData> &dataList);
//处理能量计数谱
void changeEnergyCountView(QList<ParticleData> &dataList);
signals:
void newProject(const QString &project_name);
@ -96,6 +98,8 @@ private slots:
void on_action_stop_measure_triggered();
void on_AddressCountTimer();
void on_EnergyCountTimer();
private:
QMutex _mutex_info_output;
QPlainTextEdit* _plain_edit_info_output;
@ -125,13 +129,16 @@ private:
QMutex m_energyDataMutex;//能量谱锁
//能量计数
QHash<uint, QString> m_channelEnergyCountFiles; // 通道号 -> 能量计数文件路径
QHash<uint, QMap<double, unsigned long long>> m_channelEnergyStats; // 通道号 -> 能量bin -> 计数
QMap<double, unsigned long long> m_allChannelEnergyStats; // 全通道能量bin -> 计数
QHash<uint, QString> energy_count_filename_list; // 通道号 -> 能量计数文件路径
QHash<uint, QMap<double, unsigned long long>> channel_energy_counts; // 通道号 -> 能量bin -> 计数
QMap<double, unsigned long long> all_channel_energy_counts; // 全通道能量bin -> 计数
QMutex m_energyCountMutex; // 保护能量计数数据的互斥锁
QHash<QString, std::vector<double>> m_energyScaleCoeffCache;// 能量刻度系数缓存
//道址计数视图定时器
QTimer* m_AddressCountTimer;
//能量计数视图定时器
QTimer* m_EnergyCountTimer;
};
#endif // MAINWINDOW_H