Function sitk_SetImageFromArray

Wrapping/Python/sitkNumpyArrayConversion.cxx:39–114

An internal function that performs a deep copy of the image buffer * into a python byte array. The byte array can later be converted * into a numpy array with the frombuffer method. */

   * into a numpy array with the frombuffer method.
   */
  static PyObject *
  sitk_SetImageFromArray(PyObject * SWIGUNUSEDPARM(self), PyObject * args)
  {
    PyObject * pyObj = NULL;
    PyObject * pyImage = NULL;

    Py_buffer pyBuffer;
    memset(&pyBuffer, 0, sizeof(Py_buffer));

    sitk::Image * sitkImage = NULL;
    void *        sitkBufferPtr = NULL;

    size_t len = 1;

    // We wish to support both the new PEP3118 buffer interface and the
    // older. So we first try to parse the arguments with the new buffer
    // protocol, then the old.
    if (!PyArg_ParseTuple(args, "OO", &pyObj, &pyImage))
    {
      return NULL;
    }
    if (PyObject_GetBuffer(pyObj, &pyBuffer, PyBUF_FULL_RO) != 0)
    {
      return NULL;
    }


    /* Cast over to a sitk Image. */
    {
      void * voidImage;
      int    res = 0;
      res = SWIG_ConvertPtr(pyImage, &voidImage, SWIGTYPE_p_itk__simple__Image, 0);
      if (!SWIG_IsOK(res))
      {
        SWIG_exception_fail(SWIG_ArgError(res),
                            "in method 'SetImageFromArray', argument needs to be of type 'sitk::Image *'");
      }
      sitkImage = reinterpret_cast<sitk::Image *>(voidImage);
    }

    try
    {
      if (sitkImage->GetPixelIDValue() == sitk::sitkUnknown)
      {
        PyErr_SetString(PyExc_RuntimeError, "Unknown pixel type.");
        SWIG_fail;
      }

      sitkBufferPtr = sitkImage->GetBufferAsVoid();
    }
    catch (const std::exception & e)
    {
      std::string msg = "Exception thrown in SimpleITK new Image: ";
      msg += e.what();
      PyErr_SetString(PyExc_RuntimeError, msg.c_str());
      goto fail;
    }