PyReverseProxy.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PYKERNEL_PYREVERSEPROXY_H
#define PYKERNEL_PYREVERSEPROXY_H

 
#ifdef _POSIX_C_SOURCE
# undef _POSIX_C_SOURCE
#endif
#ifdef _XOPEN_SOURCE
# undef _XOPEN_SOURCE
#endif
#include "Python.h"
#include <unordered_map>
#include <string>
#include "RootUtils/PyAthenaGILStateEnsure.h"
#include "CxxUtils/checker_macros.h"
 
 
struct PyReverseProxy
{
  ~PyReverseProxy()
  {
    RootUtils::PyGILStateEnsure gil;
    s_proxyMap.erase(m_key);
  }

  void setFunc (PyObject * func)
  {
    RootUtils::PyGILStateEnsure gil;
    Py_DECREF(m_func); 
    m_func = func;
    Py_INCREF(m_func); 
  }
 
  // getter
  void getObj (void *& obj)
  {
    obj = m_obj;
  }
 
  PyObject * toPyObj (void * obj)
  {
    RootUtils::PyGILStateEnsure gil;
    m_obj = obj;
    return PyObject_CallObject(m_func,NULL);
  }

  static PyReverseProxy * getProxy(const std::string & key)
  {
    RootUtils::PyGILStateEnsure gil;
    PyReverseProxy* & prox = s_proxyMap[key];
    if (!prox) {
      prox = new PyReverseProxy(key);
    }
    return prox;
  }
 
private:
  PyReverseProxy() 
  {}

  PyReverseProxy(const std::string &key)
    : m_key(key), m_obj(0), m_func(0)
  {}

  std::string m_key;

  void * m_obj;

  PyObject * m_func;

  static std::unordered_map<std::string, PyReverseProxy *> s_proxyMap ATLAS_THREAD_SAFE;
};
 
#endif