test/standalone/simple/PyInterp_Utils.h

   1 // Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22 //  Author : Adrien BRUNETON
  23 //
  24
  25 #ifndef PYINTERP_UTILS_H
  26 #define PYINTERP_UTILS_H
  27
  28 #include "PyInterp.h"
  29 #include <Python.h>
  30 #ifdef _DEBUG_
  31   #include <iostream>
  32 #endif
  33
  34 /**
  35  * \class PyLockWrapper
  36  * \brief Python GIL wrapper.
  37  *
  38  * Utility class wrapping the Python GIL acquisition. This makes use of the high level
  39  * API (PyGILState_Ensure and PyGILState_Release), and is hence compatible with only
  40  * one running Python interpreter (no call to Py_NewInterpreter()).
  41  * When the class is instanciated the lock is acquired. It is released at destruction time.
  42  * Copy construction (and hence assignation) is forbidden.
  43  */
  44 class PyLockWrapper
  45 {
  46
  47 public:
  48   /**
  49    * \brief Constructor. Automatically acquires GIL.
  50    */
  51   PyLockWrapper()
  52   {
  53     _gil_state = PyGILState_Ensure();
  54     // Save current thread state for later comparison
  55     _state = PyGILState_GetThisThreadState();
  56   }
  57
  58   /**
  59    * \brief Destructor. Automatically releases GIL.
  60    */
  61   ~PyLockWrapper()
  62   {
  63     PyThreadState* _currState = PyGILState_GetThisThreadState();
  64 #ifdef _DEBUG_
  65     if (_currState != _state)
  66     {
  67       std::cout << "!!!!!!!!! PyLockWrapper inconsistency - now entering infinite loop for debugging\n";
  68       while(1);
  69     }
  70 #endif
  71     PyGILState_Release(_gil_state);
  72   }
  73
  74 private:
  75   PyGILState_STATE _gil_state;
  76   PyThreadState* _state;
  77
  78   // "Rule of 3" - Forbid usage of copy operator and copy-constructor
  79   PyLockWrapper(const PyLockWrapper & another);
  80   const PyLockWrapper & operator=(const PyLockWrapper & another);
  81 };
  82
  83
  84 /**
  85  * \class PyObjWrapper
  86  * \brief Utility class to properly handle the reference counting required on Python objects.
  87  */
  88 class PyObjWrapper
  89 {
  90   PyObject* myObject;
  91 public:
  92   PyObjWrapper(PyObject* theObject) : myObject(theObject) {}
  93   PyObjWrapper() : myObject(0) {}
  94   virtual ~PyObjWrapper() { Py_XDECREF(myObject); }
  95
  96   operator PyObject*()    { return myObject;  }
  97   PyObject* operator->()  { return myObject;  }
  98   PyObject* get()         { return myObject;  }
  99   bool operator!()        { return !myObject; }
 100   bool operator==(PyObject* theObject) { return myObject == theObject; }
 101   PyObject** operator&()  { return &myObject; }
 102   PyObjWrapper& operator=(PyObjWrapper* theObjWrapper)
 103   {
 104     Py_XDECREF(myObject);
 105     myObject = theObjWrapper->myObject;
 106     return *this;
 107   }
 108 };
 109
 110 #endif
 111