Testing/VTKTestWrite.cxx

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