itkGPUBSplineTransformFactory.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 itkGPUBSplineTransformFactory_h
#define itkGPUBSplineTransformFactory_h
 
#include "itkGPUObjectFactoryBase.h"
#include "itkGPUBSplineTransform.h"
 
namespace itk
{
template <typename NDimensions>
class ITK_TEMPLATE_EXPORT GPUBSplineTransformFactory2 : public GPUObjectFactoryBase<NDimensions>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(GPUBSplineTransformFactory2);
 
  using Self = GPUBSplineTransformFactory2;
  using Superclass = GPUObjectFactoryBase<NDimensions>;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;
 
  const char *
  GetDescription() const
  {
    return "A Factory for GPUBSplineTransform";
  }
 
  itkFactorylessNewMacro(Self);
 
  itkTypeMacro(GPUBSplineTransformFactory2, GPUObjectFactoryBase);
 
  static void
  RegisterOneFactory();
 
  template <typename TType, unsigned int VImageDimension>
  void
  operator()()
  {
    // Override for spline order equal 1
    this->RegisterOverride(typeid(BSplineTransform<TType, VImageDimension, 1>).name(),
                           typeid(GPUBSplineTransform<TType, VImageDimension, 1>).name(),
                           "GPU BSplineTransform override",
                           true,
                           CreateObjectFunction<GPUBSplineTransform<TType, VImageDimension, 1>>::New());
 
    // Override for spline order equal 2
    this->RegisterOverride(typeid(BSplineTransform<TType, VImageDimension, 2>).name(),
                           typeid(GPUBSplineTransform<TType, VImageDimension, 2>).name(),
                           "GPU BSplineTransform override",
                           true,
                           CreateObjectFunction<GPUBSplineTransform<TType, VImageDimension, 2>>::New());
 
    // Override for spline order equal 3
    this->RegisterOverride(typeid(BSplineTransform<TType, VImageDimension, 3>).name(),
                           typeid(GPUBSplineTransform<TType, VImageDimension, 3>).name(),
                           "GPU BSplineTransform override",
                           true,
                           CreateObjectFunction<GPUBSplineTransform<TType, VImageDimension, 3>>::New());
  }
 
 
protected:
  GPUBSplineTransformFactory2();
  virtual ~GPUBSplineTransformFactory2() {}
 
  using RealTypeList = typelist::MakeTypeList<float, double>::Type;
 
  virtual void
  Register1D();
 
  virtual void
  Register2D();
 
  virtual void
  Register3D();
};
 
} // end namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#  include "itkGPUBSplineTransformFactory.hxx"
#endif
 
#endif /* itkGPUBSplineTransformFactory_h */