/*=========================================================================
                                                                                
  Program:   gdcm
  Module:    $RCSfile: AnonymizeNoLoad.cxx,v $
  Language:  C++
  Date:      $Date: 2005/07/20 14:48:15 $
  Version:   $Revision: 1.7 $
                                                                                
  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 "gdcmCommon.h"
#include "gdcmDebug.h"
#include "gdcmDirList.h"

#include "gdcmArgMgr.h"

#include <iostream>

int main(int argc, char *argv[])
{
   START_USAGE(usage)
   "\n AnonymizeNoLoad :\n",
   "Anonymize a gdcm-readable Dicom image even if pixels aren't gdcm readable ",
   "         Warning : Warning : the image is overwritten                     ",
   "                   to preserve image integrity, use a copy.               ",
   "usage: AnonymizeNoLoad {filein=inputFileName|dirin=inputDirectoryName}    ",
   "                       [rubout=listOfPrivateElementsToRubOut]             ", 
   "                       [ { [noshadowseq] | [noshadow][noseq] } ] [debug]  ",
   "       inputFileName : Name of the (single) file user wants to anonymize  ",
   "       inputDirectoryName : user wants to anonymize *all* the files       ",
   "                            within the (single Patient!) directory        ",
   "       listOfPrivateElementsToRubOut : group,elem (in hexa) of private    ",
   "                           Elements to rub out                            ",
   "       noshadowseq: user doesn't want to load Private Sequences           ",
   "       noshadow   : user doesn't want to load Private groups (odd number) ",
   "       noseq      : user doesn't want to load Sequences                   ",
   "       debug      : user wants to run the program in 'debug mode'         ",
   FINISH_USAGE

   // ----- Initialize Arguments Manager ------
  
   gdcm::ArgMgr *am = new gdcm::ArgMgr(argc, argv);
  
   if (am->ArgMgrDefined("usage")) 
   {
      am->ArgMgrUsage(usage); // Display 'usage'
      delete am;
      return 0;
   }

   if (am->ArgMgrDefined("debug"))
      gdcm::Debug::DebugOn();

   char *fileName = am->ArgMgrGetString("filein",(char *)0);
   char *dirName  = am->ArgMgrGetString("dirin",(char *)0);

   if ( (fileName == 0 && dirName == 0)
        ||
      (fileName != 0 && dirName != 0) )
   {
       std::cout <<std::endl
                 << "Either 'filein' or 'dirin' must be present;" << std::endl
                 << "Not both" << std::endl;
       delete am;
       return 0;
 }
 
   int loadMode = 0x00000000;
   if ( am->ArgMgrDefined("noshadowseq") )
      loadMode |= NO_SHADOWSEQ;
   else 
   {
   if ( am->ArgMgrDefined("noshadow") )
         loadMode |= NO_SHADOW;
      if ( am->ArgMgrDefined("noseq") )
         loadMode |= NO_SEQ;
   }

   int rubOutNb;
   uint16_t *elemsToRubOut = am->ArgMgrGetXInt16Enum("rubout", &rubOutNb);
 
   std::cout << " ---------------------------- rubOutNb " << rubOutNb
             << std::endl;
   delete am;  // ------ we don't need Arguments Manager any longer ------


   if ( fileName != 0 ) // ====== Deal with a single file ======
   {

   // 
   //   Parse the input file.
   // 
      gdcm::File *f;
      f = new gdcm::File( );
      f->SetLoadMode(loadMode);
      f->SetFileName( fileName );
      bool res = f->Load();

      // gdcm::File::IsReadable() is no usable here, because we deal with
      // any kind of gdcm::Readable *document*
      // not only gdcm::File (as opposed to gdcm::DicomDir)
      if ( !res ) 
      {
          std::cout <<std::endl
              << "Sorry, " << fileName <<"  not a gdcm-readable "
              << "DICOM / ACR Document"
              << std::endl;
           delete f;
           return 1;
      }
      std::cout << fileName << " is readable " << std::endl;

      // 
      //      No need to load the pixels in memory.
      //      File will be overwritten
      // 


      // 
      //  Choose the fields to anonymize.
      // 

      // Institution name 
      f->AddAnonymizeElement( 0x0008, 0x0080, "Xanadoo" ); 
      // Patient's name 
      f->AddAnonymizeElement( 0x0010, 0x0010, "g^Fantomas" );      
      // Patient's ID
      f->AddAnonymizeElement( 0x0010, 0x0020,"1515" );
      // Patient's Birthdate
      f->AddAnonymizeElement( 0x0010, 0x0030,"11.11.1111" );
      // Patient's Adress
      f->AddAnonymizeElement( 0x0010, 0x1040,"Sing-sing" );
      // Patient's Mother's Birth Name
      f->AddAnonymizeElement( 0x0010, 0x1060,"g^Vampirella" );      
      // Study Instance UID
      f->AddAnonymizeElement( 0x0020, 0x000d, "9.99.999.9999" );
      // Telephone
      f->AddAnonymizeElement(0x0010, 0x2154, "3615" );

      for (int ri=0; ri<rubOutNb; ri++)
      {
         printf("%04x,%04x\n",elemsToRubOut[2*ri], elemsToRubOut[2*ri+1]);
         f->AddAnonymizeElement((uint32_t)elemsToRubOut[2*ri], 
                                (uint32_t)elemsToRubOut[2*ri+1],"*" ); 
      }

      // Aware use will add new fields here

      // 
      //      Overwrite the file
      // 

      std::cout <<"Let's AnonymizeNoLoad " << std::endl;

      // The gdcm::File remains untouched in memory

      f->AnonymizeNoLoad();

      // No need to write the File : modif were done on disc !
      // File was overwritten ...

      std::cout <<"End AnonymizeNoLoad" << std::endl;

      // 
      //      Remove the Anonymize list
      //   
      f->ClearAnonymizeList();
 
      delete f;
      return 0;

   }
   else  // ====== Deal with a (single Patient) Directory ======
   {
      std::cout << "dirName [" << dirName << "]" << std::endl;
      gdcm::DirList dirList(dirName,1); // gets recursively the file list
      gdcm::DirListType fileList = dirList.GetFilenames();
      for( gdcm::DirListType::iterator it  = fileList.begin();
                                 it != fileList.end();
                                 ++it )
      {

         gdcm::File *f;
         f = new gdcm::File( );
         f->SetLoadMode(loadMode);
         f->SetFileName( it->c_str() );
         bool res = f->Load();

         if ( !res )
         {
            delete f; 
            continue;
         }
         // 
         //  Choose the fields to anonymize.
         // 
 
         // Institution name 
         f->AddAnonymizeElement( 0x0008, 0x0080, "Xanadoo" ); 
         // Patient's name 
         f->AddAnonymizeElement( 0x0010, 0x0010, "g^Fantomas" );   
         // Patient's ID
         f->AddAnonymizeElement( 0x0010, 0x0020,"1515" );
         // Patient's Birthdate
         f->AddAnonymizeElement( 0x0010, 0x0030,"11.11.1111" );
         // Patient's Adress
         f->AddAnonymizeElement( 0x0010, 0x1040,"Sing-sing" );
         // Patient's Mother's Birth Name
         f->AddAnonymizeElement( 0x0010, 0x1060,"g^Vampirella" );   
         // Study Instance UID
         // we may not brutaly overwrite it
         //f->AddAnonymizeElement( 0x0020, 0x000d, "9.99.999.9999" );
         // Telephone
         f->AddAnonymizeElement(0x0010, 0x2154, "3615" );

         for (int ri=0; ri<rubOutNb; ri++)
         {
            f->AddAnonymizeElement((uint32_t)elemsToRubOut[2*ri], 
                                   (uint32_t)elemsToRubOut[2*ri+1],"*" ); 
         }        
         std::cout <<"Let's AnonymizeNoLoad " << it->c_str() << std::endl;

         // The gdcm::File remains untouched in memory

         f->AnonymizeNoLoad();

         // 
         //   Remove the Anonymize list
         //

         f->ClearAnonymizeList();
    
         delete f;         
        }

     }
   return 0;
}