在OR工具中使用Python回调

Question

我希望这是一个简单的饮酒问题。我正在使用谷歌或工具优化库。它是一个用SWIG包装的C++库(我对它知之甚少)。我很难让Python回调函数工作起来。有一个C++函数

DecisionBuilder* MakePhase(const std::vector<IntVar*>& vars,
                           IndexEvaluator1* var_evaluator,
                           IntValueStrategy val_str);

随同

typedef ResultCallback1<int64, int64> IndexEvaluator1;

而相关的酒杯(我相信)是

  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);
  }

在另一个文件中我们有

%{
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)

唉，这是事情发展的方向，我总是会犯以下错误：

  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)

困难在于第二个参数中的回调函数；如果我使用一个内置值而不是回调，则操作是成功的。但是，我需要在里面有我自己的功能。

我不会在我的模块中导入任何文件。我需要这么做吗？

Answer 1

几天后，我找到了答案。如果我打电话

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

它不是在手册中引用的标准函数名Phase，而是起作用的。如果我要使用另一个参数组合，我必须使用另一个函数名，我相信。在这种情况下，似乎重载失败了。这很好，但是开发者的警告是很好的。