// Copyright (c) 2014 Josh Blum

// SPDX-License-Identifier: BSL-1.0

#include "PothosModule.hpp"

#include <cassert>

static PyTypeObject ProxyCallType = {

PyObject_HEAD_INIT(NULL)

};

static void ProxyCall_dealloc(ProxyCallObject *self)

{

delete self->proxy;

delete self->name;

Py_TYPE(self)->tp_free((PyObject*)self);

}

static int ProxyCall_init(ProxyCallObject *self, PyObject *args, PyObject *)

{

self->proxy = new PyObjectRef(PyTuple_GetItem(args, 0), REF_BORROWED);

self->name = new PyObjectRef(PyTuple_GetItem(args, 1), REF_BORROWED);

return 0;

}

PyObject* ProxyCall_call(ProxyCallObject *self, PyObject *args, PyObject *)

{

try

{

//extract string name

const auto name = PyObjectToProxy(self->name->obj).convert<std::string>();

//create call args

Pothos::ProxyVector proxyArgs;

for (int i = 0; i < PyTuple_Size(args); i++)

{

proxyArgs.push_back(PyObjectToProxy(PyTuple_GetItem(args, i)));

}

//make proxy call

auto handle = ((ProxyObject *)self->proxy->obj)->proxy->getHandle();

Pothos::Proxy proxy;

{

PyThreadStateLock lock; //proxy call could be potentially blocking

proxy = handle->call(name, proxyArgs.data(), proxyArgs.size());

}

//convert the result into a pyobject

return ProxyToPyObject(proxyEnvTranslate(proxy, getPythonProxyEnv()));

}

catch (const Pothos::Exception &ex)

{

PyErr_SetString(PyExc_RuntimeError, ex.displayText().c_str());

return nullptr;

}

}

PyObject *makeProxyCallObject(PyObject *args)

{

return PyObject_CallObject((PyObject *)&ProxyCallType, args);

}

void registerProxyCallType(PyObject *m)

{

ProxyCallType.tp_new = PyType_GenericNew;

ProxyCallType.tp_name = "PothosProxyCall";

ProxyCallType.tp_basicsize = sizeof(ProxyCallObject);

ProxyCallType.tp_dealloc = (destructor)ProxyCall_dealloc;

ProxyCallType.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE;

ProxyCallType.tp_doc = "Pothos Proxy Call binding";

ProxyCallType.tp_init = (initproc)ProxyCall_init;

ProxyCallType.tp_call = (ternaryfunc)ProxyCall_call;

if (PyType_Ready(&ProxyCallType) < 0) return;

Py_INCREF(&ProxyCallType);

PyModule_AddObject(m, "ProxyCall", (PyObject *)&ProxyCallType);

}