2 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
4 #include "bbcreaVtkVectorsExtraction.h"
5 #include "bbcreaVtkPackage.h"
6 #include "vtkDoubleArray.h"
12 BBTK_ADD_BLACK_BOX_TO_PACKAGE(creaVtk,VectorsExtraction)
13 BBTK_BLACK_BOX_IMPLEMENTATION(VectorsExtraction,bbtk::AtomicBlackBox);
15 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
17 void VectorsExtraction::Process()
20 // THE MAIN PROCESSING METHOD BODY
21 // Here we simply set the input 'In' value to the output 'Out'
22 // And print out the output value
23 // INPUT/OUTPUT ACCESSORS ARE OF THE FORM :
24 // void bbSet{Input|Output}NAME(const TYPE&)
25 // const TYPE& bbGet{Input|Output}NAME() const
27 // * NAME is the name of the input/output
28 // (the one provided in the attribute 'name' of the tag 'input')
29 // * TYPE is the C++ type of the input/output
30 // (the one provided in the attribute 'type' of the tag 'input')
31 std::cout<< "MLER VectorsExtraction Process Start" << std::endl;
32 if(bbGetInputVectorsFieldImage() != NULL )
34 vtkImageData *m_VectorImage = bbGetInputVectorsFieldImage();
36 vtkDoubleArray *velocity = vtkDoubleArray::New();
37 std::vector<double> coord = bbGetInputCoordinates();
38 std::vector<double> vec;
42 double ang1 = 0;//Angle 1
43 double ang2 = 0;//ANgle 2
44 double comp = 0;//Component x + y
45 //Find coordinate into the image
46 std::cout << "ups ." << std::endl;
47 m_idType = m_VectorImage->FindPoint(coord[0],coord[1],coord[2]);
48 std::cout << "m_idType" << m_idType << std::endl;
49 //Recuperate vector of vectorFieldImage
50 m_VectorImage->Print(std::cout);
53 int sizeArrays= m_VectorImage->GetPointData()->GetNumberOfArrays();
54 for (int iArray=0; iArray<sizeArrays;iArray++)
56 printf("%s\n", m_VectorImage->GetPointData()->GetArrayName(iArray) );
59 vtkDataArray *dataArray = m_VectorImage->GetPointData()->GetArray("velocity");
62 printf("DataArrya ok \n");
64 vecData=dataArray->GetTuple3(m_idType);
65 printf("info data: %f %f %f \n", vecData[0], vecData[1], vecData[2] );
69 bbSetOutputVector(vec);
73 velocity = vtkDoubleArray::SafeDownCast(m_VectorImage->GetPointData()->GetArray("velocity"));
74 //velocity->Print(std::cout);
77 std::cout<< "# components velocity: " << velocity->GetNumberOfComponents()<<std::endl;
78 std::cout<< "# tuples velocity: " << velocity->GetNumberOfTuples()<<std::endl;
79 vec[0] = velocity->GetComponent(m_idType,0);
80 vec[1] = velocity->GetComponent(m_idType,1);
81 vec[2] = velocity->GetComponent(m_idType,2);
82 std::cout << "velocity :" << vec[0] << "," <<vec[1] << ","<< vec[2] << std::endl;
83 //Miss calcule of angles -----
84 comp = sqrt((vec[0]*vec[0])+(vec[1]*vec[1]));
85 ang1 = atan2(vec[2],comp) * 180 / PI;
86 ang2 = atan2(vec[1],vec[0]) * 180 / PI;
87 //m_VectorImage->Print(std::cout);
88 std::cout << "ang1: " << ang1 << std::endl;
89 std::cout << "ang2: " << ang2 << std::endl;
90 bbSetOutputAngle1(ang1);
91 bbSetOutputAngle2(ang2);
92 bbSetOutputVector(vec);
93 bbSetOutputX(coord[0]);
94 bbSetOutputY(coord[1]);
95 bbSetOutputZ(coord[2]);
98 bbSetOutputAngle1(ang1);
99 bbSetOutputAngle2(ang2);
100 bbSetOutputVector(vec);
103 std::cout<< "MLER VectorsExtraction Process END" << std::endl;
106 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
108 void VectorsExtraction::bbUserSetDefaultValues()
111 // SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX
112 // Here we initialize the input 'In' to 0
113 bbSetInputVectorsFieldImage(NULL);
117 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
119 void VectorsExtraction::bbUserInitializeProcessing()
122 // THE INITIALIZATION METHOD BODY :
124 // but this is where you should allocate the internal/output pointers
130 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
132 void VectorsExtraction::bbUserFinalizeProcessing()
135 // THE FINALIZATION METHOD BODY :
137 // but this is where you should desallocate the internal/output pointers
142 // EO namespace bbcreaVtk