Testing/VTKTestWrite.cxx

   1 /*=========================================================================
   2
   3   Program:   gdcm
   4   Module:    $RCSfile: VTKTestWrite.cxx,v $
   5   Language:  C++
   6   Date:      $Date: 2005/02/09 15:31:15 $
   7   Version:   $Revision: 1.9 $
   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 "vtkGdcmWriter.h"
  20 #include "vtkImageViewer.h"
  21 #include "vtkImageData.h"
  22 #include "vtkRegressionTestImage.h"
  23 #include "vtkImageClip.h"
  24 #include "vtkImageTranslateExtent.h"
  25 #include "vtkImageAppendComponents.h"
  26 #include "vtkImageShiftScale.h"
  27
  28 #include <iostream>
  29
  30 //Generated file:
  31 #include "gdcmDataImages.h"
  32
  33 #ifndef vtkFloatingPointType
  34 #define vtkFloatingPointType float
  35 #endif
  36
  37 int VTKWriteTest(vtkTesting *t,vtkImageViewer *viewer,
  38                  std::string const &filename,
  39                  std::string const &referenceFileName)
  40 {
  41    int retVal = 0;  //by default this is an error
  42
  43    t->CleanArguments();
  44    t->AddArgument("-D");
  45    t->AddArgument( GDCM_DATA_ROOT );
  46    t->AddArgument("-V");
  47    t->AddArgument( referenceFileName.c_str() );
  48    t->AddArgument("-T");
  49    t->AddArgument( "." );
  50
  51    // Read the image
  52    vtkGdcmReader *origin = vtkGdcmReader::New();
  53    origin->SetFileName( filename.c_str() );
  54    origin->Update();
  55
  56    // Write the image
  57    vtkGdcmWriter *writer = vtkGdcmWriter::New();
  58    writer->SetFileName( "TestWrite.dcm" );
  59    writer->SetInput(origin->GetOutput());
  60    writer->Write();
  61
  62    origin->Delete();
  63    writer->Delete();
  64
  65    // Read the written image
  66    vtkGdcmReader *reader = vtkGdcmReader::New();
  67    reader->SetFileName( "TestWrite.dcm" );
  68    reader->Update();
  69
  70    double range[2];
  71    reader->GetOutput()->GetScalarRange(range);
  72    int dim[3];
  73    reader->GetOutput()->GetDimensions( dim );
  74
  75    // Show
  76    if( viewer )
  77    {
  78       viewer->SetInput ( reader->GetOutput() );
  79
  80       viewer->SetColorWindow (range[1] - range[0]);
  81       viewer->SetColorLevel (0.5 * (range[1] + range[0]));
  82
  83       viewer->SetSize(dim[0], dim[1]);
  84       if(dim[2] != 1)
  85       {
  86          //For multifame dicom, take a snapshot of the center slice (+/- 1)
  87          viewer->SetZSlice( dim[2] / 2 );
  88       }
  89       else
  90       {
  91          viewer->SetZSlice( 0 );
  92       }
  93
  94       viewer->OffScreenRenderingOff();
  95       viewer->Render();
  96       viewer->SetInput(NULL);
  97    }
  98
  99    //----------------------------------------------------------------------
 100    // Transform the image to be RGB unsigned char, due to the requests in
 101    // vtkTesting processing
 102    // The pipeline is broken after each process to keep maximum of memory
 103    vtkImageData *image=reader->GetOutput();
 104    image->Update();
 105    image->Register(NULL);
 106    reader->Delete();
 107
 108    // Get the middle slice in the image
 109    if(dim[2] != 1)
 110    {
 111       int *ext=image->GetExtent();
 112       vtkImageClip *clip=vtkImageClip::New();
 113       clip->SetInput(image);
 114       clip->SetOutputWholeExtent(ext[0],ext[1],ext[2],ext[3],
 115                                  ext[4]+dim[2] / 2,ext[4]+dim[2] / 2);
 116       clip->ClipDataOn();
 117       vtkImageTranslateExtent *translat=vtkImageTranslateExtent::New();
 118       translat->SetInput(clip->GetOutput());
 119       translat->SetTranslation(0,0,-ext[4]-dim[2] / 2);
 120
 121       image->UnRegister(NULL);
 122       image=translat->GetOutput();
 123       image->Update();
 124       image->Register(NULL);
 125       translat->SetOutput(NULL);
 126       clip->Delete();
 127       translat->Delete();
 128    }
 129
 130    // Set an unsigned char image
 131    // Shift/Scale the image
 132    image->Update();
 133    vtkImageShiftScale *iss=vtkImageShiftScale::New();
 134    iss->SetInput(image);
 135    iss->SetOutputScalarTypeToUnsignedChar();
 136    iss->SetShift(-range[0]);
 137    iss->SetScale(255.0/(range[1]-range[0]));
 138    iss->ClampOverflowOn();
 139
 140    image->UnRegister(NULL);
 141    image=iss->GetOutput();
 142    image->Update();
 143    image->Register(NULL);
 144    iss->Delete();
 145
 146    // Set 3 components to the image
 147    if(image->GetNumberOfScalarComponents()==1)
 148    {
 149       vtkImageAppendComponents *x3=vtkImageAppendComponents::New();
 150       x3->AddInput(image);
 151       x3->AddInput(image);
 152       x3->AddInput(image);
 153
 154       image->UnRegister(NULL);
 155       image=x3->GetOutput();
 156       image->Update();
 157       image->Register(NULL);
 158       x3->SetOutput(NULL);
 159       x3->Delete();
 160    }
 161    // End of transform
 162    //----------------------------------------------------------------------
 163
 164    // make test
 165    ostrstream str;
 166    retVal = t->RegressionTest(image,0.0,str);
 167    image->UnRegister(NULL);
 168
 169    if( retVal != vtkTesting::PASSED )
 170    {
 171       std::cerr << str.str();
 172    }
 173    str.rdbuf()->freeze(1);
 174
 175    if( retVal == vtkTesting::PASSED )
 176    {
 177       std::cerr << "       ... OK" << std::endl;
 178       return 0;
 179    }
 180    else
 181    {
 182       std::cerr << "       ... Failed" << std::endl;
 183       return 1;
 184    }
 185 }
 186
 187 int VTKTestWrite(int argc, char *argv[])
 188 {
 189    bool show = false;
 190    if( argc >= 2 )
 191    {
 192       if( std::string(argv[1]) == "-V" )
 193       {
 194          show = true;
 195       }
 196    }
 197
 198    int ret = 0;
 199    vtkTesting *t = vtkTesting::New();
 200    vtkImageViewer *viewer;
 201    if( show )
 202       viewer = vtkImageViewer::New();
 203    else
 204       viewer = NULL;
 205
 206    if( argc < 3+show )
 207    {
 208       std::cerr << "Usage: " << argv[0] << " [-V] image.dcm ref.png\n";
 209       std::cerr << "   -V : to view images to the screen... \n"
 210                 << "        this mode can generate errors in the test\n\n";
 211    }
 212    else
 213    {
 214       ret = VTKWriteTest(t,viewer,argv[1+show],argv[2+show]);
 215       t->Delete();
 216       if( viewer )
 217          viewer->Delete();
 218
 219       return ret;
 220    }
 221
 222    // Test for all images
 223    int i = 0;
 224    while( gdcmDataImages[i] != 0 )
 225    {
 226       std::string filename = GDCM_DATA_ROOT;
 227       filename += "/";  //doh!
 228       filename += gdcmDataImages[i];
 229       std::cerr << "Filename: " << filename << std::endl;
 230
 231       //Extract name to find the png file matching:
 232       std::string pngfile = gdcmDataImages[i++];
 233       //pngfile.replace(pngfile.size()-3, 3, "png");
 234       //More robust approach:
 235       std::string::size_type dot_pos = pngfile.rfind( "." );
 236       pngfile = pngfile.substr(0, dot_pos).append( ".png" );
 237       pngfile.insert( 0, "Baseline/");
 238       //std::cerr << "PNG file: " << pngfile << std::endl;
 239
 240       ret += VTKWriteTest(t,viewer,filename,pngfile);
 241    }
 242    t->Delete();
 243    if( viewer )
 244       viewer->Delete();
 245
 246    return ret;
 247 }