elxOpenCLMovingGenericPyramid.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright UMC Utrecht and contributors
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/
#ifndef __elxOpenCLMovingGenericPyramid_h
#define __elxOpenCLMovingGenericPyramid_h
 
#include "elxIncludes.h" // include first to avoid MSVS warning
#include "elxMovingGenericPyramid.h"
#include "itkGPUImage.h"
 
namespace elastix
{
 
template< class TElastix >
class OpenCLMovingGenericPyramid :
  public MovingGenericPyramid< TElastix >
{
public:
 
  typedef OpenCLMovingGenericPyramid                             Self;
  typedef MovingGenericPyramid< TElastix >                       Superclass;
  typedef typename MovingGenericPyramid< TElastix >::Superclass1 Superclass1;
  typedef typename MovingGenericPyramid< TElastix >::Superclass2 Superclass2;
  typedef itk::SmartPointer< Self >                              Pointer;
  typedef itk::SmartPointer< const Self >                        ConstPointer;
 
  itkNewMacro( Self );
 
  itkTypeMacro( OpenCLMovingGenericPyramid, MovingGenericPyramid );
 
  elxClassNameMacro( "OpenCLMovingGenericImagePyramid" );
 
  itkStaticConstMacro( ImageDimension, unsigned int, Superclass1::ImageDimension );
 
  typedef typename Superclass1::InputImageType             InputImageType;
  typedef typename Superclass1::OutputImageType            OutputImageType;
  typedef typename Superclass1::InputImageType::PixelType  InputImagePixelType;
  typedef typename Superclass1::OutputImageType::PixelType OutputImagePixelType;
 
  typedef typename itk::ObjectFactoryBase::Pointer ObjectFactoryBasePointer;
 
  typedef itk::GPUImage< InputImagePixelType, InputImageType::ImageDimension >
    GPUInputImageType;
  typedef typename GPUInputImageType::Pointer GPUInputImagePointer;
  typedef itk::GPUImage< OutputImagePixelType, OutputImageType::ImageDimension >
    GPUOutputImageType;
 
  typedef itk::GenericMultiResolutionPyramidImageFilter<
    GPUInputImageType,
    GPUOutputImageType, float > GPUPyramidType;
  typedef typename GPUPyramidType::Pointer GPUPyramidPointer;
 
  virtual void BeforeRegistration( void );
 
  virtual void ReadFromFile( void );
 
protected:
 
  void BeforeGenerateData( void );
 
  virtual void GenerateData( void );
 
  OpenCLMovingGenericPyramid();
  virtual ~OpenCLMovingGenericPyramid() {}
 
private:
 
  OpenCLMovingGenericPyramid( const Self & ); // purposely not implemented
  void operator=( const Self & ); // purposely not implemented
 
  void RegisterFactories( void );
 
  void UnregisterFactories( void );
 
  void SwitchingToCPUAndReport( const bool configError );
 
  void ReportToLog( void );
 
  GPUPyramidPointer                       m_GPUPyramid;
  bool                                    m_GPUPyramidReady;
  bool                                    m_GPUPyramidCreated;
  bool                                    m_ContextCreated;
  bool                                    m_UseOpenCL;
  std::vector< ObjectFactoryBasePointer > m_Factories;
};
 
} // end namespace elastix
 
#ifndef ITK_MANUAL_INSTANTIATION
#include "elxOpenCLMovingGenericPyramid.hxx"
#endif
 
#endif // end #ifndef __elxOpenCLMovingGenericPyramid_h