]> Creatis software - gdcm.git/blobdiff - src/gdcmFile.cxx
* Amelioration of code, remove some code copy. For the loading of
[gdcm.git] / src / gdcmFile.cxx
index b5de832e9f59741c52597ffb7aced3ab12230b10..666a8e96f5fdd0245ddfb14ec1d15cb82937732e 100644 (file)
@@ -3,8 +3,8 @@
   Program:   gdcm
   Module:    $RCSfile: gdcmFile.cxx,v $
   Language:  C++
-  Date:      $Date: 2004/11/16 02:54:35 $
-  Version:   $Revision: 1.155 $
+  Date:      $Date: 2004/11/24 11:17:47 $
+  Version:   $Revision: 1.161 $
                                                                                 
   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
   l'Image). All rights reserved. See Doc/License.txt or
@@ -18,6 +18,8 @@
 
 #include "gdcmFile.h"
 #include "gdcmDebug.h"
+#include "gdcmUtil.h"
+#include "gdcmBinEntry.h"
 #include <fstream>
 
 namespace gdcm 
@@ -73,7 +75,12 @@ File::File(std::string const & filename )
  */
 void File::Initialise()
 {
-   PixelConverter = NULL; //just in case
+   WriteMode = WMODE_RGB;
+   WriteType = WTYPE_IMPL_VR;
+
+   PixelConverter = new PixelConvert;
+   Archive = new DocEntryArchive( HeaderInternal );
+
    if ( HeaderInternal->IsReadable() )
    {
       ImageDataSizeRaw = ComputeDecompressedPixelDataSizeFromHeader();
@@ -86,9 +93,9 @@ void File::Initialise()
          ImageDataSize = ImageDataSizeRaw;
       }
 
-      PixelConverter = new PixelConvert;  //LEAK !
       PixelConverter->GrabInformationsFromHeader( HeaderInternal );
    }
+
    SaveInitialValues();
 }
 
@@ -99,17 +106,20 @@ void File::Initialise()
  */
 File::~File()
 { 
-   if( SelfHeader )
+   if( PixelConverter )
    {
-      delete HeaderInternal;
+      delete PixelConverter;
+   }
+   if( Archive )
+   {
+      delete Archive;
    }
-   HeaderInternal = 0;
 
-   DeleteInitialValues();
-   if( PixelConverter )
+   if( SelfHeader )
    {
-      delete PixelConverter;
+      delete HeaderInternal;
    }
+   HeaderInternal = 0;
 }
 
 /**
@@ -118,120 +128,8 @@ File::~File()
  */
 void File::SaveInitialValues()
 { 
-
    PixelRead  = -1; // no ImageData read yet.
-   LastAllocatedPixelDataLength = 0;
    Pixel_Data = 0;
-
-   InitialSpp = "";     
-   InitialPhotInt = "";
-   InitialPlanConfig = "";
-   InitialBitsAllocated = "";
-   InitialHighBit = "";
-  
-   InitialRedLUTDescr   = 0;
-   InitialGreenLUTDescr = 0;
-   InitialBlueLUTDescr  = 0;
-   InitialRedLUTData    = 0;
-   InitialGreenLUTData  = 0;
-   InitialBlueLUTData   = 0; 
-                
-   if ( HeaderInternal->IsReadable() )
-   {
-      // the following values *may* be modified 
-      // by File::GetImageDataIntoVectorRaw
-      // we save their initial value.
-      InitialSpp           = HeaderInternal->GetEntryByNumber(0x0028,0x0002);
-      InitialPhotInt       = HeaderInternal->GetEntryByNumber(0x0028,0x0004);
-      InitialPlanConfig    = HeaderInternal->GetEntryByNumber(0x0028,0x0006);
-      
-      InitialBitsAllocated = HeaderInternal->GetEntryByNumber(0x0028,0x0100);
-      InitialHighBit       = HeaderInternal->GetEntryByNumber(0x0028,0x0102);
-
-      // the following entries *may* be removed from the H table
-      // (NOT deleted ...) by File::GetImageDataIntoVectorRaw  
-      // we keep a pointer on them.
-      InitialRedLUTDescr   = HeaderInternal->GetDocEntryByNumber(0x0028,0x1101);
-      InitialGreenLUTDescr = HeaderInternal->GetDocEntryByNumber(0x0028,0x1102);
-      InitialBlueLUTDescr  = HeaderInternal->GetDocEntryByNumber(0x0028,0x1103);
-
-      InitialRedLUTData    = HeaderInternal->GetDocEntryByNumber(0x0028,0x1201);
-      InitialGreenLUTData  = HeaderInternal->GetDocEntryByNumber(0x0028,0x1202);
-      InitialBlueLUTData   = HeaderInternal->GetDocEntryByNumber(0x0028,0x1203); 
-   }
-}
-
-/**
- * \brief restores some initial values
- * \warning not end user intended
- */
-void File::RestoreInitialValues()
-{   
-   if ( HeaderInternal->IsReadable() )
-   {      
-      // the following values *may* have been modified 
-      // by File::GetImageDataIntoVectorRaw
-      // we restore their initial value.
-      if ( InitialSpp != "")
-         HeaderInternal->SetEntryByNumber(InitialSpp,0x0028,0x0002);
-      if ( InitialPhotInt != "")
-         HeaderInternal->SetEntryByNumber(InitialPhotInt,0x0028,0x0004);
-      if ( InitialPlanConfig != "")
-
-         HeaderInternal->SetEntryByNumber(InitialPlanConfig,0x0028,0x0006);
-      if ( InitialBitsAllocated != "")
-          HeaderInternal->SetEntryByNumber(InitialBitsAllocated,0x0028,0x0100);
-      if ( InitialHighBit != "")
-          HeaderInternal->SetEntryByNumber(InitialHighBit,0x0028,0x0102);
-               
-      // the following entries *may* be have been removed from the H table
-      // (NOT deleted ...) by File::GetImageDataIntoVectorRaw  
-      // we restore them.
-
-      if (InitialRedLUTDescr)
-         HeaderInternal->AddEntry(InitialRedLUTDescr);
-      if (InitialGreenLUTDescr)
-         HeaderInternal->AddEntry(InitialGreenLUTDescr);
-      if (InitialBlueLUTDescr)
-         HeaderInternal->AddEntry(InitialBlueLUTDescr);
-
-      if (InitialRedLUTData)
-         HeaderInternal->AddEntry(InitialBlueLUTDescr);
-      if (InitialGreenLUTData)
-         HeaderInternal->AddEntry(InitialGreenLUTData);
-      if (InitialBlueLUTData)
-         HeaderInternal->AddEntry(InitialBlueLUTData);
-   }
-}
-
-/**
- * \brief delete initial values (il they were saved)
- *        of InitialLutDescriptors and InitialLutData
- */
-void File::DeleteInitialValues()
-{ 
-
-// InitialLutDescriptors and InitialLutData
-// will have to be deleted if the don't belong any longer
-// to the Header H table when the header is deleted...
-
-   if ( InitialRedLUTDescr )           
-      delete InitialRedLUTDescr;
-  
-   if ( InitialGreenLUTDescr )
-      delete InitialGreenLUTDescr;
-      
-   if ( InitialBlueLUTDescr )      
-      delete InitialBlueLUTDescr; 
-       
-   if ( InitialRedLUTData )      
-      delete InitialRedLUTData;
-   
-   if ( InitialGreenLUTData != NULL)
-      delete InitialGreenLUTData;
-      
-   if ( InitialBlueLUTData != NULL)      
-      delete InitialBlueLUTData;      
 }
 
 //-----------------------------------------------------------------------------
@@ -311,6 +209,22 @@ uint8_t* File::GetImageData()
       return 0;
    }
                                                                                 
+   if ( HeaderInternal->HasLUT() && PixelConverter->BuildRGBImage() )
+   {
+      return PixelConverter->GetRGB();
+   }
+   else
+   {
+      // When no LUT or LUT conversion fails, return the decompressed
+      return PixelConverter->GetDecompressed();
+   }
+   
+/*   if ( ! GetDecompressed() )
+   {
+      // If the decompression failed nothing can be done.
+      return 0;
+   }
+                                                                                
    uint8_t* pixelData;
    if ( HeaderInternal->HasLUT() && PixelConverter->BuildRGBImage() )
    {
@@ -346,7 +260,7 @@ uint8_t* File::GetImageData()
       if ( HeaderInternal->HasLUT() )
       {
          // The LUT interpretation failed
-         std::string photometricInterpretation = "MONOCHROME1 ";
+         std::string photometricInterpretation = Util::DicomString("MONOCHROME1");
          HeaderInternal->SetEntryByNumber( photometricInterpretation,
                                            0x0028, 0x0004 );
          PixelRead = 0; // no PixelRaw
@@ -376,15 +290,10 @@ uint8_t* File::GetImageData()
    }
 
    // We say the value *is* loaded.
-   GetHeader()->SetEntryByNumber( GDCM_BINLOADED,
-      GetHeader()->GetGrPixel(), GetHeader()->GetNumPixel());
-
-   // Will be 7fe0, 0010 in standard case
-   GetHeader()->SetEntryBinAreaByNumber( pixelData, 
-      GetHeader()->GetGrPixel(), GetHeader()->GetNumPixel()); 
+   SetPixelData(pixelData);
 // END PIXELCONVERT CLEANME
 
-   return pixelData;
+   return pixelData;*/
 }
 
 /**
@@ -428,9 +337,9 @@ size_t File::GetImageDataIntoVector (void* destination, size_t maxSize)
                         "than caller's expected MaxSize");
          return 0;
       }
-      memmove( destination,
-               (void*)PixelConverter->GetRGB(),
-               PixelConverter->GetRGBSize() );
+      memcpy( destination,
+              (void*)PixelConverter->GetRGB(),
+              PixelConverter->GetRGBSize() );
       return PixelConverter->GetRGBSize();
    }
 
@@ -441,9 +350,9 @@ size_t File::GetImageDataIntoVector (void* destination, size_t maxSize)
                      "than caller's expected MaxSize");
       return 0;
    }
-   memmove( destination,
-            (void*)PixelConverter->GetDecompressed(),
-            PixelConverter->GetDecompressedSize() );
+   memcpy( destination,
+           (void*)PixelConverter->GetDecompressed(),
+           PixelConverter->GetDecompressedSize() );
    return PixelConverter->GetDecompressedSize();
 }
 
@@ -458,7 +367,8 @@ size_t File::GetImageDataIntoVector (void* destination, size_t maxSize)
  */
 uint8_t* File::GetImageDataRaw ()
 {
-   uint8_t* decompressed = GetDecompressed();
+   return GetDecompressed();
+/*   uint8_t* decompressed = GetDecompressed();
    if ( ! decompressed )
    {
       return 0;
@@ -491,17 +401,12 @@ uint8_t* File::GetImageDataRaw ()
    }
 
    // We say the value *is* loaded.
-   GetHeader()->SetEntryByNumber( GDCM_BINLOADED,
-   GetHeader()->GetGrPixel(), GetHeader()->GetNumPixel());
-   // will be 7fe0, 0010 in standard cases
-   GetHeader()->SetEntryBinAreaByNumber( decompressed,
-   GetHeader()->GetGrPixel(), GetHeader()->GetNumPixel());
+   SetPixelData(decompressed);
  
    PixelRead = 1; // PixelRaw
 // END PIXELCONVERT CLEANME
 
-   return decompressed;
+   return decompressed;*/
 }
 
 uint8_t* File::GetDecompressed()
@@ -583,7 +488,8 @@ bool File::WriteRawData(std::string const & fileName)
 
 bool File::WriteDcmImplVR (std::string const & fileName)
 {
-   return WriteBase(fileName, ImplicitVR);
+   SetWriteTypeToDcmImplVR();
+   return Write(fileName);
 }
 
 /**
@@ -596,7 +502,8 @@ bool File::WriteDcmImplVR (std::string const & fileName)
 
 bool File::WriteDcmExplVR (std::string const & fileName)
 {
-   return WriteBase(fileName, ExplicitVR);
+   SetWriteTypeToDcmExplVR();
+   return Write(fileName);
 }
 
 /**
@@ -615,7 +522,22 @@ bool File::WriteDcmExplVR (std::string const & fileName)
 
 bool File::WriteAcr (std::string const & fileName)
 {
-   return WriteBase(fileName, ACR);
+   SetWriteTypeToAcr();
+   return Write(fileName);
+}
+
+bool File::Write(std::string const& fileName)
+{
+   switch(WriteType)
+   {
+      case WTYPE_IMPL_VR:
+         return WriteBase(fileName,ImplicitVR);
+      case WTYPE_EXPL_VR:
+         return WriteBase(fileName,ExplicitVR);
+      case WTYPE_ACR:
+         return WriteBase(fileName,ACR);
+   }
+   return(false);
 }
 
 //-----------------------------------------------------------------------------
@@ -630,10 +552,10 @@ bool File::WriteAcr (std::string const & fileName)
  */
 bool File::WriteBase (std::string const & fileName, FileType type)
 {
-   if ( PixelRead == -1 && type != ExplicitVR)
+/*   if ( PixelRead == -1 && type != ExplicitVR)
    {
       return false;
-   }
+   }*/
 
    std::ofstream* fp1 = new std::ofstream(fileName.c_str(), 
                               std::ios::out | std::ios::binary);
@@ -652,6 +574,19 @@ bool File::WriteBase (std::string const & fileName, FileType type)
       fp1->write("DICM", 4);
    }
 
+   switch(WriteMode)
+   {
+      case WMODE_NATIVE :
+         SetWriteToNative();
+         break;
+      case WMODE_DECOMPRESSED :
+         SetWriteToDecompressed();
+         break;
+      case WMODE_RGB :
+         SetWriteToRGB();
+         break;
+   }
+
    // --------------------------------------------------------------
    // Special Patch to allow gdcm to re-write ACR-LibIDO formated images
    //
@@ -665,15 +600,11 @@ bool File::WriteBase (std::string const & fileName, FileType type)
    std::string rows, columns; 
    if ( HeaderInternal->GetFileType() == ACR_LIBIDO)
    {
-      rows    = HeaderInternal->GetEntryByNumber(0x0028, 0x0010);
-      columns = HeaderInternal->GetEntryByNumber(0x0028, 0x0011);
-
-      HeaderInternal->SetEntryByNumber(columns,  0x0028, 0x0010);
-      HeaderInternal->SetEntryByNumber(rows   ,  0x0028, 0x0011);
+      SetWriteToLibido();
    }
    // ----------------- End of Special Patch ----------------
       
-   uint16_t grPixel  = HeaderInternal->GetGrPixel();
+/*   uint16_t grPixel  = HeaderInternal->GetGrPixel();
    uint16_t numPixel = HeaderInternal->GetNumPixel();;
           
    DocEntry* PixelElement = 
@@ -688,7 +619,7 @@ bool File::WriteBase (std::string const & fileName, FileType type)
    {
       // we tranformed GrayLevel pixels + LUT into RGB Pixel
       PixelElement->SetLength( ImageDataSize );
-   }
+   }*/
  
    HeaderInternal->Write(fp1, type);
 
@@ -700,10 +631,13 @@ bool File::WriteBase (std::string const & fileName, FileType type)
 
    if ( HeaderInternal->GetFileType() == ACR_LIBIDO )
    {
-      HeaderInternal->SetEntryByNumber(rows   , 0x0028, 0x0010);
-      HeaderInternal->SetEntryByNumber(columns, 0x0028, 0x0011);
+      RestoreWriteFromLibido();
    }
    // ----------------- End of Special Patch ----------------
+
+   RestoreWrite();
+
+
    fp1->close ();
    delete fp1;
 
@@ -718,5 +652,164 @@ uint8_t* File::GetLutRGBA()
    return PixelConverter->GetLutRGBA();
 }
 
+//-----------------------------------------------------------------------------
+// Private
+/**
+ * \brief Set the pixel datas in the good entry of the Header
+ */
+void File::SetPixelData(uint8_t* data)
+{
+   GetHeader()->SetEntryByNumber( GDCM_BINLOADED,
+      GetHeader()->GetGrPixel(), GetHeader()->GetNumPixel());
+
+   // Will be 7fe0, 0010 in standard case
+   DocEntry* currentEntry = GetHeader()->GetDocEntryByNumber(GetHeader()->GetGrPixel(), GetHeader()->GetNumPixel());
+   if ( currentEntry )
+   {
+      if ( BinEntry* binEntry = dynamic_cast<BinEntry *>(currentEntry) )
+         // Flag is to false because datas are kept in the gdcmPixelConvert
+         binEntry->SetBinArea( data, false );
+   }
+}
+
+void File::SetWriteToNative()
+{
+// Nothing to do
+}
+
+void File::SetWriteToDecompressed()
+{
+//   if (( !HeaderInternal->HasLUT() ) || (!PixelConverter->BuildRGBImage()))
+   if(HeaderInternal->HasLUT() && PixelConverter->BuildRGBImage())
+   {
+      SetWriteToRGB();
+   } 
+   else
+   {
+      ValEntry* photInt = CopyValEntry(0x0028,0x0004);
+      photInt->SetValue("MONOCHROME1 ");
+      photInt->SetLength(12);
+
+      BinEntry* pixel = CopyBinEntry(GetHeader()->GetGrPixel(),GetHeader()->GetNumPixel());
+      pixel->SetValue(GDCM_BINLOADED);
+      pixel->SetBinArea(PixelConverter->GetDecompressed(),false);
+      pixel->SetLength(PixelConverter->GetDecompressedSize());
+
+      Archive->Push(photInt);
+      Archive->Push(pixel);
+   }
+/*   else 
+   {
+      SetWriteToRGB();
+   } */
+}
+
+void File::SetWriteToRGB()
+{
+   if(PixelConverter->BuildRGBImage())
+   {
+      ValEntry* spp = CopyValEntry(0x0028,0x0002);
+      spp->SetValue("3 ");
+      spp->SetLength(2);
+
+      ValEntry* photInt = CopyValEntry(0x0028,0x0004);
+      photInt->SetValue("RGB ");
+      photInt->SetLength(4);
+
+      ValEntry* planConfig = CopyValEntry(0x0028,0x0006);
+      planConfig->SetValue("0 ");
+      planConfig->SetLength(2);
+
+      BinEntry* pixel = CopyBinEntry(GetHeader()->GetGrPixel(),GetHeader()->GetNumPixel());
+      pixel->SetValue(GDCM_BINLOADED);
+      pixel->SetBinArea(PixelConverter->GetRGB(),false);
+      pixel->SetLength(PixelConverter->GetRGBSize());
+
+      Archive->Push(spp);
+      Archive->Push(photInt);
+      Archive->Push(planConfig);
+      Archive->Push(pixel);
+   }
+   else
+   {
+      SetWriteToDecompressed();
+   }
+}
+
+void File::RestoreWrite()
+{
+   Archive->Restore(0x0028,0x0002);
+   Archive->Restore(0x0028,0x0004);
+   Archive->Restore(0x0028,0x0006);
+   Archive->Restore(GetHeader()->GetGrPixel(),GetHeader()->GetNumPixel());
+}
+
+void File::SetWriteToLibido()
+{
+   ValEntry *oldRow = dynamic_cast<ValEntry *>(HeaderInternal->GetDocEntryByNumber(0x0028, 0x0010));
+   ValEntry *oldCol = dynamic_cast<ValEntry *>(HeaderInternal->GetDocEntryByNumber(0x0028, 0x0011));
+   
+   if( oldRow && oldCol )
+   {
+      std::string rows, columns; 
+
+      ValEntry *newRow=new ValEntry(oldRow->GetDictEntry());
+      ValEntry *newCol=new ValEntry(oldCol->GetDictEntry());
+
+      newRow->Copy(oldCol);
+      newCol->Copy(oldRow);
+
+      newRow->SetValue(oldCol->GetValue());
+      newCol->SetValue(oldRow->GetValue());
+
+      Archive->Push(newRow);
+      Archive->Push(newCol);
+   }
+}
+
+void File::RestoreWriteFromLibido()
+{
+   Archive->Restore(0x0028,0x0010);
+   Archive->Restore(0x0028,0x0011);
+}
+
+ValEntry* File::CopyValEntry(uint16_t group,uint16_t element)
+{
+   DocEntry* oldE = HeaderInternal->GetDocEntryByNumber(group, element);
+   ValEntry* newE;
+
+   if(oldE)
+   {
+      newE = new ValEntry(oldE->GetDictEntry());
+      newE->Copy(oldE);
+   }
+   else
+   {
+      newE = GetHeader()->NewValEntryByNumber(group,element);
+   }
+
+   return(newE);
+}
+
+BinEntry* File::CopyBinEntry(uint16_t group,uint16_t element)
+{
+   DocEntry* oldE = HeaderInternal->GetDocEntryByNumber(group, element);
+   BinEntry* newE;
+
+   if(oldE)
+   {
+      newE = new BinEntry(oldE->GetDictEntry());
+      newE->Copy(oldE);
+   }
+   else
+   {
+      newE = GetHeader()->NewBinEntryByNumber(group,element);
+   }
+
+
+   return(newE);
+}
+
+//-----------------------------------------------------------------------------
 } // end namespace gdcm