2 # ---------------------------------------------------------------------
4 # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
6 # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
7 # Previous Authors : Laurent Guigues, Jean-Pierre Roux
8 # CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
10 # This software is governed by the CeCILL-B license under French law and
11 # abiding by the rules of distribution of free software. You can use,
12 # modify and/ or redistribute the software under the terms of the CeCILL-B
13 # license as circulated by CEA, CNRS and INRIA at the following URL
14 # http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
15 # or in the file LICENSE.txt.
17 # As a counterpart to the access to the source code and rights to copy,
18 # modify and redistribute granted by the license, users are provided only
19 # with a limited warranty and the software's author, the holder of the
20 # economic rights, and the successive licensors have only limited
23 # The fact that you are presently reading this means that you have had
24 # knowledge of the CeCILL-B license and that you accept its terms.
25 # ------------------------------------------------------------------------
28 #include "vtkTensorsVisu.h"
30 #include "vtkProperty.h"
31 #include "vtkFloatArray.h"
34 //------------------------------------------------------------------------
35 vtkTensorsVisu::vtkTensorsVisu() : vtkVectorsTensorsVisuBase()
37 ss = vtkSphereSource::New();
38 superquadratic = vtkSuperquadricSource::New();
39 cs = vtkCubeSource::New();
40 tg = vtkTensorGlyph::New();
41 pdn = vtkPolyDataNormals::New();
42 pd = vtkPolyData::New();
43 pod = vtkPointData::New();
46 //------------------------------------------------------------------------
47 vtkTensorsVisu::~vtkTensorsVisu()
52 //------------------------------------------------------------------------
53 void vtkTensorsVisu::Process()
55 printf("EED vtkTensorsVisu::Process Start\n");
59 if(GetTypeForm()==2) // source superquadratic
61 superquadratic->SetThetaResolution(20);
62 superquadratic->SetPhiResolution(20);
63 tg->SetSource(superquadratic->GetOutput());
64 superquadratic->Print( std::cout );
66 } else if(GetTypeForm()==1) { // source sphere
67 tg->SetSource(ss->GetOutput());
68 ss->Print( std::cout );
69 } else { // source cube
70 tg->SetSource(cs->GetOutput());
71 cs->Print( std::cout );
75 //GetDataObject()->Print( std::cout );
78 tg->SetInput( GetDataObject() );
80 // tg->SetScaling(25);
82 tg->SetScaleFactor( GetScaleFactor() );
87 // //tg->SetScaleFactor(2);
88 // tg->SetScaleFactor(scale_factor);
91 //The normals are needed to generate the right colors and if
92 // not used some of the glyphs are black.
93 pdn->SetInput(tg->GetOutput());
97 //EED================================
98 //pdm->SetInput( tg->GetOutput());
99 //pdm->SetScalarRange( range );
101 //vtkPolyData *pd = tg->GetOutput();
102 //vtkPointData *pointdata = pd->GetPointData();
103 //vtkDataArray *da = pointdata->GetVectors();
105 //vtkDataArray *nda = vtkFloatArray::New();
108 //pointdata->AddArray( nda );
109 //reader->GetOutput()->GetPointData()->AddArray( nda );
111 //pdm->SetInput(pdn->GetOutput());
112 //pdm->ScalarVisibilityOn();
113 //pdm->SetLookupTable(vLutEED);
114 //pdm->SetColorModeToMapScalars();
115 //pdm->SelectColorArray("color");
116 //pdm->SetScalarModeToUsePointFieldData();
117 //pdm->ImmediateModeRenderingOn();
118 //=====================================
120 _pdm->SetInput( pdn->GetOutput() );
123 // _pdm->ImmediateModeRenderingOn();
124 // _pdm->ScalarVisibilityOn();
125 // _pdm->SetScalarModeToUsePointFieldData();
129 // _pdm->SetScalarModeToUseFieldData();
130 // _pdm->SetScalarModeToUseCellFieldData();
131 // _pdm->SetScalarModeToUsePointFieldData();
132 // _pdm->SetScalarModeToUseCellData();
133 // _pdm->SetScalarModeToUsePointData();
134 // _pdm->SetScalarModeToDefault();
138 //EED _pdm->Update();
140 _actor->SetMapper( _pdm );
141 _actor->GetProperty()->SetOpacity( GetOpacity() );
144 // tg->ColorGlyphsOff();
146 // vtkDataArray* s = tg->GetOutput()->GetPointData()->GetScalars();
147 // if (s) _pdm->SetScalarRange( s->GetRange() );
151 // tg->ExtractEigenvaluesOff();
153 // vtkDataArray* s = tg->GetOutput()->GetPointData()->GetScalars();
154 // if (s) _pdm->SetScalarRange( s->GetRange() );
157 // tg->SetColorModeToEigenvalues();
158 // tg->ThreeGlyphsOn();
160 // vtkDataArray* s = tg->GetOutput()->GetPointData()->GetScalars();
161 // if (s) _pdm->SetScalarRange( s->GetRange() );
164 tg->SetColorModeToEigenvalues();
165 // tg->SetColorModeToScalars();
166 // tg->ThreeGlyphsOn();
167 // tg->SymmetricOn();
169 // vtkDataArray* s = tg->GetOutput()->GetPointData()->GetScalars();
170 // if (s) _pdm->SetScalarRange( s->GetRange() );
174 // tg->ColorGlyphsOn();
175 // tg->ExtractEigenvaluesOn();
176 // tg->SetColorModeToEigenvalues();
180 // tg->ColorGlyphsOff();
181 // tg->ThreeGlyphsOff();
182 // tg->ExtractEigenvaluesOff();
186 // _pdm->SetLookupTable( _LutEED );
187 // _pdm->SelectColorArray( "Normals" );
190 //int tcg = tg->GetColorGlyphs();
191 // tg->SetColorGlyphs(10);
192 // tg->SetColorModeToScalars();
193 //tg->ThreeGlyphsOn();
196 //vtkDataArray* da = vtkDataArray::New();
197 pd = tg->GetOutput();
199 pod = pd->GetPointData();
201 //EED1 if(pod->GetScalars())
203 //EED1 _pdm->SetScalarRange(pod->GetScalars()->GetRange());
205 //pdm->SelectColorArray("GlyphVector");
207 //EED _pdm->Update();
210 //EED1 tg->Print( std::cout );
211 //EED1 printf("EED vtkTensorsVisu::Process ------------------------------------------\n");
212 //EED1 pd->Print( std::cout );
216 vtkPolyData *pd_pdn = pdn->GetOutput();
217 vtkPointData *pointdata_pdn = pd_pdn->GetPointData();
219 vtkPolyData *pd_tg = tg->GetOutput();
220 vtkPointData *pointdata_tg = pd_tg->GetPointData();
222 vtkPolyData *pd_do = (vtkPolyData*)GetDataObject();
223 vtkPointData *pointdata_do = pd_do->GetPointData();
226 //EED1 printf("EED vtkTensorsVisu::Process XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXx\n");
227 //EED1 pointdata->Print( std::cout );
230 int aa = pd_tg->GetNumberOfPieces ();
231 int bb = pd_tg->GetNumberOfVerts ();
232 int cc = pd_tg->GetNumberOfLines ();
233 int dd = pd_tg->GetNumberOfPolys ();
234 int ee = pd_tg->GetNumberOfStrips ();
235 int ff = pd_tg->GetNumberOfCells();
237 printf("EED vtkTensorsVisu::Process pd_tg %d %d %d %d %d %d \n", aa,bb,cc,dd,ee,ff);
240 aa = pd_pdn->GetNumberOfPieces ();
241 bb = pd_pdn->GetNumberOfVerts ();
242 cc = pd_pdn->GetNumberOfLines ();
243 dd = pd_pdn->GetNumberOfPolys ();
244 ee = pd_pdn->GetNumberOfStrips ();
245 ff = pd_pdn->GetNumberOfCells();
247 // printf("EED vtkTensorsVisu::Process pd_pdn %d %d %d %d %d %d \n", aa,bb,cc,dd,ee,ff);
250 vtkPolyData *ttt = (vtkPolyData *)GetDataObject();
251 int numPnts = ttt->GetNumberOfPoints();
252 // printf("EED vtkTensorsVisu::Process numPnts %d \n", numPnts );
254 vtkIdType numSourcePts, numSourceCells;
255 vtkPoints *sourcePts;
256 sourcePts = tg->GetSource()->GetPoints();
257 numSourcePts = sourcePts->GetNumberOfPoints();
258 numSourceCells = tg->GetSource()->GetNumberOfCells();
259 // printf("EED vtkTensorsVisu::Process numSourcePts %d numSourceCells %d\n", numSourcePts, numSourceCells);
261 vtkDataArray *tensorsArray = pointdata_do->GetArray("tensors");
264 vtkFloatArray *newScalarArray = vtkFloatArray::New();
265 newScalarArray->SetName( "NormalsEED" );
266 newScalarArray->SetNumberOfComponents(3);
267 int iNumPts,iNumSourcePts;
270 double sumEigenvalue,sumEigenvalue1,sumEigenvalue2,sumEigenvalue3;
271 double dirx,diry,dirz;
273 for (iNumPts=0;iNumPts<numPnts;iNumPts++)
275 normalValue = tensorsArray->GetTuple9(iNumPts);
276 sumEigenvalue1 = sqrt (normalValue[0]*normalValue[0] + normalValue[1]*normalValue[1] + normalValue[2]*normalValue[2]);
277 sumEigenvalue2 = sqrt (normalValue[3]*normalValue[3] + normalValue[4]*normalValue[4] + normalValue[5]*normalValue[5]);
278 sumEigenvalue3 = sqrt (normalValue[6]*normalValue[6] + normalValue[7]*normalValue[7] + normalValue[8]*normalValue[8]);
280 if ((sumEigenvalue1>=sumEigenvalue2) && (sumEigenvalue1>=sumEigenvalue3))
282 sumEigenvalue=sumEigenvalue1;
285 if ((sumEigenvalue2>sumEigenvalue1) && (sumEigenvalue2>=sumEigenvalue3))
287 sumEigenvalue=sumEigenvalue2;
290 if ((sumEigenvalue3>=sumEigenvalue1) && (sumEigenvalue3>=sumEigenvalue2))
292 sumEigenvalue=sumEigenvalue3;
296 dirx = fabs(normalValue[idEigen+0]/sumEigenvalue);
297 diry = fabs(normalValue[idEigen+1]/sumEigenvalue);
298 dirz = fabs(normalValue[idEigen+2]/sumEigenvalue);
299 //printf("%f %f %f\n",dirx, diry,dirz );
300 //printf("%f %f %f\n",sumEigenvalue1, sumEigenvalue2, sumEigenvalue3 );
301 //printf("%f %f %f\n", normalValue[0], normalValue[1], normalValue[2] );
304 for (iNumSourcePts=0;iNumSourcePts<numSourcePts;iNumSourcePts++)
306 newScalarArray->InsertTuple3( iNormalsEED, dirx,diry,dirz );
308 } // for iNumSourcePts
311 pointdata_tg->RemoveArray( "NormalsEED" );
312 pointdata_tg->AddArray( newScalarArray );
313 pointdata_pdn->Update();
319 _LutEED->SetVectorModeToRGBColors();
320 _pdm->SetLookupTable( _LutEED );
321 _pdm->SelectColorArray( "NormalsEED" );
325 _pdm->ScalarVisibilityOn();
326 _pdm->SetColorModeToMapScalars();
327 // _pdm->SetColorModeToDirectScalars(); // NOT compile
328 _pdm->SetScalarModeToUsePointFieldData();
329 _pdm->ImmediateModeRenderingOn();
332 int i,sizeDa = pointdata_tg->GetNumberOfArrays();
333 printf("EED vtkTensorsVisu::Process sizeDa %d \n", sizeDa );
334 for (i=0;i<sizeDa;i++)
336 int faces = pointdata_tg->GetArray(i)->GetSize() / numPnts;
337 //pointdata_pdn->GetArray(i)->Print( std::cout );
338 printf("EED vtkTensorsVisu::Process-name %d, %s, size=%d %d \n", i, pointdata_tg->GetArrayName(i) , pointdata_tg->GetArray(i)->GetSize(), faces );
350 printf("EED vtkTensorsVisu::Process End.\n");