src/GEOMBase/GEOMBase_Helper.cxx

   1 // Copyright (C) 2007-2021  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, or (at your option) any later version.
  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_Helper.cxx
  25 //  Author : Sergey ANIKIN, Open CASCADE S.A.S. (sergey.anikin@opencascade.com)
  26
  27 #include "GEOMBase_Helper.h"
  28 #include "GEOMBase.h"
  29 #include "GEOM_Operation.h"
  30
  31 #include <GeometryGUI.h>
  32
  33 #include <SUIT_Desktop.h>
  34 #include <SUIT_Session.h>
  35 #include <SUIT_ViewManager.h>
  36 #include <SUIT_ViewWindow.h>
  37 #include <SUIT_ViewModel.h>
  38 #include <SUIT_MessageBox.h>
  39 #include <SUIT_OverrideCursor.h>
  40 #include <SUIT_ResourceMgr.h>
  41
  42 #include <SalomeApp_Module.h>
  43 #include <SalomeApp_Application.h>
  44 #include <SalomeApp_Study.h>
  45 #include <LightApp_SelectionMgr.h>
  46 #include <LightApp_DataOwner.h>
  47 #include <SalomeApp_Tools.h>
  48 #include <SALOME_ListIO.hxx>
  49 #include "utilities.h"
  50
  51 #include <SALOME_Prs.h>
  52 #include "utilities.h"
  53
  54 #include <OCCViewer_ViewModel.h>
  55 #include <SVTK_ViewModel.h>
  56
  57 #include <TCollection_AsciiString.hxx>
  58 #include <TColStd_IndexedMapOfInteger.hxx>
  59
  60 //To disable automatic genericobj management, the following line should be commented.
  61 //Otherwise, it should be uncommented. Refer to KERNEL_SRC/src/SALOMEDSImpl/SALOMEDSImpl_AttributeIOR.cxx
  62 #define WITHGENERICOBJ
  63
  64 //================================================================
  65 // Function : getActiveView
  66 // Purpose  : Get active view window, returns 0 if no open study frame
  67 //================================================================
  68 SUIT_ViewWindow* GEOMBase_Helper::getActiveView()
  69 {
  70   SUIT_Study* activeStudy = SUIT_Session::session()->activeApplication()->activeStudy();
  71   if ( activeStudy )
  72     return SUIT_Session::session()->activeApplication()->desktop()->activeWindow();
  73
  74   return 0;
  75 }
  76
  77
  78 //================================================================
  79 // Function : getGeomEngine
  80 // Purpose  : Static method
  81 //================================================================
  82 GEOM::GEOM_Gen_ptr GEOMBase_Helper::getGeomEngine()
  83 {
  84   return GeometryGUI::GetGeomGen();
  85 }
  86
  87 //================================================================
  88 // Function : GEOMBase_Helper
  89 // Purpose  :
  90 //================================================================
  91 GEOMBase_Helper::GEOMBase_Helper( SUIT_Desktop* desktop, SUIT_ViewWindow* aVW )
  92   : myDisplayer( 0 ), myCommand( 0 ), myViewWindow( aVW ), isPreview( false ), myDesktop( desktop ),
  93     myIsApplyAndClose( false ), myIsOptimizedBrowsing( false ), myIsWaitCursorEnabled( true ),
  94     myIsDisableBrowsing(false), myIsDisplayResult(true)
  95 {
  96 }
  97
  98 //================================================================
  99 // Function : ~GEOMBase_Helper
 100 // Purpose  :
 101 //================================================================
 102 GEOMBase_Helper::~GEOMBase_Helper()
 103 {
 104   //rnv: Fix for the "IPAL21922 : WinTC5.1.4: incorrect quit salome"
 105   if ( !SUIT_Session::session()->activeApplication() || !SUIT_Session::session()->activeApplication()->desktop() )
 106     return;
 107
 108   if ( myPreview.size() )
 109     erasePreview();
 110   if ( hasCommand() )
 111     abortCommand();
 112   SalomeApp_Application* app = (SalomeApp_Application*)(SUIT_Session::session()->activeApplication());
 113   if (app) {
 114     GeometryGUI* aGeomGUI = dynamic_cast<GeometryGUI*>( app->module( "Geometry" ) );
 115     if(aGeomGUI)
 116       globalSelection(aGeomGUI->getLocalSelectionMode() , true );
 117   }
 118
 119   if (myDisplayer)
 120     delete myDisplayer;
 121   if ( !CORBA::is_nil( myOperation ) )
 122     myOperation->UnRegister();
 123 }
 124
 125 //================================================================
 126 // Function : display
 127 // Purpose  :
 128 //================================================================
 129 void GEOMBase_Helper::display( const ObjectList& objList, const bool updateView )
 130 {
 131   ObjectList::const_iterator it;
 132   for ( it = objList.begin(); it != objList.end(); it++ ) {
 133     display( *it, false );
 134   }
 135   if ( !objList.empty() && updateView )
 136     getDisplayer()->UpdateViewer();
 137 }
 138
 139 //================================================================
 140 // Function  : display
 141 // Purpose   : Display object.
 142 // Important : Object must be already in study
 143 //================================================================
 144 void GEOMBase_Helper::display( GEOM::GEOM_Object_ptr object, const bool updateView )
 145 {
 146   // Unset color of shape ( this color may be set during preview displaying )
 147   // Default color will be used
 148   //   getDisplayer()->UnsetColor();
 149   getDisplayer()->UnsetWidth();
 150
 151   MESSAGE("GEOMBase_Helper::display myTexture = "<<getDisplayer()->GetTexture());
 152
 153   // Enable activisation of selection
 154   getDisplayer()->SetToActivate( true );
 155
 156   // Display object
 157   getDisplayer()->Display( object, updateView );
 158 }
 159
 160 //================================================================
 161 // Function : erase
 162 // Purpose  :
 163 //================================================================
 164 void GEOMBase_Helper::erase( const ObjectList& objList, const bool updateView )
 165 {
 166   ObjectList::const_iterator it = objList.begin();
 167   for ( ; it != objList.end(); it++ ) {
 168     erase( *it, false );
 169   }
 170   if ( !objList.empty() && updateView )
 171     getDisplayer()->UpdateViewer();
 172 }
 173
 174 //================================================================
 175 // Function : erase
 176 // Purpose  :
 177 //================================================================
 178 void GEOMBase_Helper::erase( GEOM::GEOM_Object_ptr object, const bool updateView )
 179 {
 180   if ( !object->_is_nil() ) {
 181     QString entry = getEntry( object );
 182     QString name  = GEOMBase::GetName( object );
 183     getDisplayer()->Erase ( new SALOME_InteractiveObject(entry.toUtf8().constData(),
 184                                                          "GEOM",
 185                                                          name.toUtf8().constData() ),
 186                             true, updateView );
 187   }
 188 }
 189
 190 //================================================================
 191 // Function : redisplay
 192 // Purpose  :
 193 //================================================================
 194 void GEOMBase_Helper::redisplay( const ObjectList& objList,
 195                                  const bool withChildren,
 196                                  const bool updateView )
 197 {
 198   ObjectList::const_iterator it = objList.begin();
 199   for ( ; it != objList.end(); it++ ) {
 200     redisplay( *it, withChildren, false );
 201   }
 202   if ( !objList.empty() && updateView )
 203     getDisplayer()->UpdateViewer();
 204 }
 205
 206 //================================================================
 207 // Function : redisplay
 208 // Purpose  :
 209 //================================================================
 210 void GEOMBase_Helper::redisplay( GEOM::GEOM_Object_ptr object,
 211                                  const bool withChildren,
 212                                  const bool updateView )
 213 {
 214   if ( !object->_is_nil() ) {
 215     // Unset color of shape ( this color may be set during preview displaying )
 216     // Default color will be used
 217     getDisplayer()->UnsetColor();
 218     getDisplayer()->UnsetWidth();
 219
 220     // Enable activisation of selection
 221     getDisplayer()->SetToActivate( true );
 222
 223     QString entry = getEntry( object );
 224     QString  name = GEOMBase::GetName( object );
 225     getDisplayer()->Redisplay
 226       (new SALOME_InteractiveObject (entry.toUtf8().constData(),
 227                                      "GEOM",
 228                                      name.toUtf8().constData()),
 229        false);
 230   }
 231
 232   if ( withChildren ) {
 233     SalomeApp_Study* aDoc = getStudy();
 234     if ( aDoc && aDoc->studyDS() ) {
 235       _PTR(Study) aStudy = aDoc->studyDS();
 236       CORBA::String_var objStr = SalomeApp_Application::orb()->object_to_string(object);
 237       _PTR(SObject) aSObj (aStudy->FindObjectIOR(std::string(objStr.in())));
 238       if ( aSObj  ) {
 239         _PTR(ChildIterator) anIt ( aStudy->NewChildIterator( aSObj ) );
 240         for ( anIt->InitEx( true ); anIt->More(); anIt->Next() ) {
 241           GEOM::GEOM_Object_var aChild = GEOM::GEOM_Object::_narrow
 242             (GeometryGUI::ClientSObjectToObject(anIt->Value()));
 243           if ( !CORBA::is_nil( aChild ) ) {
 244             if ( !aChild->_is_nil() ) {
 245               QString entry = getEntry( aChild );
 246               QString  name = GEOMBase::GetName( aChild );
 247               getDisplayer()->Redisplay
 248                 ( new SALOME_InteractiveObject( entry.toUtf8().constData(),
 249                                                 "GEOM",
 250                                                 name.toUtf8().constData() ),
 251                   false );
 252             }
 253           }
 254         }
 255       }
 256     }
 257   }
 258
 259   if ( updateView )
 260     getDisplayer()->UpdateViewer();
 261 }
 262
 263 //================================================================
 264 // Function : displayPreview
 265 // Purpose  : Method for displaying preview based on execute() results
 266 //================================================================
 267 void GEOMBase_Helper::displayPreview( const bool   display,
 268                                       const bool   activate,
 269                                       const bool   update,
 270                                       const bool   toRemoveFromEngine,
 271                                       const double lineWidth,
 272                                       const int    displayMode,
 273                                       const int    color,
 274                                       const bool   append )
 275 {
 276   if(!display) {
 277     erasePreview( update );
 278     return;
 279   }
 280
 281   isPreview = true;
 282   QString msg;
 283   if ( !isValid( msg ) )
 284   {
 285     erasePreview( update );
 286     isPreview = false;
 287     return;
 288   }
 289
 290   if( !append )
 291     erasePreview( false );
 292
 293   try {
 294     SUIT_OverrideCursor wc;
 295     ObjectList objects;
 296
 297     if ( !isWaitCursorEnabled() )
 298       wc.suspend();
 299
 300     if ( !execute( objects ) || !getOperation()->IsDone() ) {
 301       wc.suspend();
 302     }
 303     else {
 304       for ( ObjectList::iterator it = objects.begin(); it != objects.end(); ++it )
 305       {
 306         GEOM::GEOM_Object_var obj = *it;
 307         displayPreview( obj, true, activate, false, lineWidth, displayMode, color );
 308         if ( toRemoveFromEngine )
 309           obj->UnRegister();
 310       }
 311     }
 312   }
 313   catch( const SALOME::SALOME_Exception& e ) {
 314     SalomeApp_Tools::QtCatchCorbaException( e );
 315   }
 316
 317   isPreview = false;
 318
 319   if ( update )
 320     updateViewer();
 321 }
 322
 323 //================================================================
 324 // Function : displayPreview
 325 // Purpose  : Method for displaying preview of resulting shape
 326 //================================================================
 327 void GEOMBase_Helper::displayPreview( GEOM::GEOM_Object_ptr object,
 328                                       const bool            append,
 329                                       const bool            activate,
 330                                       const bool            update,
 331                                       const double          lineWidth,
 332                                       const int             displayMode,
 333                                       const int             color )
 334 {
 335   SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
 336
 337   // Set color for preview shape
 338   getDisplayer()->SetColor( color == -1 ? Quantity_NOC_VIOLET : color );
 339
 340   // set width of displayed shape
 341   getDisplayer()->SetWidth( lineWidth == -1 ? resMgr->integerValue("Geometry", "preview_edge_width", -1) : lineWidth );
 342
 343   // set display mode of displayed shape
 344   int aPrevDispMode = getDisplayer()->SetDisplayMode( displayMode == -1 ? resMgr->integerValue( "Geometry", "display_mode", 0 ) : displayMode );
 345
 346   // Disable activation of selection
 347   getDisplayer()->SetToActivate( activate );
 348
 349   // Make a reference to GEOM_Object
 350   CORBA::String_var objStr = SalomeApp_Application::orb()->object_to_string( object );
 351   getDisplayer()->SetName( objStr.in() );
 352
 353   // Build prs
 354   SALOME_Prs* aPrs = getDisplayer()->BuildPrs( object );
 355   if ( aPrs == 0 || aPrs->IsNull() )
 356     return;
 357
 358   // Make preview not clippable
 359   aPrs->SetClippable (false);
 360
 361   // Display prs
 362   displayPreview( aPrs, append, update );
 363
 364   getDisplayer()->UnsetName();
 365   getDisplayer()->UnsetColor();
 366   getDisplayer()->SetDisplayMode( aPrevDispMode );
 367
 368   // Enable activation of displayed objects
 369   getDisplayer()->SetToActivate( true );
 370 }
 371
 372 //================================================================
 373 // Function : displayPreview
 374 // Purpose  : Method for displaying arbitrary preview objects (not limited to shapes)
 375 //================================================================
 376 void GEOMBase_Helper::displayPreview( const SALOME_Prs* prs,
 377                                       const bool        append,
 378                                       const bool        update )
 379 {
 380   if ( !append )
 381     erasePreview( false );
 382
 383   // remember current view frame to make correct erase preview later
 384   myViewWindow = getActiveView();
 385
 386   if ( myViewWindow == 0 )
 387     return;
 388
 389   // Display prs
 390   SUIT_ViewManager* aViewManager = myViewWindow->getViewManager();
 391   if ( aViewManager->getType() == OCCViewer_Viewer::Type() ||
 392        aViewManager->getType() == SVTK_Viewer::Type() )
 393   {
 394     SUIT_ViewModel* aViewModel = aViewManager->getViewModel();
 395     SALOME_View* aView = dynamic_cast<SALOME_View*>(aViewModel);
 396     if (aView)
 397       aView->Display( getDisplayer(), prs );
 398   }
 399
 400   // Add prs to the preview list
 401   myPreview.push_back( (SALOME_Prs*)prs );
 402
 403   // Update viewer
 404   if ( update )
 405     getDisplayer()->UpdateViewer();
 406 }
 407
 408 //================================================================
 409 // Function : erasePreview
 410 // Purpose  :
 411 //================================================================
 412 void GEOMBase_Helper::erasePreview( const bool update )
 413 {
 414   // check view frame where the preview was displayed
 415   bool vfOK = checkViewWindow() && myViewWindow;
 416   // Iterate through presentations and delete them
 417   for ( PrsList::iterator anIter = myPreview.begin(); anIter != myPreview.end(); ++anIter )
 418   {
 419     if ( vfOK )
 420     {
 421       SUIT_ViewManager* aViewManager = myViewWindow->getViewManager();
 422       if ( aViewManager->getType() == OCCViewer_Viewer::Type() ||
 423            aViewManager->getType() == SVTK_Viewer::Type() )
 424       {
 425         SUIT_ViewModel* aViewModel = aViewManager->getViewModel();
 426         SALOME_View* aView = dynamic_cast<SALOME_View*>(aViewModel);
 427         if (aView)
 428           aView->Erase( getDisplayer(), *anIter, true );
 429       }
 430     }
 431     delete *anIter;
 432   }
 433   myPreview.clear();
 434
 435   // Update viewer
 436   if ( update )
 437     updateViewer();
 438 }
 439
 440 //================================================================
 441 // Function  : localSelection
 442 // Purpose   : Activate selection of objects of a given type
 443 // IMPORTANT : Works after localSelection( ... ) method call only
 444 //================================================================
 445 void GEOMBase_Helper::activate( const int theType )
 446 {
 447   if (!getStudy()) return;
 448   _PTR(Study) aStudy = getStudy()->studyDS();
 449   _PTR(SComponent) aGeom ( aStudy->FindComponent( "GEOM" ) );
 450   if ( !aGeom )
 451     return;
 452
 453   SALOME_ListIO aList;
 454   _PTR(ChildIterator) anIter ( aStudy->NewChildIterator( aGeom ) );
 455   for ( ; anIter->More(); anIter->Next() )
 456   {
 457     _PTR(SObject) aSO ( anIter->Value() );
 458     if ( aSO )
 459     {
 460       _PTR(SObject) aRefSO;
 461       if ( !aSO->ReferencedObject( aRefSO ) )
 462       {
 463         GEOM::GEOM_Object_var anObj = GEOM::GEOM_Object::_narrow
 464           (GeometryGUI::ClientSObjectToObject(aSO));
 465         if ( !anObj->_is_nil() && anObj->GetType() == theType )
 466           aList.Append( new SALOME_InteractiveObject( aSO->GetID().c_str(), "GEOM", aSO->GetName().c_str()) );
 467       }
 468     }
 469   }
 470
 471   getDisplayer()->LocalSelection( aList, 0 );
 472 }
 473
 474 //================================================================
 475 // Function : localSelection
 476 // Purpose  : Activate selection of sub-shapes in accordance with mode
 477 //            modes are from TopAbs_ShapeEnum
 478 //================================================================
 479 void GEOMBase_Helper::localSelection( const ObjectList& theObjs, const std::list<int> modes )
 480 {
 481   SALOME_ListIO aListOfIO;
 482
 483   ObjectList::const_iterator anIter = theObjs.begin();
 484   for ( ; anIter != theObjs.end(); ++anIter )
 485   {
 486     GEOM::GEOM_Object_ptr anObj = *anIter;
 487     if ( anObj->_is_nil() )
 488       continue;
 489     QString anEntry = getEntry( anObj );
 490     if ( anEntry != "" )
 491     {
 492       //MESSAGE("anEntry: "<< anEntry.toStdString().c_str());
 493       QString aName = GEOMBase::GetName( anObj );
 494       aListOfIO.Append( new SALOME_InteractiveObject( anEntry.toUtf8().constData(),
 495                                                       "GEOM", aName.toUtf8().constData() ));
 496     }
 497   }
 498
 499   getDisplayer()->LocalSelection( aListOfIO, modes );
 500 }
 501
 502 //================================================================
 503 // Function : localSelection
 504 // Purpose  : Activate selection of sub-shapes in accordance with mode
 505 //            theMode is from TopAbs_ShapeEnum
 506 //================================================================
 507 void GEOMBase_Helper::localSelection( const ObjectList& theObjs, const int theMode )
 508 {
 509   std::list<int> modes;
 510   modes.push_back( theMode );
 511   localSelection( theObjs, modes );
 512 }
 513
 514 //================================================================
 515 // Function : localSelection
 516 // Purpose  : Activate selection of sub-shapes in accordance with mode
 517 //            modes are from TopAbs_ShapeEnum
 518 //================================================================
 519 void GEOMBase_Helper::localSelection( GEOM::GEOM_Object_ptr obj, const std::list<int> modes )
 520 {
 521   // If object is null local selection for all objects is activated
 522   if ( obj->_is_nil() ) {
 523     getDisplayer()->LocalSelection( Handle(SALOME_InteractiveObject)(), modes );
 524     return;
 525   }
 526
 527   ObjectList objList;
 528   objList.push_back( obj );
 529   localSelection( objList, modes );
 530 }
 531
 532 //================================================================
 533 // Function : localSelection
 534 // Purpose  : Activate selection of sub-shapes in accordance with mode
 535 //            mode is from TopAbs_ShapeEnum
 536 //================================================================
 537 void GEOMBase_Helper::localSelection( GEOM::GEOM_Object_ptr obj, const int mode )
 538 {
 539   std::list<int> modes;
 540   modes.push_back( mode );
 541   localSelection( obj, modes );
 542 }
 543
 544 //================================================================
 545 // Function : localSelection
 546 // Purpose  : Activate selection of sub-shapes in accordance with mode
 547 //            mode is from TopAbs_ShapeEnum
 548 //================================================================
 549 void GEOMBase_Helper::localSelection( const std::string& entry,  const std::string& name, const std::list<int> modes)
 550 {
 551   SALOME_ListIO aListOfIO;
 552   QString aName = name.c_str();
 553   aListOfIO.Append( new SALOME_InteractiveObject( entry.c_str(),
 554                                                   "GEOM", aName.toUtf8().constData() ));
 555   getDisplayer()->LocalSelection( aListOfIO, modes );
 556 }
 557
 558 //================================================================
 559 // Function : localSelection
 560 // Purpose  : Activate selection of sub-shapes in accordance with mode
 561 //            modes are from TopAbs_ShapeEnum
 562 //================================================================
 563 void GEOMBase_Helper::localSelection( const std::list<int> modes )
 564 {
 565   localSelection( GEOM::GEOM_Object::_nil(), modes );
 566 }
 567
 568 //================================================================
 569 // Function : localSelection
 570 // Purpose  : Activate selection of sub-shapes in accordance with mode
 571 //            mode is from TopAbs_ShapeEnum
 572 //================================================================
 573 void GEOMBase_Helper::localSelection( const int mode )
 574 {
 575   std::list<int> modes;
 576   modes.push_back( mode );
 577   localSelection( modes );
 578 }
 579
 580 //================================================================
 581 // Function : globalSelection
 582 // Purpose  : Activate selection of sub-shapes. Set selection filters
 583 //            in accordance with mode. theMode is from GEOMImpl_Types
 584 //================================================================
 585 void GEOMBase_Helper::globalSelection( const int theMode, const bool update )
 586 {
 587   getDisplayer()->GlobalSelection( theMode, update );
 588 }
 589
 590 //================================================================
 591 // Function : globalSelection
 592 // Purpose  : Activate selection of sub-shapes. Set selection filters
 593 //            in accordance with mode. theMode is from GEOMImpl_Types
 594 //================================================================
 595 void GEOMBase_Helper::globalSelection( const TColStd_MapOfInteger& theModes,
 596                                        const bool update )
 597 {
 598   getDisplayer()->GlobalSelection( theModes, update );
 599 }
 600
 601 //================================================================
 602 // Function : globalSelection
 603 // Purpose  : Activate selection of sub-shapes. Set selection filters
 604 //            in accordance with mode. theMode is from GEOMImpl_Types
 605 //================================================================
 606 void GEOMBase_Helper::globalSelection( const TColStd_MapOfInteger& theModes,
 607                                        const QList<int>& subShapes,
 608                                        const bool update )
 609 {
 610   getDisplayer()->GlobalSelection( theModes, update, &subShapes );
 611 }
 612
 613 //================================================================
 614 // Function : addInStudy
 615 // Purpose  : Add object in study
 616 //================================================================
 617 QString GEOMBase_Helper::addInStudy( GEOM::GEOM_Object_ptr theObj, const char* theName )
 618 {
 619   if ( !hasCommand() )
 620     return QString();
 621
 622   _PTR(Study) aStudy = getStudy()->studyDS();
 623   if ( !aStudy || theObj->_is_nil() )
 624     return QString();
 625
 626   GEOM::GEOM_Object_ptr aFatherObj = getFather( theObj );
 627
 628   SALOMEDS::SObject_var aSO =
 629     getGeomEngine()->AddInStudy(theObj, theName, aFatherObj);
 630
 631   QString anEntry;
 632   if ( !aSO->_is_nil() ) {
 633     CORBA::String_var entry = aSO->GetID();
 634     anEntry = entry.in();
 635   }
 636   // Each dialog is responsible for this method implementation,
 637   // default implementation does nothing
 638   restoreSubShapes(aSO);
 639   aSO->UnRegister();
 640
 641   return anEntry;
 642 }
 643
 644 //================================================================
 645 // Function : restoreSubShapes
 646 // Purpose  : restore tree of argument's sub-shapes under the resulting shape
 647 //================================================================
 648 void GEOMBase_Helper::restoreSubShapes (SALOMEDS::SObject_ptr /*theSObject*/)
 649 {
 650   // do nothing by default
 651
 652   // example of implementation in particular dialog:
 653   // GEOM::ListOfGO anArgs;
 654   // anArgs.length(0); // empty list means that all arguments should be restored
 655   // getGeomEngine()->RestoreSubShapesSO(theSObject, anArgs,
 656   //                                     /*theFindMethod=*/GEOM::FSM_GetInPlace,
 657   //                                     /*theInheritFirstArg=*/false);
 658 }
 659
 660 //================================================================
 661 // Function : updateObjBrowser
 662 // Purpose  : Update object browser
 663 //================================================================
 664 void GEOMBase_Helper::updateObjBrowser() const
 665 {
 666   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>(SUIT_Session::session()->activeApplication());
 667   if (app) {
 668     CAM_Module* module = app->module( "Geometry" );
 669     SalomeApp_Module* appMod = dynamic_cast<SalomeApp_Module*>( module );
 670     if ( appMod ) {
 671       appMod->updateObjBrowser();
 672     }
 673   }
 674 }
 675
 676 //================================================================
 677 // Function : updateViewer
 678 // Purpose  : Update active 3D view
 679 //================================================================
 680 void GEOMBase_Helper::updateViewer()
 681 {
 682   getDisplayer()->UpdateViewer();
 683 }
 684
 685 //================================================================
 686 // Function : getStudy
 687 // Purpose  : Returns the active study. It is recommended to use
 688 //            this method instead of direct desktop->getActiveStudy() calls
 689 //================================================================
 690 SalomeApp_Study* GEOMBase_Helper::getStudy() const
 691 {
 692   SUIT_Desktop* aDesktop = getDesktop();
 693   if (!aDesktop)
 694     return 0;
 695
 696   QList<SUIT_Application*> anAppList = SUIT_Session::session()->applications();
 697
 698   SUIT_Application* anApp = 0;
 699   QListIterator<SUIT_Application*> it( anAppList );
 700   while ( it.hasNext() )
 701   {
 702     anApp = it.next();
 703     if ( anApp && anApp->desktop() == aDesktop )
 704       break;
 705   }
 706
 707   return dynamic_cast<SalomeApp_Study*>(anApp->activeStudy());
 708 }
 709
 710 //================================================================
 711 // Function : getEntry
 712 // Purpose  :
 713 //================================================================
 714 QString GEOMBase_Helper::getEntry( GEOM::GEOM_Object_ptr object ) const
 715 {
 716   SalomeApp_Study* study = getStudy();
 717   if ( study )  {
 718     QString objIOR = GEOMBase::GetIORFromObject( object );
 719     if ( objIOR != "" ) {
 720       _PTR(SObject) SO ( study->studyDS()->FindObjectIOR( objIOR.toUtf8().constData() ) );
 721       if ( SO )
 722         return QString::fromStdString(SO->GetID());
 723     }
 724   }
 725   return "";
 726 }
 727
 728 //================================================================
 729 // Function : getDisplayer
 730 // Purpose  :
 731 //================================================================
 732 GEOM_Displayer* GEOMBase_Helper::getDisplayer()
 733 {
 734   if ( !myDisplayer )
 735     myDisplayer = new GEOM_Displayer();
 736   return myDisplayer;
 737 }
 738
 739 //================================================================
 740 // Function : clearShapeBuffer
 741 // Purpose  :
 742 //================================================================
 743 void GEOMBase_Helper::clearShapeBuffer( GEOM::GEOM_Object_ptr theObj )
 744 {
 745   GeometryGUI::ClearShapeBuffer(theObj);
 746 }
 747
 748 //================================================================
 749 // Function : openCommand
 750 // Purpose  :
 751 //================================================================
 752 bool GEOMBase_Helper::openCommand()
 753 {
 754   bool res = false;
 755   if ( !getStudy() || hasCommand() )
 756   {
 757     MESSAGE("Getting out from openCommand()")
 758       return res;
 759   }
 760
 761   GEOM::GEOM_IOperations_var anOp = GEOM::GEOM_IOperations::_narrow( getOperation() );
 762   if ( !anOp->_is_nil() ) {
 763     myCommand = new GEOM_Operation( SUIT_Session::session()->activeApplication(), anOp.in() );
 764     myCommand->start();
 765     res = true;
 766   }
 767   else
 768   {
 769     MESSAGE("anOp->_is_nil() = true")
 770       }
 771
 772   return res;
 773 }
 774
 775 //================================================================
 776 // Function : abortCommand
 777 // Purpose  :
 778 //================================================================
 779 bool GEOMBase_Helper::abortCommand()
 780 {
 781   if ( !hasCommand() )
 782     return false;
 783
 784   myCommand->abort();
 785   delete myCommand; // I don't know where to delete this object here ?
 786   myCommand = 0;
 787
 788   return true;
 789 }
 790
 791 //================================================================
 792 // Function : commitCommand
 793 // Purpose  :
 794 //================================================================
 795 bool GEOMBase_Helper::commitCommand( const char* )
 796 {
 797   if ( !hasCommand() )
 798     return false;
 799
 800   myCommand->commit();
 801   delete myCommand; // I don't know where to delete this object here ?
 802   myCommand = 0;
 803
 804   return true;
 805 }
 806
 807 //================================================================
 808 // Function : hasCommand
 809 // Purpose  :
 810 //================================================================
 811 bool GEOMBase_Helper::hasCommand() const
 812 {
 813   return (bool)myCommand;
 814 }
 815
 816 //================================================================
 817 // Function : getOperation
 818 // Purpose  :
 819 //================================================================
 820 GEOM::GEOM_IOperations_ptr GEOMBase_Helper::getOperation()
 821 {
 822   if ( myOperation->_is_nil() )
 823     myOperation = createOperation();
 824
 825   return myOperation;
 826 }
 827
 828
 829
 830 //================================================================
 831 // Function : checkViewWindow
 832 // Purpose  :
 833 //================================================================
 834 bool GEOMBase_Helper::checkViewWindow()
 835 {
 836   if ( myViewWindow ){
 837     QList<SUIT_ViewWindow*> aViewWindowsList = SUIT_Session::session()->activeApplication()->desktop()->windows();
 838     QListIterator<SUIT_ViewWindow*> it( aViewWindowsList );
 839     while ( it.hasNext() )
 840     {
 841       if ( myViewWindow == it.next() )
 842         return true;
 843     }
 844   }
 845   myViewWindow = 0;
 846   return false;
 847 }
 848
 849 //================================================================
 850 // Function : onAccept
 851 // Purpose  : This method should be called from dialog's slots onOk() and onApply()
 852 //            It performs user input validation, then it
 853 //            performs a proper operation and manages transactions, etc.
 854 //================================================================
 855 bool GEOMBase_Helper::onAccept( const bool publish, const bool useTransaction, bool erasePreviewFlag )
 856 {
 857   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 858   if ( !appStudy )
 859   {
 860     MESSAGE("appStudy is empty")
 861       return false;
 862   }
 863   _PTR(Study) aStudy = appStudy->studyDS();
 864
 865   bool aLocked = (_PTR(AttributeStudyProperties) (aStudy->GetProperties()))->IsLocked();
 866   if ( aLocked ) {
 867     MESSAGE("GEOMBase_Helper::onAccept - ActiveStudy is locked");
 868     SUIT_MessageBox::warning ( (QWidget*)SUIT_Session::session()->activeApplication()->desktop(),
 869                                QObject::tr("WRN_WARNING"),
 870                                QObject::tr("WRN_STUDY_LOCKED"),
 871                                QObject::tr("BUT_OK") );
 872     return false;
 873   }
 874
 875   QString msg;
 876   if ( !isValid( msg ) ) {
 877     showError( msg );
 878     return false;
 879   }
 880
 881   if(erasePreviewFlag)
 882     erasePreview( false );
 883
 884   bool result = false;
 885
 886   try {
 887     if ( ( !publish && !useTransaction ) || openCommand() ) {
 888       SUIT_OverrideCursor wc;
 889       SUIT_Session::session()->activeApplication()->putInfo( "" );
 890       ObjectList objects;
 891       if ( !execute( objects ) || !getOperation()->IsDone() ) {
 892         wc.suspend();
 893         abortCommand();
 894         showError();
 895       }
 896       else {
 897         QList<GEOM::GeomObjPtr> anObjectList = getSourceObjects( );
 898         addSubshapesToStudy(); // add Sub-shapes if local selection
 899         const int nbObjs = objects.size();
 900         QStringList anEntryList;
 901         int currObj = 1, aNumber = 1;
 902         for ( ObjectList::iterator it = objects.begin(); it != objects.end(); ++it, currObj++ ) {
 903           GEOM::GEOM_Object_var obj=*it;
 904           if ( publish ) {
 905             QString aName = getObjectName(obj);
 906             if (aName.isEmpty()) {
 907               aName = getNewObjectName(currObj);
 908               if ( nbObjs > 1 ) {
 909                 if (aName.isEmpty())
 910                   aName = getPrefix(obj);
 911                 if (nbObjs <= 30) {
 912                   // Try to find a unique name
 913                   aName = GEOMBase::GetDefaultName(aName, extractPrefix());
 914                 } else {
 915                   // Don't check name uniqueness in case of numerous objects
 916                   aName = aName + "_" + QString::number(aNumber++);
 917                 }
 918               } else {
 919                 // PAL6521: use a prefix, if some dialog box doesn't reimplement getNewObjectName()
 920                 if ( aName.isEmpty() )
 921                   aName = GEOMBase::GetDefaultName( getPrefix( obj ) );
 922               }
 923             }
 924             anEntryList << addInStudy( obj, aName.toUtf8().constData() );
 925             // updateView=false
 926             if( isDisplayResult() )
 927               display( obj, false );
 928 #ifdef WITHGENERICOBJ
 929             // obj has been published in study. Its refcount has been incremented.
 930             // It is safe to decrement its refcount
 931             // so that it will be destroyed when the entry in study will be removed
 932             obj->UnRegister();
 933 #endif
 934           }
 935           else {
 936             // asv : fix of PAL6454. If publish==false, then the original shape
 937             // was modified, and need to be re-cached in GEOM_Client before redisplay
 938             clearShapeBuffer( obj );
 939             // withChildren=true, updateView=false
 940             if( isDisplayResult() )
 941               redisplay( obj, true, false );
 942           }
 943         }
 944
 945         if ( nbObjs ) {
 946           commitCommand();
 947           updateObjBrowser();
 948           if( SUIT_Application* anApp = SUIT_Session::session()->activeApplication() ) {
 949             LightApp_Application* aLightApp = dynamic_cast<LightApp_Application*>( anApp );
 950             if(aLightApp) {
 951               QString anOpName( typeid(*this).name() );
 952               aLightApp->emitOperationFinished( "Geometry", anOpName, anEntryList );
 953
 954               if ( !isDisableBrowsing() )
 955                 aLightApp->browseObjects( anEntryList, isApplyAndClose(), isOptimizedBrowsing() );
 956             }
 957             anApp->putInfo( QObject::tr("GEOM_PRP_DONE") );
 958           }
 959           if ( anObjectList.count() > 0 )
 960             hideSourceObjects( anObjectList );
 961           result = true;
 962         }
 963         else
 964           abortCommand();
 965       }
 966     }
 967   }
 968   catch( const SALOME::SALOME_Exception& e ) {
 969     SalomeApp_Tools::QtCatchCorbaException( e );
 970     abortCommand();
 971     MESSAGE("Exception caught")
 972       }
 973
 974   updateViewer();
 975
 976   MESSAGE("result ="<<result)
 977     return result;
 978 }
 979
 980
 981 //================================================================
 982 // Function : showError
 983 // Purpose  : Shows a message box with information about an error taken from getOperation()->GetErrorCode()
 984 //================================================================
 985 void GEOMBase_Helper::showError()
 986 {
 987   QString msg;
 988   if ( !getOperation()->_is_nil() )
 989     msg = QObject::tr( getOperation()->GetErrorCode() );
 990
 991   if ( msg.isEmpty() )
 992     msg = QObject::tr( "GEOM_PRP_ABORT" );
 993
 994   SUIT_MessageBox::critical( SUIT_Session::session()->activeApplication()->desktop(),
 995                              QObject::tr( "GEOM_ERROR_STATUS" ),
 996                              msg,
 997                              QObject::tr( "BUT_OK" ) );
 998 }
 999
1000 //================================================================
1001 // Function : showError
1002 // Purpose  : Shows a error message followed by <msg>
1003 //================================================================
1004 void GEOMBase_Helper::showError( const QString& msg )
1005 {
1006   QString str( QObject::tr( "GEOM_INCORRECT_INPUT" ) );
1007   if ( !msg.isEmpty() )
1008     str += "\n" + msg;
1009   SUIT_MessageBox::critical(SUIT_Session::session()->activeApplication()->desktop(), QObject::tr( "GEOM_ERROR" ), str, QObject::tr( "BUT_OK" ) );
1010 }
1011
1012 //////////////////////////////////////////////////////////////////
1013 // Virtual methods to be redefined in dialogs
1014 //////////////////////////////////////////////////////////////////
1015
1016 //================================================================
1017 // Function : createOperation
1018 // Purpose  : Redefine this method to return proper IOperation reference
1019 //================================================================
1020 GEOM::GEOM_IOperations_ptr GEOMBase_Helper::createOperation()
1021 {
1022   GEOM::GEOM_IOperations_var aNilOp;
1023   return aNilOp._retn();
1024 }
1025
1026 //================================================================
1027 // Function : isValid
1028 // Purpose  : Called by onAccept(). Redefine this method to check validity of user input in dialog boxes.
1029 //================================================================
1030 bool GEOMBase_Helper::isValid( QString& )
1031 {
1032   return true;
1033 }
1034
1035 //================================================================
1036 // Function : execute
1037 // Purpose  : This method is called by onAccept().
1038 //            It should perform the required operation and put all new or modified objects into
1039 //            <objects> argument.Should return <false> if some error occurs during its execution.
1040 //================================================================
1041 bool GEOMBase_Helper::execute( ObjectList& )
1042 {
1043   return false;
1044 }
1045
1046 //================================================================
1047 // Function : getFather
1048 // Purpose  : This method is called by addInStudy(). It should return a father object
1049 //            for <theObj> or a nil reference if <theObj> should be published
1050 //            as a top-level object.
1051 //================================================================
1052 GEOM::GEOM_Object_ptr GEOMBase_Helper::getFather( GEOM::GEOM_Object_ptr )
1053 {
1054   return GEOM::GEOM_Object::_nil();
1055 }
1056
1057 //================================================================
1058 // Function : getObjectName
1059 // Purpose  : Redefine this method to return proper name for the given object
1060 //================================================================
1061 QString GEOMBase_Helper::getObjectName( GEOM::GEOM_Object_ptr ) const
1062 {
1063   return QString();
1064 }
1065
1066 //================================================================
1067 // Function : getNewObjectName
1068 // Purpose  : Redefine this method to return proper name for a new object
1069 //================================================================
1070 QString GEOMBase_Helper::getNewObjectName( int ) const
1071 {
1072   return QString();
1073 }
1074
1075 //================================================================
1076 // Function : extractPrefix
1077 // Purpose  : Redefine this method to return \c true if necessary
1078 //            to extract prefix when generating new name for the
1079 //            object(s) being created
1080 //================================================================
1081 bool GEOMBase_Helper::extractPrefix() const
1082 {
1083   return false;
1084 }
1085
1086 //================================================================
1087 // Function : getPrefix
1088 // Purpose  : Get prefix for name of created object
1089 //================================================================
1090 QString GEOMBase_Helper::getPrefix( GEOM::GEOM_Object_ptr theObj ) const
1091 {
1092   if ( !myPrefix.isEmpty() || theObj->_is_nil() )
1093     return myPrefix;
1094
1095   GEOM::shape_type aType = theObj->GetShapeType();
1096
1097   switch ( aType )
1098   {
1099   case GEOM::VERTEX   : return QObject::tr( "GEOM_VERTEX" );
1100   case GEOM::EDGE     : return QObject::tr( "GEOM_EDGE" );
1101   case GEOM::WIRE     : return QObject::tr( "GEOM_WIRE" );
1102   case GEOM::FACE     : return QObject::tr( "GEOM_FACE" );
1103   case GEOM::SHELL    : return QObject::tr( "GEOM_SHELL" );
1104   case GEOM::SOLID    : return QObject::tr( "GEOM_SOLID" );
1105   case GEOM::COMPSOLID: return QObject::tr( "GEOM_COMPOUNDSOLID" );
1106   case GEOM::COMPOUND : return QObject::tr( "GEOM_COMPOUND" );
1107   default : return "";
1108   }
1109 }
1110
1111 //================================================================
1112 // Function : getDesktop
1113 // Purpose  : Returns myDesktop field. Initialized in constructor,
1114 //            usually as dynamic_cast<SUIT_Desktop*>(parentWidget())
1115 //================================================================
1116 SUIT_Desktop* GEOMBase_Helper::getDesktop() const
1117 {
1118   return myDesktop;
1119 }
1120
1121 //================================================================
1122 // Function : selectObjects
1123 // Purpose  : Selects list of objects
1124 //================================================================
1125 bool GEOMBase_Helper::selectObjects( ObjectList& objects )
1126 {
1127   SUIT_DataOwnerPtrList aList;
1128   ObjectList::iterator anIter;
1129   for ( anIter = objects.begin(); anIter != objects.end(); ++anIter )
1130   {
1131     QString anEntry = getEntry( *anIter );
1132     LightApp_DataOwner* anOwher = new LightApp_DataOwner( anEntry );
1133     aList.append( anOwher );
1134   }
1135
1136   SUIT_Session* session = SUIT_Session::session();
1137   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
1138   if ( !app )
1139     return false;
1140
1141   LightApp_SelectionMgr* aMgr = app->selectionMgr();
1142   if ( !aMgr )
1143     return false;
1144
1145   aMgr->setSelected( aList, false );
1146
1147   return true;
1148 }
1149
1150 //================================================================
1151 // Function : findObjectInFather
1152 // Purpose  : It should return an object if its founded in study or
1153 //            return Null object if the object is not founded
1154 //================================================================
1155 GEOM::GEOM_Object_ptr GEOMBase_Helper::findObjectInFather (GEOM::GEOM_Object_ptr theFather,
1156                                                            const QString& theName)
1157 {
1158   SalomeApp_Application* app =
1159     dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
1160   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
1161   _PTR(Study) aDStudy = appStudy->studyDS();
1162   QString IOR = GEOMBase::GetIORFromObject( theFather );
1163   _PTR(SObject) SObj ( aDStudy->FindObjectIOR( IOR.toUtf8().constData() ) );
1164
1165   bool inStudy = false;
1166   GEOM::GEOM_Object_var aReturnObject;
1167   for (_PTR(ChildIterator) iit (aDStudy->NewChildIterator( SObj )); iit->More() && !inStudy; iit->Next()) {
1168     _PTR(SObject) child (iit->Value());
1169     QString aChildName = child->GetName().c_str();
1170     if (aChildName == theName) {
1171       inStudy = true;
1172       CORBA::Object_var corbaObj = GeometryGUI::ClientSObjectToObject(iit->Value());
1173       aReturnObject = GEOM::GEOM_Object::_narrow( corbaObj );
1174     }
1175   }
1176   if (inStudy)
1177     return aReturnObject._retn();
1178
1179   return GEOM::GEOM_Object::_nil();
1180 }
1181
1182 //================================================================
1183 // Function : findObjectInFather
1184 // Purpose  : It should return an object if its founded in study or
1185 //            return Null object if the object is not founded
1186 //================================================================
1187 GEOM::GEOM_Object_ptr GEOMBase_Helper::findObjectInFather( GEOM::GEOM_Object_ptr theFather,
1188                                                            int theIndex )
1189 {
1190   GEOM::GEOM_Object_var object;
1191
1192   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
1193   if ( study ) {
1194     _PTR(Study) studyDS = study->studyDS();
1195     QString IOR = GEOMBase::GetIORFromObject( theFather );
1196     _PTR(SObject) sobject( studyDS->FindObjectIOR( IOR.toLatin1().constData() ) );
1197     if ( sobject ) {
1198       _PTR(ChildIterator) it( studyDS->NewChildIterator( sobject ) );
1199       for ( ; it->More(); it->Next() ) {
1200         GEOM::GEOM_Object_var cobject = GEOM::GEOM_Object::_narrow( GeometryGUI::ClientSObjectToObject( it->Value() ) );
1201         if ( !CORBA::is_nil( cobject ) ) {
1202           GEOM::ListOfLong_var indices = cobject->GetSubShapeIndices();
1203           int length = indices->length();
1204           // VSR 18/03/2014: we need only sub-shapes with single sub-shape index (to exclude groups, etc)
1205           if ( length == 1 && indices[0] == theIndex ) {
1206             object = cobject;
1207             break;
1208           }
1209         }
1210       }
1211     }
1212   }
1213
1214   return object._retn();
1215 }
1216
1217 //================================================================
1218 // Function : addSubshapesToStudy
1219 // Purpose  : Virtual method to add sub-shapes if needs
1220 //================================================================
1221 void GEOMBase_Helper::addSubshapesToStudy()
1222 {
1223   //Implemented in Dialogs, called from Accept method
1224 }
1225
1226 //================================================================
1227 // Function : getSourceObjects
1228 // Purpose  : Virtual method to get source objects
1229 //================================================================
1230 QList<GEOM::GeomObjPtr> GEOMBase_Helper::getSourceObjects()
1231 {
1232   //Implemented in Dialogs, called from Accept method
1233   QList<GEOM::GeomObjPtr> res;
1234   return res;
1235 }
1236
1237 //================================================================
1238 // Function : getSelected
1239 // Purpose  : Get selected object by specified type
1240 //
1241 // Returns valid object if only one object of the specified type is selected
1242 // (no matter global or local selection is activated). If \a type is TopAbs_SHAPE,
1243 // geometrical object of any valid type is expected.
1244 //
1245 // \param type type of the object to be obtained from selection
1246 // \return selected geometrical object or nil object if selection is not satisfactory
1247 //================================================================
1248 GEOM::GeomObjPtr GEOMBase_Helper::getSelected( TopAbs_ShapeEnum type )
1249 {
1250   QList<TopAbs_ShapeEnum> types;
1251   types << type;
1252   return getSelected( types );
1253 }
1254
1255 //================================================================
1256 // Function : getSelected
1257 // Purpose  : Get selected object by specified types
1258 //
1259 // Returns valid object if only one object of the specified type is selected
1260 // (no matter global or local selection is activated). The list of allowed
1261 // shape types is passed via \a types. If \a types includes TopAbs_SHAPE,
1262 // geometrical object of any valid type is expected.
1263 //
1264 // \param types list of allowed shape types for the objects to be obtained from selection
1265 // \return selected geometrical object or nil object if selection is not satisfactory
1266 //================================================================
1267 GEOM::GeomObjPtr GEOMBase_Helper::getSelected( const QList<TopAbs_ShapeEnum>& types )
1268 {
1269   QList<GEOM::GeomObjPtr> selected = getSelected( types, 1 );
1270   return selected.count() > 0 ? selected[0] : GEOM::GeomObjPtr();
1271 }
1272
1273 //================================================================
1274 // Function : getSelected
1275 // Purpose  : Get selected object(s) by specified type
1276 //
1277 // Returns list of selected objects if selection satisfies specifies selection options.
1278 // (no matter global or local selection is activated). If \a type is TopAbs_SHAPE,
1279 // geometrical objects of any valid type are expected.
1280 //
1281 // The \a type parameter specifies allowed type of the object(s) being selected.
1282 // The \a count parameter specifies exact number of the objects to be retrieved from selection.
1283 // The \a strict parameter specifies policy being applied to the selection.
1284 // If \a count < 0, then any number of the selected objects is valid (including 0).
1285 // In this case, if \a strict is \c true (default), all selected objects should satisfy
1286 // the specified \a type.
1287 // If \a count > 0, only specified number of the objects is retrieved from the selection.
1288 // In this case, if \a strict is \c true (default), function returns empty list if total number of selected
1289 // objects does not correspond to the \a count parameter. Otherwise (if \a strict is \c false),
1290 // function returns valid list of objects if at least \a count objects satisfy specified \a type.
1291 //
1292 // \param type type of the object(s) to be obtained from selection
1293 // \param count number of items to be retrieved from selection
1294 // \param strict selection policy
1295 // \return list of selected geometrical objects or empty list if selection is not satisfactory
1296 //================================================================
1297 QList<GEOM::GeomObjPtr> GEOMBase_Helper::getSelected( TopAbs_ShapeEnum type, int count, bool strict )
1298 {
1299   QList<TopAbs_ShapeEnum> types;
1300   types << type;
1301   return getSelected( types, count, strict );
1302 }
1303
1304 static bool typeInList( TopAbs_ShapeEnum type, const QList<TopAbs_ShapeEnum>& types )
1305 {
1306   bool ok = false;
1307   for ( int i = 0; i < types.count() && !ok; i++ )
1308     ok = types[i] == TopAbs_SHAPE || types[i] == type;
1309   return ok;
1310 }
1311
1312 //================================================================
1313 // Function : getSelected
1314 // Purpose  : Get selected objects by specified types
1315 //
1316 // Returns list of selected objects if selection satisfies specifies selection options.
1317 // (no matter global or local selection is activated). If \a types includes TopAbs_SHAPE,
1318 // geometrical objects of any valid type are expected.
1319 //
1320 // The \a types parameter specifies allowed types of the object(s) being selected.
1321 // The \a count parameter specifies exact number of the objects to be retrieved from selection.
1322 // The \a strict parameter specifies policy being applied to the selection.
1323 // If \a count < 0, then any number of the selected objects is valid (including 0).
1324 // In this case, if \a strict is \c true (default), all selected objects should satisfy
1325 // the specified \a type.
1326 // If \a count > 0, only specified number of the objects is retrieved from the selection.
1327 // In this case, if \a strict is \c true (default), function returns empty list if total number of selected
1328 // objects does not correspond to the \a count parameter. Otherwise (if \a strict is \c false),
1329 // function returns valid list of objects if at least \a count objects satisfy specified \a type.
1330 //
1331 // \param types list of allowed shape types for the objects to be obtained from selection
1332 // \param count number of items to be retrieved from selection
1333 // \param strict selection policy
1334 // \return list of selected geometrical objects or empty list if selection is not satisfactory
1335 //================================================================
1336 QList<GEOM::GeomObjPtr> GEOMBase_Helper::getSelected( const QList<TopAbs_ShapeEnum>& types, int count, bool strict )
1337 {
1338   SUIT_Session* session = SUIT_Session::session();
1339   QList<GEOM::GeomObjPtr> result;
1340
1341   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
1342   if ( app ) {
1343     LightApp_SelectionMgr* selMgr = app->selectionMgr();
1344     if ( selMgr ) {
1345       SALOME_ListIO selected;
1346       selMgr->selectedObjects( selected );
1347       SALOME_ListIteratorOfListIO it( selected );
1348       bool stopped = false;
1349       for ( ; it.More() && !stopped; it.Next() ) {
1350         Handle(SALOME_InteractiveObject) IO = it.Value();
1351         GEOM::GeomObjPtr object = GEOMBase::ConvertIOinGEOMObject( IO );
1352         if ( object ) {
1353           TColStd_IndexedMapOfInteger subShapes;
1354           selMgr->GetIndexes( IO, subShapes );
1355           int nbSubShapes = subShapes.Extent();
1356           if ( nbSubShapes == 0 ) {
1357             // global selection
1358             if ( typeInList( (TopAbs_ShapeEnum)(object->GetShapeType()), types ) ) {
1359               result << object;
1360               if ( count > 0 ) {
1361                 if ( strict && result.count() > count ) {
1362                   result.clear();
1363                   stopped = true;
1364                 }
1365                 else if ( !strict && result.count() == count )
1366                   stopped = true;
1367               }
1368             }
1369             else if ( strict ) {
1370               result.clear();
1371               stopped = true;
1372             }
1373           }
1374           else {
1375             // local selection
1376             for ( int i = 1; i <= nbSubShapes && !stopped; i++ ) {
1377               int idx = subShapes( i );
1378               GEOM::GeomObjPtr subShape = findObjectInFather( object.get(), idx );
1379               if ( !subShape ) {
1380                 // sub-shape is not yet published in the study
1381                 GEOM::ShapesOpPtr shapesOp = getGeomEngine()->GetIShapesOperations();
1382                 subShape.take( shapesOp->GetSubShape( object.get(), idx ) ); // take ownership!
1383               }
1384               if ( typeInList( (TopAbs_ShapeEnum)(subShape->GetShapeType()), types ) ) {
1385                 result << subShape;
1386                 if ( count > 0 ) {
1387                   if ( strict && result.count() > count ) {
1388                     result.clear();
1389                     stopped = true;
1390                   }
1391                   else if ( !strict && result.count() == count )
1392                     stopped = true;
1393                 }
1394               }
1395               else if ( strict ) {
1396                 result.clear();
1397                 stopped = true;
1398               }
1399             }
1400           }
1401         }
1402       }
1403     }
1404   }
1405   return result;
1406 }
1407 //================================================================
1408 // Function : hideSourceObject
1409 // Purpose  :
1410 //================================================================
1411 void GEOMBase_Helper::hideSourceObjects( QList<GEOM::GeomObjPtr> theObject )
1412 {
1413   SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
1414   if ( resMgr->booleanValue( "Geometry", "hide_input_object", true) ) {
1415     GEOM_Displayer* aDisplayer = getDisplayer();
1416     for ( int i = 0; i < theObject.count(); i++ )
1417       aDisplayer->Erase( theObject[i].get() );
1418   }
1419 }
1420
1421 //================================================================
1422 // Function : setIsApplyAndClose
1423 // Purpose  : Set value of the flag indicating that the dialog is
1424 //            accepted by Apply & Close button
1425 //================================================================
1426 void GEOMBase_Helper::setIsApplyAndClose( const bool theFlag )
1427 {
1428   myIsApplyAndClose = theFlag;
1429 }
1430
1431 //================================================================
1432 // Function : isApplyAndClose
1433 // Purpose  : Get value of the flag indicating that the dialog is
1434 //            accepted by Apply & Close button
1435 //================================================================
1436 bool GEOMBase_Helper::isApplyAndClose() const
1437 {
1438   return myIsApplyAndClose;
1439 }
1440
1441 //================================================================
1442 // Function : setIsOptimizedBrowsing
1443 // Purpose  : Set value of the flag switching to optimized
1444 //            browsing mode (to select the first published
1445 //            object only)
1446 //================================================================
1447 void GEOMBase_Helper::setIsOptimizedBrowsing( const bool theFlag )
1448 {
1449   myIsOptimizedBrowsing = theFlag;
1450 }
1451
1452 //================================================================
1453 // Function : isOptimizedBrowsing
1454 // Purpose  : Get value of the flag switching to optimized
1455 //            browsing mode (to select the first published
1456 //            object only)
1457 //================================================================
1458 bool GEOMBase_Helper::isOptimizedBrowsing() const
1459 {
1460   return myIsOptimizedBrowsing;
1461 }