230 lines
7.5 KiB
C++
230 lines
7.5 KiB
C++
|
#include <qgspoint.h>
|
|||
|
|
#include "LampWindDataItem.h"
|
|||
|
|
#include "BaseTool.h"
|
|||
|
|
#include <QFileInfo>
|
|||
|
|
#include "ManualLabelToolWidget.h"
|
|||
|
|
#include <QgsMeshLayer.h>
|
|||
|
|
#include <qgsmeshdataprovider.h>
|
|||
|
|
#include <qgsmeshlayerutils.h>
|
|||
|
|
#include <qgsmeshdataset.h>
|
|||
|
|
#include <qgsproject.h>
|
|||
|
|
#include <qgsvectorlayer.h>
|
|||
|
|
#include <qgsmeshrenderersettings.h>
|
|||
|
|
#include <qgscoordinatereferencesystem.h>
|
|||
|
|
#include <qgsmapcanvas.h>
|
|||
|
|
#include <qgsfeature.h>
|
|||
|
|
#include <qgsgeometry.h>
|
|||
|
|
#include <qgsfield.h>
|
|||
|
|
#include <qgssymbol.h>
|
|||
|
|
#include <qgssinglesymbolrenderer.h>
|
|||
|
|
#include <qgslinesymbol.h>
|
|||
|
|
#include <qgsmarkersymbol.h>
|
|||
|
|
#include <qgsrenderer.h>
|
|||
|
|
#include <qgsmarkersymbollayer.h>
|
|||
|
|
#include <qgsmesheditor.h>
|
|||
|
|
#include <qgsmeshdatasetgroupstore.h>
|
|||
|
|
#include <qgsmeshdatasetgrouptreeview.h>
|
|||
|
|
#include <qgsvectorfieldsymbollayer.h>
|
|||
|
|
#include <qgsmemoryproviderutils.h>
|
|||
|
|
#include <qgsgraduatedsymbolrenderer.h>
|
|||
|
|
#include <omp.h>
|
|||
|
|
#include "WidgetSettingClass.h"
|
|||
|
|
#include "ImageOperatorBase.h"
|
|||
|
|
#include "Wind2ERANc.h"
|
|||
|
|
#include <QProgressDialog>
|
|||
|
|
#include <QObject>
|
|||
|
|
#include <QString>
|
|||
|
|
|
|||
|
|
LampWindDataItem::LampWindDataItem(QTreeWidget* IntreeWidget)
|
|||
|
|
:LampDataItem(IntreeWidget)
|
|||
|
|
{
|
|||
|
|
timeArr.clear();
|
|||
|
|
Filename.clear();
|
|||
|
|
Filepath.clear();
|
|||
|
|
time_filepath.clear();
|
|||
|
|
selectItems.clear();
|
|||
|
|
this->treeWidget = IntreeWidget;
|
|||
|
|
this->ItemType = u8"LampWind";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
LampWindDataItem::~LampWindDataItem()
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void LampWindDataItem::InitTreeNode()
|
|||
|
|
{
|
|||
|
|
// 创建根节点，子节点,父窗体： QTreeWidget* treeWidget;
|
|||
|
|
// 其中要求子节点选中的时候，需要能够知道根节点的文本
|
|||
|
|
|
|||
|
|
this->setText(0, this->Filename);
|
|||
|
|
this->setText(1, u8"WindData");
|
|||
|
|
this->setCheckState(0, Qt::Unchecked);
|
|||
|
|
// 添加子节点
|
|||
|
|
for (const auto& key : timeArr.keys()) {
|
|||
|
|
QTreeWidgetItem* childItem1 = new QTreeWidgetItem(this); // 指定父节点为rootItem
|
|||
|
|
childItem1->setText(0, key);
|
|||
|
|
childItem1->setText(1, "");
|
|||
|
|
childItem1->setCheckState(0, Qt::Unchecked); // 在第一列添加复选框 [3](@ref)
|
|||
|
|
}
|
|||
|
|
//treeWidget->expandAll();
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void LampWindDataItem::showData(ManualLabelToolWidget* mainWidget)
|
|||
|
|
{
|
|||
|
|
// 获取当前节点的setCheckState
|
|||
|
|
if (this->checkState(0) == Qt::Checked) {
|
|||
|
|
for (int32_t i = 0; i < this->childCount(); i++) {
|
|||
|
|
int32_t tid = timeArr[this->child(i)->text(0)];
|
|||
|
|
this->ShowWindData(mainWidget, tid, this->child(i)->text(0));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
for (int32_t i = 0; i < this->childCount(); i++) {
|
|||
|
|
if (this->child(i)->checkState(0) == Qt::Checked) {
|
|||
|
|
int32_t tid = timeArr[this->child(i)->text(0)];
|
|||
|
|
this->ShowWindData(mainWidget, tid, this->child(i)->text(0));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
QString LampWindDataItem::getDescription()
|
|||
|
|
{
|
|||
|
|
QString starttimestr = Timenanosecond2timeStr(info.firstTimestamp);
|
|||
|
|
QString endtimestr = Timenanosecond2timeStr(info.lastTimestamp);
|
|||
|
|
|
|||
|
|
return QString(u8"FilePath: %1\nFileName: %2\nHeight:%3 \nWidth:%4 \nNum:%5, \nfirstTs:%6, \nlastTs:%7, \nminLon:%8, \nmaxLon:%9, \nminLat:%10, \nmaxLat:%11, \nESPG:%12, \nTransX:%13, %14, %15, \nTransY:%15, %16, %17, \nSize:%18 \n")
|
|||
|
|
.arg(this->Filepath).arg(this->Filename)
|
|||
|
|
.arg(info.Height).arg(info.Width).arg(info.Num)
|
|||
|
|
.arg(starttimestr).arg(endtimestr)
|
|||
|
|
.arg(info.minLon).arg(info.maxLon).arg(info.minLat).arg(info.maxLat)
|
|||
|
|
.arg(info.ESPGCODE)
|
|||
|
|
.arg(info.T11).arg(info.T12).arg(info.T13)
|
|||
|
|
.arg(info.T21).arg(info.T22).arg(info.T23)
|
|||
|
|
.arg(info.fileSize);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void LampWindDataItem::setSelectItems(QList<QString> selectItemName)
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void LampWindDataItem::OpenFileData(QString path)
|
|||
|
|
{
|
|||
|
|
// 读取数据
|
|||
|
|
this->info = getDataFileInfo(path.toUtf8().constData());
|
|||
|
|
int64_t* timeArr = new int64_t[this->info.Num];
|
|||
|
|
|
|||
|
|
QString ncfilename = QFileInfo(path).completeBaseName();
|
|||
|
|
QProgressDialog* progressDialog = new QProgressDialog(this->treeWidget);
|
|||
|
|
progressDialog->setWindowTitle( (u8"请稍候")); // 设置对话框标题
|
|||
|
|
progressDialog->setLabelText( (u8"正在处理中...")); // 设置说明文字
|
|||
|
|
progressDialog->resize(150, 50);
|
|||
|
|
|
|||
|
|
|
|||
|
|
// 生成缓存文件地址
|
|||
|
|
FileCacheFolderPath = WidgetSettingClass::instance().getTempFolder() + "\\"+ ncfilename;
|
|||
|
|
QDir dir;
|
|||
|
|
dir.mkpath(FileCacheFolderPath);
|
|||
|
|
ERA5NetCDFConverter cover;
|
|||
|
|
if (get_WindDataFileTimeArr(path.toUtf8().constData(), this->info, timeArr) == 0) {
|
|||
|
|
this->timeArr.clear();
|
|||
|
|
progressDialog->setRange(0, this->info.Num); // 设置进度范围，例如从0到100 [3,4](@ref)
|
|||
|
|
for (int64_t i = 0; i < this->info.Num; i++) {
|
|||
|
|
QString timeflag = Timenanosecond2timeStr(timeArr[i]);
|
|||
|
|
QString timefilename= QString(u8"\\mesh_wind_%1.nc").arg(timeflag.replace(":", "").replace(".", "M"));
|
|||
|
|
|
|||
|
|
QString temppath = FileCacheFolderPath + "\\" + timefilename;
|
|||
|
|
time_filepath.insert(i, temppath);
|
|||
|
|
cover.convert_to_era5_netcdf(
|
|||
|
|
path.toUtf8().constData(),
|
|||
|
|
temppath.toUtf8().constData(),
|
|||
|
|
i
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
this->timeArr.insert(QString(u8"%1").arg(timeflag),i);
|
|||
|
|
if (i % 30 == 0) {
|
|||
|
|
progressDialog->setValue(i);
|
|||
|
|
progressDialog->show();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}else{}
|
|||
|
|
|
|||
|
|
this->hashCode = pathToHashCode_Qt(path);
|
|||
|
|
|
|||
|
|
this->Filepath = path;
|
|||
|
|
this->Filename=QFileInfo(path).fileName();
|
|||
|
|
progressDialog->close();
|
|||
|
|
delArrPtr(timeArr);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
void LampWindDataItem::ShowWindData(ManualLabelToolWidget* mainWidget, int64_t tid, QString timeStr)
|
|||
|
|
{
|
|||
|
|
// 创建缓存文件
|
|||
|
|
//QString temppath = WidgetSettingClass::instance().getTempFolder() + QString(u8"\\mesh_wind_%1.nc").arg(timeStr.replace(":", "").replace(".", "M"));
|
|||
|
|
|
|||
|
|
QgsMapCanvas* map_canvas = mainWidget->map_canvas; // 地图
|
|||
|
|
QString url = time_filepath[tid];
|
|||
|
|
QgsMeshLayer* meshLayer = new QgsMeshLayer(url, timeStr, "mdal");
|
|||
|
|
//qDebug() << temppath;
|
|||
|
|
QString EPSGStr = QString(u8"EPSG:%1").arg(info.ESPGCODE);
|
|||
|
|
// 获取数据提供者
|
|||
|
|
QgsMeshDataProvider* dp = meshLayer->dataProvider();
|
|||
|
|
if (!dp) {
|
|||
|
|
qDebug() << "Failed to get data provider!";
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
meshLayer->setCrs(QgsCoordinateReferenceSystem(EPSGStr));
|
|||
|
|
// 获取数据集组数量
|
|||
|
|
int gprCount = dp->datasetGroupCount();
|
|||
|
|
int targetVectorIndex = -1; // 初始化为-1，表示未找到
|
|||
|
|
|
|||
|
|
// 遍历所有数据集组，寻找第一个矢量数据集
|
|||
|
|
for (int i = 0; i < gprCount; ++i) {
|
|||
|
|
QgsMeshDatasetIndex index(i, 0); // 组索引i，时间步0
|
|||
|
|
QgsMeshDatasetGroupMetadata meta = dp->datasetGroupMetadata(index);
|
|||
|
|
|
|||
|
|
bool isVector = meta.isVector();
|
|||
|
|
QString name = meta.name();
|
|||
|
|
|
|||
|
|
if (isVector) {
|
|||
|
|
targetVectorIndex = i;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果找到了矢量数据集组
|
|||
|
|
if (targetVectorIndex != -1) {
|
|||
|
|
// 获取当前的渲染器设置
|
|||
|
|
QgsMeshRendererSettings s = meshLayer->rendererSettings();
|
|||
|
|
|
|||
|
|
// 设置激活的矢量数据集组和标量数据集组
|
|||
|
|
s.setActiveVectorDatasetGroup(targetVectorIndex);
|
|||
|
|
s.setActiveScalarDatasetGroup(targetVectorIndex);
|
|||
|
|
|
|||
|
|
// 将修改后的设置应用回图层
|
|||
|
|
meshLayer->setRendererSettings(s);
|
|||
|
|
|
|||
|
|
qDebug() << "Successfully set active dataset group to index:" << targetVectorIndex;
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
qDebug() << "No vector dataset group found!";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
mainWidget->AddLayers(meshLayer);
|
|||
|
|
|
|||
|
|
map_canvas->setExtent(meshLayer->extent());
|
|||
|
|
map_canvas->refresh();
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|