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;
33 if(bbGetInputVectorsFieldImage() != NULL )
35 vtkImageData *m_VectorImage = bbGetInputVectorsFieldImage();
37 vtkDoubleArray *velocity = vtkDoubleArray::New();
39 std::vector<double> coord = bbGetInputCoordinates();
40 std::vector<double> vec;
46 double ang1 = 0;//Angle 1
47 double ang2 = 0;//ANgle 2
48 double comp = 0;//Component x + y
50 //Find coordinate into the image
51 m_idType = m_VectorImage->FindPoint(coord[0],coord[1],coord[2]);
52 std::cout << m_idType;
54 //Recuperate vector of vectorFieldImage
55 velocity = vtkDoubleArray::SafeDownCast(m_VectorImage->GetPointData()->GetArray("velocity"));
56 //velocity->Print(std::cout);
58 std::cout<< "# components velocity: " << velocity->GetNumberOfComponents()<<std::endl;
59 std::cout<< "# tuples velocity: " << velocity->GetNumberOfTuples()<<std::endl;
61 vec[0] = velocity->GetComponent(m_idType,0);
62 vec[1] = velocity->GetComponent(m_idType,1);
63 vec[2] = velocity->GetComponent(m_idType,2);
65 std::cout << "velocity :" << vec[0] << "," <<vec[1] << ","<< vec[2] << std::endl;
68 //Miss calcule of angles -----
69 comp = sqrt((vec[0]*vec[0])+(vec[1]*vec[1]));
71 ang1 = atan2(vec[2],comp) * 180 / PI;
74 ang2 = atan2(vec[1],vec[0]) * 180 / PI;
77 //m_VectorImage->Print(std::cout);
79 std::cout << "ang1: " << ang1 << std::endl;
80 std::cout << "ang2: " << ang2 << std::endl;
82 bbSetOutputAngle1(ang1);
83 bbSetOutputAngle2(ang2);
84 bbSetOutputVector(vec);
85 bbSetOutputX(coord[0]);
86 bbSetOutputY(coord[1]);
87 bbSetOutputZ(coord[2]);
91 std::cout<< "MLER VectorsExtraction Process END" << std::endl;
95 // 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)
97 void VectorsExtraction::bbUserSetDefaultValues()
100 // SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX
101 // Here we initialize the input 'In' to 0
102 bbSetInputVectorsFieldImage(NULL);
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::bbUserInitializeProcessing()
111 // THE INITIALIZATION METHOD BODY :
113 // but this is where you should allocate the internal/output pointers
119 // 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)
121 void VectorsExtraction::bbUserFinalizeProcessing()
124 // THE FINALIZATION METHOD BODY :
126 // but this is where you should desallocate the internal/output pointers
131 // EO namespace bbcreaVtk