Function BuildNumpyArray

Wrapping/Python/mitk/Image.cpp:126–158

}

py::array BuildNumpyArray(const Image& img, const void* data, bool writeable, py::capsule capsule)
{
  auto dtype = PixelTypeToDType(img.GetPixelType());
  auto format = PixelTypeFormat(img.GetPixelType());
  auto shape = ComputeNumpyShape(img);

  std::vector<py::ssize_t> strides(shape.size());
  strides.back() = dtype.itemsize();
  for (py::ssize_t i = static_cast<py::ssize_t>(shape.size()) - 2; i >= 0; --i)
    strides[i] = strides[i + 1] * shape[i + 1];

  // pybind11's buffer_info stores a non-const void*. Mirror what the
  // library itself does in its typed buffer_info(const T*, ...) overload:
  // const_cast the pointer here and let the readonly flag enforce
  // const-ness at the Python boundary.
  py::array arr(
    py::buffer_info(
      const_cast<void*>(data),
      dtype.itemsize(),
      format,
      shape.size(),
      shape,
      strides,
      !writeable
    ),
    capsule
  );

  if (!writeable)
    arr.attr("flags").attr("writeable") = py::bool_(false);

  return arr;
}

py::array AsNumpyDirect(Image& img, bool writeable, mitk::TimeStepType time_step)
{