cd17e05e3a
git-svn-id: svn://localhost/ardour2/branches/3.0@9783 d708f5d6-7413-0410-9779-e7cbd77b26cf
171 lines
3.2 KiB
C++
171 lines
3.2 KiB
C++
#include <cmath>
|
|
|
|
#include "ardour/readable.h"
|
|
#include "ardour/transient_detector.h"
|
|
|
|
#include "i18n.h"
|
|
|
|
using namespace Vamp;
|
|
using namespace ARDOUR;
|
|
using namespace std;
|
|
|
|
/* need a static initializer function for this */
|
|
|
|
string TransientDetector::_op_id = X_("libardourvampplugins:qm-onsetdetector:2");
|
|
|
|
TransientDetector::TransientDetector (float sr)
|
|
: AudioAnalyser (sr, X_("libardourvampplugins:qm-onsetdetector"))
|
|
{
|
|
/* update the op_id */
|
|
|
|
_op_id = X_("libardourvampplugins:qm-onsetdetector");
|
|
|
|
// XXX this should load the above-named plugin and get the current version
|
|
|
|
_op_id += ":2";
|
|
|
|
threshold = 0.00;
|
|
}
|
|
|
|
TransientDetector::~TransientDetector()
|
|
{
|
|
}
|
|
|
|
string
|
|
TransientDetector::operational_identifier()
|
|
{
|
|
return _op_id;
|
|
}
|
|
|
|
int
|
|
TransientDetector::run (const std::string& path, Readable* src, uint32_t channel, AnalysisFeatureList& results)
|
|
{
|
|
current_results = &results;
|
|
int ret = analyse (path, src, channel);
|
|
|
|
current_results = 0;
|
|
|
|
return ret;
|
|
}
|
|
|
|
int
|
|
TransientDetector::use_features (Plugin::FeatureSet& features, ostream* out)
|
|
{
|
|
const Plugin::FeatureList& fl (features[0]);
|
|
|
|
for (Plugin::FeatureList::const_iterator f = fl.begin(); f != fl.end(); ++f) {
|
|
|
|
if (f->hasTimestamp) {
|
|
|
|
if (out) {
|
|
(*out) << (*f).timestamp.toString() << endl;
|
|
}
|
|
|
|
current_results->push_back (RealTime::realTime2Frame (f->timestamp, (framecnt_t) floor(sample_rate)));
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void
|
|
TransientDetector::set_threshold (float val)
|
|
{
|
|
threshold = val;
|
|
}
|
|
|
|
void
|
|
TransientDetector::set_sensitivity (float val)
|
|
{
|
|
if (plugin) {
|
|
plugin->selectProgram ("Percussive onsets");
|
|
plugin->setParameter ("sensitivity", val);
|
|
}
|
|
}
|
|
|
|
void
|
|
TransientDetector::cleanup_transients (AnalysisFeatureList& t, float sr, float gap_msecs)
|
|
{
|
|
if (t.empty()) {
|
|
return;
|
|
}
|
|
|
|
t.sort ();
|
|
|
|
/* remove duplicates or other things that are too close */
|
|
|
|
AnalysisFeatureList::iterator i = t.begin();
|
|
AnalysisFeatureList::iterator f, b;
|
|
const framecnt_t gap_frames = (framecnt_t) floor (gap_msecs * (sr / 1000.0));
|
|
|
|
while (i != t.end()) {
|
|
|
|
// move front iterator to just past i, and back iterator the same place
|
|
|
|
f = i;
|
|
++f;
|
|
b = f;
|
|
|
|
// move f until we find a new value that is far enough away
|
|
|
|
while ((f != t.end()) && (((*f) - (*i)) < gap_frames)) {
|
|
++f;
|
|
}
|
|
|
|
i = f;
|
|
|
|
// if f moved forward from b, we had duplicates/too-close points: get rid of them
|
|
|
|
if (b != f) {
|
|
t.erase (b, f);
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
TransientDetector::update_positions (Readable* src, uint32_t channel, AnalysisFeatureList& positions)
|
|
{
|
|
Plugin::FeatureSet features;
|
|
|
|
Sample* data = 0;
|
|
|
|
int buff_size = 1024;
|
|
int step_size = 64;
|
|
|
|
data = new Sample[buff_size];
|
|
|
|
AnalysisFeatureList::iterator i = positions.begin();
|
|
|
|
while (i != positions.end()) {
|
|
|
|
framecnt_t to_read;
|
|
|
|
/* read from source */
|
|
to_read = buff_size;
|
|
|
|
if (src->read (data, (*i) - buff_size, to_read, channel) != to_read) {
|
|
break;
|
|
}
|
|
|
|
// Simple heuristic for locating approx correct cut position.
|
|
|
|
for (int j = 0; j < buff_size;){
|
|
|
|
Sample s = abs (data[j]);
|
|
Sample s2 = abs (data[j + step_size]);
|
|
|
|
if ((s2 - s) > threshold){
|
|
//cerr << "Thresh exceeded. Moving pos from: " << (*i) << " to: " << (*i) - buff_size + (j + 16) << endl;
|
|
(*i) = (*i) - buff_size + (j + 24);
|
|
break;
|
|
}
|
|
|
|
j = j + step_size;
|
|
}
|
|
|
|
++i;
|
|
}
|
|
|
|
delete [] data;
|
|
}
|