Tempo ramps - use consistent naming for _locked methods, use enum_2_string for TempoS...

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

diff --git a/libs/ardour/audioanalyser.cc b/libs/ardour/audioanalyser.cc
index 4cc99a5d5e2da2a6cac03bd350ad3f8d1709f912..7013d3446f515447d703ef28f84bf234856a2d5f 100644 (file)
--- a/libs/ardour/audioanalyser.cc
+++ b/libs/ardour/audioanalyser.cc
@@ -1,13 +1,37 @@
-#include <vamp-sdk/hostext/PluginLoader.h>
+/*
+    Copyright (C) 2012 Paul Davis
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, write to the Free Software
+    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+*/
+
+#include <cstring>
+
+#include <vamp-hostsdk/PluginLoader.h>
+
+#include "pbd/gstdio_compat.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/readable.h>
+#include "ardour/audioanalyser.h"
+#include "ardour/readable.h"
+
+#include <cstring>
 
 #include "i18n.h"
 
@@ -18,13 +42,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
@@ -34,19 +64,19 @@ AudioAnalyser::initialize_plugin (AnalysisPluginKey key, float sr)
 
        PluginLoader* loader (PluginLoader::getInstance());
 
-       plugin = loader->loadPlugin (key, sr, PluginLoader::ADAPT_ALL);
+       plugin = loader->loadPlugin (key, sr, PluginLoader::ADAPT_ALL_SAFE);
 
        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
        */
 
-       bufsize = 65536;
-       stepsize = bufsize;
+       bufsize = 1024;
+       stepsize = 512;
 
        if (plugin->getMinChannelCount() > 1) {
                delete plugin;
@@ -68,64 +98,48 @@ AudioAnalyser::reset ()
                plugin->reset ();
        }
 }
-       
+
 int
 AudioAnalyser::analyse (const string& path, Readable* src, uint32_t channel)
 {
-       ofstream ofile;
-       Plugin::FeatureSet onsets;
+       stringstream outss;
+       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 };
 
-       if (!path.empty()) {
-               ofile.open (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 : &outss))) {
                        goto out;
                }
 
-               pos += stepsize;
-               
+               pos += min (stepsize, to_read);
+
                if (pos >= len) {
                        done = true;
                }
@@ -133,25 +147,21 @@ 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() ? 0 : &outss))) {
                goto out;
        }
 
        ret = 0;
 
   out:
-       /* works even if it has not been opened */
-       ofile.close ();
-
-       if (ret) {
-               g_remove (path.c_str());
-       }
-       if (data) {
-               delete data;
+       if (!ret) {
+               g_file_set_contents (path.c_str(), outss.str().c_str(), -1, NULL);
        }
 
+       delete [] data;
+
        return ret;
 }