]> Creatis software - bbtk.git/blob - packages/itkvtk/src/bbitkvtkitkImageVector2vtkImageDataVector.cxx
c26717aeee183ff5e7b992cd8c26ec8543f347b3
[bbtk.git] / packages / itkvtk / src / bbitkvtkitkImageVector2vtkImageDataVector.cxx
1 /*
2 # ---------------------------------------------------------------------
3 #
4 # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
5 #                        pour la SantÈ)
6 # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
7 # Previous Authors : Laurent Guigues, Jean-Pierre Roux
8 # CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
9 #
10 #  This software is governed by the CeCILL-B license under French law and
11 #  abiding by the rules of distribution of free software. You can  use,
12 #  modify and/ or redistribute the software under the terms of the CeCILL-B
13 #  license as circulated by CEA, CNRS and INRIA at the following URL
14 #  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
15 #  or in the file LICENSE.txt.
16 #
17 #  As a counterpart to the access to the source code and  rights to copy,
18 #  modify and redistribute granted by the license, users are provided only
19 #  with a limited warranty  and the software's author,  the holder of the
20 #  economic rights,  and the successive licensors  have only  limited
21 #  liability.
22 #
23 #  The fact that you are presently reading this means that you have had
24 #  knowledge of the CeCILL-B license and that you accept its terms.
25 # ------------------------------------------------------------------------ */
26
27
28 /*=========================================================================
29   Program:   bbtk
30   Module:    $RCSfile: bbitkvtkitkImageVector2vtkImageDataVector.cxx,v $
31   Language:  C++
32   Date:      $Date: 2012/11/16 08:50:49 $
33   Version:   $Revision: 1.2 $
34 =========================================================================*/
35
36
37
38 /**
39  *  \file 
40  *  \brief 
41  */
42 #ifdef _USE_ITK_
43 #ifdef _USE_VTK_
44
45 #include "bbitkvtkitkImageVector2vtkImageDataVector.h"
46 #include "bbitkvtkPackage.h"
47 #include "itkImageToVTKImageFilter.h"
48
49 // -------------------------------------------------------------------------
50 // LFV: avoid itk::Concept check on 4d or superior images
51 #define BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, ty, di ) \
52   if( t == typeid( itk::Image< ty, di > ) )                             \
53     this->Convert< itk::Image< ty, di > >( )
54 // -------------------------------------------------------------------------
55
56 namespace bbitkvtk 
57 {
58   BBTK_BLACK_BOX_IMPLEMENTATION(itkImageVector2vtkImageDataVector,bbtk::AtomicBlackBox);
59
60   BBTK_ADD_BLACK_BOX_TO_PACKAGE(itkvtk,itkImageVector2vtkImageDataVector);
61
62   void itkImageVector2vtkImageDataVector::Convert()
63   {
64   // suppose *all* the images have the same type // JPR
65     bbtk::TypeInfo t = bbGetInputIn()[0].type();
66
67     // Only supports images in 1, 2 or 3 dimensions
68     BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, char, 1 );
69     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, short, 1 );
70     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, int, 1 );
71     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, long, 1 );
72     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, float, 1 );
73     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, double, 1 );
74     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned char, 1 );
75     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned short, 1 );
76     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned int, 1 );
77     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned long, 1 );
78     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, char, 2 );
79     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, short, 2 );
80     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, int, 2 );
81     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, long, 2 );
82     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, float, 2 );
83     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, double, 2 );
84     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned char, 2 );
85     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned short, 2 );
86     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned int, 2 );
87     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned long, 2 );
88     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, char, 3 );
89     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, short, 3 );
90     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, int, 3 );
91     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, long, 3 );
92     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, float, 3 );
93     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, double, 3 );
94     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned char, 3 );
95     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned short, 3 );
96     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned int, 3 );
97     else BBITKVTK_itkImageVector2vtkImageDataVector_Template( t, unsigned long, 3 );
98
99     /* Deprecated
100        BBTK_TEMPLATE_ITK_IMAGE_SWITCH(t,Convert);
101     */
102   }
103
104    template<class T>
105    void itkImageVector2vtkImageDataVector::Convert()
106    {
107     bbtkDebugMessage("process",5,"==> ["<<bbGetFullName()<<"] : Convert<"
108                      <<bbtk::TypeName<T>()
109                      <<">()"<<std::endl);
110
111     typedef T itkImageType;
112     typedef itkImageType* itkImageTypePointer;
113     typedef itk::ImageToVTKImageFilter< itkImageType > ItkToVtkConnection;
114     typename ItkToVtkConnection::Pointer conv;
115
116 int vectorSize = bbGetInputIn().size();
117 std::cout << " itkImageVector2vtkImageDataVector vector size :" << vectorSize << std::endl;
118 for (int i=0; i< vectorSize ; i++)
119 {
120     // No converter yet : create it and set its input
121     if (!mConverter) 
122       {
123         bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : No converter yet : creating it"<<std::endl);
124         conv = ItkToVtkConnection::New();
125         mConverter = conv;
126         conv->SetInput( this->bbGetInputIn()[i].get< itkImageTypePointer >() );
127       }
128     else 
129       {
130         bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Converter ok "<<std::endl);
131         
132         // Input itkImageType type changed ? 
133         // Have to change the converter and set its input
134         conv = dynamic_cast<ItkToVtkConnection*>
135           ((itk::ProcessObject*)mConverter);
136         if (!conv) 
137           {
138             bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Converter of wrong input type : reacreating it "<<std::endl);
139
140             mConverter->UnRegister();
141             conv = ItkToVtkConnection::New();
142             mConverter = conv;
143             conv->SetInput( this->bbGetInputIn()[i].get<itkImageTypePointer>() );
144           }
145         // Input image type did not change but input image pointer did:
146         // set new input
147         else if ( this->bbGetInputIn()[i].get<itkImageTypePointer>() 
148                   != (itkImageType*)(conv->GetExporter()->GetInputs()[0].GetPointer()))
149           {
150             bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Converter input changed : resetting it"<<std::endl);
151
152             conv->SetInput( this->bbGetInputIn()[i].get<itkImageTypePointer>() );
153           }
154         else 
155           {
156             bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Nothing changed"<<std::endl);
157           }
158       }
159     try
160       {
161         bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Trying update"<<std::endl);  
162
163         conv->Update();
164         m_Output_Vector.push_back(conv->GetOutput());
165       }
166     catch( itk::ExceptionObject & e) 
167       {
168         bbtkError("itkImageVector2vtkImageDataVector<"
169                   <<bbtk::TypeName<T>()
170                   <<">::Convert() : "<<e);
171       } 
172     bbtkDebugMessage("process",5,"<== ["<<bbGetFullName()<<"] : Convert<"
173                      <<bbtk::TypeName<T>()
174                      <<">() *DONE*"<<std::endl);
175                      
176 } // end iterate on vector size
177
178 bbSetOutputOut(m_Output_Vector);
179    
180 }
181
182
183   void itkImageVector2vtkImageDataVector::bbUserSetDefaultValues()
184   {
185    // bbSetOutputOut(std::vector<vtkImageData*>); // syntax? JPR
186     mConverter = 0;
187   }
188
189   void itkImageVector2vtkImageDataVector::bbUserInitializeProcessing()
190         {
191         }
192         
193   void itkImageVector2vtkImageDataVector::bbUserFinalizeProcessing()    
194         {
195         }
196
197         
198 }
199 // eo namespace bbitkvtk
200
201 #endif
202 // _USE_VTK_
203 #endif
204 // _USE_ITK_