我希望这是一个简单的SWIG问题。我正在使用Google OR-Tools优化库。它是一个包含在SWIG中的C ++库(我对此知之甚少)。我很难让Python回调函数工作。有一个C ++函数
DecisionBuilder* MakePhase(const std::vector<IntVar*>& vars,
IndexEvaluator1* var_evaluator,
IntValueStrategy val_str);
与
一起typedef ResultCallback1<int64, int64> IndexEvaluator1;
和相关的SWIG(我相信)是
DecisionBuilder* VarEvalValStrPhase(
const std::vector<IntVar*>& vars,
ResultCallback1<int64, int64>* var_evaluator,
operations_research::Solver::IntValueStrategy val_str) {
return self->MakePhase(vars, var_evaluator, val_str);
}
在另一个SWIG文件中我们有
%{
static int64 PyCallback1Int64Int64(PyObject* pyfunc, int64 i) {
// () needed to force creation of one-element tuple
PyObject* pyresult = PyEval_CallFunction(pyfunc, "(l)", static_cast<long>(i));
int64 result = 0;
if (!pyresult) {
PyErr_SetString(PyExc_RuntimeError,
"ResultCallback1<int64, int64> invocation failed.");
} else {
result = PyInt_AsLong(pyresult);
Py_DECREF(pyresult);
}
return result;
}
%}
%typemap(in) ResultCallback1<int64, int64>* {
if (!PyCallable_Check($input)) {
PyErr_SetString(PyExc_TypeError, "Need a callable object!");
SWIG_fail;
}
$1 = NewPermanentCallback(&PyCallback1Int64Int64, $input);
}
在我的Python模块中,我定义了一个函数Run1,如下所示(这里是我认为应该有一些类型转换的部分,但我认为这不是Python的方式):
def Run1(index1):
return index1
并设置
selector_callback = Run1
solver = pywrapcp.Solver("graph-coloring")
最后,我打电话给
solver.Phase(nodes,
selector_callback,
solver.INT_VALUE_DEFAULT)
在这里,事情就是kablooie,我总是得到以下错误:
File "C:\dev\Python27\lib\site-packages\ortools-1.3853-py2.7-win-amd64.egg\ortools\constraint_solver\pywrapcp.py", line 457, in Phase
def Phase(self, *args): return _pywrapcp.Solver_Phase(self, *args)
NotImplementedError: Wrong number or type of arguments for overloaded function 'Solver_Phase'.
Possible C/C++ prototypes are:
operations_research::Solver::MakePhase(std::vector< operations_research::IntVar *,std::allocator< operations_research::IntVar * > > const &,operations_research::Solver::IntVarStrategy,operations_research::Solver::IntValueStrategy)
operations_research::Solver::MakePhase(std::vector< operations_research::IntervalVar *,std::allocator< operations_research::IntervalVar * > > const &,operations_research::Solver::IntervalStrategy)
operations_research::Solver::MakePhase(std::vector< operations_research::SequenceVar *,std::allocator< operations_research::SequenceVar * > > const &,operations_research::Solver::SequenceStrategy)
难点在于第二个参数中的回调函数;如果我使用其中一个内置值而不是回调,则操作成功。但是,我确实需要在那里拥有自己的功能。
我没有在我的模块中导入任何SWIG文件。我需要这样做吗?
答案 0 :(得分:1)
过了几天,我找到了答案。如果我打电话
solver.VarEvalValStrPhase(nodes,
selector_callback,
solver.INT_VALUE_DEFAULT)
代替整本手册中引用的标准函数名Phase
,它可以正常工作。如果我要使用另一个参数组合,我必须使用我认为的另一个函数名称。在这种情况下,似乎重载失败。这很好,但开发人员的警告会很好。