test/standalone/simple/PyInterp_Utils.h

   1 // Copyright (C) 2007-2022  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 #ifdef _DEBUG_
  64     if (PyGILState_GetThisThreadState() != _state)
  65     {
  66       std::cout << "!!!!!!!!! PyLockWrapper inconsistency - now entering infinite loop for debugging\n";
  67       while(1);
  68     }
  69 #endif
  70     PyGILState_Release(_gil_state);
  71   }
  72
  73 private:
  74   PyGILState_STATE _gil_state;
  75   PyThreadState* _state;
  76
  77   // "Rule of 3" - Forbid usage of copy operator and copy-constructor
  78   PyLockWrapper(const PyLockWrapper & another);
  79   const PyLockWrapper & operator=(const PyLockWrapper & another);
  80 };
  81
  82
  83 /**
  84  * \class PyObjWrapper
  85  * \brief Utility class to properly handle the reference counting required on Python objects.
  86  */
  87 class PyObjWrapper
  88 {
  89   PyObject* myObject;
  90 public:
  91   PyObjWrapper(PyObject* theObject) : myObject(theObject) {}
  92   PyObjWrapper() : myObject(0) {}
  93   virtual ~PyObjWrapper() { Py_XDECREF(myObject); }
  94
  95   operator PyObject*()    { return myObject;  }
  96   PyObject* operator->()  { return myObject;  }
  97   PyObject* get()         { return myObject;  }
  98   bool operator!()        { return !myObject; }
  99   bool operator==(PyObject* theObject) { return myObject == theObject; }
 100   PyObject** operator&()  { return &myObject; }
 101   PyObjWrapper& operator=(PyObjWrapper* theObjWrapper)
 102   {
 103     Py_XDECREF(myObject);
 104     myObject = theObjWrapper->myObject;
 105     return *this;
 106   }
 107 };
 108
 109 #endif
 110