PyUtil.h

Go to the documentation of this file.

// This file is part of DynExp.
 
#pragma once
 
#include "stdafx.h"
 
namespace py = pybind11;
 
namespace Util
{
    static constexpr auto PyTab = "    ";
 
    class PyStdoutLoggerWrapper
    {
    public:
        void write(std::string Str);
 
        void flush() {}
    };
 
    class PyGilReleasedInterpreter
    {
    public:
        PyGilReleasedInterpreter();
 
        void PrintDebugInfo();              
 
    private:
        // Do not change order. GILRelease needs to be destroyed first upon destruction.
        PyStdoutLoggerWrapper Logger;           
        py::scoped_interpreter Interpreter;     
        py::module_ Module_sys;             
        py::gil_scoped_release GILRelease;      
    };
 
    template <typename T, std::enable_if_t<
        std::is_base_of_v<py::object, T>, int> = 0
    >
    class PyObject
    {
    public:
        using ObjectType = T;
 
        PyObject() { Reset(); }
 
        PyObject(const T& Object) : Object(std::make_unique<T>(Object)) {}
 
        ~PyObject() { Remove(); }
 
        PyObject& operator=(const T& Object)
        {
            this->Object = std::make_unique<T>(Object);
            return *this;
        }
 
        auto& Get() const noexcept { return *Object; }
 
        auto& Get() noexcept { return *Object; }
 
        void Reset()
        {
            py::gil_scoped_acquire acquire;
            Object = std::make_unique<T>();
        }
 
        template <typename... ArgTs>
        py::object operator()(ArgTs&& ...Args) const
        {
            if (Object && !Object->is_none())
                return (*Object)(std::forward<ArgTs>(Args)...);
            else
                return py::none();
        }
 
    private:
        void Remove()
        {
            py::gil_scoped_acquire acquire;
            Object.reset();
        }
 
        std::unique_ptr<T> Object;
    };
}