#2671 creaVtk Feature New Normal - VectorsExtraction

[creaVtk.git] / bbtk_creaVtk_PKG / src / bbcreaVtkVectorsExtraction.cxx

diff --git a/bbtk_creaVtk_PKG/src/bbcreaVtkVectorsExtraction.cxx b/bbtk_creaVtk_PKG/src/bbcreaVtkVectorsExtraction.cxx
index f26075e66e2bc69f0f0034a946485606b7004ddc..38f2ffb1e9c8f34d1712daf04c3578bad6a550a8 100644 (file)
--- a/bbtk_creaVtk_PKG/src/bbcreaVtkVectorsExtraction.cxx
+++ b/bbtk_creaVtk_PKG/src/bbcreaVtkVectorsExtraction.cxx
@@ -5,6 +5,7 @@
 #include "bbcreaVtkPackage.h"
 #include "vtkDoubleArray.h"
 
+
 namespace bbcreaVtk
 {
 
@@ -31,24 +32,59 @@ void VectorsExtraction::Process()
 
     if(bbGetInputVectorsFieldImage() != NULL )
        {
-               vtkImageData* m_VectorImage = bbGetInputVectorsFieldImage();
+               vtkImageData         *m_VectorImage = bbGetInputVectorsFieldImage();
+               vtkIdType             m_idType;
+      vtkDoubleArray       *velocity      = vtkDoubleArray::New();
+
+      std::vector<double> coord           = bbGetInputCoordinates();
+      std::vector<double> vec;
+
+      vec.push_back(0);
+      vec.push_back(0);
+      vec.push_back(0);
 
-               std::vector<double> coord = bbGetInputCoordinates();
+      double ang1 = 0;//Angle 1
+      double ang2 = 0;//ANgle 2
+      double comp = 0;//Component x + y
 
-               vtkDoubleArray* velocity = vtkDoubleArray::New();
+      //Find coordinate into the image      
+      m_idType = m_VectorImage->FindPoint(coord[0],coord[1],coord[2]);
+      std::cout << m_idType;
+
+      //Recuperate vector of vectorFieldImage
                velocity = vtkDoubleArray::SafeDownCast(m_VectorImage->GetPointData()->GetArray("velocity"));
+      //velocity->Print(std::cout);
+
+      std::cout<< "# components velocity: " << velocity->GetNumberOfComponents()<<std::endl;
+      std::cout<< "# tuples velocity: " << velocity->GetNumberOfTuples()<<std::endl;
+
+      vec[0] = velocity->GetComponent(m_idType,0);
+      vec[1] = velocity->GetComponent(m_idType,1);
+      vec[2] = velocity->GetComponent(m_idType,2);
+
+      std::cout << "velocity :" << vec[0] << "," <<vec[1] << ","<< vec[2] << std::endl;
+
+
+      //Miss calcule of angles -----
+      comp = sqrt((vec[0]*vec[0])+(vec[1]*vec[1]));
+      
+      ang1 = atan2(vec[2],comp) * 180 / PI;
+
 
-               std::cout<< "# components velocity" << velocity->GetNumberOfComponents()<<std::endl;
+      ang2 = atan2(vec[1],vec[0]) * 180 / PI;
 
-               double ang1 = 0;
-               double ang2 = 0;
+      
+               //m_VectorImage->Print(std::cout);
 
-               ang1 = 1;
-               ang2 = 2;
-               m_VectorImage->Print(std::cout);
+      std::cout << "ang1: " << ang1 << std::endl;
+      std::cout << "ang2: " << ang2 << std::endl;
 
-               bbSetOutputAngle1(ang1);
-               bbSetOutputAngle2(ang2);
+       bbSetOutputAngle1(ang1);
+       bbSetOutputAngle2(ang2);
+      bbSetOutputVector(vec);
+      bbSetOutputX(coord[0]);
+      bbSetOutputY(coord[1]);
+      bbSetOutputZ(coord[2]);
 
        }// if image != NULL