1 // Copyright (C) 2014-2015 EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
10 // Lesser General Public License for more details.
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #include "HYDROGUI_ImportPolylineOp.h"
21 #include "HYDROGUI_DataModel.h"
22 #include "HYDROGUI_Module.h"
23 #include "HYDROGUI_UpdateFlags.h"
24 #include "HYDROGUI_Tool2.h"
25 #include <HYDROData_PolylineXY.h>
26 #include <HYDROData_Polyline3D.h>
27 #include <HYDROGUI_DataObject.h>
28 #include <HYDROData_Bathymetry.h>
29 #include <HYDROData_Iterator.h>
30 #include <HYDROData_ShapeFile.h>
31 #include <HYDROData_Tool.h>
33 #include <HYDROData_Profile.h>
35 #include <SUIT_Desktop.h>
36 #include <SUIT_FileDlg.h>
37 #include <LightApp_Application.h>
39 #include <QApplication>
42 #include <SUIT_MessageBox.h>
44 #include <HYDROGUI_ImportPolylineDlg.h>
45 #include <HYDROGUI_Module.h>
46 #include <HYDROGUI_Operation.h>
48 HYDROGUI_ImportPolylineOp::HYDROGUI_ImportPolylineOp( HYDROGUI_Module* theModule )
49 : HYDROGUI_Operation( theModule )
51 setName( tr( "IMPORT_POLYLINE" ) );
54 HYDROGUI_ImportPolylineOp::~HYDROGUI_ImportPolylineOp()
58 void HYDROGUI_ImportPolylineOp::startOperation()
60 HYDROGUI_Operation::startOperation();
62 myFileDlg = new SUIT_FileDlg( module()->getApp()->desktop(), true );
63 myFileDlg->setWindowTitle( getName() );
64 myFileDlg->setFileMode( SUIT_FileDlg::ExistingFiles );
65 myFileDlg->setNameFilter( tr("POLYLINE_FILTER") );
67 connect( myFileDlg, SIGNAL( accepted() ), this, SLOT( onApply() ) );
68 connect( myFileDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
73 NCollection_Sequence<Handle(HYDROData_Entity)> HYDROGUI_ImportPolylineOp::ImportPolyOp(
74 const QStringList& aFileNames, Handle(HYDROData_Document) theDocument,
75 HYDROGUI_Module* module, HYDROData_ShapeFile::ImportShapeType theShapeTypesToImport)
77 NCollection_Sequence<Handle(HYDROData_Entity)> importedEntities;
78 foreach (QString aFileName, aFileNames)
80 if ( aFileName.isEmpty() )
83 QString anExt = aFileName.split('.', QString::SkipEmptyParts).back();
85 bool importXY = false;
88 importXY = SUIT_MessageBox::question( module->getApp()->desktop(),
89 tr( "IMPORT_POLYLINE" ),
90 tr( "IMPORT_POLYLINE_XY_PART_ONLY" ),
91 QMessageBox::Yes | QMessageBox::No,
92 SUIT_MessageBox::Yes) == SUIT_MessageBox::Yes;
96 QApplication::setOverrideCursor( Qt::WaitCursor );
97 HYDROData_ShapeFile anImporter;
98 NCollection_IndexedDataMap<Handle(HYDROData_Entity), SHPObject*> theEntitiesToSHPObj;
99 NCollection_Sequence<Handle(HYDROData_Entity)> PolyEnt;
100 int aShapeTypeOfFile = -1;
101 int aStat = anImporter.OpenAndParse(aFileName);
102 QApplication::restoreOverrideCursor();
105 QApplication::setOverrideCursor( Qt::WaitCursor );
106 aStat = anImporter.ImportPolylines(theDocument, aFileName, theEntitiesToSHPObj, theShapeTypesToImport);
107 QApplication::restoreOverrideCursor();
108 for (int k=1;k<=theEntitiesToSHPObj.Extent();k++)
110 PolyEnt.Append(theEntitiesToSHPObj.FindKey(k));
112 if (aStat == 1 || aStat == 2)
115 QString aDBFFileName;
116 aDBFFileName = aFileName.simplified().replace( aFileName.simplified().size() - 4, 4, ".dbf");
117 bool DBF_Stat = anImporter.DBF_OpenDBF(aDBFFileName);
120 QStringList aFieldList = anImporter.DBF_GetFieldList();
121 int nbRecords = anImporter.DBF_GetNbRecords();
122 assert (theEntitiesToSHPObj.Extent() == nbRecords);
123 if (theEntitiesToSHPObj.Extent() == nbRecords)
125 int indNameAttrFound = -1;
127 bool bUseNameAttrFound = false;
128 for (QStringList::iterator it = aFieldList.begin(); it != aFieldList.end(); it++, k++)
129 if (QString::compare(*it, "name", Qt::CaseInsensitive) == 0)
131 indNameAttrFound = k;
135 if (indNameAttrFound != -1)
136 bUseNameAttrFound = SUIT_MessageBox::question( module->getApp()->desktop(),
137 tr( "IMPORT_POLYLINE" ),
138 tr( "IMPORT_POLYLINE_USE_NAME_ATTR" ),
139 QMessageBox::Yes | QMessageBox::No,
140 SUIT_MessageBox::Yes) == SUIT_MessageBox::Yes;
142 std::vector<std::vector<HYDROData_ShapeFile::DBF_AttrValue>> anAttrVV;
143 for (int i = 0; i < aFieldList.size(); i++)
145 std::vector<HYDROData_ShapeFile::DBF_AttrValue> anAttrV;
146 anAttrVV.push_back(anAttrV);
147 anImporter.DBF_GetAttributeList(i, anAttrVV[i] );
151 for (int i = 1; i <= theEntitiesToSHPObj.Extent(); i++)
153 Handle(HYDROData_PolylineXY) aPolylineXY = Handle(HYDROData_PolylineXY)::DownCast( theEntitiesToSHPObj.FindKey(i) );
154 QStringList aDBFinfo;
155 aDBFinfo << aFieldList; //first, the table header
156 for (int j = 0; j < aFieldList.size(); j++)
159 const HYDROData_ShapeFile::DBF_AttrValue& attrV = anAttrVV[j][i-1];
160 if (attrV.myFieldType == HYDROData_ShapeFile::DBF_FieldType_String)
161 aDBFinfo << attrV.myStrVal;
162 else if (attrV.myFieldType == HYDROData_ShapeFile::DBF_FieldType_Integer)
163 aDBFinfo << QString::number(attrV.myIntVal);
164 else if (attrV.myFieldType == HYDROData_ShapeFile::DBF_FieldType_Double)
165 aDBFinfo << QString::number(attrV.myDoubleVal);
169 assert (aDBFinfo.size() / 2 == aFieldList.size());
170 aPolylineXY->SetDBFInfo(aDBFinfo);
171 if (bUseNameAttrFound)
173 QString nameOfPoly = aDBFinfo[aDBFinfo.size() / 2 + indNameAttrFound];
174 if (!nameOfPoly.isEmpty())
175 aPolylineXY->SetName(nameOfPoly);
178 aPolylineXY->SetName("null_name_" + QString::number(indNULL));
187 if (anImporter.GetShapeType() == 13) //Polyline 3D
189 HYDROGUI_ImportPolylineDlg* aDLG = new HYDROGUI_ImportPolylineDlg( module->getApp()->desktop(), PolyEnt );
190 aDLG->setModal( true );
191 aDLG->setWindowTitle(tr("IMPORT_POLYLINE"));
192 //QApplication::restoreOverrideCursor();
193 if( aDLG->exec()==QDialog::Accepted )
195 NCollection_Sequence<Handle(HYDROData_Entity)> CheckedPolylines = aDLG->GetCheckedPolylines();
196 NCollection_IndexedDataMap<Handle(HYDROData_Entity), SHPObject*> CheckedEntitiesToSHPObj;
197 for (int k=1;k<=CheckedPolylines.Size();k++)
199 SHPObject* const* SHPObkP = theEntitiesToSHPObj.Seek(CheckedPolylines(k));
201 CheckedEntitiesToSHPObj.Add(CheckedPolylines(k), *SHPObkP);
203 NCollection_Sequence<Handle(HYDROData_Entity)> theEntitiesPoly3D;
204 anImporter.ImportPolylines3D(theDocument, CheckedEntitiesToSHPObj, theEntitiesPoly3D, theShapeTypesToImport);
205 PolyEnt.Append(theEntitiesPoly3D);
211 UpdateView(module, PolyEnt);
214 UpdateView(module, PolyEnt);
215 if (theShapeTypesToImport == HYDROData_ShapeFile::ImportShapeType_All) //if other flag = > no need to show this messagebox
216 SUIT_MessageBox::information(module->getApp()->desktop(),
217 tr( "IMPORT_POLYLINE" ), tr ("POLYGON_IMPORTED_AS_POLYLINE"));
222 if (theShapeTypesToImport == HYDROData_ShapeFile::ImportShapeType::ImportShapeType_Polygon)
223 aMess += tr ("POLYLINE_IMPORT_FAILED_AS_POLYGON") + ";\n";
225 aMess += tr ("POLYLINE_IMPORT_FAILED_AS_POLYLINE") + ";\n";
228 aMess += tr ("CANT_OPEN_SHP");
229 else if (aStat == -2)
230 aMess += tr ("CANT_OPEN_SHX");
232 aMess += tr ("SHAPE_TYPE_IS") + anImporter.GetShapeTypeName(aShapeTypeOfFile);
233 SUIT_MessageBox::warning( module->getApp()->desktop(), tr( "IMPORT_POLYLINE" ), aMess);
235 importedEntities.Append(PolyEnt);
237 else if ( anExt == "xy" || (importXY && anExt == "xyz"))
239 if (!HYDROData_Tool::importPolylineFromXYZ(aFileName, theDocument, true, importedEntities))
240 SUIT_MessageBox::warning( module->getApp()->desktop(), tr( "IMPORT_POLYLINE" ), tr( "NO_ONE_POLYLINE_IMPORTED" ));
242 else if (anExt == "xyz")
244 if (!HYDROData_Tool::importPolylineFromXYZ(aFileName, theDocument, false, importedEntities))
245 SUIT_MessageBox::warning( module->getApp()->desktop(), tr( "IMPORT_POLYLINE" ), tr( "NO_ONE_POLYLINE_IMPORTED" ));
248 return importedEntities;
251 void HYDROGUI_ImportPolylineOp::onApply()
259 QStringList aFileNames = myFileDlg->selectedFiles();
261 //QApplication::setOverrideCursor( Qt::WaitCursor );
264 ImportPolyOp(aFileNames, doc(), module(), HYDROData_ShapeFile::ImportShapeType_All);
266 if (!aFileNames.empty())
268 commitDocOperation();
270 module()->update( UF_Model | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init | UF_OCCViewer );
275 QApplication::restoreOverrideCursor();
278 void HYDROGUI_ImportPolylineOp::UpdateView( HYDROGUI_Module* module, NCollection_Sequence<Handle(HYDROData_Entity)>& anEntities)
280 size_t anActiveViewId = HYDROGUI_Tool::GetActiveGraphicsViewId( module );
281 if ( anActiveViewId == 0 )
282 anActiveViewId = HYDROGUI_Tool::GetActiveOCCViewId( module );
284 for (int i = 1; i <= anEntities.Size() ; i++)
286 anEntities(i)->Update();
287 module->setObjectVisible( anActiveViewId, anEntities(i), true );
288 module->setIsToUpdate( anEntities(i) );