修正了模型保存时,可能会替换内置模型

pull/1/head
剑古敛锋 2024-07-23 10:30:50 +08:00
parent 49781f86a8
commit 8f64808f65
1 changed files with 122 additions and 150 deletions

View File

@ -22,68 +22,60 @@
#include <vtkDataSetSurfaceFilter.h>
#include "MainWindow/MainWindow.h"
#include "MainWindow/SubWindowManager.h"
#include "Settings/BusAPI.h"
namespace MeshData
{
VTKdataExchange::VTKdataExchange(const QString &fileName, MeshOperation operation, GUI::MainWindow *mw, int modelId) : MeshThreadBase(fileName, operation, mw),
_fileName(fileName),
_meshData(MeshData::getInstance()),
_operation(operation),
_modelId(modelId)
namespace MeshData {
VTKdataExchange::VTKdataExchange(const QString& fileName, MeshOperation operation,
GUI::MainWindow* mw, int modelId)
: MeshThreadBase(fileName, operation, mw)
, _fileName(fileName)
, _meshData(MeshData::getInstance())
, _operation(operation)
, _modelId(modelId)
{
}
bool VTKdataExchange::read()
{
QFileInfo info(_fileName);
if (!info.exists())
if(!info.exists())
return false;
QString name = info.fileName();
QString path = info.filePath();
QString suffix = info.suffix().toLower();
vtkDataSet *dataset = nullptr;
QString name = info.fileName();
QString path = info.filePath();
QString suffix = info.suffix().toLower();
vtkDataSet* dataset = nullptr;
if (suffix == "vtk")
{
if(suffix == "vtk") {
dataset = readVTK();
}
else if (suffix == "stl")
{
} else if(suffix == "stl") {
dataset = readSTL();
}
else if (suffix == "dat")
{
} else if(suffix == "dat") {
dataset = readTecplot();
}
if (dataset != nullptr)
{
if(dataset != nullptr) {
// 网格面片三角化
emit _mainwindow->printMessage(Common::Message::Normal,u8"surface triangulation .....");
// vtkSmartPointer<vtkDataSetSurfaceFilter> surfaceFilter = vtkSmartPointer<vtkDataSetSurfaceFilter>::New();
// surfaceFilter->SetInputData(dataset);
// surfaceFilter->Update();
// dataset = surfaceFilter->GetOutput();
MeshKernal *k = new MeshKernal;
// emit _mainwindow->printMessage(Common::Message::Normal,
// u8"surface triangulation .....");
MeshKernal* k = new MeshKernal;
k->setName(name);
k->setPath(path);
if(QFile(path).exists()) {
QString appworkpath = Setting::BusAPI::instance()->getApplicationExePath();
if(path.contains(appworkpath)) {
QString emptyStr;
k->setPath(emptyStr);
}
}
int nKernal = _meshData->getKernalCount();
for (int iKernal = 0; iKernal < nKernal; ++iKernal)
{
if (!_threadRuning)
for(int iKernal = 0; iKernal < nKernal; ++iKernal) {
if(!_threadRuning)
return false;
MeshKernal *temp = _meshData->getKernalAt(iKernal);
if (temp->getPath() == path)
{ ///< MG same file update
MeshKernal* temp = _meshData->getKernalAt(iKernal);
if(temp->getPath() == path) { ///< MG same file update
_meshData->removeKernalAt(iKernal);
break;
}
}
k->setMeshData(dataset);
_meshData->appendMeshKernal(k);
return true;
@ -91,13 +83,13 @@ namespace MeshData
return false;
}
vtkDataSet *VTKdataExchange::readVTK()
vtkDataSet* VTKdataExchange::readVTK()
{
if (!_threadRuning)
if(!_threadRuning)
return nullptr;
QTextCodec *codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
_vtkReader = vtkSmartPointer<vtkDataSetReader>::New();
QTextCodec* codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
_vtkReader = vtkSmartPointer<vtkDataSetReader>::New();
_vtkReader->SetFileName(ba);
_vtkReader->SetReadAllColorScalars(false);
_vtkReader->SetReadAllScalars(false);
@ -109,37 +101,37 @@ namespace MeshData
return dataset;
}
vtkDataSet *VTKdataExchange::readSTL()
vtkDataSet* VTKdataExchange::readSTL()
{
if (!_threadRuning)
if(!_threadRuning)
return nullptr;
QTextCodec *codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
_stlReader = vtkSmartPointer<vtkSTLReader>::New();
_stlReader->SetFileName(ba);
_stlReader->SetMerging(true);
QTextCodec* codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
_stlReader = vtkSmartPointer<vtkSTLReader>::New();
_stlReader->SetFileName(_fileName.toLocal8Bit().constData());
// _stlReader->SetMerging(true);
_stlReader->Update();
return _stlReader->GetOutput();
}
vtkDataSet *VTKdataExchange::readTecplot()
vtkDataSet* VTKdataExchange::readTecplot()
{
if (!_threadRuning)
if(!_threadRuning)
return nullptr;
QTextCodec *codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
_tecplotReader = vtkSmartPointer<vtkTecplotReader>::New();
QTextCodec* codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
_tecplotReader = vtkSmartPointer<vtkTecplotReader>::New();
_tecplotReader->SetFileName(ba);
_tecplotReader->Update();
vtkMultiBlockDataSet *mudataset = _tecplotReader->GetOutput();
vtkMultiBlockDataSet* mudataset = _tecplotReader->GetOutput();
// blockDataSet = mudataset;//add for multi zone test
const int nblock = mudataset->GetNumberOfBlocks();
if (nblock <= 0)
const int nblock = mudataset->GetNumberOfBlocks();
if(nblock <= 0)
return nullptr;
vtkAppendFilter *filter = vtkAppendFilter::New();
for (int i = 0; i < nblock; ++i)
{
vtkDataSet *dataset = vtkDataSet::SafeDownCast(mudataset->GetBlock(i));
vtkAppendFilter* filter = vtkAppendFilter::New();
for(int i = 0; i < nblock; ++i) {
vtkDataSet* dataset = vtkDataSet::SafeDownCast(mudataset->GetBlock(i));
filter->AddInputData(dataset);
}
filter->Update(); // put multi zone together
@ -150,20 +142,15 @@ namespace MeshData
bool VTKdataExchange::write()
{
QFileInfo info(_fileName);
QString name = info.fileName();
QString path = info.filePath();
QString suffix = info.suffix().toLower();
QString name = info.fileName();
QString path = info.filePath();
QString suffix = info.suffix().toLower();
if (suffix == "vtk")
{
if(suffix == "vtk") {
return writeVTK();
}
else if (suffix == "stl")
{
} else if(suffix == "stl") {
return writeSTL();
}
else if (suffix == "dat")
{
} else if(suffix == "dat") {
return false;
}
return false;
@ -173,57 +160,51 @@ namespace MeshData
{
ModuleBase::ThreadTask::run();
bool result = false;
switch (_operation)
{
case MESH_READ:
emit showInformation(tr("Import VTK Mesh File From \"%1\"").arg(_fileName));
result = read();
setReadResult(result);
break;
case MESH_WRITE:
emit showInformation(tr("Export VTK Mesh File To \"%1\"").arg(_fileName));
result = write();
setWriteResult(result);
break;
switch(_operation) {
case MESH_READ:
emit showInformation(tr("Import VTK Mesh File From \"%1\"").arg(_fileName));
result = read();
setReadResult(result);
break;
case MESH_WRITE:
emit showInformation(tr("Export VTK Mesh File To \"%1\"").arg(_fileName));
result = write();
setWriteResult(result);
break;
}
defaultMeshFinished();
}
bool VTKdataExchange::writeVTK()
{
QTextCodec *codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
vtkUnstructuredGrid *ungird = vtkUnstructuredGrid::New();
QTextCodec* codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
vtkUnstructuredGrid* ungird = vtkUnstructuredGrid::New();
vtkSmartPointer<vtkAppendFilter> filter = vtkSmartPointer<vtkAppendFilter>::New();
if (_modelId > 0)
{
MeshKernal *kernal = _meshData->getKernalByID(_modelId);
if (kernal != nullptr)
{
vtkDataSet *dataset = kernal->getMeshData();
if (dataset != nullptr)
if(_modelId > 0) {
MeshKernal* kernal = _meshData->getKernalByID(_modelId);
if(kernal != nullptr) {
vtkDataSet* dataset = kernal->getMeshData();
if(dataset != nullptr)
filter->AddInputData(dataset);
}
}
else
{
} else {
const int n = _meshData->getKernalCount();
for (int i = 0; i < n; ++i)
{
if (!_threadRuning)
for(int i = 0; i < n; ++i) {
if(!_threadRuning)
return false;
MeshKernal *kernal = _meshData->getKernalAt(i);
if (kernal != nullptr)
{
vtkDataSet *dataset = kernal->getMeshData();
if (dataset != nullptr)
MeshKernal* kernal = _meshData->getKernalAt(i);
if(kernal != nullptr) {
vtkDataSet* dataset = kernal->getMeshData();
if(dataset != nullptr)
filter->AddInputData(dataset);
}
}
}
filter->Update();
ungird->CopyStructure(filter->GetOutput());
vtkSmartPointer<vtkUnstructuredGridWriter> writer = vtkSmartPointer<vtkUnstructuredGridWriter>::New();
writer->SetInputData(ungird);
// ungird->CopyStructure(filter->GetOutput());
vtkSmartPointer<vtkUnstructuredGridWriter> writer =
vtkSmartPointer<vtkUnstructuredGridWriter>::New();
writer->SetInputData(filter->GetOutput());
writer->SetFileTypeToBinary();
writer->SetFileName(_fileName.toLocal8Bit().constData());
writer->Write();
@ -232,67 +213,58 @@ namespace MeshData
bool VTKdataExchange::writeSTL()
{
QFile infile(_fileName);
if(infile.exists()){
if(infile.exists()) {
infile.remove();
}
qDebug()<<" VTKdataExchange::writeSTL start ";
QTextCodec *codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
vtkUnstructuredGrid *ungird = vtkUnstructuredGrid::New();
// vtkSmartPointer<vtkAppendFilter> filter = vtkSmartPointer<vtkAppendFilter>::New();
vtkSmartPointer<vtkGeometryFilter> cellToPointFilter = vtkSmartPointer<vtkGeometryFilter>::New();
if (_modelId > 0)
{
MeshKernal *kernal = _meshData->getKernalByID(_modelId);
if (kernal != nullptr)
{
vtkDataSet *dataset = kernal->getMeshData();
if (dataset != nullptr)
qDebug() << " VTKdataExchange::writeSTL start ";
QTextCodec* codec = QTextCodec::codecForName("GB18030");
QByteArray ba = codec->fromUnicode(_fileName);
vtkSmartPointer<vtkGeometryFilter> cellToPointFilter =
vtkSmartPointer<vtkGeometryFilter>::New();
if(_modelId > 0) {
MeshKernal* kernal = _meshData->getKernalByID(_modelId);
if(kernal != nullptr) {
vtkDataSet* dataset = kernal->getMeshData();
if(dataset != nullptr) {
cellToPointFilter->AddInputData(dataset);
}
}
}
else
{
} else {
const int n = _meshData->getKernalCount();
for (int i = 0; i < n; ++i)
{
if (!_threadRuning)
for(int i = 0; i < n; ++i) {
if(!_threadRuning)
return false;
MeshKernal *kernal = _meshData->getKernalAt(i);
if (kernal != nullptr)
{
vtkDataSet *dataset = kernal->getMeshData();
if (dataset != nullptr)
MeshKernal* kernal = _meshData->getKernalAt(i);
if(kernal != nullptr) {
vtkDataSet* dataset = kernal->getMeshData();
if(dataset != nullptr)
cellToPointFilter->AddInputData(dataset);
}
}
}
cellToPointFilter->Update();
// ungird->CopyStructure(filter->GetOutput());
// qDebug()<<"ungird , cell Count : "+QString::number(ungird->GetNumberOfCells());
vtkSmartPointer<vtkPolyData> inpolyData=vtkSmartPointer<vtkPolyData>::New();
inpolyData = cellToPointFilter->GetOutput();
if(nullptr==inpolyData){
qDebug()<<"inpolyData ,nullptr : ";
}else{
qDebug()<<"inpolyData , cell Count : "+QString::number(inpolyData->GetNumberOfCells());
vtkSmartPointer<vtkPolyData> inpolyData = vtkSmartPointer<vtkPolyData>::New();
inpolyData = cellToPointFilter->GetOutput();
if(nullptr == inpolyData) {
qDebug() << "inpolyData ,nullptr : ";
} else {
qDebug() << "inpolyData , cell Count : "
+ QString::number(inpolyData->GetNumberOfCells());
}
vtkSmartPointer<vtkSTLWriter> writer = vtkSmartPointer<vtkSTLWriter>::New();
writer->SetInputConnection(cellToPointFilter->GetOutputPort());
// writer->SetInputData(ungird);
writer->SetFileTypeToBinary();
writer->SetFileName(_fileName.toLocal8Bit().constData());
writer->Write();
qDebug()<<" VTKdataExchange::writeSTL finish !!!! "+ba;
qDebug() << " VTKdataExchange::writeSTL finish !!!! " + ba;
QFile outfile(_fileName);
return outfile.exists();// 判断文件是否存在
return outfile.exists(); // 判断文件是否存在
return true;
}
bool VTKdataExchange::writeTecplot()
{
return false;
}
}
} // namespace MeshData