]> Creatis software - bbtk.git/blob - packages/vtk/src/bbvtkMetaImageWriter.cxx
Merge branch 'vtk8itk5wx3-mingw64' of ssh://git.creatis.insa-lyon.fr/bbtk into vtk8it...
[bbtk.git] / packages / vtk / src / bbvtkMetaImageWriter.cxx
1 //===== 
2 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
3 //===== 
4 #include "bbvtkMetaImageWriter.h"
5 #include "bbvtkPackage.h"
6
7 #include <vtkExtractVOI.h> 
8 #include <vtkMetaImageWriter.h> 
9
10 #include "vtkJSONImageWriter.h"
11 #include "vtkXMLImageDataWriter.h"
12 #include "vtkXMLDataSetWriter.h"
13 #include "vtkDataSetWriter.h"
14 #include "vtkPDataSetWriter.h"
15 #include "vtkXMLWriter.h"
16
17 namespace bbvtk
18 {
19
20 BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,MetaImageWriter)
21 BBTK_BLACK_BOX_IMPLEMENTATION(MetaImageWriter,bbtk::AtomicBlackBox);
22
23
24
25 void MetaImageWriter::Save_mhdb( std::string filename  , vtkImageData* img ,int sizeB)
26 {
27         int i,j,k;
28         int ext[6];
29         int dim[3];
30         int elementType=img->GetScalarType();
31         
32         img->GetExtent(ext);
33         dim[0]=ext[1]-ext[0]+1;
34         dim[1]=ext[3]-ext[2]+1;
35         dim[2]=ext[5]-ext[4]+1; 
36         int bsx=ceil((double)dim[0]/(double)sizeB);
37         int bsy=ceil((double)dim[1]/(double)sizeB);
38         int bsz=ceil((double)dim[2]/(double)sizeB);
39         int voi[6];
40         std::string filenameBlock;      
41         std::string filenameBlockVti;   
42         if (filename.substr(filename.size()-5) == ".mhdb")  
43         { 
44                 std::string cmd;
45 //              cmd="rm -rf "+filename;
46                 system( cmd.c_str() );
47                 cmd="mkdir "+filename;
48                 system( cmd.c_str() );
49         } 
50         
51         std::string fileinfoname=filename+"/info.dat";
52         FILE *ff=fopen(fileinfoname.c_str(),"w");
53           fprintf(ff,"SizeBlock = %d\n",sizeB);
54           fprintf(ff,"DimSize = %d %d %d\n" , dim[0],dim[1],dim[2] );
55           fprintf(ff,"ElementType = %d\n", elementType );
56         fclose(ff);     
57         
58         for (i=0;i<bsx;i++)
59         {
60                 for (j=0;j<bsy;j++)
61                 {
62                         for (k=0;k<bsz;k++)
63                         {
64                                 voi[0]=i*sizeB;
65                                 voi[1]=voi[0]+(sizeB-1);
66                                 voi[2]=j*sizeB;
67                                 voi[3]=voi[2]+(sizeB-1);
68                                 voi[4]=k*sizeB;
69                                 voi[5]=voi[4]+(sizeB-1);
70                                 if (voi[1]>=dim[0]) { voi[1]=dim[0]-1; } 
71                                 if (voi[3]>=dim[1]) { voi[3]=dim[1]-1; } 
72                                 if (voi[5]>=dim[2]) { voi[5]=dim[2]-1; } 
73                                 vtkExtractVOI *extract = vtkExtractVOI::New();
74                                 extract->SetInputData( img );
75                                 extract->SetVOI(voi);   
76                                 extract->UpdateWholeExtent();
77                                 extract->Modified();
78                                 extract->Update();
79                                 filenameBlock           =filename+"/mhdb-"+ std::to_string(i)+"-"+ std::to_string(j)+"-"+ std::to_string(k)+".mha";
80                                 vtkMetaImageWriter* w = vtkMetaImageWriter::New();
81                                 w->SetInputData( extract->GetOutput() );
82                                 w->SetCompression(true);  
83                                 w->SetFileDimensionality(bbGetInputIn()->GetDataDimension());   // NTU
84                                 w->SetFileName( filenameBlock.c_str() );  
85                                 w->Write();
86                                 w->Delete();
87                                 
88 /* Borrame
89                                 filenameBlockVti=filenameBlock+"-ZLib.vti";
90                                 vtkXMLImageDataWriter *writer = vtkXMLImageDataWriter::New();
91                                 writer->SetDataModeToBinary();
92 //                              writer->SetCompressionLevel(5);
93                                 writer->SetCompressorTypeToZLib();
94                                 writer->SetFileName( filenameBlockVti.c_str() );
95                                 writer->SetInputData( bbGetInputIn() );
96                                 writer->Write();
97 */
98                                 
99                                 extract->Delete();
100                         } // for k
101                 } // for j
102         } // for i
103 }
104
105
106
107 //===== 
108 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
109 //===== 
110 void MetaImageWriter::Process()
111 {
112
113 printf("EED vtkMetaImageWriter::Process  START\n");  
114
115 // THE MAIN PROCESSING METHOD BODY
116 //   Here we simply set the input 'In' value to the output 'Out'
117 //   And print out the output value
118 // INPUT/OUTPUT ACCESSORS ARE OF THE FORM :
119 //    void bbSet{Input|Output}NAME(const TYPE&)
120 //    const TYPE& bbGet{Input|Output}NAME() const 
121 //    Where :
122 //    * NAME is the name of the input/output
123 //      (the one provided in the attribute 'name' of the tag 'input')
124 //    * TYPE is the C++ type of the input/output
125 //      (the one provided in the attribute 'type' of the tag 'input')
126
127 //EED 2017-01-01 Migration VTK7
128 #if VTK_MAJOR_VERSION <= 5
129    bbGetInputIn()->Update();
130 #else
131         // ...
132 #endif
133         std::string userGivenName = bbGetInputFilename();
134         if (bbGetInputIn()!=NULL)
135         {
136                 if (userGivenName.substr(userGivenName.size()-5) == ".mhdb")  
137                 { 
138                         Save_mhdb( userGivenName , bbGetInputIn() , bbGetInputSizeBlock() );
139                 } else {
140                                 if (!((userGivenName.substr(userGivenName.size()-4) == ".mhd") || (userGivenName.substr(userGivenName.size()-4) == ".mha")))  //JPR
141                                 { 
142                                         userGivenName += ".mhd";
143                                 } 
144                            vtkMetaImageWriter* w = vtkMetaImageWriter::New();
145         //EED 2017-01-01 Migration VTK7
146         #if VTK_MAJOR_VERSION <= 5
147                                  w->SetInput(bbGetInputIn());
148         #else
149                                  w->SetInputData(bbGetInputIn());
150         #endif
151                                  w->SetCompression(bbGetInputCompression());  
152                                  w->SetFileDimensionality(bbGetInputIn()->GetDataDimension());  // NTU
153                                  w->SetFileName(userGivenName.c_str());  
154                                  //w->SetFileName(bbGetInputFilename().c_str());                                // JPR
155                                  w->Write();
156                                  w->Delete();
157                 } // .mhdb
158         } // bbGetInputIn()!=NULL
159 printf("EED vtkMetaImageWriter::Process  END\n");  
160 }
161 //===== 
162 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
163 //===== 
164 void MetaImageWriter::bbUserSetDefaultValues()
165 {
166
167 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX 
168 //    Here we initialize the input 'In' to 0
169    bbSetInputCompression(false);
170    bbSetInputSizeBlock(20);  
171 }
172 //===== 
173 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
174 //===== 
175 void MetaImageWriter::bbUserInitializeProcessing()
176 {
177
178 //  THE INITIALIZATION METHOD BODY :
179 //    Here does nothing 
180 //    but this is where you should allocate the internal/output pointers 
181 //    if any 
182
183   
184 }
185 //===== 
186 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
187 //===== 
188 void MetaImageWriter::bbUserFinalizeProcessing()
189 {
190
191 //  THE FINALIZATION METHOD BODY :
192 //    Here does nothing 
193 //    but this is where you should desallocate the internal/output pointers 
194 //    if any
195   
196 }
197 }
198 // EO namespace bbvtk
199
200