/*=========================================================================
 *
 *  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 elxRayCastInterpolator_hxx
#define elxRayCastInterpolator_hxx

#include "elxRayCastInterpolator.h"

namespace elastix
{

/*
 * ***************** BeforeAll *****************
 */

template <typename TElastix>
int
RayCastInterpolator<TElastix>::BeforeAll()
{
  // Check if 2D-3D
  if constexpr (TElastix::FixedDimension != 3)
  {
    itkExceptionMacro("The RayCastInterpolator expects the fixed image to be 3D.");
    return 1;
  }
  if constexpr (TElastix::MovingDimension != 3)
  {
    itkExceptionMacro("The RayCastInterpolator expects the moving image to be 3D.");
    return 1;
  }

  return 0;
} // end BeforeAll()


/*
 * ***************** BeforeRegistration *****************
 */

template <typename TElastix>
void
RayCastInterpolator<TElastix>::BeforeRegistration()
{
  this->m_CombinationTransform = CombinationTransformType::New();
  this->m_CombinationTransform->SetUseComposition(true);

  unsigned int numberofparameters = this->m_Elastix->GetElxTransformBase()->GetAsITKBaseType()->GetNumberOfParameters();
  TransformParametersType preParameters(numberofparameters, 0.0);

  for (unsigned int i = 0; i < numberofparameters; ++i)
  {
    bool ret =
      this->GetConfiguration()->ReadParameter(preParameters[i], "PreParameters", this->GetComponentLabel(), i, 0);
    if (!ret)
    {
      std::cerr << " Error, not enough PreParameters are given" << std::endl;
    }
  }

  this->m_PreTransform = EulerTransformType::New();
  this->m_PreTransform->SetParameters(preParameters);
  this->m_CombinationTransform->SetInitialTransform(this->m_PreTransform);
  this->m_CombinationTransform->SetCurrentTransform(this->m_Elastix->GetElxTransformBase()->GetAsITKBaseType());

  this->SetTransform(this->m_CombinationTransform);

  PointType focalPoint{};

  for (unsigned int i = 0; i < TElastix::FixedDimension; ++i)
  {
    bool ret = this->GetConfiguration()->ReadParameter(focalPoint[i], "FocalPoint", this->GetComponentLabel(), i, 0);
    if (!ret)
    {
      std::cerr << "Error, FocalPoint not assigned" << std::endl;
    }
  }

  this->SetFocalPoint(focalPoint);

} // end BeforeRegistration()


/*
 * ***************** BeforeEachResolution *****************
 */

template <typename TElastix>
void
RayCastInterpolator<TElastix>::BeforeEachResolution()
{
  unsigned int level = (this->m_Registration->GetAsITKBaseType())->GetCurrentLevel();

  double threshold = 0.;
  this->GetConfiguration()->ReadParameter(threshold, "Threshold", this->GetComponentLabel(), level, 0);
  this->SetThreshold(threshold);

} // end BeforeEachResolution()


} // end namespace elastix

#endif // end #ifndef elxRayCastInterpolator_hxx
