Method push

src/python/pystreamingalgorithm.cpp:177–236

PyObject* PyStreamingAlgorithm::push(PyStreamingAlgorithm* self, PyObject* args) {
  vector<PyObject*> argsV = unpack(args);
  if (argsV.size() != 2) {
    PyErr_SetString(PyExc_ValueError, "Algorithm.push requires 2 arguments");
    return NULL;
  }

  if (!PyString_Check(argsV[0])) {
    PyErr_SetString(PyExc_ValueError, "Algorithm.push requires a string as the first argument");
    return NULL;
  }

  // name of the source to push onto
  string name = PyString_AS_STRING(argsV[0]);

  // check if source exists
  if (!contains(self->algo->outputs(), name)) {
    ostringstream msg;
    msg << self->algo->name() << " does not have an input named " << name;
    PyErr_SetString(PyExc_ValueError, msg.str().c_str());
    return NULL;
  }

  SourceBase& src = self->algo->output(name);

  // edt of given data
  Edt tp = typeInfoToEdt(src.typeInfo());

  #define PUSH_COPY(tname, type) { \
    type* val = (type*)tname::fromPythonCopy(argsV[1]); \
    src.push(*val); \
    delete val; \
    break; \
  }

  switch(tp) {
    case INTEGER:             PUSH_COPY(Integer, int);
    case REAL:                PUSH_COPY(PyReal, Real);
    case STRING:              PUSH_COPY(String, string);
    case STEREOSAMPLE:        PUSH_COPY(PyStereoSample, StereoSample);
    case VECTOR_STRING:       PUSH_COPY(VectorString, vector<string>);
    case VECTOR_STEREOSAMPLE: PUSH_COPY(VectorStereoSample, vector<StereoSample>);

    case VECTOR_REAL: {
      RogueVector<Real>* val = (RogueVector<Real>*)VectorReal::fromPythonRef(argsV[1]);
      src.push(*(vector<Real>*)val);
      delete val;
      break;
    }

    default:
      ostringstream msg;
      msg << "given value type not supported: " << edtToString(tp);
      PyErr_SetString(PyExc_ValueError, msg.str().c_str()); return NULL;
  }

  #undef PUSH_COPY