181 lines
6.1 KiB
C++
181 lines
6.1 KiB
C++
#include "QImageCursorWidget.h"
|
||
#include <QApplication>
|
||
#include <QMainWindow>
|
||
#include <QVBoxLayout>
|
||
#include <QWidget>
|
||
#include <memory>
|
||
#include <qgsrasteridentifyresult.h>
|
||
#include <qgsmapcanvas.h>
|
||
#include <QCheckBox>
|
||
#include <QTreeView>
|
||
#include <qgsraster.h>
|
||
#include <qgsrasterlayer.h>
|
||
#include <qgsmeshlayer.h>
|
||
|
||
|
||
|
||
|
||
QImageCursorWidget::QImageCursorWidget(QWidget *parent)
|
||
: QDialog(parent)
|
||
{
|
||
maplayer_treewidgetItem.clear();
|
||
this->initUi();
|
||
this->setWindowTitle(u8"游标窗口");
|
||
this->resize(400, 600);
|
||
}
|
||
|
||
QImageCursorWidget::~QImageCursorWidget()
|
||
{}
|
||
|
||
void QImageCursorWidget::initUi()
|
||
{
|
||
// 创建主布局（垂直分布）
|
||
QVBoxLayout* mainLayout = new QVBoxLayout(this);
|
||
|
||
// 1. 创建菜单按钮栏（这里使用QToolBar或QWidget配合QHBoxLayout模拟）
|
||
QWidget* toolBarWidget = new QWidget(this);
|
||
QHBoxLayout* toolBarLayout = new QHBoxLayout(toolBarWidget);
|
||
toolBarLayout->setContentsMargins(0, 0, 0, 0); // 可选：移除边距使布局更紧凑
|
||
|
||
// 创建复选框按钮：“时序图”、“方向图”
|
||
QCheckBox* checkBoxTimeSeries = new QCheckBox(tr(u8"时序图"), toolBarWidget);
|
||
QCheckBox* checkBoxDirection = new QCheckBox(tr(u8"方向图"), toolBarWidget);
|
||
|
||
// 将复选框添加到按钮栏布局
|
||
toolBarLayout->addWidget(checkBoxTimeSeries);
|
||
toolBarLayout->addWidget(checkBoxDirection);
|
||
toolBarLayout->addStretch(); // 添加一个弹性空间，使按钮靠左排列
|
||
|
||
// 2. 创建TreeView
|
||
treewidgetResults = new QTreeWidget(this);
|
||
treewidgetResults->setColumnCount(2);
|
||
treewidgetResults->setHeaderLabels(QStringList() << u8"属性" << u8"值");
|
||
// 3. 将按钮栏和TreeView添加到主垂直布局中
|
||
mainLayout->addWidget(toolBarWidget);
|
||
mainLayout->addWidget(treewidgetResults);
|
||
|
||
// 4. （可选）设置布局的比例，例如让TreeView占据更多空间
|
||
mainLayout->setStretchFactor(toolBarWidget, 0); // 工具栏不拉伸
|
||
mainLayout->setStretchFactor(treewidgetResults, 1); // TreeView可拉伸
|
||
|
||
|
||
// 设置当前窗口的布局为主布局
|
||
this->setLayout(mainLayout);
|
||
|
||
}
|
||
|
||
void QImageCursorWidget::showCursor(QList<QgsMapToolIdentify::IdentifyResult> result)
|
||
{
|
||
maplayer_treewidgetItem.clear();
|
||
treewidgetResults->clear();
|
||
treewidgetResults->setColumnCount(2);
|
||
treewidgetResults->setHeaderLabels(QStringList() << u8"属性" << u8"值");
|
||
int32_t layerCount = result.count();
|
||
for (int32_t i = 0; i < layerCount; i++) {
|
||
if (result[i].mLayer->type() == Qgis::LayerType::Raster) {
|
||
|
||
}
|
||
else if (result[i].mLayer->type() == Qgis::LayerType::Mesh) {
|
||
|
||
}
|
||
}
|
||
this->populateTreeWidget(result);
|
||
}
|
||
|
||
void QImageCursorWidget::populateTreeWidget(const QList<QgsMapToolIdentify::IdentifyResult>& results)
|
||
{
|
||
for (const QgsMapToolIdentify::IdentifyResult& result : results) {
|
||
if (result.mLayer->type() == Qgis::LayerType::Raster) {
|
||
addFeature(qobject_cast<QgsRasterLayer*>(result.mLayer), result.mLabel, result.mAttributes, result.mDerivedAttributes, result.mFields, result.mFeature, result.mParams);
|
||
}
|
||
else if (result.mLayer->type() == Qgis::LayerType::Mesh) {
|
||
addFeature(qobject_cast<QgsMeshLayer*>(result.mLayer), result.mLabel, result.mAttributes, result.mDerivedAttributes);
|
||
}
|
||
else {
|
||
|
||
}
|
||
}
|
||
treewidgetResults->expandAll();
|
||
}
|
||
|
||
void QImageCursorWidget::addKeyValuePair(QTreeWidgetItem* parent, const QString& key, const QString& value)
|
||
{
|
||
QTreeWidgetItem* item = new QTreeWidgetItem(parent);
|
||
item->setText(0, key);
|
||
item->setText(1, value);
|
||
}
|
||
|
||
QTreeWidgetItem* QImageCursorWidget::layerItem(QgsMapLayer* layer)
|
||
{
|
||
|
||
QTreeWidgetItem* layItem = nullptr;
|
||
if (maplayer_treewidgetItem.contains(layer)) {
|
||
layItem = maplayer_treewidgetItem[layer];
|
||
treewidgetResults->addTopLevelItem(layItem);
|
||
}
|
||
else {
|
||
layItem = new QTreeWidgetItem(treewidgetResults);
|
||
layItem->setText(0, layer->name());
|
||
maplayer_treewidgetItem.insert(layer, layItem);
|
||
treewidgetResults->addTopLevelItem(layItem);
|
||
}
|
||
return layItem;
|
||
|
||
}
|
||
|
||
|
||
|
||
void QImageCursorWidget::addFeature(QgsRasterLayer* layer,
|
||
const QString& label, const QMap<QString, QString>& attributes,
|
||
const QMap<QString, QString>& derivedAttributes, const QgsFields& fields,
|
||
const QgsFeature& feature, const QMap<QString, QVariant>& params)
|
||
{
|
||
QTreeWidgetItem* layItem = layerItem(layer);
|
||
QTreeWidgetItem* labelNode = new QTreeWidgetItem();
|
||
labelNode->setText(0, label);
|
||
// attributes
|
||
for (auto it = attributes.begin(); it != attributes.end(); ++it) {
|
||
QTreeWidgetItem* tempnode = new QTreeWidgetItem();
|
||
tempnode->setText(0, it.key());
|
||
tempnode->setText(1, it.value());
|
||
labelNode->addChild(tempnode);
|
||
}
|
||
layItem->addChild(labelNode);
|
||
// derivedAttributes
|
||
QTreeWidgetItem* driverNode = new QTreeWidgetItem();
|
||
for (auto it = derivedAttributes.begin(); it != derivedAttributes.end(); ++it) {
|
||
QTreeWidgetItem* tempnode = new QTreeWidgetItem();
|
||
tempnode->setText(0, it.key());
|
||
tempnode->setText(1, it.value());
|
||
driverNode->addChild(tempnode);
|
||
}
|
||
layItem->addChild(driverNode);
|
||
}
|
||
|
||
void QImageCursorWidget::addFeature(QgsMeshLayer* layer, const QString& label, const QMap<QString, QString>& attributes, const QMap<QString, QString>& derivedAttributes)
|
||
{
|
||
QTreeWidgetItem* layItem = layerItem(layer);
|
||
QTreeWidgetItem* labelNode = new QTreeWidgetItem();
|
||
labelNode->setText(0, label);
|
||
// attributes
|
||
for (auto it = attributes.begin(); it != attributes.end(); ++it) {
|
||
QTreeWidgetItem* tempnode = new QTreeWidgetItem();
|
||
tempnode->setText(0, it.key());
|
||
tempnode->setText(1, it.value());
|
||
labelNode->addChild(tempnode);
|
||
}
|
||
layItem->addChild(labelNode);
|
||
// derivedAttributes
|
||
QTreeWidgetItem* driverNode = new QTreeWidgetItem();
|
||
driverNode->setText(0, u8"(derivedAttributes)");
|
||
for (auto it = derivedAttributes.begin(); it != derivedAttributes.end(); ++it) {
|
||
QTreeWidgetItem* tempnode = new QTreeWidgetItem();
|
||
tempnode->setText(0, it.key());
|
||
tempnode->setText(1, it.value());
|
||
driverNode->addChild(tempnode);
|
||
}
|
||
layItem->addChild(driverNode);
|
||
}
|
||
|
||
|
||
|