]> SALOME platform Git repositories - modules/smesh.git/blob - src/OBJECT/SMESH_NodeLabelActor.cxx
Salome HOME
0022100: EDF 2413 SMESH: Take into account TRIA7
[modules/smesh.git] / src / OBJECT / SMESH_NodeLabelActor.cxx
1 // Copyright (C) 2007-2013  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 //  File   : SMESH_NodeLabelActor.cxx
23 //  Author : Roman NIKOLAEV
24 //  Module : SMESH
25 //
26 #include "SMESH_NodeLabelActor.h"
27
28 #include <VTKViewer_TransformFilter.h>
29
30 #include <vtkObjectFactory.h>
31 #include <vtkCallbackCommand.h>
32 #include <vtkMaskPoints.h>
33 #include <vtkSelectVisiblePoints.h>
34 #include <vtkLabeledDataMapper.h>
35 #include <vtkActor2D.h>
36 #include <vtkTextProperty.h>
37 #include <vtkPointData.h>
38 #include <vtkProperty2D.h>
39 #include <vtkRenderer.h>
40 #include <vtkUnstructuredGrid.h>
41
42 vtkStandardNewMacro(SMESH_NodeLabelActor);
43
44 /*!
45   Constructor.
46 */
47 SMESH_NodeLabelActor::SMESH_NodeLabelActor() {
48   //Definition of points numbering pipeline
49   //---------------------------------------
50   myPointsNumDataSet = vtkUnstructuredGrid::New();
51
52   myPtsMaskPoints = vtkMaskPoints::New();
53   myPtsMaskPoints->SetInputData(myPointsNumDataSet);
54   myPtsMaskPoints->SetOnRatio(1);
55
56   myPtsSelectVisiblePoints = vtkSelectVisiblePoints::New();
57   myPtsSelectVisiblePoints->SetInputConnection(myPtsMaskPoints->GetOutputPort());
58   myPtsSelectVisiblePoints->SelectInvisibleOff();
59   myPtsSelectVisiblePoints->SetTolerance(0.1);
60     
61   myPtsLabeledDataMapper = vtkLabeledDataMapper::New();
62   myPtsLabeledDataMapper->SetInputConnection(myPtsSelectVisiblePoints->GetOutputPort());
63   myPtsLabeledDataMapper->SetLabelFormat("%d");
64   myPtsLabeledDataMapper->SetLabelModeToLabelScalars();
65     
66   myPtsTextProp = vtkTextProperty::New();
67   myPtsTextProp->SetFontFamilyToTimes();
68   myPtsTextProp->SetFontSize(10);
69   myPtsTextProp->SetBold(1);
70   myPtsTextProp->SetItalic(0);
71   myPtsTextProp->SetShadow(0);
72   myPtsTextProp->SetColor( 1, 1, 1 );
73   myPtsLabeledDataMapper->SetLabelTextProperty(myPtsTextProp);
74
75   myIsPointsLabeled = false;
76
77   myPointLabels = vtkActor2D::New();
78   myPointLabels->SetMapper(myPtsLabeledDataMapper);
79   myPointLabels->SetVisibility(myIsPointsLabeled);
80
81   vtkCallbackCommand* callBackCommand = vtkCallbackCommand::New();
82   callBackCommand->SetClientData(this);
83   callBackCommand->SetCallback(SMESH_NodeLabelActor::ProcessEvents);
84
85   myTransformFilter->AddObserver("VTKViewer_TransformFilter::TransformationFinished",
86                                  callBackCommand);
87   callBackCommand->Delete();
88 }
89
90 /*!
91   Destructor
92 */
93 SMESH_NodeLabelActor::~SMESH_NodeLabelActor() {
94   //Deleting of points numbering pipeline
95   //---------------------------------------
96   myPointsNumDataSet->Delete();
97   
98   // commented: porting to vtk 5.0
99   //  myPtsLabeledDataMapper->RemoveAllInputs();
100   myPtsLabeledDataMapper->Delete();
101   
102   // commented: porting to vtk 5.0
103   //  myPtsSelectVisiblePoints->UnRegisterAllOutputs();
104   myPtsSelectVisiblePoints->Delete();
105   
106   // commented: porting to vtk 5.0
107   //  myPtsMaskPoints->UnRegisterAllOutputs();
108   myPtsMaskPoints->Delete();
109   myPointLabels->Delete();
110   myPtsTextProp->Delete();
111 }
112
113 void SMESH_NodeLabelActor::SetFontProperties( SMESH::LabelFont family, int size, 
114                                               bool bold, bool italic, bool shadow,
115                                               double r, double g, double b )
116 {
117   switch ( family ) {
118   case SMESH::FntArial:
119     myPtsTextProp->SetFontFamilyToArial(); break;
120   case SMESH::FntCourier:
121     myPtsTextProp->SetFontFamilyToCourier(); break;
122   case SMESH::FntTimes:
123   default:
124     myPtsTextProp->SetFontFamilyToTimes(); break;
125   }
126   myPtsTextProp->SetFontSize( size );
127   myPtsTextProp->SetBold( bold );
128   myPtsTextProp->SetItalic( italic );
129   myPtsTextProp->SetShadow( shadow ); 
130   myPtsTextProp->SetColor( r, g, b ); 
131 }
132
133 void SMESH_NodeLabelActor::SetPointsLabeled(bool theIsPointsLabeled) {
134   myTransformFilter->Update();
135   vtkDataSet* aGrid = vtkUnstructuredGrid::SafeDownCast(myTransformFilter->GetOutput());
136
137   if(!aGrid)
138     return;
139     
140   myIsPointsLabeled = theIsPointsLabeled && aGrid->GetNumberOfPoints();
141
142   if ( myIsPointsLabeled )
143   {
144     myPointsNumDataSet->ShallowCopy(aGrid);
145     vtkUnstructuredGrid *aDataSet = myPointsNumDataSet;
146     
147     int aNbElem = aDataSet->GetNumberOfPoints();
148     
149     vtkIntArray *anArray = vtkIntArray::New();
150     anArray->SetNumberOfValues( aNbElem );
151     
152     for ( vtkIdType anId = 0; anId < aNbElem; anId++ )
153     {
154       int aSMDSId = myVisualObj->GetNodeObjId( anId );
155       anArray->SetValue( anId, aSMDSId );
156     }
157     
158     aDataSet->GetPointData()->SetScalars( anArray );
159     myPtsMaskPoints->SetInputData( aDataSet );
160     myPointLabels->SetVisibility( GetVisibility() );
161     anArray->Delete();
162   }
163   else
164   {
165     myPointLabels->SetVisibility( false );
166   } 
167 }
168
169
170 void SMESH_NodeLabelActor::SetVisibility(int theMode)
171 {
172   SMESH_DeviceActor::SetVisibility(theMode);
173   myPointLabels->VisibilityOff();
174   if(myIsPointsLabeled && theMode)
175     myPointLabels->VisibilityOn();
176 }
177
178
179 void SMESH_NodeLabelActor::AddToRender(vtkRenderer* theRenderer)
180 {
181   SMESH_DeviceActor::AddToRender(theRenderer);
182   myPtsSelectVisiblePoints->SetRenderer(theRenderer);
183   theRenderer->AddActor2D(myPointLabels);
184 }
185
186 void SMESH_NodeLabelActor::RemoveFromRender(vtkRenderer* theRenderer)
187 {
188   theRenderer->RemoveActor(myPointLabels);
189   SMESH_DeviceActor::RemoveFromRender(theRenderer);
190 }
191
192 void SMESH_NodeLabelActor::UpdateLabels() {
193   if(myIsPointsLabeled)
194     SetPointsLabeled(myIsPointsLabeled);
195 }
196
197
198 void SMESH_NodeLabelActor::ProcessEvents(vtkObject* vtkNotUsed(theObject),
199                                          unsigned long theEvent,
200                                          void* theClientData,
201                                          void* vtkNotUsed(theCallData)) {
202   SMESH_NodeLabelActor* self = reinterpret_cast<SMESH_NodeLabelActor*>(theClientData);    
203   if(self)
204     self->UpdateLabels();
205 }