// -*- mode:c++ -*-
//
// Module test_array_ext.cpp
//
// Copyright (c) 2003 Raoul M. Gough
//
// Use, modification and distribution is subject to the Boost Software
// License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy
// at http://www.boost.org/LICENSE_1_0.txt)
//
// History
// =======
// 2003/10/15   rmg     File creation from testlinear.cpp
//
// $Id$
//

#include "int_wrapper.hpp"

#include <boost/python/suite/indexing/container_suite.hpp>
#include <boost/python/suite/indexing/iterator_pair.hpp>
#include <boost/python/class.hpp>
#include <boost/python/module.hpp>
#include <boost/python/def.hpp>
#include <boost/python/implicit.hpp>

// More messiness from not having a separate int_wrapper.cpp file
bool int_wrapper::our_trace_flag = true;
unsigned int_wrapper::our_object_counter = 0;

boost::python::indexing::iterator_pair<int *> get_array_plain()
{
  static int array[] = { 8, 6, 4, 2, 1, 3, 5, 7, 0 };

  return boost::python::indexing::iterator_pair<int *>
    (boost::python::indexing::begin(array)
     , boost::python::indexing::end(array));
}

boost::python::indexing::iterator_pair<int_wrapper *> get_array_wrap()
{
  static int_wrapper array[] = {
    int_wrapper(8), int_wrapper(6), int_wrapper(4), int_wrapper(2)
    , int_wrapper(1), int_wrapper(3), int_wrapper(5)
    , int_wrapper(7), int_wrapper(0) };

  return boost::python::indexing::iterator_pair<int_wrapper *>
    (boost::python::indexing::begin(array)
     , boost::python::indexing::end(array));
}

BOOST_PYTHON_MODULE(test_array_ext)
{
  boost::python::implicitly_convertible <int, int_wrapper>();

  boost::python::def ("setTrace", &int_wrapper::setTrace);

  boost::python::class_<int_wrapper>("int_wrapper", boost::python::init<int>())
    .def ("increment", &int_wrapper::increment)
    .def ("__repr__", &int_wrapper::repr)
    .def ("__cmp__", compare)
    ;

  typedef boost::python::indexing::iterator_pair<int *> Container1;

  boost::python::class_<Container1>
    ("Array", boost::python::init<int *, int *>())
    .def (boost::python::indexing::container_suite<Container1>());

  boost::python::def ("get_array_plain", get_array_plain);

  typedef boost::python::indexing::iterator_pair<int_wrapper *> Container2;

  // reference_existing_object is safe in this case, because the array
  // is static, and we never manually destroy any array elements. There
  // is also no point in using return_internal_reference to extend the
  // life of the iterator_pair object, since it has no influence on the
  // lifetimes of the array elements.
  boost::python::class_<Container2>
    ("Array_ref", boost::python::init<int_wrapper *, int_wrapper *>())
    .def (boost::python::indexing::container_suite<Container2>
          ::with_policies(boost::python::return_value_policy
                          <boost::python::reference_existing_object>()));

  boost::python::def ("get_array_wrap", get_array_wrap);
}