packages/vtk/src/bbvtkSurfaceTexture.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 "bbvtkSurfaceTexture.h"
   5 #include "bbvtkPackage.h"
   6
   7 #include <vtkPointData.h>
   8 #include <vtkMath.h>
   9
  10 namespace bbvtk
  11 {
  12
  13 BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,SurfaceTexture)
  14 BBTK_BLACK_BOX_IMPLEMENTATION(SurfaceTexture,bbtk::AtomicBlackBox);
  15 //=====
  16 // 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)
  17 //=====
  18 void SurfaceTexture::Process()
  19 {
  20 // THE MAIN PROCESSING METHOD BODY
  21 //   Here we simply set the input 'In' value to the output 'Out'
  22 //   And print out the output value
  23 // INPUT/OUTPUT ACCESSORS ARE OF THE FORM :
  24 //    void bbSet{Input|Output}NAME(const TYPE&)
  25 //    const TYPE& bbGet{Input|Output}NAME() const
  26 //    Where :
  27 //    * NAME is the name of the input/output
  28 //      (the one provided in the attribute 'name' of the tag 'input')
  29 //    * TYPE is the C++ type of the input/output
  30 //      (the one provided in the attribute 'type' of the tag 'input')
  31
  32 //    bbSetOutputOut( bbGetInputIn() );
  33 //    std::cout << "Output value = " <<bbGetOutputOut() << std::endl;
  34
  35         int     i;
  36         double  spc[3];
  37         double  range[2];
  38         int     ext[6];
  39         if ((bbGetInputImage()!=NULL) && (bbGetInputMesh()!=NULL))
  40         {
  41                 bbGetInputImage()->GetSpacing(spc);
  42                 bbGetInputImage()->GetScalarRange(range);
  43         //EED 2017-01-01 Migration VTK7
  44         #if VTK_MAJOR_VERSION <= 5
  45                 bbGetInputImage()->GetWholeExtent(ext);
  46         #else
  47                 bbGetInputImage()->GetExtent(ext);
  48         #endif
  49                 int maxX = ext[1]-ext[0]+1;
  50                 int maxY = ext[3]-ext[2]+1;
  51                 int maxZ = ext[5]-ext[4]+1;
  52
  53                 if (backColorType!=bbGetInputColorType() )
  54                 {
  55                         backColorType = bbGetInputColorType();
  56                         if (colors!=NULL)                               { colors->Delete();                     }
  57 //                      if (colorLookupTable!=NULL)             { colorLookupTable->Delete();   }
  58 //                      if (colorLookupTableWL!=NULL)   { colorLookupTableWL->Delete(); }
  59
  60
  61 //                      firsttime=false;
  62                         // Generate the colors for each point based on the color map
  63                         colors = vtkUnsignedCharArray::New();
  64                         colors->SetNumberOfComponents(3);
  65 //                      colors->SetName("ColorsEED");
  66                         // Create the color map
  67                         if (bbGetInputColorType()==1)
  68                         {
  69                                 colorLookupTableWL      = vtkWindowLevelLookupTable::New();
  70                                 colorLookupTableWL->InverseVideoOn();
  71                                 colorLookupTable        = colorLookupTableWL;
  72                         } else {
  73                                 colorLookupTable = vtkLookupTable::New();
  74                         }
  75                         colorLookupTable->SetTableRange(range[0],range[1]);
  76                         colorLookupTable->Build();
  77                         double rgba1[4];
  78                         double rgba2[4];
  79                         for (int iLookTable = 0; iLookTable<128; iLookTable++)
  80                         {
  81                           colorLookupTable->GetTableValue(      iLookTable, rgba1);
  82                           colorLookupTable->GetTableValue(256-1-iLookTable, rgba2);
  83                           colorLookupTable->SetTableValue(256-1-iLookTable , rgba1[0],rgba1[1],rgba1[2],rgba1[3]);
  84                           colorLookupTable->SetTableValue(      iLookTable , rgba2[0],rgba2[1],rgba2[2],rgba2[3]);
  85                         } // for iLookTable
  86
  87         //EED 2018-06-8 ***********************ARDS Projet***********************************************
  88                         if (bbGetInputColorType()==2)
  89                         {
  90                                 int tableSize=256;
  91                                 colorLookupTable = vtkLookupTable::New();
  92         //                      colorLookupTable->SetNumberOfTableValues(tableSize);
  93                                 colorLookupTable->SetTableRange(range[0],range[1]);
  94         //                      colorLookupTable->SetTableRange(0,11);
  95                                 colorLookupTable->Build();
  96
  97         /*
  98           for (unsigned int i = 0; i < tableSize; ++i)
  99                 {
 100                 colorLookupTable->SetTableValue(i,
 101                                    vtkMath::Random(.25, 1.0),
 102                                    vtkMath::Random(.25, 1.0),
 103                                    vtkMath::Random(.25, 1.0),
 104                                    1.0);
 105                 }
 106         */
 107                                 double rgba1[4];
 108                                 double rgba2[4];
 109                                 int i,j;
 110                                 for (int iLookTable = 0; iLookTable<tableSize*3; iLookTable++)
 111                                 {
 112                                         i=rand() % tableSize;
 113                                         j=rand() % tableSize;
 114                                         colorLookupTable->GetTableValue(i, rgba1);
 115                                         colorLookupTable->GetTableValue(j, rgba2);
 116
 117                                   colorLookupTable->SetTableValue(j, rgba1[0],rgba1[1],rgba1[2],rgba1[3]);
 118                                   colorLookupTable->SetTableValue(i, rgba2[0],rgba2[1],rgba2[2],rgba2[3]);
 119                                 } // for iLookTable
 120                         } // type 2
 121                         if (bbGetInputColorType()==3)
 122                         {
 123                                 int tableSize=13;
 124                                 colorLookupTable = vtkLookupTable::New();
 125                                 colorLookupTable->SetNumberOfTableValues(tableSize);
 126                                 colorLookupTable->SetTableRange(6,range[1]);
 127                                 colorLookupTable->Build();
 128                           colorLookupTable->SetTableValue(0, 1  ,1      ,1      ,1);
 129                           colorLookupTable->SetTableValue(1, 1  ,0      ,0      ,1);
 130                           colorLookupTable->SetTableValue(2, 0  ,1      ,0      ,1);
 131                           colorLookupTable->SetTableValue(3, 0  ,0      ,1      ,1);
 132                           colorLookupTable->SetTableValue(4, 0  ,1      ,1      ,1);
 133                           colorLookupTable->SetTableValue(5, 1  ,1      ,0      ,1);
 134                           colorLookupTable->SetTableValue(6, 1  ,0      ,1      ,1);
 135                           colorLookupTable->SetTableValue(7, 0.5,1      ,1      ,1);
 136                           colorLookupTable->SetTableValue(8, 1  ,0.5,1  ,1);
 137                           colorLookupTable->SetTableValue(9, 1  ,1      ,0.5,1);
 138                           colorLookupTable->SetTableValue(10,1  ,0.5,0.5,1);
 139                           colorLookupTable->SetTableValue(11,0.5,0.5,1  ,1);
 140                         } // type 3
 141                 }  // firsttime
 142
 143
 144
 145                 bbGetInputMesh()->GetPointData()->SetScalars(colors);
 146                 if (bbGetInputColorType()==1)
 147                 {
 148                         colorLookupTableWL->SetLevel( bbGetInputColorLevel() );
 149                         colorLookupTableWL->SetWindow( bbGetInputColorWindow() );
 150                 }
 151                 if (bbGetInputExternalLookupTable()!=NULL)
 152                 {
 153                         generalLookupTable = bbGetInputExternalLookupTable();
 154                 } else {
 155                         generalLookupTable = colorLookupTable;
 156                 }
 157                 int missingpoints = bbGetInputMesh()->GetNumberOfPoints() - colors->GetDataSize()/colors->GetNumberOfComponents();
 158                 for(i = 0; i < missingpoints; i++)
 159                 {
 160                         colors->InsertNextTuple3(0,0,0);
 161                 }
 162                 if (bbGetInputTransform()!=NULL)
 163                 {
 164                         bbGetInputTransform()->Update();
 165                 }
 166                 double gl;
 167                 double p1[3];
 168                 double p2[3];
 169                 double dcolor[3];
 170                 for(i = 0; i < bbGetInputMesh()->GetNumberOfPoints(); i++)
 171                 {
 172                         if (bbGetInputTransform()!=NULL)
 173                         {
 174                                 bbGetInputMesh()->GetPoint(i,p1);
 175                                 bbGetInputTransform()->TransformPoint(p1,p2);
 176                         } else {
 177                                 bbGetInputMesh()->GetPoint(i,p2);
 178                         }
 179                         p2[0] = p2[0]/spc[0];
 180                         p2[1] = p2[1]/spc[1];
 181                         p2[2] = p2[2]/spc[2];
 182                         if ( (p2[0]>=0) && (p2[0]<maxX) && (p2[1]>=0) && (p2[1]<maxY) &&(p2[2]>=0) && (p2[2]<maxZ)  )
 183                         {
 184                            gl =  bbGetInputImage()->GetScalarComponentAsDouble(round(p2[0]), round(p2[1]), round(p2[2]),0);
 185                         } else {
 186                                 gl=0;
 187                         }
 188                         generalLookupTable->GetColor(gl, dcolor);
 189                         colors->SetTuple3(i,255*dcolor[0],255*dcolor[1],255*dcolor[2]);
 190                 } // for i
 191                 bbGetInputMesh()->Modified();
 192         } // if Image && Mesh
 193 }
 194
 195 //=====
 196 // 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)
 197 //=====
 198 void SurfaceTexture::bbUserSetDefaultValues()
 199 {
 200
 201 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX
 202 //    Here we initialize the input 'In' to 0
 203         bbSetInputMesh(NULL);
 204         bbSetInputImage(NULL);
 205         bbSetInputColorType(0);
 206         bbSetInputColorLevel(500);
 207         bbSetInputColorWindow(500);
 208         bbSetInputTransform(NULL);
 209         bbSetInputExternalLookupTable(NULL);
 210
 211         backColorType                   = -999;
 212         colors                                  = NULL;
 213         colorLookupTable                = NULL;
 214         colorLookupTableWL              = NULL;
 215 }
 216 //=====
 217 // 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)
 218 //=====
 219 void SurfaceTexture::bbUserInitializeProcessing()
 220 {
 221
 222 //  THE INITIALIZATION METHOD BODY :
 223 //    Here does nothing
 224 //    but this is where you should allocate the internal/output pointers
 225 //    if any
 226
 227
 228 }
 229 //=====
 230 // 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)
 231 //=====
 232 void SurfaceTexture::bbUserFinalizeProcessing()
 233 {
 234
 235 //  THE FINALIZATION METHOD BODY :
 236 //    Here does nothing
 237 //    but this is where you should desallocate the internal/output pointers
 238 //    if any
 239
 240 }
 241 }
 242 // EO namespace bbvtk
 243
 244