2 Copyright (C) 2012 Paul Davis
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 #include "ardour/readable.h"
23 #include "ardour/transient_detector.h"
28 using namespace ARDOUR;
31 /* need a static initializer function for this */
33 string TransientDetector::_op_id = X_("qm-onset");
35 TransientDetector::TransientDetector (float sr)
36 : AudioAnalyser (sr, X_("libardourvampplugins:qm-onsetdetector"))
41 TransientDetector::~TransientDetector()
46 TransientDetector::operational_identifier()
52 TransientDetector::run (const std::string& path, Readable* src, uint32_t channel, AnalysisFeatureList& results)
54 current_results = &results;
55 int ret = analyse (path, src, channel);
63 TransientDetector::use_features (Plugin::FeatureSet& features, ostream* out)
65 const Plugin::FeatureList& fl (features[0]);
67 for (Plugin::FeatureList::const_iterator f = fl.begin(); f != fl.end(); ++f) {
69 if (f->hasTimestamp) {
72 (*out) << (*f).timestamp.toString() << endl;
75 current_results->push_back (RealTime::realTime2Frame (f->timestamp, (framecnt_t) floor(sample_rate)));
83 TransientDetector::set_threshold (float val)
89 TransientDetector::set_sensitivity (uint32_t mode, float val)
92 // see libs/vamp-plugins/OnsetDetect.cpp
93 //plugin->selectProgram ("General purpose"); // dftype = 3, sensitivity = 50, whiten = 0 (default)
94 //plugin->selectProgram ("Percussive onsets"); // dftype = 4, sensitivity = 40, whiten = 0
95 plugin->setParameter ("dftype", mode);
96 plugin->setParameter ("sensitivity", std::min (100.f, std::max (0.f, val)));
97 plugin->setParameter ("whiten", 0);
102 TransientDetector::cleanup_transients (AnalysisFeatureList& t, float sr, float gap_msecs)
110 /* remove duplicates or other things that are too close */
112 AnalysisFeatureList::iterator i = t.begin();
113 AnalysisFeatureList::iterator f, b;
114 const framecnt_t gap_frames = (framecnt_t) floor (gap_msecs * (sr / 1000.0));
116 while (i != t.end()) {
118 // move front iterator to just past i, and back iterator the same place
124 // move f until we find a new value that is far enough away
126 while ((f != t.end()) && gap_frames > 0 && (((*f) - (*i)) < gap_frames)) {
132 // if f moved forward from b, we had duplicates/too-close points: get rid of them
141 TransientDetector::update_positions (Readable* src, uint32_t channel, AnalysisFeatureList& positions)
143 int const buff_size = 1024;
144 int const step_size = 64;
146 Sample* data = new Sample[buff_size];
148 AnalysisFeatureList::iterator i = positions.begin();
150 while (i != positions.end()) {
152 /* read from source */
153 framecnt_t const to_read = buff_size;
155 if (src->read (data, (*i) - buff_size, to_read, channel) != to_read) {
159 // Simple heuristic for locating approx correct cut position.
161 for (int j = 0; j < (buff_size - step_size); ) {
163 Sample const s = abs (data[j]);
164 Sample const s2 = abs (data[j + step_size]);
166 if ((s2 - s) > threshold) {
167 //cerr << "Thresh exceeded. Moving pos from: " << (*i) << " to: " << (*i) - buff_size + (j + 16) << endl;
168 (*i) = (*i) - buff_size + (j + 24);