]> SALOME platform Git repositories - modules/geom.git/blob - src/GEOMBase/GEOMBase_Helper.cxx
Salome HOME
Additional fix for bug NPAL19028 (see remarks from Olivier Giorgis).
[modules/geom.git] / src / GEOMBase / GEOMBase_Helper.cxx
1 //  GEOM GEOMGUI : GUI for Geometry component
2 //
3 //  Copyright (C) 2004  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 //  This library is free software; you can redistribute it and/or
7 //  modify it under the terms of the GNU Lesser General Public
8 //  License as published by the Free Software Foundation; either
9 //  version 2.1 of the License.
10 //
11 //  This library is distributed in the hope that it will be useful,
12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 //  Lesser General Public License for more details.
15 //
16 //  You should have received a copy of the GNU Lesser General Public
17 //  License along with this library; if not, write to the Free Software
18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 //
23 //
24 //  File   : GEOMBase_Helper.cxx
25 //  Author : Sergey ANIKIN
26 //  Module : GEOM
27 //  $Header$
28
29 #include <SUIT_ViewModel.h>
30
31 #include "GEOMBase_Helper.h"
32 #include "GEOMBase.h"
33 #include "GEOM_Operation.h"
34 #include "GeometryGUI.h"
35 #include "GEOM_Displayer.h"
36 #include "GEOMImpl_Types.hxx"
37
38 #include <SUIT_Session.h>
39 #include <SUIT_ViewWindow.h>
40 #include <SUIT_MessageBox.h>
41 #include <SUIT_OverrideCursor.h>
42
43 #include <SalomeApp_Module.h>
44 #include <SalomeApp_Application.h>
45 #include <SalomeApp_Study.h>
46 #include <LightApp_SelectionMgr.h>
47 #include <LightApp_DataOwner.h>
48 #include <SalomeApp_Tools.h>
49 #include <SalomeApp_DataModel.h>
50
51 #include <OCCViewer_ViewModel.h>
52 #include <SVTK_ViewModel.h>
53
54 #include <OB_Browser.h>
55
56 #include <TColStd_MapOfInteger.hxx>
57 #include <TCollection_AsciiString.hxx>
58
59 using namespace std;
60
61 #include <SALOMEDSClient.hxx>
62
63
64 //================================================================
65 // Function : getActiveView
66 // Purpose  : Get active view window, returns 0 if no open study frame
67 //================================================================
68 static SUIT_ViewWindow* 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 )
92   : myDesktop( desktop ), myViewWindow( 0 ), myDisplayer( 0 ), myCommand( 0 ), isPreview( false )
93 {
94 }
95
96 //================================================================
97 // Function : ~GEOMBase_Helper
98 // Purpose  :
99 //================================================================
100 GEOMBase_Helper::~GEOMBase_Helper()
101 {
102   if ( !SUIT_Session::session()->activeApplication()->desktop() )
103     return;
104
105   if ( myPreview.size() )
106     erasePreview();
107   if ( hasCommand() )
108     abortCommand();
109   SalomeApp_Application* app = (SalomeApp_Application*)(SUIT_Session::session()->activeApplication());
110   if (app) {
111     GeometryGUI* aGeomGUI = dynamic_cast<GeometryGUI*>( app->module( "Geometry" ) );
112     if(aGeomGUI)
113       globalSelection(aGeomGUI->getLocalSelectionMode() , true );
114   }
115   
116   if (myDisplayer)
117     delete myDisplayer;
118 }
119
120 //================================================================
121 // Function : display
122 // Purpose  :
123 //================================================================
124 void GEOMBase_Helper::display( const ObjectList& objList, const bool updateView )
125 {
126   ObjectList::const_iterator it;
127   for ( it = objList.begin(); it != objList.end(); it++ ) {
128     display( *it, false );
129   }
130   if ( !objList.empty() && updateView )
131     getDisplayer()->UpdateViewer();
132 }
133
134 //================================================================
135 // Function  : display
136 // Purpose   : Display object.
137 // Important : Object must be already in study
138 //================================================================
139 void GEOMBase_Helper::display( GEOM::GEOM_Object_ptr object, const bool updateView )
140 {
141   // Unset color of shape ( this color may be set during preview displaying )
142   // Default color will be used
143   getDisplayer()->UnsetColor();
144   getDisplayer()->UnsetWidth();
145
146   // Enable activisation of selection
147   getDisplayer()->SetToActivate( true );
148
149   // Display object
150   getDisplayer()->Display( object, updateView );
151 }
152
153 //================================================================
154 // Function : erase
155 // Purpose  :
156 //================================================================
157 void GEOMBase_Helper::erase( const ObjectList& objList, const bool updateView )
158 {
159   ObjectList::const_iterator it = objList.begin();
160   for ( ; it != objList.end(); it++ ) {
161     erase( *it, false );
162   }
163   if ( !objList.empty() && updateView )
164     getDisplayer()->UpdateViewer();
165 }
166
167 //================================================================
168 // Function : erase
169 // Purpose  :
170 //================================================================
171 void GEOMBase_Helper::erase( GEOM::GEOM_Object_ptr object, const bool updateView )
172 {
173   if ( !object->_is_nil() ) {
174     string entry = getEntry( object );
175     getDisplayer()->Erase( new SALOME_InteractiveObject(
176       entry.c_str(), "GEOM", strdup( GEOMBase::GetName( object ) ) ), true, updateView );
177   }
178 }
179
180 //================================================================
181 // Function : redisplay
182 // Purpose  :
183 //================================================================
184 void GEOMBase_Helper::redisplay( const ObjectList& objList,
185                                  const bool withChildren,
186                                  const bool updateView )
187 {
188   ObjectList::const_iterator it = objList.begin();
189   for ( ; it != objList.end(); it++ ) {
190     redisplay( *it, withChildren, false );
191   }
192   if ( !objList.empty() && updateView )
193     getDisplayer()->UpdateViewer();
194 }
195
196 //================================================================
197 // Function : redisplay
198 // Purpose  :
199 //================================================================
200 void GEOMBase_Helper::redisplay( GEOM::GEOM_Object_ptr object,
201                                  const bool withChildren,
202                                  const bool updateView )
203 {
204   if ( !object->_is_nil() ) {
205     // Unset color of shape ( this color may be set during preview displaying )
206     // Default color will be used
207     getDisplayer()->UnsetColor();
208     getDisplayer()->UnsetWidth();
209
210     // Enable activisation of selection
211     getDisplayer()->SetToActivate( true );
212
213     string entry = getEntry( object );
214     getDisplayer()->Redisplay(new SALOME_InteractiveObject
215                               (entry.c_str(), "GEOM", strdup(GEOMBase::GetName(object))), false);
216   }
217
218   if ( withChildren ) {
219     SalomeApp_Study* aDoc = getStudy();
220     if ( aDoc && aDoc->studyDS() ) {
221       _PTR(Study) aStudy = aDoc->studyDS();
222       CORBA::String_var objStr = SalomeApp_Application::orb()->object_to_string(object);
223       _PTR(SObject) aSObj (aStudy->FindObjectIOR(string(objStr.in())));
224       if ( aSObj  ) {
225         _PTR(ChildIterator) anIt ( aStudy->NewChildIterator( aSObj ) );
226         for ( anIt->InitEx( true ); anIt->More(); anIt->Next() ) {
227           GEOM::GEOM_Object_var aChild = GEOM::GEOM_Object::_narrow
228             (GeometryGUI::ClientSObjectToObject(anIt->Value()));
229           if ( !CORBA::is_nil( aChild ) ) {
230             if ( !aChild->_is_nil() ) {
231               string entry = getEntry( aChild );
232               getDisplayer()->Redisplay( new SALOME_InteractiveObject(
233                 entry.c_str(), "GEOM", strdup( GEOMBase::GetName( aChild ) ) ), false );
234             }
235           }
236         }
237       }
238     }
239   }
240
241   if ( updateView )
242     getDisplayer()->UpdateViewer();
243 }
244
245 //================================================================
246 // Function : displayPreview
247 // Purpose  : Method for displaying preview based on execute() results
248 //================================================================
249 void GEOMBase_Helper::displayPreview( const bool   activate,
250                                       const bool   update,
251                                       const bool   toRemoveFromEngine,
252                                       const double lineWidth, 
253                                       const int    displayMode, 
254                                       const int    color )
255 {
256   isPreview = true;
257   QString msg;
258   if ( !isValid( msg ) )
259   {
260     erasePreview( update );
261     isPreview = false;
262     return;
263   }
264
265   erasePreview( false );
266
267   try {
268     SUIT_OverrideCursor wc;
269     ObjectList objects;
270     if ( !execute( objects ) || !getOperation()->IsDone() ) {
271       wc.suspend();
272     }
273     else {
274       for ( ObjectList::iterator it = objects.begin(); it != objects.end(); ++it )
275       {
276         displayPreview( *it, true, activate, false, lineWidth, displayMode, color );
277         if ( toRemoveFromEngine )
278           getGeomEngine()->RemoveObject( *it );
279       }
280     }
281   }
282   catch( const SALOME::SALOME_Exception& e ) {
283     SalomeApp_Tools::QtCatchCorbaException( e );
284   }
285
286   isPreview = false;
287
288   if ( update )
289     updateViewer();
290 }
291
292 //================================================================
293 // Function : displayPreview
294 // Purpose  : Method for displaying preview of resulting shape
295 //================================================================
296 void GEOMBase_Helper::displayPreview( GEOM::GEOM_Object_ptr object,
297                                       const bool            append,
298                                       const bool            activate,
299                                       const bool            update,
300                                       const double          lineWidth, 
301                                       const int             displayMode, 
302                                       const int             color )
303 {
304   // Set color for preview shape
305   getDisplayer()->SetColor( color == -1 ? Quantity_NOC_VIOLET : color );
306
307   // set width of displayed shape
308   getDisplayer()->SetWidth( lineWidth );
309   
310   // set display mode of displayed shape
311   int aPrevDispMode = getDisplayer()->SetDisplayMode( displayMode );
312
313   // Disable activation of selection
314   getDisplayer()->SetToActivate( activate );
315
316   // Make a reference to GEOM_Object
317   CORBA::String_var objStr = SalomeApp_Application::orb()->object_to_string( object );
318   getDisplayer()->SetName( objStr.in() );
319
320   // Build prs
321   SALOME_Prs* aPrs = getDisplayer()->BuildPrs( object );
322   if ( aPrs == 0 || aPrs->IsNull() )
323     return;
324
325   // Display prs
326   displayPreview( aPrs, append, update );
327
328   getDisplayer()->UnsetName();
329   getDisplayer()->UnsetColor();
330   getDisplayer()->SetDisplayMode( aPrevDispMode );
331
332   // Enable activation of displayed objects
333   getDisplayer()->SetToActivate( true );
334 }
335
336 //================================================================
337 // Function : displayPreview
338 // Purpose  : Method for displaying arbitrary preview objects (not limited to shapes)
339 //================================================================
340 void GEOMBase_Helper::displayPreview( const SALOME_Prs* prs,
341                                       const bool        append,
342                                       const bool        update )
343 {
344   if ( !append )
345     erasePreview( false );
346
347   // remember current view frame to make correct erase preview later
348   myViewWindow = getActiveView();
349
350   if ( myViewWindow == 0 )
351     return;
352
353   // Display prs
354   SUIT_ViewManager* aViewManager = myViewWindow->getViewManager();
355   if ( aViewManager->getType() == OCCViewer_Viewer::Type() ||
356        aViewManager->getType() == SVTK_Viewer::Type() )
357     {
358       SUIT_ViewModel* aViewModel = aViewManager->getViewModel();
359       SALOME_View* aView = dynamic_cast<SALOME_View*>(aViewModel);
360       if (aView)
361         aView->Display( prs );
362     }
363
364   // Add prs to the preview list
365   myPreview.push_back( (SALOME_Prs*)prs );
366
367   // Update viewer
368   if ( update )
369     getDisplayer()->UpdateViewer();
370 }
371
372 //================================================================
373 // Function : erasePreview
374 // Purpose  :
375 //================================================================
376 void GEOMBase_Helper::erasePreview( const bool update )
377 {
378   // check view frame where the preview was displayed
379   bool vfOK = checkViewWindow() && myViewWindow;
380   // Iterate through presentations and delete them
381   for ( PrsList::iterator anIter = myPreview.begin(); anIter != myPreview.end(); ++anIter ) {
382     if ( vfOK )
383       {
384          SUIT_ViewManager* aViewManager = myViewWindow->getViewManager();
385          if ( aViewManager->getType() == OCCViewer_Viewer::Type() ||
386               aViewManager->getType() == SVTK_Viewer::Type() )
387            {
388              SUIT_ViewModel* aViewModel = aViewManager->getViewModel();
389              SALOME_View* aView = dynamic_cast<SALOME_View*>(aViewModel);
390              if (aView)
391                aView->Erase( *anIter, true );
392            }
393       }
394     delete *anIter;
395   }
396   myPreview.clear();
397
398   // Update viewer
399   if ( update )
400     updateViewer();
401 }
402
403 //================================================================
404 // Function  : localSelection
405 // Purpose   : Activate selection of objects of a given type
406 // IMPORTANT : Works after localSelection( ... ) method call only
407 //================================================================
408 void GEOMBase_Helper::activate( const int theType )
409 {
410   if (!getStudy()) return;
411   _PTR(Study) aStudy = getStudy()->studyDS();
412   _PTR(SComponent) aGeom ( aStudy->FindComponent( "GEOM" ) );
413   if ( !aGeom )
414     return;
415
416   SALOME_ListIO aList;
417   _PTR(ChildIterator) anIter ( aStudy->NewChildIterator( aGeom ) );
418   for ( ; anIter->More(); anIter->Next() )
419   {
420     _PTR(SObject) aSO ( anIter->Value() );
421     if ( aSO )
422     {
423       _PTR(SObject) aRefSO;
424       if ( !aSO->ReferencedObject( aRefSO ) )
425       {
426         GEOM::GEOM_Object_var anObj = GEOM::GEOM_Object::_narrow
427           (GeometryGUI::ClientSObjectToObject(aSO));
428         if ( !anObj->_is_nil() && anObj->GetType() == theType )
429           aList.Append( new SALOME_InteractiveObject( aSO->GetID().c_str(), "GEOM", aSO->GetName().c_str()) );
430       }
431     }
432   }
433
434   getDisplayer()->LocalSelection( aList, 0 );
435 }
436
437 //================================================================
438 // Function : localSelection
439 // Purpose  : Activate selection of subshapes in accordance with mode
440 //            theMode is from TopAbs_ShapeEnum
441 //================================================================
442 void GEOMBase_Helper::localSelection( const ObjectList& theObjs, const int theMode )
443 {
444   SALOME_ListIO aListOfIO;
445
446   ObjectList::const_iterator anIter = theObjs.begin();
447   for ( ; anIter != theObjs.end(); ++anIter )
448   {
449     GEOM::GEOM_Object_ptr anObj = *anIter;
450     if ( anObj->_is_nil() )
451       continue;
452     string aEntry = getEntry( anObj );
453     if ( aEntry != "" )
454       aListOfIO.Append( new SALOME_InteractiveObject(
455         aEntry.c_str(), "GEOM", strdup( GEOMBase::GetName( anObj ) ) ) );
456   }
457
458   getDisplayer()->LocalSelection( aListOfIO, theMode );
459 }
460
461 //================================================================
462 // Function : localSelection
463 // Purpose  : Activate selection of subshapes in accordance with mode
464 //            theMode is from TopAbs_ShapeEnum
465 //================================================================
466 void GEOMBase_Helper::localSelection( GEOM::GEOM_Object_ptr obj, const int mode )
467 {
468   // If object is null local selection for all objects is activated
469   if ( obj->_is_nil() ) {
470     getDisplayer()->LocalSelection( Handle(SALOME_InteractiveObject)(), mode );
471     return;
472   }
473
474   ObjectList objList;
475   objList.push_back( obj );
476   localSelection( objList, mode );
477 }
478
479
480 //================================================================
481 // Function : globalSelection
482 // Purpose  : Activate selection of subshapes. Set selection filters
483 //            in accordance with mode. theMode is from GEOMImpl_Types
484 //================================================================
485 void GEOMBase_Helper::globalSelection( const int theMode, const bool update )
486 {
487   getDisplayer()->GlobalSelection( theMode, update );
488 }
489
490 //================================================================
491 // Function : globalSelection
492 // Purpose  : Activate selection of subshapes. Set selection filters
493 //            in accordance with mode. theMode is from GEOMImpl_Types
494 //================================================================
495 void GEOMBase_Helper::globalSelection( const TColStd_MapOfInteger& theModes,
496                                        const bool update )
497 {
498   getDisplayer()->GlobalSelection( theModes, update );
499 }
500
501 //================================================================
502 // Function : globalSelection
503 // Purpose  : Activate selection of subshapes. Set selection filters
504 //            in accordance with mode. theMode is from GEOMImpl_Types
505 //================================================================
506 void GEOMBase_Helper::globalSelection( const TColStd_MapOfInteger& theModes,
507                                        const QValueList<int>& subShapes,
508                                        const bool update )
509 {
510   getDisplayer()->GlobalSelection( theModes, update, &subShapes);
511 }
512
513 //================================================================
514 // Function : addInStudy
515 // Purpose  : Add object in study
516 //================================================================
517 void GEOMBase_Helper::addInStudy( GEOM::GEOM_Object_ptr theObj, const char* theName )
518 {
519   if ( !hasCommand() )
520     return;
521
522   _PTR(Study) aStudy = getStudy()->studyDS();
523   if ( !aStudy || theObj->_is_nil() )
524     return;
525
526   SALOMEDS::Study_var aStudyDS = GeometryGUI::ClientStudyToStudy(aStudy);
527
528   GEOM::GEOM_Object_ptr aFatherObj = getFather( theObj );
529
530   SALOMEDS::SObject_var aSO =
531     getGeomEngine()->AddInStudy(aStudyDS, theObj, theName, aFatherObj);
532
533   // Each dialog is responsible for this method implementation,
534   // default implementation does nothing
535   restoreSubShapes(aStudyDS, aSO);
536 }
537
538 //================================================================
539 // Function : restoreSubShapes
540 // Purpose  : restore tree of argument's sub-shapes under the resulting shape
541 //================================================================
542 void GEOMBase_Helper::restoreSubShapes (SALOMEDS::Study_ptr   /*theStudy*/,
543                                         SALOMEDS::SObject_ptr /*theSObject*/)
544 {
545   // do nothing by default
546
547   // example of implementation in particular dialog:
548   // GEOM::ListOfGO anArgs;
549   // anArgs.length(0); // empty list means that all arguments should be restored
550   // getGeomEngine()->RestoreSubShapesSO(theStudy, theSObject, anArgs,
551   //                                     /*theFindMethod=*/GEOM::FSM_GetInPlace,
552   //                                     /*theInheritFirstArg=*/false);
553 }
554
555 //================================================================
556 // Function : updateObjBrowser
557 // Purpose  : Update object browser
558 //================================================================
559 void GEOMBase_Helper::updateObjBrowser() const
560 {
561   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>(SUIT_Session::session()->activeApplication());
562   if (app) {
563     CAM_Module* module = app->module( "Geometry" );
564     SalomeApp_Module* appMod = dynamic_cast<SalomeApp_Module*>( module );
565     if ( appMod ) {
566       appMod->updateObjBrowser( true );
567     }
568   }
569 }
570
571 //================================================================
572 // Function : updateViewer
573 // Purpose  : Update active 3D view
574 //================================================================
575 void GEOMBase_Helper::updateViewer()
576 {
577   getDisplayer()->UpdateViewer();
578 }
579
580 //================================================================
581 // Function : getStudyId
582 // Purpose  : Get study Id
583 //================================================================
584 int GEOMBase_Helper::getStudyId() const
585 {
586   int anId = -1;
587   if ( getStudy() )
588     anId = getStudy()->id();
589   return anId;
590 }
591
592 //================================================================
593 // Function : getStudy
594 // Purpose  : Returns the active study. It is recommended to use
595 //            this method instead of direct desktop->getActiveStudy() calls
596 //================================================================
597 SalomeApp_Study* GEOMBase_Helper::getStudy() const
598 {
599   SUIT_Desktop* aDesktop = getDesktop();
600   if (!aDesktop)
601     return 0;
602
603   QPtrList<SUIT_Application> anAppList = SUIT_Session::session()->applications();
604
605   SUIT_Application* anApp = 0;
606   for ( QPtrListIterator<SUIT_Application> it( anAppList ); it.current() ; ++it )
607     {
608       anApp = it.current();
609       if ( anApp->desktop() == aDesktop )
610         break;
611     }
612
613   return dynamic_cast<SalomeApp_Study*>(anApp->activeStudy());
614 }
615
616 //================================================================
617 // Function : getEntry
618 // Purpose  :
619 //================================================================
620 char* GEOMBase_Helper::getEntry( GEOM::GEOM_Object_ptr object ) const
621 {
622   SalomeApp_Study* study = getStudy();
623   if ( study )  {
624     char * objIOR = GEOMBase::GetIORFromObject( object );
625     string IOR( objIOR );
626     free( objIOR );
627     if ( IOR != "" ) {
628       _PTR(SObject) SO ( study->studyDS()->FindObjectIOR( IOR ) );
629       if ( SO ) {
630               return TCollection_AsciiString((char*)SO->GetID().c_str()).ToCString();
631       }
632     }
633   }
634   return "";
635 }
636
637 //================================================================
638 // Function : getDisplayer
639 // Purpose  :
640 //================================================================
641 GEOM_Displayer* GEOMBase_Helper::getDisplayer()
642 {
643   if ( !myDisplayer )
644     myDisplayer = new GEOM_Displayer( getStudy() );
645   return myDisplayer;
646 }
647
648 //================================================================
649 // Function : clearShapeBuffer
650 // Purpose  :
651 //================================================================
652 void GEOMBase_Helper::clearShapeBuffer( GEOM::GEOM_Object_ptr theObj )
653 {
654   if ( CORBA::is_nil( theObj ) )
655     return;
656
657   CORBA::String_var IOR = SalomeApp_Application::orb()->object_to_string( theObj );
658   TCollection_AsciiString asciiIOR( (char *)IOR.in() );
659   GEOM_Client().RemoveShapeFromBuffer( asciiIOR );
660
661   if ( !getStudy() || !getStudy()->studyDS() )
662     return;
663
664   _PTR(Study) aStudy = getStudy()->studyDS();
665   _PTR(SObject) aSObj ( aStudy->FindObjectIOR( string( IOR ) ) );
666   if ( !aSObj )
667     return;
668
669   _PTR(ChildIterator) anIt ( aStudy->NewChildIterator( aSObj ) );
670   for ( anIt->InitEx( true ); anIt->More(); anIt->Next() ) {
671     _PTR(GenericAttribute) anAttr;
672     if ( anIt->Value()->FindAttribute(anAttr, "AttributeIOR") ) {
673       _PTR(AttributeIOR) anIOR ( anAttr );
674       TCollection_AsciiString asciiIOR( (char*)anIOR->Value().c_str() );
675       GEOM_Client().RemoveShapeFromBuffer( asciiIOR );
676     }
677   }
678 }
679
680 //================================================================
681 // Function : openCommand
682 // Purpose  :
683 //================================================================
684 bool GEOMBase_Helper::openCommand()
685 {
686   bool res = false;
687   if ( !getStudy() || hasCommand() )
688     return res;
689
690   GEOM::GEOM_IOperations_var anOp = GEOM::GEOM_IOperations::_narrow( getOperation() );
691   if ( !anOp->_is_nil() ) {
692     myCommand = new GEOM_Operation( SUIT_Session::session()->activeApplication(), anOp.in() );
693     myCommand->start();
694     res = true;
695   }
696
697   return res;
698 }
699
700 //================================================================
701 // Function : abortCommand
702 // Purpose  :
703 //================================================================
704 bool GEOMBase_Helper::abortCommand()
705 {
706   if ( !hasCommand() )
707     return false;
708
709   myCommand->abort();
710   myCommand = 0;
711
712   return true;
713 }
714
715 //================================================================
716 // Function : commitCommand
717 // Purpose  :
718 //================================================================
719 bool GEOMBase_Helper::commitCommand( const char* )
720 {
721   if ( !hasCommand() )
722     return false;
723
724   myCommand->commit();
725   myCommand = 0;
726
727   return true;
728 }
729
730 //================================================================
731 // Function : hasCommand
732 // Purpose  :
733 //================================================================
734 bool GEOMBase_Helper::hasCommand() const
735 {
736   return (bool)myCommand;
737 }
738
739 //================================================================
740 // Function : getOperation
741 // Purpose  :
742 //================================================================
743 GEOM::GEOM_IOperations_ptr GEOMBase_Helper::getOperation()
744 {
745   if ( myOperation->_is_nil() )
746     myOperation = createOperation();
747
748   return myOperation;
749 }
750
751
752
753 //================================================================
754 // Function : checkViewWindow
755 // Purpose  :
756 //================================================================
757 bool GEOMBase_Helper::checkViewWindow()
758 {
759   if ( myViewWindow ){
760     QPtrList<SUIT_ViewWindow> aViewWindowsList = SUIT_Session::session()->activeApplication()->desktop()->windows();
761     for ( QPtrListIterator<SUIT_ViewWindow> it( aViewWindowsList ); it.current(); ++it )
762       {
763         if ( myViewWindow == it.current() )
764           return true;
765       }
766   }
767   myViewWindow = 0;
768   return false;
769 }
770
771 //================================================================
772 // Function : onAccept
773 // Purpose  : This method should be called from dialog's slots onOk() and onApply()
774 //            It perfroms user input validation, then it
775 //            performs a proper operation and manages transactions, etc.
776 //================================================================
777 bool GEOMBase_Helper::onAccept (const bool publish, const bool useTransaction)
778 {
779   SalomeApp_Study* appStudy =
780     dynamic_cast<SalomeApp_Study*>(SUIT_Session::session()->activeApplication()->activeStudy());
781   if (!appStudy) return false;
782   _PTR(Study) aStudy = appStudy->studyDS();
783
784   bool aLocked = (_PTR(AttributeStudyProperties) (aStudy->GetProperties()))->IsLocked();
785   if ( aLocked ) {
786     MESSAGE("GEOMBase_Helper::onAccept - ActiveStudy is locked");
787     SUIT_MessageBox::warn1 ( (QWidget*)SUIT_Session::session()->activeApplication()->desktop(),
788                            QObject::tr("WRN_WARNING"),
789                            QObject::tr("WRN_STUDY_LOCKED"),
790                            QObject::tr("BUT_OK") );
791     return false;
792   }
793
794   QString msg;
795   if ( !isValid( msg ) ) {
796     showError( msg );
797     return false;
798   }
799
800   erasePreview( false );
801
802   try {
803     if ( ( !publish && !useTransaction ) || openCommand() ) {
804       SUIT_OverrideCursor wc;
805       SUIT_Session::session()->activeApplication()->putInfo( "" );
806       ObjectList objects;
807       if ( !execute( objects ) || !getOperation()->IsDone() ) {
808         wc.suspend();
809         abortCommand();
810         showError();
811       }
812       else {
813         addSubshapesToStudy(); // add Subshapes if local selection
814         const int nbObjs = objects.size();
815         int aNumber = 1;
816         for ( ObjectList::iterator it = objects.begin(); it != objects.end(); ++it ) {
817           if ( publish ) {
818             QString aName = getNewObjectName();
819             if ( nbObjs > 1 ) {
820               if (aName.isEmpty())
821                 aName = getPrefix(*it);
822               if (nbObjs <= 30) {
823                 // Try to find a unique name
824                 aName = GEOMBase::GetDefaultName(aName);
825               } else {
826                 // Don't check name uniqueness in case of numerous objects
827                 aName = aName + "_" + QString::number(aNumber++);
828               }
829             } else {
830               // PAL6521: use a prefix, if some dialog box doesn't reimplement getNewObjectName()
831               if ( aName.isEmpty() )
832                 aName = GEOMBase::GetDefaultName( getPrefix( *it ) );
833             }
834             addInStudy( *it, aName.latin1() );
835             // updateView=false
836             display( *it, false );
837           }
838           else {
839             // asv : fix of PAL6454. If publish==false, then the original shape
840             // was modified, and need to be re-cached in GEOM_Client before redisplay
841             clearShapeBuffer( *it );
842             // withChildren=true, updateView=false
843             redisplay( *it, true, false );
844           }
845         }
846
847         if ( nbObjs ) {
848           commitCommand();
849           updateObjBrowser();
850           SUIT_Session::session()->activeApplication()->putInfo( QObject::tr("GEOM_PRP_DONE") );
851         }
852         else
853           abortCommand();
854       }
855     }
856   }
857   catch( const SALOME::SALOME_Exception& e ) {
858     SalomeApp_Tools::QtCatchCorbaException( e );
859     abortCommand();
860   }
861
862   updateViewer();
863
864   return true;
865 }
866
867
868 //================================================================
869 // Function : showError
870 // Purpose  : Shows a message box with infromation about an error taken from getOperation()->GetErrorCode()
871 //================================================================
872 void GEOMBase_Helper::showError()
873 {
874   QString msg;
875   if ( !getOperation()->_is_nil() )
876     msg = QObject::tr( getOperation()->GetErrorCode() );
877
878   if ( msg.isEmpty() )
879     msg = QObject::tr( "GEOM_PRP_ABORT" );
880
881   SUIT_MessageBox::error1( SUIT_Session::session()->activeApplication()->desktop(),
882                            QObject::tr( "GEOM_ERROR_STATUS" ),
883                            msg,
884                            QObject::tr( "BUT_OK" ) );
885 }
886
887 //================================================================
888 // Function : showError
889 // Purpose  : Shows a error message followed by <msg>
890 //================================================================
891 void GEOMBase_Helper::showError( const QString& msg )
892 {
893   QString str( QObject::tr( "GEOM_INCORRECT_INPUT" ) );
894   if ( !msg.isEmpty() )
895     str += "\n" + msg;
896   SUIT_MessageBox::error1(SUIT_Session::session()->activeApplication()->desktop(), QObject::tr( "GEOM_ERROR" ), str, QObject::tr( "BUT_OK" ) );
897 }
898
899 //////////////////////////////////////////////////////////////////
900 // Virtual methods to be redefined in dialogs
901 //////////////////////////////////////////////////////////////////
902
903 //================================================================
904 // Function : createOperation
905 // Purpose  : Redefine this method to return proper IOperation reference
906 //================================================================
907 GEOM::GEOM_IOperations_ptr GEOMBase_Helper::createOperation()
908 {
909   GEOM::GEOM_IOperations_var aNilOp;
910   return aNilOp._retn();
911 }
912
913 //================================================================
914 // Function : isValid
915 // Purpose  : Called by onAccept(). Redefine this method to check validity of user input in dialog boxes.
916 //================================================================
917 bool GEOMBase_Helper::isValid( QString& )
918 {
919   return true;
920 }
921
922 //================================================================
923 // Function : execute
924 // Purpose  : This method is called by onAccept().
925 //            It should perform the required operation and put all new or modified objects into
926 //            <objects> argument.Should return <false> if some error occurs during its execution.
927 //================================================================
928 bool GEOMBase_Helper::execute( ObjectList& objects )
929 {
930   return false;
931 }
932
933 //================================================================
934 // Function : getFather
935 // Purpose  : This method is called by addInStudy(). It should return a father object
936 //            for <theObj> or a nil reference if <theObj> should be published
937 //            as a top-level object.
938 //================================================================
939 GEOM::GEOM_Object_ptr GEOMBase_Helper::getFather( GEOM::GEOM_Object_ptr theObj )
940 {
941   return GEOM::GEOM_Object::_nil();
942 }
943
944 //================================================================
945 // Function : getNewObjectName
946 // Purpose  : Redefine this method to return proper name for a new object
947 //================================================================
948 const char* GEOMBase_Helper::getNewObjectName() const
949 {
950   return "";
951 }
952
953 //================================================================
954 // Function : getPrefix
955 // Purpose  : Get prefix for name of created object
956 //================================================================
957 QString GEOMBase_Helper::getPrefix( GEOM::GEOM_Object_ptr theObj ) const
958 {
959   if ( !myPrefix.isEmpty() || theObj->_is_nil() )
960     return myPrefix;
961
962   //TopoDS_Shape aShape;
963   //if ( !GEOMBase::GetShape( theObj, aShape ) )
964   //  return "";
965   //
966   //long aType = aShape.ShapeType();
967   GEOM::shape_type aType = theObj->GetShapeType();
968
969   switch ( aType )
970   {
971     case GEOM::VERTEX   : return QObject::tr( "GEOM_VERTEX" );
972     case GEOM::EDGE     : return QObject::tr( "GEOM_EDGE" );
973     case GEOM::WIRE     : return QObject::tr( "GEOM_WIRE" );
974     case GEOM::FACE     : return QObject::tr( "GEOM_FACE" );
975     case GEOM::SHELL    : return QObject::tr( "GEOM_SHELL" );
976     case GEOM::SOLID    : return QObject::tr( "GEOM_SOLID" );
977     case GEOM::COMPSOLID: return QObject::tr( "GEOM_COMPOUNDSOLID" );
978     case GEOM::COMPOUND : return QObject::tr( "GEOM_COMPOUND" );
979     default : return "";
980   }
981 }
982
983 //================================================================
984 // Function : selectedIO
985 // Purpose  : Return the list of selected SALOME_InteractiveObject's
986 //================================================================
987 const SALOME_ListIO& GEOMBase_Helper::selectedIO()
988 {
989   mySelected.Clear();
990
991   SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
992   if ( app ) {
993     LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
994     if ( aSelMgr )
995       aSelMgr->selectedObjects( mySelected );
996   }
997
998   return mySelected;
999 }
1000
1001 //================================================================
1002 // Function : IObjectCount
1003 // Purpose  : Return the number of selected objects
1004 //================================================================
1005 int GEOMBase_Helper::IObjectCount()
1006 {
1007   return selectedIO().Extent();
1008 }
1009
1010 //================================================================
1011 // Function : firstIObject
1012 // Purpose  :  Return the first selected object in the selected object list
1013 //================================================================
1014 Handle(SALOME_InteractiveObject) GEOMBase_Helper::firstIObject()
1015 {
1016   const SALOME_ListIO& aList = selectedIO();
1017   return aList.Extent() > 0 ? aList.First() : Handle(SALOME_InteractiveObject)();
1018 }
1019
1020 //================================================================
1021 // Function : lastIObject
1022 // Purpose  : Return the last selected object in the selected object list
1023 //================================================================
1024 Handle(SALOME_InteractiveObject) GEOMBase_Helper::lastIObject()
1025 {
1026   const SALOME_ListIO& aList = selectedIO();
1027   return aList.Extent() > 0 ? aList.Last() : Handle(SALOME_InteractiveObject)();
1028 }
1029
1030 //================================================================
1031 // Function : getDesktop
1032 // Purpose  : Returns myDesktop field.  Initialized in constructor, usually as dynamic_cast<SUIT_Desktop*>(parentWidget())
1033 //================================================================
1034 SUIT_Desktop* GEOMBase_Helper::getDesktop() const
1035 {
1036   return myDesktop;
1037 }
1038
1039 //================================================================
1040 // Function : selectObjects
1041 // Purpose  : Selects list of objects 
1042 //================================================================
1043 bool GEOMBase_Helper::selectObjects( ObjectList& objects )
1044 {
1045   SUIT_DataOwnerPtrList aList;
1046   ObjectList::iterator anIter;
1047   for ( anIter = objects.begin(); anIter != objects.end(); ++anIter )
1048   {
1049     string entry = getEntry( *anIter );
1050     QString aEntry( entry.c_str() );
1051     LightApp_DataOwner* anOwher = new LightApp_DataOwner( aEntry );
1052     aList.append( anOwher );
1053   }
1054   
1055   SUIT_Session* session = SUIT_Session::session();
1056   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
1057   if ( !app )
1058     return false;
1059
1060   LightApp_SelectionMgr* aMgr = app->selectionMgr();
1061   if ( !aMgr )
1062     return false;
1063   
1064   aMgr->setSelected( aList, false );
1065   
1066   return true;
1067 }
1068   
1069 //================================================================
1070 // Function : findObjectInFather
1071 // Purpose  : It should return an object if its founded in study or
1072 //            return Null object if the object is not founded
1073 //================================================================
1074 GEOM::GEOM_Object_ptr GEOMBase_Helper::findObjectInFather( GEOM::GEOM_Object_ptr theFather, const char* theName)
1075 {
1076   SalomeApp_Application* app =
1077     dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
1078   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
1079   _PTR(Study) aDStudy = appStudy->studyDS();
1080   string IOR = GEOMBase::GetIORFromObject( theFather );
1081   _PTR(SObject) SObj ( aDStudy->FindObjectIOR( IOR ) );
1082
1083   bool inStudy = false;
1084   GEOM::GEOM_Object_var aReturnObject;
1085   for (_PTR(ChildIterator) iit (aDStudy->NewChildIterator( SObj )); iit->More() && !inStudy; iit->Next()) {
1086     _PTR(SObject) child (iit->Value());
1087     QString aChildName = child->GetName();
1088     if (aChildName == theName) {
1089       inStudy = true;
1090       CORBA::Object_var corbaObj = GeometryGUI::ClientSObjectToObject(iit->Value());
1091       aReturnObject = GEOM::GEOM_Object::_narrow( corbaObj );
1092     }
1093   }
1094   if (inStudy)
1095     return aReturnObject._retn();
1096
1097   return GEOM::GEOM_Object::_nil();
1098 }
1099   
1100 //================================================================
1101 // Function : addSubshapesToStudy
1102 // Purpose  : Virtual method to add subshapes if needs
1103 //================================================================  
1104 void GEOMBase_Helper::addSubshapesToStudy()
1105 {
1106   //Impemented in Dialogs, called from Accept method
1107 }
1108
1109 //================================================================
1110 // Function : addSubshapesToFather
1111 // Purpose  : Method to add Father Subshapes to Study if it`s not exist
1112 //================================================================  
1113 void GEOMBase_Helper::addSubshapesToFather( QMap<QString, GEOM::GEOM_Object_var>& theMap )
1114 {
1115   //GetStudyDS
1116   SalomeApp_Application* app =
1117     dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
1118   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
1119   _PTR(Study) aDStudy = appStudy->studyDS();
1120
1121   GEOM::GEOM_IGroupOperations_var anOp = getGeomEngine()->GetIGroupOperations( getStudyId() );
1122  
1123   for( QMap<QString, GEOM::GEOM_Object_var>::Iterator it = theMap.begin(); it != theMap.end(); it++ )
1124     {
1125       if ( !anOp->_is_nil() ) {
1126         GEOM::GEOM_Object_var aFatherObj = anOp->GetMainShape( it.data() );
1127         if ( !aFatherObj->_is_nil() ) { 
1128         GEOM::GEOM_Object_var aFindedObject = findObjectInFather(aFatherObj, it.key() );
1129       
1130         //Add Object to study if its not exist
1131         if ( aFindedObject == GEOM::GEOM_Object::_nil() )
1132           GeometryGUI::GetGeomGen()->AddInStudy(GeometryGUI::ClientStudyToStudy(aDStudy),
1133                                               it.data(), it.key(), aFatherObj );
1134         }
1135       }
1136       else {
1137         //cout << " anOperations is NULL! " << endl;
1138       }
1139     }
1140 }  
1141
1142