itkObjectToObjectOptimizerBase.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright NumFOCUS
 *
 *  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
 *
 *         https://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 itkObjectToObjectOptimizerBase_h
#define itkObjectToObjectOptimizerBase_h
 
#include "ITKOptimizersv4Export.h"
 
#include "itkOptimizerParameters.h"
#include "itkOptimizerParameterScalesEstimator.h"
#include "itkObjectToObjectMetricBase.h"
#include "itkIntTypes.h"
 
namespace itk
{
class ObjectToObjectOptimizerBaseTemplateEnums
{
public:
  enum class StopConditionObjectToObjectOptimizer : uint8_t
  {
    MAXIMUM_NUMBER_OF_ITERATIONS,
    COSTFUNCTION_ERROR,
    UPDATE_PARAMETERS_ERROR,
    STEP_TOO_SMALL,
    CONVERGENCE_CHECKER_PASSED,
    GRADIENT_MAGNITUDE_TOLEARANCE,
    OTHER_ERROR
  };
};
// Define how to print enumeration
using StopConditionObjectToObjectOptimizerEnum =
  ObjectToObjectOptimizerBaseTemplateEnums::StopConditionObjectToObjectOptimizer;
extern ITKOptimizersv4_EXPORT std::ostream &
                              operator<<(std::ostream &                                                                       out,
           const ObjectToObjectOptimizerBaseTemplateEnums::StopConditionObjectToObjectOptimizer value);

template <typename TInternalComputationValueType = double>
class ITK_TEMPLATE_EXPORT ObjectToObjectOptimizerBaseTemplate : public Object
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(ObjectToObjectOptimizerBaseTemplate);

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

  itkOverrideGetNameOfClassMacro(ObjectToObjectOptimizerBaseTemplate);

  using ScalesType = OptimizerParameters<TInternalComputationValueType>;
  using ScalesEstimatorType = OptimizerParameterScalesEstimatorTemplate<TInternalComputationValueType>;

  using ParametersType = OptimizerParameters<TInternalComputationValueType>;

  using MetricType = ObjectToObjectMetricBaseTemplate<TInternalComputationValueType>;
  using MetricTypePointer = typename MetricType::Pointer;

  using DerivativeType = typename MetricType::DerivativeType;

  using NumberOfParametersType = typename MetricType::NumberOfParametersType;

  using MeasureType = typename MetricType::MeasureType;

  using StopConditionReturnStringType = std::string;

  using StopConditionDescriptionType = std::ostringstream;

  itkSetObjectMacro(Metric, MetricType);
  itkGetModifiableObjectMacro(Metric, MetricType);

  itkGetConstReferenceMacro(CurrentMetricValue, MeasureType);

  virtual const MeasureType &
  GetValue() const;

  // itkSetMacro( Scales, ScalesType );
  virtual void
  SetScales(const ScalesType & scales)
  {
    this->m_Scales = scales;
  }


  itkGetConstReferenceMacro(Scales, ScalesType);

  itkGetConstReferenceMacro(ScalesAreIdentity, bool);

  itkSetMacro(Weights, ScalesType);

  itkGetConstReferenceMacro(Weights, ScalesType);

  itkGetConstReferenceMacro(WeightsAreIdentity, bool);

  bool
  GetScalesInitialized() const;

  itkSetObjectMacro(ScalesEstimator, ScalesEstimatorType);

  itkSetMacro(DoEstimateScales, bool);
  itkGetConstReferenceMacro(DoEstimateScales, bool);
  itkBooleanMacro(DoEstimateScales);

  virtual void
  SetNumberOfWorkUnits(ThreadIdType number);
 
#if !defined(ITK_LEGACY_REMOVE)
  itkLegacyMacro(virtual void SetNumberOfThreads(ThreadIdType number))
  {
    return this->SetNumberOfWorkUnits(number);
  }
  itkLegacyMacro(virtual const ThreadIdType & GetNumberOfThreads() const)
  {
    return this->m_NumberOfWorkUnits;
  }
#endif // !ITK_LEGACY_REMOVE

  itkGetConstReferenceMacro(NumberOfWorkUnits, ThreadIdType);

  itkGetConstMacro(CurrentIteration, SizeValueType);

  itkSetMacro(NumberOfIterations, SizeValueType);

  itkGetConstMacro(NumberOfIterations, SizeValueType);

  virtual const ParametersType &
  GetCurrentPosition() const;

  virtual void
  StartOptimization(bool doOnlyInitialization = false);

  virtual StopConditionReturnStringType
  GetStopConditionDescription() const = 0;

  virtual bool
  CanUseScales() const
  {
    return true;
  }
 
protected:
  ObjectToObjectOptimizerBaseTemplate();
 
  ~ObjectToObjectOptimizerBaseTemplate() override;
 
  MetricTypePointer m_Metric{};
  ThreadIdType      m_NumberOfWorkUnits{};
  SizeValueType     m_CurrentIteration{};
  SizeValueType     m_NumberOfIterations{};

  MeasureType m_CurrentMetricValue{};

  ScalesType m_Scales{};

  ScalesType m_Weights{};

  bool m_ScalesAreIdentity{};

  typename ScalesEstimatorType::Pointer m_ScalesEstimator{};

  bool m_WeightsAreIdentity{};

  bool m_DoEstimateScales{};
 
  void
  PrintSelf(std::ostream & os, Indent indent) const override;
};

using ObjectToObjectOptimizerBase = ObjectToObjectOptimizerBaseTemplate<double>;
 
} // end namespace itk
 
#endif

#ifndef ITK_TEMPLATE_EXPLICIT_ObjectToObjectOptimizerBaseTemplate
// Explicit instantiation is required to ensure correct dynamic_cast
// behavior across shared libraries.
//
// IMPORTANT: Since within the same compilation unit,
//            ITK_TEMPLATE_EXPLICIT_<classname> defined and undefined states
//            need to be considered. This code *MUST* be *OUTSIDE* the header
//            guards.
//
#if defined(ITKOptimizersv4_EXPORTS)
//   We are building this library
#  define ITKOptimizersv4_EXPORT_EXPLICIT ITK_FORWARD_EXPORT
#else
//   We are using this library
#  define ITKOptimizersv4_EXPORT_EXPLICIT ITKOptimizersv4_EXPORT
#endif
namespace itk
{
 
ITK_GCC_PRAGMA_DIAG_PUSH()
ITK_GCC_PRAGMA_DIAG(ignored "-Wattributes")
 
#if defined(_MSC_VER)
#  pragma warning(disable : 4661) // no suitable definition provided for explicit template instantiation request
#endif
extern template class ITKOptimizersv4_EXPORT_EXPLICIT ObjectToObjectOptimizerBaseTemplate<double>;
extern template class ITKOptimizersv4_EXPORT_EXPLICIT ObjectToObjectOptimizerBaseTemplate<float>;
 
ITK_GCC_PRAGMA_DIAG_POP()
 
} // end namespace itk
#undef ITKOptimizersv4_EXPORT_EXPLICIT
#endif