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