Salome HOME
Merge from V5_1_main 14/05/2010
[modules/geom.git] / src / GEOMToolsGUI / GEOMToolsGUI.cxx
1 //  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
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.
10 //
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.
15 //
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
19 //
20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 // GEOM GEOMGUI : GUI for Geometry component
24 // File   : GEOMBase_Tools.cxx
25 // Author : Damien COQUERET, Open CASCADE S.A.S.
26 //
27 #include "GEOMToolsGUI.h"
28 #include "GEOMToolsGUI_DeleteDlg.h"
29
30 #include <GeometryGUI.h>
31 #include "GeometryGUI_Operations.h"
32 #include <GEOMBase.h>
33 #include <GEOM_Operation.h>
34 #include <GEOM_Displayer.h>
35
36 #include <SUIT_Session.h>
37 #include <SUIT_OverrideCursor.h>
38 #include <SUIT_MessageBox.h>
39 #include <SUIT_Tools.h>
40 #include <SUIT_FileDlg.h>
41 #include <SUIT_Desktop.h>
42 #include <SUIT_ViewModel.h>
43 #include <SUIT_ViewManager.h>
44
45 #include <SalomeApp_Application.h>
46 #include <SalomeApp_Study.h>
47 #include <LightApp_SelectionMgr.h>
48 #include <GEOMImpl_Types.hxx>
49
50 #include <SALOME_ListIO.hxx>
51 #include <SALOME_ListIteratorOfListIO.hxx>
52 #include <SALOME_Prs.h>
53
54 // QT Includes
55 #include <QApplication>
56 #include <QMap>
57 #include <QRegExp>
58
59 // OCCT Includes
60 #include <TCollection_AsciiString.hxx>
61
62 typedef QMap<QString, QString> FilterMap;
63 static QString lastUsedFilter;
64
65 //=======================================================================
66 // function : getFileName
67 // purpose  : Selection of a file name for Import/Export. Returns also
68 //            the selected file type code through <filter> argument.
69 //=======================================================================
70 static QString getFileName( QWidget*           parent,
71                             const QString&     initial,
72                             const FilterMap&   filterMap,
73                             const QStringList& filters,
74                             const QString&     caption,
75                             bool               open,
76                             QString&           format,
77                             bool               showCurrentDirInitially = false )
78 {
79   //QStringList filters;
80   QString aBrepFilter;
81   for ( FilterMap::const_iterator it = filterMap.begin(); it != filterMap.end(); ++it ) {
82     //filters.push_back( it.key() );
83     if ( it.key().contains( "BREP", Qt::CaseInsensitive ) )
84       aBrepFilter = it.key();
85   }
86
87   SUIT_FileDlg* fd = new SUIT_FileDlg( parent, open, true, true );
88   if ( !caption.isEmpty() )
89     fd->setWindowTitle( caption );
90
91   if ( !initial.isEmpty() )
92     fd->selectFile( initial );
93   
94   if ( showCurrentDirInitially && SUIT_FileDlg::getLastVisitedPath().isEmpty() )
95     fd->setDirectory( QDir::currentPath() );
96
97   fd->setFilters( filters );
98   
99   if ( !lastUsedFilter.isEmpty() && filterMap.contains( lastUsedFilter ) ) {
100     fd->selectFilter( lastUsedFilter );
101   }
102   else if ( !aBrepFilter.isEmpty() ) {
103     fd->selectFilter( aBrepFilter );
104   }
105
106   QString filename;
107   if ( fd->exec() == QDialog::Accepted ) {
108     filename = fd->selectedFile();
109     format = filterMap[fd->selectedFilter()];
110     lastUsedFilter = fd->selectedFilter();
111   }
112
113   delete fd;
114   qApp->processEvents();
115   return filename;
116 }
117
118 //=======================================================================
119 // function : getFileNames
120 // purpose  : Select list of files for Import operation. Returns also
121 //            the selected file type code through <format> argument.
122 //=======================================================================
123 static QStringList getFileNames( QWidget*           parent,
124                                  const QString&     initial,
125                                  const FilterMap&   filterMap,
126                                  const QString&     caption,
127                                  QString&           format,
128                                  bool               showCurrentDirInitially = false)
129 {
130   QString aBrepFilter;
131   QStringList allFilters;
132   QStringList filters;
133   QRegExp re( "\\((.*)\\)" );
134   re.setMinimal( true );
135   for ( FilterMap::const_iterator it = filterMap.begin(); it != filterMap.end(); ++it ) {
136     if ( it.value().contains( "BREP", Qt::CaseInsensitive ) && aBrepFilter.isEmpty() )
137       aBrepFilter = it.key();
138     filters.append( it.key() );
139     int pos = 0;
140     while ( re.indexIn( it.key(), pos ) >= 0 ) {
141       QString f = re.cap(1);
142       pos = re.pos() + f.length() + 2;
143       allFilters.append( f.simplified() );
144     }
145   }
146   filters.append( QObject::tr( "GEOM_ALL_IMPORT_FILES" ).arg( allFilters.join( " " ) ) );
147   
148   SUIT_FileDlg fd( parent, true, true, true );
149   fd.setFileMode( SUIT_FileDlg::ExistingFiles );     
150   if ( !caption.isEmpty() )
151     fd.setWindowTitle( caption );
152   if ( !initial.isEmpty() )
153     fd.selectFile( initial );
154   
155   if ( showCurrentDirInitially && SUIT_FileDlg::getLastVisitedPath().isEmpty() )
156     fd.setDirectory( QDir::currentPath() );
157   
158   fd.setFilters( filters );
159   
160   if ( !lastUsedFilter.isEmpty() && filterMap.contains( lastUsedFilter ) )
161     fd.selectFilter( lastUsedFilter );
162   else if ( !aBrepFilter.isEmpty() )
163     fd.selectFilter( aBrepFilter );
164
165   QStringList filenames;
166   if ( fd.exec() ) {
167     filenames = fd.selectedFiles();
168     format = filterMap.contains( fd.selectedFilter() ) ? filterMap[ fd.selectedFilter() ] : QString();
169     lastUsedFilter = fd.selectedFilter();
170   }
171   qApp->processEvents();
172   return filenames;
173 }
174
175 //=======================================================================
176 // function : getParentComponent
177 // purpose  : Get object's parent component entry
178 //=======================================================================
179 static QString getParentComponent( _PTR( SObject ) obj )
180 {
181   if ( obj ) {
182     _PTR(SComponent) comp = obj->GetFatherComponent();
183     if ( comp )
184       return QString( comp->GetID().c_str() );
185   }
186   return QString();
187 }
188
189 //=====================================================================================
190 // function : inUse
191 // purpose  : check if the object(s) passed as the the second arguments are used
192 //            by the other objects in the study
193 //=====================================================================================
194 static bool inUse( _PTR(Study) study, const QString& component, const QMap<QString,QString>& objects )
195 {
196   _PTR(SObject) comp = study->FindObjectID( component.toLatin1().data() );
197   if ( !comp )
198     return false;
199
200   // collect all GEOM objects being deleted
201   QMap<QString, GEOM::GEOM_Object_var> gobjects;
202   QMap<QString, QString>::ConstIterator oit;
203   std::list<_PTR(SObject)> aSelectedSO;
204   for ( oit = objects.begin(); oit != objects.end(); ++oit ) {
205     _PTR(SObject) so = study->FindObjectID( oit.key().toLatin1().data() );
206     if ( !so )
207       continue;
208     aSelectedSO.push_back(so);
209     CORBA::Object_var corbaObj_rem = GeometryGUI::ClientSObjectToObject( so );
210     GEOM::GEOM_Object_var geomObj_rem = GEOM::GEOM_Object::_narrow( corbaObj_rem );
211     if( CORBA::is_nil( geomObj_rem ) ) 
212       continue;
213     gobjects.insert( oit.key(), geomObj_rem );
214   }
215
216   // Search References with other Modules
217   std::list< _PTR(SObject) >::iterator itSO = aSelectedSO.begin();
218   for ( ; itSO != aSelectedSO.end(); ++itSO ) {
219     std::vector<_PTR(SObject)> aReferences = study->FindDependances( *itSO  );    
220     int aRefLength = aReferences.size();
221     if (aRefLength) {
222       for (int i = 0; i < aRefLength; i++) {
223         _PTR(SObject) firstSO( aReferences[i] );
224         _PTR(SComponent) aComponent = firstSO->GetFatherComponent();
225         QString type = aComponent->ComponentDataType().c_str();
226         if ( type == "SMESH" )
227           return true;
228       }
229     }
230   }
231
232   // browse through all GEOM data tree
233   _PTR(ChildIterator) it ( study->NewChildIterator( comp ) );
234   for ( it->InitEx( true ); it->More(); it->Next() ) {
235     _PTR(SObject) child( it->Value() );
236     CORBA::Object_var corbaObj = GeometryGUI::ClientSObjectToObject( child );
237     GEOM::GEOM_Object_var geomObj = GEOM::GEOM_Object::_narrow( corbaObj );
238     if( CORBA::is_nil( geomObj ) ) 
239       continue;
240
241     GEOM::ListOfGO_var list = geomObj->GetDependency();
242     if( list->length() == 0 ) 
243       continue;
244
245     for( int i = 0; i < list->length(); i++ ) {
246       bool depends = false;
247       bool deleted = false;
248       QMap<QString, GEOM::GEOM_Object_var>::Iterator git;
249       for ( git = gobjects.begin(); git != gobjects.end() && ( !depends || !deleted ); ++git ) {
250         depends = depends || list[i]->_is_equivalent( *git );
251         deleted = deleted || git.key() == child->GetID().c_str() ;//geomObj->_is_equivalent( *git );
252       }
253       if ( depends && !deleted )
254         return true;
255     }
256   }
257   return false;
258 }
259
260
261 //=======================================================================
262 // function : GEOMToolsGUI()
263 // purpose  : Constructor
264 //=======================================================================
265 GEOMToolsGUI::GEOMToolsGUI( GeometryGUI* parent )
266 : GEOMGUI( parent )
267 {
268 }
269
270
271 //=======================================================================
272 // function : ~GEOMToolsGUI()
273 // purpose  : Destructor
274 //=======================================================================
275 GEOMToolsGUI::~GEOMToolsGUI()
276 {
277 }
278
279
280 //=======================================================================
281 // function : OnGUIEvent()
282 // purpose  :
283 //=======================================================================
284 bool GEOMToolsGUI::OnGUIEvent(int theCommandID, SUIT_Desktop* parent)
285 {
286   getGeometryGUI()->EmitSignalDeactivateDialog();
287
288   switch ( theCommandID ) {
289   case GEOMOp::OpDelete:         // EDIT - DELETE
290     OnEditDelete();
291     break;
292   case GEOMOp::OpImport:         // FILE - IMPORT
293     Import();
294     break;
295   case GEOMOp::OpExport:         // FILE - EXPORT
296     Export();
297     break;
298   case GEOMOp::OpCheckGeom:      // TOOLS - CHECK GEOMETRY
299     OnCheckGeometry();
300     break;
301   case GEOMOp::OpSelectVertex:   // POPUP - SELECT ONLY - VERTEX
302     OnSelectOnly( GEOM_POINT );
303     break;
304   case GEOMOp::OpSelectEdge:     // POPUP - SELECT ONLY - EDGE
305     OnSelectOnly( GEOM_EDGE );
306     break;
307   case GEOMOp::OpSelectWire:     // POPUP - SELECT ONLY - WIRE
308     OnSelectOnly( GEOM_WIRE );
309     break;
310   case GEOMOp::OpSelectFace:     // POPUP - SELECT ONLY - FACE
311     OnSelectOnly( GEOM_FACE );
312     break;
313   case GEOMOp::OpSelectShell:    // POPUP - SELECT ONLY - SHELL
314     OnSelectOnly( GEOM_SHELL );
315     break;
316   case GEOMOp::OpSelectSolid:    // POPUP - SELECT ONLY - SOLID
317     OnSelectOnly( GEOM_SOLID );
318     break;
319   case GEOMOp::OpSelectCompound: // POPUP - SELECT ONLY - COMPOUND
320     OnSelectOnly( GEOM_COMPOUND );
321     break;
322   case GEOMOp::OpSelectAll:      // POPUP - SELECT ONLY - SELECT ALL
323     OnSelectOnly( GEOM_ALLOBJECTS );
324     break;
325   case GEOMOp::OpRename:         // POPUP - RENAME
326     OnRename();
327     break;
328   case GEOMOp::OpDeflection:     // POPUP - DEFLECTION ANGLE
329     OnDeflection();
330     break;
331   case GEOMOp::OpColor:          // POPUP - COLOR
332     OnColor();
333     break;
334   case GEOMOp::OpTransparency:   // POPUP - TRANSPARENCY
335     OnTransparency();
336     break;
337   case GEOMOp::OpIsos:           // POPUP - ISOS
338     OnNbIsos();
339     break;
340   case GEOMOp::OpAutoColor:      // POPUP - AUTO COLOR
341     OnAutoColor();
342     break;
343   case GEOMOp::OpNoAutoColor:    // POPUP - DISABLE AUTO COLOR
344     OnDisableAutoColor();
345     break;
346   case GEOMOp::OpShowChildren:   // POPUP - SHOW CHILDREN
347   case GEOMOp::OpHideChildren:   // POPUP - HIDE CHILDREN
348     OnShowHideChildren( theCommandID == GEOMOp::OpShowChildren );
349     break;
350   case GEOMOp::OpPointMarker:    // POPUP - POINT MARKER
351     OnPointMarker();
352     break;
353   default:
354     SUIT_Session::session()->activeApplication()->putInfo(tr("GEOM_PRP_COMMAND").arg(theCommandID));
355     break;
356   }
357   return true;
358 }
359
360
361 //===============================================================================
362 // function : OnEditDelete()
363 // purpose  :
364 //===============================================================================
365 void GEOMToolsGUI::OnEditDelete()
366 {
367   SALOME_ListIO selected;
368   SalomeApp_Application* app =
369     dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
370   if ( !app )
371     return;
372
373   LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
374   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
375   if ( !aSelMgr || !appStudy )
376     return;
377
378   // get selection
379   aSelMgr->selectedObjects( selected, "ObjectBrowser", false );
380   if ( selected.IsEmpty() )
381     return;
382
383   _PTR(Study) aStudy = appStudy->studyDS();
384   
385   // check if study is locked
386   if ( _PTR(AttributeStudyProperties)( aStudy->GetProperties() )->IsLocked() ) {
387     SUIT_MessageBox::warning( app->desktop(),
388                               tr("WRN_WARNING"),
389                               tr("WRN_STUDY_LOCKED") );
390     return; // study is locked
391   }
392   
393   // get GEOM component
394   CORBA::String_var geomIOR = app->orb()->object_to_string( GeometryGUI::GetGeomGen() );
395   QString geomComp = getParentComponent( aStudy->FindObjectIOR( geomIOR.in() ) );
396
397   // check each selected object: if belongs to GEOM, if not reference...
398   QMap<QString,QString> toBeDeleted;
399   QMap<QString,QString> allDeleted;
400   bool isComponentSelected = false;
401   for ( SALOME_ListIteratorOfListIO It( selected ); It.More(); It.Next() ) {
402     Handle(SALOME_InteractiveObject) anIObject = It.Value();
403     if ( !anIObject->hasEntry() )
404       continue; // invalid object
405     // ...
406     QString entry = anIObject->getEntry();
407     _PTR(SObject) obj = aStudy->FindObjectID( entry.toLatin1().data() );
408     // check parent component
409     QString parentComp = getParentComponent( obj );
410     if ( parentComp != geomComp )  {
411       SUIT_MessageBox::warning( app->desktop(),
412                                 QObject::tr("ERR_ERROR"),
413                                 QObject::tr("NON_GEOM_OBJECTS_SELECTED").arg( getGeometryGUI()->moduleName() ) );
414       return; // not GEOM object selected
415     }
416
417     ///////////////////////////////////////////////////////
418     // if GEOM component is selected, so skip other checks
419     if ( isComponentSelected ) continue; 
420     ///////////////////////////////////////////////////////
421         
422     // check if object is reference
423     _PTR(SObject) refobj;
424     if ( obj && obj->ReferencedObject( refobj ) ) {
425       // get the main object by reference IPAL 21354
426       obj = refobj;
427       entry = obj->GetID().c_str();
428     }
429     // ...
430     QString aName = obj->GetName().c_str();
431     if ( entry == geomComp ) {
432       // GEOM component is selected, skip other checks
433       isComponentSelected = true;
434       continue;
435     }
436     toBeDeleted.insert( entry, aName );
437     allDeleted.insert( entry, aName ); // skip GEOM component
438     // browse through all children recursively
439     _PTR(ChildIterator) it ( aStudy->NewChildIterator( obj ) );
440     for ( it->InitEx( true ); it->More(); it->Next() ) {
441       _PTR(SObject) child( it->Value() );
442       if ( child && child->ReferencedObject( refobj ) )
443         continue; // skip references
444       aName = child->GetName().c_str();
445       if ( !aName.isEmpty() )
446         allDeleted.insert( child->GetID().c_str(), aName );
447     }
448   }
449   
450   // is there is anything to delete?
451   if ( !isComponentSelected && allDeleted.count() <= 0 )
452     return; // nothing to delete
453
454   // show confirmation dialog box
455   GEOMToolsGUI_DeleteDlg dlg( app->desktop(), allDeleted, isComponentSelected );
456   if ( !dlg.exec() )
457     return; // operation is cancelled by user
458   
459   // get currently opened views
460   QList<SALOME_View*> views;
461   SALOME_View* view;
462   ViewManagerList vmans = app->viewManagers();
463   SUIT_ViewManager* vman;
464   foreach ( vman, vmans ) {
465     SUIT_ViewModel* vmod = vman->getViewModel();
466     view = dynamic_cast<SALOME_View*> ( vmod ); // must work for OCC and VTK views
467     if ( view )
468       views.append( view );
469   }
470   
471   _PTR(StudyBuilder) aStudyBuilder (aStudy->NewBuilder());
472   GEOM_Displayer* disp = new GEOM_Displayer( appStudy );
473   
474   if ( isComponentSelected ) {
475     // GEOM component is selected: delete all objects recursively
476     _PTR(SObject) comp = aStudy->FindObjectID( geomComp.toLatin1().data() );
477     if ( !comp )
478       return;
479     _PTR(ChildIterator) it ( aStudy->NewChildIterator( comp ) );
480     // remove top-level objects only
481     for ( it->InitEx( false ); it->More(); it->Next() ) {
482       _PTR(SObject) child( it->Value() );
483       // remove object from GEOM engine
484       removeObjectWithChildren( child, aStudy, views, disp );
485       // remove object from study
486       aStudyBuilder->RemoveObjectWithChildren( child );
487     }
488   }
489   else {
490     // GEOM component is not selected: check if selected objects are in use
491     if ( inUse( aStudy, geomComp, allDeleted ) ) {
492       SUIT_MessageBox::warning( app->desktop(),
493                                 QObject::tr("WRN_WARNING"),
494                                 QObject::tr("DEP_OBJECT") );
495       return; // object(s) in use
496     }
497     // ... and then delete all objects
498     QMap<QString, QString>::Iterator it;
499     for ( it = toBeDeleted.begin(); it != toBeDeleted.end(); ++it ) {
500       _PTR(SObject) obj ( aStudy->FindObjectID( it.key().toLatin1().data() ) );
501       // remove object from GEOM engine
502       removeObjectWithChildren( obj, aStudy, views, disp );
503       // remove objects from study
504       aStudyBuilder->RemoveObjectWithChildren( obj );
505     }
506   }
507   
508   selected.Clear();
509   aSelMgr->setSelectedObjects( selected );
510   getGeometryGUI()->updateObjBrowser();
511   app->updateActions(); //SRN: To update a Save button in the toolbar
512 }
513
514 //=====================================================================================
515 // function : Import
516 // purpose  : BRep, Iges, Step, ...
517 //=====================================================================================
518 bool GEOMToolsGUI::Import()
519 {
520   SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( getGeometryGUI()->getApp() );
521   if ( !app ) return false;
522
523   SalomeApp_Study* stud = dynamic_cast<SalomeApp_Study*> ( app->activeStudy() );
524   if ( !stud ) {
525     MESSAGE ( "FAILED to cast active study to SalomeApp_Study" );
526     return false;
527   }
528   _PTR(Study) aStudy = stud->studyDS();
529
530   // check if study is locked
531   bool aLocked = (_PTR(AttributeStudyProperties)(aStudy->GetProperties()))->IsLocked();
532   if ( aLocked ) {
533     SUIT_MessageBox::warning( app->desktop(),
534                               QObject::tr("WRN_WARNING"),
535                               QObject::tr("WRN_STUDY_LOCKED") );
536     return false;
537   }
538
539   // check if GEOM engine is available
540   GEOM::GEOM_Gen_var eng = GeometryGUI::GetGeomGen();
541   if ( CORBA::is_nil( eng ) ) {
542     SUIT_MessageBox::critical( app->desktop(),
543                                QObject::tr("WRN_WARNING"),
544                                QObject::tr( "GEOM Engine is not started" ) );
545     return false;
546   }
547
548   GEOM::GEOM_IInsertOperations_var aInsOp = eng->GetIInsertOperations( aStudy->StudyId() );
549   if ( aInsOp->_is_nil() )
550     return false;
551
552   // obtain a list of available import formats
553   FilterMap aMap;
554   GEOM::string_array_var aFormats, aPatterns;
555   aInsOp->ImportTranslators( aFormats, aPatterns );
556
557   for ( int i = 0, n = aFormats->length(); i < n; i++ )
558     aMap.insert( (char*)aPatterns[i], (char*)aFormats[i] );
559
560   // select files to be imported
561   QString fileType;
562   QStringList fileNames = getFileNames( app->desktop(), "", aMap,
563                                         tr( "GEOM_MEN_IMPORT" ), fileType, true );
564
565   // set Wait cursor
566   SUIT_OverrideCursor wc;
567
568   if ( fileNames.count() == 0 )
569     return false; // nothing selected, return
570
571   QStringList errors;
572
573   QList< GEOM::GEOM_Object_var > objsForDisplay;
574
575   
576   // iterate through all selected files
577
578   SUIT_MessageBox::StandardButton igesAnswer = SUIT_MessageBox::NoButton;
579   SUIT_MessageBox::StandardButton acisAnswer = SUIT_MessageBox::NoButton;
580
581   for ( int i = 0; i < fileNames.count(); i++ ) {
582     QString fileName = fileNames[i];
583
584     if ( fileName.isEmpty() )
585       continue;
586
587     QString aCurrentType;
588     if ( fileType.isEmpty() ) {
589       // file type is not defined, try to detect
590       QString ext = QFileInfo( fileName ).suffix().toUpper();
591       QRegExp re( "\\*\\.(\\w+)" );
592       for ( FilterMap::const_iterator it = aMap.begin(); 
593             it != aMap.end() && aCurrentType.isEmpty(); ++it ) {
594         int pos = 0;
595         while ( re.indexIn( it.key(), pos ) >= 0 ) {
596           QString f = re.cap(1).trimmed().toUpper();
597           if ( ext == f ) { aCurrentType = it.value(); break; }
598           pos = re.pos() + re.cap(1).length() + 2;
599         }
600       }
601     }
602     else {
603       aCurrentType = fileType;
604     }
605
606     if ( aCurrentType.isEmpty() ) {
607       errors.append( QString( "%1 : %2" ).arg( fileName ).arg( tr( "GEOM_UNSUPPORTED_TYPE" ) ) );
608       continue;
609     }
610
611     GEOM_Operation* anOp = new GEOM_Operation( app, aInsOp.in() );
612     try {
613       app->putInfo( tr( "GEOM_PRP_LOADING" ).arg( SUIT_Tools::file( fileName, /*withExten=*/true ) ) );
614       anOp->start();
615
616       CORBA::String_var fileN = fileName.toLatin1().constData();
617       CORBA::String_var fileT = aCurrentType.toLatin1().constData();
618
619       // skl 29.05.2009
620       if ( aCurrentType == "IGES" ) {
621         GEOM::GEOM_Object_var anObj = aInsOp->Import( fileN, "IGES_UNIT" );
622         bool needConvert = false;
623         TCollection_AsciiString aUnitName = aInsOp->GetErrorCode();
624         if ( aUnitName.SubString( 1, 4 ) == "UNIT" )
625           needConvert = aUnitName.SubString( 6, aUnitName.Length() ) != "M";
626
627         if ( needConvert ) {
628           if ( igesAnswer == SUIT_MessageBox::NoToAll ) {
629             // converting for all files is already approved
630             fileT = "IGES_SCALE";
631           }
632           else if ( igesAnswer != SUIT_MessageBox::YesToAll ) {
633             SUIT_MessageBox::StandardButtons btns = SUIT_MessageBox::Yes | SUIT_MessageBox::No;
634             if ( i < fileNames.count()-1 ) btns = btns | SUIT_MessageBox::YesToAll | SUIT_MessageBox::NoToAll;
635             igesAnswer = SUIT_MessageBox::question( app->desktop(),
636                                                     "Question",//tr("WRN_WARNING"),
637                                                     tr("GEOM_SCALE_DIMENSIONS"),
638                                                     btns | SUIT_MessageBox::Cancel,
639                                                     SUIT_MessageBox::No );
640             switch ( igesAnswer ) {
641             case SUIT_MessageBox::Cancel:
642               return false;                // cancel (break) import operation
643             case SUIT_MessageBox::Yes:
644             case SUIT_MessageBox::YesToAll:
645               break;                       // scaling is confirmed
646             case SUIT_MessageBox::No:
647             case SUIT_MessageBox::NoAll:
648               fileT = "IGES_SCALE";
649             default:
650               break;                       // scaling is rejected
651             } // switch ( igesAnswer )
652           } // if ( igeAnswer != NoToAll )
653         } // if ( needConvert )
654       } // if ( aCurrentType == "IGES" )
655       else if ( aCurrentType == "ACIS" ) {
656         if ( acisAnswer != SUIT_MessageBox::YesToAll && acisAnswer != SUIT_MessageBox::NoToAll ) {
657           SUIT_MessageBox::StandardButtons btns = SUIT_MessageBox::Yes | SUIT_MessageBox::No;
658           if ( i < fileNames.count()-1 ) btns = btns | SUIT_MessageBox::YesToAll | SUIT_MessageBox::NoToAll;
659           acisAnswer = SUIT_MessageBox::question( app->desktop(),
660                                                   "Question",//tr("WRN_WARNING"),
661                                                   tr("GEOM_PUBLISH_NAMED_SHAPES"),
662                                                   btns | SUIT_MessageBox::Cancel,
663                                                   SUIT_MessageBox::No );
664           if ( acisAnswer == SUIT_MessageBox::Cancel )
665             return false; // cancel (break) import operation
666         } // if ( acisAnswer != YesToAll && acisAnswer != NoToAll )
667       } // else if ( aCurrentType == "ACIS" )
668       
669       GEOM::GEOM_Object_var anObj = aInsOp->Import( fileN, fileT );
670
671       if ( !anObj->_is_nil() && aInsOp->IsDone() ) {
672         QString aPublishObjName = 
673           GEOMBase::GetDefaultName( SUIT_Tools::file( fileName, /*withExten=*/true ) );
674         
675         SALOMEDS::Study_var aDSStudy = GeometryGUI::ClientStudyToStudy( aStudy );
676         GeometryGUI::GetGeomGen()->PublishInStudy( aDSStudy,
677                                                    SALOMEDS::SObject::_nil(),
678                                                    anObj,
679                                                    aPublishObjName.toLatin1().constData() );
680         
681         objsForDisplay.append( anObj );
682         
683         if ( aCurrentType == "ACIS" ) {
684           if ( acisAnswer == SUIT_MessageBox::Yes || acisAnswer == SUIT_MessageBox::YesToAll )
685             GeometryGUI::GetGeomGen()->PublishNamedShapesInStudy( aDSStudy, anObj );
686         }
687
688         anOp->commit();
689       }
690       else {
691         anOp->abort();
692         errors.append( QString( "%1 : %2" ).arg( fileName ).arg( aInsOp->GetErrorCode() ) );
693       }
694     }
695     catch( const SALOME::SALOME_Exception& S_ex ) {
696       anOp->abort();
697       errors.append( QString( "%1 : %2" ).arg( fileName ).arg( tr( "GEOM_UNKNOWN_IMPORT_ERROR" ) ) );
698     }
699   }
700
701   // update object browser
702   getGeometryGUI()->updateObjBrowser( true );
703
704   // display imported model (if only one file is selected)
705   if ( objsForDisplay.count() == 1 )
706     GEOM_Displayer( stud ).Display( objsForDisplay[0].in() );
707
708   if ( errors.count() > 0 ) {
709     SUIT_MessageBox::critical( app->desktop(),
710                                QObject::tr( "GEOM_ERROR" ),
711                                QObject::tr( "GEOM_IMPORT_ERRORS" ) + "\n" + errors.join( "\n" ) );
712   }
713
714   app->updateActions(); //SRN: To update a Save button in the toolbar
715
716   return objsForDisplay.count() > 0;
717 }
718
719
720 //=====================================================================================
721 // function : Export
722 // purpose  : BRep, Iges, Step
723 //=====================================================================================
724 bool GEOMToolsGUI::Export()
725 {
726   SalomeApp_Application* app = getGeometryGUI()->getApp();
727   if (!app) return false;
728
729   SalomeApp_Study* stud = dynamic_cast<SalomeApp_Study*> ( app->activeStudy() );
730   if ( !stud ) {
731     MESSAGE ( "FAILED to cast active study to SalomeApp_Study" );
732     return false;
733   }
734   _PTR(Study) aStudy = stud->studyDS();
735
736   GEOM::GEOM_Gen_var eng = GeometryGUI::GetGeomGen();
737   if ( CORBA::is_nil( eng ) ) {
738     SUIT_MessageBox::critical( app->desktop(),
739                                QObject::tr("WRN_WARNING"),
740                                QObject::tr( "GEOM Engine is not started" ) );
741     return false;
742   }
743
744   GEOM::GEOM_IInsertOperations_var aInsOp = eng->GetIInsertOperations( aStudy->StudyId() );
745   if ( aInsOp->_is_nil() )
746     return false;
747
748   // Obtain a list of available export formats
749   FilterMap aMap;
750   QStringList filters;
751   GEOM::string_array_var aFormats, aPatterns;
752   aInsOp->ExportTranslators( aFormats, aPatterns );
753   for ( int i = 0, n = aFormats->length(); i < n; i++ ) {
754     aMap.insert( (char*)aPatterns[i], (char*)aFormats[i] );
755     filters.push_back( (char*)aPatterns[i] );
756   }
757
758   // Get selected objects
759   LightApp_SelectionMgr* sm = app->selectionMgr();
760   if ( !sm )
761     return false;
762
763   SALOME_ListIO selectedObjects;
764   sm->selectedObjects( selectedObjects );
765   bool appropriateObj = false;
766
767   SALOME_ListIteratorOfListIO It( selectedObjects );
768   for(;It.More();It.Next()) {
769     Handle(SALOME_InteractiveObject) IObject = It.Value();
770     Standard_Boolean found;
771     GEOM::GEOM_Object_var anObj = GEOMBase::ConvertIOinGEOMObject(IObject, found);
772
773     if ( !found || anObj->_is_nil() )
774       continue;
775
776     QString fileType;
777     QString file = getFileName(app->desktop(), QString( IObject->getName() ), aMap, filters,
778                                tr("GEOM_MEN_EXPORT"), false, fileType, true);
779
780     // User has pressed "Cancel" --> stop the operation
781     if ( file.isEmpty() || fileType.isEmpty() )
782       return false;
783
784     GEOM_Operation* anOp = new GEOM_Operation( app, aInsOp.in() );
785     try {
786       SUIT_OverrideCursor wc;
787
788       app->putInfo( tr("GEOM_PRP_EXPORT").arg(SUIT_Tools::file( file, /*withExten=*/true )) );
789
790       anOp->start();
791
792
793       aInsOp->Export( anObj, file.toStdString().c_str(), fileType.toLatin1().constData() );
794
795       if ( aInsOp->IsDone() )
796         anOp->commit();
797       else
798         {
799           anOp->abort();
800           wc.suspend();
801           SUIT_MessageBox::critical( app->desktop(),
802                                      QObject::tr( "GEOM_ERROR" ),
803                                      QObject::tr("GEOM_PRP_ABORT") + "\n" + QString( aInsOp->GetErrorCode() ) );
804           return false;
805         }
806     }
807     catch (const SALOME::SALOME_Exception& S_ex) {
808       //QtCatchCorbaException(S_ex);
809       anOp->abort();
810       return false;
811     }
812     appropriateObj = true;
813   }
814
815   if ( !appropriateObj )
816     SUIT_MessageBox::warning( app->desktop(),
817                               QObject::tr("WRN_WARNING"),
818                               QObject::tr("GEOM_WRN_NO_APPROPRIATE_SELECTION") );
819   return true;
820 }
821
822 //=====================================================================================
823 // function : RemoveObjectWithChildren
824 // purpose  : used by OnEditDelete() method
825 //=====================================================================================
826 void GEOMToolsGUI::removeObjectWithChildren(_PTR(SObject) obj,
827                                             _PTR(Study) aStudy,
828                                             QList<SALOME_View*> views,
829                                             GEOM_Displayer* disp)
830 {
831   // iterate through all children of obj
832   for (_PTR(ChildIterator) it (aStudy->NewChildIterator(obj)); it->More(); it->Next()) {
833     _PTR(SObject) child (it->Value());
834     removeObjectWithChildren(child, aStudy, views, disp);
835   }
836
837   // erase object and remove it from engine
838   _PTR(GenericAttribute) anAttr;
839   if (obj->FindAttribute(anAttr, "AttributeIOR")) {
840     _PTR(AttributeIOR) anIOR (anAttr);
841
842     // Delete shape in Client
843     const TCollection_AsciiString ASCIor ((char*)anIOR->Value().c_str());
844     getGeometryGUI()->GetShapeReader().RemoveShapeFromBuffer(ASCIor);
845
846     CORBA::Object_var corbaObj = GeometryGUI::ClientSObjectToObject(obj);
847     GEOM::GEOM_Object_var geomObj = GEOM::GEOM_Object::_narrow( corbaObj );
848     if (!CORBA::is_nil(geomObj)) {
849       // Erase graphical object
850       QListIterator<SALOME_View*> it( views );
851       while ( it.hasNext() )
852         if ( SALOME_View* view = it.next() )
853           disp->Erase(geomObj, true, view);
854       
855       // Remove object from Engine
856       // We can't directly remove object from engine. All we can do is to unpublish the object
857       // from the study. Another client could be using the object.
858       // Unpublishing is done just after in aStudyBuilder->RemoveObjectWithChildren( child );
859       //GeometryGUI::GetGeomGen()->RemoveObject( geomObj );
860     }
861   }
862 }
863
864 //=================================================================================
865 // function : deactivate()
866 // purpose  : Called when GEOM component is deactivated
867 //=================================================================================
868 void GEOMToolsGUI::deactivate()
869 {
870   SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
871   if ( app ) {
872     SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
873     GEOM_Displayer aDisp (appStudy);
874     aDisp.GlobalSelection();
875     getGeometryGUI()->setLocalSelectionMode(GEOM_ALLOBJECTS);
876   }
877 }
878
879 //=====================================================================================
880 // EXPORTED METHODS
881 //=====================================================================================
882 extern "C"
883 {
884 #ifdef WIN32
885   __declspec( dllexport )
886 #endif
887   GEOMGUI* GetLibGUI( GeometryGUI* parent )
888   {
889     return new GEOMToolsGUI( parent );
890   }
891 }