Changes in python/lib/gen_code.py [1164fd85:bf54364]

File:

• python/lib/gen_code.py (modified) (3 diffs)

python/lib/gen_code.py

@@ -75 +75 @@
         'onset': '1',
         'pitch': '1',
+        'notes': '3',
         'wavetable': 'self->hop_size',
         'sampler': 'self->hop_size',
@@ -82 +83 @@
         'filterbank': 'self->n_filters',
         'tss': 'self->hop_size',
+        }
+
+objinputsize = {
+        'mfcc': 'self->buf_size / 2 + 1',
+        'notes': 'self->hop_size',
+        'onset': 'self->hop_size',
+        'pitch': 'self->hop_size',
+        'sampler': 'self->hop_size',
+        'specdesc': 'self->buf_size / 2 + 1',
+        'tempo': 'self->hop_size',
+        'wavetable': 'self->hop_size',
         }
 
@@ -377 +389 @@
         return NULL;
     }}""".format(input_param, pytoaubio = pytoaubio_fn[input_param['type']])
-        out += """
+        if self.shortname in objinputsize:
+            out += """
+
+    if (self->{0[name]}.length != {expected_size}) {{
+        PyErr_Format (PyExc_ValueError,
+            "input size of {shortname} should be %d, not %d",
+            {expected_size}, self->{0[name]}.length);
+        return NULL;
+    }}""".format(input_param, expected_size = objinputsize[self.shortname], **self.__dict__)
+        else:
+            out += """
 
     // TODO: check input sizes"""