packages/itkvtk/src/bbitkvtkitkImage2vtkImageData.cxx

   1 /*=========================================================================
   2   Program:   bbtk
   3   Module:    $RCSfile: bbitkvtkitkImage2vtkImageData.cxx,v $
   4   Language:  C++
   5   Date:      $Date: 2008/11/12 12:47:05 $
   6   Version:   $Revision: 1.6 $
   7 =========================================================================*/
   8
   9 /* ---------------------------------------------------------------------
  10
  11 * Copyright (c) CREATIS-LRMN (Centre de Recherche en Imagerie Medicale)
  12 * Authors : Eduardo Davila, Laurent Guigues, Jean-Pierre Roux
  13 *
  14 *  This software is governed by the CeCILL-B license under French law and
  15 *  abiding by the rules of distribution of free software. You can  use,
  16 *  modify and/ or redistribute the software under the terms of the CeCILL-B
  17 *  license as circulated by CEA, CNRS and INRIA at the following URL
  18 *  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  19 *  or in the file LICENSE.txt.
  20 *
  21 *  As a counterpart to the access to the source code and  rights to copy,
  22 *  modify and redistribute granted by the license, users are provided only
  23 *  with a limited warranty  and the software's author,  the holder of the
  24 *  economic rights,  and the successive licensors  have only  limited
  25 *  liability.
  26 *
  27 *  The fact that you are presently reading this means that you have had
  28 *  knowledge of the CeCILL-B license and that you accept its terms.
  29 * ------------------------------------------------------------------------ */
  30
  31 /**
  32  *  \file
  33  *  \brief
  34  */
  35 #ifdef _USE_ITK_
  36 #ifdef _USE_VTK_
  37
  38 #include "bbitkvtkitkImage2vtkImageData.h"
  39 #include "bbitkvtkPackage.h"
  40 // VtkToItkConnection
  41 //#include "itkVTKImageToImageFilter.h"
  42 // ItkToVtkConnection
  43 #include "itkImageToVTKImageFilter.h"
  44
  45
  46 namespace bbitkvtk
  47 {
  48   BBTK_BLACK_BOX_IMPLEMENTATION(itkImage2vtkImageData,bbtk::AtomicBlackBox);
  49
  50   BBTK_ADD_BLACK_BOX_TO_PACKAGE(itkvtk,itkImage2vtkImageData);
  51
  52
  53
  54
  55   void itkImage2vtkImageData::Convert()
  56   {
  57     bbtk::TypeInfo t = bbGetInputIn().type();
  58     BBTK_TEMPLATE_ITK_IMAGE_SWITCH(t,Convert);
  59   }
  60
  61
  62
  63    template<class T>
  64    void itkImage2vtkImageData::Convert()
  65    {
  66     bbtkDebugMessage("process",5,"==> ["<<bbGetFullName()<<"] : Convert<"
  67                      <<bbtk::TypeName<T>()
  68                      <<">()"<<std::endl);
  69
  70     typedef T itkImageType;
  71         typedef itkImageType* itkImageTypePointer;
  72     typedef itk::ImageToVTKImageFilter< itkImageType > ItkToVtkConnection;
  73     typename ItkToVtkConnection::Pointer conv;
  74
  75     // No converter yet : create it and set its input
  76     if (!mConverter)
  77       {
  78         bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : No converter yet : creating it"<<std::endl);
  79         conv = ItkToVtkConnection::New();
  80         mConverter = conv;
  81         conv->SetInput( this->bbGetInputIn().get< itkImageTypePointer >() );
  82       }
  83     else
  84       {
  85         bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Converter ok "<<std::endl);
  86
  87         // Input itkImageType type changed ?
  88         // Have to change the converter and set its input
  89         conv = dynamic_cast<ItkToVtkConnection*>
  90           ((itk::ProcessObject*)mConverter);
  91         if (!conv)
  92           {
  93             bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Converter of wrong input type : reacreating it "<<std::endl);
  94
  95             mConverter->UnRegister();
  96             conv = ItkToVtkConnection::New();
  97             mConverter = conv;
  98             conv->SetInput( this->bbGetInputIn().get<itkImageTypePointer>() );
  99           }
 100         // Input image type did not change but input image pointer did:
 101         // set new input
 102         else if ( this->bbGetInputIn().get<itkImageTypePointer>()
 103                   != (itkImageType*)(conv->GetExporter()->GetInputs()[0].GetPointer()))
 104           {
 105             bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Converter input changed : resetting it"<<std::endl);
 106
 107             conv->SetInput( this->bbGetInputIn().get<itkImageTypePointer>() );
 108           }
 109         else
 110           {
 111             bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Nothing changed"<<std::endl);
 112
 113           }
 114       }
 115     try
 116       {
 117
 118         bbtkDebugMessage("process",5,"    ["<<bbGetFullName()<<"] : Trying update"<<std::endl);
 119
 120         conv->Update();
 121         bbSetOutputOut(conv->GetOutput());
 122       }
 123     catch( itk::ExceptionObject & e)
 124       {
 125         bbtkError("itkImage2vtkImageData<"
 126                   <<bbtk::TypeName<T>()
 127                   <<">::Convert() : "<<e);
 128       }
 129     bbtkDebugMessage("process",5,"<== ["<<bbGetFullName()<<"] : Convert<"
 130                      <<bbtk::TypeName<T>()
 131                      <<">() *DONE*"<<std::endl);
 132
 133
 134  }
 135
 136   void itkImage2vtkImageData::bbUserConstructor()
 137   {
 138     bbSetOutputOut(NULL);
 139     mConverter = 0;
 140   }
 141   void itkImage2vtkImageData::bbUserCopyConstructor()
 142   {
 143     bbSetOutputOut(NULL);
 144     mConverter = 0;
 145   }
 146   void itkImage2vtkImageData::bbUserDestructor()
 147   {
 148   }
 149
 150
 151 }
 152 // eo namespace bbitkvtk
 153
 154 #endif
 155 // _USE_VTK_
 156 #endif
 157 // _USE_ITK_