testDicomReader.cpp

Go to the documentation of this file.

 
#include <iostream>
#include "DGtal/images/ImageContainerBySTLVector.h"
#include "DGtal/images/ImageContainerByITKImage.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/io/readers/DicomReader.h"
 
#include "ConfigTest.h"
 
 
using namespace std;
using namespace DGtal;
 
// Functions for testing class DicomReader.
 
template <typename Image3D>
bool testDicomReader()
{
  const std::string filename = testPath + "samples/dicomSample/1629.dcm";
  Image3D image = DicomReader< Image3D >::importDicom( filename );
 
  trace.info() << image <<endl;
 
  unsigned int nbVal=0, nbPos = 0;
  typename Image3D::ConstRange r = image.constRange();
  for ( typename Image3D::ConstRange::ConstIterator it=r.begin(), itend=r.end() ; it != itend ; ++it )
  {
    nbVal++;
    if ( (*it) > 0 ) nbPos++;
  }
 
  trace.info() << "Number of points with (val>0) = " << nbVal << endl;
 
  return nbVal==2130048 && nbPos==296030;
}
 
static
std::vector<std::string>
getFirstDicomSerieFileNames(const std::string &path)
{
  typedef itk::GDCMSeriesFileNames NamesGeneratorType;
  NamesGeneratorType::Pointer nameGenerator = NamesGeneratorType::New();
  nameGenerator->SetUseSeriesDetails( true );
  nameGenerator->SetDirectory( path );
  
  typedef itk::GDCMSeriesFileNames::SeriesUIDContainerType SeriesIdContainer;
  const SeriesIdContainer & seriesUID = nameGenerator->GetSeriesUIDs();
  
  if (! seriesUID.empty() )
  {
    return nameGenerator->GetFileNames( *(seriesUID.begin()) );
  }
  return std::vector<std::string>();
}
 
template <typename Image3D>
bool testDicomReaderFromDirectory()
{
  const std::string path = testPath + "samples/dicomSample";
  const std::vector<std::string> fileNames = getFirstDicomSerieFileNames( path );
  Image3D image = DicomReader< Image3D >::importDicomSeries( fileNames );
 
  trace.info() << image <<endl;
 
  unsigned int nbVal=0, nbPos = 0;
  typename Image3D::ConstRange r = image.constRange();
  for ( typename Image3D::ConstRange::ConstIterator it=r.begin(), itend=r.end() ; it != itend ; ++it )
  {
    nbVal++;
    if ( (*it) > 0 ) nbPos++;
  }
 
  trace.info() << "Number of points with (val>0) = " << nbVal << endl;
 
  return nbVal==2130048 && nbPos==296030;
}
 
 
template <typename PixelType>
bool testSpatialInformation()
{
  const std::string path = testPath + "samples/dicomSample";
 
  const std::vector<std::string> fileNames = getFirstDicomSerieFileNames( path );
 
  typedef ImageContainerByITKImage<Z3i::Domain,  PixelType> Image3D;
  Image3D image = DicomReader< Image3D >::importDicomSeries( fileNames );
  typename Image3D::ITKImagePointer dgtal_itk = image.getITKImagePointer();
 
  typedef itk::Image<PixelType, 3> ItkImage;
  typedef itk::ImageSeriesReader<ItkImage> ItkReader;
  typename ItkReader::Pointer reader = ItkReader::New();
  reader->SetFileNames( fileNames );
 
  reader->Update();
  
  typename ItkImage::Pointer itk = reader->GetOutput();
 
  const bool ok1 = ( dgtal_itk->GetSpacing() == itk->GetSpacing() );
  const bool ok2 = ( dgtal_itk->GetOrigin() == itk->GetOrigin() );
  const bool ok3 = ( dgtal_itk->GetDirection() == itk->GetDirection() );
  return ok1 && ok2 && ok3;
}
 
 
bool testIOException()
{
  //Default image selector = STLVector
  typedef ImageContainerBySTLVector<Z3i::Domain,  int > Image3D;
 
  std::string filename = testPath + "samples/null.dcm";
  try {
    Image3D image = DicomReader< Image3D >::importDicom( filename );
  }
  catch(exception& e) {
    trace.info() << "Exception catched. Message : " << e.what()<<endl;
    return true;
  }
 
  return false;
}
 
 
 
// Standard services - public :
 
int main( int argc, char** argv )
{
  trace.beginBlock ( "Testing class DicomReader" );
  trace.info() << "Args:";
  for ( int i = 0; i < argc; ++i )
    trace.info() << " " << argv[ i ];
  trace.info() << endl;
 
  bool res = testDicomReader<ImageContainerBySTLVector<Z3i::Domain, unsigned char> >()
    && testDicomReader<ImageContainerBySTLVector<Z3i::Domain, uint16_t> >()
    && testDicomReader<ImageContainerByITKImage<Z3i::Domain, unsigned char> >()
    && testDicomReader<ImageContainerByITKImage<Z3i::Domain, uint16_t> >()
    && testDicomReaderFromDirectory<ImageContainerBySTLVector<Z3i::Domain, unsigned char> >()
    && testDicomReaderFromDirectory<ImageContainerByITKImage<Z3i::Domain, uint16_t> >()
    && testSpatialInformation<unsigned char>()
    && testSpatialInformation<uint16_t>()
    && testIOException();
  trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
  trace.endBlock();
 
  return !res;
}
//                                                                           //