X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=tools%2FclitkWriteDicomSeriesGenericFilter.txx;h=741ca234a99da038edb030fd82214903e062ca95;hb=7a5d28f9a67bbf9f6eec89368010a5a08ca0838d;hp=21612e4eaba3b927a5bc84ba3bc781c5f1b4727f;hpb=0b7c9b1e1215634b02cbd38d4e4ba101d6111ba8;p=clitk.git diff --git a/tools/clitkWriteDicomSeriesGenericFilter.txx b/tools/clitkWriteDicomSeriesGenericFilter.txx index 21612e4..741ca23 100644 --- a/tools/clitkWriteDicomSeriesGenericFilter.txx +++ b/tools/clitkWriteDicomSeriesGenericFilter.txx @@ -1,7 +1,7 @@ /*========================================================================= Program: vv http://www.creatis.insa-lyon.fr/rio/vv - Authors belong to: + Authors belong to: - University of LYON http://www.universite-lyon.fr/ - Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr - CREATIS CNRS laboratory http://www.creatis.insa-lyon.fr @@ -20,191 +20,346 @@ /* ================================================= * @file clitkWriteDicomSeriesGenericFilter.txx - * @author - * @date - * - * @brief - * + * @author + * @date + * + * @brief + * ===================================================*/ +// clitk +#include "clitkResampleImageWithOptionsFilter.h" + namespace clitk { - //----------------------------------------------------------- - // Constructor - //----------------------------------------------------------- - template - WriteDicomSeriesGenericFilter::WriteDicomSeriesGenericFilter() - { - m_Verbose=false; - m_InputFileName=""; +//----------------------------------------------------------- +// Constructor +//----------------------------------------------------------- +template +WriteDicomSeriesGenericFilter::WriteDicomSeriesGenericFilter() +{ + m_Verbose=false; + m_InputFileName=""; +} + + +//----------------------------------------------------------- +// Update +//----------------------------------------------------------- +template +void WriteDicomSeriesGenericFilter::Update() +{ + // Read the Dimension and PixelType + int Dimension; + std::string PixelType; + ReadImageDimensionAndPixelType(m_InputFileName, Dimension, PixelType); + + + // Call UpdateWithDim + if(Dimension==2) UpdateWithDim<2>(PixelType); + else if(Dimension==3) UpdateWithDim<3>(PixelType); + // else if (Dimension==4)UpdateWithDim<4>(PixelType); + else { + std::cout<<"Error, Only for 2 or 3 Dimensions!!!"< +template +void +WriteDicomSeriesGenericFilter::UpdateWithDim(std::string PixelType) +{ + if (m_Verbose) std::cout << "Image was detected to be "< - void WriteDicomSeriesGenericFilter::Update() - { - // Read the Dimension and PixelType - int Dimension; - std::string PixelType; - ReadImageDimensionAndPixelType(m_InputFileName, Dimension, PixelType); + if(PixelType == "short") { + if (m_Verbose) std::cout << "Launching filter in "<< Dimension <<"D and signed short..." << std::endl; + UpdateWithDimAndPixelType(); + } + // else if(PixelType == "unsigned_short"){ + // if (m_Verbose) std::cout << "Launching filter in "<< Dimension <<"D and unsigned_short..." << std::endl; + // UpdateWithDimAndPixelType(); + // } - - // Call UpdateWithDim - if(Dimension==2) UpdateWithDim<2>(PixelType); - else if(Dimension==3) UpdateWithDim<3>(PixelType); - // else if (Dimension==4)UpdateWithDim<4>(PixelType); - else - { - std::cout<<"Error, Only for 2 or 3 Dimensions!!!"<(); } - //------------------------------------------------------------------- - // Update with the number of dimensions - //------------------------------------------------------------------- - template - template - void - WriteDicomSeriesGenericFilter::UpdateWithDim(std::string PixelType) - { - if (m_Verbose) std::cout << "Image was detected to be "<(); + // } + else { + if (m_Verbose) std::cout << "Launching filter in "<< Dimension <<"D and float..." << std::endl; + UpdateWithDimAndPixelType(); + } +} + +//------------------------------------------------------------------- +// Update with the number of dimensions and the pixeltype read from +// the dicom files. The MHD files may be resampled to match the +// dicom spacing (and number of slices). Rounding errors in resampling +// are handled by removing files when generating the output dicom +// series. +//------------------------------------------------------------------- +template +template +void +WriteDicomSeriesGenericFilter::UpdateWithDimAndPixelType() +{ - if(PixelType == "short"){ - if (m_Verbose) std::cout << "Launching filter in "<< Dimension <<"D and signed short..." << std::endl; - UpdateWithDimAndPixelType(); + // ImageTypes + typedef itk::Image InputImageType; + typedef itk::Image OutputImageType; + + // Read the dicom directory + typedef itk::ImageSeriesReader< InputImageType > ReaderType; + typedef itk::GDCMImageIO ImageIOType; + typedef itk::GDCMSeriesFileNames NamesGeneratorType; + + ImageIOType::Pointer gdcmIO = ImageIOType::New(); + NamesGeneratorType::Pointer namesGenerator = NamesGeneratorType::New(); + namesGenerator->SetInputDirectory( m_ArgsInfo.inputDir_arg ); + namesGenerator->SetOutputDirectory( m_ArgsInfo.outputDir_arg ); + typename ReaderType::FileNamesContainer filenames_in = namesGenerator->GetInputFileNames(); + typename ReaderType::FileNamesContainer filenames_out = namesGenerator->GetOutputFileNames(); + + // Output the dicom files + unsigned int numberOfFilenames = filenames_in.size(); + if (m_Verbose) { + std::cout << numberOfFilenames <<" were read in the directory "<(); - // } + } + + // Read the series + typename ReaderType::Pointer reader = ReaderType::New(); + reader->SetImageIO( gdcmIO ); + reader->SetFileNames( filenames_in ); + try { + reader->Update(); + } catch (itk::ExceptionObject &excp) { + std::cerr << "Error: Exception thrown while reading the DICOM series!!" << std::endl; + std::cerr << excp << std::endl; + } + + // Read the input (MHD file) + typedef typename InputImageType::RegionType RegionType; + typedef typename RegionType::SizeType SizeType; + typedef itk::ImageFileReader InputReaderType; + typename InputReaderType::Pointer volumeReader = InputReaderType::New(); + volumeReader->SetFileName( m_InputFileName); + volumeReader->Update(); + + typename InputImageType::Pointer input = volumeReader->GetOutput(); + if (input->GetSpacing() != reader->GetOutput()->GetSpacing()) + { + // resampling is carried out on the fly if resolution between + // the input mhd and input dicom series is different + if (m_Verbose) + std::cout << "Warning: The image spacing differs between the MHD file and the input dicom series. Performing resampling with default options (for advanced options, use clitkResampleImage)." << std::endl; - else if (PixelType == "unsigned_char"){ - if (m_Verbose) std::cout << "Launching filter in "<< Dimension <<"D and unsigned_char..." << std::endl; - UpdateWithDimAndPixelType(); - } + // Filter + typedef clitk::ResampleImageWithOptionsFilter ResampleImageFilterType; + typename ResampleImageFilterType::Pointer filter = ResampleImageFilterType::New(); + filter->SetInput(input); + filter->SetVerboseOptions(m_Verbose); + filter->SetGaussianFilteringEnabled(false); + filter->SetDefaultPixelValue(0); + filter->SetOutputSpacing(reader->GetOutput()->GetSpacing()); + const SizeType& input_size = reader->GetOutput()->GetLargestPossibleRegion().GetSize(); + SizeType output_size; + for (unsigned int i = 0; i < Dimension; i++) + output_size[i] = input_size[i]; + filter->SetOutputSize(output_size); + filter->Update(); + input = filter->GetOutput(); + } + + // In some cases, due to resampling approximation issues, + // the number of slices in the MHD file may be different (smaller) + // from the number of files in the template dicom directory. + // To avoid ITK generating an exception, we reduce the number + // of DCM files to be considered, and a warning is printed + // in verbose mode + const RegionType volumeRegion = input->GetLargestPossibleRegion(); + const SizeType& volumeSize = volumeRegion.GetSize(); + if (Dimension == 3 && volumeSize[2] < numberOfFilenames) + { + if (m_Verbose) + std::cout << "Warning: The number of files in " << m_ArgsInfo.inputDir_arg << " (" << filenames_in.size() << " files) is greater than the number of slices in MHD (" << volumeSize[2] << " slices). Using only " << volumeSize[2] << " files." << std::endl; - // else if (PixelType == "char"){ - // if (m_Verbose) std::cout << "Launching filter in "<< Dimension <<"D and signed_char..." << std::endl; - // UpdateWithDimAndPixelType(); - // } - else { - if (m_Verbose) std::cout << "Launching filter in "<< Dimension <<"D and float..." << std::endl; - UpdateWithDimAndPixelType(); - } + filenames_in.resize(volumeSize[2]); + filenames_out.resize(filenames_in.size()); + numberOfFilenames = filenames_in.size(); } + // Modify the meta dictionary + typedef itk::MetaDataDictionary DictionaryType; + const std::vector* dictionary = reader->GetMetaDataDictionaryArray(); + + // Get keys + unsigned int numberOfKeysGiven=0; + if(m_ArgsInfo.midP_flag && m_ArgsInfo.key_given) + std::cerr<<"Error: both keys and midP option are given"< - template - void - WriteDicomSeriesGenericFilter::UpdateWithDimAndPixelType() + for (unsigned int i = 0; i < numberOfKeysGiven; i++) { + std::string entryId(m_ArgsInfo.key_arg[i] ); + std::string value( m_ArgsInfo.tag_arg[i] ); + for(unsigned int fni = 0; fni( *((*dictionary)[fni]), entryId, value ); + } + + // Output directory and filenames + itksys::SystemTools::MakeDirectory( m_ArgsInfo.outputDir_arg ); // create if it doesn't exist + typedef itk::ImageSeriesWriter SeriesWriterType; + typename SeriesWriterType::Pointer seriesWriter = SeriesWriterType::New(); + + seriesWriter->SetInput( input ); + seriesWriter->SetImageIO( gdcmIO ); + + seriesWriter->SetFileNames( filenames_out ); + seriesWriter->SetMetaDataDictionaryArray( dictionary ); + + // Write + try { + seriesWriter->Update(); + } catch( itk::ExceptionObject & excp ) { + std::cerr << "Error: Exception thrown while writing the series!!" << std::endl; + std::cerr << excp << std::endl; + } + +} + +/* +//------------------------------------------------------------------- +// Update with the number of dimensions and the pixeltype +//------------------------------------------------------------------- +template +template +void +WriteDicomSeriesGenericFilter::UpdateWithDimAndPixelType() +{ + + // ImageTypes + typedef itk::Image InputImageType; + typedef itk::Image OutputImageType; + + // Read the input (volumetric) + typedef itk::ImageFileReader InputReaderType; + typename InputReaderType::Pointer volumeReader = InputReaderType::New(); + volumeReader->SetFileName( m_InputFileName); + volumeReader->Update(); + typename InputImageType::Pointer input= volumeReader->GetOutput(); + + // Read the dicom directory + typedef itk::ImageSeriesReader< InputImageType > ReaderType; + typedef itk::GDCMImageIO ImageIOType; + typedef itk::GDCMSeriesFileNames NamesGeneratorType; + + ImageIOType::Pointer gdcmIO = ImageIOType::New(); + NamesGeneratorType::Pointer namesGenerator = NamesGeneratorType::New(); + namesGenerator->SetInputDirectory( m_ArgsInfo.inputDir_arg ); + namesGenerator->SetOutputDirectory( m_ArgsInfo.outputDir_arg ); + typename ReaderType::FileNamesContainer filenames_in = namesGenerator->GetInputFileNames(); + typename ReaderType::FileNamesContainer filenames_out = namesGenerator->GetOutputFileNames(); + + // Output the dicom files + unsigned int numberOfFilenames = filenames_in.size(); + if (m_Verbose) { + std::cout << numberOfFilenames <<" were read in the directory "<SetImageIO( gdcmIO ); + reader->SetFileNames( filenames_in ); + try { + reader->Update(); + } catch (itk::ExceptionObject &excp) { + std::cerr << "Error: Exception thrown while reading the DICOM series!!" << std::endl; + std::cerr << excp << std::endl; + } - // ImageTypes - typedef itk::Image InputImageType; - typedef itk::Image OutputImageType; - - // Read the input (volumetric) - typedef itk::ImageFileReader InputReaderType; - typename InputReaderType::Pointer volumeReader = InputReaderType::New(); - volumeReader->SetFileName( m_InputFileName); - volumeReader->Update(); - typename InputImageType::Pointer input= volumeReader->GetOutput(); - - // Read the dicom directory - typedef itk::ImageSeriesReader< InputImageType > ReaderType; - typedef itk::GDCMImageIO ImageIOType; - typedef itk::GDCMSeriesFileNames NamesGeneratorType; - - ImageIOType::Pointer gdcmIO = ImageIOType::New(); - NamesGeneratorType::Pointer namesGenerator = NamesGeneratorType::New(); - namesGenerator->SetInputDirectory( m_ArgsInfo.inputDir_arg ); - typename ReaderType::FileNamesContainer filenames = namesGenerator->GetInputFileNames(); - - // Output the dicom files - unsigned int numberOfFilenames = filenames.size(); - if (m_Verbose) - { - std::cout << numberOfFilenames <<" were read in the directory "<* dictionary = reader->GetMetaDataDictionaryArray(); - - // Get keys - unsigned int numberOfKeysGiven=0; - if(m_ArgsInfo.midP_flag && m_ArgsInfo.key_given) - std::cerr<<"Error: both keys and midP option are given"<( *((*dictionary)[fni]), entryId, value ); - } - - // Output directory and filenames - itksys::SystemTools::MakeDirectory( m_ArgsInfo.outputDir_arg ); // create if it doesn't exist - typedef itk::ImageSeriesWriter SeriesWriterType; - typename SeriesWriterType::Pointer seriesWriter = SeriesWriterType::New(); - - seriesWriter->SetInput( volumeReader->GetOutput() ); - seriesWriter->SetImageIO( gdcmIO ); - namesGenerator->SetOutputDirectory( m_ArgsInfo.outputDir_arg ); - seriesWriter->SetFileNames( namesGenerator->GetOutputFileNames() ); - //seriesWriter->SetMetaDataDictionaryArray( dictionary ); - seriesWriter->SetMetaDataDictionaryArray( dictionary ); - - // Write - try - { - seriesWriter->Update(); - } - catch( itk::ExceptionObject & excp ) - { - std::cerr << "Error: Exception thrown while writing the series!!" << std::endl; - std::cerr << excp << std::endl; - } + // Modify the meta dictionary + typedef itk::MetaDataDictionary DictionaryType; + const std::vector* dictionary = reader->GetMetaDataDictionaryArray(); + + // Get keys + unsigned int numberOfKeysGiven=0; + if(m_ArgsInfo.midP_flag && m_ArgsInfo.key_given) + std::cerr<<"Error: both keys and midP option are given"<( *((*dictionary)[fni]), entryId, value ); } + // Output directory and filenames + itksys::SystemTools::MakeDirectory( m_ArgsInfo.outputDir_arg ); // create if it doesn't exist + typedef itk::ImageSeriesWriter SeriesWriterType; + typename SeriesWriterType::Pointer seriesWriter = SeriesWriterType::New(); + + seriesWriter->SetInput( volumeReader->GetOutput() ); + seriesWriter->SetImageIO( gdcmIO ); + + seriesWriter->SetFileNames( filenames_out ); + seriesWriter->SetMetaDataDictionaryArray( dictionary ); + + // Write + try { + seriesWriter->Update(); + } catch( itk::ExceptionObject & excp ) { + std::cerr << "Error: Exception thrown while writing the series!!" << std::endl; + std::cerr << excp << std::endl; + } + +} +*/ }//end clitk - + #endif //#define clitkWriteDicomSeriesGenericFilter_txx