1 // Copyright (C) 2007-2015 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License, or (at your option) any later version.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 #include "HYDROGUI_ImportPolylineOp.h"
25 #include "HYDROGUI_DataModel.h"
26 #include "HYDROGUI_Module.h"
27 #include "HYDROGUI_UpdateFlags.h"
28 #include "HYDROGUI_Tool.h"
29 #include <HYDROData_PolylineXY.h>
30 #include <HYDROData_Polyline3D.h>
31 #include <HYDROGUI_DataObject.h>
32 #include <HYDROData_Bathymetry.h>
33 #include <HYDROData_Iterator.h>
35 #include <HYDROData_Profile.h>
37 #include <SUIT_Desktop.h>
38 #include <SUIT_FileDlg.h>
39 #include <LightApp_Application.h>
41 #include <QApplication>
44 #include <QMessageBox>
47 HYDROGUI_ImportPolylineOp::HYDROGUI_ImportPolylineOp( HYDROGUI_Module* theModule )
48 : HYDROGUI_Operation( theModule )
50 setName( tr( "IMPORT_POLYLINE" ) );
53 HYDROGUI_ImportPolylineOp::~HYDROGUI_ImportPolylineOp()
57 void HYDROGUI_ImportPolylineOp::startOperation()
59 HYDROGUI_Operation::startOperation();
61 myFileDlg = new SUIT_FileDlg( module()->getApp()->desktop(), true );
62 myFileDlg->setWindowTitle( getName() );
63 myFileDlg->setFileMode( SUIT_FileDlg::ExistingFiles );
64 myFileDlg->setFilter( tr("POLYLINE_FILTER") );
66 connect( myFileDlg, SIGNAL( accepted() ), this, SLOT( onApply() ) );
67 connect( myFileDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
72 void HYDROGUI_ImportPolylineOp::onApply()
80 QString aFileName = myFileDlg->selectedFile();
81 if ( aFileName.isEmpty() )
87 QString anExt = aFileName.split('.', QString::SkipEmptyParts).back();
92 aHSHP = SHPOpen( aFileName.toAscii().data(), "rb" );
95 QApplication::setOverrideCursor( Qt::WaitCursor );
99 HYDROData_Iterator anIter( doc() );
101 QStringList anExistingNames;
102 std::vector<int> anAllowedIndexes;
103 for( ; anIter.More(); anIter.Next() )
104 anExistingNames.push_back(anIter.Current()->GetName());
106 QFileInfo aFileInfo(aFileName);
107 QString aBaseFileName = aFileInfo.baseName();
109 if (aHSHP->nShapeType == 3 || aHSHP->nShapeType == 23)
112 for (;anAllowedIndexes.size() < mySHPObjects.size();)
114 if (!anExistingNames.contains(aBaseFileName + "_PolyXY_" + QString::number(anInd)))
116 anAllowedIndexes.push_back(anInd);
123 for (size_t i = 0; i < mySHPObjects.size(); i++ )
125 ProcessSHPPolyXY(mySHPObjects[i], aBaseFileName, anAllowedIndexes[i]);
128 else if (aHSHP->nShapeType == 13)
131 for (;anAllowedIndexes.size() < mySHPObjects.size();)
133 if (!anExistingNames.contains(aBaseFileName + "_PolyXY_" + QString::number(anInd)) &&
134 !anExistingNames.contains(aBaseFileName + "_Poly3D_" + QString::number(anInd)) &&
135 !anExistingNames.contains(aBaseFileName + "_Bath_" + QString::number(anInd)))
137 anAllowedIndexes.push_back(anInd);
143 for (size_t i = 0; i < mySHPObjects.size(); i++ )
144 ProcessSHPPoly3D(mySHPObjects[i], aBaseFileName, anAllowedIndexes[i]);
146 commitDocOperation();
149 for (size_t i = 0; i < mySHPObjects.size(); i++ )
150 free (mySHPObjects[i]);
152 mySHPObjects.clear();
155 else if (anExt == "sx")
157 QFile aFile (aFileName);
158 aFile.open(QIODevice::ReadOnly);
162 QApplication::setOverrideCursor( Qt::WaitCursor );
168 commitDocOperation();
173 module()->update( UF_Model | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init );
175 QApplication::restoreOverrideCursor();
178 void HYDROGUI_ImportPolylineOp::ProcessSX()
180 Handle(HYDROData_PolylineXY) aPolylineXY = Handle(HYDROData_PolylineXY)::DownCast( doc()->CreateObject( KIND_POLYLINEXY ) );
182 Handle(HYDROData_Polyline3D) aPolylineObj = Handle(HYDROData_Polyline3D)::DownCast( doc()->CreateObject( KIND_POLYLINE ) );
184 Handle(HYDROData_Bathymetry) aBath = Handle(HYDROData_Bathymetry)::DownCast( doc()->CreateObject( KIND_BATHYMETRY ) );
185 HYDROData_Bathymetry::AltitudePoints aAPoints;
187 int aNSect = myCurveBlocks.size();
188 for ( int i = 0 ; i < aNSect ; i++ )
190 bool aSectClosure = true;
191 HYDROData_PolylineXY::SectionType aSectType = HYDROData_PolylineXY::SECTION_POLYLINE;
192 aPolylineXY->AddSection( TCollection_AsciiString(myCurveBlocks[i].myName.toStdString().c_str()), aSectType, myCurveBlocks[i].myIsConnected );
194 for ( int k = 0 ; k < myCurveBlocks[i].myXYZPoints.size() ; k+=3 )
196 HYDROData_PolylineXY::Point aSectPoint;
197 aSectPoint.SetX( myCurveBlocks[i].myXYZPoints[k].X() );
198 aSectPoint.SetY( myCurveBlocks[i].myXYZPoints[k].Y() );
199 aPolylineXY->AddPoint( i, aSectPoint );
201 aAPoints.Append(myCurveBlocks[i].myXYZPoints[k]);
204 QString aFileName = myFileDlg->selectedFile();
205 QFileInfo aFileInfo(aFileName);
206 QString aBaseFileName = aFileInfo.baseName();
207 QString aBathName = aBaseFileName + "_bath_1";
208 QString aPolyXYName = aBaseFileName + "_polyXY_1";
209 QString aPoly3DName = aBaseFileName + "_poly3D_1";
212 for (;HYDROGUI_Tool::FindObjectByName( module(), aBathName, KIND_BATHYMETRY) ||
213 HYDROGUI_Tool::FindObjectByName( module(), aPolyXYName, KIND_POLYLINEXY) ||
214 HYDROGUI_Tool::FindObjectByName( module(), aPoly3DName, KIND_POLYLINE);)
216 aBathName = aBaseFileName + "_bath_" + QString::number(anInd);
217 aPolyXYName = aBaseFileName + "_polyXY_" + QString::number(anInd);
218 aPoly3DName = aBaseFileName + "_poly3D_" + QString::number(anInd);
222 aPolylineXY->SetName( aPolyXYName );
223 aPolylineXY->SetWireColor(HYDROData_PolylineXY::DefaultWireColor());
224 aPolylineXY->Update();
226 aBath->SetAltitudePoints(aAPoints);
227 aBath->SetName( aBathName );
229 aPolylineObj->SetPolylineXY (aPolylineXY, false);
230 aPolylineObj->SetAltitudeObject(aBath);
232 aPolylineObj->SetBorderColor( HYDROData_Polyline3D::DefaultBorderColor() );
233 aPolylineObj->SetName( aPoly3DName );
235 aPolylineObj->Update();
237 size_t anActiveViewId = HYDROGUI_Tool::GetActiveGraphicsViewId( module() );
238 if ( anActiveViewId == 0 )
239 anActiveViewId = HYDROGUI_Tool::GetActiveOCCViewId( module() );
241 module()->setObjectVisible( anActiveViewId, aPolylineXY, true );
242 module()->setObjectVisible( anActiveViewId, aPolylineObj, true );
244 module()->setIsToUpdate( aPolylineObj );
247 void HYDROGUI_ImportPolylineOp::ProcessSHPPolyXY(SHPObject* anObj, QString theFileName, int theInd)
249 //if (anObj->nSHPType != SHPT_ARC && anObj->nSHPType != SHPT_ARCM)
251 Handle(HYDROData_PolylineXY) aPolylineXY = Handle(HYDROData_PolylineXY)::DownCast( doc()->CreateObject( KIND_POLYLINEXY ) );
253 int nParts = anObj->nParts;
254 for ( int i = 0 ; i < nParts ; i++ )
256 int StartIndex = anObj->panPartStart[i];
259 EndIndex = anObj->panPartStart[i + 1];
261 EndIndex = anObj->nVertices;
263 bool IsClosed = false;
264 HYDROData_PolylineXY::SectionType aSectType = HYDROData_PolylineXY::SECTION_POLYLINE;
265 if (anObj->padfX[StartIndex] == anObj->padfX[EndIndex - 1] &&
266 anObj->padfY[StartIndex] == anObj->padfY[EndIndex - 1] )
269 aPolylineXY->AddSection( TCollection_AsciiString( ("poly_section_" + QString::number(i)).data()->toAscii()), aSectType, true);
272 aPolylineXY->AddSection( TCollection_AsciiString( ("poly_section_" + QString::number(i)).data()->toAscii()), aSectType, false);
276 for ( int k = StartIndex; k < EndIndex ; k++ )
278 HYDROData_PolylineXY::Point aSectPoint;
279 aSectPoint.SetX( anObj->padfX[k] );
280 aSectPoint.SetY( anObj->padfY[k] );
281 aPolylineXY->AddPoint( i, aSectPoint );
286 aPolylineXY->SetWireColor( HYDROData_PolylineXY::DefaultWireColor() );
287 aPolylineXY->SetName( theFileName + "_PolyXY_" + QString::number(theInd) );
289 aPolylineXY->Update();
291 size_t anActiveViewId = HYDROGUI_Tool::GetActiveGraphicsViewId( module() );
292 if ( anActiveViewId == 0 )
293 anActiveViewId = HYDROGUI_Tool::GetActiveOCCViewId( module() );
295 module()->setObjectVisible( anActiveViewId, aPolylineXY, true );
297 module()->setIsToUpdate( aPolylineXY );
300 void HYDROGUI_ImportPolylineOp::ProcessSHPPoly3D(SHPObject* anObj, QString theFileName, int theInd)
302 Handle(HYDROData_PolylineXY) aPolylineXY = Handle(HYDROData_PolylineXY)::DownCast( doc()->CreateObject( KIND_POLYLINEXY ) );
304 Handle(HYDROData_Polyline3D) aPolylineObj = Handle(HYDROData_Polyline3D)::DownCast( doc()->CreateObject( KIND_POLYLINE ) );
306 Handle(HYDROData_Bathymetry) aBath = Handle(HYDROData_Bathymetry)::DownCast( doc()->CreateObject( KIND_BATHYMETRY ) );
307 HYDROData_Bathymetry::AltitudePoints aAPoints;
309 int nParts = anObj->nParts;
310 for ( int i = 0 ; i < nParts ; i++ )
312 //bool aSectClosure = true;
313 int StartIndex = anObj->panPartStart[i];
316 EndIndex = anObj->panPartStart[i + 1];
318 EndIndex = anObj->nVertices;
320 bool IsClosed = false;
321 HYDROData_PolylineXY::SectionType aSectType = HYDROData_PolylineXY::SECTION_POLYLINE;
322 if (anObj->padfX[StartIndex] == anObj->padfX[EndIndex - 1] &&
323 anObj->padfY[StartIndex] == anObj->padfY[EndIndex - 1] &&
324 anObj->padfZ[StartIndex] == anObj->padfZ[EndIndex - 1])
327 aPolylineXY->AddSection( TCollection_AsciiString( ("poly_section_" + QString::number(i)).data()->toAscii()), aSectType, true );
330 aPolylineXY->AddSection( TCollection_AsciiString( ("poly_section_" + QString::number(i)).data()->toAscii()), aSectType, false );
334 for ( int k = StartIndex ; k < EndIndex ; k++ )
336 HYDROData_PolylineXY::Point aSectPoint;
337 aSectPoint.SetX( anObj->padfX[k] );
338 aSectPoint.SetY( anObj->padfY[k] );
339 aPolylineXY->AddPoint( i, aSectPoint );
340 aAPoints.Append(gp_XYZ (anObj->padfX[k], anObj->padfY[k], anObj->padfZ[k]));
345 QString aBathName = theFileName + "_bath_" + QString::number(theInd);
346 QString aPolyXYName = theFileName + "_polyXY_" + QString::number(theInd);
347 QString aPoly3DName = theFileName + "_poly3D_" + QString::number(theInd);
349 aPolylineXY->SetName( aPolyXYName );
350 aPolylineXY->SetWireColor(HYDROData_PolylineXY::DefaultWireColor());
351 aPolylineXY->Update();
353 aBath->SetAltitudePoints(aAPoints);
354 aBath->SetName( aBathName );
356 aPolylineObj->SetPolylineXY (aPolylineXY, false);
357 aPolylineObj->SetAltitudeObject(aBath);
359 aPolylineObj->SetBorderColor( HYDROData_Polyline3D::DefaultBorderColor() );
360 aPolylineObj->SetName( aPoly3DName );
362 aPolylineObj->Update();
364 size_t anActiveViewId = HYDROGUI_Tool::GetActiveGraphicsViewId( module() );
365 if ( anActiveViewId == 0 )
366 anActiveViewId = HYDROGUI_Tool::GetActiveOCCViewId( module() );
368 module()->setObjectVisible( anActiveViewId, aPolylineXY, true );
369 module()->setObjectVisible( anActiveViewId, aPolylineObj, true );
371 module()->setIsToUpdate( aPolylineObj );
375 bool HYDROGUI_ImportPolylineOp::Parse( QFile& theFile)
377 if ( !theFile.isOpen() )
384 myCurveBlocks.clear();
385 bool aTotStat = true;
387 aLine = theFile.readLine().simplified();
388 aList = aLine.split( ' ', QString::SkipEmptyParts );
390 for (;!theFile.atEnd();)
392 if (aList[0] == "B" && (aList[1] == "C" || aList[1] == "P" || aList[1] == "N"))
394 HYDROGUI_CurveBlock aCurveBlockInfo;
396 aCurveBlockInfo.myType = 1;
397 else if (aList[1] == "P")
398 aCurveBlockInfo.myType = 2;
399 else if (aList[1] == "N")
400 aCurveBlockInfo.myType = 2;
402 if (aList.size() == 9)
404 for (int j = 2; j < 8; j++)
405 aCurveBlockInfo.myRefCoords.push_back(aList[j].toDouble());
406 aCurveBlockInfo.myRefRatio = aList[8].toDouble();
412 aBLine = theFile.readLine().simplified();
413 aBList = aBLine.split( ' ', QString::SkipEmptyParts );
415 if (aBList[0] == "CP")
417 if (aBList.size() == 2 && (aBList[1] == "0" || aBList[1] == "1" || aBList[1] == "2"))
418 aCurveBlockInfo.myCurvePlane = aBList[1].toInt();
419 else if (aBList.size() == 3 && (aBList[1] == "0" || aBList[1] == "1") && (aBList[2] == "0" || aBList[2] == "1"))
421 aCurveBlockInfo.myIsClosed = aBList[1].toInt();
422 aCurveBlockInfo.myIsClosed = aBList[2].toInt();
426 for (int j = 1; j < aBList.size(); j++)
427 aCurveBlockInfo.myAdditionalCurveInfo.push_back(aBList[j].toDouble());
430 if (aBList[0] == "CN")
432 for (int i = 1; i < aBList.size(); i++)
433 Name += aBList[i] + "_";
434 Name.remove(Name.size() - 1, 1);
435 aCurveBlockInfo.myName = Name;
437 } while (!theFile.atEnd() && aBLine[0] == 'C' );
443 if (aBList.size() >= 3 && aBLine[0] != 'B' && aBLine[0] != 'C') {
445 anXYZ.SetX (aBList[0].toDouble(&aStat));
446 aTotStat = aTotStat && aStat;
447 anXYZ.SetY (aBList[1].toDouble(&aStat));
448 aTotStat = aTotStat && aStat;
449 anXYZ.SetZ (aBList[2].toDouble(&aStat));
450 aTotStat = aTotStat && aStat;
452 aCurveBlockInfo.myXYZPoints.push_back(anXYZ);
454 aBLine = theFile.readLine().simplified();
455 aBList = aBLine.split( ' ', QString::SkipEmptyParts );
460 } while (!theFile.atEnd()/* && aBLine[0] == 'B'*/ );
462 myCurveBlocks.push_back(aCurveBlockInfo);
467 aLine = theFile.readLine().simplified();
468 aList = aLine.split( ' ', QString::SkipEmptyParts );
477 void HYDROGUI_ImportPolylineOp::Parse(SHPHandle theHandle)
480 mySHPObjects.clear();
481 SHPGetInfo( theHandle, NULL, &aShapeType, NULL, NULL );
482 if (aShapeType == 3 || aShapeType == 13 || aShapeType == 23)
484 for (int i = 0; i < theHandle->nRecords; i++)
485 mySHPObjects.push_back(SHPReadObject(theHandle, i));