bbtk/src/bbcreaMaracasVisuShowNPoints_Tools.cxx

   1 //=====
   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)
   3 //=====
   4 #include "bbcreaMaracasVisuShowNPoints_Tools.h"
   5 #include "bbcreaMaracasVisuPackage.h"
   6
   7 #include <vtkPointData.h>
   8 #include <vtkDataArray.h>
   9 #include <vtkMath.h>
  10
  11
  12 namespace bbcreaMaracasVisu
  13 {
  14
  15 BBTK_ADD_BLACK_BOX_TO_PACKAGE(creaMaracasVisu,ShowNPoints_Tools)
  16 BBTK_BLACK_BOX_IMPLEMENTATION(ShowNPoints_Tools,bbtk::AtomicBlackBox);
  17
  18
  19
  20 void ShowNPoints_Tools::NearestPointToMesh( vtkPoints *points,vtkStaticPointLocator *pointLocator,double *spc,double *p,double *pM)
  21 {
  22     p[0]  = p[0] * spc[0];
  23     p[1]  = p[1] * spc[1];
  24     p[2]  = p[2] * spc[2];
  25     points->GetPoint( pointLocator->FindClosestPoint(p) , pM );
  26     pM[0] = pM[0] / spc[0];
  27     pM[1] = pM[1] / spc[1];
  28     pM[2] = pM[2] / spc[2];
  29 }
  30
  31 void ShowNPoints_Tools::CreatePatch_3points()
  32 {
  33         //printf("PG ShowNPoints_Tools::CreatePatch_3points  Entered patch 3 points !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n");
  34         WidgetShowNPoints* wsp = bbGetInputWidgetShowNPoints();
  35
  36         std::vector<double> lstX = wsp->GetModelShowNPoints()->GetLstPointsX();
  37         std::vector<double> lstY = wsp->GetModelShowNPoints()->GetLstPointsY();
  38         std::vector<double> lstZ = wsp->GetModelShowNPoints()->GetLstPointsZ();
  39
  40         double pA[3];
  41     pA[0] = lstX[0];
  42     pA[1] = lstY[0];
  43     pA[2] = lstZ[0];
  44     double pB[3];
  45     pB[0] = lstX[1];
  46     pB[1] = lstY[1];
  47     pB[2] = lstZ[1];
  48     double pC[3];
  49     pC[0] = lstX[2];
  50     pC[1] = lstY[2];
  51     pC[2] = lstZ[2];
  52
  53         double v[3];
  54         double u[3];
  55         vtkMath::Subtract(pC,pA,v);
  56         vtkMath::Subtract(pB,pA,u);
  57
  58         /**point in AC closest to B.
  59         *formula t=(V)dot(U)/(V)dot(V)
  60         */
  61         double t = vtkMath::Dot(v,u)/vtkMath::Dot(v,v);
  62         /**nP found point
  63         *formula A+t(C-A) -> A+tV
  64         */
  65         vtkMath::MultiplyScalar(v, t);
  66         double nP[3];
  67         vtkMath::Add(pA, v, nP);
  68
  69         //calculate direction vector from found point to B
  70         double dirVector[3];
  71         vtkMath::Subtract(pB, nP, dirVector);
  72
  73         double pointAdd[3];
  74         double pointSub[3];
  75
  76         //Add and subtract direction vector to A and C to find the 4 points to create the patch,
  77
  78         std::vector<double> resListX;
  79         std::vector<double> resListY;
  80         std::vector<double> resListZ;
  81
  82         vtkMath::Add(pA, dirVector, pointAdd);
  83         vtkMath::Subtract(pA, dirVector, pointSub);
  84
  85         resListX.push_back(pointSub[0]);
  86         resListY.push_back(pointSub[1]);
  87         resListZ.push_back(pointSub[2]);
  88
  89         resListX.push_back(pointAdd[0]);
  90         resListY.push_back(pointAdd[1]);
  91         resListZ.push_back(pointAdd[2]);
  92
  93
  94         vtkMath::Add(pC, dirVector, pointAdd);
  95         vtkMath::Subtract(pC, dirVector, pointSub);
  96
  97         resListX.push_back(pointAdd[0]);
  98         resListY.push_back(pointAdd[1]);
  99         resListZ.push_back(pointAdd[2]);
 100
 101         resListX.push_back(pointSub[0]);
 102         resListY.push_back(pointSub[1]);
 103         resListZ.push_back(pointSub[2]);
 104
 105         //Create patch given the previously calculated points (4 points)
 106         CreatePatch_Points(resListX, resListY, resListZ);
 107 }
 108
 109 void ShowNPoints_Tools::CreatePatch_4points()
 110 {
 111         WidgetShowNPoints* wsp = bbGetInputWidgetShowNPoints();
 112
 113         std::vector<double> lstX = wsp->GetModelShowNPoints()->GetLstPointsX();
 114         std::vector<double> lstY = wsp->GetModelShowNPoints()->GetLstPointsY();
 115         std::vector<double> lstZ = wsp->GetModelShowNPoints()->GetLstPointsZ();
 116         CreatePatch_Points(lstX, lstY, lstZ);
 117 }
 118
 119 void ShowNPoints_Tools::InitCreatePatch_Points()
 120 {
 121         WidgetShowNPoints* wsp = bbGetInputWidgetShowNPoints();
 122         wsp->StopAutoAddPoints();
 123     wsp->StopTrackPoint();
 124         if((wsp->GetLstModelShowNPointsSize()==1) && (bbGetInputMesh()!=NULL )){
 125                 if(wsp->GetModelShowNPoints()->GetLstPointsSize()==4){
 126                         CreatePatch_4points();
 127                 }
 128                 else if(wsp->GetModelShowNPoints()->GetLstPointsSize()==3){
 129                         CreatePatch_3points();
 130                 }
 131                 else{
 132                         printf("PG ShowNPoints_Tools::CreatePatch_Npoints  Warning patch not apply. Number of points is invalid. Group should have 3 or 4 points\n");
 133                 }
 134         }
 135         else{
 136                 printf("PG ShowNPoints_Tools::CreatePatch_Npoints  Warning patch not apply. groups or mesh invalid. Need 1 group of 3 or 4 points, and a mesh\n");
 137         }
 138 }
 139
 140 void ShowNPoints_Tools::CreatePatch_Points(std::vector<double> lstX, std::vector<double> lstY, std::vector<double> lstZ)
 141 {
 142     WidgetShowNPoints* wsp = bbGetInputWidgetShowNPoints();
 143     std::vector<long int>   lstIdNormalSurface;
 144     double                  spc[3];
 145                             spc[0]          = bbGetInputSpacing()[0];
 146                             spc[1]          = bbGetInputSpacing()[1];
 147                             spc[2]          = bbGetInputSpacing()[2];
 148     vtkPoints               *points         = bbGetInputMesh()->GetPoints();
 149     vtkStaticPointLocator   *pointLocator   = vtkStaticPointLocator::New();
 150     pointLocator->SetDataSet( bbGetInputMesh() );
 151     pointLocator->BuildLocator();
 152     wsp->StopTrackPoint();
 153     double p[3],pM[3];
 154     double dx,dy,dz;
 155
 156     wsp->DeleteAllPoints_();
 157     //wsp->ErasePoint( 0 );
 158     //wsp->ErasePoint( 0 );
 159     //wsp->ErasePoint( 0 );
 160     //wsp->ErasePoint( 0 );
 161     // --- Group 0 ---
 162     dx = lstX[1]-lstX[0];
 163     dy = lstY[1]-lstY[0];
 164     dz = lstZ[1]-lstZ[0];
 165     //
 166     //
 167     double part = 0.25;
 168     for(int sect = 0; sect < 5; sect++)
 169     {
 170         p[0] = lstX[0] + dx*(sect*part);
 171                 p[1] = lstY[0] + dy*(sect*part);
 172                 p[2] = lstZ[0] + dz*(sect*part);
 173                 if(sect == 4){
 174                         p[0] = p[0] + 0.5*(((lstX[1]+lstX[2])/2)-p[0]);
 175                         p[1] = p[1] + 0.5*(((lstY[1]+lstY[2])/2)-p[1]);
 176                         p[2] = p[2] + 0.5*(((lstZ[1]+lstZ[2])/2)-p[2]);
 177                 }
 178                 if(sect == 0){
 179                         p[0] = p[0] + 0.5*(((lstX[0]+lstX[3])/2)-p[0]);
 180                         p[1] = p[1] + 0.5*(((lstY[0]+lstY[3])/2)-p[1]);
 181                         p[2] = p[2] + 0.5*(((lstZ[0]+lstZ[3])/2)-p[2]);
 182                 }
 183                 NearestPointToMesh(points, pointLocator, spc, p,pM);
 184                 wsp->InsertPoint(pM[0] ,pM[1], pM[2],"");
 185
 186                 if(sect == 0 || sect == 4) lstIdNormalSurface.push_back( pointLocator->FindClosestPoint(pM) );
 187     }
 188
 189     // --- Group 1 ---
 190     wsp->InsertCollectionAfter_();
 191       p[0] = (lstX[0]+lstX[3])/2;
 192       p[1] = (lstY[0]+lstY[3])/2;
 193       p[2] = (lstZ[0]+lstZ[3])/2;
 194       NearestPointToMesh(points, pointLocator, spc, p,pM);
 195       wsp->InsertPoint(pM[0] ,pM[1], pM[2],"");
 196       p[0] = (lstX[1]+lstX[2])/2;
 197       p[1] = (lstY[1]+lstY[2])/2;
 198       p[2] = (lstZ[1]+lstZ[2])/2;
 199       NearestPointToMesh(points, pointLocator, spc, p,pM);
 200       wsp->InsertPoint(pM[0] ,pM[1], pM[2],"");
 201       p[0] = (lstX[0]+lstX[1]+lstX[2]+lstX[3])/4;          p[1] = (lstY[0]+lstY[1]+lstY[2]+lstY[3])/4;          p[2] = (lstZ[0]+lstZ[1]+lstZ[2]+lstZ[3])/4;
 202       wsp->InsertPoint(p[0] ,p[1], p[2],"");
 203
 204     // --- Group 2 ---
 205
 206     wsp->InsertCollectionAfter_();
 207
 208     dx = lstX[2]-lstX[3];
 209     dy = lstY[2]-lstY[3];
 210     dz = lstZ[2]-lstZ[3];
 211     for(int sect = 0; sect < 5; sect++)
 212     {
 213         p[0] = lstX[3] + dx*(sect*part);
 214                 p[1] = lstY[3] + dy*(sect*part);
 215                 p[2] = lstZ[3] + dz*(sect*part);
 216                 if(sect == 4){
 217                         p[0] = p[0] + 0.5*(((lstX[1]+lstX[2])/2)-p[0]);
 218                         p[1] = p[1] + 0.5*(((lstY[1]+lstY[2])/2)-p[1]);
 219                         p[2] = p[2] + 0.5*(((lstZ[1]+lstZ[2])/2)-p[2]);
 220                 }
 221                 if(sect == 0){
 222                         p[0] = p[0] + 0.5*(((lstX[0]+lstX[3])/2)-p[0]);
 223                         p[1] = p[1] + 0.5*(((lstY[0]+lstY[3])/2)-p[1]);
 224                         p[2] = p[2] + 0.5*(((lstZ[0]+lstZ[3])/2)-p[2]);
 225                 }
 226                 NearestPointToMesh(points, pointLocator, spc, p,pM);
 227                 wsp->InsertPoint(pM[0] ,pM[1], pM[2],"");
 228
 229                 if(sect == 0 || sect == 4) lstIdNormalSurface.push_back( pointLocator->FindClosestPoint(pM) );
 230     }
 231
 232      pointLocator->Delete();
 233
 234     // Check normals
 235 //        1. Recorrer las normales de lstIdNormals y calcular el promedio  -> V1
 236     double          *nValue;
 237         double                  n1[3];
 238     vtkPointData    *pointdata  =  bbGetInputMesh()->GetPointData();
 239     vtkDataArray    *dataarray  = pointdata->GetNormals();
 240         int i,size = lstIdNormalSurface.size();
 241         n1[0]=0;
 242         n1[1]=0;
 243         n1[2]=0;
 244         for (i=0; i<size; i++)
 245         {
 246             nValue        = dataarray->GetTuple3( lstIdNormalSurface[i] );
 247                 n1[0] = n1[0] + nValue[0];
 248                 n1[1] = n1[1] + nValue[1];
 249                 n1[2] = n1[2] + nValue[2];
 250         } // for i
 251         n1[0] = n1[0]/size;
 252         n1[1] = n1[1]/size;
 253         n1[2] = n1[2]/size;
 254
 255 //        2. Calcular el promedio de 4 normales de la nueva superficie     -> V2
 256     double pC[3];
 257     double pM1[3];
 258     double pM2[3];
 259     double n2[3];
 260     n2[0] = 0;
 261     n2[1] = 0;
 262     n2[2] = 0;
 263     // Collection 0 with 4 points
 264     // Collection 1 with 3 points
 265     // Collection 2 with 4 points
 266     wsp->GetCollectionPoint(1,1, pC);
 267
 268     wsp->GetCollectionPoint(0,0, pM);
 269     vtkMath::Subtract(pM,pC,pM1);
 270     wsp->GetCollectionPoint(0,3, pM);
 271     vtkMath::Subtract(pM,pC,pM2);
 272     vtkMath::Cross(pM1,pM2,pM);
 273     n2[0] = pM[0];
 274     n2[1] = pM[1];
 275     n2[2] = pM[2];
 276
 277     wsp->GetCollectionPoint(0,3, pM);
 278     vtkMath::Subtract(pM,pC,pM1);
 279     wsp->GetCollectionPoint(2,3, pM);
 280     vtkMath::Subtract(pM,pC,pM2);
 281     vtkMath::Cross(pM1,pM2,pM);
 282     n2[0] = n2[0] + pM[0];
 283     n2[1] = n2[1] + pM[1];
 284     n2[2] = n2[2] + pM[2];
 285
 286     wsp->GetCollectionPoint(2,3, pM);
 287     vtkMath::Subtract(pM,pC,pM1);
 288     wsp->GetCollectionPoint(2,0, pM);
 289     vtkMath::Subtract(pM,pC,pM2);
 290     vtkMath::Cross(pM1,pM2,pM);
 291     n2[0] = n2[0] + pM[0];
 292     n2[1] = n2[1] + pM[1];
 293     n2[2] = n2[2] + pM[2];
 294
 295     wsp->GetCollectionPoint(2,0, pM);
 296     vtkMath::Subtract(pM,pC,pM1);
 297     wsp->GetCollectionPoint(0,0, pM);
 298     vtkMath::Subtract(pM,pC,pM2);
 299     vtkMath::Cross(pM1,pM2,pM);
 300     n2[0] = n2[0] + pM[0];
 301     n2[1] = n2[1] + pM[1];
 302     n2[2] = n2[2] + pM[2];
 303
 304     n2[0] = n2[0] / 4;
 305     n2[1] = n2[1] / 4;
 306     n2[2] = n2[2] / 4;
 307
 308 //        3. Calcular el angulo entre V1 y V2
 309     double angle = vtkMath::AngleBetweenVectors(n1,n2) * 180 / vtkMath::Pi();
 310 //        4. Si el angulo es major de 90 Invertir las normales de la superficie actual
 311     if (angle<90)
 312     {
 313         vtkMath::MultiplyScalar(n2, -1);
 314         wsp->InvertLstPoints_();
 315     } // if angle
 316     std::vector<double> normalOut(n2, n2 + 3);
 317     double norm = vtkMath::Norm(n2);
 318     normalOut[0] = normalOut[0]/norm;
 319         normalOut[1] = normalOut[1]/norm;
 320         normalOut[2] = normalOut[2]/norm;
 321     bbSetOutputOut(normalOut);
 322
 323     // --- Finish ---
 324     wsp->SetOutputBox();
 325     wsp->UndoRedo_SaveCollection();
 326 }
 327
 328 void ShowNPoints_Tools::MovePatchCenter()
 329 {
 330         std::vector<double> params = bbGetInputParams();
 331         if(params.size() == 4)
 332         {
 333                 if(params[3] != 1 && params[3] != -1)
 334                 {
 335                         printf("PG ShowNPoints_Tools::MovePatchCenter()  Warning params are wrong. direction of movement should be 1 or -1\n");
 336                         return;
 337                 }
 338
 339                 double centerPoint[3];
 340                 std::vector<double> normal, modPoint;
 341
 342                 normal.push_back(params[0]);
 343                 normal.push_back(params[1]);
 344                 normal.push_back(params[2]);
 345
 346                 int direction = -1 * params[3];
 347
 348                 WidgetShowNPoints* wsp = bbGetInputWidgetShowNPoints();
 349                 wsp->GetCollectionPoint(1,1, centerPoint);
 350                 modPoint.push_back(centerPoint[0] + direction * normal[0]);
 351                 modPoint.push_back(centerPoint[1] + direction * normal[1]);
 352                 modPoint.push_back(centerPoint[2] + direction * normal[2]);
 353
 354                 wsp->SetCollectionPoint(1, 1, modPoint);
 355
 356         // --- Finish ---
 357         wsp->SetOutputBox();
 358         wsp->UndoRedo_SaveCollection();
 359         }else{
 360                 printf("PG ShowNPoints_Tools::MovePatchCenter()  Warning params are wrong. Need 4: normal x y z and direction of movement (1 or -1)\n");
 361         }
 362 }
 363
 364 //=====
 365 // 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)
 366 //=====
 367 void ShowNPoints_Tools::Process()
 368 {
 369
 370 // THE MAIN PROCESSING METHOD BODY
 371 //   Here we simply set the input 'In' value to the output 'Out'
 372 //   And print out the output value
 373 // INPUT/OUTPUT ACCESSORS ARE OF THE FORM :
 374 //    void bbSet{Input|Output}NAME(const TYPE&)
 375 //    const TYPE& bbGet{Input|Output}NAME() const
 376 //    Where :
 377 //    * NAME is the name of the input/output
 378 //      (the one provided in the attribute 'name' of the tag 'input')
 379 //    * TYPE is the C++ type of the input/output
 380 //      (the one provided in the attribute 'type' of the tag 'input')
 381
 382 //    bbSetOutputOut( bbGetInputIn() );
 383 //    std::cout << "Output value = " <<bbGetOutputOut() << std::endl;
 384
 385     if (bbGetInputWidgetShowNPoints()!=NULL)
 386     {
 387         if (bbGetInputType()==1)
 388         {
 389             bbGetInputWidgetShowNPoints()->OnAutoAddPoints_tool();
 390         } // if Type
 391
 392         if (bbGetInputType()==5)
 393         {
 394             bbGetInputWidgetShowNPoints()->OnAddPoint_();
 395         } // if Type
 396         if (bbGetInputType()==10)
 397         {
 398             bbGetInputWidgetShowNPoints()->OnInsertPoint_();
 399         } // if Type
 400         if (bbGetInputType()==20)
 401         {
 402             bbGetInputWidgetShowNPoints()->OnTrackPoint_tool();
 403         } // if Type
 404         if (bbGetInputType()==30)
 405         {
 406             bbGetInputWidgetShowNPoints()->OnSetPoint_();
 407         } // if Type
 408         if (bbGetInputType()==40)
 409         {
 410             bbGetInputWidgetShowNPoints()->OnErasePoint_();
 411         } // if Type
 412         if (bbGetInputType()==50)
 413         {
 414             bbGetInputWidgetShowNPoints()->OnDeleteAllPoints_();
 415         } // if Type
 416         if (bbGetInputType()==100)
 417         {
 418             bbGetInputWidgetShowNPoints()->OnInsertCollectionAfter_();
 419         } // if Type
 420         if (bbGetInputType()==110)
 421         {
 422             bbGetInputWidgetShowNPoints()->OnDeleteCollection_();
 423         } // if Type
 424         if (bbGetInputType()==120)
 425         {
 426             bbGetInputWidgetShowNPoints()->OnResetCollections_();
 427         } // if Type
 428         if (bbGetInputType()==200)
 429         {
 430             InitCreatePatch_Points();
 431         } // if Type
 432         if (bbGetInputType()==210)
 433         {
 434             bbGetInputWidgetShowNPoints()->OnInvertLstPoints_();
 435         } // if Type
 436         if(bbGetInputType()==220)
 437         {
 438                 MovePatchCenter();
 439         } // if Type
 440     } // if bbGetInputWidgetShowNPoints
 441 }
 442 //=====
 443 // 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)
 444 //=====
 445 void ShowNPoints_Tools::bbUserSetDefaultValues()
 446 {
 447 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX
 448 //    Here we initialize the input 'In' to 0
 449    bbSetInputType(0);
 450    bbSetInputMesh(NULL);
 451    bbSetInputWidgetShowNPoints(NULL);
 452
 453     std::vector<double> spc;
 454     spc.push_back(1);
 455     spc.push_back(1);
 456     spc.push_back(1);
 457     bbSetInputSpacing(spc);
 458 }
 459 //=====
 460 // 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)
 461 //=====
 462 void ShowNPoints_Tools::bbUserInitializeProcessing()
 463 {
 464 //  THE INITIALIZATION METHOD BODY :
 465 //    Here does nothing
 466 //    but this is where you should allocate the internal/output pointers
 467 //    if any
 468 }
 469 //=====
 470 // 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)
 471 //=====
 472 void ShowNPoints_Tools::bbUserFinalizeProcessing()
 473 {
 474 //  THE FINALIZATION METHOD BODY :
 475 //    Here does nothing
 476 //    but this is where you should desallocate the internal/output pointers
 477 //    if any
 478 }
 479
 480 } // EO namespace bbcreaMaracasVisu
 481
 482