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
  36
  37         vtkImageData *newImage;
  38         long int newHeaderSize;
  39     std::string newFileName=bbGetInputFileName()+"-OneSlice";
  40
  41     int sx,sy,sz;
  42     char mystring[250];
  43     char strTmp[20];
  44         bool ok=true;
  45     FILE *ffIn  = fopen(bbGetInputFileName().c_str(),"r+");
  46         if (ffIn!=NULL){
  47             FILE *ffOut = fopen(newFileName.c_str(),"w+");
  48
  49         while(!feof(ffIn))
  50         {
  51             strcpy(mystring,"\n");
  52                         fgets(mystring,250,ffIn);
  53                 if (strncmp("NDims",mystring,5)==0) {strcpy(mystring,"NDims = 2\n");}
  54                 if (strncmp("DimSize",mystring,6)==0)
  55                         {
  56                                 sscanf(mystring,"%s %s %d %d %d",strTmp, strTmp, &sx, &sy,&sz);
  57                                 sprintf(mystring,"DimSize = %d %d\n",sx,sy);
  58                         newHeaderSize = sx*sy*bbGetInputSlice();
  59                                 if (bbGetInputSlice()>=sz) {ok=false;}
  60                         } // if
  61                 if (strncmp("ElementType = MET_USHORT",mystring,24)==0) { newHeaderSize=newHeaderSize*2; }
  62                 if (strncmp("ElementType = MET_SHORT",mystring,23)==0) { newHeaderSize=newHeaderSize*2; }
  63                 if (strncmp("ElementType = MET_FLOAT",mystring,23)==0) { newHeaderSize=newHeaderSize*2; }
  64                 if (strncmp("Offset",mystring,6)==0) {strcpy(mystring,"Offset = 0 0 0\n");}
  65                 if (strncmp("HeaderSize",mystring,10)==0) {strcpy(mystring,"");}
  66                 if (strncmp("ElementDataFile",mystring,15)==0)
  67                         {
  68                                 fprintf(ffOut,"HeaderSize = %ld\n\n", newHeaderSize );
  69                         } // if
  70                 fprintf(ffOut,mystring);
  71         } // while
  72                 fclose(ffIn);
  73                 fclose(ffOut);
  74
  75                 if (ok==true)
  76                 {
  77                         vtkMetaImageReader * reader = vtkMetaImageReader::New();
  78                         reader->SetFileName( newFileName.c_str() );
  79                         reader->Update();
  80                         newImage = reader->GetOutput();
  81                 } // if ok
  82         } else {
  83                 ok=false;
  84         }
  85
  86         if (ok==false)
  87         {
  88                 int i,j;
  89                 int sizeX, sizeY, sizeZ;
  90                 sizeX = 200;
  91                 sizeY = sizeX;
  92                 sizeZ = 1;
  93                 newImage = vtkImageData::New();
  94                 newImage->Initialize();
  95                 newImage->SetScalarTypeToUnsignedChar();
  96                 newImage->SetSpacing( 1,1,1 );
  97                 newImage->SetDimensions(  sizeX,sizeY,sizeZ );
  98                 newImage->SetWholeExtent(0,  sizeX-1,0,sizeY-1,0,sizeZ-1 );
  99                 newImage->SetExtent(0,  sizeX-1,0,sizeY-1,0,sizeZ-1 );
 100                 newImage->SetNumberOfScalarComponents(1);
 101                 newImage->AllocateScalars();
 102                 newImage->Update();
 103                 memset ( (void*)newImage->GetScalarPointer(), 0, sizeX*sizeY*1 );
 104                 for (i=0; i<sizeX; i++)
 105                 {
 106                         newImage->SetScalarComponentFromDouble(i,i,0, 0, 255 );
 107                         newImage->SetScalarComponentFromDouble(i,sizeY-1-i,0, 0, 255 );
 108                 } // for i
 109         } // if ok
 110
 111         bbSetOutputOut( newImage );
 112
 113 }
 114 //=====
 115 // 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)
 116 //=====
 117 void ReadMHDPlane::bbUserSetDefaultValues()
 118 {
 119
 120 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX
 121 //    Here we initialize the input 'In' to 0
 122    bbSetInputFileName("");
 123    bbSetInputSlice(0);
 124
 125 }
 126 //=====
 127 // 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)
 128 //=====
 129 void ReadMHDPlane::bbUserInitializeProcessing()
 130 {
 131
 132 //  THE INITIALIZATION METHOD BODY :
 133 //    Here does nothing
 134 //    but this is where you should allocate the internal/output pointers
 135 //    if any
 136
 137
 138 }
 139 //=====
 140 // 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)
 141 //=====
 142 void ReadMHDPlane::bbUserFinalizeProcessing()
 143 {
 144
 145 //  THE FINALIZATION METHOD BODY :
 146 //    Here does nothing
 147 //    but this is where you should desallocate the internal/output pointers
 148 //    if any
 149
 150 }
 151 }
 152 // EO namespace bbcreaVtk
 153
 154