comparison runner/FeatureExtractionManager.cpp @ 8:f98b58fb294f

* Set block & step sizes if not properly specified

 
 #include <vamp-hostsdk/PluginChannelAdapter.h>
 #include <vamp-hostsdk/PluginBufferingAdapter.h>
 #include <vamp-hostsdk/PluginInputDomainAdapter.h>
 #include <vamp-hostsdk/PluginSummarisingAdapter.h>
+#include <vamp-hostsdk/PluginWrapper.h>
 #include <vamp-hostsdk/PluginLoader.h>
 
 #include <iostream>
 
 using namespace std;

 using Vamp::HostExt::PluginLoader;
 using Vamp::HostExt::PluginChannelAdapter;
 using Vamp::HostExt::PluginBufferingAdapter;
 using Vamp::HostExt::PluginInputDomainAdapter;
 using Vamp::HostExt::PluginSummarisingAdapter;
+using Vamp::HostExt::PluginWrapper;
 
 #include "data/fileio/FileSource.h"
 #include "data/fileio/AudioFileReader.h"
 #include "data/fileio/AudioFileReaderFactory.h"
 #include "data/fileio/PlaylistFileReader.h"

             }
 
             cerr << "NOTE: Loaded and initialised plugin " << plugin
                  << " for transform \""
                  << transform.getIdentifier().toStdString() << "\"" << endl;
+
+        } else {
+
+            if (transform.getStepSize() == 0 || transform.getBlockSize() == 0) {
+
+                PluginWrapper *pw = dynamic_cast<PluginWrapper *>(plugin);
+                if (pw) {
+                    PluginBufferingAdapter *pba =
+                        pw->getWrapper<PluginBufferingAdapter>();
+                    if (pba) {
+                        size_t actualStepSize = 0;
+                        size_t actualBlockSize = 0;
+                        pba->getActualStepAndBlockSizes(actualStepSize,
+                                                        actualBlockSize);
+                        if (transform.getStepSize() == 0) {
+                            transform.setStepSize(actualStepSize);
+                        }
+                        if (transform.getBlockSize() == 0) {
+                            transform.setBlockSize(actualBlockSize);
+                        }
+                    }
+                }
+            }
         }
 
         if (transform.getOutput() == "") {
             transform.setOutput
                 (plugin->getOutputDescriptors()[0].identifier.c_str());