#include <nanobind/nanobind.h>
#include <nanobind/stl/vector.h>
#include <nanobind/stl/string.h>
#include <nanobind/stl/tuple.h>
#include <nanobind/stl/list.h>
#include <nanobind/ndarray.h>
#include <nanobind/stl/shared_ptr.h>


#include <exception>
#include <vector>
#include <string>
#include <algorithm>

#include "itkImage.h"

#include "antscore/itkSurfaceImageCurvature.h"

#include "antsImage.h"

namespace nb = nanobind;
using namespace nb::literals;

AntsImage<itk::Image<float, 3>> weingartenImageCurvature( AntsImage<itk::Image<float, 3>> myimage,
                                      float sigma, int opt )
{
  typedef itk::Image<float, 3> ImageType;

  typedef typename ImageType::Pointer       ImagePointerType;
  enum { ImageDimension = ImageType::ImageDimension };
  typedef itk::SurfaceImageCurvature<ImageType> ParamType;
  typename ParamType::Pointer Parameterizer = ParamType::New();
  typename ImageType::Pointer input = myimage.ptr;
  typename ImageType::DirectionType imgdir = input->GetDirection();
  typename ImageType::DirectionType iddir = input->GetDirection();
  iddir.SetIdentity();
  input->SetDirection( iddir );
  Parameterizer->SetInputImage(input);
  Parameterizer->SetNeighborhoodRadius( 1. );
  if( sigma <= 0.5 )
      {
      sigma = 1.66;
      }
  Parameterizer->SetSigma(sigma);
  Parameterizer->SetUseLabel(false);
  Parameterizer->SetUseGeodesicNeighborhood(false);
  float sign = 1.0;
  Parameterizer->SetkSign(sign);
  Parameterizer->SetThreshold(0);
  if( opt != 5 && opt != 6 )
      {
      Parameterizer->ComputeFrameOverDomain( 3 );
      }
  else
      {
      Parameterizer->ComputeFrameOverDomain( opt );
      }
  typename ImageType::Pointer output = Parameterizer->GetFunctionImage();
  output->SetDirection( imgdir );

  AntsImage<ImageType> outImage = { output };
  return outImage;

}


void local_weingartenImageCurvature(nb::module_ &m)
{
  m.def("weingartenImageCurvature", &weingartenImageCurvature);
}