]> Creatis software - creaVtk.git/blobdiff - bbtk_creaVtk_PKG/src/bbcreaVtkCreateMeshFromPoints.cxx
#3501 new box GlyphPolyDataNormals (Pablo)
[creaVtk.git] / bbtk_creaVtk_PKG / src / bbcreaVtkCreateMeshFromPoints.cxx
index 534d0029101cb8ca4cce1d167bf095e7721817ac..a6948c45ee344c4fe7d6f3966b8ec98ba0df7242 100644 (file)
@@ -5,6 +5,7 @@
 #include "bbcreaVtkPackage.h"
 
 #include "vtkTriangleStrip.h"
+#include <vtkMath.h>
 
 namespace bbcreaVtk
 {
@@ -38,10 +39,9 @@ void CreateMeshFromPoints::Process()
                std::vector<int> lstIndexs              = bbGetInputLstIndexs();
                if ( (lstIndexs.size()<=1) || (lstX.size()==0) || (lstX.size()!=lstY.size()) || (lstY.size()!=lstZ.size()) )
                {
-                       printf("Warnning! CreateMeshFromPoints::Process: List of points X Y Z  and LstIndexes is not correct\n");
+                       printf("PG CreateMeshFromPoints::Process: List of points X Y Z  and LstIndexes is not correct\n");
                        bbSetOutputOut(NULL);
                } else  {
-            printf("EED CreateMeshFromPoints::Process  initial=%d  final=%d\n" ,  lstIndexs[0] ,   lstIndexs[  lstIndexs.size()-1 ] );
                        int ii,sizeSegment1,sizeSegment2;
                        int endSegment;
 //                     vtkSmartPointer<vtkPoints> points = vtkSmartPointer<vtkPoints>::New();
@@ -83,7 +83,21 @@ void CreateMeshFromPoints::Process()
                                } // for ii 
                                iGeneral=iGeneral+sizeSegment1;
                                cells->InsertNextCell(triangleStrip);
-                       } //for  LstIndexs         
+                       } //for  LstIndexs
+
+                       
+                       if(bbGetInputCloseSurface()){                   
+                               //false = Open Contour
+                               //true = Closed Contour
+                               if(bbGetInputOpenClose()){                      
+                                       CloseContourSides(lstIndexs, true);
+                               }
+                               else{
+                                       CloseOpenContourSurface(lstIndexs);
+                               }
+                       }
+                       
+                       
 //                     vtkPolyData *polydata = vtkPolyData::New();
                        if (polydata!=NULL) polydata->Delete();
                        polydata = vtkPolyData::New();
@@ -101,6 +115,204 @@ void CreateMeshFromPoints::Process()
                        triangle->Update();
                        bbSetOutputOut( clean->GetOutput() );
                }// if listXYZ size
+               //printf("PG CreateMeshFromPoints::Process: End\n");
+}
+
+/**
+* Closes the sides of the contour
+* iterates in one way or the other, depending on the order of the points and calculated vectors.
+* uPointOrder: Points are order in a U shape
+* lstIndexs: number of points on each spline
+*/
+void CreateMeshFromPoints::CloseContourSides(std::vector<int> lstIndexs, bool uPointOrder){
+       int     sizeLstIdexes = lstIndexs.size();
+       int sizePoints = bbGetInputLstX().size();
+
+       int firstIndex, end, centroidId, numPointsFace, contraryId;
+       int increment = uPointOrder?1:sizeLstIdexes;
+       double centroid[3];
+       
+       for(int facesIdx = 0; facesIdx < 2; facesIdx++){
+               std::fill(std::begin(centroid), std::end(centroid), 0);
+               if(facesIdx == 0){
+                       firstIndex = 0;
+                       numPointsFace = uPointOrder?lstIndexs[0]: sizeLstIdexes;
+                       end = uPointOrder?firstIndex + numPointsFace:sizePoints - lstIndexs[sizeLstIdexes - 1] + 1;
+                       contraryId = sizePoints-1;
+               }
+               else{
+                       firstIndex = uPointOrder?sizePoints - lstIndexs[sizeLstIdexes-1]:lstIndexs[0]-1;
+                       numPointsFace = uPointOrder?lstIndexs[sizeLstIdexes-1]:sizeLstIdexes;
+                       end = uPointOrder?firstIndex + numPointsFace:sizePoints;
+                       contraryId = 0;
+               }
+               if(numPointsFace > 1){
+                       double lastPoint[3] = {};
+                       bool validCentroid = CalcValidCentroid(centroid, firstIndex, end, increment, numPointsFace);
+                       if(validCentroid){
+                               bool normalOrder = isPointingCorrectly(firstIndex, firstIndex+increment, centroid, contraryId);
+                               centroidId = points->InsertNextPoint(centroid[0], centroid[1], centroid[2]);
+                               vtkSmartPointer<vtkTriangleStrip> triangleStrip1 = vtkSmartPointer<vtkTriangleStrip>::New();
+                               triangleStrip1->GetPointIds()->SetNumberOfIds(numPointsFace*2+1);
+                               if(normalOrder){        //(facesIdx == 0 && uPointOrder) || (facesIdx == 1 && !uPointOrder)){
+                                       int initial = firstIndex;
+                                       int triangleIndex = 0;
+                                       for(int index = initial; index < end; index+=increment){
+                                               triangleStrip1->GetPointIds()->SetId(triangleIndex,index);
+                                               if(index+increment >= end){
+                                                       triangleStrip1->GetPointIds()->SetId(triangleIndex+1,initial);
+                                                       triangleStrip1->GetPointIds()->SetId(triangleIndex+2,centroidId);
+                                               }
+                                               else{
+                                                       triangleStrip1->GetPointIds()->SetId(triangleIndex+1,centroidId);
+                                               }
+                                               triangleIndex+=2;
+                                       }
+                                       cells->InsertNextCell(triangleStrip1);
+                               }
+                               else{
+                                       int initial = firstIndex-1;
+                                       int triangleIndex = 0;
+                                       int triangleStripStart = end-1;
+                                       for(int index = triangleStripStart; index > initial; index-=increment){
+                                               triangleStrip1->GetPointIds()->SetId(triangleIndex,index);
+                                               if(index-increment <= initial){
+                                                       triangleStrip1->GetPointIds()->SetId(triangleIndex+1,triangleStripStart);
+                                                       triangleStrip1->GetPointIds()->SetId(triangleIndex+2,centroidId);
+                                               }
+                                               else{
+                                                       triangleStrip1->GetPointIds()->SetId(triangleIndex+1,centroidId);
+                                               }
+                                               triangleIndex+=2;
+                                       }
+                                       cells->InsertNextCell(triangleStrip1);
+                               }
+                       }//if validCentroid
+               }//if numPointsFace
+       }//for facesIdx
+
+}
+
+/**
+* Checks if the normal from firstPointId, secPointId and centroid points away 
+* from the vector centroid to contrPointId.
+* Used to check that the order used to create the new polygons is correct.
+*/
+bool CreateMeshFromPoints::isPointingCorrectly( int firstPointId, int secPointId, double(&centroid)[3], int contrPointId) {
+
+       double firstPoint[3], secPoint[3], contrPoint[3];
+       points->GetPoint(firstPointId, firstPoint);
+       points->GetPoint(secPointId, secPoint);
+       
+       double firstVect[3], secVect[3], normal[3], contrVect[3];
+       
+       vtkMath::Subtract(firstPoint, centroid, firstVect);
+       vtkMath::Subtract(secPoint, centroid, secVect);
+       
+       points->GetPoint(contrPointId, contrPoint);
+       vtkMath::Subtract(contrPoint, centroid, contrVect);
+       
+       vtkMath::Cross(firstVect, secVect, normal);
+       double dotCalc;
+       dotCalc = vtkMath::Dot(normal, contrVect);
+       
+       return dotCalc<0;
+}
+
+/**
+* Checks if points on each side of the shapes represent a curve.
+*/
+bool CreateMeshFromPoints::CheckLinePointOrder(){
+       int sizePoints = bbGetInputLstX().size();
+       std::vector<int> lstIndexs = bbGetInputLstIndexs();
+       double point1[3], point2[3], point3[3];
+       double center[3];
+       double firstRadiusSum = 0;
+       double secondRadiusSum = 0;
+       for(int i = 0; i < lstIndexs[0] && lstIndexs[0] > 3; i+=3){
+               if(i+3 <= lstIndexs[0]){
+                       points->GetPoint(i, point1);
+                       points->GetPoint(i+1, point2);
+                       points->GetPoint(i+2, point3);
+                       firstRadiusSum += vtkMath::Solve3PointCircle(point1, point2, point3, center);
+               }
+       }
+       
+       for(int i = 0; i < sizePoints && lstIndexs.size() > 3; i+=(lstIndexs[0]*3)){
+               if(i+(3*lstIndexs[0]) <= sizePoints){
+                       points->GetPoint(i, point1);
+                       points->GetPoint(i+lstIndexs[0], point2);
+                       points->GetPoint(i+(2*lstIndexs[0]), point3);
+                       secondRadiusSum += vtkMath::Solve3PointCircle(point1, point2, point3, center);
+               }
+       }
+       
+       return firstRadiusSum > secondRadiusSum;
+}
+
+/**
+* Closes an open contour
+* lstIndexs: number of points on each spline
+*/
+void CreateMeshFromPoints::CloseOpenContourSurface(std::vector<int> lstIndexs){
+       int     sizeLstIdexes = lstIndexs.size();
+       int sizeLstX = bbGetInputLstX().size();
+       bool linePointOrder = CheckLinePointOrder();
+       CloseContourSides(lstIndexs, !linePointOrder);
+       CloseContourBottom(!linePointOrder);
+}
+
+/**
+* Calculates centroid.
+* centroid: array to store calculation
+* start: start index of points to use
+* end: end index of points to use
+* increment: increment to be used in point iteration
+* numPoints: number of points used to calculate the centroid.
+* Returns a bool indicating the validity of the centroid calculated.
+* False = invalid centroid = all points are the same.
+*/
+bool CreateMeshFromPoints::CalcValidCentroid(double(&centroid)[3], int start, int end, int increment, int numPoints){
+       double currPoint[3] = {}, prevPoint[3] = {};
+       bool samePoint = true;
+       for(int i = start; i < end; i+=increment){
+               points->GetPoint(i, currPoint);
+               if(samePoint && (currPoint[0] != prevPoint[0] || currPoint[1] != prevPoint[1] || currPoint[2] != prevPoint[2])){
+                       samePoint = false;
+               }
+               centroid[0] += currPoint[0];
+               centroid[1] += currPoint[1];
+               centroid[2] += currPoint[2];
+               std::copy(std::begin(currPoint), std::end(currPoint), prevPoint);
+       }
+       centroid[0] /= numPoints;
+       centroid[1] /= numPoints;
+       centroid[2] /= numPoints;
+       
+       return !samePoint;
+}
+
+/**
+* Closes the bottom of the given countour.
+* Should only be used when its an open contour.
+* uPointOrder: points are ordered in U shape
+*/
+void CreateMeshFromPoints::CloseContourBottom(bool uPointOrder){
+       std::vector<int> lstIndexs = bbGetInputLstIndexs();
+       int sizeLstIdexes = lstIndexs.size();
+       int sizeLstX = bbGetInputLstX().size();
+       
+       vtkSmartPointer<vtkTriangleStrip> triangleStripBottom = vtkSmartPointer<vtkTriangleStrip>::New();
+       triangleStripBottom->GetPointIds()->SetNumberOfIds(sizeLstIdexes*2);
+       int triangleIndex = 0, currentId = 0, nextId = 0;
+       for(int splineIndex = 0; splineIndex < sizeLstIdexes;splineIndex++){
+               triangleStripBottom->GetPointIds()->SetId(triangleIndex, currentId);
+               nextId = uPointOrder?currentId + lstIndexs[splineIndex] - 1:sizeLstX - sizeLstIdexes + splineIndex;
+               triangleStripBottom->GetPointIds()->SetId(triangleIndex+1, nextId);
+               triangleIndex+=2;
+               currentId = uPointOrder?nextId + 1: splineIndex+1;
+       }
+       cells->InsertNextCell(triangleStripBottom);
 }
 
 //===== 
@@ -112,6 +324,8 @@ void CreateMeshFromPoints::bbUserSetDefaultValues()
 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX 
 //    Here we initialize the input 'In' to 0
 //   bbSetInputIn(0);
+       bbSetInputCloseSurface(false);
+       bbSetInputOpenClose(false);
        points          = NULL;
        cells           = NULL;
        polydata        = NULL;