]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_MeshEditPreview.cxx
Salome HOME
Porting SMESH module to Qt 4
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_MeshEditPreview.cxx
1 // SMESH SMESHGUI : GUI for SMESH component
2 //
3 // Copyright (C) 2003  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 // File   : SMESHGUI_MeshEditPreview.cxx
23 // Author : Open CASCADE S.A.S.
24 //
25
26 // SMESH includes
27 #include "SMESHGUI_MeshEditPreview.h"
28
29 #include "SMESHGUI_VTKUtils.h"
30
31 #include <SMESH_Actor.h>
32 #include <SMESH_ActorUtils.h>
33
34 // SALOME GUI includes
35 #include <VTKViewer_CellLocationsArray.h>
36 #include <SVTK_ViewWindow.h>
37
38 // VTK includes
39 #include <vtkPoints.h>
40 #include <vtkIdList.h>
41 #include <vtkCellArray.h>
42 #include <vtkUnsignedCharArray.h>
43 #include <vtkUnstructuredGrid.h>
44 #include <vtkUnstructuredGridWriter.h>
45 #include <vtkDataSetMapper.h>
46 #include <vtkProperty.h>
47
48 // Qt includes
49 #include <QColor>
50
51 // IDL includes
52 #include <SALOMEconfig.h>
53 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
54
55 //================================================================================
56 /*!
57  * \brief Constructor
58  */
59 //================================================================================
60
61 SMESHGUI_MeshEditPreview::SMESHGUI_MeshEditPreview(SVTK_ViewWindow* theViewWindow):
62   myViewWindow(theViewWindow)
63 {
64   myGrid = vtkUnstructuredGrid::New();
65
66   // Create and display actor
67   vtkDataSetMapper* aMapper = vtkDataSetMapper::New();
68   aMapper->SetInput( myGrid );
69
70   myPreviewActor = SALOME_Actor::New();
71   myPreviewActor->SetInfinitive(true);
72   myPreviewActor->VisibilityOn();
73   myPreviewActor->PickableOff();
74
75   vtkFloatingPointType anRGB[3];
76   SMESH::GetColor( "SMESH", "selection_element_color", anRGB[0], anRGB[1], anRGB[2], QColor( 0, 170, 255 ) );
77   SetColor( anRGB[0], anRGB[1], anRGB[2] );
78
79   myPreviewActor->SetMapper( aMapper );
80   aMapper->Delete();
81
82   myViewWindow->AddActor(myPreviewActor);
83
84 }
85
86 //================================================================================
87 /*!
88  * \brief Destroy
89  */
90 //================================================================================
91
92 SMESHGUI_MeshEditPreview::~SMESHGUI_MeshEditPreview()
93 {
94   myGrid->Delete();
95
96   myViewWindow->RemoveActor(myPreviewActor);
97   myPreviewActor->Delete();
98
99 }
100
101 //================================================================================
102 /*!
103  * \brief Returns vtk cell type
104  */
105 //================================================================================
106
107 vtkIdType getCellType( const SMDSAbs_ElementType theType,
108                        const bool thePoly,
109                        const int theNbNodes )
110 {
111   switch( theType ) 
112   {
113   case SMDSAbs_Node:              return VTK_VERTEX;
114   case SMDSAbs_Edge: 
115     if( theNbNodes == 2 )         return VTK_LINE;
116     else if ( theNbNodes == 3 )   return VTK_QUADRATIC_EDGE;
117     else return VTK_EMPTY_CELL;
118
119   case SMDSAbs_Face  :
120     if (thePoly && theNbNodes>2 ) return VTK_POLYGON;
121     else if ( theNbNodes == 3 )   return VTK_TRIANGLE;
122     else if ( theNbNodes == 4 )   return VTK_QUAD;
123     else if ( theNbNodes == 6 )   return VTK_QUADRATIC_TRIANGLE;
124     else if ( theNbNodes == 8 )   return VTK_QUADRATIC_QUAD;
125     else return VTK_EMPTY_CELL;
126
127   case SMDSAbs_Volume:
128     if (thePoly && theNbNodes>3 ) return VTK_CONVEX_POINT_SET;
129     else if ( theNbNodes == 4 )   return VTK_TETRA;
130     else if ( theNbNodes == 5 )   return VTK_PYRAMID;
131     else if ( theNbNodes == 6 )   return VTK_WEDGE;
132     else if ( theNbNodes == 8 )   return VTK_HEXAHEDRON;
133     else if ( theNbNodes == 10 )  {
134       return VTK_QUADRATIC_TETRA;
135     }
136     else if ( theNbNodes == 20 )  {
137       return VTK_QUADRATIC_HEXAHEDRON;
138     }
139     else if ( theNbNodes==13 || theNbNodes==15 )  {
140       return VTK_CONVEX_POINT_SET;
141     }
142     else return VTK_EMPTY_CELL;
143
144   default: return VTK_EMPTY_CELL;
145   }
146 }
147
148 //================================================================================
149 /*!
150  * \brief Set preview data
151  */
152 //================================================================================
153
154 void SMESHGUI_MeshEditPreview::SetData (const SMESH::MeshPreviewStruct* previewData)
155 {
156   // Create points
157   const SMESH::nodes_array& aNodesXYZ = previewData->nodesXYZ;
158   vtkPoints* aPoints = vtkPoints::New();
159   aPoints->SetNumberOfPoints(aNodesXYZ.length());
160
161   for ( int i = 0; i < aNodesXYZ.length(); i++ ) {
162     aPoints->SetPoint( i, aNodesXYZ[i].x, aNodesXYZ[i].y, aNodesXYZ[i].z );
163   }
164   myGrid->SetPoints(aPoints);
165
166   aPoints->Delete();
167
168   // Create cells
169   const SMESH::long_array&  anElemConnectivity = previewData->elementConnectivities;
170   const SMESH::types_array& anElemTypes = previewData->elementTypes;
171
172   vtkIdType aCellsSize = anElemConnectivity.length() + anElemTypes.length();
173   vtkIdType aNbCells = anElemTypes.length();
174
175   vtkCellArray* aConnectivity = vtkCellArray::New();
176   aConnectivity->Allocate( aCellsSize, 0 );
177
178   vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
179   aCellTypesArray->SetNumberOfComponents( 1 );
180   aCellTypesArray->Allocate( aNbCells * aCellTypesArray->GetNumberOfComponents() );
181
182   vtkIdList *anIdList = vtkIdList::New();
183   int aNodePos = 0;
184
185   for ( int i = 0; i < anElemTypes.length(); i++ ) {
186     const SMESH::ElementSubType& anElementSubType = anElemTypes[i];
187     SMDSAbs_ElementType aType = SMDSAbs_ElementType(anElementSubType.SMDS_ElementType);
188     vtkIdType aNbNodes = anElementSubType.nbNodesInElement;
189     anIdList->SetNumberOfIds( aNbNodes );
190
191     for ( vtkIdType aNodeId = 0; aNodeId < aNbNodes; aNodeId++ ){
192       anIdList->SetId( aNodeId, anElemConnectivity[aNodePos] );
193       aNodePos++;
194     }
195
196     aConnectivity->InsertNextCell( anIdList );
197     aCellTypesArray->InsertNextValue( getCellType( aType,
198                                                    anElemTypes[i].isPoly, 
199                                                    aNbNodes ) );
200   }
201   anIdList->Delete();
202
203   // Insert cells in grid
204   VTKViewer_CellLocationsArray* aCellLocationsArray = VTKViewer_CellLocationsArray::New();
205   aCellLocationsArray->SetNumberOfComponents( 1 );
206   aCellLocationsArray->SetNumberOfTuples( aNbCells );
207
208   aConnectivity->InitTraversal();
209   for( vtkIdType idType = 0, *pts, npts; aConnectivity->GetNextCell( npts, pts ); idType++ )
210     aCellLocationsArray->SetValue( idType, aConnectivity->GetTraversalLocation( npts ) );
211
212   myGrid->SetCells( aCellTypesArray, aCellLocationsArray, aConnectivity );
213
214   myPreviewActor->GetMapper()->Update();
215
216   aCellTypesArray->Delete();
217   aCellLocationsArray->Delete();
218   aConnectivity->Delete();
219
220   SetVisibility(true);
221 }
222
223 //================================================================================
224 /*!
225  * \brief Set visibility
226  */
227 //================================================================================
228
229 void SMESHGUI_MeshEditPreview::SetVisibility (bool theVisibility)
230 {
231   myPreviewActor->SetVisibility(theVisibility);
232   SMESH::RepaintCurrentView();
233 }
234
235 //================================================================================
236 /*!
237  * \brief Set preview color
238  */
239 //================================================================================
240
241 void SMESHGUI_MeshEditPreview::SetColor(double R, double G, double B)
242 {
243   myPreviewActor->SetColor( R, G, B );
244 }
245
246 //================================================================================
247 /*!
248  * \brief Get preview actor
249  */
250 //================================================================================
251 SALOME_Actor* SMESHGUI_MeshEditPreview::GetActor() const
252
253   return myPreviewActor;
254 }