Make stem export export from right before any processors.

[ardour.git] / libs / ardour / audioanalyser.cc

diff --git a/libs/ardour/audioanalyser.cc b/libs/ardour/audioanalyser.cc
index 4cc99a5d5e2da2a6cac03bd350ad3f8d1709f912..d477026a24df8998be326335c5d710176111ddac 100644 (file)
--- a/libs/ardour/audioanalyser.cc
+++ b/libs/ardour/audioanalyser.cc
@@ -1,13 +1,18 @@
-#include <vamp-sdk/hostext/PluginLoader.h>
+#include <cstring>
+
+#include "vamp-hostsdk/PluginLoader.h"
+
 #include <glibmm/miscutils.h>
 #include <glibmm/fileutils.h>
 #include <glib/gstdio.h> // for g_remove()
 
-#include <pbd/error.h>
+#include "pbd/error.h"
+#include "pbd/failed_constructor.h"
+
+#include "ardour/audioanalyser.h"
+#include "ardour/readable.h"
 
-#include <ardour/audioanalyser.h>
-#include <ardour/readable.h>
-#include <ardour/readable.h>
+#include <cstring>
 
 #include "i18n.h"
 
@@ -18,13 +23,19 @@ using namespace ARDOUR;
 
 AudioAnalyser::AudioAnalyser (float sr, AnalysisPluginKey key)
        : sample_rate (sr)
-       , plugin (0)
        , plugin_key (key)
 {
+       /* create VAMP plugin and initialize */
+
+       if (initialize_plugin (plugin_key, sample_rate)) {
+               error << string_compose (_("cannot load VAMP plugin \"%1\""), key) << endmsg;
+               throw failed_constructor();
+       }
 }
 
 AudioAnalyser::~AudioAnalyser ()
 {
+       delete plugin;
 }
 
 int
@@ -39,7 +50,7 @@ AudioAnalyser::initialize_plugin (AnalysisPluginKey key, float sr)
        if (!plugin) {
                error << string_compose (_("VAMP Plugin \"%1\" could not be loaded"), key) << endmsg;
                return -1;
-       } 
+       }
 
        /* we asked for the buffering adapter, so set the blocksize to
           something that makes for efficient disk i/o
@@ -68,64 +79,62 @@ AudioAnalyser::reset ()
                plugin->reset ();
        }
 }
-       
+
 int
 AudioAnalyser::analyse (const string& path, Readable* src, uint32_t channel)
 {
        ofstream ofile;
-       Plugin::FeatureSet onsets;
+       Plugin::FeatureSet features;
        int ret = -1;
        bool done = false;
        Sample* data = 0;
-       nframes64_t len = src->readable_length();
-       nframes64_t pos = 0;
+       framecnt_t len = src->readable_length();
+       framepos_t pos = 0;
        float* bufs[1] = { 0 };
+       string tmp_path;
 
        if (!path.empty()) {
-               ofile.open (path.c_str());
+
+               /* store data in tmp file, not the real one */
+
+               tmp_path = path;
+               tmp_path += ".tmp";
+
+               ofile.open (tmp_path.c_str());
                if (!ofile) {
                        goto out;
                }
        }
 
-       /* create VAMP percussion onset plugin and initialize */
-       
-       if (plugin == 0) {
-               if (initialize_plugin (plugin_key, sample_rate)) {
-                       goto out;
-               } 
-       } 
-
        data = new Sample[bufsize];
        bufs[0] = data;
 
        while (!done) {
 
-               nframes64_t to_read;
-               
+               framecnt_t to_read;
+
                /* read from source */
 
-               to_read = min ((len - pos), bufsize);
-               
+               to_read = min ((len - pos), (framecnt_t) bufsize);
+
                if (src->read (data, pos, to_read, channel) != to_read) {
-                       cerr << "bad read\n";
                        goto out;
                }
 
                /* zero fill buffer if necessary */
 
                if (to_read != bufsize) {
-                       memset (data + to_read, 0, (bufsize - to_read));
+                       memset (data + to_read, 0, (bufsize - to_read) * sizeof (Sample));
                }
-               
-               onsets = plugin->process (bufs, RealTime::fromSeconds ((double) pos / sample_rate));
 
-               if (use_features (onsets, (path.empty() ? &ofile : 0))) {
+               features = plugin->process (bufs, RealTime::fromSeconds ((double) pos / sample_rate));
+
+               if (use_features (features, (path.empty() ? 0 : &ofile))) {
                        goto out;
                }
 
-               pos += stepsize;
-               
+               pos += min (stepsize, to_read);
+
                if (pos >= len) {
                        done = true;
                }
@@ -133,9 +142,9 @@ AudioAnalyser::analyse (const string& path, Readable* src, uint32_t channel)
 
        /* finish up VAMP plugin */
 
-       onsets = plugin->getRemainingFeatures ();
+       features = plugin->getRemainingFeatures ();
 
-       if (use_features (onsets, (path.empty() ? &ofile : 0))) {
+       if (use_features (features, (path.empty() ? &ofile : 0))) {
                goto out;
        }
 
@@ -146,12 +155,14 @@ AudioAnalyser::analyse (const string& path, Readable* src, uint32_t channel)
        ofile.close ();
 
        if (ret) {
-               g_remove (path.c_str());
-       }
-       if (data) {
-               delete data;
+               g_remove (tmp_path.c_str());
+       } else if (!path.empty()) {
+               /* move the data file to the requested path */
+               g_rename (tmp_path.c_str(), path.c_str());
        }
 
+       delete [] data;
+
        return ret;
 }