tools/clitkImage2DicomDoseGenericFilter.txx

   1 /*=========================================================================
   2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
   3
   4   Authors belong to:
   5   - University of LYON              http://www.universite-lyon.fr/
   6   - Léon Bérard cancer center       http://www.centreleonberard.fr
   7   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
   8
   9   This software is distributed WITHOUT ANY WARRANTY; without even
  10   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  11   PURPOSE.  See the copyright notices for more information.
  12
  13   It is distributed under dual licence
  14
  15   - BSD        See included LICENSE.txt file
  16   - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  17 ===========================================================================**/
  18 #ifndef clitkImage2DicomDoseGenericFilter_txx
  19 #define clitkImage2DicomDoseGenericFilter_txx
  20
  21 /* =================================================
  22  * @file   clitkImage2DicomDosemGenericFilter.txx
  23  * @author
  24  * @date
  25  *
  26  * @brief
  27  *
  28  ===================================================*/
  29
  30 #include "math.h"
  31
  32 #include "clitkImage2DicomDoseGenericFilter.h"
  33 #include "clitkCommon.h"
  34 #include "itkMinimumMaximumImageCalculator.h"
  35 #include "itkImageRegionIterator.h"
  36 #include "itkImageRegionConstIterator.h"
  37 #if GDCM_MAJOR_VERSION >= 2
  38 #include "gdcmUIDGenerator.h"
  39 #include <gdcmImageHelper.h>
  40 #include <gdcmAttribute.h>
  41 #include <gdcmReader.h>
  42 #include <gdcmWriter.h>
  43 #else
  44 #include "gdcmFile.h"
  45 #include "gdcmUtil.h"
  46 #endif
  47
  48 //#include "gdcmBase.h"
  49 //#include "gdcmDocEntry.h"
  50
  51
  52
  53 namespace clitk
  54 {
  55
  56
  57 //-----------------------------------------------------------
  58 // Constructor
  59 //-----------------------------------------------------------
  60 template<class args_info_type>
  61 Image2DicomDoseGenericFilter<args_info_type>::Image2DicomDoseGenericFilter()
  62 {
  63   m_Verbose=false;
  64   m_InputFileName="";
  65 }
  66
  67
  68 //-----------------------------------------------------------
  69 // Update
  70 //-----------------------------------------------------------
  71 template<class args_info_type>
  72 void Image2DicomDoseGenericFilter<args_info_type>::Update()
  73 {
  74   // Read the Dimension and PixelType
  75   int Dimension;
  76   std::string PixelType;
  77   ReadImageDimensionAndPixelType(m_InputFileName, Dimension, PixelType);
  78
  79
  80   // Call UpdateWithDim
  81   if(Dimension==2) UpdateWithDim<2>(PixelType);
  82   else if(Dimension==3) UpdateWithDim<3>(PixelType);
  83   // else if (Dimension==4)UpdateWithDim<4>(PixelType);
  84   else {
  85     std::cout<<"Error, Only for 2 or 3  Dimensions!!!"<<std::endl ;
  86     return;
  87   }
  88 }
  89
  90 //-------------------------------------------------------------------
  91 // Update with the number of dimensions
  92 //-------------------------------------------------------------------
  93 template<class args_info_type>
  94 template<unsigned int Dimension>
  95 void
  96 Image2DicomDoseGenericFilter<args_info_type>::UpdateWithDim(std::string PixelType)
  97 {
  98   if (m_Verbose) std::cout << "Image was detected to be "<<Dimension<<"D and "<< PixelType<<"..."<<std::endl;
  99
 100   if(PixelType == "short") {
 101     if (m_Verbose) std::cout << "Launching filter in "<< Dimension <<"D and signed short..." << std::endl;
 102     UpdateWithDimAndPixelType<Dimension, signed short>();
 103   }
 104   else if(PixelType == "unsigned_short"){
 105     if (m_Verbose) std::cout  << "Launching filter in "<< Dimension <<"D and unsigned_short..." << std::endl;
 106     UpdateWithDimAndPixelType<Dimension, unsigned short>();
 107   }
 108
 109   else if (PixelType == "unsigned_char") {
 110     if (m_Verbose) std::cout  << "Launching filter in "<< Dimension <<"D and unsigned_char..." << std::endl;
 111     UpdateWithDimAndPixelType<Dimension, unsigned char>();
 112   }
 113
 114   //     else if (PixelType == "char"){
 115   //       if (m_Verbose) std::cout  << "Launching filter in "<< Dimension <<"D and signed_char..." << std::endl;
 116   //       UpdateWithDimAndPixelType<Dimension, signed char>();
 117   //     }
 118   else if (PixelType == "double") {
 119     if (m_Verbose) std::cout  << "Launching filter in "<< Dimension <<"D and double..." << std::endl;
 120     UpdateWithDimAndPixelType<Dimension, double>();
 121   }
 122   else {
 123     if (m_Verbose) std::cout  << "Launching filter in "<< Dimension <<"D and float..." << std::endl;
 124     UpdateWithDimAndPixelType<Dimension, float>();
 125   }
 126 }
 127
 128 //-------------------------------------------------------------------
 129 // Update with the number of dimensions and the pixeltype read from
 130 // the dicom files. The MHD files may be resampled to match the
 131 // dicom spacing (and number of slices). Rounding errors in resampling
 132 // are handled by removing files when generating the output dicom
 133 // series.
 134 //-------------------------------------------------------------------
 135 template<class args_info_type>
 136 template <unsigned int Dimension, class  PixelType>
 137 void
 138 Image2DicomDoseGenericFilter<args_info_type>::UpdateWithDimAndPixelType()
 139 {
 140
 141 #if GDCM_MAJOR_VERSION == 2
 142   // ImageTypes
 143   typedef itk::Image<PixelType, Dimension> InputImageType;
 144   typedef unsigned short int OutputPixelType;
 145   typedef itk::Image<OutputPixelType, Dimension> OutputImageType;
 146   typedef itk::ImageFileReader< InputImageType >     ReaderType;
 147   typedef itk::ImageSeriesReader< InputImageType >     ReaderSeriesType;
 148   typedef itk::ImageSeriesWriter<OutputImageType, OutputImageType>  WriterSerieType;
 149   typedef itk::ImageRegionIterator< InputImageType > IteratorType;
 150   typedef itk::MinimumMaximumImageCalculator <InputImageType> ImageCalculatorFilterType;
 151   typedef itk::GDCMImageIO ImageIOType;
 152
 153   //-----------------------------------------------------------------------------
 154   // opening image input file
 155   typename ReaderType::Pointer reader = ReaderType::New();
 156   const char * filename = m_ArgsInfo.input_arg;
 157   reader->SetFileName( filename );
 158   reader->Update();
 159   typename InputImageType::Pointer image = reader->GetOutput();
 160   std::ostringstream value;
 161
 162   // Read Dicom model file
 163   typename ReaderSeriesType::Pointer readerSeries = ReaderSeriesType::New();
 164   ImageIOType::Pointer gdcmIO = ImageIOType::New();
 165   std::string filename_out = m_ArgsInfo.output_arg;
 166   gdcmIO->LoadPrivateTagsOn();
 167   gdcmIO->KeepOriginalUIDOn();
 168   typename ReaderSeriesType::FileNamesContainer fileNames;
 169   fileNames.push_back(m_ArgsInfo.DicomInputFile_arg);
 170   readerSeries->SetImageIO( gdcmIO );
 171   readerSeries->SetFileNames( fileNames );
 172   try {
 173     readerSeries->Update();
 174   } catch (itk::ExceptionObject &excp) {
 175     std::cerr << "Error: Exception thrown while reading the DICOM model file !!" << std::endl;
 176     std::cerr << excp << std::endl;
 177   }
 178
 179   // update output dicom keys/tags
 180   typename ReaderSeriesType::DictionaryRawPointer inputDict = (*(readerSeries->GetMetaDataDictionaryArray()))[0];
 181   typename ReaderSeriesType::DictionaryArrayType outputArray;
 182   typename ReaderSeriesType::DictionaryRawPointer outputDict = new typename ReaderSeriesType::DictionaryType;
 183   CopyDictionary (*inputDict, *outputDict);
 184
 185   // origin
 186   typename InputImageType::PointType origin = image->GetOrigin();
 187   value.str("");
 188   value<<origin[0]<<'\\'<<origin[1]<<'\\'<<origin[2];
 189   itk::EncapsulateMetaData<std::string>(*outputDict, "0020|0032", value.str());
 190   DD(origin);
 191
 192   // size
 193   typename InputImageType::SizeType imageSize = image->GetLargestPossibleRegion().GetSize();
 194   //DD(imageSize);
 195   int NbCols=imageSize[0];      // col
 196   int NbRows=imageSize[1];      // row
 197   int NbFrames=imageSize[2];    // frame
 198   itk::EncapsulateMetaData<std::string>(*outputDict, "0028|0008", NumberToString(NbFrames));
 199   itk::EncapsulateMetaData<std::string>(*outputDict, "0028|0010", NumberToString(NbRows));
 200   itk::EncapsulateMetaData<std::string>(*outputDict, "0028|0011", NumberToString(NbCols));
 201   DD(NbCols);
 202   DD(NbRows);
 203   DD(NbFrames);
 204
 205   // spacing
 206   typename InputImageType::SpacingType Spacing = image->GetSpacing();
 207   value.str("");
 208   value<<Spacing[0]<<'\\'<<Spacing[1];
 209   itk::EncapsulateMetaData<std::string>(*outputDict, "0028|0030", value.str());
 210   value.str("");
 211   value<<Spacing[2];
 212   itk::EncapsulateMetaData<std::string>(*outputDict, "0018|0050", value.str());
 213   DD(Spacing);
 214
 215   // offset
 216   float offset = 0.;
 217   value.str("");
 218   value << offset;
 219   for (int i=1; i<NbFrames ; i++){
 220     offset+=Spacing[2];
 221     value << '\\';
 222     value << offset;
 223   }
 224   itk::EncapsulateMetaData<std::string>(*outputDict, "3004|000c", value.str());
 225   DD(value.str());
 226
 227   // scaling
 228   typename ImageCalculatorFilterType::Pointer imageCalculatorFilter = ImageCalculatorFilterType::New();
 229   imageCalculatorFilter->SetImage(image);
 230   imageCalculatorFilter->ComputeMaximum();
 231   float highestValue=imageCalculatorFilter->GetMaximum();
 232   double doseScaling = highestValue/(pow(2,16)-1);
 233   value.str("");
 234   value<<doseScaling;
 235   itk::EncapsulateMetaData<std::string>(*outputDict, "3004|000e", value.str());
 236   DD(value.str());
 237
 238   // image data
 239   typename OutputImageType::Pointer imageOutput = OutputImageType::New();
 240   imageOutput->SetRegions(image->GetLargestPossibleRegion());
 241   imageOutput->SetSpacing(image->GetSpacing());
 242   imageOutput->SetOrigin(image->GetOrigin());
 243   imageOutput->SetDirection(image->GetDirection());
 244   imageOutput->Allocate();
 245
 246   typename itk::ImageRegionConstIterator<InputImageType> imageIterator(image,image->GetLargestPossibleRegion());
 247   typename itk::ImageRegionIterator<OutputImageType> imageOutputIterator(imageOutput,imageOutput->GetLargestPossibleRegion());
 248   while(!imageIterator.IsAtEnd()) {
 249     // Set the current pixel to white
 250     imageOutputIterator.Set((signed short int)(imageIterator.Get()/doseScaling));
 251
 252     ++imageIterator;
 253     ++imageOutputIterator;
 254   }
 255
 256   // Output directory and filenames
 257   typename WriterSerieType::Pointer writerSerie = WriterSerieType::New();
 258   outputArray.push_back(outputDict);
 259   writerSerie->SetInput( imageOutput );
 260   writerSerie->SetImageIO( gdcmIO );
 261   typename ReaderSeriesType::FileNamesContainer fileNamesOutput;
 262   fileNamesOutput.push_back(filename_out);
 263   writerSerie->SetFileNames( fileNamesOutput );
 264   writerSerie->SetMetaDataDictionaryArray(&outputArray);
 265
 266   // Write
 267   try {
 268     if (m_ArgsInfo.verbose_flag)
 269       std::cout << writerSerie << std::endl;
 270     writerSerie->Update();
 271   } catch( itk::ExceptionObject & excp ) {
 272     std::cerr << "Error: Exception thrown while writing the series!!" << std::endl;
 273     std::cerr << excp << std::endl;
 274   }
 275
 276   //Read sequence dicom tag with gdcm
 277   gdcm::Reader readerTemplateGDCM;
 278   readerTemplateGDCM.SetFileName( fileNames[0].c_str() );
 279   readerTemplateGDCM.Read();
 280   gdcm::File &fileTemplate = readerTemplateGDCM.GetFile();
 281   gdcm::DataSet &dsTemplate = fileTemplate.GetDataSet();
 282   const unsigned int ptr_len = 42;
 283   char *ptrTemplate = new char[ptr_len];
 284   memset(ptrTemplate,0,ptr_len);
 285
 286   const gdcm::DataElement &referenceRTPlanSq = dsTemplate.GetDataElement(gdcm::Tag(0x300c, 0x02));
 287
 288   //Copy/Write sequence dicom tag with gdcm
 289   gdcm::Reader readerOutputGDCM;
 290   readerOutputGDCM.SetFileName( fileNamesOutput[0].c_str() );
 291   readerOutputGDCM.Read();
 292   gdcm::File &file = readerOutputGDCM.GetFile();
 293   gdcm::DataSet &dsOutput = file.GetDataSet();
 294
 295   dsOutput.Insert(referenceRTPlanSq);
 296   gdcm::Writer w;
 297   w.SetFile( file );
 298   w.SetFileName( fileNamesOutput[0].c_str() );
 299   w.Write();
 300
 301 //---------------------------------------------------------------------------------------
 302 //WRITE DICOM BIS
 303 // The previous way of writting DICOM-RT-DOSE works only for ITK
 304 // and resulting RT-DOSE files are not readable by commercial systems.
 305 // The nex step is to copy again the output file with another syntax, allowing to make RT-DOSE files readable by commercial systems.
 306 // see Jean-Pierre Roux comment below.
 307
 308 /*gdcm::FileHelper *fh = new gdcm::FileHelper(args_info.OutputFile_arg);
 309    void *imageData;
 310    int dataSize;
 311
 312    dataSize  = fh->GetImageDataRawSize();
 313    imageData = fh->GetImageDataRaw();// somewhat important : Loads the Pixels in memory !
 314
 315    fh->SetWriteModeToRaw();
 316    fh->SetWriteTypeToDcmExplVR();
 317
 318    bool res = fh->Write(args_info.OutputFile_arg);
 319
 320    if(!res)
 321       std::cout <<"Fail to write [" << args_info.OutputFile_arg << "]" <<std::endl;
 322    else std::cout<<"\n DICOM File re-written, using the FileHelper syntax, in order to be processed by commercial systems !"<<std::endl;
 323
 324 delete fh; */
 325 /*  gdcm::Writer w;
 326   w.SetFile(mDCMFile);
 327   w.SetFileName(m_ArgsInfo.output_arg);
 328   w.Write();*/
 329 //   fh->Delete();
 330
 331 //---------------------------------------------------------------------------------------
 332 //---------------------------------------------------------------------------------------
 333 // Jean-Pierre Roux help for DICOM-RT-DOSE writting
 334 //---------------------------------------------------------------------------------------
 335 //---------------------------------------------------------------------------------------
 336
 337 //      de      Jean-Pierre Roux <jpr@creatis.insa-lyon.fr>
 338 // répondre à jpr@creatis.insa-lyon.fr
 339 // à   Loic Grevillot <loic.grevillot@gmail.com>
 340 // cc   jpr@creatis.insa-lyon.fr,
 341 // Joël Schaerer <joel.schaerer@gmail.com>,
 342 // David Sarrut <David.Sarrut@creatis.insa-lyon.fr>,
 343 // Joel Schaerer <joel.schaerer@creatis.insa-lyon.fr>
 344 // date 12 juillet 2010 12:23
 345 // objet        Re: DICOM RT DOSE
 346 //
 347 // masquer les détails 12:23 (Il y a 21 heures)
 348 //
 349 // Bonjour,
 350 //
 351 // J'aurais écrit à peut prèt la même chose.
 352 // (Ci après un extrait de mon code -Example/ReWrite.cxx-)
 353 //
 354 // L'utilisation d'un FileHelper (qui ne changera rien dans ce cas précis) est une mesure de précaution, car,  l'élément 7FE0|0010 peut être compressé (ce qui n'est pas le cas pour tes images), que la manière de stocker les pixels ainsi compressés était parfois un peu ... curieuse.
 355 // Je décompresse, et réécrit non compressé.
 356 //
 357 // ============================
 358 //    gdcm::File *f = new gdcm::File();
 359 //    f->SetMaxSizeLoadEntry(0x7fffffff);
 360 //    f->SetFileName( fileName );
 361 //    bool res = f->Load();
 362 //    if ( !res )
 363 //    {
 364 //       delete f;
 365 //       return 0;
 366 //    }
 367 //
 368 //    if (!f->IsReadable())
 369 //    {
 370 //        std::cerr << "Sorry, not a Readable DICOM / ACR File"  <<std::endl;
 371 //        delete f;
 372 //        return 0;
 373 //    }
 374 //
 375 //    gdcm::FileHelper *fh = new gdcm::FileHelper(f);
 376 //    void *imageData;
 377 //    int dataSize;
 378 //
 379 //    dataSize  = fh->GetImageDataRawSize();
 380 //    imageData = fh->GetImageDataRaw();// somewhat important : Loads the Pixels in memory !
 381 //
 382 //    fh->SetWriteModeToRaw();
 383 //    fh->SetWriteTypeToDcmExplVR();
 384 //    res = fh->Write(outputFileName);
 385 //
 386 //    if(!res)
 387 //       std::cout <<"Fail to write [" << outputFileName << "]" <<std::endl;
 388 //
 389 //    f->Delete();
 390 //    fh->Delete();
 391
 392 //---------------------------------------------------------------------------------------
 393 //---------------------------------------------------------------------------------------
 394 // pour supprimer des tags:
 395 //---------------------------------------------------------------------------------------
 396 //---------------------------------------------------------------------------------------
 397 //      SOLUCE 1 QUI MARCHE POUR UN SQItem SIMPLE
 398 /*
 399 gdcm::DocEntry *a;
 400 //a= ((gdcm::SQItem*)mDCMFile)->GetDocEntry(0xfffe,0xe00d);
 401 a= ((gdcm::SQItem*)mDCMFile)->GetDocEntry(0x3004,0x000e);
 402 ((gdcm::SQItem*)mDCMFile)->RemoveEntry(a);
 403 mDCMFile->Write (args_info.OutputFile_arg, type);
 404 */
 405
 406 //      SOLUCE 2 QUI MARCHE POUR UNE SeqEntry->SQItem
 407 /*
 408 std::cout<<"\ntest correction fichier apres ecriture\n"<<std::endl;
 409 gdcm::SeqEntry *seqEntry = mDCMFile->GetSeqEntry(0x300c,0x0002);
 410 gdcm::SQItem* currentItem = seqEntry->GetFirstSQItem();
 411 gdcm::DocEntry *a;
 412 //a= currentItem->GetDocEntry(0x0008,0x1155);
 413 a= currentItem->GetDocEntry(0xfffe,0xe00d);
 414 currentItem->RemoveEntry(a);
 415 mDCMFile->Write (args_info.OutputFile_arg, type);
 416 */
 417
 418 //gdcm::DocEntry *a;
 419 //a=GetDocEntry(0x7fe0,0x0000);
 420 //((gdcm::SQItem*)mDCMFile)->RemoveEntry(a);
 421
 422 //-----------------------------------------------------------------------------------
 423   std::cout <<"\n## DICOM Image to RT DOSE application is ended..."<<std::endl;
 424   std::cout <<"#########################################################\n" << std::endl;
 425
 426 #else
 427   std::cout << "Use GDCM2" << std::endl;
 428 #endif
 429 }
 430 //---------------------------------------------------------------------------
 431
 432
 433 //---------------------------------------------------------------------------
 434 void CopyDictionary (itk::MetaDataDictionary &fromDict, itk::MetaDataDictionary &toDict)
 435 {
 436   typedef itk::MetaDataDictionary DictionaryType;
 437
 438   DictionaryType::ConstIterator itr = fromDict.Begin();
 439   DictionaryType::ConstIterator end = fromDict.End();
 440   typedef itk::MetaDataObject< std::string > MetaDataStringType;
 441
 442   while( itr != end )
 443     {
 444     itk::MetaDataObjectBase::Pointer  entry = itr->second;
 445
 446     MetaDataStringType::Pointer entryvalue =
 447       dynamic_cast<MetaDataStringType *>( entry.GetPointer() ) ;
 448     if( entryvalue )
 449       {
 450       std::string tagkey   = itr->first;
 451       std::string tagvalue = entryvalue->GetMetaDataObjectValue();
 452       itk::EncapsulateMetaData<std::string>(toDict, tagkey, tagvalue);
 453       }
 454     ++itr;
 455     }
 456 }
 457 //---------------------------------------------------------------------------
 458
 459 //---------------------------------------------------------------------------
 460 template <typename T> std::string NumberToString ( T Number )
 461 {
 462    std::ostringstream ss;
 463    ss << Number;
 464    return ss.str();
 465 }
 466 //---------------------------------------------------------------------------
 467
 468
 469 }//end clitk
 470
 471 #endif //#define clitkImage2DicomDoseGenericFilter_txx