NDArray.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#include "MantidPythonInterface/core/NDArray.h"
 
#include <boost/python/detail/prefix.hpp> // Safe include of Python.h
#include <boost/python/tuple.hpp>
#define PY_ARRAY_UNIQUE_SYMBOL CORE_ARRAY_API
#include <numpy/arrayobject.h>
 
using namespace boost::python;
 
namespace Mantid::PythonInterface {
 
void importNumpy() {
  if (!Py_IsInitialized()) {
    throw std::runtime_error("Library requires an active Python interpreter.\n"
                             "Call Py_Initialize at an appropriate point in the application.");
  }
 
  if (_import_array() < 0) {
    PyErr_Print();
    PyErr_SetString(PyExc_ImportError, "numpy.core.multiarray failed to import");
  }
}
 
PyTypeObject *ndarrayType() { return &PyArray_Type; }
 
// -----------------------------------------------------------------------------
// NDArray
// -----------------------------------------------------------------------------
 
bool NDArray::check(const object &obj) { return PyArray_Check(obj.ptr()); }
 
NDArray::NDArray(const object &obj) : object(detail::borrowed_reference(obj.ptr())) {}
 
Py_intptr_t const *NDArray::get_shape() const { return PyArray_DIMS(reinterpret_cast<PyArrayObject *>(this->ptr())); }
 
int NDArray::get_nd() const { return PyArray_NDIM(reinterpret_cast<PyArrayObject *>(this->ptr())); }
 
void *NDArray::get_data() const { return PyArray_DATA(reinterpret_cast<PyArrayObject *>(this->ptr())); }
 
char NDArray::get_typecode() const { return PyArray_DESCR(reinterpret_cast<PyArrayObject *>(this->ptr()))->type; }
 
NDArray NDArray::astype(char dtype, bool copy) const {
  auto callable = object(handle<>(PyObject_GetAttrString(this->ptr(), const_cast<char *>("astype"))));
  auto args = tuple();
  auto kwargs = object(handle<>(Py_BuildValue(const_cast<char *>("{s:c,s:i}"), "dtype", dtype, "copy", copy ? 1 : 0)));
  return NDArray(boost::python::detail::new_reference(PyObject_Call(callable.ptr(), args.ptr(), kwargs.ptr())));
}
 
} // namespace Mantid::PythonInterface
 
// -----------------------------------------------------------------------------
// object_manager_traits specialisation for NDArray
// -----------------------------------------------------------------------------
namespace boost::python::converter {
 
using Mantid::PythonInterface::ndarrayType;
 
bool object_manager_traits<Mantid::PythonInterface::NDArray>::check(PyObject *obj) {
  return ::PyObject_IsInstance(obj, (PyObject *)ndarrayType());
}
 
python::detail::new_reference object_manager_traits<Mantid::PythonInterface::NDArray>::adopt(PyObject *obj) {
  return python::detail::new_reference(python::pytype_check(ndarrayType(), obj));
}
 
PyTypeObject const *object_manager_traits<Mantid::PythonInterface::NDArray>::get_pytype() { return ndarrayType(); }
} // namespace boost::python::converter