bbtk_creaVtk_PKG/src/bbcreaVtkReadMHDPlane.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 "bbcreaVtkReadMHDPlane.h"
   5 #include "bbcreaVtkPackage.h"
   6
   7 #include "stdio.h"
   8
   9 namespace bbcreaVtk
  10 {
  11
  12 BBTK_ADD_BLACK_BOX_TO_PACKAGE(creaVtk,ReadMHDPlane)
  13 BBTK_BLACK_BOX_IMPLEMENTATION(ReadMHDPlane,bbtk::AtomicBlackBox);
  14 //=====
  15 // 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)
  16 //=====
  17 void ReadMHDPlane::Process()
  18 {
  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         std::string inputfilename;
  36
  37         if (bbGetInputDirectionPlane()=="XY")
  38         {
  39                 inputfilename=bbGetInputFileName();
  40         } // if XY
  41
  42
  43         if ((bbGetInputDirectionPlane()=="YZ") || (bbGetInputDirectionPlane()=="ZX"))
  44         {
  45
  46                 //-- Split FileName
  47                 std::size_t found               =       bbGetInputFileName().find_last_of("/\\");
  48                 std::string path                =       bbGetInputFileName().substr(0,found+1);
  49                 std::string filename    =       bbGetInputFileName().substr(found+1);
  50 #ifdef _WIN32
  51                 path=path+"YZ_ZX\\";
  52 #else
  53                 path=path+"YZ_ZX/";
  54 #endif
  55
  56
  57                 if (bbGetInputDirectionPlane()=="YZ")
  58                 {
  59                         inputfilename = path+filename+"_YZ.mhd";
  60                 } // if YZ
  61
  62                 if (bbGetInputDirectionPlane()=="ZX")
  63                 {
  64                         inputfilename = path+filename+"_ZX.mhd";
  65                 } // XZ
  66         } // if YZ || XZ
  67
  68         vtkImageData *newImage;
  69         long int newHeaderSize;
  70     std::string newFileName=inputfilename+"-OneSlice";
  71
  72     int sx,sy,sz;
  73     char mystring[250];
  74     char strTmp[20];
  75         bool ok=true;
  76     FILE *ffIn  = fopen(inputfilename.c_str(),"r");
  77         if (ffIn!=NULL)
  78         {
  79             FILE *ffOut = fopen(newFileName.c_str(),"w");
  80
  81         while(!feof(ffIn))
  82         {
  83             strcpy(mystring,"\n");
  84                         fgets(mystring,250,ffIn);
  85                 if (strncmp("NDims",mystring,5)==0) {strcpy(mystring,"NDims = 2\n");}
  86                 if (strncmp("DimSize",mystring,6)==0)
  87                         {
  88                                 sscanf(mystring,"%s %s %d %d %d",strTmp, strTmp, &sx, &sy,&sz);
  89                                 sprintf(mystring,"DimSize = %d %d\n",sx,sy);
  90                         newHeaderSize = sx*sy*bbGetInputSlice();
  91                                 if (bbGetInputSlice()>=sz) {ok=false;}
  92                         } // if
  93                 if (strncmp("ElementType = MET_CHAR",mystring,22)==0)   { newHeaderSize=newHeaderSize*1; }
  94                 if (strncmp("ElementType = MET_UCHAR",mystring,23)==0)  { newHeaderSize=newHeaderSize*1; }
  95                 if (strncmp("ElementType = MET_USHORT",mystring,24)==0) { newHeaderSize=newHeaderSize*2; }
  96                 if (strncmp("ElementType = MET_SHORT",mystring,23)==0)  { newHeaderSize=newHeaderSize*2; }
  97                 if (strncmp("ElementType = MET_FLOAT",mystring,23)==0)  { newHeaderSize=newHeaderSize*4; }
  98                 if (strncmp("ElementType = VTK_FLOAT",mystring,23)==0)  { newHeaderSize=newHeaderSize*4; }
  99                 if (strncmp("Offset",mystring,6)==0) {strcpy(mystring,"Offset = 0 0 0\n");}
 100                 if (strncmp("HeaderSize",mystring,10)==0) {strcpy(mystring,"");}
 101                 if (strncmp("ElementDataFile",mystring,15)==0)
 102                         {
 103                                 fprintf(ffOut,"HeaderSize = %ld\n\n", newHeaderSize );
 104                         } // if
 105                 fprintf(ffOut,mystring);
 106         } // while
 107                 fclose(ffIn);
 108                 fclose(ffOut);
 109
 110                 if (ok==true)
 111                 {
 112                         vtkMetaImageReader * reader = vtkMetaImageReader::New();
 113                         reader->SetFileName( newFileName.c_str() );
 114                         reader->Update();
 115                         newImage = reader->GetOutput();
 116                 } // if ok
 117         } else {
 118                 ok=false;
 119                 printf("EED ERROR: Problem openin:%s\n", inputfilename.c_str() );
 120         }
 121
 122         if (ok==false)
 123         {
 124                 int i;
 125                 int sizeX, sizeY, sizeZ;
 126                 sizeX = 200;
 127                 sizeY = sizeX;
 128                 sizeZ = 1;
 129                 newImage = vtkImageData::New();
 130                 newImage->Initialize();
 131                 newImage->SetScalarTypeToUnsignedChar();
 132                 newImage->SetSpacing( 1,1,1 );
 133                 newImage->SetDimensions(  sizeX,sizeY,sizeZ );
 134                 newImage->SetWholeExtent(0,  sizeX-1,0,sizeY-1,0,sizeZ-1 );
 135                 newImage->SetExtent(0,  sizeX-1,0,sizeY-1,0,sizeZ-1 );
 136                 newImage->SetNumberOfScalarComponents(1);
 137                 newImage->AllocateScalars();
 138                 newImage->Update();
 139                 memset ( (void*)newImage->GetScalarPointer(), 0, sizeX*sizeY*1 );
 140                 for (i=0; i<sizeX; i++)
 141                 {
 142                         newImage->SetScalarComponentFromDouble(i,i,0, 0, 255 );
 143                         newImage->SetScalarComponentFromDouble(i,sizeY-1-i,0, 0, 255 );
 144                 } // for i
 145         } // if ok
 146
 147         bbSetOutputOut( newImage );
 148
 149 }
 150 //=====
 151 // 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)
 152 //=====
 153 void ReadMHDPlane::bbUserSetDefaultValues()
 154 {
 155
 156 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX
 157 //    Here we initialize the input 'In' to 0
 158    bbSetInputFileName("");
 159    bbSetInputSlice(0);
 160    bbSetInputDirectionPlane("XY");
 161
 162 }
 163 //=====
 164 // 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)
 165 //=====
 166 void ReadMHDPlane::bbUserInitializeProcessing()
 167 {
 168
 169 //  THE INITIALIZATION METHOD BODY :
 170 //    Here does nothing
 171 //    but this is where you should allocate the internal/output pointers
 172 //    if any
 173
 174
 175 }
 176 //=====
 177 // 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)
 178 //=====
 179 void ReadMHDPlane::bbUserFinalizeProcessing()
 180 {
 181
 182 //  THE FINALIZATION METHOD BODY :
 183 //    Here does nothing
 184 //    but this is where you should desallocate the internal/output pointers
 185 //    if any
 186
 187 }
 188 }
 189 // EO namespace bbcreaVtk
 190
 191