comparison PyPluginObject.cpp @ 41:55fcd0e3e513

Fix handling of 2D NumPy arrays

         if (conv.error) {
             PyErr_SetString(PyExc_TypeError, conv.getError().str().c_str());
             return data;
         }
 
+        if ((int)data.size() != channels) {
+//            cerr << "Wrong number of channels: got " << data.size() << ", expected " << channels << endl;
+            PyErr_SetString(PyExc_TypeError, "Wrong number of channels");
+            return vector<vector<float> >();
+        }
+        
     } else {
         
         if (!PyList_Check(pyBuffer)) {
             PyErr_SetString(PyExc_TypeError, "List of NumPy Array required for process input.");
             return data;
         }
 
         if (PyList_GET_SIZE(pyBuffer) != channels) {
-            cerr << "Wrong number of channels: got " << PyList_GET_SIZE(pyBuffer) << ", expected " << channels << endl;
+//            cerr << "Wrong number of channels: got " << PyList_GET_SIZE(pyBuffer) << ", expected " << channels << endl;
             PyErr_SetString(PyExc_TypeError, "Wrong number of channels");
             return data;
         }
 
         for (int c = 0; c < channels; ++c) {
             PyObject *cbuf = PyList_GET_ITEM(pyBuffer, c);
             data.push_back(conv.PyValue_To_FloatVector(cbuf));
         }
-    
-        for (int c = 0; c < channels; ++c) {
-            if ((int)data[c].size() != blockSize) {
-                cerr << "Wrong number of samples on channel " << c << ": expected " << blockSize << " (plugin's block size), got " << data[c].size() << endl;
-                PyErr_SetString(PyExc_TypeError, "Wrong number of samples");
-                return vector<vector<float> >();
-            }
+    }
+    
+    for (int c = 0; c < channels; ++c) {
+        if ((int)data[c].size() != blockSize) {
+//            cerr << "Wrong number of samples on channel " << c << ": expected " << blockSize << " (plugin's block size), got " << data[c].size() << endl;
+            PyErr_SetString(PyExc_TypeError, "Wrong number of samples");
+            return vector<vector<float> >();
         }
     }
     
     return data;
 }