366 lines
19 KiB
C++
366 lines
19 KiB
C++
#include "panels/columns/CategorySection.hpp"
|
||
|
||
#include <QComboBox>
|
||
#include <QDate>
|
||
#include <map>
|
||
#include <set>
|
||
#include "panels/columns/DateRangeEdit.hpp"
|
||
#include <QHBoxLayout>
|
||
#include <QLabel>
|
||
#include <QMenu>
|
||
#include <QPoint>
|
||
#include <QPushButton>
|
||
#include <QSet>
|
||
#include <QSignalBlocker>
|
||
#include <QToolButton>
|
||
#include <QTreeWidget>
|
||
#include <QTreeWidgetItemIterator>
|
||
#include <QVBoxLayout>
|
||
|
||
#include "Theme.hpp"
|
||
#include "panels/DatasetListPanel.hpp"
|
||
#include "repo/DatasetFieldDictionary.hpp"
|
||
|
||
namespace geopro::app {
|
||
|
||
using geopro::data::DsRow;
|
||
using geopro::data::DsTypeFields;
|
||
|
||
CategorySection::CategorySection(const CategorySpec& spec, geopro::data::DatasetFieldDictionary* dict,
|
||
QWidget* parent)
|
||
: QWidget(parent), spec_(spec), dict_(dict) {
|
||
auto* root = new QVBoxLayout(this);
|
||
root->setContentsMargins(0, 0, 0, 0);
|
||
root->setSpacing(0);
|
||
|
||
// 数据类型段头（可折叠，规范§4.3/§6）：chevron + 标题(title 字号·半粗) ｜「+ 新增三维体」(右,仅反演类)。
|
||
// 段头加浅底 + 底分隔线作视觉分段；去原生小三角(难看)→chevron 文本前缀，随主题/hover 变色。
|
||
auto* headerRow = new QWidget(this);
|
||
headerRow->setObjectName(QStringLiteral("secHeader"));
|
||
applyTokenizedStyleSheet(headerRow,
|
||
QStringLiteral("QWidget#secHeader{background:{{bg/panel-subtle}};"
|
||
"border-bottom:1px solid {{divider}};}"));
|
||
auto* hl = new QHBoxLayout(headerRow);
|
||
hl->setContentsMargins(space::kMd, space::kSm, space::kSm, space::kSm);
|
||
hl->setSpacing(space::kSm);
|
||
header_ = new QToolButton(headerRow);
|
||
header_->setCheckable(true);
|
||
header_->setChecked(true);
|
||
header_->setArrowType(Qt::NoArrow);
|
||
header_->setToolButtonStyle(Qt::ToolButtonTextOnly);
|
||
header_->setCursor(Qt::PointingHandCursor);
|
||
applyTokenizedStyleSheet(
|
||
header_, QStringLiteral("QToolButton{border:none;background:transparent;padding:0;"
|
||
"font-size:%1px;font-weight:%2;color:{{text/primary}};}"
|
||
"QToolButton:hover{color:{{accent/primary}};}")
|
||
.arg(scaledPx(type::kTitle))
|
||
.arg(type::kWeightSemibold));
|
||
auto syncHeader = [this] {
|
||
header_->setText((header_->isChecked() ? QStringLiteral("▾ ") : QStringLiteral("▸ "))
|
||
+ QString::fromStdString(spec_.title));
|
||
};
|
||
syncHeader();
|
||
hl->addWidget(header_);
|
||
hl->addStretch(1);
|
||
if (spec_.canGenerateVolume) {
|
||
auto* gen = new QToolButton(headerRow);
|
||
gen->setText(QStringLiteral("+ 新增三维体"));
|
||
gen->setCursor(Qt::PointingHandCursor);
|
||
// 次级强调按钮(规范§6.7)：描边 accent + accent 文字，hover 浅强调底；非裸文字。
|
||
applyTokenizedStyleSheet(
|
||
gen, QStringLiteral(
|
||
"QToolButton{border:1px solid {{accent/primary}};border-radius:%1px;"
|
||
"color:{{accent/primary}};background:transparent;padding:%2px %3px;font-size:%4px;}"
|
||
"QToolButton:hover{background:{{bg/selected}};}"
|
||
"QToolButton:pressed{background:{{bg/hover}};}")
|
||
.arg(radius::kSm)
|
||
.arg(scaledPx(space::kXxs))
|
||
.arg(scaledPx(space::kMd))
|
||
.arg(scaledPx(type::kCaption)));
|
||
connect(gen, &QToolButton::clicked, this, [this] {
|
||
emit generateVolumeRequested(QString::fromStdString(spec_.dsTypeCode), checkedDsIds());
|
||
});
|
||
hl->addWidget(gen);
|
||
}
|
||
root->addWidget(headerRow);
|
||
|
||
body_ = new QWidget(this);
|
||
auto* body = new QVBoxLayout(body_);
|
||
body->setContentsMargins(space::kMd, space::kSm, space::kMd, space::kMd);
|
||
body->setSpacing(space::kSm);
|
||
|
||
// 筛选行：采集时间范围（在前）+ 装置类型（在后，仅 ERT 类）。
|
||
auto* filterRow = new QHBoxLayout();
|
||
filterRow->setSpacing(space::kSm);
|
||
dateRange_ = new DateRangeEdit(body_);
|
||
connect(dateRange_, &DateRangeEdit::rangeChanged, this, [this] { rebuildList(); });
|
||
filterRow->addWidget(dateRange_, 1);
|
||
if (spec_.hasArrayTypeFilter) {
|
||
arrayCombo_ = new QComboBox(body_);
|
||
arrayCombo_->addItem(QStringLiteral("全部装置"), QString());
|
||
connect(arrayCombo_, qOverload<int>(&QComboBox::currentIndexChanged), this,
|
||
[this](int) { rebuildList(); });
|
||
filterRow->addWidget(arrayCombo_);
|
||
}
|
||
body->addLayout(filterRow);
|
||
|
||
// 段体：可勾选数据树（勾选=渲染）。复用数据列表卡片委托与 populateDatasetList。
|
||
list_ = new QTreeWidget(body_);
|
||
list_->setHeaderHidden(true);
|
||
list_->setRootIsDecorated(true);
|
||
list_->setIndentation(14); // 紧凑父子缩进（默认 20 太宽）
|
||
// #7：段体不出内层滚动条——内容超出时整段拉长，由 CategoryAnalysisTab 外层 QScrollArea 统一滚动。
|
||
list_->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
|
||
list_->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
|
||
applyDatasetCardDelegate(list_);
|
||
connect(list_, &QTreeWidget::itemChanged, this, [this](QTreeWidgetItem* it, int) {
|
||
// 异常行复选框 = 该异常显隐（异常不进渲染勾选集，单独走 anomalyVisibilityChanged → setAnomalyVisible）。
|
||
if (it && it->data(0, kDsDdCodeRole).toString() == QStringLiteral("dd_anomaly"))
|
||
emit anomalyVisibilityChanged(it->data(0, kDsIdRole).toString(),
|
||
it->checkState(0) == Qt::Checked);
|
||
emitChecked();
|
||
});
|
||
connect(list_, &QTreeWidget::itemDoubleClicked, this, [this](QTreeWidgetItem* it, int) {
|
||
const QString id = it->data(0, kDsIdRole).toString();
|
||
if (id.isEmpty()) return;
|
||
emit detailRequested(id, it->data(0, kDsDdCodeRole).toString(), it->data(0, kDsNameRole).toString());
|
||
});
|
||
if (spec_.id == "voxel") { // 仅三维体段提供右键操作菜单（体/切片/异常）
|
||
list_->setContextMenuPolicy(Qt::CustomContextMenu);
|
||
connect(list_, &QTreeWidget::customContextMenuRequested, this, &CategorySection::showContextMenu);
|
||
// 树选中切片/异常 → VTK 高亮联动（正向 list→VTK）。
|
||
connect(list_, &QTreeWidget::itemSelectionChanged, this, [this] {
|
||
const auto items = list_->selectedItems();
|
||
if (items.isEmpty()) return;
|
||
QTreeWidgetItem* it = items.first();
|
||
const QString id = it->data(0, kDsIdRole).toString();
|
||
const QString dd = it->data(0, kDsDdCodeRole).toString();
|
||
if (!id.isEmpty() && dd != QStringLiteral("container")) emit datasetSelected(id, dd);
|
||
});
|
||
}
|
||
body->addWidget(list_, 1);
|
||
|
||
root->addWidget(body_, 1);
|
||
|
||
connect(header_, &QToolButton::toggled, this, [this, syncHeader](bool on) {
|
||
body_->setVisible(on);
|
||
syncHeader(); // ▾(展开)/▸(折叠) 切换
|
||
emit collapsedChanged(); // 外层据此把折叠段 stretch 归 0、展开段吸收余量 → 折叠向上收
|
||
});
|
||
}
|
||
|
||
bool CategorySection::isExpanded() const { return header_ && header_->isChecked(); }
|
||
|
||
void CategorySection::setStructure(const std::vector<geopro::data::StructNode>& nodes) {
|
||
structure_ = nodes; // 容器分层（项目根/GS/TM→ds）在 Task 12 接入真实结构后据此构建。
|
||
}
|
||
|
||
void CategorySection::setDatasets(const std::vector<DsRow>& rows) {
|
||
rows_ = rows;
|
||
refreshArrayCombo();
|
||
rebuildList();
|
||
}
|
||
|
||
void CategorySection::selectItem(const QString& dsId) {
|
||
const QSignalBlocker block(list_); // 程序化选中(VTK→list)不回发 datasetSelected，避免环路
|
||
for (QTreeWidgetItemIterator it(list_); *it; ++it)
|
||
if (!dsId.isEmpty() && (*it)->data(0, kDsIdRole).toString() == dsId) {
|
||
list_->setCurrentItem(*it);
|
||
return;
|
||
}
|
||
list_->setCurrentItem(nullptr); // 空 dsId / 未找到 → 清选中
|
||
}
|
||
|
||
void CategorySection::setChecked(const QString& dsId, bool on) {
|
||
for (QTreeWidgetItemIterator it(list_); *it; ++it)
|
||
if ((*it)->data(0, kDsIdRole).toString() == dsId &&
|
||
((*it)->flags() & Qt::ItemIsUserCheckable)) {
|
||
(*it)->setCheckState(0, on ? Qt::Checked : Qt::Unchecked); // 触发 itemChanged→emitChecked→渲染
|
||
return;
|
||
}
|
||
}
|
||
|
||
void CategorySection::refreshArrayCombo() {
|
||
if (!spec_.hasArrayTypeFilter || !arrayCombo_) return;
|
||
const QString prev = arrayCombo_->currentData().toString();
|
||
const QSignalBlocker block(arrayCombo_);
|
||
arrayCombo_->clear();
|
||
arrayCombo_->addItem(QStringLiteral("全部装置"), QString());
|
||
// 按「加载到本段的数据自身携带的类型值」的范围筛选（用户口径）。装置/arrayType 不在 ds 行数据上
|
||
// （实测 data/page 列表无该字段），故用 ds 已带的类型名 typeName（多为中文）作筛选维度；
|
||
// typeName 缺失时回退 dsTypeCode + 全局枚举翻译。列出组内出现过的类型值。
|
||
QSet<QString> seen;
|
||
for (const auto& r : rows_) {
|
||
QString display = QString::fromStdString(r.typeName);
|
||
QString code = display;
|
||
if (display.isEmpty() && !r.dsTypeCode.empty()) { // 无类型名 → 用 dsTypeCode（必要时枚举翻译）
|
||
code = QString::fromStdString(r.dsTypeCode);
|
||
display = code;
|
||
if (dict_) {
|
||
const auto& en = dict_->arrayTypeEnum();
|
||
const auto it = en.find(r.dsTypeCode);
|
||
if (it != en.end()) display = QString::fromStdString(it->second);
|
||
}
|
||
}
|
||
if (code.isEmpty() || seen.contains(code)) continue;
|
||
seen.insert(code);
|
||
arrayCombo_->addItem(display, code); // itemData=类型值（passesFilters 据此比对 typeName/dsTypeCode）
|
||
}
|
||
const int idx = arrayCombo_->findData(prev); // 尽量保留上次选择
|
||
arrayCombo_->setCurrentIndex(idx >= 0 ? idx : 0);
|
||
}
|
||
|
||
bool CategorySection::passesFilters(const DsRow& row) const {
|
||
// 类型筛选（"全部"=空不筛）：按 ds 自身类型值（typeName，回退 dsTypeCode）命中选中项。
|
||
if (spec_.hasArrayTypeFilter && arrayCombo_) {
|
||
const QString sel = arrayCombo_->currentData().toString();
|
||
if (!sel.isEmpty()) {
|
||
const QString t = !row.typeName.empty() ? QString::fromStdString(row.typeName)
|
||
: QString::fromStdString(row.dsTypeCode);
|
||
if (t != sel) return false;
|
||
}
|
||
}
|
||
// 采集时间范围（collectTime 经 dict 取；缺则回退 createTime；空范围不约束）。
|
||
const QDate from = dateRange_ ? dateRange_->from() : QDate();
|
||
const QDate to = dateRange_ ? dateRange_->to() : QDate();
|
||
if (from.isValid() || to.isValid()) {
|
||
const DsTypeFields* f = dict_ ? dict_->fields(spec_.dsTypeCode) : nullptr;
|
||
std::string ts = f ? collectTimeOf(row, *f) : std::string();
|
||
if (ts.empty()) ts = row.createTime;
|
||
const QDate d = QDate::fromString(QString::fromStdString(ts).left(10), QStringLiteral("yyyy-MM-dd"));
|
||
if (d.isValid()) {
|
||
if (from.isValid() && d < from) return false;
|
||
if (to.isValid() && d > to) return false;
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
void CategorySection::rebuildList() {
|
||
// 增量保留：记住当前已勾选的 ds，重建后复原（仍存在的项保持勾选）。否则每次刷新(勾选对象/建体/
|
||
// 存切片/建异常都会触发)清空全部勾选 → 渲染被重置，体验极差（用户反馈：必须增量更新）。
|
||
std::set<std::string> wasChecked, wasSeen; // wasSeen=重建前所有可勾选项(区分"新项" vs "曾取消")
|
||
for (QTreeWidgetItemIterator it(list_); *it; ++it) {
|
||
if (!((*it)->flags() & Qt::ItemIsUserCheckable)) continue;
|
||
const std::string id = (*it)->data(0, kDsIdRole).toString().toStdString();
|
||
wasSeen.insert(id);
|
||
if ((*it)->checkState(0) == Qt::Checked) wasChecked.insert(id);
|
||
}
|
||
|
||
std::vector<DsRow> filtered;
|
||
filtered.reserve(rows_.size());
|
||
for (const auto& r : rows_)
|
||
if (passesFilters(r)) filtered.push_back(r);
|
||
|
||
// 从项目根的层级树：容器节点（结构，剪枝仅留有 ds 的路径）+ ds（挂 parentId 或 structParentId）。
|
||
std::vector<DsRow> display;
|
||
if (!structure_.empty()) {
|
||
std::map<std::string, const geopro::data::StructNode*> byId;
|
||
for (const auto& n : structure_) byId[n.id] = &n;
|
||
std::set<std::string> keep; // 收集每个 ds 的结构归属向上的祖先链
|
||
for (const auto& d : filtered) {
|
||
std::string p = d.structParentId;
|
||
while (!p.empty() && byId.count(p) && !keep.count(p)) {
|
||
keep.insert(p);
|
||
p = byId[p]->parentId;
|
||
}
|
||
}
|
||
for (const auto& n : structure_)
|
||
if (keep.count(n.id)) {
|
||
DsRow c;
|
||
c.id = n.id;
|
||
c.dsName = n.name;
|
||
c.ddCode = "container";
|
||
c.parentId = n.parentId;
|
||
display.push_back(std::move(c));
|
||
}
|
||
}
|
||
// 分段树是「项目/GS/TM」组织树：ds 应挂到其结构容器(TM)下。但派生 ds(如反演剖面)带派生父
|
||
// parentId（指向原始 ds），若该原始 ds 不在本段(被筛掉/属别类型段)，按 parentId 挂载会失败 →
|
||
// ds 浮到树根平铺（用户实测 bug：ds 没挂在 tm 下）。故：派生父在本段则保留派生嵌套(如 体>切片)，
|
||
// 否则回退挂到结构容器 structParentId(TM)。
|
||
std::set<std::string> presentIds;
|
||
for (const auto& d : filtered) presentIds.insert(d.id);
|
||
for (const auto& d : filtered) {
|
||
DsRow x = d;
|
||
if (x.parentId.empty() || !presentIds.count(x.parentId)) x.parentId = x.structParentId;
|
||
display.push_back(std::move(x));
|
||
}
|
||
|
||
{
|
||
const QSignalBlocker block(list_);
|
||
populateDatasetList(list_, display, /*append=*/false);
|
||
for (QTreeWidgetItemIterator it(list_); *it; ++it) {
|
||
// 容器节点（项目根/GS/TM）只作层级骨架——明确去掉复选框、不可勾选。
|
||
if ((*it)->data(0, kDsDdCodeRole).toString() == QStringLiteral("container")) {
|
||
(*it)->setFlags((*it)->flags() & ~Qt::ItemIsUserCheckable);
|
||
continue;
|
||
}
|
||
(*it)->setFlags((*it)->flags() | Qt::ItemIsUserCheckable);
|
||
const std::string id = (*it)->data(0, kDsIdRole).toString().toStdString();
|
||
const bool isAnomaly = (*it)->data(0, kDsDdCodeRole).toString() == QStringLiteral("dd_anomaly");
|
||
// 复原勾选：曾勾选→勾；曾出现但未勾→不勾；新项→异常默认勾(显示),其余默认不勾。
|
||
const Qt::CheckState st = wasChecked.count(id) ? Qt::Checked
|
||
: wasSeen.count(id) ? Qt::Unchecked
|
||
: (isAnomaly ? Qt::Checked : Qt::Unchecked);
|
||
(*it)->setCheckState(0, st);
|
||
}
|
||
}
|
||
list_->expandAll(); // 展开容器层级（项目根/GS/TM），让体/切片/异常可见
|
||
// #7：段体无内层滚动条 → list 最小高度 = 可见项内容总高（与 DatasetCardDelegate::sizeHint 一致：
|
||
// 有副标题 52，否则 30）。内容多则撑大本段、超视口由外层滚动；内容少则被 stretch 拉到平分高度。
|
||
int contentH = 2 * list_->frameWidth() + 4;
|
||
for (QTreeWidgetItemIterator hit(list_); *hit; ++hit)
|
||
if (!(*hit)->isHidden())
|
||
contentH += (*hit)->text(0).contains(QLatin1Char('\n')) ? 52 : 30;
|
||
list_->setMinimumHeight(contentH);
|
||
emitChecked(); // 上抛复原后的勾选集（保持渲染，不再清空 → 控制器据 diff 增量保留已渲染图元）
|
||
}
|
||
|
||
QStringList CategorySection::checkedDsIds() const {
|
||
QStringList ids;
|
||
for (QTreeWidgetItemIterator it(list_); *it; ++it)
|
||
if ((*it)->checkState(0) == Qt::Checked) ids << (*it)->data(0, kDsIdRole).toString();
|
||
return ids;
|
||
}
|
||
|
||
void CategorySection::emitChecked() { emit checkedDatasetsChanged(checkedDsIds()); }
|
||
|
||
void CategorySection::showContextMenu(const QPoint& pos) {
|
||
using geopro::render::interact::SliceAxis;
|
||
QTreeWidgetItem* it = list_->itemAt(pos);
|
||
if (!it) return;
|
||
const QString id = it->data(0, kDsIdRole).toString();
|
||
const QString ddCode = it->data(0, kDsDdCodeRole).toString();
|
||
if (id.isEmpty() || ddCode == QStringLiteral("container")) return; // 容器节点无操作菜单
|
||
const QString name = it->data(0, kDsNameRole).toString();
|
||
QMenu menu(this);
|
||
menu.addAction(QStringLiteral("详情"), this,
|
||
[this, id, ddCode, name] { emit detailRequested(id, ddCode, name); });
|
||
if (ddCode == QStringLiteral("dd_voxel")) { // 三维体
|
||
QMenu* sl = menu.addMenu(QStringLiteral("生成切片")); // id=被右键的三维体 dsId（切片建到该体上）
|
||
sl->addAction(QStringLiteral("上下"), this, [this, id] { emit sliceRequested(SliceAxis::UpDown, id); });
|
||
sl->addAction(QStringLiteral("前后"), this, [this, id] { emit sliceRequested(SliceAxis::FrontBack, id); });
|
||
sl->addAction(QStringLiteral("左右"), this, [this, id] { emit sliceRequested(SliceAxis::LeftRight, id); });
|
||
sl->addAction(QStringLiteral("任意"), this, [this, id] { emit sliceRequested(SliceAxis::Oblique, id); });
|
||
menu.addAction(QStringLiteral("色阶…"), this, [this, id] { emit colorScaleRequested(id); });
|
||
} else if (ddCode == QStringLiteral("dd_slice")) { // 切片
|
||
menu.addAction(QStringLiteral("保存位姿"), this, [this, id] { emit sliceSaveRequested(id); });
|
||
menu.addAction(QStringLiteral("另存为…"), this, [this, id] { emit sliceSaveAsRequested(id); });
|
||
QMenu* ex = menu.addMenu(QStringLiteral("导出"));
|
||
ex->addAction(QStringLiteral("图片"), this, [this, id] { emit sliceExportImageRequested(id); });
|
||
ex->addAction(QStringLiteral("dat"), this, [this, id] { emit sliceExportDatRequested(id); });
|
||
menu.addAction(QStringLiteral("色阶…"), this, [this, id] { emit colorScaleRequested(id); });
|
||
menu.addSeparator();
|
||
menu.addAction(QStringLiteral("删除"), this, [this, id, ddCode] { emit deleteDatasetRequested(id, ddCode); });
|
||
} else if (ddCode == QStringLiteral("dd_anomaly")) { // 异常
|
||
menu.addAction(QStringLiteral("显示"), this, [this, id] { emit anomalyVisibilityChanged(id, true); });
|
||
menu.addAction(QStringLiteral("隐藏"), this, [this, id] { emit anomalyVisibilityChanged(id, false); });
|
||
menu.addSeparator();
|
||
menu.addAction(QStringLiteral("删除"), this, [this, id, ddCode] { emit deleteDatasetRequested(id, ddCode); });
|
||
}
|
||
menu.exec(list_->viewport()->mapToGlobal(pos));
|
||
}
|
||
|
||
} // namespace geopro::app
|