Salome HOME
Fix for bug PAL13958(iso don't run (regressive bug)).
[modules/geom.git] / src / GEOMGUI / GEOM_Displayer.cxx
1 //  GEOM GEOMGUI : GUI for Geometry component
2 //
3 //  Copyright (C) 2003  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   : GEOM_Displayer.cxx
25 //  Author : Vadim SANDLER
26 //  Module : GEOM
27 //  $Header$
28
29 #include "GEOM_Displayer.h"
30
31 #include "GeometryGUI.h"
32
33 #include "GEOM_TypeFilter.h"
34 #include "GEOM_EdgeFilter.h"
35 #include "GEOM_FaceFilter.h"
36 #include "GEOM_PreviewFilter.h"
37 #include "GEOM_LogicalFilter.h"
38 #include "GEOM_OCCFilter.h"
39
40 #include "GEOM_Actor.h"
41 #include "GEOM_AssemblyBuilder.h"
42 #include "GEOM_AISShape.hxx"
43 #include "GEOM_AISTrihedron.hxx"
44 #include "GEOM_VTKTrihedron.hxx"
45
46 #include <SUIT_Desktop.h>
47 #include <SUIT_ViewWindow.h>
48 #include <SUIT_Session.h>
49 #include <SUIT_Tools.h>
50 #include <SUIT_ViewManager.h>
51
52 #include <SalomeApp_Study.h>
53 #include <SalomeApp_Application.h>
54 #include <LightApp_SelectionMgr.h>
55 #include <SalomeApp_TypeFilter.h>
56 #include <SalomeApp_Tools.h>
57
58 #include <SALOME_ListIteratorOfListIO.hxx>
59
60 #include <SOCC_Prs.h>
61 #include <SOCC_ViewModel.h>
62
63 #include <SVTK_Prs.h>
64 #include <SVTK_ViewModel.h>
65
66 #include <SALOMEDSClient.hxx>
67 #include <SALOMEDSClient_SObject.hxx>
68
69 // OCCT Includes
70 #include <AIS_Drawer.hxx>
71 #include <AIS_ListIteratorOfListOfInteractive.hxx>
72 #include <Prs3d_IsoAspect.hxx>
73 #include <Prs3d_PointAspect.hxx>
74 #include <Graphic3d_AspectMarker3d.hxx>
75 #include <StdSelect_TypeOfEdge.hxx>
76 #include <StdSelect_TypeOfFace.hxx>
77 #include <TopoDS_Face.hxx>
78 #include <BRep_Tool.hxx>
79 #include <Geom_Plane.hxx>
80 #include <Geom_Axis2Placement.hxx>
81 #include <gp_Pln.hxx>
82 #include <TColStd_MapOfInteger.hxx>
83 #include <TColStd_MapIteratorOfMapOfInteger.hxx>
84
85 // VTK Includes
86 #include <vtkActorCollection.h>
87 #include <vtkProperty.h>
88
89 // STL Includes
90 #include <cstring>
91
92 // CORBA Headers
93 #include CORBA_CLIENT_HEADER(SALOMEDS_Attributes)
94
95 #include "GEOMImpl_Types.hxx"
96
97 using namespace std;
98
99 //================================================================
100 // Function : getActiveStudy
101 // Purpose  : Get active study, returns 0 if no open study frame
102 //================================================================
103 static inline SalomeApp_Study* getActiveStudy()
104 {
105   SUIT_Session* session = SUIT_Session::session();
106   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
107   if ( app )
108     return ( SalomeApp_Study* )app->activeStudy();
109   return 0;
110 }
111
112 //================================================================
113 // Function : getTopAbsMode
114 // Purpose  : Get TopAbs_ShapeEnum value corresponding to the
115 //            one from GEOMImpl_Types.h
116 //================================================================
117 static inline int getTopAbsMode( const int implType )
118 {
119   switch ( implType )
120   {
121     case GEOM_COMPOUND  : return TopAbs_COMPOUND;
122     case GEOM_SOLID     : return TopAbs_SOLID;
123     case GEOM_SHELL     : return TopAbs_SHELL;
124     case GEOM_FACE      : return TopAbs_FACE;
125     case GEOM_WIRE      : return TopAbs_WIRE;
126     case GEOM_EDGE      : return TopAbs_EDGE;
127     case GEOM_POINT     : return TopAbs_VERTEX;
128     default             : return -1;
129   }
130 }
131
132 //================================================================
133 // Function : getFilter
134 // Purpose  : Get filter corresponding to the type of object
135 //            from GEOMImpl_Types.h
136 //================================================================
137 SUIT_SelectionFilter* GEOM_Displayer::getFilter( const int theMode )
138 {
139   SUIT_SelectionFilter* aFilter;
140
141   int aTopAbsMode = getTopAbsMode( theMode );
142   if ( aTopAbsMode != -1 )
143     aFilter = new GEOM_TypeFilter( getStudy(), aTopAbsMode, true ); //@ aFilter = new GEOM_TypeFilter( ( TopAbs_ShapeEnum )aTopAbsMode );
144   else
145     switch ( theMode )
146       {
147       case GEOM_LINE      : aFilter = new GEOM_EdgeFilter( getStudy(), StdSelect_Line ); break;
148       case GEOM_CIRCLE    : aFilter = new GEOM_EdgeFilter( getStudy(), StdSelect_Circle ); break;
149
150       case GEOM_PLANE     : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Plane ); break;
151       case GEOM_CYLINDER  : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Cylinder ); break;
152       case GEOM_SPHERE    : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Sphere ); break;
153       case GEOM_TORUS     : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Torus ); break;
154       case GEOM_REVOLUTION: aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Revol ); break;
155       case GEOM_CONE      : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Cone ); break;
156
157       case GEOM_PREVIEW   : aFilter = new GEOM_PreviewFilter( getStudy() ); break;
158
159       case GEOM_ALLSHAPES : aFilter = new GEOM_SelectionFilter(getStudy(), true ); break;
160       case GEOM_ALLGEOM   : aFilter = new SalomeApp_TypeFilter( getStudy(), "GEOM" ); break;
161
162       default             : aFilter = new GEOM_TypeFilter( getStudy(), theMode ); break;
163       }
164
165   return aFilter;
166 }
167
168 //================================================================
169 // Function : getEntry
170 // Purpose  :
171 //================================================================
172 static string getEntry( GEOM::GEOM_Object_ptr object )
173 {
174   SUIT_Session* session = SUIT_Session::session();
175   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
176   if ( app )
177   {
178     string IOR = app->orb()->object_to_string( object );
179     if ( IOR != "" )
180     {
181       SalomeApp_Study* study = ( SalomeApp_Study* )app->activeStudy();
182       _PTR(SObject) SO ( study->studyDS()->FindObjectIOR( IOR ) );
183       if ( SO )
184         return SO->GetID();
185     }
186   }
187   return "";
188 }
189
190 //================================================================
191 // Function : getName
192 // Purpose  :
193 //================================================================
194 static string getName( GEOM::GEOM_Object_ptr object )
195 {
196   SUIT_Session* session = SUIT_Session::session();
197   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
198   if ( app )
199   {
200     string IOR = app->orb()->object_to_string( object );
201     if ( IOR != "" )
202     {
203       SalomeApp_Study* study = ( SalomeApp_Study* )app->activeStudy();
204       _PTR(SObject) aSObj ( study->studyDS()->FindObjectIOR( IOR ) );
205
206       _PTR(GenericAttribute) anAttr;
207
208       if ( aSObj && aSObj->FindAttribute( anAttr, "AttributeName") )
209       {
210         _PTR(AttributeName) aNameAttr( anAttr );
211         return aNameAttr->Value();
212       }
213     }
214   }
215
216   return "";
217 }
218
219 //=================================================================
220 /*!
221  *  GEOM_Displayer::GEOM_Displayer
222  *  Constructor
223  */
224 //=================================================================
225 GEOM_Displayer::GEOM_Displayer( SalomeApp_Study* st )
226 {
227   if( st )
228     myApp = dynamic_cast<SalomeApp_Application*>( st->application() );
229   else
230     myApp = 0;
231
232   /* Shading Color */
233   SUIT_Session* session = SUIT_Session::session();
234   SUIT_ResourceMgr* resMgr = session->resourceMgr();
235
236   QColor col = resMgr->colorValue( "Geometry", "shading_color", QColor( 255, 0, 0 ) );
237   myShadingColor = SalomeApp_Tools::color( col );
238
239   myDisplayMode = resMgr->integerValue("Geometry", "display_mode", 0);
240
241   myColor = -1;
242   // This color is used for shape displaying. If it is equal -1 then
243   // default color is used.
244
245   myWidth = -1;
246   myType = -1;
247
248   myToActivate = true;
249   // This parameter is used for activisation/deactivisation of objects to be displayed
250
251   myViewFrame = 0;
252 }
253
254 //=================================================================
255 /*!
256  *  GEOM_Displayer::~GEOM_Displayer
257  *  Destructor
258  */
259 //=================================================================
260 GEOM_Displayer::~GEOM_Displayer()
261 {
262 }
263
264 //=================================================================
265 /*!
266  *  GEOM_Displayer::Display
267  *  Display interactive object in the current viewer
268  */
269 //=================================================================
270 void GEOM_Displayer::Display( const Handle(SALOME_InteractiveObject)& theIO,
271                              const bool updateViewer,
272                              SALOME_View* theViewFrame )
273 {
274   SALOME_View* vf = theViewFrame ? theViewFrame : GetActiveView();
275   if ( vf )
276   {
277     SALOME_Prs* prs = buildPresentation( theIO->getEntry(), vf );
278
279     if ( prs )
280     {
281       vf->BeforeDisplay( this );
282       vf->Display( prs );
283       vf->AfterDisplay( this );
284
285       if ( updateViewer )
286         vf->Repaint();
287
288       delete prs;  // delete presentation because displayer is its owner
289     }
290   }
291 }
292
293 //=================================================================
294 /*!
295  *  GEOM_Displayer::Display
296  *  This overloaded Display() method can be useful for operations
297  *  not using dialog boxes.
298  */
299 //=================================================================
300 void GEOM_Displayer::Display( GEOM::GEOM_Object_ptr theObj, const bool updateViewer )
301 {
302   if ( theObj->_is_nil() )
303     return;
304
305   string entry = getEntry( theObj );
306   if ( entry != "" ) {
307     Display(new SALOME_InteractiveObject(entry.c_str(), "GEOM", getName(theObj).c_str()),
308             updateViewer);
309   }
310 }
311
312 //=================================================================
313 /*!
314  *  GEOM_Displayer::Erase
315  *  Erase interactive object in the current viewer
316  */
317 //=================================================================
318 void GEOM_Displayer::Erase( const Handle(SALOME_InteractiveObject)& theIO,
319                             const bool forced,
320                             const bool updateViewer,
321                             SALOME_View* theViewFrame )
322 {
323   if ( theIO.IsNull() )
324     return;
325
326   SALOME_View* vf = theViewFrame ? theViewFrame : GetActiveView();
327
328   if ( vf ) {
329     SALOME_Prs* prs = vf->CreatePrs( theIO->getEntry() );
330     if ( prs ) {
331       vf->Erase( prs, forced );
332       if ( updateViewer )
333         vf->Repaint();
334       delete prs;  // delete presentation because displayer is its owner
335     }
336   }
337 }
338
339 //=================================================================
340 /*!
341  *  GEOM_Displayer::Erase
342  *  Erase geometry object in the current viewer
343  */
344 //=================================================================
345 void GEOM_Displayer::Erase( GEOM::GEOM_Object_ptr theObj,
346                             const bool forced,
347                             const bool updateViewer )
348 {
349   string entry = getEntry( theObj );
350   if ( entry != "" )
351   {
352     Erase(new SALOME_InteractiveObject(entry.c_str(), "GEOM", getName(theObj).c_str()),
353           forced, updateViewer);
354   }
355 }
356
357 //=================================================================
358 /*!
359  *  GEOM_Displayer::Redisplay
360  *  Redisplay (erase and then display again) interactive object
361  *  in the current viewer
362  */
363 //=================================================================
364 void GEOM_Displayer::Redisplay( const Handle(SALOME_InteractiveObject)& theIO,
365                                 const bool updateViewer )
366 {
367   // Remove the object permanently (<forced> == true)
368   SUIT_Session* ses = SUIT_Session::session();
369   SUIT_Application* app = ses->activeApplication();
370   if ( app )
371   {
372     SUIT_Desktop* desk = app->desktop();
373     QPtrList<SUIT_ViewWindow> wnds = desk->windows();
374     SUIT_ViewWindow* wnd;
375     for ( wnd = wnds.first(); wnd; wnd = wnds.next() )
376     {
377       SUIT_ViewManager* vman = wnd->getViewManager();
378       if ( vman )
379       {
380         SUIT_ViewModel* vmodel = vman->getViewModel();
381         if ( vmodel )
382         {
383           SALOME_View* view = dynamic_cast<SALOME_View*>(vmodel);
384           if ( view )
385           {
386             if ( view->isVisible( theIO ) || view == GetActiveView() )
387             {
388               Erase( theIO, true, false, view );
389               Display( theIO, updateViewer, view );
390             }
391           }
392         }
393       }
394     }
395   }
396 }
397
398 //=================================================================
399 /*!
400  *  GEOM_Displayer::Display
401  *  Calls Display() method for each object in the given list
402  */
403 //=================================================================
404 void GEOM_Displayer::Display( const SALOME_ListIO& theIOList, const bool updateViewer )
405 {
406   SALOME_ListIteratorOfListIO Iter( theIOList );
407   for ( ; Iter.More(); Iter.Next() ) {
408     Display( Iter.Value(), false );
409   }
410   if ( updateViewer )
411     UpdateViewer();
412 }
413
414 //=================================================================
415 /*!
416  *  GEOM_Displayer::Erase
417  *  Calls Erase() method for each object in the given list
418  */
419 //=================================================================
420 void GEOM_Displayer::Erase( const SALOME_ListIO& theIOList,
421                             const bool forced,
422                             const bool updateViewer )
423 {
424   SALOME_ListIteratorOfListIO Iter( theIOList );
425   for ( ; Iter.More(); Iter.Next() )
426     Erase( Iter.Value(), forced, false );
427
428   if ( updateViewer )
429     UpdateViewer();
430 }
431
432 //=================================================================
433 /*!
434  *  GEOM_Displayer::Redisplay
435  *  Calls Redisplay() method for each object in the given list
436  */
437 //=================================================================
438 void GEOM_Displayer::Redisplay( const SALOME_ListIO& theIOList, const bool updateViewer )
439 {
440   SALOME_ListIteratorOfListIO Iter( theIOList );
441   for ( ; Iter.More(); Iter.Next() )
442     Redisplay( Iter.Value(), false );
443
444   if ( updateViewer )
445     UpdateViewer();
446 }
447
448 //=================================================================
449 /*!
450  *  GEOM_Displayer::Update
451  *  Update OCC presentaion
452  *  [ Reimplemented from SALOME_Displayer ]
453  */
454 //=================================================================
455 void GEOM_Displayer::Update( SALOME_OCCPrs* prs )
456 {
457   SOCC_Prs* occPrs = dynamic_cast<SOCC_Prs*>( prs );
458   if ( !occPrs )
459     return;
460
461   if ( myType == GEOM_MARKER && !myShape.IsNull() && myShape.ShapeType() == TopAbs_FACE )
462   {
463     TopoDS_Face aFace = TopoDS::Face( myShape );
464     Handle(Geom_Plane) aPlane = Handle(Geom_Plane)::DownCast( BRep_Tool::Surface( aFace ) );
465     if ( !aPlane.IsNull() )
466     {
467       gp_Ax3 aPos = aPlane->Pln().Position();
468       Handle(Geom_Axis2Placement) aPlc = new Geom_Axis2Placement( aPos.Ax2() );
469
470       Handle(GEOM_AISTrihedron) aTrh;
471
472       if ( occPrs->IsNull() )
473       {
474         aTrh = new GEOM_AISTrihedron( aPlc );
475
476         if ( HasColor() )
477           aTrh->SetColor( (Quantity_NameOfColor)GetColor() );
478
479         if ( HasWidth() )
480           aTrh->SetWidth( GetWidth() );
481
482         if ( !myIO.IsNull() )
483         {
484           aTrh->setIO( myIO );
485           aTrh->SetOwner( myIO );
486         }
487
488         occPrs->AddObject( aTrh );
489       }
490       else
491       {
492         AIS_ListOfInteractive aList;
493         occPrs->GetObjects( aList );
494         AIS_ListIteratorOfListOfInteractive anIter( aList );
495         for ( ; anIter.More(); anIter.Next() )
496         {
497           aTrh = Handle(GEOM_AISTrihedron)::DownCast( anIter.Value() );
498           if ( !aTrh.IsNull() )
499           {
500             aTrh->SetComponent( aPlc );
501             aTrh->SetToUpdate();
502           }
503         }
504       }
505
506       occPrs->SetToActivate( ToActivate() );
507     }
508   }
509   else
510   {
511     // if presentation is empty we try to create new one
512     if ( occPrs->IsNull() )
513     {
514       if ( !myShape.IsNull() )
515       {
516         Handle(GEOM_AISShape) AISShape = new GEOM_AISShape( myShape, "" );
517         // Temporary staff: vertex must be infinite for correct visualization
518         AISShape->SetInfiniteState( myShape.Infinite() || myShape.ShapeType() == TopAbs_VERTEX );
519
520         // Setup shape properties here ..., e.g. display mode, color, transparency, etc
521         AISShape->SetDisplayMode( myDisplayMode );
522         AISShape->SetShadingColor( myShadingColor );
523
524         // Set color and number for iso lines
525         SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
526         QColor col = aResMgr->colorValue( "Geometry", "isos_color", QColor(int(0.5*255), int(0.5*255), int(0.5*255)) );
527         Quantity_Color aColor = SalomeApp_Tools::color( col );
528         int anUIsoNumber = aResMgr->integerValue("OCCViewer", "iso_number_u", 1);
529         int aVIsoNumber  = aResMgr->integerValue("OCCViewer", "iso_number_v", 1);
530                 
531         Handle(Prs3d_IsoAspect) anAspect = AISShape->Attributes()->UIsoAspect();
532         anAspect->SetNumber( anUIsoNumber );
533         anAspect->SetColor( aColor );
534         AISShape->Attributes()->SetUIsoAspect( anAspect );
535         
536         anAspect = AISShape->Attributes()->VIsoAspect();
537         anAspect->SetNumber( aVIsoNumber );
538         anAspect->SetColor( aColor );
539         AISShape->Attributes()->SetVIsoAspect( anAspect );
540         
541         if ( HasColor() )
542         {
543           AISShape->SetColor( (Quantity_NameOfColor)GetColor() );
544           if ( myShape.ShapeType() == TopAbs_VERTEX )
545           {
546             Handle(Prs3d_PointAspect) anAspect = AISShape->Attributes()->PointAspect();
547             anAspect->SetColor( (Quantity_NameOfColor)GetColor() );
548             AISShape->Attributes()->SetPointAspect( anAspect );
549           }
550         }
551         else
552           {
553             if ( myShape.ShapeType() == TopAbs_VERTEX )
554               {
555                 col = aResMgr->colorValue( "Geometry", "point_color", QColor( 255, 255, 0 ) );
556                 aColor = SalomeApp_Tools::color( col );
557                 
558                 Handle(Prs3d_PointAspect) anAspect = AISShape->Attributes()->PointAspect();
559                 anAspect->SetColor( aColor );
560                 AISShape->Attributes()->SetPointAspect( anAspect );
561               }
562             else
563               {
564                 // Set line aspect
565                 col = aResMgr->colorValue( "Geometry", "wireframe_color", QColor( 255, 255, 0 ) );
566                 aColor = SalomeApp_Tools::color( col );
567                 
568                 Handle(Prs3d_LineAspect) anAspect = AISShape->Attributes()->LineAspect();
569                 anAspect->SetColor( aColor );
570                 AISShape->Attributes()->SetLineAspect( anAspect );
571
572                 // Set unfree boundaries aspect
573                 anAspect = AISShape->Attributes()->UnFreeBoundaryAspect();
574                 anAspect->SetColor( aColor );
575                 AISShape->Attributes()->SetUnFreeBoundaryAspect( anAspect );
576
577                 // Set free boundaries aspect
578                 col = aResMgr->colorValue( "Geometry", "free_bound_color", QColor( 0, 255, 0 ) );
579                 aColor = SalomeApp_Tools::color( col );
580                 
581                 anAspect = AISShape->Attributes()->FreeBoundaryAspect();
582                 anAspect->SetColor( aColor );
583                 AISShape->Attributes()->SetFreeBoundaryAspect( anAspect );
584                 
585                 // Set wire aspect
586                 col = aResMgr->colorValue( "Geometry", "line_color", QColor( 255, 0, 0 ) );
587                 aColor = SalomeApp_Tools::color( col );
588                 
589                 anAspect = AISShape->Attributes()->WireAspect();
590                 anAspect->SetColor( aColor );
591                 AISShape->Attributes()->SetWireAspect( anAspect );
592               }
593           }
594
595         if ( HasWidth() )
596           AISShape->SetWidth( GetWidth() );
597
598         if ( !myIO.IsNull() )
599         {
600           AISShape->setIO( myIO );
601           AISShape->SetOwner( myIO );
602         }
603         else if ( !myName.empty() )
604         {
605           // Workaround to allow selection of temporary objects
606           static int tempId = 0;
607           char buf[50];
608           sprintf( buf, "TEMP_%d", tempId++ );
609           Handle( SALOME_InteractiveObject ) anObj =
610             new SALOME_InteractiveObject( buf, "GEOM", myName.c_str() );
611           AISShape->setIO( anObj );
612           AISShape->SetOwner( anObj );
613         }
614         // AISShape->SetName(???); ??? necessary to set name ???
615         occPrs->AddObject( AISShape );
616
617         // In accordance with ToActivate() value object will be activated/deactivated
618         // when it will be displayed
619         occPrs->SetToActivate( ToActivate() );
620       }
621     }
622     // if presentation is found -> set again shape for it
623     else
624     {
625       if ( !myShape.IsNull() )
626       {
627         AIS_ListOfInteractive IOList;
628         occPrs->GetObjects( IOList );
629         AIS_ListIteratorOfListOfInteractive Iter( IOList );
630         for ( ; Iter.More(); Iter.Next() )
631         {
632           Handle(GEOM_AISShape) AISShape = Handle(GEOM_AISShape)::DownCast( Iter.Value() );
633           if ( AISShape.IsNull() )
634             continue;
635           if ( AISShape->Shape() != myShape )
636           {
637             AISShape->Set( myShape );
638             AISShape->UpdateSelection();
639             AISShape->SetToUpdate();
640           }
641           if ( !myIO.IsNull() )
642           {
643             AISShape->setIO( myIO );
644             AISShape->SetOwner( myIO );
645           }
646         }
647       }
648     }
649   }
650 }
651
652 //=================================================================
653 /*!
654  *  GEOM_Displayer::Update
655  *  Update VTK presentaion
656  *  [ Reimplemented from SALOME_Displayer ]
657  */
658 //=================================================================
659 void GEOM_Displayer::Update( SALOME_VTKPrs* prs )
660 {
661   SVTK_Prs* vtkPrs = dynamic_cast<SVTK_Prs*>( prs );
662   if ( !vtkPrs || myShape.IsNull() )
663     return;
664
665   vtkActorCollection* theActors = 0;
666
667   if ( myType == GEOM_MARKER && !myShape.IsNull() && myShape.ShapeType() == TopAbs_FACE )
668   {
669     myToActivate = false;
670     GEOM_VTKTrihedron* aTrh = GEOM_VTKTrihedron::New();
671
672     if ( HasColor() )
673     {
674       Quantity_Color aColor( (Quantity_NameOfColor)GetColor() );
675       aTrh->SetColor( aColor.Red(), aColor.Green(), aColor.Blue() );
676     }
677
678     Handle(Geom_Plane) aPlane =
679       Handle(Geom_Plane)::DownCast( BRep_Tool::Surface( TopoDS::Face( myShape ) ) );
680     if ( aPlane.IsNull() )
681       return;
682
683     gp_Ax2 anAx2 = aPlane->Pln().Position().Ax2();
684     aTrh->SetPlacement( new Geom_Axis2Placement( anAx2 ) );
685
686 //    if ( SVTK_Viewer* vf = dynamic_cast<SVTK_Viewer*>( GetActiveView() ) )
687 //      aTrh->SetSize( 0.5 * vf->GetTrihedronSize() );
688
689     vtkPrs->AddObject( aTrh );
690
691     theActors = vtkActorCollection::New();
692     theActors->AddItem( aTrh );
693   }
694   else
695     theActors = GEOM_AssemblyBuilder::BuildActors( myShape, 0, 0, Standard_True );
696
697   theActors->InitTraversal();
698
699   vtkActor* anActor = (vtkActor*)theActors->GetNextActor();
700
701   vtkProperty* aProp = 0;
702
703   if ( HasColor() || HasWidth() )
704   {
705     aProp = vtkProperty::New();
706     aProp->SetRepresentationToWireframe();
707   }
708
709   if ( HasColor() )
710   {
711     Quantity_Color aColor( (Quantity_NameOfColor)GetColor() );
712     aProp->SetColor( aColor.Red(), aColor.Green(), aColor.Blue() );
713   }
714
715   if ( HasWidth() )
716   {
717     aProp->SetLineWidth( GetWidth() );
718   }
719
720   while ( anActor != NULL )
721   {
722     SALOME_Actor* GActor = SALOME_Actor::SafeDownCast( anActor );
723
724     GActor->setIO( myIO );
725
726     if ( aProp )
727     {
728       GActor->SetProperty( aProp );
729       GActor->SetPreviewProperty( aProp );
730
731       GEOM_Actor* aGeomGActor = GEOM_Actor::SafeDownCast( anActor );
732       if ( aGeomGActor != 0 )
733       {
734         aGeomGActor->SetShadingProperty( aProp );
735         aGeomGActor->SetWireframeProperty( aProp );
736       }
737     }
738
739     if ( myToActivate )
740       GActor->PickableOn();
741     else
742       GActor->PickableOff();
743
744     vtkPrs->AddObject( GActor );
745
746     anActor = (vtkActor*)theActors->GetNextActor();
747   }
748
749   if ( aProp )
750     aProp->Delete();
751
752   theActors->Delete();
753 }
754
755 //=================================================================
756 /*!
757  *  GEOM_Displayer::BuildPrs
758  *  Build presentation accordint to the current viewer type
759  */
760 //=================================================================
761 SALOME_Prs* GEOM_Displayer::BuildPrs( GEOM::GEOM_Object_ptr theObj )
762 {
763   if ( theObj->_is_nil() )
764     return 0;
765
766   myViewFrame = GetActiveView();
767   if ( myViewFrame == 0 )
768     return 0;
769
770   SALOME_Prs* aPrs = myViewFrame->CreatePrs();
771   if ( aPrs == 0 )
772     return 0;
773
774   internalReset();
775   setShape( GEOM_Client().GetShape( GeometryGUI::GetGeomGen(), theObj ) );
776   myType = theObj->GetType();
777
778   // Update presentation
779   UpdatePrs( aPrs );
780
781   return aPrs;
782 }
783
784 //=================================================================
785 /*!
786  *  GEOM_Displayer::BuildPrs
787  *  Build presentation accordint to the current viewer type
788  */
789 //=================================================================
790 SALOME_Prs* GEOM_Displayer::BuildPrs( const TopoDS_Shape& theShape )
791 {
792   myViewFrame = GetActiveView();
793   if ( theShape.IsNull() || myViewFrame == 0 )
794     return 0;
795
796   SALOME_Prs* aPrs = myViewFrame->CreatePrs();
797   if ( aPrs == 0 )
798     return 0;
799
800   internalReset();
801   setShape( theShape );
802   myType = -1;
803
804   UpdatePrs( aPrs );
805
806   return aPrs;
807 }
808
809 //=================================================================
810 /*!
811  *  GEOM_Displayer::buildPresentation
812  *  Builds/finds object's presentation for the current viewer
813  *  Calls corresponding Update() method by means of double dispatch
814  *  [ internal ]
815  */
816 //=================================================================
817 SALOME_Prs* GEOM_Displayer::buildPresentation( const QString& entry,
818                                                SALOME_View* theViewFrame )
819 {
820   SALOME_Prs* prs = 0;
821   internalReset();
822
823   myViewFrame = theViewFrame ? theViewFrame : GetActiveView();
824
825   if ( myViewFrame )
826   {
827     prs = LightApp_Displayer::buildPresentation( entry, theViewFrame );
828     if ( prs )
829     {
830       Handle( SALOME_InteractiveObject ) theIO = new SALOME_InteractiveObject();
831       theIO->setEntry( entry.latin1() );
832       if ( !theIO.IsNull() )
833       {
834         // set interactive object
835         setIO( theIO );
836         //  Find SOBject (because shape should be published previously)
837         SUIT_Session* session = SUIT_Session::session();
838         SUIT_Application* app = session->activeApplication();
839         if ( app )
840         {
841           SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
842           if ( study )
843           {
844             _PTR(SObject) SO ( study->studyDS()->FindObjectID( theIO->getEntry() ) );
845             if ( SO )
846             {
847               // get CORBA reference to data object
848               CORBA::Object_var object = GeometryGUI::ClientSObjectToObject(SO);
849               if ( !CORBA::is_nil( object ) )
850               {
851                 // downcast to GEOM object
852                 GEOM::GEOM_Object_var GeomObject = GEOM::GEOM_Object::_narrow( object );
853                 if ( !GeomObject->_is_nil() )
854                 {
855                   // finally set shape
856                   setShape( GEOM_Client().GetShape( GeometryGUI::GetGeomGen(), GeomObject ) );
857                   myType = GeomObject->GetType();
858                 }
859               }
860             }
861           }
862         }
863       }
864       UpdatePrs( prs );  // Update presentation by using of the double dispatch
865     }
866   }
867   return prs;
868 }
869
870 //=================================================================
871 /*!
872  *  GEOM_Displayer::internalReset
873  *  Resets internal data
874  *  [internal]
875  */
876 //=================================================================
877 void GEOM_Displayer::internalReset()
878 {
879   myIO.Nullify();
880   myShape.Nullify();
881 }
882
883 //=================================================================
884 /*!
885  *  GEOM_Displayer::LocalSelection
886  *  Activate selection of CAD shapes with activisation of selection
887  *  of their sub-shapes (with opened local context for OCC viewer)
888  */
889 //=================================================================
890 void GEOM_Displayer::LocalSelection( const Handle(SALOME_InteractiveObject)& theIO, const int theMode )
891 {
892   SUIT_Session* session = SUIT_Session::session();
893   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
894   if ( !app )
895     return;
896
897   LightApp_SelectionMgr* sm = app->selectionMgr();
898
899   // remove all filters from selection
900   sm->clearFilters();
901
902   SALOME_View* vf = GetActiveView();
903   if ( vf ) {
904     if (!theIO.IsNull() && !vf->isVisible(theIO))
905       Display(theIO);
906     SALOME_Prs* prs = vf->CreatePrs( theIO.IsNull() ? 0 : theIO->getEntry() );
907     vf->LocalSelection( prs, theMode );
908     delete prs;  // delete presentation because displayer is its owner
909   }
910 }
911
912 //=================================================================
913 /*!
914  *  GEOM_Displayer::globalSelection
915  *  Activate selection of CAD shapes without activisation of selection
916  *  of their sub-shapes (without opened local context for OCC viewer)
917  */
918 //=================================================================
919 void GEOM_Displayer::GlobalSelection( const int theMode, const bool update )
920 {
921   TColStd_MapOfInteger aModes;
922   aModes.Add( theMode );
923   GlobalSelection( aModes, update );
924 }
925
926 //=================================================================
927 /*!
928  *  GEOM_Displayer::globalSelection
929  *  Activate selection of CAD shapes without activisation of selection
930  *  of their sub-shapes (without opened local context for OCC viewer)
931  */
932 //=================================================================
933 void GEOM_Displayer::GlobalSelection( const TColStd_MapOfInteger& theModes,
934                                       const bool update )
935 {
936   SUIT_Session* session = SUIT_Session::session();
937   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
938   if ( !app )
939     return;
940
941   SALOME_View* vf = GetActiveView();
942   if ( vf == 0 )
943     return;
944
945   // Close local context
946   vf->GlobalSelection( update );
947
948   // Set selection filters in accordance with current mode
949   LightApp_SelectionMgr* sm = app->selectionMgr();
950   if ( !sm )
951     return;
952
953   // Remove from selection temporary objects if necessary
954   if ( !theModes.Contains( GEOM_PREVIEW ) )
955     clearTemporary( sm );
956
957   //@ aSel->ClearIndex();
958
959   sm->clearFilters();
960
961   // Remove filters from AIS_InteractiveContext
962   Handle(AIS_InteractiveContext) ic;
963   SOCC_Viewer* viewer = dynamic_cast<SOCC_Viewer*>( vf );
964   if ( viewer )
965     {
966       ic = viewer->getAISContext();
967       if ( !ic.IsNull() )
968         ic->RemoveFilters();
969     }
970
971   if ( theModes.Contains( GEOM_ALLOBJECTS ) )
972     return;
973
974   SUIT_SelectionFilter* aFilter;
975   if ( theModes.Extent() == 1 )
976     {
977       int aMode = TColStd_MapIteratorOfMapOfInteger( theModes ).Key();
978       aFilter = getFilter( aMode );
979     }
980   else if ( theModes.Extent() > 1 )
981     {
982       TColStd_MapOfInteger aTopAbsModes;
983       TColStd_MapIteratorOfMapOfInteger anIter( theModes );
984       QPtrList<SUIT_SelectionFilter> aListOfFilters;
985       for ( ; anIter.More(); anIter.Next() )
986         {
987           SUIT_SelectionFilter* aFilter = getFilter( anIter.Key() );
988           if ( aFilter )
989             aListOfFilters.append( aFilter );
990         }
991
992       aFilter = new GEOM_LogicalFilter( aListOfFilters, GEOM_LogicalFilter::LO_OR );
993     }
994   else
995     return;
996
997   if ( aFilter )
998     {
999       sm->installFilter( aFilter );
1000       if ( !ic.IsNull() )
1001         {
1002           Handle(GEOM_OCCFilter) anOCCFilter = new GEOM_OCCFilter( sm );
1003           ic->AddFilter( anOCCFilter );
1004         }
1005     }
1006 }
1007
1008 //=================================================================
1009 /*!
1010  *  GEOM_Displayer::LocalSelection
1011  *  Activate selection of CAD shapes with activisation of selection
1012  *  of their sub-shapes (with opened local context for OCC viewer)
1013  */
1014 //=================================================================
1015 void GEOM_Displayer::LocalSelection( const SALOME_ListIO& theIOList, const int theMode )
1016 {
1017   SALOME_ListIteratorOfListIO Iter( theIOList );
1018   for ( ; Iter.More(); Iter.Next() )
1019     LocalSelection( Iter.Value(), theMode );
1020 }
1021
1022 //=================================================================
1023 /*!
1024  *  GEOM_Displayer::BeforeDisplay
1025  *  Called before displaying of pars. Close local context
1026  *  [ Reimplemented from SALOME_Displayer ]
1027  */
1028 //=================================================================
1029 void GEOM_Displayer::BeforeDisplay( SALOME_View* v, const SALOME_OCCViewType& )
1030 {
1031   SOCC_Viewer* vf = dynamic_cast<SOCC_Viewer*>( v );
1032   if ( vf )
1033   {
1034     Handle(AIS_InteractiveContext) ic = vf->getAISContext();
1035     if ( !ic.IsNull() )
1036     {
1037       if ( ic->HasOpenedContext() )
1038       ic->CloseAllContexts();
1039     }
1040   }
1041 }
1042
1043 void GEOM_Displayer::AfterDisplay( SALOME_View*, const SALOME_OCCViewType& )
1044 {
1045 }
1046
1047
1048 //=================================================================
1049 /*!
1050  *  GEOM_Displayer::SetColor
1051  *  Set color for shape displaying. If it is equal -1 then default color is used.
1052  *  Available values are from Quantity_NameOfColor enumeration
1053  */
1054 //=================================================================
1055 void GEOM_Displayer::SetColor( const int color )
1056 {
1057   myColor = color;
1058 }
1059
1060 int GEOM_Displayer::GetColor() const
1061 {
1062   return myColor;
1063 }
1064
1065 bool GEOM_Displayer::HasColor() const
1066 {
1067   return myColor != -1;
1068 }
1069
1070 void GEOM_Displayer::UnsetColor()
1071 {
1072   myColor = -1;
1073 }
1074
1075 //=================================================================
1076 /*!
1077  *  GEOM_Displayer::SetWidth
1078  *  Set width of shape displaying. If it is equal -1 then default width is used.
1079  */
1080 //=================================================================
1081 void GEOM_Displayer::SetWidth( const double width )
1082 {
1083   myWidth = width;
1084 }
1085
1086 double GEOM_Displayer::GetWidth() const
1087 {
1088   return myWidth;
1089 }
1090
1091 bool GEOM_Displayer::HasWidth() const
1092 {
1093   return myWidth != -1;
1094 }
1095
1096 void GEOM_Displayer::UnsetWidth()
1097 {
1098   myWidth = -1;
1099 }
1100
1101 //=================================================================
1102 /*!
1103  *  GEOM_Displayer::SetToActivate
1104  *  This method is used for activisation/deactivisation of objects to be displayed
1105  */
1106 //=================================================================
1107 void GEOM_Displayer::SetToActivate( const bool toActivate )
1108 {
1109   myToActivate = toActivate;
1110 }
1111 bool GEOM_Displayer::ToActivate() const
1112 {
1113   return myToActivate;
1114 }
1115
1116 //=================================================================
1117 /*!
1118  *  GEOM_Displayer::clearTemporary
1119  *  Removes from selection temporary objects
1120  */
1121 //=================================================================
1122 void GEOM_Displayer::clearTemporary( LightApp_SelectionMgr* theSelMgr )
1123 {
1124   SALOME_ListIO selected, toSelect;
1125   theSelMgr->selectedObjects( selected );
1126
1127   for (  SALOME_ListIteratorOfListIO it( selected ) ; it.More(); it.Next() ) {
1128     Handle(SALOME_InteractiveObject) io = it.Value();
1129     if ( !io.IsNull() && io->hasEntry() && strncmp( io->getEntry(), "TEMP_", 5 ) != 0 )
1130       toSelect.Append( it.Value() );
1131   }
1132
1133   theSelMgr->setSelectedObjects( toSelect, true );
1134 }
1135
1136 void GEOM_Displayer::SetName( const char* theName )
1137 {
1138   myName = theName;
1139 }
1140
1141 void GEOM_Displayer::UnsetName()
1142 {
1143   myName = "";
1144 }
1145
1146 SalomeApp_Study* GEOM_Displayer::getStudy() const
1147 {
1148   return dynamic_cast<SalomeApp_Study*>( myApp->activeStudy() );
1149 }
1150
1151 void GEOM_Displayer::setIO( const Handle(SALOME_InteractiveObject)& theIO )
1152 {
1153   myIO = theIO;
1154 }
1155
1156 void GEOM_Displayer::setShape( const TopoDS_Shape& theShape )
1157 {
1158   myShape = theShape;
1159 }
1160
1161 bool GEOM_Displayer::canBeDisplayed( const QString& /*entry*/, const QString& viewer_type ) const
1162 {
1163   return viewer_type==SOCC_Viewer::Type() || viewer_type==SVTK_Viewer::Type();
1164 }