src/GEOMToolsGUI/GEOMToolsGUI.cxx

   1 //  Copyright (C) 2007-2008  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 // GEOM GEOMGUI : GUI for Geometry component
  23 // File   : GEOMBase_Tools.cxx
  24 // Author : Damien COQUERET, Open CASCADE S.A.S.
  25 //
  26 #include "GEOMToolsGUI.h"
  27 #include "GEOMToolsGUI_DeleteDlg.h"
  28
  29 #include <GeometryGUI.h>
  30 #include <GEOMBase.h>
  31 #include <GEOM_Operation.h>
  32 #include <GEOM_Displayer.h>
  33
  34 #include <SUIT_Session.h>
  35 #include <SUIT_OverrideCursor.h>
  36 #include <SUIT_MessageBox.h>
  37 #include <SUIT_Tools.h>
  38 #include <SUIT_FileDlg.h>
  39 #include <SUIT_Desktop.h>
  40 #include <SUIT_ViewModel.h>
  41 #include <SUIT_ViewManager.h>
  42
  43 #include <SalomeApp_Application.h>
  44 #include <SalomeApp_Study.h>
  45 #include <LightApp_SelectionMgr.h>
  46 #include <GEOMImpl_Types.hxx>
  47
  48 #include <SALOME_ListIO.hxx>
  49 #include <SALOME_ListIteratorOfListIO.hxx>
  50 #include <SALOME_Prs.h>
  51
  52 // QT Includes
  53 #include <QApplication>
  54 #include <QMap>
  55 #include <QRegExp>
  56
  57 // OCCT Includes
  58 #include <TCollection_AsciiString.hxx>
  59
  60 using namespace std;
  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   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   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     {
 290     case 31: // COPY
 291       {
 292         OnEditCopy();
 293         break;
 294       }
 295     case 33: // DELETE
 296       {
 297         OnEditDelete();
 298         break;
 299       }
 300     case 111: // IMPORT BREP
 301     case 112: // IMPORT IGES
 302     case 113: // IMPORT STEP
 303       {
 304         Import();
 305         break;
 306       }
 307     case 121: // EXPORT BREP
 308     case 122: // EXPORT IGES
 309     case 123: // EXPORT STEP
 310       {
 311         Export();
 312         break;
 313       }
 314     case 2171: // POPUP VIEWER - SELECT ONLY - VERTEX
 315       {
 316         OnSelectOnly( GEOM_POINT );
 317         break;
 318       }
 319     case 2172: // POPUP VIEWER - SELECT ONLY - EDGE
 320       {
 321         OnSelectOnly( GEOM_EDGE );
 322         break;
 323       }
 324     case 2173: // POPUP VIEWER - SELECT ONLY - WIRE
 325       {
 326         OnSelectOnly( GEOM_WIRE );
 327         break;
 328       }
 329     case 2174: // POPUP VIEWER - SELECT ONLY - FACE
 330       {
 331         OnSelectOnly( GEOM_FACE );
 332         break;
 333       }
 334     case 2175: // POPUP VIEWER - SELECT ONLY - SHELL
 335       {
 336         OnSelectOnly( GEOM_SHELL );
 337         break;
 338       }
 339     case 2176: // POPUP VIEWER - SELECT ONLY - SOLID
 340       {
 341         OnSelectOnly( GEOM_SOLID );
 342         break;
 343       }
 344     case 2177: // POPUP VIEWER - SELECT ONLY - COMPOUND
 345       {
 346         OnSelectOnly( GEOM_COMPOUND );
 347         break;
 348       }
 349     case 2178: // POPUP VIEWER - SELECT ONLY - SELECT ALL
 350       {
 351         OnSelectOnly( GEOM_ALLOBJECTS );
 352         break;
 353       }
 354     case 411: // SETTINGS - ADD IN STUDY
 355       {
 356         // SAN -- TO BE REMOVED !!!
 357         break;
 358       }
 359     case 412: // SETTINGS - SHADING COLOR
 360       {
 361         OnSettingsColor();
 362         break;
 363       }
 364     case 804: // ADD IN STUDY - POPUP VIEWER
 365       {
 366         // SAN -- TO BE REMOVED !!!!
 367         break;
 368       }
 369     case 901: // RENAME
 370       {
 371         OnRename();
 372         break;
 373       }
 374     case 5103: // CHECK GEOMETRY
 375       {
 376         OnCheckGeometry();
 377         break;
 378       }
 379     case 8032: // COLOR - POPUP VIEWER
 380       {
 381         OnColor();
 382         break;
 383       }
 384     case 8033: // TRANSPARENCY - POPUP VIEWER
 385       {
 386         OnTransparency();
 387         break;
 388       }
 389     case 8034: // ISOS - POPUP VIEWER
 390       {
 391         OnNbIsos();
 392         break;
 393       }
 394     case 8035: // AUTO COLOR - POPUP VIEWER
 395       {
 396         OnAutoColor();
 397         break;
 398       }
 399     case 8036: // DISABLE AUTO COLOR - POPUP VIEWER
 400       {
 401         OnDisableAutoColor();
 402         break;
 403       }
 404     case 9024 : // OPEN - OBJBROSER POPUP
 405       {
 406         OnOpen();
 407         break;
 408       }
 409     default:
 410       {
 411         SUIT_Session::session()->activeApplication()->putInfo(tr("GEOM_PRP_COMMAND").arg(theCommandID));
 412         break;
 413       }
 414     }
 415   return true;
 416 }
 417
 418
 419 //===============================================================================
 420 // function : OnEditDelete()
 421 // purpose  :
 422 //===============================================================================
 423 void GEOMToolsGUI::OnEditDelete()
 424 {
 425   SALOME_ListIO selected;
 426   SalomeApp_Application* app =
 427     dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
 428   if ( !app )
 429     return;
 430
 431   LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
 432   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
 433   if ( !aSelMgr || !appStudy )
 434     return;
 435
 436   // get selection
 437   aSelMgr->selectedObjects( selected, "ObjectBrowser", false );
 438   if ( selected.IsEmpty() )
 439     return;
 440
 441   _PTR(Study) aStudy = appStudy->studyDS();
 442
 443   // check if study is locked
 444   if ( _PTR(AttributeStudyProperties)( aStudy->GetProperties() )->IsLocked() ) {
 445     SUIT_MessageBox::warning( app->desktop(),
 446                               tr("WRN_WARNING"),
 447                               tr("WRN_STUDY_LOCKED") );
 448     return; // study is locked
 449   }
 450
 451   // get GEOM component
 452   CORBA::String_var geomIOR = app->orb()->object_to_string( GeometryGUI::GetGeomGen() );
 453   QString geomComp = getParentComponent( aStudy->FindObjectIOR( geomIOR.in() ) );
 454
 455   // check each selected object: if belongs to GEOM, if not reference...
 456   QMap<QString,QString> toBeDeleted;
 457   QMap<QString,QString> allDeleted;
 458   bool isComponentSelected = false;
 459   for ( SALOME_ListIteratorOfListIO It( selected ); It.More(); It.Next() ) {
 460     Handle(SALOME_InteractiveObject) anIObject = It.Value();
 461     if ( !anIObject->hasEntry() )
 462       continue; // invalid object
 463     // ...
 464     QString entry = anIObject->getEntry();
 465     _PTR(SObject) obj = aStudy->FindObjectID( entry.toLatin1().data() );
 466     // check parent component
 467     QString parentComp = getParentComponent( obj );
 468     if ( parentComp != geomComp )  {
 469       SUIT_MessageBox::warning( app->desktop(),
 470                                 QObject::tr("ERR_ERROR"),
 471                                 QObject::tr("NON_GEOM_OBJECTS_SELECTED").arg( getGeometryGUI()->moduleName() ) );
 472       return; // not GEOM object selected
 473     }
 474
 475     ///////////////////////////////////////////////////////
 476     // if GEOM component is selected, so skip other checks
 477     if ( isComponentSelected ) continue;
 478     ///////////////////////////////////////////////////////
 479
 480     // check if object is reference
 481     _PTR(SObject) refobj;
 482     if ( obj && obj->ReferencedObject( refobj ) )
 483       continue; // skip references
 484     // ...
 485     QString aName = obj->GetName().c_str();
 486     if ( entry == geomComp ) {
 487       // GEOM component is selected, skip other checks
 488       isComponentSelected = true;
 489       continue;
 490     }
 491     toBeDeleted.insert( entry, aName );
 492     allDeleted.insert( entry, aName ); // skip GEOM component
 493     // browse through all children recursively
 494     _PTR(ChildIterator) it ( aStudy->NewChildIterator( obj ) );
 495     for ( it->InitEx( true ); it->More(); it->Next() ) {
 496       _PTR(SObject) child( it->Value() );
 497       if ( child && child->ReferencedObject( refobj ) )
 498         continue; // skip references
 499       aName = child->GetName().c_str();
 500       if ( !aName.isEmpty() )
 501         allDeleted.insert( child->GetID().c_str(), aName );
 502     }
 503   }
 504
 505   // is there is anything to delete?
 506   if ( !isComponentSelected && allDeleted.count() <= 0 )
 507     return; // nothing to delete
 508
 509   // show confirmation dialog box
 510   GEOMToolsGUI_DeleteDlg dlg( app->desktop(), allDeleted, isComponentSelected );
 511   if ( !dlg.exec() )
 512     return; // operation is cancelled by user
 513
 514   // get currently opened views
 515   QList<SALOME_View*> views;
 516   SALOME_View* view;
 517   ViewManagerList vmans = app->viewManagers();
 518   SUIT_ViewManager* vman;
 519   foreach ( vman, vmans ) {
 520     SUIT_ViewModel* vmod = vman->getViewModel();
 521     view = dynamic_cast<SALOME_View*> ( vmod ); // must work for OCC and VTK views
 522     if ( view )
 523       views.append( view );
 524   }
 525
 526   _PTR(StudyBuilder) aStudyBuilder (aStudy->NewBuilder());
 527   GEOM_Displayer* disp = new GEOM_Displayer( appStudy );
 528
 529   if ( isComponentSelected ) {
 530     // GEOM component is selected: delete all objects recursively
 531     _PTR(SObject) comp = aStudy->FindObjectID( geomComp.toLatin1().data() );
 532     if ( !comp )
 533       return;
 534     _PTR(ChildIterator) it ( aStudy->NewChildIterator( comp ) );
 535     // remove top-level objects only
 536     for ( it->InitEx( false ); it->More(); it->Next() ) {
 537       _PTR(SObject) child( it->Value() );
 538       // remove object from GEOM engine
 539       removeObjectWithChildren( child, aStudy, views, disp );
 540       // remove object from study
 541       aStudyBuilder->RemoveObjectWithChildren( child );
 542     }
 543   }
 544   else {
 545     // GEOM component is not selected: check if selected objects are in use
 546     if ( inUse( aStudy, geomComp, allDeleted ) ) {
 547       SUIT_MessageBox::warning( app->desktop(),
 548                                 QObject::tr("WRN_WARNING"),
 549                                 QObject::tr("DEP_OBJECT") );
 550       return; // object(s) in use
 551     }
 552     // ... and then delete all objects
 553     QMap<QString, QString>::Iterator it;
 554     for ( it = toBeDeleted.begin(); it != toBeDeleted.end(); ++it ) {
 555       _PTR(SObject) obj ( aStudy->FindObjectID( it.key().toLatin1().data() ) );
 556       // remove object from GEOM engine
 557       removeObjectWithChildren( obj, aStudy, views, disp );
 558       // remove objects from study
 559       aStudyBuilder->RemoveObjectWithChildren( obj );
 560     }
 561   }
 562
 563   selected.Clear();
 564   aSelMgr->setSelectedObjects( selected );
 565   getGeometryGUI()->updateObjBrowser();
 566   app->updateActions(); //SRN: To update a Save button in the toolbar
 567 }
 568
 569
 570 //==============================================================================
 571 // function : OnEditCopy()
 572 // purpose  :
 573 //==============================================================================
 574 void GEOMToolsGUI::OnEditCopy()
 575 {
 576 /*
 577  SALOME_Selection* Sel = SALOME_Selection::Selection(QAD_Application::getDesktop()->getActiveStudy()->getSelection() );
 578   GEOM::string_array_var listIOR = new GEOM::string_array;
 579
 580   const SALOME_ListIO& List = Sel->StoredIObjects();
 581
 582   myGeomBase->ConvertListOfIOInListOfIOR(List, listIOR);
 583
 584   Sel->ClearIObjects();
 585
 586   SALOMEDS::Study_var aStudy = QAD_Application::getDesktop()->getActiveStudy()->getStudyDocument();
 587   int aStudyID = aStudy->StudyId();
 588
 589   for (unsigned int ind = 0; ind < listIOR->length();ind++) {
 590     GEOM::GEOM_Object_var aShapeInit = myGeom->GetIORFromString(listIOR[ind]);
 591     try {
 592       GEOM::GEOM_IInsertOperations_var IOp =  myGeom->GetIInsertOperations(aStudyID);
 593       GEOM::GEOM_Object_var result = IOp->MakeCopy(aShapeInit);
 594       myGeomBase->Display(result);
 595     }
 596     catch  (const SALOME::SALOME_Exception& S_ex) {
 597       QtCatchCorbaException(S_ex);
 598     }
 599   }
 600
 601   QAD_Application::getDesktop()->putInfo(tr("GEOM_PRP_READY"));
 602 */
 603 }
 604
 605 //=====================================================================================
 606 // function : Import
 607 // purpose  : BRep, Iges, Step
 608 //=====================================================================================
 609 bool GEOMToolsGUI::Import()
 610 {
 611   SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( getGeometryGUI()->getApp() );
 612   if (! app) return false;
 613
 614   SalomeApp_Study* stud = dynamic_cast<SalomeApp_Study*> ( app->activeStudy() );
 615   if ( !stud ) {
 616     MESSAGE ( "FAILED to cast active study to SalomeApp_Study" );
 617     return false;
 618   }
 619   _PTR(Study) aStudy = stud->studyDS();
 620
 621   // check if study is locked
 622   bool aLocked = (_PTR(AttributeStudyProperties)(aStudy->GetProperties()))->IsLocked();
 623   if ( aLocked ) {
 624     SUIT_MessageBox::warning( app->desktop(),
 625                               QObject::tr("WRN_WARNING"),
 626                               QObject::tr("WRN_STUDY_LOCKED") );
 627     return false;
 628   }
 629
 630   // check if GEOM engine is available
 631   GEOM::GEOM_Gen_var eng = GeometryGUI::GetGeomGen();
 632   if ( CORBA::is_nil( eng ) ) {
 633     SUIT_MessageBox::critical( app->desktop(),
 634                                QObject::tr("WRN_WARNING"),
 635                                QObject::tr( "GEOM Engine is not started" ) );
 636     return false;
 637   }
 638
 639   GEOM::GEOM_IInsertOperations_var aInsOp = eng->GetIInsertOperations( aStudy->StudyId() );
 640   if ( aInsOp->_is_nil() )
 641     return false;
 642
 643   // obtain a list of available import formats
 644   FilterMap aMap;
 645   GEOM::string_array_var aFormats, aPatterns;
 646   aInsOp->ImportTranslators( aFormats, aPatterns );
 647
 648   for ( int i = 0, n = aFormats->length(); i < n; i++ )
 649     aMap.insert( (char*)aPatterns[i], (char*)aFormats[i] );
 650
 651   // select files to be imported
 652   QString fileType;
 653   QStringList fileNames = getFileNames( app->desktop(), "", aMap,
 654                                         tr( "GEOM_MEN_IMPORT" ), fileType, true );
 655
 656   // set Wait cursor
 657   SUIT_OverrideCursor wc;
 658
 659   if ( fileNames.count() == 0 )
 660     return false; // nothing selected, return
 661
 662   QStringList errors;
 663
 664   QList< GEOM::GEOM_Object_var > objsForDisplay;
 665
 666   // iterate through all selected files
 667   for ( QStringList::ConstIterator it = fileNames.begin(); it != fileNames.end(); ++it ) {
 668     QString fileName = *it;
 669
 670     if ( fileName.isEmpty() )
 671       continue;
 672
 673     QString aCurrentType;
 674     if ( fileType.isEmpty() ) {
 675       // file type is not defined, try to detect
 676       QString ext = QFileInfo( fileName ).suffix().toUpper();
 677       QRegExp re( "\\*\\.(\\w+)" );
 678       for ( FilterMap::const_iterator it = aMap.begin();
 679             it != aMap.end() && aCurrentType.isEmpty(); ++it ) {
 680         int pos = 0;
 681         while ( re.indexIn( it.key(), pos ) >= 0 ) {
 682           QString f = re.cap(1).trimmed().toUpper();
 683           if ( ext == f ) { aCurrentType = it.value(); break; }
 684           pos = re.pos() + re.cap(1).length() + 2;
 685         }
 686       }
 687     }
 688     else {
 689       aCurrentType = fileType;
 690     }
 691
 692     if ( aCurrentType.isEmpty() ) {
 693       errors.append( QString( "%1 : %2" ).arg( fileName ).arg( tr( "GEOM_UNSUPPORTED_TYPE" ) ) );
 694       continue;
 695     }
 696
 697     GEOM_Operation* anOp = new GEOM_Operation( app, aInsOp.in() );
 698     try {
 699       app->putInfo( tr( "GEOM_PRP_LOADING" ).arg( SUIT_Tools::file( fileName, /*withExten=*/true ) ) );
 700       anOp->start();
 701
 702       CORBA::String_var fileN = fileName.toLatin1().constData();
 703       CORBA::String_var fileT = aCurrentType.toLatin1().constData();
 704       GEOM::GEOM_Object_var anObj = aInsOp->Import( fileN, fileT );
 705
 706       if ( !anObj->_is_nil() && aInsOp->IsDone() ) {
 707         QString aPublishObjName =
 708           GEOMBase::GetDefaultName( SUIT_Tools::file( fileName, /*withExten=*/true ) );
 709
 710         SALOMEDS::Study_var aDSStudy = GeometryGUI::ClientStudyToStudy( aStudy );
 711         GeometryGUI::GetGeomGen()->PublishInStudy( aDSStudy,
 712                                                    SALOMEDS::SObject::_nil(),
 713                                                    anObj,
 714                                                    aPublishObjName.toLatin1().constData() );
 715
 716         objsForDisplay.append( anObj );
 717
 718         anOp->commit();
 719       }
 720       else {
 721         anOp->abort();
 722         errors.append( QString( "%1 : %2" ).arg( fileName ).arg( aInsOp->GetErrorCode() ) );
 723       }
 724     }
 725     catch( const SALOME::SALOME_Exception& S_ex ) {
 726       anOp->abort();
 727       errors.append( QString( "%1 : %2" ).arg( fileName ).arg( tr( "GEOM_UNKNOWN_IMPORT_ERROR" ) ) );
 728     }
 729   }
 730
 731   // update object browser
 732   getGeometryGUI()->updateObjBrowser( true );
 733
 734   // display imported model (if only one file is selected)
 735   if ( objsForDisplay.count() == 1 )
 736     GEOM_Displayer( stud ).Display( objsForDisplay[0].in() );
 737
 738   if ( errors.count() > 0 ) {
 739     SUIT_MessageBox::critical( app->desktop(),
 740                                QObject::tr( "GEOM_ERROR" ),
 741                                QObject::tr( "GEOM_IMPORT_ERRORS" ) + "\n" + errors.join( "\n" ) );
 742   }
 743
 744   app->updateActions(); //SRN: To update a Save button in the toolbar
 745
 746   return objsForDisplay.count() > 0;
 747 }
 748
 749
 750 //=====================================================================================
 751 // function : Export
 752 // purpose  : BRep, Iges, Step
 753 //=====================================================================================
 754 bool GEOMToolsGUI::Export()
 755 {
 756   SalomeApp_Application* app = getGeometryGUI()->getApp();
 757   if (!app) return false;
 758
 759   SalomeApp_Study* stud = dynamic_cast<SalomeApp_Study*> ( app->activeStudy() );
 760   if ( !stud ) {
 761     MESSAGE ( "FAILED to cast active study to SalomeApp_Study" );
 762     return false;
 763   }
 764   _PTR(Study) aStudy = stud->studyDS();
 765
 766   GEOM::GEOM_Gen_var eng = GeometryGUI::GetGeomGen();
 767   if ( CORBA::is_nil( eng ) ) {
 768     SUIT_MessageBox::critical( app->desktop(),
 769                                QObject::tr("WRN_WARNING"),
 770                                QObject::tr( "GEOM Engine is not started" ) );
 771     return false;
 772   }
 773
 774   GEOM::GEOM_IInsertOperations_var aInsOp = eng->GetIInsertOperations( aStudy->StudyId() );
 775   if ( aInsOp->_is_nil() )
 776     return false;
 777
 778   // Obtain a list of available export formats
 779   FilterMap aMap;
 780   QStringList filters;
 781   GEOM::string_array_var aFormats, aPatterns;
 782   aInsOp->ExportTranslators( aFormats, aPatterns );
 783   for ( int i = 0, n = aFormats->length(); i < n; i++ ) {
 784     aMap.insert( (char*)aPatterns[i], (char*)aFormats[i] );
 785     filters.push_back( (char*)aPatterns[i] );
 786   }
 787
 788   // Get selected objects
 789   LightApp_SelectionMgr* sm = app->selectionMgr();
 790   if ( !sm )
 791     return false;
 792
 793   SALOME_ListIO selectedObjects;
 794   sm->selectedObjects( selectedObjects );
 795   bool appropriateObj = false;
 796
 797   SALOME_ListIteratorOfListIO It( selectedObjects );
 798   for(;It.More();It.Next()) {
 799     Handle(SALOME_InteractiveObject) IObject = It.Value();
 800     Standard_Boolean found;
 801     GEOM::GEOM_Object_var anObj = GEOMBase::ConvertIOinGEOMObject(IObject, found);
 802
 803     if ( !found || anObj->_is_nil() )
 804       continue;
 805
 806     QString fileType;
 807     QString file = getFileName(app->desktop(), QString( IObject->getName() ), aMap, filters,
 808                                tr("GEOM_MEN_EXPORT"), false, fileType, true);
 809
 810     // User has pressed "Cancel" --> stop the operation
 811     if ( file.isEmpty() || fileType.isEmpty() )
 812       return false;
 813
 814     GEOM_Operation* anOp = new GEOM_Operation( app, aInsOp.in() );
 815     try {
 816       SUIT_OverrideCursor wc;
 817
 818       app->putInfo( tr("GEOM_PRP_EXPORT").arg(SUIT_Tools::file( file, /*withExten=*/true )) );
 819
 820       anOp->start();
 821
 822
 823       aInsOp->Export( anObj, file.toStdString().c_str(), fileType.toLatin1().constData() );
 824
 825       if ( aInsOp->IsDone() )
 826         anOp->commit();
 827       else
 828         {
 829           anOp->abort();
 830           wc.suspend();
 831           SUIT_MessageBox::critical( app->desktop(),
 832                                      QObject::tr( "GEOM_ERROR" ),
 833                                      QObject::tr("GEOM_PRP_ABORT") + "\n" + QString( aInsOp->GetErrorCode() ) );
 834           return false;
 835         }
 836     }
 837     catch (const SALOME::SALOME_Exception& S_ex) {
 838       //QtCatchCorbaException(S_ex);
 839       anOp->abort();
 840       return false;
 841     }
 842     appropriateObj = true;
 843   }
 844
 845   if ( !appropriateObj )
 846     SUIT_MessageBox::warning( app->desktop(),
 847                               QObject::tr("WRN_WARNING"),
 848                               QObject::tr("GEOM_WRN_NO_APPROPRIATE_SELECTION") );
 849   return true;
 850 }
 851
 852 //=====================================================================================
 853 // function : RemoveObjectWithChildren
 854 // purpose  : to be used by OnEditDelete() method
 855 //=====================================================================================
 856 void GEOMToolsGUI::removeObjectWithChildren(_PTR(SObject) obj,
 857                                             _PTR(Study) aStudy,
 858                                             QList<SALOME_View*> views,
 859                                             GEOM_Displayer* disp)
 860 {
 861   // iterate through all children of obj
 862   for (_PTR(ChildIterator) it (aStudy->NewChildIterator(obj)); it->More(); it->Next()) {
 863     _PTR(SObject) child (it->Value());
 864     removeObjectWithChildren(child, aStudy, views, disp);
 865   }
 866
 867   // erase object and remove it from engine
 868   _PTR(GenericAttribute) anAttr;
 869   if (obj->FindAttribute(anAttr, "AttributeIOR")) {
 870     _PTR(AttributeIOR) anIOR (anAttr);
 871
 872     // Delete shape in Client
 873     const TCollection_AsciiString ASCIor ((char*)anIOR->Value().c_str());
 874     getGeometryGUI()->GetShapeReader().RemoveShapeFromBuffer(ASCIor);
 875
 876     CORBA::Object_var corbaObj = GeometryGUI::ClientSObjectToObject(obj);
 877     GEOM::GEOM_Object_var geomObj = GEOM::GEOM_Object::_narrow( corbaObj );
 878     if (!CORBA::is_nil(geomObj)) {
 879       // Erase graphical object
 880       QListIterator<SALOME_View*> it( views );
 881       while ( it.hasNext() )
 882         if ( SALOME_View* view = it.next() )
 883           disp->Erase(geomObj, true, view);
 884
 885       // Remove object from Engine
 886       GeometryGUI::GetGeomGen()->RemoveObject( geomObj );
 887     }
 888   }
 889 }
 890
 891 //=================================================================================
 892 // function : deactivate()
 893 // purpose  : Called when GEOM component is deactivated
 894 //=================================================================================
 895 void GEOMToolsGUI::deactivate()
 896 {
 897   SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
 898   if ( app ) {
 899     SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
 900     GEOM_Displayer aDisp (appStudy);
 901     aDisp.GlobalSelection();
 902     getGeometryGUI()->setLocalSelectionMode(GEOM_ALLOBJECTS);
 903   }
 904 }
 905
 906 //=====================================================================================
 907 // EXPORTED METHODS
 908 //=====================================================================================
 909 extern "C"
 910 {
 911 #ifdef WIN32
 912   __declspec( dllexport )
 913 #endif
 914   GEOMGUI* GetLibGUI( GeometryGUI* parent )
 915   {
 916     return new GEOMToolsGUI( parent );
 917   }
 918 }