]> Creatis software - gdcm.git/blob - Testing/VTKTestRead.cxx
ENH: for now do not support our own bugs
[gdcm.git] / 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 }