#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..e753ae38515c463e790784035831ee3a2a574c84 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,46 @@ 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;
+
+      double ang1 = 0;//Angle 1
+      double ang2 = 0;//ANgle 2
+
+     // double  *v = static_cast<double *> (m_VectorImage->GetScalarPointer(coord[0],coord[1],coord[2]));
 
-               std::vector<double> coord = bbGetInputCoordinates();
 
-               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;
 
-               std::cout<< "# components velocity" << velocity->GetNumberOfComponents()<<std::endl;
 
-               double ang1 = 0;
-               double ang2 = 0;
+      //Miss calcule of angles -----
 
-               ang1 = 1;
-               ang2 = 2;
+      ang1 = vec[0];
+               ang2 = vec[1];
                m_VectorImage->Print(std::cout);
 
                bbSetOutputAngle1(ang1);
                bbSetOutputAngle2(ang2);
+      bbSetOutputVector(vec);
 
        }// if image != NULL