/*
- Copyright (C) 2004 Paul Davis
+ Copyright (C) 2004 Paul Davis
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
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
+#include <cmath>
-#include <pbd/basename.h>
+#include "pbd/basename.h"
-#include <ardour/types.h>
-#include <ardour/quantize.h>
-#include <ardour/session.h>
-#include <ardour/smf_source.h>
-#include <ardour/midi_region.h>
-#include <ardour/tempo.h>
+#include "ardour/quantize.h"
+#include "ardour/midi_model.h"
-#include "i18n.h"
+#include "pbd/i18n.h"
using namespace std;
+using namespace PBD;
using namespace ARDOUR;
-
-/** Quantize notes, valid for MIDI regions only.
+/** Quantize notes
*
- * Q is the quantize value in beats, ie 1.0 = quantize to beats,
+ * grid parameters are the quantize value in beats, ie 1.0 = quantize to beats,
* 0.25 = quantize to beats/4, etc.
*/
-Quantize::Quantize (Session& s, double q)
- : Filter (s)
- , _q(q)
+
+Quantize::Quantize (bool snap_start, bool snap_end,
+ double start_grid, double end_grid,
+ float strength, float swing, float threshold)
+ : _snap_start (snap_start)
+ , _snap_end (snap_end)
+ , _start_grid(start_grid)
+ , _end_grid(end_grid)
+ , _strength (strength/100.0)
+ , _swing (swing/100.0)
+ , _threshold (threshold)
{
}
{
}
-int
-Quantize::run (boost::shared_ptr<Region> r)
+static double
+swing_position (double pos, double grid, double swing, double offset)
{
- boost::shared_ptr<MidiRegion> region = boost::dynamic_pointer_cast<MidiRegion>(r);
- if (!region)
- return -1;
-
- // FIXME: how to make a whole file region if it isn't?
- //assert(region->whole_file());
-
- boost::shared_ptr<MidiSource> src = region->midi_source(0);
- src->load_model();
-
- boost::shared_ptr<MidiModel> model = src->model();
-
- // FIXME: Model really needs to be switched to beat time (double) ASAP
-
- const Tempo& t = session.tempo_map().tempo_at(r->start());
- const Meter& m = session.tempo_map().meter_at(r->start());
-
- double q_frames = _q * (m.frames_per_bar(t, session.frame_rate()) / (double)m.beats_per_bar());
-
- for (MidiModel::Notes::iterator i = model->notes().begin(); i != model->notes().end(); ++i) {
- const double new_time = lrint(i->time() / q_frames) * q_frames;
- const double new_dur = ((i->time() != 0 && new_dur < (q_frames * 1.5))
- ? q_frames
- : lrint(i->duration() / q_frames) * q_frames);
-
- i->set_time(new_time);
- i->set_duration(new_dur);
+ /* beats start out numbered at zero.
+ *
+ * every other position on the start-quantize-grid is
+ * optionally swung, meaning that its position is moved
+ * somewhere between its natural position and 2/3 of
+ * the way to the next start-quantize-grid position.
+ *
+ * so, if the _start grid is 0.5, the beat at 0 isn't
+ * swung, but something at 0.5 is, the beat at 1 isn't
+ * swung, but something at 1.5 is.
+ *
+ * if the start grid is 1.0, the beat at 0 isn't swung,
+ * but the beat at 1.0 is. the beat at 2.0 isn't swung,
+ * but the beat at 3.0 is. and so on.
+ *
+ * so the criterion for a position being swung is
+ * whether or not ((possible_grid_position / grid) % 2) != 0
+ */
+
+ const bool swing_quantize_grid_position = pos > 0.0 && fmod ((pos/grid), 2.0) != 0;
+ const bool swing_previous_grid_position = pos > grid && fmod ((pos-grid)/grid, 2.0) != 0;
+
+ /* one of these will not be subject to swing */
+
+ double swung_pos = pos;
+ double swung_previous_grid_position;
+
+ if (pos > grid) {
+ swung_previous_grid_position = pos - grid;
+ } else {
+ swung_previous_grid_position = 0.0;
+ }
+
+ if (swing_previous_grid_position) {
+ swung_previous_grid_position = swung_previous_grid_position + (2.0/3.0 * swing * grid);
}
- return 0;
-#if 0
- SourceList nsrcs;
- SourceList::iterator si;
- nframes_t blocksize = 256 * 1024;
- Sample* buf = 0;
- nframes_t fpos;
- nframes_t fstart;
- nframes_t to_read;
- int ret = -1;
-
- boost::shared_ptr<AudioRegion> region = boost::dynamic_pointer_cast<AudioRegion>(r);
- if (!region)
- return ret;
-
- /* create new sources */
-
- if (make_new_sources (region, nsrcs)) {
- goto out;
+ if (swing_quantize_grid_position) {
+ swung_pos = swung_pos + (2.0/3.0 * swing * grid);
}
- fstart = region->start();
+ /* now correct for start-of-model offset */
+
+ pos += offset;
- if (blocksize > region->length()) {
- blocksize = region->length();
+ if (fabs (pos - swung_pos) > fabs (pos - swung_previous_grid_position)) {
+ pos = swung_previous_grid_position;
+ } else {
+ pos = swung_pos;
}
- fpos = max (fstart, (fstart + region->length() - blocksize));
- buf = new Sample[blocksize];
- to_read = blocksize;
+ return pos;
+}
- /* now read it backwards */
+Command*
+Quantize::operator () (boost::shared_ptr<MidiModel> model,
+ Evoral::Beats position,
+ std::vector<Evoral::Sequence<Evoral::Beats>::Notes>& seqs)
+{
+ /* TODO: Rewrite this to be precise with fixed point? */
- while (to_read) {
+ /* Calculate offset from start of model to next closest quantize step,
+ to quantize relative to actual session beats (etc.) rather than from the
+ start of the model.
+ */
+ const double round_pos = round(position.to_double() / _start_grid) * _start_grid;
+ const double offset = round_pos - position.to_double();
- uint32_t n;
+ MidiModel::NoteDiffCommand* cmd = new MidiModel::NoteDiffCommand (model, "quantize");
- for (n = 0, si = nsrcs.begin(); n < region->n_channels(); ++n, ++si) {
+ for (std::vector<Evoral::Sequence<Evoral::Beats>::Notes>::iterator s = seqs.begin(); s != seqs.end(); ++s) {
- /* read it in */
-
- if (region->audio_source (n)->read (buf, fpos, to_read) != to_read) {
- goto out;
- }
+ for (Evoral::Sequence<MidiModel::TimeType>::Notes::iterator i = (*s).begin(); i != (*s).end(); ++i) {
- /* swap memory order */
-
- for (nframes_t i = 0; i < to_read/2; ++i) {
- swap (buf[i],buf[to_read-1-i]);
- }
-
- /* write it out */
+ /* compute new start + end points WITHOUT the offset
+ * caused by the start of the model (see above).
+ *
+ * these versions of new_start and new_end are
+ * guaranteed to precisely align with the quantize grid(s).
+ */
+
+ double new_start = round (((*i)->time().to_double() - offset) / _start_grid) * _start_grid;
+ double new_end = round (((*i)->end_time().to_double() - offset) / _end_grid) * _end_grid;
+
+ if (_swing) {
- boost::shared_ptr<AudioSource> asrc(boost::dynamic_pointer_cast<AudioSource>(*si));
+ new_start = swing_position (new_start, _start_grid, _swing, offset);
+ new_end = swing_position (new_end, _end_grid, _swing, offset);
- if (asrc && asrc->write (buf, to_read) != to_read) {
- goto out;
+ } else {
+
+ /* now correct for start-of-model offset */
+
+ new_start += offset;
+ new_end += offset;
}
- }
- if (fpos > fstart + blocksize) {
- fpos -= to_read;
- to_read = blocksize;
- } else {
- to_read = fpos - fstart;
- fpos = fstart;
- }
- };
+ double delta = new_start - (*i)->time().to_double();
+
- ret = finish (region, nsrcs);
+ if (fabs (delta) >= _threshold) {
+ if (_snap_start) {
+ delta *= _strength;
+ cmd->change ((*i), MidiModel::NoteDiffCommand::StartTime,
+ (*i)->time() + delta);
+ }
+ }
- out:
+ if (_snap_end) {
+ delta = new_end - (*i)->end_time().to_double();
- if (buf) {
- delete [] buf;
- }
+ if (fabs (delta) >= _threshold) {
+ Evoral::Beats new_dur(new_end - new_start);
+
+ if (!new_dur) {
+ new_dur = Evoral::Beats(_end_grid);
+ }
- if (ret) {
- for (si = nsrcs.begin(); si != nsrcs.end(); ++si) {
- boost::shared_ptr<AudioSource> asrc(boost::dynamic_pointer_cast<AudioSource>(*si));
- asrc->mark_for_remove ();
+ cmd->change ((*i), MidiModel::NoteDiffCommand::Length, new_dur);
+ }
+ }
}
}
-
- return ret;
-#endif
+
+ return cmd;
}