Mercurial > hg > svcore
view transform/TransformFactory.cpp @ 47:bac8b14ab355
* Add menu for re-adding existing layers
* Fix layer tree window so that it at least approximates correct
* Add bundled operations in command history, for use with things like
multiple consecutive changes to a parameter value
* Disambiguate plugins that happen to have identical descriptions
* Add spectral centroid plugin (could use some parameters!)
* Some other fixes
author | Chris Cannam |
---|---|
date | Fri, 17 Mar 2006 17:38:28 +0000 |
parents | 742e6882e187 |
children | 39ae3dee27b9 |
line wrap: on
line source
/* -*- c-basic-offset: 4 -*- vi:set ts=8 sts=4 sw=4: */ /* A waveform viewer and audio annotation editor. Chris Cannam, Queen Mary University of London, 2005-2006 This is experimental software. Not for distribution. */ #include "TransformFactory.h" #include "BeatDetectTransform.h" #include "BeatDetectionFunctionTransform.h" #include "FeatureExtractionPluginTransform.h" #include "plugin/FeatureExtractionPluginFactory.h" #include <iostream> TransformFactory * TransformFactory::m_instance = new TransformFactory; TransformFactory * TransformFactory::instance() { return m_instance; } TransformFactory::~TransformFactory() { } TransformFactory::TransformList TransformFactory::getAllTransforms() { if (m_transforms.empty()) populateTransforms(); TransformList list; for (TransformMap::const_iterator i = m_transforms.begin(); i != m_transforms.end(); ++i) { list.push_back(TransformDesc(i->first, i->second)); } return list; } void TransformFactory::populateTransforms() { std::vector<QString> fexplugs = FeatureExtractionPluginFactory::getAllPluginIdentifiers(); std::map<QString, QString> makers; for (size_t i = 0; i < fexplugs.size(); ++i) { QString pluginId = fexplugs[i]; FeatureExtractionPluginFactory *factory = FeatureExtractionPluginFactory::instanceFor(pluginId); if (!factory) { std::cerr << "WARNING: TransformFactory::populateTransforms: No feature extraction plugin factory for instance " << pluginId.toLocal8Bit().data() << std::endl; continue; } FeatureExtractionPlugin *plugin = factory->instantiatePlugin(pluginId, 48000); if (!plugin) { std::cerr << "WARNING: TransformFactory::populateTransforms: Failed to instantiate plugin " << pluginId.toLocal8Bit().data() << std::endl; continue; } QString pluginDescription = plugin->getDescription().c_str(); FeatureExtractionPlugin::OutputList outputs = plugin->getOutputDescriptors(); for (size_t j = 0; j < outputs.size(); ++j) { QString transformName = QString("%1:%2") .arg(pluginId).arg(outputs[j].name.c_str()); QString userDescription; if (outputs.size() == 1) { userDescription = pluginDescription; } else { userDescription = QString("%1: %2") .arg(pluginDescription) .arg(outputs[j].description.c_str()); } m_transforms[transformName] = userDescription; makers[transformName] = plugin->getMaker().c_str(); } } // disambiguate plugins with similar descriptions std::map<QString, int> descriptions; for (TransformMap::iterator i = m_transforms.begin(); i != m_transforms.end(); ++i) { QString name = i->first, description = i->second; ++descriptions[description]; ++descriptions[QString("%1 [%2]").arg(description).arg(makers[name])]; } std::map<QString, int> counts; TransformMap newMap; for (TransformMap::iterator i = m_transforms.begin(); i != m_transforms.end(); ++i) { QString name = i->first, description = i->second; if (descriptions[description] > 1) { description = QString("%1 [%2]").arg(description).arg(makers[name]); if (descriptions[description] > 1) { description = QString("%1 <%2>") .arg(description).arg(++counts[description]); } } newMap[name] = description; } m_transforms = newMap; } QString TransformFactory::getTransformDescription(TransformName name) { if (m_transforms.find(name) != m_transforms.end()) { return m_transforms[name]; } else return ""; } QString TransformFactory::getTransformFriendlyName(TransformName name) { QString description = getTransformDescription(name); int i = description.indexOf(':'); if (i >= 0) { return description.remove(0, i + 2); } else { return description; } } Transform * TransformFactory::createTransform(TransformName name, Model *inputModel) { return createTransform(name, inputModel, true); } Transform * TransformFactory::createTransform(TransformName name, Model *inputModel, bool start) { Transform *transform = 0; if (name == BeatDetectTransform::getName()) { transform = new BeatDetectTransform(inputModel); } else if (name == BeatDetectionFunctionTransform::getName()) { transform = new BeatDetectionFunctionTransform(inputModel); } else { QString id = name.section(':', 0, 2); QString output = name.section(':', 3); if (FeatureExtractionPluginFactory::instanceFor(id)) { transform = new FeatureExtractionPluginTransform(inputModel, id, output); } else { std::cerr << "TransformFactory::createTransform: Unknown transform " << name.toStdString() << std::endl; } } if (start && transform) transform->start(); transform->setObjectName(name); return transform; } Model * TransformFactory::transform(TransformName name, Model *inputModel) { Transform *t = createTransform(name, inputModel, false); if (!t) return 0; connect(t, SIGNAL(finished()), this, SLOT(transformFinished())); t->start(); return t->detachOutputModel(); } void TransformFactory::transformFinished() { QObject *s = sender(); Transform *transform = dynamic_cast<Transform *>(s); if (!transform) { std::cerr << "WARNING: TransformFactory::transformFinished: sender is not a transform" << std::endl; return; } transform->wait(); // unnecessary but reassuring delete transform; }