Salome HOME
Added a check on proximity between last point and current point in 3D sketch construc...
[modules/geom.git] / src / GEOMGUI / GEOMGUI_OCCSelector.cxx
index c0868ae60170cbb3c8631cf473d7cda25d566bdb..e2008bfa88b8beae8828fba4b73a45e8a681bcab 100644 (file)
@@ -1,21 +1,27 @@
-// Copyright (C) 2005  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+// 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
-// 
+//
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
-// License as published by the Free Software Foundation; either 
+// License as published by the Free Software Foundation; either
 // version 2.1 of the License.
-// 
-// This library is distributed in the hope that it will be useful 
-// but WITHOUT ANY WARRANTY; without even the implied warranty of 
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 // Lesser General Public License for more details.
 //
-// You should have received a copy of the GNU Lesser General Public  
-// License along with this library; if not, write to the Free Software 
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
 //
-// See http://www.salome-platform.org/
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+
+// File   : GEOMGUI_OCCSelector.cxx
+// Author : Alexander SOLOVYOV, Open CASCADE S.A.S. (alexander.solovyov@opencascade.com)
 //
 #include "GEOMGUI_OCCSelector.h"
 
 
 #include <OCCViewer_ViewModel.h>
 
+#include <SUIT_Session.h>
+#include <SalomeApp_Study.h>
 #include <SALOME_InteractiveObject.hxx>
 
-#include <AIS_ListOfInteractive.hxx>
-#include <AIS_ListIteratorOfListOfInteractive.hxx>
+#include <StdSelect_BRepOwner.hxx>
+#include <SelectMgr_Selection.hxx>
 #include <SelectMgr_EntityOwner.hxx>
+#include <SelectMgr_IndexedMapOfOwner.hxx>
+#include <SelectBasics_SensitiveEntity.hxx>
+
 #include <AIS_Shape.hxx>
-#include <TopTools_IndexedMapOfShape.hxx>
+#include <AIS_ListOfInteractive.hxx>
+#include <AIS_ListIteratorOfListOfInteractive.hxx>
 #include <TopExp.hxx>
-#include <SelectMgr_IndexedMapOfOwner.hxx>
+#include <TopTools_IndexedMapOfShape.hxx>
 #include <TColStd_ListIteratorOfListOfInteger.hxx>
-#include <SelectMgr_Selection.hxx>
-#include <SelectBasics_SensitiveEntity.hxx>
 #include <TColStd_IndexedMapOfInteger.hxx>
-#include <SelectMgr_IndexedMapOfOwner.hxx>
 #include <NCollection_DataMap.hxx>
 
 
 //================================================================
 // Function : GEOMGUI_OCCSelector
-// Purpose  : 
+// Purpose  :
 //================================================================
 GEOMGUI_OCCSelector::GEOMGUI_OCCSelector( OCCViewer_Viewer* viewer, SUIT_SelectionMgr* mgr )
 : LightApp_OCCSelector( viewer, mgr )
@@ -51,7 +60,7 @@ GEOMGUI_OCCSelector::GEOMGUI_OCCSelector( OCCViewer_Viewer* viewer, SUIT_Selecti
 
 //================================================================
 // Function : ~GEOMGUI_OCCSelector
-// Purpose  : 
+// Purpose  :
 //================================================================
 GEOMGUI_OCCSelector::~GEOMGUI_OCCSelector()
 {
@@ -59,79 +68,97 @@ GEOMGUI_OCCSelector::~GEOMGUI_OCCSelector()
 
 //================================================================
 // Function : getSelection
-// Purpose  : 
+// Purpose  :
 //================================================================
 void GEOMGUI_OCCSelector::getSelection( SUIT_DataOwnerPtrList& aList ) const
 {
   OCCViewer_Viewer* vw = viewer();
-  if ( !vw )
+  if (!vw)
     return;
 
   Handle(AIS_InteractiveContext) ic = vw->getAISContext();
-  
-  if ( ic->HasOpenedContext() )
+
+  if (ic->HasOpenedContext())
+  {
+    TopoDS_Shape curBigShape;
+    TopTools_IndexedMapOfShape subShapes;
+
+    for (ic->InitSelected(); ic->MoreSelected(); ic->NextSelected())
     {
-      for ( ic->InitSelected(); ic->MoreSelected(); ic->NextSelected() )
-       {
-         Handle(SelectMgr_EntityOwner) anOwner = ic->SelectedOwner();
-         if ( anOwner.IsNull() )
-           continue;
-         
-         Handle(AIS_InteractiveObject) io = Handle(AIS_InteractiveObject)::DownCast( anOwner->Selectable() );
-         
-         QString entryStr = entry( io );
-         int index = -1; 
-         
-         if ( anOwner->ComesFromDecomposition() ) // == Local Selection
-           {
-             TopoDS_Shape subShape = anOwner->Shape();
-             Handle(AIS_Shape) aisShape = Handle(AIS_Shape)::DownCast( io );
-             if ( !aisShape.IsNull() )
-               {
-                 TopoDS_Shape bigShape = aisShape->Shape();
-                 
-                 TopTools_IndexedMapOfShape subShapes;
-                 TopExp::MapShapes( bigShape, subShapes );
-                 index = subShapes.FindIndex( subShape );
-               }
-           }
-         
-         if ( !entryStr.isEmpty() )
-           {
-             LightApp_DataOwner* owner;
-             if ( index > -1 ) // Local Selection
-               owner = new LightApp_DataSubOwner( entryStr, index );
-             else // Global Selection
-               owner = new LightApp_DataOwner( entryStr );
-
-             aList.append( SUIT_DataOwnerPtr( owner ) );
-           }
-       }
+      Handle(StdSelect_BRepOwner) anOwner = Handle(StdSelect_BRepOwner)::DownCast(ic->SelectedOwner());
+      if (anOwner.IsNull())
+        continue;
+
+      Handle(AIS_InteractiveObject) io = Handle(AIS_InteractiveObject)::DownCast(anOwner->Selectable());
+
+      QString entryStr = entry(io);
+      int index = -1;
+
+      if (anOwner->ComesFromDecomposition()) // == Local Selection
+      {
+        TopoDS_Shape subShape = anOwner->Shape();
+        Handle(AIS_Shape) aisShape = Handle(AIS_Shape)::DownCast(io);
+        if (!aisShape.IsNull())
+        {
+          TopoDS_Shape bigShape = aisShape->Shape();
+
+          if (!bigShape.IsEqual(curBigShape))
+          {
+            curBigShape = bigShape;
+            subShapes.Clear();
+            TopExp::MapShapes(bigShape, subShapes);
+          }
+          index = subShapes.FindIndex(subShape);
+        }
+      }
+
+      if (!entryStr.isEmpty())
+      {
+        Handle(SALOME_InteractiveObject) anIO = Handle(SALOME_InteractiveObject)::DownCast(io->GetOwner()); 
+        LightApp_DataOwner* owner;
+        if (index > -1) // Local Selection
+          owner = new LightApp_DataSubOwner (entryStr, index);
+        else if ( !anIO.IsNull() ) // Global Selection
+          owner = new LightApp_DataOwner( anIO );
+
+        aList.append(SUIT_DataOwnerPtr(owner));
+      }
     }
+  }
   else
+  {
+    for (ic->InitCurrent(); ic->MoreCurrent(); ic->NextCurrent())
     {
-      for ( ic->InitCurrent(); ic->MoreCurrent(); ic->NextCurrent() )
-       {
-         Handle(AIS_InteractiveObject) io = ic->Current();
-         
-         QString entryStr = entry( io );
-         
-         if ( !entryStr.isEmpty() )
-           {
-             LightApp_DataOwner* owner = new LightApp_DataOwner( entryStr );
-             aList.append( SUIT_DataOwnerPtr( owner ) );
-           }
-       }
+      Handle(AIS_InteractiveObject) io = ic->Current();
+
+      QString entryStr = entry( io );
+
+      if ( !entryStr.isEmpty() )
+      {
+        Handle(SALOME_InteractiveObject) anIO = Handle(SALOME_InteractiveObject)::DownCast(io->GetOwner()); 
+        if ( !anIO.IsNull() ) {
+          LightApp_DataOwner* owner = new LightApp_DataOwner( anIO );
+          aList.append( SUIT_DataOwnerPtr( owner ) );
+        }
+      }
     }
+  }
+
+  // add externally selected objects
+  SUIT_DataOwnerPtrList::const_iterator anExtIter;
+  for (anExtIter = mySelectedExternals.begin(); anExtIter != mySelectedExternals.end(); anExtIter++)
+  {
+    aList.append(*anExtIter);
+  }
 }
 
 //================================================================
 // Function : getEntityOwners
-// Purpose  : 
+// Purpose  :
 //================================================================
 static void getEntityOwners( const Handle(AIS_InteractiveObject)& theObj,
-                            const Handle(AIS_InteractiveContext)& theIC,
-                            SelectMgr_IndexedMapOfOwner& theMap )
+                             const Handle(AIS_InteractiveContext)& theIC,
+                             SelectMgr_IndexedMapOfOwner& theMap )
 {
   if ( theObj.IsNull() || theIC.IsNull() )
     return;
@@ -150,19 +177,19 @@ static void getEntityOwners( const Handle(AIS_InteractiveObject)& theObj,
     for ( sel->Init(); sel->More(); sel->Next() ) {
       Handle(SelectBasics_SensitiveEntity) entity = sel->Sensitive();
       if ( entity.IsNull() )
-       continue;
+        continue;
 
       Handle(SelectMgr_EntityOwner) owner =
-       Handle(SelectMgr_EntityOwner)::DownCast(entity->OwnerId());
+        Handle(SelectMgr_EntityOwner)::DownCast(entity->OwnerId());
       if ( !owner.IsNull() )
-       theMap.Add( owner );
+        theMap.Add( owner );
     }
   }
 }
 
 //================================================================
 // Function : setSelection
-// Purpose  : 
+// Purpose  :
 //================================================================
 void GEOMGUI_OCCSelector::setSelection( const SUIT_DataOwnerPtrList& aList )
 {
@@ -172,14 +199,12 @@ void GEOMGUI_OCCSelector::setSelection( const SUIT_DataOwnerPtrList& aList )
 
   Handle(AIS_InteractiveContext) ic = vw->getAISContext();
 
-#ifndef WNT
-  NCollection_DataMap<TCollection_AsciiString, TColStd_IndexedMapOfInteger> indexesMap; // "entry - list_of_int" map for LOCAL selection
-#else
-  NCollection_DataMap<Standard_CString, TColStd_IndexedMapOfInteger> indexesMap; // "entry - list_of_int" map for LOCAL selection
-#endif
+  // "entry - list_of_int" map for LOCAL selection
+  NCollection_DataMap<TCollection_AsciiString, TColStd_IndexedMapOfInteger> indexesMap;
+
   QMap<QString,int> globalSelMap; // only Key=entry from this map is used.  value(int) is NOT used at all.
   SelectMgr_IndexedMapOfOwner ownersmap; // map of owners to be selected
-  
+
   AIS_ListOfInteractive aDispList;
   ic->DisplayedObjects( aDispList );
 
@@ -191,29 +216,29 @@ void GEOMGUI_OCCSelector::setSelection( const SUIT_DataOwnerPtrList& aList )
     if ( subOwner )
     {
       QString entry = subOwner->entry();
-#ifndef WNT
-      if ( indexesMap.IsBound( TCollection_AsciiString((char*)entry.latin1())))
-#else
-         if ( indexesMap.IsBound( (char*)entry.latin1()))
-#endif
+      if ( indexesMap.IsBound( TCollection_AsciiString(entry.toLatin1().data())))
       {
-       TColStd_IndexedMapOfInteger& subIndexes = indexesMap.ChangeFind((char*)entry.latin1());
-       subIndexes.Add( subOwner->index() );
-       //indexesMap.replace( entry, subIndexes );
+        TColStd_IndexedMapOfInteger& subIndexes = indexesMap.ChangeFind(entry.toLatin1().data());
+        subIndexes.Add( subOwner->index() );
+        //indexesMap.replace( entry, subIndexes );
       }
       else
       {
-       TColStd_IndexedMapOfInteger subIndexes;
-       subIndexes.Add( subOwner->index() );
-       indexesMap.Bind((char*)entry.latin1(), subIndexes);
+        TColStd_IndexedMapOfInteger subIndexes;
+        subIndexes.Add( subOwner->index() );
+        indexesMap.Bind(entry.toLatin1().data(), subIndexes);
       }
-    } 
+    }
     else // the owner is NOT a sub owner, maybe it is a DataOwner == GLOBAL selection
     {
       const LightApp_DataOwner* owner = dynamic_cast<const LightApp_DataOwner*>( (*itr).operator->() );
       if ( owner )
       {
-       globalSelMap[owner->entry()] = 1;
+        SalomeApp_Study* appStudy =
+          dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
+        QString anEntry = appStudy->referencedToEntry( owner->entry() );
+
+        globalSelMap[anEntry] = 1;
       }
     }
   }
@@ -229,48 +254,77 @@ void GEOMGUI_OCCSelector::setSelection( const SUIT_DataOwnerPtrList& aList )
       SelectMgr_IndexedMapOfOwner owners;
       getEntityOwners( io, ic, owners ); // get all owners
 
-      for  ( int i = 1, n = owners.Extent(); i <= n; i++ ) 
+      int i, n = owners.Extent();
+
+      // 1. Prepare map of shapes for local selection
+      TopTools_IndexedMapOfShape aMapOfShapes;
+      bool isLocal = false;
+
+      Handle(StdSelect_BRepOwner) anOwner;
+      for (i = 1; i <= n && !isLocal; i++)
+      {
+        anOwner = Handle(StdSelect_BRepOwner)::DownCast(owners( i ));
+        if (!anOwner.IsNull() && anOwner->HasShape())
+        {
+          if (anOwner->ComesFromDecomposition() || !globalSelMap.contains(entryStr))
+          {
+            // has a local selection
+            Handle(AIS_Shape) aisShape = Handle(AIS_Shape)::DownCast( io );
+            if (!aisShape.IsNull() && indexesMap.IsBound(entryStr.toLatin1().data()))
+            {
+              isLocal = true;
+              TopoDS_Shape shape = aisShape->Shape();
+              aMapOfShapes.Clear();
+              TopExp::MapShapes(shape, aMapOfShapes);
+            }
+          }
+        }
+      }
+
+      // 2. Process all owners
+      for (i = 1; i <= n; i++)
       {
-       Handle(SelectMgr_EntityOwner) anOwner = owners( i );
-       if ( anOwner.IsNull() || !anOwner->HasShape() )
-         continue;
-
-       // GLOBAL selection
-       if ( !anOwner->ComesFromDecomposition() && globalSelMap.contains( entryStr ) ) 
-       {
-         ownersmap.Add( anOwner );
-       }
-       // LOCAL selection
-       else
-       {
-         Handle(AIS_Shape) aisShape = Handle(AIS_Shape)::DownCast( io );
-
-         if ( !aisShape.IsNull() && indexesMap.IsBound( (char*)entryStr.latin1() ) )
-         {
-           TopoDS_Shape shape = aisShape->Shape();
-           TopTools_IndexedMapOfShape aMapOfShapes;
-           TopExp::MapShapes( shape, aMapOfShapes );
-           const TColStd_IndexedMapOfInteger& subIndexes = indexesMap.ChangeFind((char*)entryStr.latin1());
-
-           const TopoDS_Shape& aSubShape = anOwner->Shape();
-           int  aSubShapeId = aMapOfShapes.FindIndex( aSubShape );
-
-           // check if the "sub_shape_index" is found in the "map of indexes for this entry",
-           // which was passes in the parameter
-           if ( subIndexes.Contains( aSubShapeId ) )
-           {
-             ownersmap.Add( anOwner );
-           }
-         }
-        } // end of local selection
+        anOwner = Handle(StdSelect_BRepOwner)::DownCast(owners( i ));
+
+        if ( anOwner.IsNull() || !anOwner->HasShape() )
+          continue;
+
+        // GLOBAL selection
+        if ( !anOwner->ComesFromDecomposition() && globalSelMap.contains( entryStr ) )
+        {
+          ownersmap.Add( anOwner );
+          globalSelMap[entryStr]++;
+        }
+        // LOCAL selection
+        else
+        {
+          if (isLocal)
+          {
+            const TColStd_IndexedMapOfInteger& subIndexes =
+              indexesMap.ChangeFind(entryStr.toLatin1().data());
+
+            const TopoDS_Shape& aSubShape = anOwner->Shape();
+            int aSubShapeId = aMapOfShapes.FindIndex( aSubShape );
+
+            // check if the "sub_shape_index" is found in the "map of indexes for this entry",
+            // which was passes in the parameter
+            if ( subIndexes.Contains( aSubShapeId ) )
+            {
+              ownersmap.Add( anOwner );
+            }
+          }
+        } // end of LOCAL selection
       } // end of for(owners)
-    }// end of if(entry)
-  }// end of for(AIS_all_ios)
+    } // end of if(entry)
+  } // end of for(AIS_all_ios)
 
   vw->unHighlightAll( false );
 
   // DO the selection
-  for  ( int i = 1, n = ownersmap.Extent(); i <= n; i++ ) 
+  int i = 1, n = ownersmap.Extent();
+  bool isAutoHilight = ic->AutomaticHilight();
+  ic->SetAutomaticHilight(Standard_False); //Bug 17269: for better performance
+  for  (; i <= n; i++)
   {
     Handle(SelectMgr_EntityOwner) owner = ownersmap( i );
     if ( owner->State() )
@@ -281,6 +335,28 @@ void GEOMGUI_OCCSelector::setSelection( const SUIT_DataOwnerPtrList& aList )
     else
       ic->AddOrRemoveSelected( Handle(AIS_InteractiveObject)::DownCast(owner->Selectable()), false );
   }
+  ic->SetAutomaticHilight(isAutoHilight); //Bug 17269: restore mode
+  if (n < 3000)
+    ic->HilightSelected(/*updateviewer*/Standard_True);
+  else
+    vw->update();
 
-  vw->update();
+  // fill extra selected
+  mySelectedExternals.clear();
+  for ( SUIT_DataOwnerPtrList::const_iterator itr2 = aList.begin(); itr2 != aList.end(); ++itr2 )
+  {
+    const LightApp_DataSubOwner* subOwner =
+      dynamic_cast<const LightApp_DataSubOwner*>( (*itr2).operator->() );
+    if ( !subOwner )
+    {
+      const LightApp_DataOwner* owner = dynamic_cast<const LightApp_DataOwner*>( (*itr2).operator->() );
+      if ( owner )
+      {
+        SalomeApp_Study* appStudy =
+          dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
+        QString anEntry = appStudy->referencedToEntry( owner->entry() );
+        if (globalSelMap[anEntry] == 1) mySelectedExternals.append(*itr2);
+      }
+    }
+  }
 }