]> Creatis software - gdcm.git/blob - Testing/VTKTestReadSeq.cxx
ENH: for now do not support our own bugs
[gdcm.git] / Testing / VTKTestReadSeq.cxx
1 /*=========================================================================
2                                                                                 
3   Program:   gdcm
4   Module:    $RCSfile: VTKTestReadSeq.cxx,v $
5   Language:  C++
6   Date:      $Date: 2007/09/18 07:58:38 $
7   Version:   $Revision: 1.12 $
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 <iostream>
28 #include <sstream>
29
30 //Generated file:
31 #include "gdcmDataSeqImages.h"
32
33 #ifndef vtkFloatingPointType
34 #define vtkFloatingPointType float
35 #endif
36
37 int VTKReadSeqTest(vtkTesting *t, vtkImageViewer *viewer,
38                    std::string const &filename, 
39                    std::string const &referenceFileName)
40 {
41    int retVal; // = 0;  (to avoid bcc 5.5 warnings)
42
43    // Set the reader   
44    vtkGdcmReader *reader = vtkGdcmReader::New();
45
46    char *newFileDcm = new char[filename.size()+1];
47    int fileCount = 0;
48    for(int i=0;i<9;i++)
49    {
50       fileCount = i;
51       sprintf(newFileDcm,filename.c_str(),i);
52
53       // Test the existance of the file
54       ifstream opened(newFileDcm,std::ios::in | std::ios::binary);
55       if(opened)
56       {
57          reader->AddFileName(newFileDcm);
58          opened.close();
59       }
60       else
61          break;
62    }
63    delete[] newFileDcm;
64    reader->Update();
65
66    double range[2];
67    reader->GetOutput()->GetScalarRange(range);
68
69    // Show
70    if( viewer )
71    {
72       viewer->SetInput ( reader->GetOutput() );
73       viewer->OffScreenRenderingOff();
74
75       viewer->SetColorWindow (range[1] - range[0]);
76       viewer->SetColorLevel (0.5 * (range[1] + range[0]));
77
78       int dim[3];
79       reader->GetOutput()->GetDimensions( dim );
80       viewer->SetSize(dim[0], dim[1]);
81       viewer->SetZSlice( 0 );
82       viewer->Render();
83       viewer->SetInput(NULL);
84    }
85
86    // make test
87    int ret = 0;
88    std::ostringstream str;
89    char *newFilePng = new char[referenceFileName.size()+1];
90    for(int j=0;j<fileCount;j++)
91    {
92       sprintf(newFilePng,referenceFileName.c_str(),j);
93
94       t->CleanArguments();
95       t->AddArgument("-D");
96       t->AddArgument( GDCM_DATA_ROOT );
97       t->AddArgument("-V");
98       t->AddArgument( newFilePng );
99       t->AddArgument("-T");
100       t->AddArgument( "." );
101
102       //----------------------------------------------------------------------
103       // Transform the image to be RGB unsigned char, due to the requests in
104       // vtkTesting processing
105       // The pipeline is broken after each process to keep maximum of memory
106       vtkImageData *image=reader->GetOutput();
107       image->Update();
108       image->Register(NULL);
109
110       // Get the middle slice in the image
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],j,j);
115       clip->ClipDataOn();
116       vtkImageTranslateExtent *translat=vtkImageTranslateExtent::New();
117       translat->SetInput(clip->GetOutput());
118       translat->SetTranslation(0,0,-j);
119
120       image->UnRegister(NULL);
121       image=translat->GetOutput();
122       image->Update();
123       image->Register(NULL);
124       translat->SetOutput(NULL);
125       clip->Delete();
126       translat->Delete();
127
128       // Set an unsigned char image
129       // Shift/Scale the image 
130       image->Update();
131       vtkImageShiftScale *iss=vtkImageShiftScale::New();
132       iss->SetInput(image);
133       iss->SetOutputScalarTypeToUnsignedChar();
134       iss->SetShift(-range[0]);
135       iss->SetScale(255.0/(range[1]-range[0]));
136       iss->ClampOverflowOn();
137
138       image->UnRegister(NULL);
139       image=iss->GetOutput();
140       image->Update();
141       image->Register(NULL);
142       iss->Delete();
143
144       // Set 3 components to the image
145       if(image->GetNumberOfScalarComponents()==1)
146       {
147          vtkImageAppendComponents *x3=vtkImageAppendComponents::New();
148          x3->AddInput(image);
149          x3->AddInput(image);
150          x3->AddInput(image);
151
152          image->UnRegister(NULL);
153          image=x3->GetOutput();
154          image->Update();
155          image->Register(NULL);
156          x3->SetOutput(NULL);
157          x3->Delete();
158       }
159       // End of transform
160       //----------------------------------------------------------------------
161
162       // make test
163       retVal = t->RegressionTest(image,2.0,str);
164       image->UnRegister(NULL);
165
166       if( retVal != vtkTesting::PASSED )
167       {
168          std::cerr << str.str();
169       }
170
171       if( retVal == vtkTesting::PASSED )
172       {
173          std::cerr << "       ...Slice " << j << ": OK" << std::endl;
174       }
175       else
176       {
177          std::cerr << "       ...Slice " << j << ": Failed" << std::endl;
178          ret++;
179       }
180    }
181
182    delete[] newFilePng;
183    reader->Delete();
184
185    return ret;
186 }
187
188 int VTKTestReadSeq(int argc, char *argv[])
189 {
190    bool show = false;
191    if( argc >= 2 )
192    {
193       if( std::string(argv[1]) == "-V" )
194       {
195          show = true;
196       }
197    }
198
199    int ret = 0;
200    vtkTesting *t = vtkTesting::New();
201    vtkImageViewer *viewer;
202    if( show )
203       viewer = vtkImageViewer::New();
204    else
205       viewer = NULL;
206
207    if( argc < 3+show )
208    {
209       std::cerr << "Usage: " << argv[0] << " [-V] image%d.dcm ref%d.png\n";
210       std::cerr << "   -V : to view images to the screen... \n"
211                 << "        this mode can generate errors in the test\n";
212       std::cerr << "   %d : this will be replaced by a number at execution.\n"
213                 << "        It will be from 0 to 9 only with a step of 1\n\n";
214    }
215    else
216    {
217       ret = VTKReadSeqTest(t,viewer,argv[1+show],argv[2+show]);
218       t->Delete();
219       if( viewer )
220          viewer->Delete();
221
222       return ret;
223    }
224
225    // Test for all images
226    int i = 0;
227    while( gdcmDataSeqImages[i] != 0 )
228    {
229       std::string filename = GDCM_DATA_ROOT;
230       filename += "/";  //doh!
231       filename += gdcmDataSeqImages[i];
232       std::cerr << "Filename: " << filename << std::endl;
233
234       //Extract name to find the png file matching:
235       std::string pngfile = gdcmDataSeqImages[i++];
236       std::string::size_type dot_pos = pngfile.rfind( "." );
237       pngfile = pngfile.substr(0, dot_pos).append( ".png" );
238       pngfile.insert( 0, "Baseline/");
239       
240       ret += VTKReadSeqTest(t,viewer,filename,pngfile);
241    }
242    t->Delete();
243    if( viewer )
244       viewer->Delete();
245
246    return ret;
247 }