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