1 // Copyright (C) 2007-2012 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 // File : SMESH_NodeLabelActor.cxx
23 // Author : Roman NIKOLAEV
26 #include "SMESH_NodeLabelActor.h"
28 #include <VTKViewer_TransformFilter.h>
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>
42 vtkStandardNewMacro(SMESH_NodeLabelActor);
47 SMESH_NodeLabelActor::SMESH_NodeLabelActor() {
48 //Definition of points numbering pipeline
49 //---------------------------------------
50 myPointsNumDataSet = vtkUnstructuredGrid::New();
52 myPtsMaskPoints = vtkMaskPoints::New();
53 myPtsMaskPoints->SetInput(myPointsNumDataSet);
54 myPtsMaskPoints->SetOnRatio(1);
56 myPtsSelectVisiblePoints = vtkSelectVisiblePoints::New();
57 myPtsSelectVisiblePoints->SetInput(myPtsMaskPoints->GetOutput());
58 myPtsSelectVisiblePoints->SelectInvisibleOff();
59 myPtsSelectVisiblePoints->SetTolerance(0.1);
61 myPtsLabeledDataMapper = vtkLabeledDataMapper::New();
62 myPtsLabeledDataMapper->SetInput(myPtsSelectVisiblePoints->GetOutput());
63 myPtsLabeledDataMapper->SetLabelFormat("%d");
64 myPtsLabeledDataMapper->SetLabelModeToLabelScalars();
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);
75 myIsPointsLabeled = false;
77 myPointLabels = vtkActor2D::New();
78 myPointLabels->SetMapper(myPtsLabeledDataMapper);
79 myPointLabels->SetVisibility(myIsPointsLabeled);
81 vtkCallbackCommand* callBackCommand = vtkCallbackCommand::New();
82 callBackCommand->SetClientData(this);
83 callBackCommand->SetCallback(SMESH_NodeLabelActor::ProcessEvents);
85 myTransformFilter->AddObserver("VTKViewer_TransformFilter::TransformationFinished",
87 callBackCommand->Delete();
93 SMESH_NodeLabelActor::~SMESH_NodeLabelActor() {
94 //Deleting of points numbering pipeline
95 //---------------------------------------
96 myPointsNumDataSet->Delete();
98 // commented: porting to vtk 5.0
99 // myPtsLabeledDataMapper->RemoveAllInputs();
100 myPtsLabeledDataMapper->Delete();
102 // commented: porting to vtk 5.0
103 // myPtsSelectVisiblePoints->UnRegisterAllOutputs();
104 myPtsSelectVisiblePoints->Delete();
106 // commented: porting to vtk 5.0
107 // myPtsMaskPoints->UnRegisterAllOutputs();
108 myPtsMaskPoints->Delete();
109 myPointLabels->Delete();
110 myPtsTextProp->Delete();
113 void SMESH_NodeLabelActor::SetFontProperties( SMESH::LabelFont family, int size,
114 bool bold, bool italic, bool shadow,
115 vtkFloatingPointType r, vtkFloatingPointType g, vtkFloatingPointType b )
118 case SMESH::FntArial:
119 myPtsTextProp->SetFontFamilyToArial(); break;
120 case SMESH::FntCourier:
121 myPtsTextProp->SetFontFamilyToCourier(); break;
122 case SMESH::FntTimes:
124 myPtsTextProp->SetFontFamilyToTimes(); break;
126 myPtsTextProp->SetFontSize( size );
127 myPtsTextProp->SetBold( bold );
128 myPtsTextProp->SetItalic( italic );
129 myPtsTextProp->SetShadow( shadow );
130 myPtsTextProp->SetColor( r, g, b );
133 void SMESH_NodeLabelActor::SetPointsLabeled(bool theIsPointsLabeled) {
134 myTransformFilter->Update();
135 vtkDataSet* aGrid = vtkUnstructuredGrid::SafeDownCast(myTransformFilter->GetOutput());
140 myIsPointsLabeled = theIsPointsLabeled && aGrid->GetNumberOfPoints();
142 if ( myIsPointsLabeled )
144 myPointsNumDataSet->ShallowCopy(aGrid);
145 vtkUnstructuredGrid *aDataSet = myPointsNumDataSet;
147 int aNbElem = aDataSet->GetNumberOfPoints();
149 vtkIntArray *anArray = vtkIntArray::New();
150 anArray->SetNumberOfValues( aNbElem );
152 for ( vtkIdType anId = 0; anId < aNbElem; anId++ )
154 int aSMDSId = myVisualObj->GetNodeObjId( anId );
155 anArray->SetValue( anId, aSMDSId );
158 aDataSet->GetPointData()->SetScalars( anArray );
159 myPtsMaskPoints->SetInput( aDataSet );
160 myPointLabels->SetVisibility( GetVisibility() );
165 myPointLabels->SetVisibility( false );
170 void SMESH_NodeLabelActor::SetVisibility(int theMode)
172 SMESH_DeviceActor::SetVisibility(theMode);
173 myPointLabels->VisibilityOff();
174 if(myIsPointsLabeled && theMode)
175 myPointLabels->VisibilityOn();
179 void SMESH_NodeLabelActor::AddToRender(vtkRenderer* theRenderer)
181 SMESH_DeviceActor::AddToRender(theRenderer);
182 myPtsSelectVisiblePoints->SetRenderer(theRenderer);
183 theRenderer->AddActor2D(myPointLabels);
186 void SMESH_NodeLabelActor::RemoveFromRender(vtkRenderer* theRenderer)
188 theRenderer->RemoveActor(myPointLabels);
189 SMESH_DeviceActor::RemoveFromRender(theRenderer);
192 void SMESH_NodeLabelActor::UpdateLabels() {
193 if(myIsPointsLabeled)
194 SetPointsLabeled(myIsPointsLabeled);
198 void SMESH_NodeLabelActor::ProcessEvents(vtkObject* vtkNotUsed(theObject),
199 unsigned long theEvent,
201 void* vtkNotUsed(theCallData)) {
202 SMESH_NodeLabelActor* self = reinterpret_cast<SMESH_NodeLabelActor*>(theClientData);
204 self->UpdateLabels();