packages/vtk/src/bbvtkMetaImageWriter.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 "bbvtkMetaImageWriter.h"
   5 #include "bbvtkPackage.h"
   6
   7 #include <vtkExtractVOI.h>
   8 #include <vtkMetaImageWriter.h>
   9
  10 #include "vtkJSONImageWriter.h"
  11 #include "vtkXMLImageDataWriter.h"
  12 #include "vtkXMLDataSetWriter.h"
  13 #include "vtkDataSetWriter.h"
  14 #include "vtkPDataSetWriter.h"
  15 #include "vtkXMLWriter.h"
  16
  17 namespace bbvtk
  18 {
  19
  20 BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,MetaImageWriter)
  21 BBTK_BLACK_BOX_IMPLEMENTATION(MetaImageWriter,bbtk::AtomicBlackBox);
  22
  23
  24
  25 void MetaImageWriter::Save_mhdb( std::string filename  , vtkImageData* img ,int sizeB)
  26 {
  27         int i,j,k;
  28         int ext[6];
  29         int dim[3];
  30         int elementType=img->GetScalarType();
  31
  32         img->GetExtent(ext);
  33         dim[0]=ext[1]-ext[0]+1;
  34         dim[1]=ext[3]-ext[2]+1;
  35         dim[2]=ext[5]-ext[4]+1;
  36         int bsx=ceil((double)dim[0]/(double)sizeB);
  37         int bsy=ceil((double)dim[1]/(double)sizeB);
  38         int bsz=ceil((double)dim[2]/(double)sizeB);
  39         int voi[6];
  40         std::string filenameBlock;
  41         std::string filenameBlockVti;
  42         if (filename.substr(filename.size()-5) == ".mhdb")
  43         {
  44                 std::string cmd;
  45 //              cmd="rm -rf "+filename;
  46                 system( cmd.c_str() );
  47                 cmd="mkdir "+filename;
  48                 system( cmd.c_str() );
  49         }
  50
  51         std::string fileinfoname=filename+"/info.dat";
  52         FILE *ff=fopen(fileinfoname.c_str(),"w");
  53           fprintf(ff,"SizeBlock = %d\n",sizeB);
  54           fprintf(ff,"DimSize = %d %d %d\n" , dim[0],dim[1],dim[2] );
  55           fprintf(ff,"ElementType = %d\n", elementType );
  56         fclose(ff);
  57
  58         for (i=0;i<bsx;i++)
  59         {
  60                 for (j=0;j<bsy;j++)
  61                 {
  62                         for (k=0;k<bsz;k++)
  63                         {
  64                                 voi[0]=i*sizeB;
  65                                 voi[1]=voi[0]+(sizeB-1);
  66                                 voi[2]=j*sizeB;
  67                                 voi[3]=voi[2]+(sizeB-1);
  68                                 voi[4]=k*sizeB;
  69                                 voi[5]=voi[4]+(sizeB-1);
  70                                 if (voi[1]>=dim[0]) { voi[1]=dim[0]-1; }
  71                                 if (voi[3]>=dim[1]) { voi[3]=dim[1]-1; }
  72                                 if (voi[5]>=dim[2]) { voi[5]=dim[2]-1; }
  73                                 vtkExtractVOI *extract = vtkExtractVOI::New();
  74                                 extract->SetInputData( img );
  75                                 extract->SetVOI(voi);
  76                                 extract->UpdateWholeExtent();
  77                                 extract->Modified();
  78                                 extract->Update();
  79                                 filenameBlock           =filename+"/mhdb-"+ std::to_string(i)+"-"+ std::to_string(j)+"-"+ std::to_string(k)+".mha";
  80                                 vtkMetaImageWriter* w = vtkMetaImageWriter::New();
  81                                 w->SetInputData( extract->GetOutput() );
  82                                 w->SetCompression(true);
  83                                 w->SetFileDimensionality(bbGetInputIn()->GetDataDimension());   // NTU
  84                                 w->SetFileName( filenameBlock.c_str() );
  85                                 w->Write();
  86                                 w->Delete();
  87
  88 /* Borrame
  89                                 filenameBlockVti=filenameBlock+"-ZLib.vti";
  90                                 vtkXMLImageDataWriter *writer = vtkXMLImageDataWriter::New();
  91                                 writer->SetDataModeToBinary();
  92 //                              writer->SetCompressionLevel(5);
  93                                 writer->SetCompressorTypeToZLib();
  94                                 writer->SetFileName( filenameBlockVti.c_str() );
  95                                 writer->SetInputData( bbGetInputIn() );
  96                                 writer->Write();
  97 */
  98
  99                                 extract->Delete();
 100                         } // for k
 101                 } // for j
 102         } // for i
 103 }
 104
 105
 106
 107 //=====
 108 // 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)
 109 //=====
 110 void MetaImageWriter::Process()
 111 {
 112
 113 printf("EED vtkMetaImageWriter::Process  START\n");
 114
 115 // THE MAIN PROCESSING METHOD BODY
 116 //   Here we simply set the input 'In' value to the output 'Out'
 117 //   And print out the output value
 118 // INPUT/OUTPUT ACCESSORS ARE OF THE FORM :
 119 //    void bbSet{Input|Output}NAME(const TYPE&)
 120 //    const TYPE& bbGet{Input|Output}NAME() const
 121 //    Where :
 122 //    * NAME is the name of the input/output
 123 //      (the one provided in the attribute 'name' of the tag 'input')
 124 //    * TYPE is the C++ type of the input/output
 125 //      (the one provided in the attribute 'type' of the tag 'input')
 126
 127 //EED 2017-01-01 Migration VTK7
 128 #if VTK_MAJOR_VERSION <= 5
 129    bbGetInputIn()->Update();
 130 #else
 131         // ...
 132 #endif
 133         std::string userGivenName = bbGetInputFilename();
 134         if (bbGetInputIn()!=NULL)
 135         {
 136                 if (userGivenName.substr(userGivenName.size()-5) == ".mhdb")
 137                 {
 138                         Save_mhdb( userGivenName , bbGetInputIn() , bbGetInputSizeBlock() );
 139                 } else {
 140                                 if (!((userGivenName.substr(userGivenName.size()-4) == ".mhd") || (userGivenName.substr(userGivenName.size()-4) == ".mha")))  //JPR
 141                                 {
 142                                         userGivenName += ".mhd";
 143                                 }
 144                            vtkMetaImageWriter* w = vtkMetaImageWriter::New();
 145         //EED 2017-01-01 Migration VTK7
 146         #if VTK_MAJOR_VERSION <= 5
 147                                  w->SetInput(bbGetInputIn());
 148         #else
 149                                  w->SetInputData(bbGetInputIn());
 150         #endif
 151                                  w->SetCompression(bbGetInputCompression());
 152                                  w->SetFileDimensionality(bbGetInputIn()->GetDataDimension());  // NTU
 153                                  w->SetFileName(userGivenName.c_str());
 154                                  //w->SetFileName(bbGetInputFilename().c_str());                                // JPR
 155                                  w->Write();
 156                                  w->Delete();
 157                 } // .mhdb
 158         } // bbGetInputIn()!=NULL
 159 printf("EED vtkMetaImageWriter::Process  END\n");
 160 }
 161 //=====
 162 // 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)
 163 //=====
 164 void MetaImageWriter::bbUserSetDefaultValues()
 165 {
 166
 167 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX
 168 //    Here we initialize the input 'In' to 0
 169    bbSetInputCompression(false);
 170    bbSetInputSizeBlock(20);
 171 }
 172 //=====
 173 // 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)
 174 //=====
 175 void MetaImageWriter::bbUserInitializeProcessing()
 176 {
 177
 178 //  THE INITIALIZATION METHOD BODY :
 179 //    Here does nothing
 180 //    but this is where you should allocate the internal/output pointers
 181 //    if any
 182
 183
 184 }
 185 //=====
 186 // 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)
 187 //=====
 188 void MetaImageWriter::bbUserFinalizeProcessing()
 189 {
 190
 191 //  THE FINALIZATION METHOD BODY :
 192 //    Here does nothing
 193 //    but this is where you should desallocate the internal/output pointers
 194 //    if any
 195
 196 }
 197 }
 198 // EO namespace bbvtk
 199
 200