]> Creatis software - creaVtk.git/blobdiff - bbtk_creaVtk_PKG/src/bbcreaVtkReadMHDPlane.cxx
#2711 creaVtk Feature New Normal - new Box ReadMHDPlane
[creaVtk.git] / bbtk_creaVtk_PKG / src / bbcreaVtkReadMHDPlane.cxx
index 3b8fb24e373d5cd7593918fec5c2dd7b83b67077..d6818392ee6566160afc72b6f71da87257fe5e43 100644 (file)
@@ -32,6 +32,34 @@ BBTK_BLACK_BOX_IMPLEMENTATION(ReadMHDPlane,bbtk::AtomicBlackBox);
 // 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)
 //===== 
 
+
+vtkImageData* ReadMHDPlane::CreateDefaultImage()
+{
+       int i;
+       int sizeX, sizeY, sizeZ;
+       sizeX = 200;
+       sizeY = sizeX;
+       sizeZ = 1;
+       vtkImageData *newImage = vtkImageData::New();
+       newImage->Initialize();
+       newImage->SetScalarTypeToUnsignedChar();
+       newImage->SetSpacing( 1,1,1 );
+       newImage->SetDimensions(  sizeX,sizeY,sizeZ );
+       newImage->SetWholeExtent(0,  sizeX-1,0,sizeY-1,0,sizeZ-1 );
+       newImage->SetExtent(0,  sizeX-1,0,sizeY-1,0,sizeZ-1 );
+       newImage->SetNumberOfScalarComponents(1);
+       newImage->AllocateScalars();
+       newImage->Update();
+       memset ( (void*)newImage->GetScalarPointer(), 0, sizeX*sizeY*1 );
+       for (i=0; i<sizeX; i++)
+       {
+               newImage->SetScalarComponentFromDouble(i,i,0, 0, 255 );
+               newImage->SetScalarComponentFromDouble(i,sizeY-1-i,0, 0, 255 );
+       } // for i
+       return newImage;
+} 
+
+
 void ReadMHDPlane::ReadNormalMHD()
 {
 // THE MAIN PROCESSING METHOD BODY
@@ -50,6 +78,8 @@ void ReadMHDPlane::ReadNormalMHD()
 //    std::cout << "Output value = " <<bbGetOutputOut() << std::endl;
 
        std::string inputfilename;
+       int slice;
+       int width;
 
        if (bbGetInputDirectionPlane()=="XY")
        {
@@ -82,6 +112,13 @@ void ReadMHDPlane::ReadNormalMHD()
                } // XZ
        } // if YZ || XZ
 
+       width = bbGetInputWidth();
+       if (width<=0 ) { width=1; }
+
+       slice = bbGetInputSlice();
+       if (slice<0 ) { slice=0; }
+
+
        vtkImageData *newImage;
        long long int newHeaderSize;
     std::string newFileName=inputfilename+"-OneSlice";
@@ -99,12 +136,26 @@ void ReadMHDPlane::ReadNormalMHD()
        {
            strcpy(mystring,"\n");
                        fgets(mystring,250,ffIn);
-               if (strncmp("NDims",mystring,5)==0) {strcpy(mystring,"NDims = 2\n");}
+               if (strncmp("NDims",mystring,5)==0) {
+                               if (width==1) 
+                               { 
+                                       strcpy(mystring,"NDims = 2\n"); 
+                               } else {
+                                       strcpy(mystring,"NDims = 3\n"); 
+                               }
+                       }
                if (strncmp("DimSize",mystring,6)==0) 
                        {
                                sscanf(mystring,"%s %s %d %d %d",strTmp, strTmp, &sx, &sy,&sz);
-                               sprintf(mystring,"DimSize = %d %d\n",sx,sy);
-                       newHeaderSize = (long long int)sx*(long long int)sy*(long long int)bbGetInputSlice();
+
+                               if (width==1) 
+                               { 
+                                       sprintf(mystring,"DimSize = %d %d\n",sx,sy);
+                               } else {
+                                       sprintf(mystring,"DimSize = %d %d %d\n",sx,sy,width);
+                               }
+
+                       newHeaderSize = (long long int)sx*(long long int)sy*(long long int)slice;
                                if (bbGetInputSlice()>=sz) {ok=false;}
                        } // if
                if (strncmp("ElementType = MET_CHAR",mystring,22)==0)   { newHeaderSize=newHeaderSize*1; }
@@ -138,35 +189,26 @@ void ReadMHDPlane::ReadNormalMHD()
 
        if (ok==false)
        {
-               int i;
-               int sizeX, sizeY, sizeZ;
-               sizeX = 200;
-               sizeY = sizeX;
-               sizeZ = 1;
-               newImage = vtkImageData::New();
-               newImage->Initialize();
-               newImage->SetScalarTypeToUnsignedChar();
-               newImage->SetSpacing( 1,1,1 );
-               newImage->SetDimensions(  sizeX,sizeY,sizeZ );
-               newImage->SetWholeExtent(0,  sizeX-1,0,sizeY-1,0,sizeZ-1 );
-               newImage->SetExtent(0,  sizeX-1,0,sizeY-1,0,sizeZ-1 );
-               newImage->SetNumberOfScalarComponents(1);
-               newImage->AllocateScalars();
-               newImage->Update();
-               memset ( (void*)newImage->GetScalarPointer(), 0, sizeX*sizeY*1 );
-               for (i=0; i<sizeX; i++)
-               {
-                       newImage->SetScalarComponentFromDouble(i,i,0, 0, 255 );
-                       newImage->SetScalarComponentFromDouble(i,sizeY-1-i,0, 0, 255 );
-               } // for i
+               newImage=CreateDefaultImage();
        } // if ok
 
        bbSetOutputOut( newImage );
-
 }
 
+
+
+
+
 void ReadMHDPlane::Read64lseek()
 {
+       int             slice;
+       int             width;
+
+       width = bbGetInputWidth();
+       if (width<=0 ) { width=1; }
+
+       slice = bbGetInputSlice();
+       if (slice<0 ) { slice=0; }
 
        int             dimX=-1,dimY=-1,dimZ=-1;
        int             dim=-1;
@@ -176,7 +218,7 @@ void ReadMHDPlane::Read64lseek()
        float           ox=-1,oy=-1,oz=-1;
        long int        headersize=0;
 
-       vtkImageData *newImage;
+       vtkImageData *newImage=NULL;
 
        char mystring[250];
        char strTmp[30];
@@ -184,8 +226,6 @@ void ReadMHDPlane::Read64lseek()
        FILE *ffIn      = fopen(bbGetInputFileName().c_str(),"r");
        long long dataSize;
 
-
-
        if (ffIn!=NULL)
        {
                newImage = vtkImageData::New();
@@ -202,18 +242,18 @@ void ReadMHDPlane::Read64lseek()
                if (strncmp("HeaderSize",mystring,10)==0)               { sscanf(mystring,"%s %s %ld"           ,strTmp, strTmp, &headersize);                  }
                        if (strncmp("ElementDataFile",mystring,15)==0)  { sscanf(mystring,"%s %s %s"            ,strTmp, strTmp, strTmp2); elementdatafile=strTmp2;             }
 
-               if (strncmp("ElementType = MET_CHAR",mystring,22)==0)   { newImage->SetScalarTypeToChar();                      dataSize=sizeof(char);                  }
-               if (strncmp("ElementType = VTK_CHAR",mystring,22)==0)   { newImage->SetScalarTypeToChar();                      dataSize=sizeof(char);                  }
-               if (strncmp("ElementType = MET_UCHAR",mystring,23)==0)  { newImage->SetScalarTypeToUnsignedChar();      dataSize=sizeof(unsigned char); }
+               if (strncmp("ElementType = MET_CHAR",mystring,22)==0)                   { newImage->SetScalarTypeToChar();                      dataSize=sizeof(char);                  }
+               if (strncmp("ElementType = VTK_CHAR",mystring,22)==0)                   { newImage->SetScalarTypeToChar();                      dataSize=sizeof(char);                  }
+               if (strncmp("ElementType = MET_UCHAR",mystring,23)==0)                  { newImage->SetScalarTypeToUnsignedChar();      dataSize=sizeof(unsigned char); }
                if (strncmp("ElementType = VTK_UNSIGNED_CHAR",mystring,31)==0)  { newImage->SetScalarTypeToUnsignedChar();      dataSize=sizeof(unsigned char); }
-               if (strncmp("ElementType = MET_USHORT",mystring,24)==0) { newImage->SetScalarTypeToUnsignedShort();     dataSize=sizeof(unsigned short);}
+               if (strncmp("ElementType = MET_USHORT",mystring,24)==0)                 { newImage->SetScalarTypeToUnsignedShort();     dataSize=sizeof(unsigned short);}
                if (strncmp("ElementType = VTK_UNSIGNED_SHORT",mystring,32)==0) { newImage->SetScalarTypeToUnsignedShort();     dataSize=sizeof(unsigned short);}
-               if (strncmp("ElementType = MET_SHORT",mystring,23)==0)  { newImage->SetScalarTypeToShort();                     dataSize=sizeof(short);                 }
-               if (strncmp("ElementType = VTK_SHORT",mystring,23)==0)  { newImage->SetScalarTypeToShort();                     dataSize=sizeof(short);                 }
-               if (strncmp("ElementType = MET_FLOAT",mystring,23)==0)  { newImage->SetScalarTypeToFloat();                     dataSize=sizeof(float);                 }
-               if (strncmp("ElementType = VTK_FLOAT",mystring,23)==0)  { newImage->SetScalarTypeToFloat();                     dataSize=sizeof(float);                 }
-               if (strncmp("ElementType = MET_DOUBLE",mystring,24)==0) { newImage->SetScalarTypeToDouble();            dataSize=sizeof(double);                }
-               if (strncmp("ElementType = VTK_DOUBLE",mystring,24)==0) { newImage->SetScalarTypeToDouble();            dataSize=sizeof(double);                }
+               if (strncmp("ElementType = MET_SHORT",mystring,23)==0)                  { newImage->SetScalarTypeToShort();                     dataSize=sizeof(short);                 }
+               if (strncmp("ElementType = VTK_SHORT",mystring,23)==0)                  { newImage->SetScalarTypeToShort();                     dataSize=sizeof(short);                 }
+               if (strncmp("ElementType = MET_FLOAT",mystring,23)==0)                  { newImage->SetScalarTypeToFloat();                     dataSize=sizeof(float);                 }
+               if (strncmp("ElementType = VTK_FLOAT",mystring,23)==0)                  { newImage->SetScalarTypeToFloat();                     dataSize=sizeof(float);                 }
+               if (strncmp("ElementType = MET_DOUBLE",mystring,24)==0)                 { newImage->SetScalarTypeToDouble();            dataSize=sizeof(double);                }
+               if (strncmp("ElementType = VTK_DOUBLE",mystring,24)==0)                 { newImage->SetScalarTypeToDouble();            dataSize=sizeof(double);                }
 
        } // while
                fclose(ffIn);
@@ -221,15 +261,15 @@ void ReadMHDPlane::Read64lseek()
                newImage->Initialize();
 
 
-               int fd;
-               long long ret;
+               int             fd;
+               long long       ret;
                std::size_t found;
                std::string filename;
-               found                                   =       bbGetInputFileName().find_last_of("/\\");
+               found                                   = bbGetInputFileName().find_last_of("/\\");
                filename                                = bbGetInputFileName().substr(0,found+1) + elementdatafile ;
-               long long pos;
+               long long       pos;
 
-               long long lsize                 = dimX*dimY*1 *dataSize;
+               long long lsize                 = dimX*dimY*width *dataSize;
 
 
 #if defined(_WIN32)
@@ -250,22 +290,18 @@ void ReadMHDPlane::Read64lseek()
                if (bbGetInputDirectionPlane()=="XY") 
                { 
                        newImage->SetSpacing( spcX,spcY,spcZ );
-                       newImage->SetDimensions(  dimX,dimY,1 );
-                       newImage->SetWholeExtent(0,  dimX-1,0,dimY-1,0,0 );
-                       newImage->SetExtent(0,  dimX-1,0,dimY-1,0,0 );
+                       newImage->SetDimensions(  dimX,dimY,width );
+                       newImage->SetWholeExtent(0,  dimX-1,0,dimY-1,0,width-1 );
+                       newImage->SetExtent(0,  dimX-1,0,dimY-1,0,width-1 );
                        newImage->SetNumberOfScalarComponents(1);
                        newImage->AllocateScalars();
                        newImage->Update();
-                       pos = dimX*dimY*(long long)bbGetInputSlice()*dataSize;
-
-
-
+                       pos = dimX*dimY*(long long)slice*dataSize;
 #if defined(_WIN32)
                        if (_lseeki64( fd, pos, SEEK_SET ) < 0)
 #else
                        if (lseek64(fd, pos, SEEK_SET) < 0) 
 #endif // defined(_WIN32)
-
                        {
                                printf("EED ReadMHDPlane::Read64lseek \n");
                                fprintf(stderr, "Failed seeking to %lld, %s\n", pos, strerror(errno));
@@ -279,91 +315,119 @@ void ReadMHDPlane::Read64lseek()
                        }
                }  // if PLANE XY
 
-               /*
                if (bbGetInputDirectionPlane()=="XZ") 
                { 
-                       long long int j;
                        newImage->SetSpacing( spcX,spcZ,spcY );
-                       newImage->SetDimensions(  dimX,dimZ,1 );
-                       newImage->SetWholeExtent(0,  dimX-1,0,dimZ-1,0,0 );
-                       newImage->SetExtent(0,  dimX-1,0,dimZ-1,0,0 );
+                       newImage->SetDimensions(  dimX,dimZ,width );
+                       newImage->SetWholeExtent(0,  dimX-1,0,dimZ-1,0,width-1 );
+                       newImage->SetExtent(0,  dimX-1,0,dimZ-1,0,width-1 );
                        newImage->SetNumberOfScalarComponents(1);
                        newImage->AllocateScalars();
                        newImage->Update();
-                       long long sizeBytesPlane = dimX*dimY*dataSize;
-                       pos = dimX*(long long int)bbGetInputSlice()*dataSize;
-                       void *pImage = newImage->GetScalarPointer();
-                       for (j=0;j<dimZ;j++)
+                       int iWidth;
+                       for (iWidth=0;iWidth<width;iWidth++)
                        {
-printf("EED EED ReadMHDPlane::Read64lseek XZ j=%d \n", j);
-                               if (lseek64(fd, pos + j*sizeBytesPlane , SEEK_SET) < 0) 
-                               {
-                                       printf("EED ReadMHDPlane::Read64lseek \n");
-                                       fprintf(stderr, "Failed seeking to %lld, %s\n", pos, strerror(errno));
-                                       exit(1);
-                               }
-
-                               if ((ret = read(fd, pImage , dimX*dataSize)) < 0) 
-                               {
-                                       fprintf(stderr, "Failed reading: %s\n", strerror(errno));
-                                       exit(1);
-                               }
-                               pImage=pImage+dimX*dataSize;
-                       } // for j
+                               copy_XZ_plane(fd,newImage,slice+iWidth,iWidth,dimX,dimY,dimZ,dataSize);
+                       }
                }  // if PLANE XZ
 
 
                if (bbGetInputDirectionPlane()=="YZ") 
                { 
-                       long long int j;
-                       long long int i;
                        newImage->SetSpacing( spcY,spcZ,spcX );
-                       newImage->SetDimensions(  dimY,dimZ,1 );
-                       newImage->SetWholeExtent(0,  dimY-1,0,dimZ-1,0,0 );
-                       newImage->SetExtent(0,  dimY-1,0,dimZ-1,0,0 );
+                       newImage->SetDimensions(  dimY,dimZ,width );
+                       newImage->SetWholeExtent(0,  dimY-1,0,dimZ-1,0,width-1 );
+                       newImage->SetExtent(0,  dimY-1,0,dimZ-1,0,width-1 );
                        newImage->SetNumberOfScalarComponents(1);
                        newImage->AllocateScalars();
                        newImage->Update();
-                       long long sizeBytesPlane = dimX*dimY*dataSize;
-//                     pos = dimX*(long long int)bbGetInputSlice()*dataSize;
-                       void *pImage;
-                       for (j=0;j<dimZ;j++)
+                       int iWidth;
+                       for (iWidth=0;iWidth<width;iWidth++)
                        {
-printf("EED EED ReadMHDPlane::Read64lseek YZ i=%d j=%d \n", i,j);
-                               for (i=0;i<dimY;i++)
-                               {
-                                       pos = ((long long int)bbGetInputSlice() + i*dimX + j*dimX*dimY)*dataSize ;
-                                       if (lseek64(fd, pos  , SEEK_SET) < 0) 
-                                       {
-                                               printf("EED ReadMHDPlane::Read64lseek \n");
-                                               fprintf(stderr, "Failed seeking to %lld, %s\n", pos, strerror(errno));
-                                               exit(1);
-                                       }
-
-                                       pImage=newImage->GetScalarPointer(i, j,0 );
-                                       if ((ret = read(fd, pImage , dataSize)) < 0) 
-                                       {
-                                               fprintf(stderr, "Failed reading: %s\n", strerror(errno));
-                                               exit(1);
-                                       }
-                               }
-                       } // for j
-               }  // if PLANE YZ
-               */
-       
+                               copy_YZ_plane(fd,newImage,slice+iWidth,iWidth,dimX,dimY,dimZ,dataSize);
+                       }
+
+               }  // if PLANE YZ       
 #if defined(_WIN32)
                _close (fd);
 #else
                close (fd);
 #endif // defined(_WIN32)
+       } else {
+               newImage=CreateDefaultImage();
+       } // if  ffIn
+
+       bbSetOutputOut( newImage );
+}
+
+
+void ReadMHDPlane::copy_YZ_plane(int fd,vtkImageData *newImage,int slice,int iWidth,int dimX,int dimY,int dimZ,int dataSize)
+{
+       long long int j;
+       long long int i;
+       long long       ret;
+       long long       pos;
+       long long sizeBytesPlane = dimX*dimY*dataSize;
+       char *pImage;
+       for (j=0;j<dimZ;j++)
+       {
+               for (i=0;i<dimY;i++)
+               {
+                       pos = ((long long int)slice+iWidth + i*dimX + j*dimX*dimY)*dataSize ;
+#if defined(_WIN32)
+                       if (_lseeki64( fd, pos , SEEK_SET ) < 0)
+#else
+                       if (lseek64(fd, pos , SEEK_SET) < 0) 
+#endif // defined(_WIN32)
+                       {
+                               printf("EED ReadMHDPlane::Read64lseek \n");
+                               fprintf(stderr, "Failed seeking to %lld, %s\n", pos, strerror(errno));
+                               exit(1);
+                       }
+
+                       pImage=(char*)(newImage->GetScalarPointer(i, j,iWidth ));
+                       if ((ret = read(fd, pImage , dataSize)) < 0) 
+                       {
+                               fprintf(stderr, "Failed reading: %s\n", strerror(errno));
+                               exit(1);
+                       }
+                       } // for i
+               } // for j
+}
 
-               bbSetOutputOut( newImage );
-       } // if
 
+void ReadMHDPlane::copy_XZ_plane(int fd,vtkImageData *newImage,int slice,int iWidth,int dimX,int dimY,int dimZ,int dataSize)
+{
+       long long int j;
+       long long       ret;
+       long long       pos;
+       pos = dimX*(long long int)slice*dataSize;
+       long long sizeBytesPlane = dimX*dimY*dataSize;
+       char *pImage = (char*)( newImage->GetScalarPointer(0,0,iWidth) );
+       for (j=0;j<dimZ;j++)
+       {
+#if defined(_WIN32)
+               if (_lseeki64( fd, pos + j*sizeBytesPlane , SEEK_SET ) < 0)
+#else
+               if (lseek64(fd, pos + j*sizeBytesPlane , SEEK_SET) < 0) 
+#endif // defined(_WIN32)
+               {
+                       printf("EED ReadMHDPlane::Read64lseek \n");
+                       fprintf(stderr, "Failed seeking to %lld, %s\n", pos, strerror(errno));
+                       exit(1);
+               }
 
+               if ((ret = read(fd, pImage , dimX*dataSize)) < 0) 
+               {
+                       fprintf(stderr, "Failed reading: %s\n", strerror(errno));
+                       exit(1);
+               }
+               pImage = pImage+dimX*dataSize;
+       } // for j
 }
 
 
+
 void ReadMHDPlane::Process()
 {
        if (bbGetInputType()==0)
@@ -387,7 +451,8 @@ void ReadMHDPlane::bbUserSetDefaultValues()
 //    Here we initialize the input 'In' to 0
    bbSetInputFileName("");
    bbSetInputSlice(0);
-   bbSetInputType(0);
+   bbSetInputType(1);
+   bbSetInputWidth(1);
    bbSetInputDirectionPlane("XY");
   
 }