0030726: Visualization - change Selected state of owner in AIS_Selection
[occt.git] / src / AIS / AIS_Selection.cxx
1 // Copyright (c) 1998-1999 Matra Datavision
2 // Copyright (c) 1999-2014 OPEN CASCADE SAS
3 //
4 // This file is part of Open CASCADE Technology software library.
5 //
6 // This library is free software; you can redistribute it and/or modify it under
7 // the terms of the GNU Lesser General Public License version 2.1 as published
8 // by the Free Software Foundation, with special exception defined in the file
9 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
10 // distribution for complete text of the license and disclaimer of any warranty.
11 //
12 // Alternatively, this file may be used under the terms of Open CASCADE
13 // commercial license or contractual agreement.
14
15 #include <AIS_Selection.hxx>
16
17 #include <AIS_InteractiveObject.hxx>
18
19 IMPLEMENT_STANDARD_RTTIEXT(AIS_Selection, Standard_Transient)
20
21 namespace
22 {
23   static const Standard_Integer THE_MaxSizeOfResult = 100000;
24 }
25
26 //=======================================================================
27 //function : AIS_Selection
28 //purpose  :
29 //=======================================================================
30 AIS_Selection::AIS_Selection()
31 {
32   // for maximum performance on medium selections (< 100000 objects)
33   myResultMap.ReSize (THE_MaxSizeOfResult);
34 }
35
36 //=======================================================================
37 //function : Clear
38 //purpose  :
39 //=======================================================================
40 void AIS_Selection::Clear()
41 {
42   for (AIS_NListOfEntityOwner::Iterator aSelIter (Objects()); aSelIter.More(); aSelIter.Next())
43   {
44     const Handle(SelectMgr_EntityOwner) anObject = aSelIter.Value();
45     anObject->SetSelected (Standard_False);
46   }
47   myresult.Clear();
48   myResultMap.Clear();
49   myIterator = AIS_NListOfEntityOwner::Iterator();
50 }
51
52 //=======================================================================
53 //function : Select
54 //purpose  :
55 //=======================================================================
56 AIS_SelectStatus AIS_Selection::Select (const Handle(SelectMgr_EntityOwner)& theObject)
57 {
58   if (theObject.IsNull()
59   || !theObject->HasSelectable())
60   {
61     return AIS_SS_NotDone;
62   }
63
64   if (!myResultMap.IsBound (theObject))
65   {
66     AIS_NListOfEntityOwner::Iterator aListIter;
67     myresult.Append  (theObject, aListIter);
68     myResultMap.Bind (theObject, aListIter);
69     theObject->SetSelected (Standard_True);
70     return AIS_SS_Added;
71   }
72
73   AIS_NListOfEntityOwner::Iterator aListIter = myResultMap.Find (theObject);
74   if (myIterator == aListIter)
75   {
76     if (myIterator.More())
77     {
78       myIterator.Next();
79     }
80     else
81     {
82       myIterator = AIS_NListOfEntityOwner::Iterator();
83     }
84   }
85
86   // In the mode of advanced mesh selection only one owner is created for all selection modes.
87   // It is necessary to check the current detected entity
88   // and remove the owner from map only if the detected entity is the same as previous selected (IsForcedHilight call)
89   if (theObject->IsForcedHilight())
90   {
91     return AIS_SS_Added;
92   }
93
94   myresult.Remove (aListIter);
95   myResultMap.UnBind (theObject);
96   theObject->SetSelected (Standard_False);
97
98   // update list iterator for next object in <myresult> list if any
99   if (aListIter.More())
100   {
101     const Handle(SelectMgr_EntityOwner)& aNextObject = aListIter.Value();
102     if (myResultMap.IsBound (aNextObject))
103     {
104       myResultMap (aNextObject) = aListIter;
105     }
106     else
107     {
108       myResultMap.Bind (aNextObject, aListIter);
109     }
110   }
111   return AIS_SS_Removed;
112 }
113
114 //=======================================================================
115 //function : AddSelect
116 //purpose  :
117 //=======================================================================
118 AIS_SelectStatus AIS_Selection::AddSelect (const Handle(SelectMgr_EntityOwner)& theObject)
119 {
120   if (theObject.IsNull()
121   || !theObject->HasSelectable()
122   ||  myResultMap.IsBound (theObject))
123   {
124     return AIS_SS_NotDone;
125   }
126
127   AIS_NListOfEntityOwner::Iterator aListIter;
128   myresult.Append  (theObject, aListIter);
129   myResultMap.Bind (theObject, aListIter);
130   theObject->SetSelected (Standard_True);
131   return AIS_SS_Added;
132 }