#include <numpy/arrayobject.h>
#include "pybind11/numpy.h"     // complex
#include "pybind11/pybind11.h"
#include "pybind11/operators.h"
#include "numpy/arrayobject.h"
//#include "ndarray/pybind11.h"
#include "xtensor/xtensor.hpp"
#include "xtensor/xcontainer.hpp"
#include "xtensor-python/pyarray.hpp"

#include <complex>

namespace py = pybind11;

PYBIND11_PLUGIN (summer) {
  if (_import_array() < 0) {
    PyErr_SetString(PyExc_ImportError, "numpy.core.multiarray failed to import");
    return nullptr;
  }

  py::module m("summer", "pybind11 example plugin");

  // m.def("sum", [](xt::pyarray<double> const& x) {
  //     xt::xtensor<double, 1> tmp(x);
  //     double sum = 0;
  //     for (auto e : tmp)
  //       sum += e;
  //     return sum;
  //   });

  m.def("sum", [](xt::pyarray<std::complex<double>> const& x) {
      std::complex<double> sum = 0;
      for (auto e : x)
        sum += e;
      return sum;
    },
    py::arg("inp").noconvert()
    );

  m.def("sum", [](xt::pyarray<double> const& x) {
      double sum = 0;
      for (auto e : x)
        sum += e;
      return sum;
    },
    py::arg("inp").noconvert()
    );


  return m.ptr();
}