python/lib/gen_code.py: switch to using PyObjects instead of fvec, cvec, fmat

Paul Brossier committed 10 years ago

Commit 21e84081 parent b055b4e2 files · +42 −26Browse files

Showing the changes to python (2 of 2 files). Show all

Unified · Split

2 files

python/ext/py-filter.c +17 −12
python/lib/gen_code.py +25 −14

   aubio_filter_t * o;
   uint_t order;
   fvec_t vec;
-  fvec_t *out;
+  PyObject *out;
+  fvec_t c_out;
 } Py_filter;
 static char Py_filter_doc[] = "filter object";
   if (self->o == NULL) {
     return -1;
   }
-  self->out = new_fvec(Py_default_vector_length);
+  self->out = NULL;
   return 0;
 }
 static void
 Py_filter_del (Py_filter * self)
 {
-  del_fvec(self->out);
+  Py_XDECREF(self->out);
   del_aubio_filter (self->o);
   Py_TYPE(self)->tp_free ((PyObject *) self);
 }
 static PyObject *
 Py_filter_do(Py_filter * self, PyObject * args)
 {
   PyObject *input;
     return NULL;
   }
-  // reallocate the output if needed
+  // initialize output now
-  if (self->vec.length != self->out->length) {
+  if (self->out == NULL) {
-    del_fvec(self->out);
+    self->out = new_py_fvec(self->vec.length);
-    self->out = new_fvec(self->vec.length);
+  }
+  Py_INCREF(self->out);
+  if (!PyAubio_ArrayToCFvec(self->out, &(self->c_out)) ) {
+    return NULL;
   }
   // compute the function
-  aubio_filter_do_outplace (self->o, &(self->vec), self->out);
+  aubio_filter_do_outplace (self->o, &(self->vec), &(self->c_out));
-  return PyAubio_CFvecToArray(self->out);
+  return self->out;
 }
 static PyObject *
 Py_filter_set_c_weighting (Py_filter * self, PyObject *args)
 {
   uint_t err = 0;
   Py_RETURN_NONE;
 }
 static PyObject *
 Py_filter_set_a_weighting (Py_filter * self, PyObject *args)
 {
   uint_t err = 0;

python/lib/gen_code.py+25−14

         }
 newfromtype_fn = {
-        'fvec_t*': 'new_fvec',
+        'fvec_t*': 'new_py_fvec',
-        'fmat_t*': 'new_fmat',
+        'fmat_t*': 'new_py_fmat',
-        'cvec_t*': 'new_cvec',
+        'cvec_t*': 'new_py_cvec',
         }
 delfromtype_fn = {
-        'fvec_t*': 'del_fvec',
+        'fvec_t*': 'Py_DECREF',
-        'fmat_t*': 'del_fmat',
+        'fmat_t*': 'Py_DECREF',
-        'cvec_t*': 'del_cvec',
+        'cvec_t*': 'Py_DECREF',
         }
 param_init = {
         self.outputs = get_params_types_names(self.do_proto)[2:]
         self.do_inputs = [get_params_types_names(self.do_proto)[1]]
         self.do_outputs = get_params_types_names(self.do_proto)[2:]
-        self.outputs_flat = get_output_params(self.do_proto)
+        struct_output_str = ["PyObject *{0[name]}; {1} c_{0[name]}".format(i, i['type'][:-1]) for i in self.do_outputs]
-        self.output_results = "; ".join(self.outputs_flat)
+        self.struct_outputs = ";\n    ".join(struct_output_str)
         #print ("input_params: ", map(split_type, get_input_params(self.do_proto)))
         #print ("output_params", map(split_type, get_output_params(self.do_proto)))
     // do input vectors
     {do_inputs_list};
     // output results
-    {output_results};
+    {struct_outputs};
 }} Py_{shortname};
 """
         # fmat_t* / fvec_t* / cvec_t* inputs -> full fvec_t /.. struct in Py_{shortname}
     if (!PyArg_ParseTuple (args, "{pyparamtypes}", {refs})) {{
         return NULL;
     }}""".format(refs = refs, pyparamtypes = pyparamtypes, **self.__dict__)
-        for p in input_params:
+        for input_param in input_params:
             out += """
     if (!{pytoaubio}(py_{0[name]}, &(self->{0[name]}))) {{
         return NULL;
     }}""".format(input_param, pytoaubio = pytoaubio_fn[input_param['type']])
+        out += """
+    // TODO: check input sizes"""
+        for output_param in output_params:
+            out += """
+    Py_INCREF(self->{0[name]});
+    if (!{pytoaubio}(self->{0[name]}, &(self->c_{0[name]}))) {{
+        return NULL;
+    }}""".format(output_param, pytoaubio = pytoaubio_fn[output_param['type']])
         do_fn = get_name(self.do_proto)
         inputs = ", ".join(['&(self->'+p['name']+')' for p in input_params])
+        c_outputs = ", ".join(["&(self->c_%s)" % p['name'] for p in self.do_outputs])
         outputs = ", ".join(["self->%s" % p['name'] for p in self.do_outputs])
         out += """
-    {do_fn}(self->o, {inputs}, {outputs});
+    {do_fn}(self->o, {inputs}, {c_outputs});
-    return (PyObject *) {aubiotonumpy} ({outputs});
+    return {outputs};
 }}
 """.format(
         do_fn = do_fn,
-        aubiotonumpy = pyfromaubio_fn[output['type']],
+        inputs = inputs, outputs = outputs, c_outputs = c_outputs,
-        inputs = inputs, outputs = outputs,
         )
         return out

python/lib/gen_code.py: switch to using PyObjects instead of fvec, cvec, fmat

Keyboard shortcuts

General

Issues and pull requests

Issues, pull requests and Inbox

Repository code