]> Creatis software - bbtk.git/blob - packages/itkvtk/src/bbitkvtkitkImage2vtkImageData.cxx
77240acb1fd5dd9dd3406472827166c7fb2f1adc
[bbtk.git] / packages / itkvtk / src / bbitkvtkitkImage2vtkImageData.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: bbitkvtkitkImage2vtkImageData.cxx,v $
31   Language:  C++
32   Date:      $Date: 2012/11/16 08:50:49 $
33   Version:   $Revision: 1.12 $
34 =========================================================================*/
35
36
37 /**
38  *  \file 
39  *  \brief 
40  */
41 #ifdef _USE_ITK_
42 #ifdef _USE_VTK_
43
44 #include "bbitkvtkitkImage2vtkImageData.h"
45 #include "bbitkvtkPackage.h"
46 #include "itkImageToVTKImageFilter.h"
47
48 // -------------------------------------------------------------------------
49 // LFV: avoid itk::Concept check on 4d or superior images
50 #define BBITKVTK_itkImage2vtkImageData_Template( t, ty, di )            \
51   if( t == typeid( itk::Image< ty, di > ) )                             \
52     this->Convert< itk::Image< ty, di > >( )
53 // -------------------------------------------------------------------------
54
55 namespace bbitkvtk 
56 {
57   BBTK_BLACK_BOX_IMPLEMENTATION(itkImage2vtkImageData,bbtk::AtomicBlackBox);
58
59   BBTK_ADD_BLACK_BOX_TO_PACKAGE(itkvtk,itkImage2vtkImageData);
60
61   void itkImage2vtkImageData::Convert()
62   {
63     bbtk::TypeInfo t = bbGetInputIn().type();
64
65     // Only supports images in 1, 2 or 3 dimensions
66     BBITKVTK_itkImage2vtkImageData_Template( t, char, 1 );
67     else BBITKVTK_itkImage2vtkImageData_Template( t, short, 1 );
68     else BBITKVTK_itkImage2vtkImageData_Template( t, int, 1 );
69     else BBITKVTK_itkImage2vtkImageData_Template( t, long, 1 );
70     else BBITKVTK_itkImage2vtkImageData_Template( t, float, 1 );
71     else BBITKVTK_itkImage2vtkImageData_Template( t, double, 1 );
72     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned char, 1 );
73     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned short, 1 );
74     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned int, 1 );
75     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned long, 1 );
76     else BBITKVTK_itkImage2vtkImageData_Template( t, char, 2 );
77     else BBITKVTK_itkImage2vtkImageData_Template( t, short, 2 );
78     else BBITKVTK_itkImage2vtkImageData_Template( t, int, 2 );
79     else BBITKVTK_itkImage2vtkImageData_Template( t, long, 2 );
80     else BBITKVTK_itkImage2vtkImageData_Template( t, float, 2 );
81     else BBITKVTK_itkImage2vtkImageData_Template( t, double, 2 );
82     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned char, 2 );
83     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned short, 2 );
84     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned int, 2 );
85     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned long, 2 );
86     else BBITKVTK_itkImage2vtkImageData_Template( t, char, 3 );
87     else BBITKVTK_itkImage2vtkImageData_Template( t, short, 3 );
88     else BBITKVTK_itkImage2vtkImageData_Template( t, int, 3 );
89     else BBITKVTK_itkImage2vtkImageData_Template( t, long, 3 );
90     else BBITKVTK_itkImage2vtkImageData_Template( t, float, 3 );
91     else BBITKVTK_itkImage2vtkImageData_Template( t, double, 3 );
92     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned char, 3 );
93     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned short, 3 );
94     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned int, 3 );
95     else BBITKVTK_itkImage2vtkImageData_Template( t, unsigned long, 3 );
96
97     /* Deprecated
98        BBTK_TEMPLATE_ITK_IMAGE_SWITCH(t,Convert);
99     */
100   }
101
102    template<class T>
103    void itkImage2vtkImageData::Convert()
104    {
105     bbtkDebugMessage("process",5,"==> ["<<bbGetFullName()<<"] : Convert<"
106                      <<bbtk::TypeName<T>()
107                      <<">()"<<std::endl);
108
109     typedef T itkImageType;
110     typedef itkImageType* itkImageTypePointer;
111     typedef itk::ImageToVTKImageFilter< itkImageType > ItkToVtkConnection;
112     typename ItkToVtkConnection::Pointer conv;
113
114     // No converter yet : create it and set its input
115     if (!mConverter) 
116       {
117         bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : No converter yet : creating it"<<std::endl);
118         conv = ItkToVtkConnection::New();
119         mConverter = conv;
120         conv->SetInput( this->bbGetInputIn().get< itkImageTypePointer >() );
121       }
122     else 
123       {
124         bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Converter ok "<<std::endl);
125
126         // Input itkImageType type changed ? 
127         // Have to change the converter and set its input
128         conv = dynamic_cast<ItkToVtkConnection*>
129                            ((itk::ProcessObject*)mConverter);
130         if (!conv) 
131           {
132             bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Converter of wrong input type : reacreating it "<<std::endl);
133
134             mConverter->UnRegister();
135             conv = ItkToVtkConnection::New();
136             mConverter = conv;
137             conv->SetInput( this->bbGetInputIn().get<itkImageTypePointer>() );
138           }
139         // Input image type did not change but input image pointer did:
140         // set new input
141         else if ( this->bbGetInputIn().get<itkImageTypePointer>() 
142                   != (itkImageType*)(conv->GetExporter()->GetInputs()[0].GetPointer()))
143           {
144             bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Converter input changed : resetting it"<<std::endl);
145
146             conv->SetInput( this->bbGetInputIn().get<itkImageTypePointer>() );
147           }
148         else 
149           {
150             bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Nothing changed"<<std::endl);
151           }
152       }
153     try
154       {
155         
156         bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Trying update"<<std::endl);  
157
158         conv->Update();
159         bbSetOutputOut(conv->GetOutput());
160       }
161     catch( itk::ExceptionObject & e) 
162       {
163         bbtkError("itkImage2vtkImageData<"
164                   <<bbtk::TypeName<T>()
165                   <<">::Convert() : "<<e);
166       } 
167     bbtkDebugMessage("process",5,"<== ["<<bbGetFullName()<<"] : Convert<"
168                      <<bbtk::TypeName<T>()
169                      <<">() *DONE*"<<std::endl);
170  }
171
172   void itkImage2vtkImageData::bbUserSetDefaultValues()
173   {
174     bbSetOutputOut(NULL);
175     mConverter = 0;
176   }
177
178   void itkImage2vtkImageData::bbUserInitializeProcessing()
179   {
180   }
181
182   void itkImage2vtkImageData::bbUserFinalizeProcessing()
183   {
184   }
185
186
187 }
188 // eo namespace bbitkvtk
189
190 #endif
191 // _USE_VTK_
192 #endif
193 // _USE_ITK_