Salome HOME
correct previous integration (Porting to Python 2.6)
[modules/smesh.git] / src / OBJECT / SMESH_PreviewActorsCollection.cxx
1 //  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 //  This library is free software; you can redistribute it and/or
7 //  modify it under the terms of the GNU Lesser General Public
8 //  License as published by the Free Software Foundation; either
9 //  version 2.1 of the License.
10 //
11 //  This library is distributed in the hope that it will be useful,
12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 //  Lesser General Public License for more details.
15 //
16 //  You should have received a copy of the GNU Lesser General Public
17 //  License along with this library; if not, write to the Free Software
18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 //  SMESH OBJECT : interactive object for SMESH visualization
23 //  File   : SMESH_PreviewActorsCollection.cxx
24 //  Author : 
25 //  Module : SMESH
26 //
27
28 #include "SMESH_PreviewActorsCollection.h"
29
30 #include "utilities.h"
31
32 #include "SALOME_InteractiveObject.hxx"
33
34 // OCC includes
35 #include <TopoDS.hxx>
36 #include <TopExp.hxx>
37 #include <TopExp_Explorer.hxx>
38
39 // VTK includes
40 #include <vtkUnstructuredGrid.h>
41 #include <vtkScalarBarActor.h>
42 #include <vtkPlane.h>
43 #include <vtkRenderer.h>
44 #include <vtkProperty.h>
45
46 #include "VTKViewer_Actor.h"
47
48 #include "SVTK_DeviceActor.h"
49 #include "SALOME_Actor.h"
50
51 // QT
52 #include <QString>
53 #include <QColor>
54
55 #ifdef _DEBUG_
56 static int MYDEBUG = 0;
57 #else
58 static int MYDEBUG = 0;
59 #endif
60
61 using namespace std;
62
63
64 //vtkStandardNewMacro(SMESH_PreviewActorsCollection);
65
66
67 SMESH_PreviewActorsCollection
68 ::SMESH_PreviewActorsCollection()
69 {
70   if(MYDEBUG) MESSAGE("SMESH_PreviewActorsCollection - "<<this);
71 }
72
73
74 SMESH_PreviewActorsCollection
75 ::~SMESH_PreviewActorsCollection()
76 {
77   if(MYDEBUG) MESSAGE("~SMESH_PreviewActorsCollection - "<<this);
78   if (myRenderer)
79     RemoveFromRender(myRenderer);
80
81   QMap<int, GEOM_Actor*>::iterator iter = myMapOfActors.begin();
82   for ( ; iter != myMapOfActors.end(); ++iter )
83     if ( GEOM_Actor* anActor = iter.value() )
84       anActor->Delete();
85   myMapOfActors.clear();
86 }
87
88 bool SMESH_PreviewActorsCollection::Init( const TopoDS_Shape& theShape, TopAbs_ShapeEnum theType, const QString& theEntry )
89 {
90   myMainShape = theShape;
91   myMapOfActors.clear();
92   myMapOfShapes.Clear();
93
94   if ( theShape.IsNull() )
95     return false;
96
97   Handle( SALOME_InteractiveObject ) anIO = new SALOME_InteractiveObject();
98   anIO->setEntry( theEntry.toLatin1().constData() );
99   
100   // get indexes of seleted elements
101   TopExp::MapShapes(theShape, myMapOfShapes);
102
103   TopExp_Explorer exp( theShape, theType );
104   for ( ; exp.More(); exp.Next() ) {
105     int index = myMapOfShapes.FindIndex( exp.Current() );
106     if ( index && !myMapOfActors.contains( index ) ) { 
107       // create actor if the index is present
108       if ( GEOM_Actor* anActor = createActor( exp.Current() )) {
109         // Create new entry for actor
110         QString aString = theEntry;
111         aString += QString("_%1").arg( index ); // add index to actor entry
112
113         // Create interactive object
114         Handle( SALOME_InteractiveObject ) anIO = new SALOME_InteractiveObject();
115         anIO->setEntry( aString.toLatin1().constData() );
116
117         // Init Actor
118         anActor->SetVectorMode( true );
119         anActor->setIO( anIO );
120         anActor->SetSelector( mySelector );
121         anActor->SetPickable( true );
122         anActor->SetResolveCoincidentTopology( true );
123
124         // Add Actor to the Actors Map
125         myMapOfActors.insert(index, anActor);
126       }
127     }
128   }
129   mySelector->ClearIObjects();
130
131   return true;
132 }
133
134 GEOM_Actor* SMESH_PreviewActorsCollection::createActor(const TopoDS_Shape& shape)
135 {
136   GEOM_Actor* actor = GEOM_Actor::New();
137   actor->SetShape(shape,0,0);
138
139   //Color Properties
140   /*    
141         vtkProperty* aProp = vtkProperty::New();
142         vtkProperty* aHLProp = vtkProperty::New();
143         vtkProperty* aPHLProp = vtkProperty::New();
144         
145         aProp->SetColor( 255, 0, 0);
146         actor->SetProperty(aProp);
147
148         aHLProp->SetColor( 255, 255, 255);
149         actor->SetHighlightProperty(aHLProp);
150
151         aPHLProp->SetColor( 155, 155, 155);
152         aPHLProp->SetLineWidth ( 3 );
153         aPHLProp->SetOpacity ( 0.75 );
154         actor->SetPreHighlightProperty(aPHLProp);
155
156         aProp->Delete();
157         aHLProp->Delete();
158         aPHLProp->Delete();
159   */
160
161   return actor;
162 }
163
164 GEOM_Actor* SMESH_PreviewActorsCollection::GetActorByIndex(int index)
165 {
166   return myMapOfActors.value(index);
167 }
168
169 int SMESH_PreviewActorsCollection::GetIndexByShape( const TopoDS_Shape& theShape )
170 {
171   return myMapOfShapes.FindIndex( theShape );
172 }
173
174 void SMESH_PreviewActorsCollection::AddToRender(vtkRenderer* theRenderer)
175 {
176   myRenderer = theRenderer;
177
178   QMap<int, GEOM_Actor*>::iterator iter = myMapOfActors.begin();
179   for ( ; iter != myMapOfActors.end(); ++iter )
180     iter.value()->AddToRender( theRenderer );
181 }
182
183 void SMESH_PreviewActorsCollection::RemoveFromRender(vtkRenderer* theRenderer){
184   QMap<int, GEOM_Actor*>::iterator iter = myMapOfActors.begin();
185   for ( ; iter != myMapOfActors.end(); ++iter )
186     iter.value()->RemoveFromRender( theRenderer );
187 }
188
189 void SMESH_PreviewActorsCollection::SetSelector(SVTK_Selector* theSelector)
190 {
191   mySelector = theSelector;
192 }
193
194 void SMESH_PreviewActorsCollection::HighlightAll( bool theHighlight ){
195   QMap<int, GEOM_Actor*>::iterator iter = myMapOfActors.begin();
196   for ( ; iter != myMapOfActors.end(); ++iter )
197     iter.value()->Highlight( theHighlight );
198 }
199
200 void SMESH_PreviewActorsCollection::HighlightID( int index ){
201   GEOM_Actor* anActor = GetActorByIndex( index );
202   if ( anActor && !anActor->isHighlighted() )
203     anActor->Highlight( true );
204 }
205
206 void SMESH_PreviewActorsCollection::SetShown( bool shown ){
207   QMap<int, GEOM_Actor*>::iterator iter = myMapOfActors.begin();
208   for ( ; iter != myMapOfActors.end(); ++iter )
209     iter.value()->SetVisibility( shown );
210 }