1 // Copyright (C) 2007-2012 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 // GEOM GEOMGUI : GUI for Geometry component
24 // File : GEOMBase_Helper.cxx
25 // Author : Sergey ANIKIN, Open CASCADE S.A.S. (sergey.anikin@opencascade.com)
27 #include "GEOMBase_Helper.h"
29 #include "GEOM_Operation.h"
31 #include <GeometryGUI.h>
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>
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_ListIteratorOfListIO.hxx>
50 #include <SALOME_Prs.h>
52 #include <OCCViewer_ViewModel.h>
53 #include <SVTK_ViewModel.h>
55 #include <TColStd_MapOfInteger.hxx>
56 #include <TCollection_AsciiString.hxx>
57 #include <TColStd_IndexedMapOfInteger.hxx>
59 //To disable automatic genericobj management, the following line should be commented.
60 //Otherwise, it should be uncommented. Refer to KERNEL_SRC/src/SALOMEDSImpl/SALOMEDSImpl_AttributeIOR.cxx
61 #define WITHGENERICOBJ
63 //================================================================
64 // Function : getActiveView
65 // Purpose : Get active view window, returns 0 if no open study frame
66 //================================================================
67 static SUIT_ViewWindow* getActiveView()
69 SUIT_Study* activeStudy = SUIT_Session::session()->activeApplication()->activeStudy();
71 return SUIT_Session::session()->activeApplication()->desktop()->activeWindow();
77 //================================================================
78 // Function : getGeomEngine
79 // Purpose : Static method
80 //================================================================
81 GEOM::GEOM_Gen_ptr GEOMBase_Helper::getGeomEngine()
83 return GeometryGUI::GetGeomGen();
86 //================================================================
87 // Function : GEOMBase_Helper
89 //================================================================
90 GEOMBase_Helper::GEOMBase_Helper( SUIT_Desktop* desktop )
91 : myDesktop( desktop ), myViewWindow( 0 ), myDisplayer( 0 ), myCommand( 0 ), isPreview( false ),
92 myIsApplyAndClose( false ), myIsOptimizedBrowsing( false ), myIsWaitCursorEnabled( true ),
93 myIsDisableBrowsing(false)
97 //================================================================
98 // Function : ~GEOMBase_Helper
100 //================================================================
101 GEOMBase_Helper::~GEOMBase_Helper()
103 //rnv: Fix for the "IPAL21922 : WinTC5.1.4: incorrect quit salome"
104 if ( !SUIT_Session::session()->activeApplication() || !SUIT_Session::session()->activeApplication()->desktop() )
107 if ( myPreview.size() )
111 SalomeApp_Application* app = (SalomeApp_Application*)(SUIT_Session::session()->activeApplication());
113 GeometryGUI* aGeomGUI = dynamic_cast<GeometryGUI*>( app->module( "Geometry" ) );
115 globalSelection(aGeomGUI->getLocalSelectionMode() , true );
120 if ( !CORBA::is_nil( myOperation ) )
121 myOperation->UnRegister();
124 //================================================================
125 // Function : display
127 //================================================================
128 void GEOMBase_Helper::display( const ObjectList& objList, const bool updateView )
130 ObjectList::const_iterator it;
131 for ( it = objList.begin(); it != objList.end(); it++ ) {
132 display( *it, false );
134 if ( !objList.empty() && updateView )
135 getDisplayer()->UpdateViewer();
138 //================================================================
139 // Function : display
140 // Purpose : Display object.
141 // Important : Object must be already in study
142 //================================================================
143 void GEOMBase_Helper::display( GEOM::GEOM_Object_ptr object, const bool updateView )
145 // Unset color of shape ( this color may be set during preview displaying )
146 // Default color will be used
147 // getDisplayer()->UnsetColor();
148 getDisplayer()->UnsetWidth();
150 MESSAGE("GEOMBase_Helper::display myTexture = "<<getDisplayer()->GetTexture())
152 // Enable activisation of selection
153 getDisplayer()->SetToActivate( true );
156 getDisplayer()->Display( object, updateView );
159 //================================================================
162 //================================================================
163 void GEOMBase_Helper::erase( const ObjectList& objList, const bool updateView )
165 ObjectList::const_iterator it = objList.begin();
166 for ( ; it != objList.end(); it++ ) {
169 if ( !objList.empty() && updateView )
170 getDisplayer()->UpdateViewer();
173 //================================================================
176 //================================================================
177 void GEOMBase_Helper::erase( GEOM::GEOM_Object_ptr object, const bool updateView )
179 if ( !object->_is_nil() ) {
180 QString entry = getEntry( object );
181 getDisplayer()->Erase( new SALOME_InteractiveObject(
182 entry.toLatin1().constData(),
183 "GEOM", strdup( GEOMBase::GetName( object ).toLatin1().constData() ) ), true, updateView );
187 //================================================================
188 // Function : redisplay
190 //================================================================
191 void GEOMBase_Helper::redisplay( const ObjectList& objList,
192 const bool withChildren,
193 const bool updateView )
195 ObjectList::const_iterator it = objList.begin();
196 for ( ; it != objList.end(); it++ ) {
197 redisplay( *it, withChildren, false );
199 if ( !objList.empty() && updateView )
200 getDisplayer()->UpdateViewer();
203 //================================================================
204 // Function : redisplay
206 //================================================================
207 void GEOMBase_Helper::redisplay( GEOM::GEOM_Object_ptr object,
208 const bool withChildren,
209 const bool updateView )
211 if ( !object->_is_nil() ) {
212 // Unset color of shape ( this color may be set during preview displaying )
213 // Default color will be used
214 getDisplayer()->UnsetColor();
215 getDisplayer()->UnsetWidth();
217 // Enable activisation of selection
218 getDisplayer()->SetToActivate( true );
220 QString entry = getEntry( object );
221 getDisplayer()->Redisplay(new SALOME_InteractiveObject
222 (entry.toLatin1().constData(), "GEOM", strdup(GEOMBase::GetName(object).toLatin1().constData())), false);
225 if ( withChildren ) {
226 SalomeApp_Study* aDoc = getStudy();
227 if ( aDoc && aDoc->studyDS() ) {
228 _PTR(Study) aStudy = aDoc->studyDS();
229 CORBA::String_var objStr = SalomeApp_Application::orb()->object_to_string(object);
230 _PTR(SObject) aSObj (aStudy->FindObjectIOR(std::string(objStr.in())));
232 _PTR(ChildIterator) anIt ( aStudy->NewChildIterator( aSObj ) );
233 for ( anIt->InitEx( true ); anIt->More(); anIt->Next() ) {
234 GEOM::GEOM_Object_var aChild = GEOM::GEOM_Object::_narrow
235 (GeometryGUI::ClientSObjectToObject(anIt->Value()));
236 if ( !CORBA::is_nil( aChild ) ) {
237 if ( !aChild->_is_nil() ) {
238 QString entry = getEntry( aChild );
239 getDisplayer()->Redisplay( new SALOME_InteractiveObject(
240 entry.toLatin1().constData(), "GEOM", strdup( GEOMBase::GetName( aChild ).toLatin1().constData() ) ), false );
249 getDisplayer()->UpdateViewer();
252 //================================================================
253 // Function : displayPreview
254 // Purpose : Method for displaying preview based on execute() results
255 //================================================================
256 void GEOMBase_Helper::displayPreview( const bool display,
259 const bool toRemoveFromEngine,
260 const double lineWidth,
261 const int displayMode,
265 erasePreview( update );
271 if ( !isValid( msg ) )
273 erasePreview( update );
278 erasePreview( false );
281 SUIT_OverrideCursor wc;
284 if ( !isWaitCursorEnabled() )
287 if ( !execute( objects ) || !getOperation()->IsDone() ) {
291 for ( ObjectList::iterator it = objects.begin(); it != objects.end(); ++it )
293 GEOM::GEOM_Object_var obj=*it;
294 displayPreview( obj, true, activate, false, lineWidth, displayMode, color );
295 if ( toRemoveFromEngine )
300 catch( const SALOME::SALOME_Exception& e ) {
301 SalomeApp_Tools::QtCatchCorbaException( e );
310 //================================================================
311 // Function : displayPreview
312 // Purpose : Method for displaying preview of resulting shape
313 //================================================================
314 void GEOMBase_Helper::displayPreview( GEOM::GEOM_Object_ptr object,
318 const double lineWidth,
319 const int displayMode,
322 // Set color for preview shape
323 getDisplayer()->SetColor( color == -1 ? Quantity_NOC_VIOLET : color );
325 // set width of displayed shape
328 SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
329 lw = resMgr->integerValue("Geometry", "preview_edge_width", -1);
331 getDisplayer()->SetWidth( lw );
333 // set display mode of displayed shape
334 int aPrevDispMode = getDisplayer()->SetDisplayMode( displayMode );
336 // Disable activation of selection
337 getDisplayer()->SetToActivate( activate );
339 // Make a reference to GEOM_Object
340 CORBA::String_var objStr = SalomeApp_Application::orb()->object_to_string( object );
341 getDisplayer()->SetName( objStr.in() );
344 SALOME_Prs* aPrs = getDisplayer()->BuildPrs( object );
345 if ( aPrs == 0 || aPrs->IsNull() )
349 displayPreview( aPrs, append, update );
351 getDisplayer()->UnsetName();
352 getDisplayer()->UnsetColor();
353 getDisplayer()->SetDisplayMode( aPrevDispMode );
355 // Enable activation of displayed objects
356 getDisplayer()->SetToActivate( true );
359 //================================================================
360 // Function : displayPreview
361 // Purpose : Method for displaying arbitrary preview objects (not limited to shapes)
362 //================================================================
363 void GEOMBase_Helper::displayPreview( const SALOME_Prs* prs,
368 erasePreview( false );
370 // remember current view frame to make correct erase preview later
371 myViewWindow = getActiveView();
373 if ( myViewWindow == 0 )
377 SUIT_ViewManager* aViewManager = myViewWindow->getViewManager();
378 if ( aViewManager->getType() == OCCViewer_Viewer::Type() ||
379 aViewManager->getType() == SVTK_Viewer::Type() )
381 SUIT_ViewModel* aViewModel = aViewManager->getViewModel();
382 SALOME_View* aView = dynamic_cast<SALOME_View*>(aViewModel);
384 aView->Display( prs );
387 // Add prs to the preview list
388 myPreview.push_back( (SALOME_Prs*)prs );
392 getDisplayer()->UpdateViewer();
395 //================================================================
396 // Function : erasePreview
398 //================================================================
399 void GEOMBase_Helper::erasePreview( const bool update )
401 // check view frame where the preview was displayed
402 bool vfOK = checkViewWindow() && myViewWindow;
403 // Iterate through presentations and delete them
404 for ( PrsList::iterator anIter = myPreview.begin(); anIter != myPreview.end(); ++anIter ) {
407 SUIT_ViewManager* aViewManager = myViewWindow->getViewManager();
408 if ( aViewManager->getType() == OCCViewer_Viewer::Type() ||
409 aViewManager->getType() == SVTK_Viewer::Type() )
411 SUIT_ViewModel* aViewModel = aViewManager->getViewModel();
412 SALOME_View* aView = dynamic_cast<SALOME_View*>(aViewModel);
414 aView->Erase( *anIter, true );
426 //================================================================
427 // Function : localSelection
428 // Purpose : Activate selection of objects of a given type
429 // IMPORTANT : Works after localSelection( ... ) method call only
430 //================================================================
431 void GEOMBase_Helper::activate( const int theType )
433 if (!getStudy()) return;
434 _PTR(Study) aStudy = getStudy()->studyDS();
435 _PTR(SComponent) aGeom ( aStudy->FindComponent( "GEOM" ) );
440 _PTR(ChildIterator) anIter ( aStudy->NewChildIterator( aGeom ) );
441 for ( ; anIter->More(); anIter->Next() )
443 _PTR(SObject) aSO ( anIter->Value() );
446 _PTR(SObject) aRefSO;
447 if ( !aSO->ReferencedObject( aRefSO ) )
449 GEOM::GEOM_Object_var anObj = GEOM::GEOM_Object::_narrow
450 (GeometryGUI::ClientSObjectToObject(aSO));
451 if ( !anObj->_is_nil() && anObj->GetType() == theType )
452 aList.Append( new SALOME_InteractiveObject( aSO->GetID().c_str(), "GEOM", aSO->GetName().c_str()) );
457 getDisplayer()->LocalSelection( aList, 0 );
460 //================================================================
461 // Function : localSelection
462 // Purpose : Activate selection of sub-shapes in accordance with mode
463 // theMode is from TopAbs_ShapeEnum
464 //================================================================
465 void GEOMBase_Helper::localSelection( const ObjectList& theObjs, const int theMode )
467 SALOME_ListIO aListOfIO;
469 ObjectList::const_iterator anIter = theObjs.begin();
470 for ( ; anIter != theObjs.end(); ++anIter )
472 GEOM::GEOM_Object_ptr anObj = *anIter;
473 if ( anObj->_is_nil() )
475 QString anEntry = getEntry( anObj );
477 aListOfIO.Append( new SALOME_InteractiveObject(
478 anEntry.toLatin1().constData(), "GEOM", strdup( GEOMBase::GetName( anObj ).toLatin1().constData() ) ) );
481 getDisplayer()->LocalSelection( aListOfIO, theMode );
484 //================================================================
485 // Function : localSelection
486 // Purpose : Activate selection of sub-shapes in accordance with mode
487 // theMode is from TopAbs_ShapeEnum
488 //================================================================
489 void GEOMBase_Helper::localSelection( GEOM::GEOM_Object_ptr obj, const int mode )
491 // If object is null local selection for all objects is activated
492 if ( obj->_is_nil() ) {
493 getDisplayer()->LocalSelection( Handle(SALOME_InteractiveObject)(), mode );
498 objList.push_back( obj );
499 localSelection( objList, mode );
503 //================================================================
504 // Function : globalSelection
505 // Purpose : Activate selection of sub-shapes. Set selection filters
506 // in accordance with mode. theMode is from GEOMImpl_Types
507 //================================================================
508 void GEOMBase_Helper::globalSelection( const int theMode, const bool update )
510 getDisplayer()->GlobalSelection( theMode, update );
513 //================================================================
514 // Function : globalSelection
515 // Purpose : Activate selection of sub-shapes. Set selection filters
516 // in accordance with mode. theMode is from GEOMImpl_Types
517 //================================================================
518 void GEOMBase_Helper::globalSelection( const TColStd_MapOfInteger& theModes,
521 getDisplayer()->GlobalSelection( theModes, update );
524 //================================================================
525 // Function : globalSelection
526 // Purpose : Activate selection of sub-shapes. Set selection filters
527 // in accordance with mode. theMode is from GEOMImpl_Types
528 //================================================================
529 void GEOMBase_Helper::globalSelection( const TColStd_MapOfInteger& theModes,
530 const QList<int>& subShapes,
533 getDisplayer()->GlobalSelection( theModes, update, &subShapes );
536 //================================================================
537 // Function : addInStudy
538 // Purpose : Add object in study
539 //================================================================
540 QString GEOMBase_Helper::addInStudy( GEOM::GEOM_Object_ptr theObj, const char* theName )
545 _PTR(Study) aStudy = getStudy()->studyDS();
546 if ( !aStudy || theObj->_is_nil() )
549 SALOMEDS::Study_var aStudyDS = GeometryGUI::ClientStudyToStudy(aStudy);
551 GEOM::GEOM_Object_ptr aFatherObj = getFather( theObj );
553 SALOMEDS::SObject_var aSO =
554 getGeomEngine()->AddInStudy(aStudyDS, theObj, theName, aFatherObj);
557 if ( !aSO->_is_nil() )
558 anEntry = aSO->GetID();
560 // Each dialog is responsible for this method implementation,
561 // default implementation does nothing
562 restoreSubShapes(aStudyDS, aSO);
568 //================================================================
569 // Function : restoreSubShapes
570 // Purpose : restore tree of argument's sub-shapes under the resulting shape
571 //================================================================
572 void GEOMBase_Helper::restoreSubShapes (SALOMEDS::Study_ptr /*theStudy*/,
573 SALOMEDS::SObject_ptr /*theSObject*/)
575 // do nothing by default
577 // example of implementation in particular dialog:
578 // GEOM::ListOfGO anArgs;
579 // anArgs.length(0); // empty list means that all arguments should be restored
580 // getGeomEngine()->RestoreSubShapesSO(theStudy, theSObject, anArgs,
581 // /*theFindMethod=*/GEOM::FSM_GetInPlace,
582 // /*theInheritFirstArg=*/false);
585 //================================================================
586 // Function : updateObjBrowser
587 // Purpose : Update object browser
588 //================================================================
589 void GEOMBase_Helper::updateObjBrowser() const
591 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>(SUIT_Session::session()->activeApplication());
593 CAM_Module* module = app->module( "Geometry" );
594 SalomeApp_Module* appMod = dynamic_cast<SalomeApp_Module*>( module );
596 appMod->updateObjBrowser( true );
601 //================================================================
602 // Function : updateViewer
603 // Purpose : Update active 3D view
604 //================================================================
605 void GEOMBase_Helper::updateViewer()
607 getDisplayer()->UpdateViewer();
610 //================================================================
611 // Function : getStudyId
612 // Purpose : Get study Id
613 //================================================================
614 int GEOMBase_Helper::getStudyId() const
618 anId = getStudy()->id();
622 //================================================================
623 // Function : getStudy
624 // Purpose : Returns the active study. It is recommended to use
625 // this method instead of direct desktop->getActiveStudy() calls
626 //================================================================
627 SalomeApp_Study* GEOMBase_Helper::getStudy() const
629 SUIT_Desktop* aDesktop = getDesktop();
633 QList<SUIT_Application*> anAppList = SUIT_Session::session()->applications();
635 SUIT_Application* anApp = 0;
636 QListIterator<SUIT_Application*> it( anAppList );
637 while ( it.hasNext() )
640 if ( anApp && anApp->desktop() == aDesktop )
644 return dynamic_cast<SalomeApp_Study*>(anApp->activeStudy());
647 //================================================================
648 // Function : getEntry
650 //================================================================
651 QString GEOMBase_Helper::getEntry( GEOM::GEOM_Object_ptr object ) const
653 SalomeApp_Study* study = getStudy();
655 QString objIOR = GEOMBase::GetIORFromObject( object );
656 if ( objIOR != "" ) {
657 _PTR(SObject) SO ( study->studyDS()->FindObjectIOR( objIOR.toLatin1().constData() ) );
659 return QString::fromStdString(SO->GetID());
665 //================================================================
666 // Function : getDisplayer
668 //================================================================
669 GEOM_Displayer* GEOMBase_Helper::getDisplayer()
672 myDisplayer = new GEOM_Displayer( getStudy() );
676 //================================================================
677 // Function : clearShapeBuffer
679 //================================================================
680 void GEOMBase_Helper::clearShapeBuffer( GEOM::GEOM_Object_ptr theObj )
682 if ( CORBA::is_nil( theObj ) )
685 CORBA::String_var IOR = SalomeApp_Application::orb()->object_to_string( theObj );
686 TCollection_AsciiString asciiIOR( (char *)IOR.in() );
687 GEOM_Client::get_client().RemoveShapeFromBuffer( asciiIOR );
689 if ( !getStudy() || !getStudy()->studyDS() )
692 _PTR(Study) aStudy = getStudy()->studyDS();
693 _PTR(SObject) aSObj ( aStudy->FindObjectIOR( std::string( IOR ) ) );
697 _PTR(ChildIterator) anIt ( aStudy->NewChildIterator( aSObj ) );
698 for ( anIt->InitEx( true ); anIt->More(); anIt->Next() ) {
699 _PTR(GenericAttribute) anAttr;
700 if ( anIt->Value()->FindAttribute(anAttr, "AttributeIOR") ) {
701 _PTR(AttributeIOR) anIOR ( anAttr );
702 TCollection_AsciiString asciiIOR( (char*)anIOR->Value().c_str() );
703 GEOM_Client::get_client().RemoveShapeFromBuffer( asciiIOR );
708 //================================================================
709 // Function : openCommand
711 //================================================================
712 bool GEOMBase_Helper::openCommand()
715 if ( !getStudy() || hasCommand() )
717 MESSAGE("Getting out from openCommand()")
721 GEOM::GEOM_IOperations_var anOp = GEOM::GEOM_IOperations::_narrow( getOperation() );
722 if ( !anOp->_is_nil() ) {
723 myCommand = new GEOM_Operation( SUIT_Session::session()->activeApplication(), anOp.in() );
729 MESSAGE("anOp->_is_nil() = true")
735 //================================================================
736 // Function : abortCommand
738 //================================================================
739 bool GEOMBase_Helper::abortCommand()
745 delete myCommand; // I don't know where to delete this object here ?
751 //================================================================
752 // Function : commitCommand
754 //================================================================
755 bool GEOMBase_Helper::commitCommand( const char* )
761 delete myCommand; // I don't know where to delete this object here ?
767 //================================================================
768 // Function : hasCommand
770 //================================================================
771 bool GEOMBase_Helper::hasCommand() const
773 bool res = (bool) myCommand;
774 return (bool)myCommand;
777 //================================================================
778 // Function : getOperation
780 //================================================================
781 GEOM::GEOM_IOperations_ptr GEOMBase_Helper::getOperation()
783 if ( myOperation->_is_nil() )
784 myOperation = createOperation();
791 //================================================================
792 // Function : checkViewWindow
794 //================================================================
795 bool GEOMBase_Helper::checkViewWindow()
798 QList<SUIT_ViewWindow*> aViewWindowsList = SUIT_Session::session()->activeApplication()->desktop()->windows();
799 QListIterator<SUIT_ViewWindow*> it( aViewWindowsList );
800 while ( it.hasNext() )
802 if ( myViewWindow == it.next() )
810 //================================================================
811 // Function : onAccept
812 // Purpose : This method should be called from dialog's slots onOk() and onApply()
813 // It perfroms user input validation, then it
814 // performs a proper operation and manages transactions, etc.
815 //================================================================
816 bool GEOMBase_Helper::onAccept( const bool publish, const bool useTransaction, bool erasePreviewFlag )
818 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
821 MESSAGE("appStudy is empty")
824 _PTR(Study) aStudy = appStudy->studyDS();
826 bool aLocked = (_PTR(AttributeStudyProperties) (aStudy->GetProperties()))->IsLocked();
828 MESSAGE("GEOMBase_Helper::onAccept - ActiveStudy is locked");
829 SUIT_MessageBox::warning ( (QWidget*)SUIT_Session::session()->activeApplication()->desktop(),
830 QObject::tr("WRN_WARNING"),
831 QObject::tr("WRN_STUDY_LOCKED"),
832 QObject::tr("BUT_OK") );
837 if ( !isValid( msg ) ) {
843 erasePreview( false );
848 if ( ( !publish && !useTransaction ) || openCommand() ) {
849 SUIT_OverrideCursor wc;
850 SUIT_Session::session()->activeApplication()->putInfo( "" );
852 if ( !execute( objects ) || !getOperation()->IsDone() ) {
858 addSubshapesToStudy(); // add Sub-shapes if local selection
859 const int nbObjs = objects.size();
860 QStringList anEntryList;
862 for ( ObjectList::iterator it = objects.begin(); it != objects.end(); ++it ) {
863 GEOM::GEOM_Object_var obj=*it;
865 QString aName = getNewObjectName();
868 aName = getPrefix(obj);
870 // Try to find a unique name
871 aName = GEOMBase::GetDefaultName(aName, extractPrefix());
873 // Don't check name uniqueness in case of numerous objects
874 aName = aName + "_" + QString::number(aNumber++);
877 // PAL6521: use a prefix, if some dialog box doesn't reimplement getNewObjectName()
878 if ( aName.isEmpty() )
879 aName = GEOMBase::GetDefaultName( getPrefix( obj ) );
881 anEntryList << addInStudy( obj, aName.toLatin1().constData() );
883 display( obj, false );
884 #ifdef WITHGENERICOBJ
885 // obj has been published in study. Its refcount has been incremented.
886 // It is safe to decrement its refcount
887 // so that it will be destroyed when the entry in study will be removed
892 // asv : fix of PAL6454. If publish==false, then the original shape
893 // was modified, and need to be re-cached in GEOM_Client before redisplay
894 clearShapeBuffer( obj );
895 // withChildren=true, updateView=false
896 redisplay( obj, true, false );
903 if( SUIT_Application* anApp = SUIT_Session::session()->activeApplication() ) {
904 LightApp_Application* aLightApp = dynamic_cast<LightApp_Application*>( anApp );
905 if(aLightApp && !isDisableBrowsing() )
906 aLightApp->browseObjects( anEntryList, isApplyAndClose(), isOptimizedBrowsing() );
907 anApp->putInfo( QObject::tr("GEOM_PRP_DONE") );
916 catch( const SALOME::SALOME_Exception& e ) {
917 SalomeApp_Tools::QtCatchCorbaException( e );
919 MESSAGE("Exception catched")
924 MESSAGE("result ="<<result)
929 //================================================================
930 // Function : showError
931 // Purpose : Shows a message box with infromation about an error taken from getOperation()->GetErrorCode()
932 //================================================================
933 void GEOMBase_Helper::showError()
936 if ( !getOperation()->_is_nil() )
937 msg = QObject::tr( getOperation()->GetErrorCode() );
940 msg = QObject::tr( "GEOM_PRP_ABORT" );
942 SUIT_MessageBox::critical( SUIT_Session::session()->activeApplication()->desktop(),
943 QObject::tr( "GEOM_ERROR_STATUS" ),
945 QObject::tr( "BUT_OK" ) );
948 //================================================================
949 // Function : showError
950 // Purpose : Shows a error message followed by <msg>
951 //================================================================
952 void GEOMBase_Helper::showError( const QString& msg )
954 QString str( QObject::tr( "GEOM_INCORRECT_INPUT" ) );
955 if ( !msg.isEmpty() )
957 SUIT_MessageBox::critical(SUIT_Session::session()->activeApplication()->desktop(), QObject::tr( "GEOM_ERROR" ), str, QObject::tr( "BUT_OK" ) );
960 //////////////////////////////////////////////////////////////////
961 // Virtual methods to be redefined in dialogs
962 //////////////////////////////////////////////////////////////////
964 //================================================================
965 // Function : createOperation
966 // Purpose : Redefine this method to return proper IOperation reference
967 //================================================================
968 GEOM::GEOM_IOperations_ptr GEOMBase_Helper::createOperation()
970 GEOM::GEOM_IOperations_var aNilOp;
971 return aNilOp._retn();
974 //================================================================
975 // Function : isValid
976 // Purpose : Called by onAccept(). Redefine this method to check validity of user input in dialog boxes.
977 //================================================================
978 bool GEOMBase_Helper::isValid( QString& )
983 //================================================================
984 // Function : execute
985 // Purpose : This method is called by onAccept().
986 // It should perform the required operation and put all new or modified objects into
987 // <objects> argument.Should return <false> if some error occurs during its execution.
988 //================================================================
989 bool GEOMBase_Helper::execute( ObjectList& objects )
994 //================================================================
995 // Function : getFather
996 // Purpose : This method is called by addInStudy(). It should return a father object
997 // for <theObj> or a nil reference if <theObj> should be published
998 // as a top-level object.
999 //================================================================
1000 GEOM::GEOM_Object_ptr GEOMBase_Helper::getFather( GEOM::GEOM_Object_ptr theObj )
1002 return GEOM::GEOM_Object::_nil();
1005 //================================================================
1006 // Function : getNewObjectName
1007 // Purpose : Redefine this method to return proper name for a new object
1008 //================================================================
1009 QString GEOMBase_Helper::getNewObjectName() const
1011 return QString::null;
1014 //================================================================
1015 // Function : extractPrefix
1016 // Purpose : Redefine this method to return \c true if necessary
1017 // to extract prefix when generating new name for the
1018 // object(s) being created
1019 //================================================================
1020 bool GEOMBase_Helper::extractPrefix() const
1025 //================================================================
1026 // Function : getPrefix
1027 // Purpose : Get prefix for name of created object
1028 //================================================================
1029 QString GEOMBase_Helper::getPrefix( GEOM::GEOM_Object_ptr theObj ) const
1031 if ( !myPrefix.isEmpty() || theObj->_is_nil() )
1034 GEOM::shape_type aType = theObj->GetShapeType();
1038 case GEOM::VERTEX : return QObject::tr( "GEOM_VERTEX" );
1039 case GEOM::EDGE : return QObject::tr( "GEOM_EDGE" );
1040 case GEOM::WIRE : return QObject::tr( "GEOM_WIRE" );
1041 case GEOM::FACE : return QObject::tr( "GEOM_FACE" );
1042 case GEOM::SHELL : return QObject::tr( "GEOM_SHELL" );
1043 case GEOM::SOLID : return QObject::tr( "GEOM_SOLID" );
1044 case GEOM::COMPSOLID: return QObject::tr( "GEOM_COMPOUNDSOLID" );
1045 case GEOM::COMPOUND : return QObject::tr( "GEOM_COMPOUND" );
1046 default : return "";
1050 //================================================================
1051 // Function : getDesktop
1052 // Purpose : Returns myDesktop field. Initialized in constructor,
1053 // usually as dynamic_cast<SUIT_Desktop*>(parentWidget())
1054 //================================================================
1055 SUIT_Desktop* GEOMBase_Helper::getDesktop() const
1060 //================================================================
1061 // Function : selectObjects
1062 // Purpose : Selects list of objects
1063 //================================================================
1064 bool GEOMBase_Helper::selectObjects( ObjectList& objects )
1066 SUIT_DataOwnerPtrList aList;
1067 ObjectList::iterator anIter;
1068 for ( anIter = objects.begin(); anIter != objects.end(); ++anIter )
1070 QString anEntry = getEntry( *anIter );
1071 LightApp_DataOwner* anOwher = new LightApp_DataOwner( anEntry );
1072 aList.append( anOwher );
1075 SUIT_Session* session = SUIT_Session::session();
1076 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
1080 LightApp_SelectionMgr* aMgr = app->selectionMgr();
1084 aMgr->setSelected( aList, false );
1089 //================================================================
1090 // Function : findObjectInFather
1091 // Purpose : It should return an object if its founded in study or
1092 // return Null object if the object is not founded
1093 //================================================================
1094 GEOM::GEOM_Object_ptr GEOMBase_Helper::findObjectInFather (GEOM::GEOM_Object_ptr theFather,
1095 const QString& theName)
1097 SalomeApp_Application* app =
1098 dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
1099 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
1100 _PTR(Study) aDStudy = appStudy->studyDS();
1101 QString IOR = GEOMBase::GetIORFromObject( theFather );
1102 _PTR(SObject) SObj ( aDStudy->FindObjectIOR( IOR.toLatin1().constData() ) );
1104 bool inStudy = false;
1105 GEOM::GEOM_Object_var aReturnObject;
1106 for (_PTR(ChildIterator) iit (aDStudy->NewChildIterator( SObj )); iit->More() && !inStudy; iit->Next()) {
1107 _PTR(SObject) child (iit->Value());
1108 QString aChildName = child->GetName().c_str();
1109 if (aChildName == theName) {
1111 CORBA::Object_var corbaObj = GeometryGUI::ClientSObjectToObject(iit->Value());
1112 aReturnObject = GEOM::GEOM_Object::_narrow( corbaObj );
1116 return aReturnObject._retn();
1118 return GEOM::GEOM_Object::_nil();
1121 //================================================================
1122 // Function : findObjectInFather
1123 // Purpose : It should return an object if its founded in study or
1124 // return Null object if the object is not founded
1125 //================================================================
1126 GEOM::GEOM_Object_ptr GEOMBase_Helper::findObjectInFather( GEOM::GEOM_Object_ptr theFather,
1129 GEOM::GEOM_Object_var object;
1132 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
1134 _PTR(Study) studyDS = study->studyDS();
1135 QString IOR = GEOMBase::GetIORFromObject( theFather );
1136 _PTR(SObject) sobject( studyDS->FindObjectIOR( IOR.toLatin1().constData() ) );
1138 _PTR(ChildIterator) it( studyDS->NewChildIterator( sobject ) );
1139 for ( ; it->More() && !found; it->Next() ) {
1140 GEOM::GEOM_Object_var cobject = GEOM::GEOM_Object::_narrow( GeometryGUI::ClientSObjectToObject( it->Value() ) );
1141 if ( !CORBA::is_nil( cobject ) ) {
1142 GEOM::ListOfLong_var indices = cobject->GetSubShapeIndices();
1143 int length = indices->length();
1144 for ( int i = 0; i < length && !found; i++ ) {
1145 if ( indices[i] == theIndex ) {
1155 return object._retn();
1158 //================================================================
1159 // Function : addSubshapesToStudy
1160 // Purpose : Virtual method to add sub-shapes if needs
1161 //================================================================
1162 void GEOMBase_Helper::addSubshapesToStudy()
1164 //Impemented in Dialogs, called from Accept method
1167 //================================================================
1168 // Function : getSelected
1169 // Purpose : Get selected object by specified type
1171 // Returns valid object if only one object of the specified type is selected
1172 // (no matter global or local selection is activated). If \a type is TopAbs_SHAPE,
1173 // geometrical object of any valid type is expected.
1175 // \param type type of the object to be obtained from selection
1176 // \return selected geometrical object or nil object if selection is not satisfactory
1177 //================================================================
1178 GEOM::GeomObjPtr GEOMBase_Helper::getSelected( TopAbs_ShapeEnum type )
1180 QList<TopAbs_ShapeEnum> types;
1182 return getSelected( types );
1185 //================================================================
1186 // Function : getSelected
1187 // Purpose : Get selected object by specified types
1189 // Returns valid object if only one object of the specified type is selected
1190 // (no matter global or local selection is activated). The list of allowed
1191 // shape types is passed via \a types. If \a types includes TopAbs_SHAPE,
1192 // geometrical object of any valid type is expected.
1194 // \param types list of allowed shape types for the objects to be obtained from selection
1195 // \return selected geometrical object or nil object if selection is not satisfactory
1196 //================================================================
1197 GEOM::GeomObjPtr GEOMBase_Helper::getSelected( const QList<TopAbs_ShapeEnum>& types )
1199 QList<GEOM::GeomObjPtr> selected = getSelected( types, 1 );
1200 return selected.count() > 0 ? selected[0] : GEOM::GeomObjPtr();
1203 //================================================================
1204 // Function : getSelected
1205 // Purpose : Get selected object(s) by specified type
1207 // Returns list of selected objects if selection satisfies specifies selection options.
1208 // (no matter global or local selection is activated). If \a type is TopAbs_SHAPE,
1209 // geometrical objects of any valid type are expected.
1211 // The \a type parameter specifies allowed type of the object(s) being selected.
1212 // The \a count parameter specifies exact number of the objects to be retrieved from selection.
1213 // The \a strict parameter specifies policy being applied to the selection.
1214 // If \a count < 0, then any number of the selected objects is valid (including 0).
1215 // In this case, if \a strict is \c true (default), all selected objects should satisfy
1216 // the specified \a type.
1217 // If \a count > 0, only specified number of the objects is retrieved from the selection.
1218 // In this case, if \a strict is \c true (default), function returns empty list if total number of selected
1219 // objects does not correspond to the \a count parameter. Otherwise (if \a strict is \c false),
1220 // function returns valid list of objects if at least \a count objects satisfy specified \a type.
1222 // \param type type of the object(s) to be obtained from selection
1223 // \param count number of items to be retrieved from selection
1224 // \param strict selection policy
1225 // \return list of selected geometrical objects or empty list if selection is not satisfactory
1226 //================================================================
1227 QList<GEOM::GeomObjPtr> GEOMBase_Helper::getSelected( TopAbs_ShapeEnum type, int count, bool strict )
1229 QList<TopAbs_ShapeEnum> types;
1231 return getSelected( types, count, strict );
1234 static bool typeInList( TopAbs_ShapeEnum type, const QList<TopAbs_ShapeEnum>& types )
1237 for ( int i = 0; i < types.count() && !ok; i++ )
1238 ok = types[i] == TopAbs_SHAPE || types[i] == type;
1242 //================================================================
1243 // Function : getSelected
1244 // Purpose : Get selected objects by specified types
1246 // Returns list of selected objects if selection satisfies specifies selection options.
1247 // (no matter global or local selection is activated). If \a types includes TopAbs_SHAPE,
1248 // geometrical objects of any valid type are expected.
1250 // The \a types parameter specifies allowed types of the object(s) being selected.
1251 // The \a count parameter specifies exact number of the objects to be retrieved from selection.
1252 // The \a strict parameter specifies policy being applied to the selection.
1253 // If \a count < 0, then any number of the selected objects is valid (including 0).
1254 // In this case, if \a strict is \c true (default), all selected objects should satisfy
1255 // the specified \a type.
1256 // If \a count > 0, only specified number of the objects is retrieved from the selection.
1257 // In this case, if \a strict is \c true (default), function returns empty list if total number of selected
1258 // objects does not correspond to the \a count parameter. Otherwise (if \a strict is \c false),
1259 // function returns valid list of objects if at least \a count objects satisfy specified \a type.
1261 // \param types list of allowed shape types for the objects to be obtained from selection
1262 // \param count number of items to be retrieved from selection
1263 // \param strict selection policy
1264 // \return list of selected geometrical objects or empty list if selection is not satisfactory
1265 //================================================================
1266 QList<GEOM::GeomObjPtr> GEOMBase_Helper::getSelected( const QList<TopAbs_ShapeEnum>& types, int count, bool strict )
1268 SUIT_Session* session = SUIT_Session::session();
1269 QList<GEOM::GeomObjPtr> result;
1271 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
1273 LightApp_SelectionMgr* selMgr = app->selectionMgr();
1275 SALOME_ListIO selected;
1276 selMgr->selectedObjects( selected );
1277 SALOME_ListIteratorOfListIO it( selected );
1278 bool stopped = false;
1279 for ( ; it.More() && !stopped; it.Next() ) {
1280 Handle(SALOME_InteractiveObject) IO = it.Value();
1281 GEOM::GeomObjPtr object = GEOMBase::ConvertIOinGEOMObject( IO );
1283 TColStd_IndexedMapOfInteger subShapes;
1284 selMgr->GetIndexes( IO, subShapes );
1285 int nbSubShapes = subShapes.Extent();
1286 if ( nbSubShapes == 0 ) {
1288 if ( typeInList( (TopAbs_ShapeEnum)(object->GetShapeType()), types ) ) {
1291 if ( strict && result.count() > count ) {
1295 else if ( !strict && result.count() == count )
1299 else if ( strict ) {
1306 for ( int i = 1; i <= nbSubShapes && !stopped; i++ ) {
1307 int idx = subShapes( i );
1308 GEOM::GeomObjPtr subShape = findObjectInFather( object.get(), idx );
1310 // sub-shape is not yet published in the study
1311 GEOM::ShapesOpPtr shapesOp = getGeomEngine()->GetIShapesOperations( getStudyId() );
1312 subShape.take( shapesOp->GetSubShape( object.get(), idx ) ); // take ownership!
1314 if ( typeInList( (TopAbs_ShapeEnum)(subShape->GetShapeType()), types ) ) {
1317 if ( strict && result.count() > count ) {
1321 else if ( !strict && result.count() == count )
1325 else if ( strict ) {
1338 //================================================================
1339 // Function : setIsApplyAndClose
1340 // Purpose : Set value of the flag indicating that the dialog is
1341 // accepted by Apply & Close button
1342 //================================================================
1343 void GEOMBase_Helper::setIsApplyAndClose( const bool theFlag )
1345 myIsApplyAndClose = theFlag;
1348 //================================================================
1349 // Function : isApplyAndClose
1350 // Purpose : Get value of the flag indicating that the dialog is
1351 // accepted by Apply & Close button
1352 //================================================================
1353 bool GEOMBase_Helper::isApplyAndClose() const
1355 return myIsApplyAndClose;
1358 //================================================================
1359 // Function : setIsOptimizedBrowsing
1360 // Purpose : Set value of the flag switching to optimized
1361 // browsing mode (to select the first published
1363 //================================================================
1364 void GEOMBase_Helper::setIsOptimizedBrowsing( const bool theFlag )
1366 myIsOptimizedBrowsing = theFlag;
1369 //================================================================
1370 // Function : isOptimizedBrowsing
1371 // Purpose : Get value of the flag switching to optimized
1372 // browsing mode (to select the first published
1374 //================================================================
1375 bool GEOMBase_Helper::isOptimizedBrowsing() const
1377 return myIsOptimizedBrowsing;