Logo Search packages:      
Sourcecode: ardour version File versions  Download package

audioanalyser.cc

#include <cstring>
#include <vamp-sdk/hostext/PluginLoader.h>
#include <glibmm/miscutils.h>
#include <glibmm/fileutils.h>
#include <glib/gstdio.h> // for g_remove()

#include <pbd/error.h>

#include <ardour/audioanalyser.h>
#include <ardour/readable.h>
#include <ardour/readable.h>

#include <cstring>

#include "i18n.h"

using namespace std;
using namespace Vamp;
using namespace PBD;
using namespace ARDOUR;

AudioAnalyser::AudioAnalyser (float sr, AnalysisPluginKey key)
      : sample_rate (sr)
      , 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
AudioAnalyser::initialize_plugin (AnalysisPluginKey key, float sr)
{
      using namespace Vamp::HostExt;

      PluginLoader* loader (PluginLoader::getInstance());

      plugin = loader->loadPlugin (key, sr, PluginLoader::ADAPT_ALL);

      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;

      if (plugin->getMinChannelCount() > 1) {
            delete plugin;
            return -1;
      }

      if (!plugin->initialise (1, stepsize, bufsize)) {
            delete plugin;
            return -1;
      }

      return 0;
}

void
AudioAnalyser::reset ()
{
      if (plugin) {
            plugin->reset ();
      }
}
      
int
AudioAnalyser::analyse (const string& path, Readable* src, uint32_t channel)
{
      ofstream ofile;
      Plugin::FeatureSet features;
      int ret = -1;
      bool done = false;
      Sample* data = 0;
      nframes64_t len = src->readable_length();
      nframes64_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;
            
            /* read from source */

            to_read = min ((len - pos), bufsize);
            
            if (src->read (data, pos, to_read, channel) != to_read) {
                  goto out;
            }

            /* zero fill buffer if necessary */

            if (to_read != bufsize) {
                  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))) {
                  goto out;
            }

            pos += min (stepsize, to_read);

            if (pos >= len) {
                  done = true;
            }
      }

      /* finish up VAMP plugin */

      features = plugin->getRemainingFeatures ();

      if (use_features (features, (path.empty() ? &ofile : 0))) {
            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 (data) {
            delete [] data;
      }

      return ret;
}


Generated by  Doxygen 1.6.0   Back to index