vtkITKImageToImageFilter.h

Go to the documentation of this file.

/*=========================================================================
 
  Copyright Brigham and Women's Hospital (BWH) All Rights Reserved.
 
  See COPYRIGHT.txt
  or http://www.slicer.org/copyright/copyright.txt for details.
 
  Program:   vtkITK
  Module:    $HeadURL$
  Date:      $Date$
  Version:   $Revision$
 
==========================================================================*/
 
#ifndef __vtkITKImageToImageFilter_h
#define __vtkITKImageToImageFilter_h
 
#include "vtkITK.h"
 
// ITK includes
#include <itkCommand.h>
#include <itkProcessObject.h>
 
// VTK includes
#include <vtkCommand.h>
#include <vtkImageAlgorithm.h>
#include <vtkImageCast.h>
#include <vtkImageData.h>
#include <vtkImageExport.h>
#include <vtkImageImport.h>
#include <vtkVersion.h>
 
// STD includes
#include <iostream>
 
#undef itkExceptionMacro
#define itkExceptionMacro(x)                                                     \
  {                                                                              \
    ::std::ostringstream message;                                                \
    message << "itk::ERROR: " << this->GetNameOfClass() << "(" << this << "): "; \
    std::cout << message.str().c_str() << std::endl;                             \
  }
 
#undef itkGenericExceptionMacro
#define itkGenericExceptionMacro(x)          \
  {                                          \
    ::std::ostringstream message;            \
    message << "itk::ERROR: " x;             \
    std::cout << message.str() << std::endl; \
  }

class VTK_ITK_EXPORT vtkITKImageToImageFilter : public vtkImageAlgorithm
{
public:
  static vtkITKImageToImageFilter* New()
  {
    vtkITKImageToImageFilter* result = new vtkITKImageToImageFilter;
#ifdef VTK_HAS_INITIALIZE_OBJECT_BASE
    result->InitializeObjectBase();
#endif
    return result;
  };
 
  vtkTypeMacro(vtkITKImageToImageFilter, vtkImageAlgorithm);
 
  void PrintSelf(ostream& os, vtkIndent indent) override
  {
    Superclass::PrintSelf(os, indent);
    this->vtkExporter->PrintSelf(os, indent);
    this->vtkImporter->PrintSelf(os, indent);
  };

  vtkMTimeType GetMTime() override
  {
    vtkMTimeType t1, t2;
 
    t1 = this->Superclass::GetMTime();
    t2 = this->vtkExporter->GetMTime();
    if (t2 > t1)
    {
      t1 = t2;
    }
    t2 = this->vtkImporter->GetMTime();
    if (t2 > t1)
    {
      t1 = t2;
    }
    return t1;
  };

  void Modified() override
  {
    this->Superclass::Modified();
    if (this->m_Process)
    {
      m_Process->Modified();
    }
  };

  void DebugOn() override { this->m_Process->DebugOn(); };

  void DebugOff() override { this->m_Process->DebugOff(); };

  void SetNumberOfThreads(int val) { this->m_Process->SetNumberOfWorkUnits(val); };

  int GetNumberOfThreads() { return this->m_Process->GetNumberOfWorkUnits(); };

  void SetOutput(vtkDataObject* d) override { this->vtkImporter->SetOutput(d); };
  virtual vtkImageData* GetOutput() { return this->vtkImporter->GetOutput(); };
  virtual vtkImageData* GetOutput(int idx) { return (vtkImageData*)this->vtkImporter->GetOutput(idx); };

  virtual void SetInput(vtkImageData* Input) { this->vtkCast->SetInputData(Input); };
 
  void SetInputConnection(vtkAlgorithmOutput* input) override { this->vtkCast->SetInputConnection(input); };
 
  void SetInputConnection(int port, vtkAlgorithmOutput* input) override { this->vtkCast->SetInputConnection(port, input); };

  virtual vtkDataObject* GetInput() { return (vtkDataObject::SafeDownCast(this->vtkCast->GetInput())); };

  using vtkAlgorithm::Update;
  void Update() override
  {
    this->vtkCast->Update();
    this->vtkImporter->Update();
  }
  void Update(int port) override
  {
    this->vtkCast->Update();
    this->vtkImporter->Update(port);
  }
  void HandleProgressEvent()
  {
    if (this->m_Process)
    {
      this->UpdateProgress(m_Process->GetProgress());
    }
  };
  void HandleStartEvent() { this->InvokeEvent(vtkCommand::StartEvent, nullptr); };
  void HandleEndEvent() { this->InvokeEvent(vtkCommand::EndEvent, nullptr); };
 
protected:
  void ExecuteData(vtkDataObject*) override { vtkWarningMacro(<< "This filter does not respond to Update(). Doing a GetOutput->Update() instead."); }
 
  vtkITKImageToImageFilter()
  {
    this->vtkCast = vtkImageCast::New();
    this->vtkExporter = vtkImageExport::New();
    this->vtkImporter = vtkImageImport::New();
    this->vtkExporter->SetInputConnection(this->vtkCast->GetOutputPort());
    this->m_Process = nullptr;
    this->m_ProgressCommand = MemberCommand::New();
    this->m_ProgressCommand->SetCallbackFunction(this, &vtkITKImageToImageFilter::HandleProgressEvent);
    this->m_StartEventCommand = MemberCommand::New();
    this->m_StartEventCommand->SetCallbackFunction(this, &vtkITKImageToImageFilter::HandleStartEvent);
    this->m_EndEventCommand = MemberCommand::New();
    this->m_EndEventCommand->SetCallbackFunction(this, &vtkITKImageToImageFilter::HandleEndEvent);
  };
  ~vtkITKImageToImageFilter() override
  {
    vtkDebugMacro("Destructing vtkITKImageToImageFilter");
    this->vtkExporter->Delete();
    this->vtkImporter->Delete();
    this->vtkCast->Delete();
  };

  void LinkITKProgressToVTKProgress(itk::ProcessObject* process)
  {
    if (process)
    {
      this->m_Process = process;
      this->m_Process->AddObserver(itk::ProgressEvent(), this->m_ProgressCommand);
      this->m_Process->AddObserver(itk::StartEvent(), this->m_StartEventCommand);
      this->m_Process->AddObserver(itk::EndEvent(), this->m_EndEventCommand);
    }
  };
 
  typedef itk::SimpleMemberCommand<vtkITKImageToImageFilter> MemberCommand;
  typedef MemberCommand::Pointer MemberCommandPointer;
 
  itk::ProcessObject::Pointer m_Process;
  MemberCommandPointer m_ProgressCommand;
  MemberCommandPointer m_StartEventCommand;
  MemberCommandPointer m_EndEventCommand;

  vtkImageCast* vtkCast;
  vtkImageImport* vtkImporter;
  vtkImageExport* vtkExporter;
 
private:
  vtkITKImageToImageFilter(const vtkITKImageToImageFilter&) = delete;
  void operator=(const vtkITKImageToImageFilter&) = delete;
};
 
#endif