#include <cctbx/boost_python/flex_fwd.h>

#include <boost/python/module.hpp>
#include <boost/python/def.hpp>
#include <boost/python/class.hpp>
#include <boost/python/args.hpp>
#include <boost/python/return_value_policy.hpp>
#include <boost/python/return_by_value.hpp>
#include <boost/optional.hpp>

#include <mmtbx/den/den.h>
#include <cctbx/geometry_restraints/proxy_select.h>
#include <scitbx/array_family/boost_python/shared_wrapper.h>

namespace mmtbx { namespace den {
namespace {

  struct simple_restraints_wrappers : boost::python::pickle_suite
  {
    typedef den_simple_proxy w_t;

    static boost::python::tuple
      getinitargs(w_t const& self)
    {
      return boost::python::make_tuple(self.i_seqs,
        self.eq_distance,
        self.eq_distance_start,
        self.weight
        );
    }

    static void
    wrap()
    {
      using namespace boost::python;
      typedef return_value_policy<return_by_value> rbv;
      class_<w_t>("den_simple_proxy", no_init)
        .def(init<
          af::tiny<unsigned, 2> const&, double, double, double >((
            arg("i_seqs"),
            arg("eq_distance"),
            arg("eq_distance_start"),
            arg("weight"))))
        .add_property("i_seqs", make_getter(&w_t::i_seqs, rbv()))
        .def_readwrite("eq_distance", &w_t::eq_distance)
        .def_readwrite("eq_distance_start", &w_t::eq_distance_start)
        .def_readwrite("weight", &w_t::weight)
        .def_pickle(simple_restraints_wrappers())
        ;
      {
        typedef return_internal_reference<> rir;
        scitbx::af::boost_python::shared_wrapper<den_simple_proxy, rir>::wrap(
          "shared_den_simple_proxy")
          .def("proxy_select",
            (af::shared<w_t>(*)(
              af::const_ref<w_t> const&,
              std::size_t,
              af::const_ref<std::size_t> const&))
            cctbx::geometry_restraints::shared_proxy_select, (
              arg("n_seq"), arg("iselection")));
      }

      def("den_simple_residual_sum",
        (double(*)(
          af::const_ref<scitbx::vec3<double> > const&,
          af::const_ref<den_simple_proxy> const&,
          af::ref<scitbx::vec3<double> > const&,
          double,
          double))
        den_simple_residual_sum, (
          arg("sites_cart"),
          arg("proxies"),
          arg("gradient_array"),
          arg("den_weight") = 1.0));

      def("den_update_eq_distances",
        (void(*)(
          af::const_ref<scitbx::vec3<double> > const&,
          af::ref<den_simple_proxy> const&,
          double,
          double))
        den_update_eq_distances, (
          arg("sites_cart"),
          arg("proxies"),
          arg("gamma"),
          arg("kappa")));
    }
  }
  ;
} // namespace anonymous

namespace boost_python {

  void wrap_den() {
    simple_restraints_wrappers::wrap();
  }

} // namespace boost_python

}} //namespace mmtbx::den

BOOST_PYTHON_MODULE(mmtbx_den_restraints_ext)
{
  mmtbx::den::boost_python::wrap_den();
}