src/SALOMESDS/SALOMESDS_Auto.hxx

   1 // Copyright (C) 2018  CEA/DEN, EDF R&D, OPEN CASCADE
   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 // Author : Anthony GEAY (EDF R&D)
  20
  21 #pragma once
  22
  23 #include "Python.h"
  24
  25 namespace SALOME
  26 {
  27   class AutoPyRef
  28   {
  29   public:
  30     AutoPyRef(PyObject *pyobj=nullptr):_pyobj(pyobj) { }
  31     ~AutoPyRef() { release(); }
  32     AutoPyRef(const AutoPyRef& other):_pyobj(other._pyobj) { if(_pyobj) Py_XINCREF(_pyobj); }
  33     AutoPyRef(AutoPyRef&& other):_pyobj(other._pyobj) { other._pyobj=nullptr; }
  34     AutoPyRef& operator=(const AutoPyRef& other) { if(_pyobj==other._pyobj) return *this; release(); _pyobj=other._pyobj; Py_XINCREF(_pyobj); return *this; }
  35     AutoPyRef& operator=(AutoPyRef&& other) { if(_pyobj==other._pyobj) return *this; release(); _pyobj=other._pyobj; other._pyobj=nullptr; return *this; }
  36     operator PyObject *() { return _pyobj; }
  37     void set(PyObject *pyobj) { if(pyobj==_pyobj) return ; release(); _pyobj=pyobj; }
  38     PyObject *get() { return _pyobj; }
  39     bool isNull() const { return _pyobj==nullptr; }
  40     PyObject *retn() { if(_pyobj) Py_XINCREF(_pyobj); return _pyobj; }
  41   private:
  42     void release() { if(_pyobj) Py_XDECREF(_pyobj); _pyobj=nullptr; }
  43   private:
  44     PyObject *_pyobj;
  45   };
  46 }
  47