Function get_dag_value

graphlearn/python/c/py_wrapper.h:639–669

}

PyObject* get_dag_value(GetDagValuesResponse* res,
                        int32_t node_id,
                        const std::string& key) {
  auto t = res->GetValue(node_id, key);
  if (t == nullptr) {
    Py_RETURN_NONE;
  }
  int32_t size = t->Size();
  npy_intp shape[1];
  shape[0] = size;

  PyObject* obj = nullptr;

  switch (t->DType()) {
  case kInt32:
    obj = PyArrayFromIntVector(t->GetInt32(), shape);
    break;
  case kInt64:
    obj = PyArrayFromInt64Vector(t->GetInt64(), shape);
    break;
  case kFloat:
    obj = PyArrayFromFloatVector(t->GetFloat(), shape);
    break;
  case kString:
    obj = PyArrayFromStringVector(t->GetString(), shape);
    break;
  default:
    break;
  }
  return PyArray_Return(reinterpret_cast<PyArrayObject*>(obj));
}

#endif  // GRAPHLEARN_PYTHON_C_PY_WRAPPER_H_