Memory bug

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

diff --git a/bbtk_creaVtk_PKG/src/bbcreaVtkPolyDataNormals.cxx b/bbtk_creaVtk_PKG/src/bbcreaVtkPolyDataNormals.cxx
index 5db5dcb7f72018dbf6081671ab042cab4ab58f4a..5c0ddab5284d55b2e98517e7c520b65d604c6dce 100644 (file)
--- a/bbtk_creaVtk_PKG/src/bbcreaVtkPolyDataNormals.cxx
+++ b/bbtk_creaVtk_PKG/src/bbcreaVtkPolyDataNormals.cxx
@@ -5,7 +5,12 @@
 #include "bbcreaVtkPackage.h"
 
 #include <vtkPolyDataNormals.h>
+//#include <vtkTriangleMeshPointNormals.h>
 
+#include <vtkPolyData.h>
+#include <vtkPointData.h>
+#include <vtkDataArray.h>
+#include <vtkMath.h>
 
 namespace bbcreaVtk
 {
@@ -17,7 +22,6 @@ BBTK_BLACK_BOX_IMPLEMENTATION(PolyDataNormals,bbtk::AtomicBlackBox);
 //===== 
 void PolyDataNormals::Process()
 {
-
 // THE MAIN PROCESSING METHOD BODY
 //   Here we simply set the input 'In' value to the output 'Out'
 //   And print out the output value
@@ -31,6 +35,7 @@ void PolyDataNormals::Process()
 //      (the one provided in the attribute 'type' of the tag 'input')
 //    bbSetOutputOut( bbGetInputIn() );
 //    std::cout << "Output value = " <<bbGetOutputOut() << std::endl;
+    
     if (bbGetInputIn()!=NULL)
     {
         vtkPolyDataNormals* normal = vtkPolyDataNormals::New();
@@ -41,11 +46,97 @@ void PolyDataNormals::Process()
         } else {
             normal->ComputePointNormalsOn();
         }
-        //        normal->SetFeatureAngle(60.0);
+        normal->SetFeatureAngle(360);
+        normal->Update();
+    
+        /*
+        vtkTriangleMeshPointNormals* normal = vtkTriangleMeshPointNormals::New();
+        normal->SetInputData( bbGetInputIn() );
         normal->Update();
+        */
+        
+        // >>>>>>>>>>>>>>>>>>>> Calculate the MeanNormal
+        vtkPolyData     *polydata   = normal->GetOutput();
+        vtkPointData    *pointdata  = polydata->GetPointData();
+        vtkDataArray    *dataarray;
+        double          *pValue;
+        /*
+        int i,size=pointdata->GetNumberOfArrays();
+        for(i=0;i<size;i++)
+        {
+            dataarray=pointdata->GetArray(i);
+            printf("EED creaVtkCallbackPointPicker::Execute dataarray=%s  n=%ld p=%ld\n", dataarray->GetName(),dataarray->GetNumberOfValues() ,polydata->GetNumberOfPoints() );
+        } // for i
+        */
+        dataarray   = pointdata->GetNormals();
+        std::vector<double> meanNormal;
+        meanNormal.push_back(0);
+        meanNormal.push_back(0);
+        meanNormal.push_back(0);
+        if (dataarray!=NULL)
+        {
+            int i , size=dataarray->GetNumberOfTuples();
+            for (i=0; i<size; i++)
+            {
+                pValue        = dataarray->GetTuple3( i );
+                meanNormal[0] = meanNormal[0] + pValue[0];
+                meanNormal[1] = meanNormal[1] + pValue[1];
+                meanNormal[2] = meanNormal[2] + pValue[2];
+            } // for i
+            
+            if (size!=0)
+            {
+                meanNormal[0] = meanNormal[0] / size;
+                meanNormal[1] = meanNormal[1] / size;
+                meanNormal[2] = meanNormal[2] / size;
+                double magnitude = sqrt( meanNormal[0]*meanNormal[0] + meanNormal[1]*meanNormal[1]  + meanNormal[2]*meanNormal[2] );
+                meanNormal[0] = meanNormal[0] / magnitude;
+                meanNormal[1] = meanNormal[1] / magnitude;
+                meanNormal[2] = meanNormal[2] / magnitude;
+            } else {
+                meanNormal[0] = 1;
+                meanNormal[1] = 0;
+                meanNormal[2] = 0;
+            } // if size
+        } // if dataarray
+        // <<<<<<<<<<<<<<<<<<<<< Calculate the MeanNormal
+
+        if ((bbGetInputType()==1) && (dataarray!=NULL) )
+        {
+            double meanN[3];
+            double wxyz[4];
+            double meanNormalresult[3];
+            double ang;
+            double crossResult[3];
+            meanN[0] = meanNormal[0];
+            meanN[1] = meanNormal[1];
+            meanN[2] = meanNormal[2];
+            double magnitude;
+            int i , size=dataarray->GetNumberOfTuples();
+            for (i=0; i<size; i++)
+            {
+                pValue  = dataarray->GetTuple3( i );
+                ang     = vtkMath::AngleBetweenVectors( pValue, meanN );
+                vtkMath::Cross(meanN,pValue,crossResult);
+                wxyz[0] = ang * bbGetInputParam();
+                wxyz[1] = crossResult[0];
+                wxyz[2] = crossResult[1];
+                wxyz[3] = crossResult[2];
+                vtkMath::RotateVectorByWXYZ(meanN, wxyz ,meanNormalresult);
+                magnitude = sqrt(   meanNormalresult[0]*meanNormalresult[0] +  meanNormalresult[1]*meanNormalresult[1] + meanNormalresult[2]* meanNormalresult[2]  );
+                meanNormalresult[0] = meanNormalresult[0] / magnitude;
+                meanNormalresult[1] = meanNormalresult[1] / magnitude;
+                meanNormalresult[2] = meanNormalresult[2] / magnitude;
+                dataarray->SetTuple3( i , meanNormalresult[0], meanNormalresult[1], meanNormalresult[2] );
+            } // for
+        } // if Type==1  &&  dataarray
+        bbSetOutputMeanNormal( meanNormal );
         bbSetOutputOut( normal->GetOutput() );
-    } // In
+    } else {
+        bbSetOutputOut( NULL );
+    }// if In
 }
+
 //===== 
 // 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)
 //===== 
@@ -53,34 +144,32 @@ void PolyDataNormals::bbUserSetDefaultValues()
 {
 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX 
 //    Here we initialize the input 'In' to 0
-   bbSetInputIn(NULL);
+    bbSetInputIn(NULL);
+    bbSetOutputOut( NULL );
 }
+
 //===== 
 // 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)
 //===== 
 void PolyDataNormals::bbUserInitializeProcessing()
 {
-
 //  THE INITIALIZATION METHOD BODY :
 //    Here does nothing 
 //    but this is where you should allocate the internal/output pointers 
-//    if any 
-
-  
+//    if any
 }
+
 //===== 
 // 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)
 //===== 
 void PolyDataNormals::bbUserFinalizeProcessing()
 {
-
 //  THE FINALIZATION METHOD BODY :
 //    Here does nothing 
 //    but this is where you should desallocate the internal/output pointers 
 //    if any
-  
 }
-}
-// EO namespace bbcreaVtk
+
+}// EO namespace bbcreaVtk