/*=========================================================================
                                                                                
  Program:   gdcm
  Module:    $RCSfile: exImageLighten.cxx,v $
  Language:  C++
  Date:      $Date: 2005/07/07 17:31:54 $
  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
  http://www.creatis.insa-lyon.fr/Public/Gdcm/License.html for details.
                                                                                
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.
                                                                                
=========================================================================*/

#include "gdcmFile.h"
#include "gdcmFileHelper.h"
#include "gdcmDocument.h"
#include "gdcmValEntry.h"
#include "gdcmBinEntry.h"
#include "gdcmSeqEntry.h"

#include <stdlib.h> // for exit

int main(int argc, char *argv[])
{ 
   std::cout << "-----------------------------------------------" << std::endl;
   std::cout << "Removes from a full gdcm-readable  Dicom image"  << std::endl;
   std::cout << " all the 'Shadow groups' and the 'Sequence' entries"
          << " Warning :  'Compressed images' are uncompressed" 
          << std::endl
          << "         :  'YBR images' are converted into 'RGB images'"
          << std::endl
          << "         :  'RGB planes' are converted into 'RGB pixels'"
          << std::endl
          << "         :  'Palette Color images' are kept 'as is'"
          << std::endl;
   std::cout << "-----------------------------------------------" << std::endl;

   if( argc < 3 )
    {
       std::cerr << "Usage " << argv[0]  << " Source image.dcm  " 
                 << " Output image.dcm " << std::endl;
       return 1;
    }

   std::string fileName       = argv[1];
   std::string output         = argv[2];

// ============================================================
//   Read the input image.
// ============================================================

   std::cout << argv[1] << std::endl;

   gdcm::File *f = new gdcm::File();
   f->SetLoadMode( 0x00000000);
   f->SetFileName( fileName );
   bool res = f->Load();        

   if (!res) {
       std::cerr << "Sorry, " << fileName <<"  not a gdcm-readable "
                 << "DICOM / ACR File"
                 <<std::endl;
       return 0;
   }
   std::cout << " ... is readable " << std::endl;

// ============================================================
//   Read the Pixels
//
// ============================================================
 
   // Pixel Reading must be done here, to be sure 
   // to load the Palettes Color (if any)

   // First, create a gdcm::FileHelper
   gdcm::FileHelper *fh = new gdcm::FileHelper(f);

   // Load the pixels, DO NOT transform LUT (if any) into RGB Pixels 
   uint8_t *imageDataRaw = fh->GetImageDataRaw();
   // Get the image data size
   size_t dataRawSize    = fh->GetImageDataRawSize();

// ============================================================
//   Create a new gdcm::Filehelper, to hold new image.
// ============================================================

   gdcm::FileHelper *copy = new gdcm::FileHelper( output );

// ============================================================
//   Selective copy of the entries (including Pixel Element).
// ============================================================

   gdcm::DocEntry *d = f->GetFirstEntry();
 
   d = f->GetFirstEntry();
   while(d)
   {
      // We skip SeqEntries, since user cannot do much with them
      if ( !(dynamic_cast<gdcm::SeqEntry*>(d))
      // We skip Shadow Groups, since nobody knows what they mean
           && !( d->GetGroup()%2 ) )
      { 

         if ( gdcm::BinEntry *b = dynamic_cast<gdcm::BinEntry*>(d) )
         {              
            copy->GetFile()->InsertBinEntry( b->GetBinArea(),b->GetLength(),
                                             b->GetGroup(),b->GetElement(),
                                             b->GetVR() );
         }
         else if ( gdcm::ValEntry *v = dynamic_cast<gdcm::ValEntry*>(d) )
         {   
             copy->GetFile()->InsertValEntry( v->GetValue(),
                                              v->GetGroup(),v->GetElement(),
                                              v->GetVR() ); 
         }
         else
         {
          // We skip gdcm::SeqEntries
         }
      }
      d = f->GetNextEntry();
   }
   
   // User wants to keep the Palette Color -if any- 
   // and write the image as it was
   copy->SetImageData(imageDataRaw, dataRawSize);
   copy->SetWriteModeToRaw();
   copy->WriteDcmExplVR( output );

   std::cout << std::endl
             << "------------------------------------------------------------"
             << std::endl;
   delete f;
   delete fh;
   delete copy;

   exit (0);
}