view transform/PluginTransform.cpp @ 178:3e5a32a2acf4

* Ensure transformed version of a model is the same duration as the original
  model (don't end it when the input data runs out, as was previously done,
  because this will result in too short a model when latency compensation is
  used)

/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */

/*
    Sonic Visualiser
    An audio file viewer and annotation editor.
    Centre for Digital Music, Queen Mary, University of London.
    This file copyright 2006 QMUL.
   
    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.  See the file
    COPYING included with this distribution for more information.
*/

#include "PluginTransform.h"

#include "vamp-sdk/PluginHostAdapter.h"

PluginTransform::PluginTransform(Model *inputModel,
				 const ExecutionContext &context) :
    Transform(inputModel),
    m_context(context)
{
}

PluginTransform::ExecutionContext::ExecutionContext(int _c, size_t _bs) :
    channel(_c),
    domain(Vamp::Plugin::TimeDomain),
    stepSize(_bs ? _bs : 1024),
    blockSize(_bs ? _bs : 1024),
    windowType(HanningWindow)
{
}

PluginTransform::ExecutionContext::ExecutionContext(int _c, size_t _ss,
                                                    size_t _bs, WindowType _wt) :
    channel(_c),
    domain(Vamp::Plugin::FrequencyDomain),
    stepSize(_ss ? _ss : (_bs ? _bs / 2 : 512)),
    blockSize(_bs ? _bs : 1024),
    windowType(_wt)
{
}

PluginTransform::ExecutionContext::ExecutionContext(int _c,
                                                    const Vamp::PluginBase *_plugin) :
    channel(_c),
    domain(Vamp::Plugin::TimeDomain),
    stepSize(0),
    blockSize(0),
    windowType(HanningWindow)
{
    makeConsistentWithPlugin(_plugin);
}

bool
PluginTransform::ExecutionContext::operator==(const ExecutionContext &c)
{
    return (c.channel == channel &&
            c.domain == domain &&
            c.stepSize == stepSize &&
            c.blockSize == blockSize &&
            c.windowType == windowType);
}

void
PluginTransform::ExecutionContext::makeConsistentWithPlugin(const Vamp::PluginBase *_plugin)
{
    const Vamp::Plugin *vp = dynamic_cast<const Vamp::Plugin *>(_plugin);
    if (!vp) {
        vp = dynamic_cast<const Vamp::PluginHostAdapter *>(_plugin); //!!! why?
    }

    if (!vp) {
        domain = Vamp::Plugin::TimeDomain;
        if (!stepSize) {
            if (!blockSize) blockSize = 1024;
            stepSize = blockSize;
        } else {
            if (!blockSize) blockSize = stepSize;
        }
    } else {
        domain = vp->getInputDomain();
        if (!stepSize) stepSize = vp->getPreferredStepSize();
        if (!blockSize) blockSize = vp->getPreferredBlockSize();
        if (!blockSize) blockSize = 1024;
        if (!stepSize) {
            if (domain == Vamp::Plugin::FrequencyDomain) {
                stepSize = blockSize/2;
            } else {
                stepSize = blockSize;
            }
        }
    }
}