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