在OR Tools中通过SWIG使用Python回调

时间:2015-03-17 03:20:09

标签: python c++ swig

我希望这是一个简单的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文件。我需要这样做吗?

1 个答案:

答案 0 :(得分:1)

过了几天,我找到了答案。如果我打电话

solver.VarEvalValStrPhase(nodes,
             selector_callback,
             solver.INT_VALUE_DEFAULT)

代替整本手册中引用的标准函数名Phase,它可以正常工作。如果我要使用另一个参数组合,我必须使用我认为的另一个函数名称。在这种情况下,似乎重载失败。这很好,但开发人员的警告会很好。