|
|
#include "nmWxDisplaySettings.h"
|
|
|
#include <QDebug>
|
|
|
#include <QQueue>
|
|
|
|
|
|
#include "nmDataAnalyzeManager.h"
|
|
|
#include "nmGuiPlot.h"
|
|
|
#include "ZxObjBase.h"
|
|
|
#include <QString>
|
|
|
|
|
|
#include "ZxBaseUtil.h"
|
|
|
|
|
|
// --- CustomDisPlayTreeWidgetItem 实现 ---
|
|
|
CustomDisPlayTreeWidgetItem::CustomDisPlayTreeWidgetItem(QTreeWidget *pParent)
|
|
|
: QTreeWidgetItem(pParent)
|
|
|
{
|
|
|
m_pCheckBox = new QCheckBox();
|
|
|
m_pCheckBox->setChecked(true); // 默认选中
|
|
|
// 注意：这里不立即设置setItemWidget，因为treeWidget()可能返回nullptr
|
|
|
// nmWxDisplaySettings的构造函数中会统一设置
|
|
|
}
|
|
|
|
|
|
CustomDisPlayTreeWidgetItem::CustomDisPlayTreeWidgetItem(QTreeWidgetItem *pParent)
|
|
|
: QTreeWidgetItem(pParent)
|
|
|
{
|
|
|
m_pCheckBox = new QCheckBox();
|
|
|
m_pCheckBox->setChecked(true); // 默认选中
|
|
|
}
|
|
|
|
|
|
// --- CustomDisPlayTreeWidget 实现 ---
|
|
|
CustomDisPlayTreeWidget::CustomDisPlayTreeWidget(QWidget *pParent) : QTreeWidget(pParent)
|
|
|
{
|
|
|
// 禁用交互功能，使树看起来是静态的
|
|
|
setItemsExpandable(false); // 禁止用户展开/折叠节点
|
|
|
// setAttribute(Qt::WA_TransparentForMouseEvents); // 移除此行，否则鼠标事件（如复选框点击）会失效
|
|
|
setMouseTracking(true); // 启用鼠标跟踪，以便正确处理鼠标移动事件
|
|
|
|
|
|
setSelectionMode(QAbstractItemView::NoSelection); // 禁止选择项
|
|
|
setFocusPolicy(Qt::NoFocus); // 禁止获取焦点，避免虚线框
|
|
|
setAllColumnsShowFocus(false); // 禁止显示焦点矩形
|
|
|
// 隐藏滚动条
|
|
|
//setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
|
|
|
setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
|
|
|
|
|
|
// 视觉样式
|
|
|
setStyleSheet(
|
|
|
//"QTreeWidget {"
|
|
|
//" border: none;"
|
|
|
//" background: transparent;"
|
|
|
//"}"
|
|
|
//"QTreeWidget::item {"
|
|
|
//" background: transparent;"
|
|
|
//" border: none; /* 确保项没有边框 */"
|
|
|
//"}"
|
|
|
"QTreeWidget::item:hover {"
|
|
|
" background: transparent; /* 确保hover时背景不变 */"
|
|
|
"}"
|
|
|
"QTreeWidget::branch {"
|
|
|
" background: transparent; /* 隐藏默认分支线 */"
|
|
|
"}"
|
|
|
);
|
|
|
|
|
|
// 初始化连接线缓存
|
|
|
m_connectionLines.clear();
|
|
|
|
|
|
connect(verticalScrollBar(), SIGNAL(valueChanged(int)),
|
|
|
this, SLOT(updateConnectionLines()));
|
|
|
}
|
|
|
|
|
|
// 更新根节点连接线位置
|
|
|
void CustomDisPlayTreeWidget::updateConnectionLines()
|
|
|
{
|
|
|
m_connectionLines.clear();
|
|
|
|
|
|
int rootCount = topLevelItemCount();
|
|
|
if(rootCount <= 1) return;
|
|
|
|
|
|
// 考虑滚动偏移量
|
|
|
int scrollOffset = verticalScrollBar()->value();
|
|
|
|
|
|
// 获取第一个根节点位置
|
|
|
QTreeWidgetItem* firstRoot = topLevelItem(0);
|
|
|
if(!firstRoot) return;
|
|
|
QRect firstRect = visualItemRect(firstRoot);
|
|
|
firstRect.translate(0, -scrollOffset); // 调整滚动偏移
|
|
|
|
|
|
// 获取最后一个根节点位置
|
|
|
QTreeWidgetItem* lastRoot = topLevelItem(rootCount - 1);
|
|
|
if(!lastRoot) return;
|
|
|
QRect lastRect = visualItemRect(lastRoot);
|
|
|
lastRect.translate(0, -scrollOffset); // 调整滚动偏移
|
|
|
|
|
|
const int baseOffset = 10;
|
|
|
int lineStartX = lastRect.left() - baseOffset;
|
|
|
|
|
|
// 计算连接线中点Y坐标
|
|
|
int firstMidY = firstRect.center().y();
|
|
|
int lastMidY = lastRect.center().y();
|
|
|
|
|
|
// 保存连接线
|
|
|
ConnectionLine line;
|
|
|
line.start = QPoint(lineStartX, lastMidY);
|
|
|
line.end = QPoint(lineStartX, firstMidY);
|
|
|
m_connectionLines.append(line);
|
|
|
}
|
|
|
|
|
|
void CustomDisPlayTreeWidget::drawBranches(QPainter *pPainter, const QRect &rect, const QModelIndex &index) const
|
|
|
{
|
|
|
QTreeWidgetItem *item = itemFromIndex(index);
|
|
|
if(!item) return;
|
|
|
|
|
|
int level = 0;
|
|
|
for(QTreeWidgetItem *parent = item->parent(); parent; parent = parent->parent()) {
|
|
|
level++;
|
|
|
}
|
|
|
|
|
|
pPainter->save();
|
|
|
pPainter->setPen(QPen(Qt::black, 1.2, Qt::DotLine));
|
|
|
|
|
|
const int baseOffset = 10;
|
|
|
const int lineLength = 10;
|
|
|
int lineY = rect.center().y();
|
|
|
|
|
|
int dynamicOffset = baseOffset;
|
|
|
if(level == 1) dynamicOffset += 20;
|
|
|
else if(level == 2) dynamicOffset += 40;
|
|
|
|
|
|
int lineStartX = rect.left() + dynamicOffset;
|
|
|
int lineEndX = lineStartX + lineLength;
|
|
|
|
|
|
// 1. 绘制水平线
|
|
|
pPainter->drawLine(lineStartX, lineY, lineEndX, lineY);
|
|
|
|
|
|
// 2. 绘制垂直线
|
|
|
if(level > 0 && item->parent() != nullptr) {
|
|
|
// 仅绘制到当前项范围的垂直线，不要延伸到整个高度
|
|
|
int verticalLength = lineLength;
|
|
|
int verticalStartY = lineY - verticalLength;
|
|
|
int verticalEndY = lineY + verticalLength;
|
|
|
|
|
|
// 当前节点在父节点中的索引
|
|
|
int childIndex = item->parent()->indexOfChild(item);
|
|
|
int childCount = item->parent()->childCount();
|
|
|
|
|
|
// 绘制从当前节点到顶部的垂直线
|
|
|
pPainter->drawLine(lineStartX, lineY, lineStartX, verticalStartY);
|
|
|
|
|
|
// 如果不是最后一个子节点，则绘制向下的垂直线
|
|
|
if (childIndex < childCount - 1) {
|
|
|
pPainter->drawLine(lineStartX, lineY, lineStartX, verticalEndY);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
pPainter->restore();
|
|
|
}
|
|
|
|
|
|
// 重写绘制事件以绘制所有内容，包括QTreeWidget默认内容和我们的特殊连接线
|
|
|
void CustomDisPlayTreeWidget::paintEvent(QPaintEvent *pEvent)
|
|
|
{
|
|
|
QTreeWidget::paintEvent(pEvent); // 先绘制原始树
|
|
|
|
|
|
// 绘制缓存的连接线
|
|
|
QPainter painter(viewport());
|
|
|
painter.setPen(QPen(Qt::black, 1.2, Qt::DotLine));
|
|
|
|
|
|
foreach(const ConnectionLine &line, m_connectionLines) {
|
|
|
painter.drawLine(line.start, line.end);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// --- nmWxDisplaySettings 类实现 ---
|
|
|
nmWxDisplaySettings::nmWxDisplaySettings(QWidget *pParent) : QDialog(pParent)
|
|
|
{
|
|
|
setWindowTitle(tr("Display settings")); // 设置窗口标题
|
|
|
resize(313, 416); // 设置窗口大小
|
|
|
|
|
|
// 初始化UI组件
|
|
|
m_pTreeWidget = new CustomDisPlayTreeWidget(this);
|
|
|
m_pTreeWidget->resize(276, 299);
|
|
|
m_pTreeWidget->setColumnCount(1); // 只有一列
|
|
|
m_pTreeWidget->setHeaderHidden(true); // 隐藏表头
|
|
|
|
|
|
// 刷新显示哪些图元
|
|
|
nmDataAnalyzeManager::getCurrentInstance()->refreshChildItemsDisplay();
|
|
|
|
|
|
// 获取保存的显示设置 - 现在使用QVector
|
|
|
QVector<DisplaySetting> vecDisplaySettings = nmDataAnalyzeManager::getCurrentInstance()->getDisplaySettings();
|
|
|
|
|
|
// 遍历显示设置并构建树
|
|
|
for(int i = 0; i < vecDisplaySettings.size(); ++i) {
|
|
|
const DisplaySetting& setting = vecDisplaySettings[i];
|
|
|
const QString& categoryName = setting.key;
|
|
|
const CategoryDisplayInfo& info = setting.value;
|
|
|
|
|
|
// 创建父节点
|
|
|
CustomDisPlayTreeWidgetItem *pParentItem = new CustomDisPlayTreeWidgetItem(m_pTreeWidget);
|
|
|
m_pTreeWidget->setItemWidget(pParentItem, 0, pParentItem->checkBox());
|
|
|
pParentItem->checkBox()->setText(categoryName);
|
|
|
connect(pParentItem->checkBox(), SIGNAL(stateChanged(int)),
|
|
|
this, SLOT(onCheckBoxStateChanged(int)));
|
|
|
pParentItem->checkBox()->setChecked(info.isRootChecked);
|
|
|
|
|
|
// 创建子节点
|
|
|
for(int j = 0; j < info.childItems.size(); ++j) {
|
|
|
const ChildCategoryDisplayInfo& childInfo = info.childItems[j];
|
|
|
|
|
|
CustomDisPlayTreeWidgetItem *pChildItem = new CustomDisPlayTreeWidgetItem(pParentItem);
|
|
|
QCheckBox* pChildCheckBox = pChildItem->checkBox();
|
|
|
m_pTreeWidget->setItemWidget(pChildItem, 0, pChildCheckBox);
|
|
|
pChildCheckBox->setText(childInfo.name);
|
|
|
connect(pChildCheckBox, SIGNAL(stateChanged(int)),
|
|
|
this, SLOT(onCheckBoxStateChanged(int)));
|
|
|
pChildCheckBox->setChecked(childInfo.isChecked);
|
|
|
pChildCheckBox->setEnabled(childInfo.isEnabled);
|
|
|
}
|
|
|
|
|
|
// 更新子项的启用状态
|
|
|
updateChildItemsEnableState(pParentItem, info.isRootChecked);
|
|
|
}
|
|
|
|
|
|
m_pTreeWidget->expandAll(); // 展开所有节点，实现静态树效果
|
|
|
m_pTreeWidget->updateConnectionLines(); // 初始展开后立即更新连接线
|
|
|
|
|
|
// --- 底部按钮 ---
|
|
|
m_pImageManagerButton = new QPushButton(tr("Image manager"), this);
|
|
|
m_pApplyButton = new QPushButton(tr("Apply"), this);
|
|
|
m_pOkButton = new QPushButton(tr("OK"), this);
|
|
|
m_pCancelButton = new QPushButton(tr("Cancel"), this);
|
|
|
|
|
|
// 连接按钮的信号到槽
|
|
|
connect(m_pImageManagerButton, SIGNAL(clicked()), this, SLOT(onImageManagerClicked()));
|
|
|
connect(m_pApplyButton, SIGNAL(clicked()), this, SLOT(onApplyClicked()));
|
|
|
connect(m_pOkButton, SIGNAL(clicked()), this, SLOT(onOkClicked()));
|
|
|
connect(m_pCancelButton, SIGNAL(clicked()), this, SLOT(onCancelClicked()));
|
|
|
|
|
|
// --- 布局设置 ---
|
|
|
// 声明pButtonsVerticalLayout变量
|
|
|
QVBoxLayout *pButtonsVerticalLayout = new QVBoxLayout();
|
|
|
pButtonsVerticalLayout->setContentsMargins(0, 0, 0, 0); // 移除额外边距，让按钮紧凑
|
|
|
|
|
|
// 为 Image Manager 按钮创建单独的水平布局
|
|
|
QHBoxLayout *pImageManagerLayout = new QHBoxLayout();
|
|
|
//pImageManagerLayout->addStretch(); // 左侧伸缩空间，将按钮推向右侧
|
|
|
pImageManagerLayout->addWidget(m_pImageManagerButton);
|
|
|
// pImageManagerLayout->addStretch(); // 右侧伸缩空间，将按钮推向左侧，实现居中
|
|
|
pImageManagerLayout->setContentsMargins(0, 0, 0, 0); // 移除额外边距
|
|
|
|
|
|
// 为 Apply, OK, Cancel 按钮创建另一个水平布局
|
|
|
QHBoxLayout* pButtonLayout = new QHBoxLayout();
|
|
|
//pButtonLayout->addStretch(); // 左侧留白，推向右侧
|
|
|
pButtonLayout->addWidget(m_pApplyButton);
|
|
|
pButtonLayout->addWidget(m_pOkButton);
|
|
|
pButtonLayout->addWidget(m_pCancelButton);
|
|
|
//pButtonLayout->addStretch(); // 右侧留白，推向左侧
|
|
|
pButtonLayout->setContentsMargins(0, 0, 0, 0); // 移除额外边距
|
|
|
|
|
|
// 将两个水平按钮布局添加到垂直按钮布局中
|
|
|
pButtonsVerticalLayout->addLayout(pImageManagerLayout);
|
|
|
pButtonsVerticalLayout->addLayout(pButtonLayout);
|
|
|
|
|
|
// 设置主垂直布局
|
|
|
m_pMainLayout = new QVBoxLayout(this);
|
|
|
m_pMainLayout->addWidget(m_pTreeWidget);
|
|
|
// 在树控件和按钮之间添加伸缩空间，确保树控件占据大部分垂直空间
|
|
|
//m_pMainLayout->addStretch();
|
|
|
// 将包含所有按钮的垂直布局添加到主布局的底部
|
|
|
m_pMainLayout->addLayout(pButtonsVerticalLayout);
|
|
|
|
|
|
// 设置对话框的布局
|
|
|
setLayout(m_pMainLayout);
|
|
|
}
|
|
|
|
|
|
nmWxDisplaySettings::~nmWxDisplaySettings()
|
|
|
{
|
|
|
}
|
|
|
|
|
|
void nmWxDisplaySettings::onCheckBoxStateChanged(int state)
|
|
|
{
|
|
|
QCheckBox *pCheckBox = qobject_cast<QCheckBox*>(sender());
|
|
|
|
|
|
if(!pCheckBox) return;
|
|
|
|
|
|
// 查找与触发信号的复选框关联的QTreeWidgetItem
|
|
|
QTreeWidgetItem *pChangedItem = nullptr;
|
|
|
// 使用BFS（广度优先搜索）或DFS（深度优先搜索）来查找项是可靠的方法
|
|
|
// 这里使用BFS
|
|
|
QQueue<QTreeWidgetItem*> itemsToProcess;
|
|
|
|
|
|
// 将所有顶层项添加到队列
|
|
|
for(int i = 0; i < m_pTreeWidget->topLevelItemCount(); ++i) {
|
|
|
itemsToProcess.enqueue(m_pTreeWidget->topLevelItem(i));
|
|
|
}
|
|
|
|
|
|
while(!itemsToProcess.isEmpty()) {
|
|
|
QTreeWidgetItem *pCurrentItem = itemsToProcess.dequeue();
|
|
|
|
|
|
// 检查当前项是否包含触发信号的复选框
|
|
|
if(m_pTreeWidget->itemWidget(pCurrentItem, 0) == pCheckBox) {
|
|
|
pChangedItem = pCurrentItem;
|
|
|
break; // 找到对应的项，退出循环
|
|
|
}
|
|
|
|
|
|
// 将当前项的所有子项添加到队列
|
|
|
for(int i = 0; i < pCurrentItem->childCount(); ++i) {
|
|
|
itemsToProcess.enqueue(pCurrentItem->child(i));
|
|
|
}
|
|
|
}
|
|
|
|
|
|
if(pChangedItem) {
|
|
|
// 更新子项的启用状态
|
|
|
updateChildItemsEnableState(pChangedItem, state != Qt::Unchecked);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
void nmWxDisplaySettings::updateChildItemsEnableState(QTreeWidgetItem *pItem, bool bEnabled)
|
|
|
{
|
|
|
// 递归遍历子项，设置复选框的启用状态
|
|
|
for(int i = 0; i < pItem->childCount(); ++i) {
|
|
|
QTreeWidgetItem *pChild = pItem->child(i);
|
|
|
|
|
|
// 动态转换确保是我们的CustomDisPlayTreeWidgetItem类型
|
|
|
if(CustomDisPlayTreeWidgetItem *pCustomDisPlayChild = dynamic_cast<CustomDisPlayTreeWidgetItem * >(pChild)) {
|
|
|
pCustomDisPlayChild->checkBox()->setEnabled(bEnabled);
|
|
|
// 递归调用，处理更深层级的子项
|
|
|
updateChildItemsEnableState(pChild, bEnabled);
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
void nmWxDisplaySettings::onImageManagerClicked()
|
|
|
{
|
|
|
qDebug("Image Manager button clicked!");
|
|
|
// 这里可以打开一个新的对话框或执行其他逻辑
|
|
|
}
|
|
|
|
|
|
void nmWxDisplaySettings::onApplyClicked()
|
|
|
{
|
|
|
// 存储更新后的显示设置
|
|
|
QVector<DisplaySetting> vecUpdatedSettings;
|
|
|
|
|
|
// 遍历所有顶层项
|
|
|
for(int i = 0; i < m_pTreeWidget->topLevelItemCount(); ++i) {
|
|
|
CustomDisPlayTreeWidgetItem* pTopItem =
|
|
|
dynamic_cast<CustomDisPlayTreeWidgetItem*>(m_pTreeWidget->topLevelItem(i));
|
|
|
|
|
|
if(pTopItem) {
|
|
|
DisplaySetting setting;
|
|
|
setting.key = pTopItem->checkBox()->text();
|
|
|
setting.value.isRootChecked = pTopItem->checkBox()->isChecked();
|
|
|
|
|
|
// 收集子项信息
|
|
|
for(int j = 0; j < pTopItem->childCount(); ++j) {
|
|
|
CustomDisPlayTreeWidgetItem* pChildItem =
|
|
|
dynamic_cast<CustomDisPlayTreeWidgetItem*>(pTopItem->child(j));
|
|
|
|
|
|
if(pChildItem) {
|
|
|
QCheckBox* pChildCheckBox = pChildItem->checkBox();
|
|
|
ChildCategoryDisplayInfo childInfo;
|
|
|
childInfo.name = pChildCheckBox->text();
|
|
|
childInfo.isChecked = pChildCheckBox->isChecked();
|
|
|
childInfo.isEnabled = pChildCheckBox->isEnabled();
|
|
|
|
|
|
setting.value.childItems.append(childInfo);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
vecUpdatedSettings.append(setting);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 更新数据中心的显示设置
|
|
|
nmDataAnalyzeManager::getCurrentInstance()->setDisplaySettings(vecUpdatedSettings);
|
|
|
|
|
|
// 更新绘图
|
|
|
nmGuiPlot* pPlot = nmDataAnalyzeManager::getCurrentInstance()->getPlot();
|
|
|
|
|
|
if(pPlot) {
|
|
|
// 遍历更新后的设置并应用到绘图
|
|
|
for (int i = 0; i < vecUpdatedSettings.size(); ++i) {
|
|
|
const DisplaySetting& setting = vecUpdatedSettings.at(i);
|
|
|
QString sCategoryKey = setting.key;
|
|
|
CategoryDisplayInfo categoryInfo = setting.value;
|
|
|
bool isRootChecked = categoryInfo.isRootChecked; // 整体是否选中
|
|
|
|
|
|
if (categoryInfo.childItems.size() == 0 || isRootChecked == false) { // 无子节点，或者父节点为未选中状态
|
|
|
// 判断类别
|
|
|
if (_isSame(sCategoryKey,tr("Contour"))) {
|
|
|
pPlot->setContourVisibility(isRootChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("Images"))) {
|
|
|
pPlot->setImagesVisibility(isRootChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("Faults"))) {
|
|
|
pPlot->setAllFaultsVisible(isRootChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("Wells"))) {
|
|
|
pPlot->setAllWellsVisible(isRootChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("Fractures"))) {
|
|
|
pPlot->setAllFracturesVisible(isRootChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("Regions"))) {
|
|
|
pPlot->setAllRegionsVisible(isRootChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("RegionMarks"))) {
|
|
|
pPlot->setAllRegionMarksVisible(isRootChecked);
|
|
|
}
|
|
|
} else { // 包含子节点，并且父节点处于选中状态
|
|
|
|
|
|
// 遍历子节点（使用QVector的迭代器）
|
|
|
QVector<ChildCategoryDisplayInfo>::const_iterator childIt;
|
|
|
for (childIt = categoryInfo.childItems.constBegin();
|
|
|
childIt != categoryInfo.childItems.constEnd();
|
|
|
++childIt) {
|
|
|
const ChildCategoryDisplayInfo& child = *childIt;
|
|
|
// 判断类别
|
|
|
if (_isSame(sCategoryKey,tr("Faults"))) {
|
|
|
pPlot->setFaultVisible(child.name, child.isChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("Wells"))) {
|
|
|
pPlot->setWellVisible(child.name, child.isChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("Fractures"))) {
|
|
|
pPlot->setFractureVisible(child.name, child.isChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("Regions"))) {
|
|
|
pPlot->setRegionVisible(child.name, child.isChecked);
|
|
|
} else if (_isSame(sCategoryKey,tr("RegionMarks"))) {
|
|
|
pPlot->setRegionMarkVisible(child.name, child.isChecked);
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
pPlot->update();
|
|
|
}
|
|
|
}
|
|
|
|
|
|
void nmWxDisplaySettings::onOkClicked()
|
|
|
{
|
|
|
// 应用当前设置并关闭对话框
|
|
|
onApplyClicked();
|
|
|
accept();
|
|
|
}
|
|
|
|
|
|
void nmWxDisplaySettings::onCancelClicked()
|
|
|
{
|
|
|
// 放弃当前设置并关闭对话框
|
|
|
reject();
|
|
|
}
|