0031458: Visualization - refine classes across Prs3d and StdPrs packages
[occt.git] / src / MeshVS / MeshVS_MeshOwner.cxx
1 // Created on: 2007-01-25
2 // Created by: Sergey KOCHETKOV
3 // Copyright (c) 2007-2014 OPEN CASCADE SAS
4 //
5 // This file is part of Open CASCADE Technology software library.
6 //
7 // This library is free software; you can redistribute it and/or modify it under
8 // the terms of the GNU Lesser General Public License version 2.1 as published
9 // by the Free Software Foundation, with special exception defined in the file
10 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
11 // distribution for complete text of the license and disclaimer of any warranty.
12 //
13 // Alternatively, this file may be used under the terms of Open CASCADE
14 // commercial license or contractual agreement.
15
16
17 #include <MeshVS_DataSource.hxx>
18 #include <MeshVS_Mesh.hxx>
19 #include <MeshVS_MeshOwner.hxx>
20 #include <PrsMgr_PresentationManager.hxx>
21 #include <PrsMgr_PresentationManager3d.hxx>
22 #include <SelectMgr_SelectableObject.hxx>
23 #include <Standard_Type.hxx>
24 #include <TColStd_HPackedMapOfInteger.hxx>
25 #include <TColStd_MapIteratorOfPackedMapOfInteger.hxx>
26
27 IMPLEMENT_STANDARD_RTTIEXT(MeshVS_MeshOwner,SelectMgr_EntityOwner)
28
29 #ifndef MeshVS_PRSBUILDERHXX
30 #include <MeshVS_PrsBuilder.hxx>
31 #endif
32
33
34 //================================================================
35 // Function : Constructor MeshVS_MeshOwner
36 // Purpose  :
37 //================================================================
38 MeshVS_MeshOwner::MeshVS_MeshOwner (const SelectMgr_SelectableObject* theSelObj,
39                                     const Handle(MeshVS_DataSource)& theDS,
40                                     const Standard_Integer           thePriority)
41 : SelectMgr_EntityOwner ( theSelObj, thePriority )
42 {
43   myLastID = -1;
44   if( !theDS.IsNull() )
45     myDataSource = theDS;
46 }
47
48 //================================================================
49 // Function : GetDataSource
50 // Purpose  :
51 //================================================================
52 const Handle(MeshVS_DataSource)& MeshVS_MeshOwner::GetDataSource () const
53 {
54   return myDataSource;
55 }
56
57 //================================================================
58 // Function : GetSelectedNodes
59 // Purpose  :
60 //================================================================
61 const Handle(TColStd_HPackedMapOfInteger)& MeshVS_MeshOwner::GetSelectedNodes () const
62 {
63   return mySelectedNodes;
64 }
65
66 //================================================================
67 // Function : GetSelectedElements
68 // Purpose  :
69 //================================================================
70 const Handle(TColStd_HPackedMapOfInteger)& MeshVS_MeshOwner::GetSelectedElements () const
71 {
72   return mySelectedElems;
73 }
74
75 //================================================================
76 // Function : AddSelectedEntities
77 // Purpose  :
78 //================================================================
79 void MeshVS_MeshOwner::AddSelectedEntities (const Handle(TColStd_HPackedMapOfInteger)& Nodes,
80                                             const Handle(TColStd_HPackedMapOfInteger)& Elems)
81 {
82   if( mySelectedNodes.IsNull() )
83     mySelectedNodes = Nodes;
84   else if( !Nodes.IsNull() )
85     mySelectedNodes->ChangeMap().Unite( Nodes->Map() );
86   if( mySelectedElems.IsNull() )
87     mySelectedElems = Elems;
88   else if( !Elems.IsNull() )
89     mySelectedElems->ChangeMap().Unite( Elems->Map() );
90 }
91
92 //================================================================
93 // Function : ClearSelectedEntities
94 // Purpose  :
95 //================================================================
96 void MeshVS_MeshOwner::ClearSelectedEntities ()
97 {
98   mySelectedNodes.Nullify();
99   mySelectedElems.Nullify();
100 }
101
102 //================================================================
103 // Function : GetDetectedNodes
104 // Purpose  :
105 //================================================================
106 const Handle(TColStd_HPackedMapOfInteger)& MeshVS_MeshOwner::GetDetectedNodes () const
107 {
108   return myDetectedNodes;
109 }
110
111 //================================================================
112 // Function : GetDetectedElements
113 // Purpose  :
114 //================================================================
115 const Handle(TColStd_HPackedMapOfInteger)& MeshVS_MeshOwner::GetDetectedElements () const
116 {
117   return myDetectedElems;
118 }
119
120 //================================================================
121 // Function : SetDetectedEntities
122 // Purpose  :
123 //================================================================
124 void MeshVS_MeshOwner::SetDetectedEntities (const Handle(TColStd_HPackedMapOfInteger)& Nodes,
125                                             const Handle(TColStd_HPackedMapOfInteger)& Elems)
126 {
127   myDetectedNodes = Nodes;
128   myDetectedElems = Elems;
129   if (IsSelected()) SetSelected (Standard_False);
130 }
131
132 //================================================================
133 // Function : HilightWithColor
134 // Purpose  :
135 //================================================================
136 void MeshVS_MeshOwner::HilightWithColor (const Handle(PrsMgr_PresentationManager3d)& thePM,
137                                          const Handle(Prs3d_Drawer)& theStyle,
138                                          const Standard_Integer /*theMode*/)
139 {
140   Handle( SelectMgr_SelectableObject ) aSelObj;
141   if ( HasSelectable() )
142     aSelObj = Selectable();
143
144   if ( thePM->IsImmediateModeOn() && aSelObj->IsKind( STANDARD_TYPE( MeshVS_Mesh ) ) )
145   {
146     // Update last detected entity ID
147     Handle(TColStd_HPackedMapOfInteger) aNodes = GetDetectedNodes();
148     Handle(TColStd_HPackedMapOfInteger) aElems = GetDetectedElements(); 
149     if( !aNodes.IsNull() && aNodes->Map().Extent() == 1 )
150     {
151       TColStd_MapIteratorOfPackedMapOfInteger anIt( aNodes->Map() );
152       if( myLastID != anIt.Key() )
153       {
154         myLastID = anIt.Key();
155       }
156     }
157     else if( !aElems.IsNull() && aElems->Map().Extent() == 1 )
158     {
159       TColStd_MapIteratorOfPackedMapOfInteger anIt( aElems->Map() );
160       if( myLastID != anIt.Key() )
161       {
162         myLastID = anIt.Key();
163       }
164     }
165
166     // hilight detected entities
167     Handle( MeshVS_Mesh ) aMesh = Handle( MeshVS_Mesh )::DownCast ( aSelObj );
168     aMesh->HilightOwnerWithColor ( thePM, theStyle, this );
169   }
170 }
171
172 void MeshVS_MeshOwner::Unhilight (const Handle(PrsMgr_PresentationManager)& thePM, const Standard_Integer )
173 {
174   SelectMgr_EntityOwner::Unhilight (thePM);
175
176   Handle(TColStd_HPackedMapOfInteger) aNodes = GetDetectedNodes();
177   Handle(TColStd_HPackedMapOfInteger) aElems = GetDetectedElements();  
178   if( ( !aNodes.IsNull() && !aNodes->Map().Contains( myLastID ) ) ||
179       ( !aElems.IsNull() && !aElems->Map().Contains( myLastID ) ) )
180     return;
181   // Reset last detected ID
182   myLastID = -1;
183 }
184
185 Standard_Boolean MeshVS_MeshOwner::IsForcedHilight () const
186 {
187   Standard_Boolean aHilight = Standard_True;
188   Standard_Integer aKey = -1;
189   if( myLastID > 0 )
190   {
191     // Check the detected entity and 
192     // allow to hilight it if it differs from the last detected entity <myLastID>
193     Handle(TColStd_HPackedMapOfInteger) aNodes = GetDetectedNodes();
194     if( !aNodes.IsNull() && aNodes->Map().Extent() == 1 )
195     {
196       TColStd_MapIteratorOfPackedMapOfInteger anIt( aNodes->Map() );
197       aKey = anIt.Key();
198       if( myLastID == aKey )
199       {  
200          aHilight = Standard_False;
201       }
202     }  
203     Handle(TColStd_HPackedMapOfInteger) aElems = GetDetectedElements();
204     if( !aElems.IsNull() && aElems->Map().Extent() == 1 )
205     {
206       TColStd_MapIteratorOfPackedMapOfInteger anIt( aElems->Map() );
207       aKey = anIt.Key();
208       if( myLastID == aKey )
209       {
210           aHilight = Standard_False;
211       }
212     }
213   } 
214   return aHilight;
215 }