cannam@1: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
cannam@1: 
cannam@1: /*
cannam@1:     Vamp
cannam@1: 
cannam@1:     An API for audio analysis and feature extraction plugins.
cannam@1: 
cannam@1:     Centre for Digital Music, Queen Mary, University of London.
cannam@1:     Copyright 2006 Chris Cannam.
cannam@1:   
cannam@1:     Permission is hereby granted, free of charge, to any person
cannam@1:     obtaining a copy of this software and associated documentation
cannam@1:     files (the "Software"), to deal in the Software without
cannam@1:     restriction, including without limitation the rights to use, copy,
cannam@1:     modify, merge, publish, distribute, sublicense, and/or sell copies
cannam@1:     of the Software, and to permit persons to whom the Software is
cannam@1:     furnished to do so, subject to the following conditions:
cannam@1: 
cannam@1:     The above copyright notice and this permission notice shall be
cannam@1:     included in all copies or substantial portions of the Software.
cannam@1: 
cannam@1:     THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
cannam@1:     EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
cannam@1:     MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
cannam@6:     NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR
cannam@1:     ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
cannam@1:     CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
cannam@1:     WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
cannam@1: 
cannam@1:     Except as contained in this notice, the names of the Centre for
cannam@1:     Digital Music; Queen Mary, University of London; and Chris Cannam
cannam@1:     shall not be used in advertising or otherwise to promote the sale,
cannam@1:     use or other dealings in this Software without prior written
cannam@1:     authorization.
cannam@1: */
cannam@1: 
cannam@1: #include "PluginHostAdapter.h"
cannam@1: 
cannam@1: namespace Vamp
cannam@1: {
cannam@1: 
cannam@1: PluginHostAdapter::PluginHostAdapter(const VampPluginDescriptor *descriptor,
cannam@1:                                      float inputSampleRate) :
cannam@1:     Plugin(inputSampleRate),
cannam@1:     m_descriptor(descriptor)
cannam@1: {
cannam@15: //    std::cerr << "PluginHostAdapter::PluginHostAdapter (plugin = " << descriptor->name << ")" << std::endl;
cannam@1:     m_handle = m_descriptor->instantiate(m_descriptor, inputSampleRate);
cannam@15:     if (!m_handle) {
cannam@15: //        std::cerr << "WARNING: PluginHostAdapter: Plugin instantiation failed for plugin " << m_descriptor->name << std::endl;
cannam@15:     }
cannam@1: }
cannam@1: 
cannam@1: PluginHostAdapter::~PluginHostAdapter()
cannam@1: {
cannam@15: //    std::cerr << "PluginHostAdapter::~PluginHostAdapter (plugin = " << m_descriptor->name << ")" << std::endl;
cannam@1:     if (m_handle) m_descriptor->cleanup(m_handle);
cannam@1: }
cannam@1: 
cannam@32: std::vector<std::string>
cannam@32: PluginHostAdapter::getPluginPath()
cannam@32: {
cannam@32:     std::vector<std::string> path;
cannam@32:     std::string envPath;
cannam@32: 
cannam@32:     char *cpath = getenv("VAMP_PATH");
cannam@32:     if (cpath) envPath = cpath;
cannam@32: 
cannam@32: #ifdef _WIN32
cannam@32: #define PATH_SEPARATOR ';'
cannam@32: #define DEFAULT_VAMP_PATH "%ProgramFiles%\\Vamp Plugins"
cannam@32: #else
cannam@32: #define PATH_SEPARATOR ':'
cannam@32: #ifdef __APPLE__
cannam@32: #define DEFAULT_VAMP_PATH "/Library/Audio/Plug-Ins/Vamp/:$HOME/Library/Audio/Plug-Ins/Vamp"
cannam@32: #else
cannam@32: #define DEFAULT_VAMP_PATH "/usr/local/lib/vamp:/usr/lib/vamp:$HOME/vamp:$HOME/.vamp"
cannam@32: #endif
cannam@32: #endif
cannam@32: 
cannam@32:     if (envPath == "") {
cannam@32:         envPath = DEFAULT_VAMP_PATH;
cannam@32:         char *chome = getenv("HOME");
cannam@32:         if (chome) {
cannam@32:             std::string home(chome);
cannam@32:             std::string::size_type f;
cannam@32:             while ((f = envPath.find("$HOME")) != std::string::npos &&
cannam@32:                     f < envPath.length()) {
cannam@32:                 envPath.replace(f, 5, home);
cannam@32:             }
cannam@32:         }
cannam@32: #ifdef _WIN32
cannam@32:         char *cpfiles = getenv("ProgramFiles");
cannam@32:         if (!cpfiles) cpfiles = "C:\\Program Files";
cannam@32:         std::string pfiles(cpfiles);
cannam@32:         std::string::size_type f;
cannam@32:         while ((f = envPath.find("%ProgramFiles%")) != std::string::npos &&
cannam@32:                f < envPath.length()) {
cannam@32:             envPath.replace(f, 14, pfiles);
cannam@32:         }
cannam@32: #endif
cannam@32:     }
cannam@32: 
cannam@32:     std::string::size_type index = 0, newindex = 0;
cannam@32: 
cannam@32:     while ((newindex = envPath.find(PATH_SEPARATOR, index)) < envPath.size()) {
cannam@32: 	path.push_back(envPath.substr(index, newindex - index));
cannam@32: 	index = newindex + 1;
cannam@32:     }
cannam@32:     
cannam@32:     path.push_back(envPath.substr(index));
cannam@32: 
cannam@32:     return path;
cannam@32: }
cannam@32: 
cannam@1: bool
cannam@1: PluginHostAdapter::initialise(size_t channels,
cannam@1:                               size_t stepSize,
cannam@1:                               size_t blockSize)
cannam@1: {
cannam@1:     if (!m_handle) return false;
cannam@1:     return m_descriptor->initialise(m_handle, channels, stepSize, blockSize) ?
cannam@1:         true : false;
cannam@1: }
cannam@1: 
cannam@1: void
cannam@1: PluginHostAdapter::reset()
cannam@1: {
cannam@1:     if (!m_handle) return;
cannam@1:     m_descriptor->reset(m_handle);
cannam@1: }
cannam@1: 
cannam@1: PluginHostAdapter::InputDomain
cannam@1: PluginHostAdapter::getInputDomain() const
cannam@1: {
cannam@1:     if (m_descriptor->inputDomain == vampFrequencyDomain) {
cannam@1:         return FrequencyDomain;
cannam@1:     } else {
cannam@1:         return TimeDomain;
cannam@1:     }
cannam@1: }
cannam@1: 
cannam@1: std::string
cannam@1: PluginHostAdapter::getName() const
cannam@1: {
cannam@1:     return m_descriptor->name;
cannam@1: }
cannam@1: 
cannam@1: std::string
cannam@1: PluginHostAdapter::getDescription() const
cannam@1: {
cannam@1:     return m_descriptor->description;
cannam@1: }
cannam@1: 
cannam@1: std::string
cannam@1: PluginHostAdapter::getMaker() const
cannam@1: {
cannam@1:     return m_descriptor->maker;
cannam@1: }
cannam@1: 
cannam@1: int
cannam@1: PluginHostAdapter::getPluginVersion() const
cannam@1: {
cannam@1:     return m_descriptor->pluginVersion;
cannam@1: }
cannam@1: 
cannam@1: std::string
cannam@1: PluginHostAdapter::getCopyright() const
cannam@1: {
cannam@1:     return m_descriptor->copyright;
cannam@1: }
cannam@1: 
cannam@1: PluginHostAdapter::ParameterList
cannam@1: PluginHostAdapter::getParameterDescriptors() const
cannam@1: {
cannam@1:     ParameterList list;
cannam@1:     for (unsigned int i = 0; i < m_descriptor->parameterCount; ++i) {
cannam@1:         const VampParameterDescriptor *spd = m_descriptor->parameters[i];
cannam@1:         ParameterDescriptor pd;
cannam@1:         pd.name = spd->name;
cannam@1:         pd.description = spd->description;
cannam@1:         pd.unit = spd->unit;
cannam@1:         pd.minValue = spd->minValue;
cannam@1:         pd.maxValue = spd->maxValue;
cannam@1:         pd.defaultValue = spd->defaultValue;
cannam@1:         pd.isQuantized = spd->isQuantized;
cannam@1:         pd.quantizeStep = spd->quantizeStep;
cannam@9:         if (pd.isQuantized && spd->valueNames) {
cannam@9:             for (unsigned int j = 0; spd->valueNames[j]; ++j) {
cannam@9:                 pd.valueNames.push_back(spd->valueNames[j]);
cannam@9:             }
cannam@9:         }
cannam@1:         list.push_back(pd);
cannam@1:     }
cannam@1:     return list;
cannam@1: }
cannam@1: 
cannam@1: float
cannam@1: PluginHostAdapter::getParameter(std::string param) const
cannam@1: {
cannam@1:     if (!m_handle) return 0.0;
cannam@1: 
cannam@1:     for (unsigned int i = 0; i < m_descriptor->parameterCount; ++i) {
cannam@1:         if (param == m_descriptor->parameters[i]->name) {
cannam@1:             return m_descriptor->getParameter(m_handle, i);
cannam@1:         }
cannam@1:     }
cannam@1: 
cannam@1:     return 0.0;
cannam@1: }
cannam@1: 
cannam@1: void
cannam@1: PluginHostAdapter::setParameter(std::string param, 
cannam@1:                                 float value)
cannam@1: {
cannam@1:     if (!m_handle) return;
cannam@1: 
cannam@1:     for (unsigned int i = 0; i < m_descriptor->parameterCount; ++i) {
cannam@1:         if (param == m_descriptor->parameters[i]->name) {
cannam@1:             m_descriptor->setParameter(m_handle, i, value);
cannam@1:             return;
cannam@1:         }
cannam@1:     }
cannam@1: }
cannam@1: 
cannam@1: PluginHostAdapter::ProgramList
cannam@1: PluginHostAdapter::getPrograms() const
cannam@1: {
cannam@1:     ProgramList list;
cannam@1:     
cannam@1:     for (unsigned int i = 0; i < m_descriptor->programCount; ++i) {
cannam@1:         list.push_back(m_descriptor->programs[i]);
cannam@1:     }
cannam@1:     
cannam@1:     return list;
cannam@1: }
cannam@1: 
cannam@1: std::string
cannam@1: PluginHostAdapter::getCurrentProgram() const
cannam@1: {
cannam@1:     if (!m_handle) return "";
cannam@1: 
cannam@1:     int pn = m_descriptor->getCurrentProgram(m_handle);
cannam@1:     return m_descriptor->programs[pn];
cannam@1: }
cannam@1: 
cannam@1: void
cannam@1: PluginHostAdapter::selectProgram(std::string program)
cannam@1: {
cannam@1:     if (!m_handle) return;
cannam@1: 
cannam@1:     for (unsigned int i = 0; i < m_descriptor->programCount; ++i) {
cannam@1:         if (program == m_descriptor->programs[i]) {
cannam@1:             m_descriptor->selectProgram(m_handle, i);
cannam@1:             return;
cannam@1:         }
cannam@1:     }
cannam@1: }
cannam@1: 
cannam@1: size_t
cannam@1: PluginHostAdapter::getPreferredStepSize() const
cannam@1: {
cannam@1:     if (!m_handle) return 0;
cannam@1:     return m_descriptor->getPreferredStepSize(m_handle);
cannam@1: }
cannam@1: 
cannam@1: size_t
cannam@1: PluginHostAdapter::getPreferredBlockSize() const
cannam@1: {
cannam@1:     if (!m_handle) return 0;
cannam@1:     return m_descriptor->getPreferredBlockSize(m_handle);
cannam@1: }
cannam@1: 
cannam@26: size_t
cannam@26: PluginHostAdapter::getMinChannelCount() const
cannam@26: {
cannam@26:     if (!m_handle) return 0;
cannam@26:     return m_descriptor->getMinChannelCount(m_handle);
cannam@26: }
cannam@26: 
cannam@26: size_t
cannam@26: PluginHostAdapter::getMaxChannelCount() const
cannam@26: {
cannam@26:     if (!m_handle) return 0;
cannam@26:     return m_descriptor->getMaxChannelCount(m_handle);
cannam@26: }
cannam@26: 
cannam@1: PluginHostAdapter::OutputList
cannam@1: PluginHostAdapter::getOutputDescriptors() const
cannam@1: {
cannam@1:     OutputList list;
cannam@15:     if (!m_handle) {
cannam@15: //        std::cerr << "PluginHostAdapter::getOutputDescriptors: no handle " << std::endl;
cannam@15:         return list;
cannam@15:     }
cannam@1: 
cannam@1:     unsigned int count = m_descriptor->getOutputCount(m_handle);
cannam@1: 
cannam@1:     for (unsigned int i = 0; i < count; ++i) {
cannam@1:         VampOutputDescriptor *sd = m_descriptor->getOutputDescriptor(m_handle, i);
cannam@1:         OutputDescriptor d;
cannam@1:         d.name = sd->name;
cannam@1:         d.description = sd->description;
cannam@1:         d.unit = sd->unit;
cannam@9:         d.hasFixedBinCount = sd->hasFixedBinCount;
cannam@9:         d.binCount = sd->binCount;
cannam@9:         if (d.hasFixedBinCount) {
cannam@9:             for (unsigned int j = 0; j < sd->binCount; ++j) {
cannam@9:                 d.binNames.push_back(sd->binNames[j] ? sd->binNames[j] : "");
cannam@9:             }
cannam@1:         }
cannam@1:         d.hasKnownExtents = sd->hasKnownExtents;
cannam@1:         d.minValue = sd->minValue;
cannam@1:         d.maxValue = sd->maxValue;
cannam@1:         d.isQuantized = sd->isQuantized;
cannam@1:         d.quantizeStep = sd->quantizeStep;
cannam@1: 
cannam@1:         switch (sd->sampleType) {
cannam@1:         case vampOneSamplePerStep:
cannam@1:             d.sampleType = OutputDescriptor::OneSamplePerStep; break;
cannam@1:         case vampFixedSampleRate:
cannam@1:             d.sampleType = OutputDescriptor::FixedSampleRate; break;
cannam@1:         case vampVariableSampleRate:
cannam@1:             d.sampleType = OutputDescriptor::VariableSampleRate; break;
cannam@1:         }
cannam@1: 
cannam@1:         d.sampleRate = sd->sampleRate;
cannam@1: 
cannam@1:         list.push_back(d);
cannam@1: 
cannam@1:         m_descriptor->releaseOutputDescriptor(sd);
cannam@1:     }
cannam@1: 
cannam@1:     return list;
cannam@1: }
cannam@1: 
cannam@1: PluginHostAdapter::FeatureSet
cannam@1: PluginHostAdapter::process(float **inputBuffers,
cannam@1:                                             RealTime timestamp)
cannam@1: {
cannam@1:     FeatureSet fs;
cannam@1:     if (!m_handle) return fs;
cannam@1: 
cannam@1:     int sec = timestamp.sec;
cannam@1:     int nsec = timestamp.nsec;
cannam@1:     
cannam@12:     VampFeatureList *features = m_descriptor->process(m_handle,
cannam@1:                                                       inputBuffers,
cannam@1:                                                       sec, nsec);
cannam@1:     
cannam@1:     convertFeatures(features, fs);
cannam@1:     m_descriptor->releaseFeatureSet(features);
cannam@1:     return fs;
cannam@1: }
cannam@1: 
cannam@1: PluginHostAdapter::FeatureSet
cannam@1: PluginHostAdapter::getRemainingFeatures()
cannam@1: {
cannam@1:     FeatureSet fs;
cannam@1:     if (!m_handle) return fs;
cannam@1:     
cannam@12:     VampFeatureList *features = m_descriptor->getRemainingFeatures(m_handle); 
cannam@1: 
cannam@1:     convertFeatures(features, fs);
cannam@1:     m_descriptor->releaseFeatureSet(features);
cannam@1:     return fs;
cannam@1: }
cannam@1: 
cannam@1: void
cannam@12: PluginHostAdapter::convertFeatures(VampFeatureList *features,
cannam@1:                                    FeatureSet &fs)
cannam@1: {
cannam@7:     if (!features) return;
cannam@7: 
cannam@12:     unsigned int outputs = m_descriptor->getOutputCount(m_handle);
cannam@12: 
cannam@12:     for (unsigned int i = 0; i < outputs; ++i) {
cannam@1:         
cannam@12:         VampFeatureList &list = features[i];
cannam@1: 
cannam@1:         if (list.featureCount > 0) {
cannam@1: 
cannam@1:             for (unsigned int j = 0; j < list.featureCount; ++j) {
cannam@1:                 
cannam@1:                 Feature feature;
cannam@1:                 feature.hasTimestamp = list.features[j].hasTimestamp;
cannam@1:                 feature.timestamp = RealTime(list.features[j].sec,
cannam@1:                                              list.features[j].nsec);
cannam@1: 
cannam@1:                 for (unsigned int k = 0; k < list.features[j].valueCount; ++k) {
cannam@1:                     feature.values.push_back(list.features[j].values[k]);
cannam@1:                 }
cannam@7: 
cannam@7:                 if (list.features[j].label) {
cannam@7:                     feature.label = list.features[j].label;
cannam@7:                 }
cannam@1: 
cannam@1:                 fs[i].push_back(feature);
cannam@1:             }
cannam@1:         }
cannam@1:     }
cannam@1: }
cannam@1: 
cannam@1: }