Salome HOME
EDF 2153 : Translation
[modules/geom.git] / src / GEOMGUI / GEOMGUI_Selection.cxx
index d7977004b83faaff7ab2e8806fc752856843b98a..0aa1da20ae232d6838bdd125d459acbe79f4c308 100644 (file)
@@ -1,4 +1,4 @@
-// Copyright (C) 2007-2011  CEA/DEN, EDF R&D, OPEN CASCADE
+// Copyright (C) 2007-2012  CEA/DEN, EDF R&D, OPEN CASCADE
 //
 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
 //
 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
-//
 
 // File   : GEOMGUI_Selection.cxx
 // Author : Alexander SOLOVYOV, Open CASCADE S.A.S. (alexander.solovyov@opencascade.com)
-//
+
 #include "GEOMGUI_Selection.h"
 
 #include "GeometryGUI.h"
 #include "GEOM_Displayer.h"
 
+#include "Material_Model.h"
+
 #include <GEOM_Constants.h>
 
 #include <SalomeApp_Application.h>
 #include <GEOMImpl_Types.hxx>
 
 #include <GEOM_AISShape.hxx>
+#include <GEOM_VTKPropertyMaterial.hxx>
 
 // OCCT Includes
 #include <AIS.hxx>
 #include <AIS_InteractiveObject.hxx>
 #include <AIS_ListOfInteractive.hxx>
+#include <AIS_GraphicTool.hxx>
+#include <AIS_Drawer.hxx>
+#include <Aspect_TypeOfFacingModel.hxx>
+#include <Prs3d_ShadingAspect.hxx>
+#include<Graphic3d_MaterialAspect.hxx>
 
 // VTK Includes
 #include <vtkActorCollection.h>
@@ -151,14 +158,18 @@ QVariant GEOMGUI_Selection::parameter( const int idx, const QString& p ) const
     v = isVectorsMode( idx );
   else if ( p == "topLevel" )
     v = topLevel( idx );
-  else if ( p == "hasHiddenChildren" )
-    v = hasHiddenChildren( idx );
-  else if ( p == "hasShownChildren" )
-    v = hasShownChildren( idx );
+  else if ( p == "hasChildren" )
+    v = hasChildren( idx );
+  else if ( p == "hasConcealedChildren" )
+    v = hasConcealedChildren( idx );
+  else if ( p == "hasDisclosedChildren" )
+    v = hasDisclosedChildren( idx );
   else if ( p == "compoundOfVertices" )
     v = compoundOfVertices( idx );
   else if ( p == "imported" )
     v = isImported( idx );
+  else if ( p == "isPhysicalMaterial" )
+    v = isPhysicalMaterial(idx);
   else
     v = LightApp_Selection::parameter( idx, p );
 
@@ -301,7 +312,13 @@ QString GEOMGUI_Selection::displayMode( const int index ) const
          if ( lst.Extent() ) {
            Handle(AIS_InteractiveObject) io = lst.First();
            if ( !io.IsNull() ) {
-             int dm = io->DisplayMode();
+        int dm;
+        Handle(GEOM_AISShape) aSh = Handle(GEOM_AISShape)::DownCast(io);
+        if(!aSh.IsNull()) {
+          dm = aSh->isTopLevel() ? aSh->prevDisplayMode() : aSh->DisplayMode();
+        } else {
+          dm = io->DisplayMode();
+        }
              OCC_DISPLAY_MODE_TO_STRING( res, dm );
              if ( res.isEmpty() ) { // return default display mode of AIS_InteractiveContext
                OCCViewer_Viewer* occViewer = (OCCViewer_Viewer*)SUIT_Session::session()->activeApplication()->
@@ -313,7 +330,7 @@ QString GEOMGUI_Selection::displayMode( const int index ) const
            }
          }
        }
-       else if ( viewType == SVTK_Viewer::Type() ) { // assuming VTK
+  else if ( viewType == SVTK_Viewer::Type() ) { // assuming VTK
          SVTK_Prs* vtkPrs = dynamic_cast<SVTK_Prs*>( prs );
          vtkActorCollection* lst = vtkPrs ? vtkPrs->GetObjects() : 0;
          if ( lst ) {
@@ -430,7 +447,23 @@ bool GEOMGUI_Selection::isCompoundOfVertices( GEOM::GEOM_Object_ptr obj )
   return ret;
 }
 
-bool GEOMGUI_Selection::hasHiddenChildren( const int index ) const
+bool GEOMGUI_Selection::hasChildren( const int index ) const
+{
+  bool ok = false;
+  SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
+
+  if ( appStudy ) {
+    QString anEntry = entry( index );
+    _PTR(Study) study = appStudy->studyDS();
+    if ( study && !anEntry.isEmpty() ) {
+      _PTR(SObject) aSO( study->FindObjectID( anEntry.toStdString() ) );
+      ok = hasChildren( aSO );
+    }
+  }
+  return ok;
+}
+
+bool GEOMGUI_Selection::hasConcealedChildren( const int index ) const
 {
   bool OK = false;
   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
@@ -446,7 +479,7 @@ bool GEOMGUI_Selection::hasHiddenChildren( const int index ) const
   return OK;
 }
 
-bool GEOMGUI_Selection::hasShownChildren( const int index ) const
+bool GEOMGUI_Selection::hasDisclosedChildren( const int index ) const
 {
   bool OK = false;
   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
@@ -517,7 +550,7 @@ QString GEOMGUI_Selection::selectionMode() const
 
 bool GEOMGUI_Selection::topLevel( const int index ) const {
   bool res = false;
-  
+
 #ifdef USE_VISUAL_PROP_MAP
   bool found = false;
   QVariant v = visibleProperty( entry( index ), TOP_LEVEL_PROP );
@@ -551,3 +584,58 @@ bool GEOMGUI_Selection::topLevel( const int index ) const {
   }
   return res;
 }
+
+bool GEOMGUI_Selection::isPhysicalMaterial( const int idx ) const{
+   bool res = false;
+
+#ifdef USE_VISUAL_PROP_MAP
+   bool found = false;
+   QVariant v = visibleProperty( entry( idx ), MATERIAL_PROP );
+   if ( v.canConvert<QString>() ) {
+     Material_Model material;
+     material.fromProperties( v.toString() );
+     res = material.isPhysical();
+     found = true;
+   }
+
+   if ( !found ) {
+#endif
+     SALOME_View* view = GEOM_Displayer::GetActiveView();
+     QString viewType = activeViewType();
+     if ( view ) {
+       SALOME_Prs* prs = view->CreatePrs( entry( idx ).toLatin1().constData() );
+       if ( prs ) {
+         if ( viewType == OCCViewer_Viewer::Type() ) { // assuming OCC
+           SOCC_Prs* occPrs = (SOCC_Prs*) prs;
+           AIS_ListOfInteractive lst;
+           occPrs->GetObjects( lst );
+           if ( lst.Extent() ) {
+             Handle(AIS_InteractiveObject) io = lst.First();
+             if ( !io.IsNull() ) {
+               Handle(GEOM_AISShape) aSh = Handle(GEOM_AISShape)::DownCast(io);
+               if ( !aSh.IsNull() )
+                 res = (bool) aSh->Attributes()->ShadingAspect()->
+                        Material(Aspect_TOFM_BOTH_SIDE).MaterialType( Graphic3d_MATERIAL_PHYSIC );
+             }
+           }
+         }
+         else if ( viewType == SVTK_Viewer::Type() ) { // assuming VTK
+               SVTK_Prs* vtkPrs = dynamic_cast<SVTK_Prs*>( prs );
+               vtkActorCollection* lst = vtkPrs ? vtkPrs->GetObjects() : 0;
+               if ( lst ) {
+                 lst->InitTraversal();
+                 vtkActor* actor = lst->GetNextActor();
+                 if ( actor ) {
+              GEOM_Actor* aGeomGActor = GEOM_Actor::SafeDownCast( actor );
+                   if ( aGeomGActor ) {
+                GEOM_VTKPropertyMaterial* mat  = GEOM_VTKPropertyMaterial::SafeDownCast(aGeomGActor->GetProperty());
+                res = mat->GetPhysical();
+                   } // if ( salome actor )
+                 } // if ( actor )
+          } // if ( lst == vtkPrs->GetObjects() )
+         }
+       }
+     }
+   }
+   return res;
+}