Testing/VTKTestRead.cxx

   1 /*=========================================================================
   2
   3   Program:   gdcm
   4   Module:    $RCSfile: VTKTestRead.cxx,v $
   5   Language:  C++
   6   Date:      $Date: 2007/09/18 07:58:38 $
   7   Version:   $Revision: 1.15 $
   8
   9   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
  10   l'Image). All rights reserved. See Doc/License.txt or
  11   http://www.creatis.insa-lyon.fr/Public/Gdcm/License.html for details.
  12
  13      This software is distributed WITHOUT ANY WARRANTY; without even
  14      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  15      PURPOSE.  See the above copyright notices for more information.
  16
  17 =========================================================================*/
  18 #include "vtkGdcmReader.h"
  19 #include "vtkImageViewer.h"
  20 #include "vtkImageData.h"
  21 #include "vtkRegressionTestImage.h"
  22 #include "vtkImageClip.h"
  23 #include "vtkImageTranslateExtent.h"
  24 #include "vtkImageAppendComponents.h"
  25 #include "vtkImageShiftScale.h"
  26
  27 #include "vtkPNGWriter.h"
  28
  29 #include <iostream>
  30 #include <sstream>
  31
  32 //Generated file:
  33 #include "gdcmDataImages.h"
  34
  35 #ifndef vtkFloatingPointType
  36 #define vtkFloatingPointType float
  37 #endif
  38
  39 int VTKReadTest(vtkTesting *t,vtkImageViewer *viewer,
  40                 std::string const &filename,
  41                 std::string const &referenceFileName)
  42 {
  43    int retVal; // = 0; (to avoid bcc5.5 warnings)
  44
  45    t->CleanArguments();
  46    t->AddArgument("-D");
  47    t->AddArgument( GDCM_DATA_ROOT );
  48    t->AddArgument("-V");
  49    t->AddArgument( referenceFileName.c_str() );
  50    t->AddArgument("-T");
  51    t->AddArgument( "." );
  52
  53    vtkGdcmReader *reader = vtkGdcmReader::New();
  54    reader->SetFileName( filename.c_str() );
  55    reader->Update();
  56
  57    double range[2];
  58    reader->GetOutput()->GetScalarRange(range);
  59    int dim[3];
  60    reader->GetOutput()->GetDimensions( dim );
  61
  62    // Show
  63    if(viewer)
  64    {
  65       viewer->SetInput ( reader->GetOutput() );
  66
  67       viewer->SetColorWindow (range[1] - range[0]);
  68       viewer->SetColorLevel (0.5 * (range[1] + range[0]));
  69
  70       viewer->SetSize(dim[0], dim[1]);
  71       if(dim[2] != 1)
  72       {
  73          //For multifame dicom, take a snapshot of the center slice (+/- 1)
  74          viewer->SetZSlice( dim[2] / 2 );
  75       }
  76       else
  77       {
  78          viewer->SetZSlice( 0 );
  79       }
  80
  81       viewer->OffScreenRenderingOff();
  82       viewer->Render();
  83       viewer->SetInput(NULL);
  84    }
  85
  86    //----------------------------------------------------------------------
  87    // Transform the image to be RGB unsigned char, due to the requests in
  88    // vtkTesting processing
  89    // The pipeline is broken after each process to keep maximum of memory
  90    vtkImageData *image=reader->GetOutput();
  91    image->Update();
  92    image->Register(NULL);
  93    reader->Delete();
  94
  95    // Get the middle slice in the image
  96    if(dim[2] != 1)
  97    {
  98       int *ext=image->GetExtent();
  99       vtkImageClip *clip=vtkImageClip::New();
 100       clip->SetInput(image);
 101       clip->SetOutputWholeExtent(ext[0],ext[1],ext[2],ext[3],
 102                                  ext[4]+dim[2] / 2,ext[4]+dim[2] / 2);
 103       clip->ClipDataOn();
 104       vtkImageTranslateExtent *translat=vtkImageTranslateExtent::New();
 105       translat->SetInput(clip->GetOutput());
 106       translat->SetTranslation(0,0,-ext[4]-dim[2] / 2);
 107
 108       image->UnRegister(NULL);
 109       image=translat->GetOutput();
 110       image->Update();
 111       image->Register(NULL);
 112       translat->SetOutput(NULL);
 113       clip->Delete();
 114       translat->Delete();
 115    }
 116
 117    // Set an unsigned char image
 118    // Shift/Scale the image
 119    vtkImageShiftScale *iss=vtkImageShiftScale::New();
 120    iss->SetInput(image);
 121    iss->SetOutputScalarTypeToUnsignedChar();
 122    iss->SetShift(-range[0]);
 123    iss->SetScale(255.0/(range[1]-range[0]));
 124    iss->ClampOverflowOn();
 125
 126    image->UnRegister(NULL);
 127    image=iss->GetOutput();
 128    image->Update();
 129    image->Register(NULL);
 130    iss->Delete();
 131
 132    // Set 3 components to the image
 133    if(image->GetNumberOfScalarComponents()==1)
 134    {
 135       vtkImageAppendComponents *x3=vtkImageAppendComponents::New();
 136       x3->AddInput(image);
 137       x3->AddInput(image);
 138       x3->AddInput(image);
 139
 140       image->UnRegister(NULL);
 141       image=x3->GetOutput();
 142       image->Update();
 143       image->Register(NULL);
 144       x3->SetOutput(NULL);
 145       x3->Delete();
 146    }
 147    // End of transform
 148    //----------------------------------------------------------------------
 149
 150    // make test
 151    std::ostringstream str;
 152    retVal = t->RegressionTest(image,0.0,str);
 153    image->UnRegister(NULL);
 154
 155    if( retVal != vtkTesting::PASSED )
 156    {
 157       std::cerr << str.str();
 158    }
 159
 160    if( retVal == vtkTesting::PASSED )
 161    {
 162       std::cerr << "       ... OK" << std::endl;
 163       return 0;
 164    }
 165    else
 166    {
 167       std::cerr << "       ... Failed" << std::endl;
 168       return 1;
 169    }
 170
 171 }
 172
 173 int VTKTestRead(int argc, char *argv[])
 174 {
 175    bool show = false;
 176    if( argc >= 2 )
 177    {
 178       if( std::string(argv[1]) == "-V" )
 179       {
 180          show = true;
 181       }
 182    }
 183
 184    int ret = 0;
 185    vtkTesting *t = vtkTesting::New();
 186    vtkImageViewer *viewer;
 187    if( show )
 188       viewer = vtkImageViewer::New();
 189    else
 190       viewer = NULL;
 191
 192    if( argc < 3+show )
 193    {
 194       std::cerr << "Usage: " << argv[0] << " [-V] image.dcm ref.png\n";
 195       std::cerr << "   -V : to view images to the screen... \n"
 196                 << "        this mode can generate errors in the test\n\n";
 197    }
 198    else
 199    {
 200       ret = VTKReadTest(t,viewer,argv[1+show],argv[2+show]);
 201       t->Delete();
 202       if( viewer )
 203          viewer->Delete();
 204
 205       return ret;
 206    }
 207
 208    //gdcm::Debug::DebugOn(); // Comment out when no bug is to be tracked
 209
 210    // Test for all images
 211    int i = 0;
 212    while( gdcmDataImages[i] != 0 )
 213    {
 214       std::string filename = GDCM_DATA_ROOT;
 215       filename += "/";  //doh!
 216       filename += gdcmDataImages[i];
 217       std::cerr << "Filename: " << filename << std::endl;
 218
 219       //Extract name to find the png file matching:
 220       std::string pngfile = gdcmDataImages[i++];
 221       //pngfile.replace(pngfile.size()-3, 3, "png");
 222       //More robust approach:
 223       std::string::size_type dot_pos = pngfile.rfind( "." );
 224       pngfile = pngfile.substr(0, dot_pos).append( ".png" );
 225       pngfile.insert( 0, "Baseline/");
 226       //std::cerr << "PNG file: " << pngfile << std::endl;
 227
 228       ret += VTKReadTest(t,viewer,filename,pngfile);
 229    }
 230    t->Delete();
 231    if( viewer )
 232       viewer->Delete();
 233
 234    return ret;
 235 }