X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;ds=sidebyside;f=libs%2Fardour%2Faudioanalyser.cc;h=f90edd380a094d2840daa7af61c87f814180fcc0;hb=1389fa9e2246a1b7faf193932a7c3ca4ba559c64;hp=9da404e5bf5e7c3df83eec5b3be7349fc4418c00;hpb=80c3677c837cc9eb432df3d65a34aba543fa7258;p=ardour.git diff --git a/libs/ardour/audioanalyser.cc b/libs/ardour/audioanalyser.cc index 9da404e5bf..f90edd380a 100644 --- a/libs/ardour/audioanalyser.cc +++ b/libs/ardour/audioanalyser.cc @@ -1,15 +1,39 @@ -#include +/* + 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 + +#include + +#include "pbd/gstdio_compat.h" #include #include -#include // for g_remove() -#include +#include "pbd/error.h" +#include "pbd/failed_constructor.h" + +#include "ardour/audioanalyser.h" +#include "ardour/readable.h" -#include -#include -#include +#include -#include "i18n.h" +#include "pbd/i18n.h" using namespace std; using namespace Vamp; @@ -21,11 +45,11 @@ AudioAnalyser::AudioAnalyser (float sr, AnalysisPluginKey key) , 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 () @@ -40,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; @@ -74,44 +98,30 @@ AudioAnalyser::reset () plugin->reset (); } } - + int AudioAnalyser::analyse (const string& path, Readable* src, uint32_t channel) { - ofstream ofile; + 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 }; - string tmp_path; - - if (!path.empty()) { - - /* 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; - } - } 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) { goto out; } @@ -119,16 +129,16 @@ AudioAnalyser::analyse (const string& path, Readable* src, uint32_t channel) /* 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)); } - + features = plugin->process (bufs, RealTime::fromSeconds ((double) pos / sample_rate)); - if (use_features (features, (path.empty() ? 0 : &ofile))) { + if (use_features (features, (path.empty() ? 0 : &outss))) { goto out; } - pos += stepsize; + pos += min (stepsize, to_read); if (pos >= len) { done = true; @@ -139,26 +149,18 @@ AudioAnalyser::analyse (const string& path, Readable* src, uint32_t channel) features = plugin->getRemainingFeatures (); - if (use_features (features, (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 (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()); + if (!ret) { + g_file_set_contents (path.c_str(), outss.str().c_str(), -1, NULL); } - if (data) { - delete [] data; - } + delete [] data; return ret; }