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 "vtkDoubleArray.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");
58 if(GetTypeForm()==2) // source superquadratic
60 superquadratic->SetThetaResolution(20);
61 superquadratic->SetPhiResolution(20);
62 #if VTK_MAJOR_VERSION <= 5
63 tg->SetSource(superquadratic->GetOutput());
65 tg->SetSourceData(superquadratic->GetOutput());
67 } else if(GetTypeForm()==1) { // source sphere
68 #if VTK_MAJOR_VERSION <= 5
69 tg->SetSource(ss->GetOutput());
71 tg->SetSourceData(ss->GetOutput());
73 } else { // source cube
74 #if VTK_MAJOR_VERSION <= 5
75 tg->SetSource(cs->GetOutput());
77 tg->SetSourceData(cs->GetOutput());
81 #if VTK_MAJOR_VERSION <= 5
82 tg->SetInput( GetDataObject() );
84 tg->SetInputData( GetDataObject() );
87 // tg->SetScaling(25);
88 tg->SetScaleFactor( GetScaleFactor() );
90 printf("EED vtkTensorsVisu::Process() 1\n");
93 //The normals are needed to generate the right colors and if
94 // not used some of the glyphs are black.
95 //EED 2017-01-01 Migration VTK7
96 #if VTK_MAJOR_VERSION <= 5
97 pdn->SetInput(tg->GetOutput());
98 _pdm->SetInput( pdn->GetOutput() );
100 printf("EED vtkTensorsVisu::Process() 1.1\n");
101 pdn->SetInputData(tg->GetOutput());
102 printf("EED vtkTensorsVisu::Process() 1.2\n");
103 _pdm->SetInputData( pdn->GetOutput() );
106 printf("EED vtkTensorsVisu::Process() 1.3\n");
107 _actor->SetMapper( _pdm );
108 printf("EED vtkTensorsVisu::Process() 1.4\n");
109 _actor->GetProperty()->SetOpacity( GetOpacity() );
114 printf("EED vtkTensorsVisu::Process() 1.5\n");
115 tg->SetColorModeToEigenvalues();
116 // tg->SetColorModeToScalars();
117 printf("EED vtkTensorsVisu::Process() 1.6\n");
120 printf("EED vtkTensorsVisu::Process() 1.7\n");
124 //EED 2017-01-01 Migration VTK7
125 #if VTK_MAJOR_VERSION <= 5
126 pd = tg->GetOutput();
130 pd = tg->GetOutput();
133 pod = pd->GetPointData();
134 printf("EED vtkTensorsVisu::Process() 1.8\n");
136 printf("EED vtkTensorsVisu::Process() 1.9\n");
139 vtkPolyData *pd_pdn = pdn->GetOutput();
140 vtkPointData *pointdata_pdn = pd_pdn->GetPointData();
142 vtkPolyData *pd_tg = tg->GetOutput();
143 vtkPointData *pointdata_tg = pd_tg->GetPointData();
145 printf("EED vtkTensorsVisu::Process() 1.20\n");
146 vtkPolyData *pd_do = (vtkPolyData*)GetDataObject();
147 vtkPointData *pointdata_do = pd_do->GetPointData();
149 int numPnts = ((vtkPolyData *)GetDataObject())->GetNumberOfPoints();
151 vtkIdType numSourcePts;
152 // vtkIdType numSourceCells;
154 printf("EED vtkTensorsVisu::Process() 1.22\n");
155 vtkPoints *sourcePts = tg->GetSource()->GetPoints();
156 printf("EED vtkTensorsVisu::Process() 1.23\n");
159 numSourcePts = sourcePts->GetNumberOfPoints();
160 // numSourceCells = tg->GetSource()->GetNumberOfCells();
163 printf("EED vtkTensorsVisu::Process() 2\n");
166 vtkDataArray *tensorsArray = pointdata_do->GetArray("tensors");
169 vtkDoubleArray *newScalarArray = vtkDoubleArray::New();
170 newScalarArray->SetName( "NormalsEED" );
171 newScalarArray->SetNumberOfComponents(3);
172 int iNumPts,iNumSourcePts;
175 double sumEigenvalue,sumEigenvalue1,sumEigenvalue2,sumEigenvalue3;
176 unsigned char dirx,diry,dirz;
178 for (iNumPts=0;iNumPts<numPnts;iNumPts++)
180 normalValue = tensorsArray->GetTuple9(iNumPts);
181 sumEigenvalue1 = sqrt (normalValue[0]*normalValue[0] + normalValue[1]*normalValue[1] + normalValue[2]*normalValue[2]);
182 sumEigenvalue2 = sqrt (normalValue[3]*normalValue[3] + normalValue[4]*normalValue[4] + normalValue[5]*normalValue[5]);
183 sumEigenvalue3 = sqrt (normalValue[6]*normalValue[6] + normalValue[7]*normalValue[7] + normalValue[8]*normalValue[8]);
185 if ((sumEigenvalue1>=sumEigenvalue2) && (sumEigenvalue1>=sumEigenvalue3))
187 sumEigenvalue=sumEigenvalue1;
190 if ((sumEigenvalue2>sumEigenvalue1) && (sumEigenvalue2>=sumEigenvalue3))
192 sumEigenvalue=sumEigenvalue2;
195 if ((sumEigenvalue3>=sumEigenvalue1) && (sumEigenvalue3>=sumEigenvalue2))
197 sumEigenvalue=sumEigenvalue3;
201 dirx = (unsigned char)( 255*fabs(normalValue[idEigen+0]/sumEigenvalue) );
202 diry = (unsigned char)( 255*fabs(normalValue[idEigen+1]/sumEigenvalue) );
203 dirz = (unsigned char)( 255*fabs(normalValue[idEigen+2]/sumEigenvalue) );
205 for (iNumSourcePts=0;iNumSourcePts<numSourcePts;iNumSourcePts++)
207 // newScalarArray->InsertTuple3( iNormalsEED, dirx,diry,dirz);
208 newScalarArray->InsertTuple3( iNormalsEED, normalValue[idEigen+0],normalValue[idEigen+1],normalValue[idEigen+2] );
210 } // for iNumSourcePts
215 pointdata_tg->RemoveArray( "NormalsEED" );
216 pointdata_tg->AddArray( newScalarArray );
217 // pointdata_pdn->Update();
225 printf("EED vtkTensorsVisu::Process() 3\n");
227 _pdm->ScalarVisibilityOn();
228 _pdm->SetColorModeToMapScalars();
229 //_pdm->SetColorModeToDefault();
230 // _pdm->SetColorModeToDirectScalars(); // NOT compile
231 _pdm->SetScalarModeToUsePointFieldData();
232 _pdm->ImmediateModeRenderingOn();
234 _LutEED->SetVectorModeToComponent();
235 //_LutEED->SetVectorModeToRGBColors();
236 //_LutEED->SetVectorModeToMagnitud();
237 _pdm->SetLookupTable( _LutEED );
238 _pdm->SelectColorArray( "NormalsEED" );
240 //EED 2017-01-01 Migration VTK7
241 #if VTK_MAJOR_VERSION <= 5
248 int i,sizeDa = pointdata_tg->GetNumberOfArrays();
249 printf("EED vtkTensorsVisu::Process sizeDa %d \n", sizeDa );
250 for (i=0;i<sizeDa;i++)
252 int faces = pointdata_tg->GetArray(i)->GetSize() / numPnts;
253 //pointdata_pdn->GetArray(i)->Print( std::cout );
254 printf("EED vtkTensorsVisu::Process-name %d, %s, size=%d %d \n", i, pointdata_tg->GetArrayName(i) , pointdata_tg->GetArray(i)->GetSize(), faces );
261 printf("EED vtkTensorsVisu::Process() 4\n");
265 printf("EED vtkTensorsVisu::Process() END\n");