src/Basics/PythonCppUtils.hxx

   1 // Copyright (C) 2021-2024  CEA, EDF
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #pragma once
  21
  22 #include <Python.h>
  23
  24 class AutoGIL
  25 {
  26 public:
  27   AutoGIL():_gstate(PyGILState_Ensure()) { }
  28   ~AutoGIL() { PyGILState_Release(_gstate); }
  29 private:
  30   PyGILState_STATE _gstate;
  31 };
  32
  33 class AutoPyRef
  34 {
  35 public:
  36   AutoPyRef(PyObject *pyobj=nullptr):_pyobj(pyobj) { }
  37   virtual ~AutoPyRef() { release(); }
  38   AutoPyRef(const AutoPyRef& other):_pyobj(other._pyobj) { if(_pyobj) Py_XINCREF(_pyobj); }
  39   AutoPyRef(AutoPyRef&& other) = default;
  40   AutoPyRef& operator=(const AutoPyRef& other) { if(_pyobj==other._pyobj) return *this; release(); _pyobj=other._pyobj; Py_XINCREF(_pyobj); return *this; }
  41   operator PyObject *() { return _pyobj; }
  42   void set(PyObject *pyobj) { if(pyobj==_pyobj) return ; release(); _pyobj=pyobj; }
  43   PyObject *get() { return _pyobj; }
  44   bool isNull() const { return _pyobj==0; }
  45   PyObject *retn() { if(_pyobj) Py_XINCREF(_pyobj); return _pyobj; }
  46 protected:
  47   void release() { if(_pyobj) Py_XDECREF(_pyobj); _pyobj=nullptr; }
  48 private:
  49   PyObject *_pyobj = nullptr;
  50 };
  51
  52 class AutoPyRefGilSafe : public AutoPyRef
  53 {
  54 public:
  55   AutoPyRefGilSafe(PyObject *pyobj=nullptr):AutoPyRef(pyobj) { }
  56   ~AutoPyRefGilSafe() { AutoGIL agil; release(); }
  57 };