Deal with 'float' and 'double' Raw files.

[gdcm.git] / Example / RawToDicomStack.cxx

diff --git a/Example/RawToDicomStack.cxx b/Example/RawToDicomStack.cxx
index 314debc838afa0b5c07b263a3ca25b0b47628e8c..70e0212d00c39204429023edcaf09bf44cfacd26 100755 (executable)
--- a/Example/RawToDicomStack.cxx
+++ b/Example/RawToDicomStack.cxx
@@ -3,8 +3,8 @@
   Program:   gdcm
   Module:    $RCSfile: RawToDicomStack.cxx,v $
   Language:  C++
-  Date:      $Date: 2008/06/12 13:26:39 $
-  Version:   $Revision: 1.1 $
+  Date:      $Date: 2009/01/14 14:07:40 $
+  Version:   $Revision: 1.3 $
                                                                                 
   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
   l'Image). All rights reserved. See Doc/License.txt or
@@ -30,26 +30,31 @@
 #include <iostream>
 #include <sstream>
 
-typedef char * PS8;
-typedef unsigned char * PU8;
-typedef short int * PS16;
+typedef char               * PS8;
+typedef unsigned char      * PU8;
+typedef short int          * PS16;
 typedef unsigned short int * PU16;
-typedef int * PS32;
-typedef unsigned int * PU32;
+typedef int                * PS32;
+typedef unsigned int       * PU32;
+typedef float              * PF32;
+typedef double             * PD64;
 
 #define CRR(t1,t2)     { for(int i=0;i<nY;i++)                   \
                            for(int j=0;j<nX;j++)                 \
-                             for(int k=0;k<samplesPerPixel;k++)  \
-                                 *((t2)planePixels + i*nX +j+k) = *(((t1)pixels)+ nbPlanes*nX*nY + i*nX +j+k);\
+                             for(int k=0;k<samplesPerPixel;k++)  {\
+                                 *((t2)planePixelsOut + i*nX +j+k) = *(((t1)pixels)+ nbPlanes*nX*nY + i*nX +j+k);\
+     }                                                                              \
                        }
     
 #define CFR(PPt)  switch ( pixelTypeOutCode ) {     \
-          case -8   : CRR(PPt,PS8); break;          \
-          case 8    : CRR(PPt,PU8); break;          \
+          case -8   : CRR(PPt,PS8);  break;         \
+          case 8    : CRR(PPt,PU8);  break;         \
           case -16  : CRR(PPt,PS16); break;         \
           case 16   : CRR(PPt,PU16); break;         \
           case -32  : CRR(PPt,PS32); break;         \
           case 32   : CRR(PPt,PU32); break;         \
+          case 33   : CRR(PPt,PF32); break;         \
+          case 64   : CRR(PPt,PD64); break;         \
           }
 
 
@@ -94,7 +99,7 @@ int main(int argc, char *argv[])
    "                   rows=nb of Rows                                        ",
    "                   lines=nb of Lines,                                     ",
    "                   [frames = nb of Frames] //defaulted to 1               ",
-   "                   pixeltype={8U|8S|16U|16S|32U|32S}                      ",
+   "                   pixeltype={8U|8S|16U|16S|32U|32S|32F|64D}              ",
    "                   pixeltypeout={8U|8S|16U|16S|32U|32S}                   ",   
    "                   [{b|l}] b:BigEndian,l:LittleEndian default : l         ",
    "                   [samples = {1|3}}       //(1:Gray,3:RGB) defaulted to 1",
@@ -122,7 +127,7 @@ int main(int argc, char *argv[])
    const char *inputFileName          = am->ArgMgrGetString("filein");
    const char *outputSkeletonFileName = am->ArgMgrGetString("fileout");
    
-  const char *patientName = am->ArgMgrGetString("patientname", "g^Fantomas");
+   const char *patientName = am->ArgMgrGetString("patientname", "g^Fantomas");
    
    int nX = am->ArgMgrWantInt("rows", usage);
    int nY = am->ArgMgrWantInt("lines", usage);
@@ -219,6 +224,19 @@ int main(int argc, char *argv[])
       pixelSign = 0;
       pixelTypeCode = 32;       
    }
+   else if (strPixelType == "32F")
+   {
+      pixelSize = 4;
+      pixelSign = 0;
+      pixelTypeCode = 33;       
+   }   
+   
+   else if (strPixelType == "64D")
+   {
+      pixelSize = 8;
+      pixelSign = 0;
+      pixelTypeCode = 64;       
+   }
    else
    {
       std::cout << "Wrong 'pixeltype' (" << strPixelType << ")" << std::endl;
@@ -228,7 +246,7 @@ int main(int argc, char *argv[])
    std::string strPixelTypeOut(pixelTypeOut);
    int pixelSignOut;
    int pixelSizeOut;
-   int pixelTypeOutCode; // for the swirch case
+   int pixelTypeOutCode; // for the switch case
     
    if (strPixelTypeOut == "8S")
    {
@@ -288,37 +306,40 @@ int main(int argc, char *argv[])
    // Read the pixels
 
    int singlePlaneDataSize =  nX*nY*samplesPerPixel*pixelSizeOut;
-   int dataSize            =  singlePlaneDataSize*nZ;
+   int dataSizeIn          =  nX*nY*samplesPerPixel*pixelSize*nZ;
 
-   uint8_t *pixels = new uint8_t[dataSize];
-   uint8_t *planePixels = new uint8_t[singlePlaneDataSize];   
+   uint8_t *pixels         = new uint8_t[dataSizeIn];
+   uint8_t *planePixelsOut = new uint8_t[singlePlaneDataSize];   
    
-   Fp->read((char*)pixels, (size_t)dataSize);
-     
+   Fp->read((char*)pixels, (size_t)dataSizeIn);
+
    if ( pixelSize !=1 && ( (l && bigEndian) || (b && ! bigEndian) ) )
    {  
-      ConvertSwapZone(pixelSize, pixels, dataSize);   
+      ConvertSwapZone(pixelSize, pixels, dataSizeIn);   
    }
 
 // iterate on the planes.
 
    char outputFileName[200];
    
-for(int nbPlanes=0; nbPlanes<nZ; nbPlanes++)
-{
+   for(int nbPlanes=0; nbPlanes<nZ; nbPlanes++)
+   {
 
-sprintf(outputFileName, "%s.%04d.dcm", outputSkeletonFileName,nbPlanes);
+      sprintf(outputFileName, "%s.%04d.dcm", outputSkeletonFileName,nbPlanes);
 
 // Copy (and convert) pixels of a single plane
 
-  switch ( pixelTypeCode ) {
-    case 8    : CFR(PU8);  break;
-    case -8   : CFR(PS8);  break;
-    case -16  : CFR(PU16); break;
-    case 16   : CFR(PS16); break;
-    case -32  : CFR(PS32); break;
-    case 32   : CFR(PU32); break;
-  }
+     switch ( pixelTypeCode ) 
+     {
+       case 8    : CFR(PU8);  break;
+       case -8   : CFR(PS8);  break;
+       case -16  : CFR(PU16); break;
+       case 16   : CFR(PS16); break;
+       case -32  : CFR(PS32); break;
+       case 32   : CFR(PU32); break;
+       case 33   : CFR(PF32); break;
+       case 64   : CFR(PD64); break;       
+     }
 
 // Create an empty FileHelper
 
@@ -356,15 +377,15 @@ sprintf(outputFileName, "%s.%04d.dcm", outputSkeletonFileName,nbPlanes);
    // Set the pixel type
    
    str.str("");
-   str << pixelSize*8;
+   str << pixelSizeOut*8;
    fileToBuild->InsertEntryString(str.str(),0x0028,0x0100, "US"); // Bits Allocated
 
    str.str("");
-   str << pixelSize*8;
+   str << pixelSizeOut*8;
    fileToBuild->InsertEntryString(str.str(),0x0028,0x0101, "US"); // Bits Stored
 
    str.str("");
-   str << ( pixelSize*8 - 1 );
+   str << ( pixelSizeOut*8 - 1 );
    fileToBuild->InsertEntryString(str.str(),0x0028,0x0102, "US"); // High Bit
 
    str.str("");
@@ -393,7 +414,7 @@ sprintf(outputFileName, "%s.%04d.dcm", outputSkeletonFileName,nbPlanes);
       fileH->SetPhotometricInterpretationToMonochrome1();
      
 // Set the image Pixel Data (plane)
-   fileH->SetImageData(planePixels,singlePlaneDataSize);
+   fileH->SetImageData(planePixelsOut,singlePlaneDataSize);
 
 // Set the writting mode and write the image
    fileH->SetWriteModeToRaw();
@@ -412,6 +433,6 @@ sprintf(outputFileName, "%s.%04d.dcm", outputSkeletonFileName,nbPlanes);
 } // end iterate on the planes
 
    delete[] pixels;
-   delete[] planePixels;
+   delete[] planePixelsOut;
    return 1;
 }