1 // Copyright (C) 2005 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
4 // This library is free software; you can redistribute it and/or
5 // modify it under the terms of the GNU Lesser General Public
6 // License as published by the Free Software Foundation; either
7 // version 2.1 of the License.
9 // This library is distributed in the hope that it will be useful
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 // Lesser General Public License for more details.
14 // You should have received a copy of the GNU Lesser General Public
15 // License along with this library; if not, write to the Free Software
16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 // File : GEOMGUI_OCCSelector.cxx
21 // Author : Alexander SOLOVYOV, Open CASCADE S.A.S. (alexander.solovyov@opencascade.com)
24 #include "GEOMGUI_OCCSelector.h"
26 #include <LightApp_DataSubOwner.h>
28 #include <OCCViewer_ViewModel.h>
30 #include <AIS_ListOfInteractive.hxx>
31 #include <AIS_ListIteratorOfListOfInteractive.hxx>
32 #include <SelectMgr_EntityOwner.hxx>
33 #include <AIS_Shape.hxx>
34 #include <TopTools_IndexedMapOfShape.hxx>
36 #include <SelectMgr_IndexedMapOfOwner.hxx>
37 #include <TColStd_ListIteratorOfListOfInteger.hxx>
38 #include <SelectMgr_Selection.hxx>
39 #include <SelectBasics_SensitiveEntity.hxx>
40 #include <StdSelect_BRepOwner.hxx>
41 #include <TColStd_IndexedMapOfInteger.hxx>
42 #include <NCollection_DataMap.hxx>
44 #include <SUIT_Session.h>
45 #include <SalomeApp_Study.h>
48 //================================================================
49 // Function : GEOMGUI_OCCSelector
51 //================================================================
52 GEOMGUI_OCCSelector::GEOMGUI_OCCSelector( OCCViewer_Viewer* viewer, SUIT_SelectionMgr* mgr )
53 : LightApp_OCCSelector( viewer, mgr )
57 //================================================================
58 // Function : ~GEOMGUI_OCCSelector
60 //================================================================
61 GEOMGUI_OCCSelector::~GEOMGUI_OCCSelector()
65 //================================================================
66 // Function : getSelection
68 //================================================================
69 void GEOMGUI_OCCSelector::getSelection( SUIT_DataOwnerPtrList& aList ) const
71 OCCViewer_Viewer* vw = viewer();
75 Handle(AIS_InteractiveContext) ic = vw->getAISContext();
77 if ( ic->HasOpenedContext() )
79 for ( ic->InitSelected(); ic->MoreSelected(); ic->NextSelected() )
81 Handle(StdSelect_BRepOwner) anOwner = Handle(StdSelect_BRepOwner)::DownCast(ic->SelectedOwner());
82 if ( anOwner.IsNull() )
85 Handle(AIS_InteractiveObject) io = Handle(AIS_InteractiveObject)::DownCast( anOwner->Selectable() );
87 QString entryStr = entry( io );
90 if ( anOwner->ComesFromDecomposition() ) // == Local Selection
92 TopoDS_Shape subShape = anOwner->Shape();
93 Handle(AIS_Shape) aisShape = Handle(AIS_Shape)::DownCast( io );
94 if ( !aisShape.IsNull() )
96 TopoDS_Shape bigShape = aisShape->Shape();
98 TopTools_IndexedMapOfShape subShapes;
99 TopExp::MapShapes( bigShape, subShapes );
100 index = subShapes.FindIndex( subShape );
104 if ( !entryStr.isEmpty() )
106 LightApp_DataOwner* owner;
107 if ( index > -1 ) // Local Selection
108 owner = new LightApp_DataSubOwner( entryStr, index );
109 else // Global Selection
110 owner = new LightApp_DataOwner( entryStr );
112 aList.append( SUIT_DataOwnerPtr( owner ) );
118 for ( ic->InitCurrent(); ic->MoreCurrent(); ic->NextCurrent() )
120 Handle(AIS_InteractiveObject) io = ic->Current();
122 QString entryStr = entry( io );
124 if ( !entryStr.isEmpty() )
126 LightApp_DataOwner* owner = new LightApp_DataOwner( entryStr );
127 aList.append( SUIT_DataOwnerPtr( owner ) );
131 // add externally selected objects
132 SUIT_DataOwnerPtrList::const_iterator anExtIter;
133 for(anExtIter = mySelectedExternals.begin(); anExtIter != mySelectedExternals.end(); anExtIter++) {
134 aList.append(*anExtIter);
139 //================================================================
140 // Function : getEntityOwners
142 //================================================================
143 static void getEntityOwners( const Handle(AIS_InteractiveObject)& theObj,
144 const Handle(AIS_InteractiveContext)& theIC,
145 SelectMgr_IndexedMapOfOwner& theMap )
147 if ( theObj.IsNull() || theIC.IsNull() )
150 TColStd_ListOfInteger modes;
151 theIC->ActivatedModes( theObj, modes );
153 TColStd_ListIteratorOfListOfInteger itr( modes );
154 for (; itr.More(); itr.Next() ) {
156 if ( !theObj->HasSelection( m ) )
159 Handle(SelectMgr_Selection) sel = theObj->Selection( m );
161 for ( sel->Init(); sel->More(); sel->Next() ) {
162 Handle(SelectBasics_SensitiveEntity) entity = sel->Sensitive();
163 if ( entity.IsNull() )
166 Handle(SelectMgr_EntityOwner) owner =
167 Handle(SelectMgr_EntityOwner)::DownCast(entity->OwnerId());
168 if ( !owner.IsNull() )
174 //================================================================
175 // Function : setSelection
177 //================================================================
178 void GEOMGUI_OCCSelector::setSelection( const SUIT_DataOwnerPtrList& aList )
180 OCCViewer_Viewer* vw = viewer();
184 Handle(AIS_InteractiveContext) ic = vw->getAISContext();
186 NCollection_DataMap<TCollection_AsciiString, TColStd_IndexedMapOfInteger> indexesMap; // "entry - list_of_int" map for LOCAL selection
187 QMap<QString,int> globalSelMap; // only Key=entry from this map is used. value(int) is NOT used at all.
188 SelectMgr_IndexedMapOfOwner ownersmap; // map of owners to be selected
190 AIS_ListOfInteractive aDispList;
191 ic->DisplayedObjects( aDispList );
193 // build a map of data owner indexes to be selected.
194 // "entry - to - list_of_ids" map
195 for ( SUIT_DataOwnerPtrList::const_iterator itr = aList.begin(); itr != aList.end(); ++itr )
197 const LightApp_DataSubOwner* subOwner = dynamic_cast<const LightApp_DataSubOwner*>( (*itr).operator->() );
200 QString entry = subOwner->entry();
202 if ( indexesMap.IsBound( TCollection_AsciiString(entry.toLatin1().data())))
204 if ( indexesMap.IsBound( (char*)entry.toLatin1()))
207 TColStd_IndexedMapOfInteger& subIndexes = indexesMap.ChangeFind(entry.toLatin1().data());
208 subIndexes.Add( subOwner->index() );
209 //indexesMap.replace( entry, subIndexes );
213 TColStd_IndexedMapOfInteger subIndexes;
214 subIndexes.Add( subOwner->index() );
215 indexesMap.Bind(entry.toLatin1().data(), subIndexes);
218 else // the owner is NOT a sub owner, maybe it is a DataOwner == GLOBAL selection
220 const LightApp_DataOwner* owner = dynamic_cast<const LightApp_DataOwner*>( (*itr).operator->() );
223 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
224 QString anEntry = appStudy->referencedToEntry( owner->entry() );
226 globalSelMap[anEntry] = 1;
231 // get all owners. Fill "entry - list_of_owners" map.
232 for ( AIS_ListIteratorOfListOfInteractive it( aDispList ); it.More(); it.Next() )
234 Handle(AIS_InteractiveObject) io = it.Value();
235 QString entryStr = entry( io );
236 if ( !entryStr.isEmpty() )
238 //EntryToListOfOwnerMap entryOwnersMap; // "entry - list_of_owners" map. temporary use.
239 SelectMgr_IndexedMapOfOwner owners;
240 getEntityOwners( io, ic, owners ); // get all owners
242 for ( int i = 1, n = owners.Extent(); i <= n; i++ )
245 Handle(StdSelect_BRepOwner) anOwner = Handle(StdSelect_BRepOwner)::DownCast(owners( i ));
247 if ( anOwner.IsNull() || !anOwner->HasShape() )
251 if ( !anOwner->ComesFromDecomposition() && globalSelMap.contains( entryStr ) )
253 ownersmap.Add( anOwner );
254 globalSelMap[entryStr]++;
259 Handle(AIS_Shape) aisShape = Handle(AIS_Shape)::DownCast( io );
261 if ( !aisShape.IsNull() && indexesMap.IsBound( entryStr.toLatin1().data() ) )
263 TopoDS_Shape shape = aisShape->Shape();
264 TopTools_IndexedMapOfShape aMapOfShapes;
265 TopExp::MapShapes( shape, aMapOfShapes );
266 const TColStd_IndexedMapOfInteger& subIndexes = indexesMap.ChangeFind(entryStr.toLatin1().data());
268 const TopoDS_Shape& aSubShape = anOwner->Shape();
269 int aSubShapeId = aMapOfShapes.FindIndex( aSubShape );
271 // check if the "sub_shape_index" is found in the "map of indexes for this entry",
272 // which was passes in the parameter
273 if ( subIndexes.Contains( aSubShapeId ) )
275 ownersmap.Add( anOwner );
278 } // end of local selection
279 } // end of for(owners)
281 }// end of for(AIS_all_ios)
283 vw->unHighlightAll( false );
286 for ( int i = 1, n = ownersmap.Extent(); i <= n; i++ )
288 Handle(SelectMgr_EntityOwner) owner = ownersmap( i );
289 if ( owner->State() )
292 if ( ic->HasOpenedContext() )
293 ic->AddOrRemoveSelected( owner, false );
295 ic->AddOrRemoveSelected( Handle(AIS_InteractiveObject)::DownCast(owner->Selectable()), false );
300 // fill extra selected
301 mySelectedExternals.clear();
302 for ( SUIT_DataOwnerPtrList::const_iterator itr2 = aList.begin(); itr2 != aList.end(); ++itr2 ) {
303 const LightApp_DataSubOwner* subOwner = dynamic_cast<const LightApp_DataSubOwner*>( (*itr2).operator->() );
306 const LightApp_DataOwner* owner = dynamic_cast<const LightApp_DataOwner*>( (*itr2).operator->() );
309 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
310 QString anEntry = appStudy->referencedToEntry( owner->entry() );
311 if (globalSelMap[anEntry] == 1) mySelectedExternals.append(*itr2);