1 // Copyright (C) 2007-2013 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.
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_ImportGeomObjectOp.h"
25 #include "HYDROGUI_GeomObjectDlg.h"
27 #include "HYDROGUI_DataModel.h"
28 #include "HYDROGUI_Module.h"
29 #include "HYDROGUI_Tool.h"
30 #include "HYDROGUI_UpdateFlags.h"
32 #include <HYDROData_Obstacle.h>
33 #include <HYDROData_PolylineXY.h>
34 #include <HYDROData_Iterator.h>
38 #include <SalomeApp_Study.h>
40 #include <LightApp_Application.h>
41 #include <LightApp_UpdateFlags.h>
43 #include <SUIT_Desktop.h>
44 #include <SUIT_MessageBox.h>
48 HYDROGUI_ImportGeomObjectOp::HYDROGUI_ImportGeomObjectOp( HYDROGUI_Module* theModule,
51 : HYDROGUI_Operation( theModule ),
52 myOpType( theOpType ),
53 myGEOMOp( theGEOMOp ),
55 myIsToShowPanel( true )
57 if ( myOpType == ImportSelectedAsPolyline ) {
58 setName( tr( "IMPORT_GEOM_OBJECT_AS_POLYLINE" ) );
60 setName( tr( "IMPORT_GEOM_OBJECT_AS_OBSTACLE" ) );
64 HYDROGUI_ImportGeomObjectOp::~HYDROGUI_ImportGeomObjectOp()
68 void HYDROGUI_ImportGeomObjectOp::startOperation()
70 // Get GEOM objects to import
71 myGeomObjects.clear();
73 if ( myOpType == ImportSelectedAsObstacle ) {
75 HYDROGUI_Tool::GetSelectedGeomObjects( module(), getObstacleTypes() );
76 } else if ( myOpType == ImportSelectedAsPolyline ) {
78 HYDROGUI_Tool::GetSelectedGeomObjects( module(), getPolylineTypes() );
81 // Do not show the panel if more than one GEOM objects are selected
82 myIsToShowPanel = myIsToShowPanel && ( myGeomObjects.count() <= 1 );
84 HYDROGUI_Operation::startOperation();
86 HYDROGUI_GeomObjectDlg* aPanel = 0;
88 if ( myIsToShowPanel ) {
90 aPanel = ::qobject_cast<HYDROGUI_GeomObjectDlg*>( inputPanel() );
93 // Reset the panel state
99 // Pass the existing object names to the panel
100 QStringList anExistingNames;
101 if ( myOpType == ImportCreatedAsObstacle ||
102 myOpType == ImportSelectedAsObstacle ) {
103 anExistingNames = HYDROGUI_Tool::FindExistingObjectsNames( doc(), KIND_OBSTACLE );
104 } else if ( myOpType == ImportSelectedAsPolyline ) {
105 anExistingNames = HYDROGUI_Tool::FindExistingObjectsNames( doc(), KIND_POLYLINEXY );
108 aPanel->setObjectNames( anExistingNames );
116 // Activate GEOM module operation in case of the corresponding operation type
117 if ( myOpType == ImportCreatedAsObstacle && myGEOMOp > 0 ) {
118 LightApp_Application* anApp = module()->getApp();
120 connect( anApp, SIGNAL( operationFinished( const QString&, const QString&, const QStringList& ) ),
121 this, SLOT( onExternalOperationFinished( const QString&, const QString&, const QStringList& ) ) );
123 module()->getApp()->activateOperation( "Geometry", myGEOMOp );
128 void HYDROGUI_ImportGeomObjectOp::abortOperation()
130 LightApp_Application* anApp = module()->getApp();
132 anApp->disconnect( this );
135 closeExternalOperationDlg();
137 HYDROGUI_Operation::abortOperation();
140 void HYDROGUI_ImportGeomObjectOp::commitOperation()
142 closeExternalOperationDlg();
144 HYDROGUI_Operation::commitOperation();
147 bool HYDROGUI_ImportGeomObjectOp::processApply( int& theUpdateFlags,
148 QString& theErrorMsg )
150 // Get active SalomeApp_Study
151 SalomeApp_Study* aStudy =
152 dynamic_cast<SalomeApp_Study*>( module()->getApp()->activeStudy() );
157 // Check that GEOM objects list is not empty
158 if ( myGeomObjects.isEmpty() ) {
159 theErrorMsg = tr( "NO_GEOM_OBJECT_TO_IMPORT" );
163 QString anObjectName;
164 Handle(HYDROData_Entity) anObjectToEdit;
165 ObjectKind anObjectKind =
166 myOpType == ImportSelectedAsPolyline ? KIND_POLYLINEXY : KIND_OBSTACLE;
168 if ( myGeomObjects.count() == 1 ) {
170 HYDROGUI_GeomObjectDlg* aPanel = ::qobject_cast<HYDROGUI_GeomObjectDlg*>( inputPanel() );
173 anObjectName = aPanel->getObjectName().simplified();
174 if ( anObjectName.isEmpty() ) {
175 theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
179 // Get object to edit
180 QString anEditedName = aPanel->getEditedObjectName().simplified();
182 if ( !anEditedName.isEmpty() ) {
183 anObjectToEdit = HYDROGUI_Tool::FindObjectByName( module(), anEditedName, anObjectKind );
187 if( anObjectToEdit.IsNull() || anObjectToEdit->GetName() != anObjectName ) {
188 // check that there are no other objects with the same name in the document
189 Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), anObjectName/*, anObjectKind*/ );
190 if( !anObject.IsNull() ) {
191 theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( anObjectName );
199 // Get the GEOM object as SObject
200 foreach ( const QString& anEntry, myGeomObjects ) {
201 _PTR(SObject) aSObject( aStudy->studyDS()->FindObjectID( qPrintable(anEntry)) );
203 // Get the corresponding TopoDS_Shape
204 TopoDS_Shape aShape = GEOMBase::GetShapeFromIOR( aSObject->GetIOR().c_str() );
205 if ( aShape.IsNull() ) {
209 // Create/edit an object
210 Handle(HYDROData_Entity) anObject;
212 if ( anObjectToEdit.IsNull() ) {
213 if ( myOpType == ImportCreatedAsObstacle || myOpType == ImportSelectedAsObstacle ) {
214 anObject = doc()->CreateObject( KIND_OBSTACLE );
215 Handle(HYDROData_Obstacle) anObstacle = Handle(HYDROData_Obstacle)::DownCast( anObject );
216 anObstacle->SetFillingColor( HYDROData_Obstacle::DefaultFillingColor() );
217 anObstacle->SetBorderColor( HYDROData_Obstacle::DefaultBorderColor() );
218 } else if ( myOpType == ImportSelectedAsPolyline ) {
219 anObject = doc()->CreateObject( KIND_POLYLINEXY );
220 Handle(HYDROData_PolylineXY) aPolylineObj = Handle(HYDROData_PolylineXY)::DownCast( anObject );
221 aPolylineObj->SetWireColor( HYDROData_PolylineXY::DefaultWireColor() );
224 anObject = anObjectToEdit;
228 if ( anObjectName.isEmpty() ) {
229 QString aName = QString::fromStdString( aSObject->GetName() );
230 anObjectName = HYDROGUI_Tool::GenerateObjectName(
231 module(), aName, QStringList(), true );
233 if ( anObject->GetName() != anObjectName ) {
234 anObject->SetName( anObjectName );
237 anObjectName.clear();
240 if ( myOpType == ImportCreatedAsObstacle || myOpType == ImportSelectedAsObstacle ) {
241 Handle(HYDROData_Obstacle) anObstacle = Handle(HYDROData_Obstacle)::DownCast( anObject );
242 anObstacle->SetShape3D( aShape );
244 } else if ( myOpType == ImportSelectedAsPolyline ) {
245 Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast( anObject );
246 anIsOk = aPolyline->ImportShape( aShape );
248 /* TODO: check it before start operation
249 if ( anIsOk && !aPolyline->IsEditable() )
251 anIsOk = SUIT_MessageBox::question( module()->getApp()->desktop(),
252 tr( "POLYLINE_IS_UNRECOGNIZED_TLT" ),
253 tr( "POLYLINE_IS_UNRECOGNIZED_MSG" ),
254 QMessageBox::Yes | QMessageBox::No,
255 QMessageBox::No ) == QMessageBox::Yes;
256 setPrintErrorMessage( anIsOk );
261 // Check operation status
264 module()->setIsToUpdate( anObject );
265 theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
273 HYDROGUI_InputPanel* HYDROGUI_ImportGeomObjectOp::createInputPanel() const
275 HYDROGUI_InputPanel* aPanel = 0;
276 if ( myIsToShowPanel ) {
277 QString anObjectTypeName =
278 myOpType == ImportSelectedAsPolyline ? tr("DEFAULT_POLYLINE_NAME") :
279 tr("DEFAULT_OBSTACLE_NAME");
280 aPanel = new HYDROGUI_GeomObjectDlg( module(), getName(), anObjectTypeName );
286 void HYDROGUI_ImportGeomObjectOp::updateDefaultName()
289 HYDROGUI_GeomObjectDlg* aPanel = ::qobject_cast<HYDROGUI_GeomObjectDlg*>( inputPanel() );
294 // Set the current GEOM object name to the panel
295 if ( myGeomObjects.count() == 1 ) {
296 SalomeApp_Study* aStudy =
297 dynamic_cast<SalomeApp_Study*>( module()->getApp()->activeStudy() );
299 QString anEntry = myGeomObjects.first();
300 _PTR(SObject) aSObject( aStudy->studyDS()->FindObjectID( qPrintable(anEntry) ) );
302 aPanel->setDefaultName( QString::fromStdString(aSObject->GetName()) );
309 * Called when the operation perfomed by another module is finished.
310 * \param theModuleName the name of the module which perfomed the operation
311 * \param theOperationName the operation name
312 * \param theEntryList the list of the created objects entries
314 void HYDROGUI_ImportGeomObjectOp::onExternalOperationFinished(
315 const QString& theModuleName, const QString& theOperationName,
316 const QStringList& theEntryList )
318 // Process "Geometry" module operations with non-empty list of created objects only
319 if ( theModuleName != "Geometry" || theEntryList.isEmpty() ) {
323 // Store the operation name
324 myGEOMOpName = theOperationName;
326 // Store the geom objects entries list
327 myGeomObjects = theEntryList;
329 // Update the default name of the HYDRO object
333 void HYDROGUI_ImportGeomObjectOp::closeExternalOperationDlg()
335 if ( myGEOMOpName.isEmpty() ) {
339 SUIT_Desktop* aDesktop = module()->getApp()->desktop();
341 QList<QDialog*> aDialogs = aDesktop->findChildren<QDialog*>();
342 foreach ( QDialog* aDlg, aDialogs ) {
343 if ( typeid(*aDlg).name() == myGEOMOpName ) {
351 QList<GEOM::shape_type> HYDROGUI_ImportGeomObjectOp::getObstacleTypes()
353 QList<GEOM::shape_type> aTypes;
355 aTypes << GEOM::COMPOUND << GEOM::COMPOUND << GEOM::SOLID <<
356 GEOM::SHELL << GEOM::FACE << GEOM::SHAPE;
361 QList<GEOM::shape_type> HYDROGUI_ImportGeomObjectOp::getPolylineTypes()
363 QList<GEOM::shape_type> aTypes;
365 aTypes << GEOM::WIRE << GEOM::EDGE;