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