itkGPUTransformCopier.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 itkGPUTransformCopier_h
#define itkGPUTransformCopier_h
 
#include "itkTransform.h"
 
namespace itk
{
template <typename TTypeList, typename NDimensions, typename TTransform, typename TOutputTransformPrecisionType>
class ITK_TEMPLATE_EXPORT GPUTransformCopier : public Object
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(GPUTransformCopier);

  using Self = GPUTransformCopier;
  using Superclass = Object;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;

  itkNewMacro(Self);

  itkOverrideGetNameOfClassMacro(GPUTransformCopier);

  using CPUTransformType = TTransform;
  using CPUTransformConstPointer = typename CPUTransformType::ConstPointer;
  using CPUParametersType = typename CPUTransformType::ParametersType;
  using CPUFixedParametersType = typename CPUTransformType::FixedParametersType;
  using CPUScalarType = typename CPUTransformType::ScalarType;

  itkStaticConstMacro(InputSpaceDimension, unsigned int, CPUTransformType::InputSpaceDimension);
  itkStaticConstMacro(OutputSpaceDimension, unsigned int, CPUTransformType::OutputSpaceDimension);

  using GPUScalarType = TOutputTransformPrecisionType;
  using GPUTransformType = Transform<GPUScalarType, InputSpaceDimension, OutputSpaceDimension>;
  using GPUTransformPointer = typename GPUTransformType::Pointer;
  using GPUParametersType = typename GPUTransformType::ParametersType;
  using GPUFixedParametersType = typename GPUTransformType::FixedParametersType;

  itkSetConstObjectMacro(InputTransform, CPUTransformType);

  itkGetModifiableObjectMacro(Output, GPUTransformType);

  itkGetConstMacro(ExplicitMode, bool);
  itkSetMacro(ExplicitMode, bool);

  void
  Update();
 
#ifdef ITK_USE_CONCEPT_CHECKING
  // Begin concept checking
  itkConceptMacro(OutputTransformPrecisionTypeIsFloatingPointCheck,
                  (Concept::IsFloatingPoint<TOutputTransformPrecisionType>));
  // End concept checking
#endif
 
protected:
  GPUTransformCopier();
  ~GPUTransformCopier() override {}
  void
  PrintSelf(std::ostream & os, Indent indent) const override;

  bool
  CopyTransform(const CPUTransformConstPointer & fromTransform, GPUTransformPointer & toTransform);

  void
  CastCopyTransformParameters(const CPUTransformConstPointer & fromTransform, GPUTransformPointer & toTransform);

  void
  CastCopyParameters(const CPUParametersType & from, GPUParametersType & to);

  void
  CastCopyFixedParameters(const CPUFixedParametersType & from, GPUFixedParametersType & to);
 
private:
  bool
  CopyBSplineTransform(const CPUTransformConstPointer & fromTransform, GPUTransformPointer & toTransform);

  template <unsigned int Dimension>
  struct ITK_TEMPLATE_EXPORT TransformSpaceDimensionToType{};

  template <unsigned int InputSpaceDimension>
  bool
  CopyEuler2DTransform(const CPUTransformConstPointer &,
                       GPUTransformPointer &,
                       TransformSpaceDimensionToType<InputSpaceDimension>)
  {
    return false;
  }
 

  template <unsigned int InputSpaceDimension>
  bool
  CopyEuler3DTransform(const CPUTransformConstPointer &,
                       GPUTransformPointer &,
                       TransformSpaceDimensionToType<InputSpaceDimension>)
  {
    return false;
  }
 

  bool
  CopyEuler2DTransform(const CPUTransformConstPointer & fromTransform,
                       GPUTransformPointer &            toTransform,
                       TransformSpaceDimensionToType<2>);

  bool
  CopyEuler3DTransform(const CPUTransformConstPointer & fromTransform,
                       GPUTransformPointer &            toTransform,
                       TransformSpaceDimensionToType<3>);

  template <unsigned int InputSpaceDimension>
  bool
  CopySimilarity2DTransform(const CPUTransformConstPointer &,
                            GPUTransformPointer &,
                            TransformSpaceDimensionToType<InputSpaceDimension>)
  {
    return false;
  }
 

  template <unsigned int InputSpaceDimension>
  bool
  CopySimilarity3DTransform(const CPUTransformConstPointer &,
                            GPUTransformPointer &,
                            TransformSpaceDimensionToType<InputSpaceDimension>)
  {
    return false;
  }
 

  bool
  CopySimilarity2DTransform(const CPUTransformConstPointer & fromTransform,
                            GPUTransformPointer &            toTransform,
                            TransformSpaceDimensionToType<2>);

  bool
  CopySimilarity3DTransform(const CPUTransformConstPointer & fromTransform,
                            GPUTransformPointer &            toTransform,
                            TransformSpaceDimensionToType<3>);
 
private:
  CPUTransformConstPointer m_InputTransform{};
  GPUTransformPointer      m_Output{};
  ModifiedTimeType         m_InternalTransformTime{};
  bool                     m_ExplicitMode{};
};
 
} // end namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#  include "itkGPUTransformCopier.hxx"
#endif
 
#endif /* itkGPUTransformCopier_h */