wxPython/include/wx/wxPython/twoitem.h

   1 ////////////////////////////////////////////////////////////////////////////
   2 // Name:        twoitem.h
   3 // Purpose:     A template function to help with converting a python object
   4 //              to some wx class that takes two integer value parameters.
   5 //              Factored out of wxPython_int.h
   6 //
   7 // Author:      Robin Dunn
   8 //
   9 // Created:     25-April-2006
  10 // RCS-ID:      $Id$
  11 // Copyright:   (c) 2006 by Total Control Software
  12 // Licence:     wxWindows license
  13 /////////////////////////////////////////////////////////////////////////////
  14
  15 #ifndef __twoitem_h__
  16 #define __twoitem_h__
  17
  18 template<class T>
  19 bool wxPyTwoIntItem_helper(PyObject* source, T** obj, const wxChar* name)
  20 {
  21     // If source is an object instance then it may already be the right type
  22     if (wxPySwigInstance_Check(source)) {
  23         T* ptr;
  24         if (! wxPyConvertSwigPtr(source, (void **)&ptr, name))
  25             goto error;
  26         *obj = ptr;
  27         return true;
  28     }
  29     // otherwise a 2-tuple of integers is expected
  30     else if (PySequence_Check(source) && PyObject_Length(source) == 2) {
  31         PyObject* o1 = PySequence_GetItem(source, 0);
  32         PyObject* o2 = PySequence_GetItem(source, 1);
  33         if (!PyNumber_Check(o1) || !PyNumber_Check(o2)) {
  34             Py_DECREF(o1);
  35             Py_DECREF(o2);
  36             goto error;
  37         }
  38         **obj = T(PyInt_AsLong(o1), PyInt_AsLong(o2));
  39         Py_DECREF(o1);
  40         Py_DECREF(o2);
  41         return true;
  42     }
  43
  44  error:
  45     wxString msg;
  46     msg.Printf(wxT("Expected a 2-tuple of integers or a %s object."), name);
  47     PyErr_SetString(PyExc_TypeError, msg.mb_str());
  48     return false;
  49 }
  50
  51
  52 #endif