try not thinning when loading old-school automation lists
[ardour.git] / libs / ardour / quantize.cc
index 5bee86d0c491a562065fabcc5d1b7b14449cf92a..a4543d773c75dd035fb3059171673561e837cb75 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    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"
 
 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 (Session& s, bool snap_start, bool snap_end,
+                   double start_grid, double end_grid,
+                   float strength, float swing, float threshold)
+       : session (s)
+       , _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)
 {
 }
 
@@ -47,122 +53,79 @@ Quantize::~Quantize ()
 {
 }
 
-int
-Quantize::run (boost::shared_ptr<Region> r)
+Command*
+Quantize::operator () (boost::shared_ptr<MidiModel> model,
+                       double position,
+                       std::vector<Evoral::Sequence<Evoral::MusicalTime>::Notes>& seqs)
 {
-       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);
-       }
+       /* 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 = ceil(position / _start_grid) * _start_grid;
+       const double offset    = round_pos - position;
 
-       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;
-       }
+       bool even;
+       MidiModel::NoteDiffCommand* cmd = new MidiModel::NoteDiffCommand (model, "quantize");
 
-       fstart = region->start();
+       for (std::vector<Evoral::Sequence<Evoral::MusicalTime>::Notes>::iterator s = seqs.begin(); s != seqs.end(); ++s) {
 
-       if (blocksize > region->length()) {
-               blocksize = region->length();
-       }
+               even = false;
 
-       fpos = max (fstart, (fstart + region->length() - blocksize));
-       buf = new Sample[blocksize];
-       to_read = blocksize;
+               for (Evoral::Sequence<MidiModel::TimeType>::Notes::iterator i = (*s).begin(); i != (*s).end(); ++i) {
 
-       /* now read it backwards */
+                       double new_start = round ((*i)->time() / _start_grid) * _start_grid + offset;
+                       double new_end = round ((*i)->end_time() / _end_grid) * _end_grid + offset;
 
-       while (to_read) {
+                       if (_swing > 0.0 && !even) {
 
-               uint32_t n;
+                               double next_grid = new_start + _start_grid;
 
-               for (n = 0, si = nsrcs.begin(); n < region->n_channels(); ++n, ++si) {
+                               /* find a spot 2/3 (* swing factor) of the way between the grid point
+                                  we would put this note at, and the nominal position of the next note.
+                               */
 
-                       /* read it in */
-                       
-                       if (region->audio_source (n)->read (buf, fpos, to_read) != to_read) {
-                               goto out;
-                       }
+                               new_start = new_start + (2.0/3.0 * _swing * (next_grid - new_start));
+
+                       } else if (_swing < 0.0 && !even) {
+
+                               double prev_grid = new_start - _start_grid;
+
+                               /* find a spot 2/3 (* swing factor) of the way between the grid point
+                                  we would put this note at, and the nominal position of the previous note.
+                               */
+
+                               new_start = new_start - (2.0/3.0 * _swing * (new_start - prev_grid));
 
-                       /* swap memory order */
-                       
-                       for (nframes_t i = 0; i < to_read/2; ++i) {
-                               swap (buf[i],buf[to_read-1-i]);
                        }
-                       
-                       /* write it out */
 
-                       boost::shared_ptr<AudioSource> asrc(boost::dynamic_pointer_cast<AudioSource>(*si));
+                       double delta = new_start - (*i)->time();
 
-                       if (asrc && asrc->write (buf, to_read) != to_read) {
-                               goto out;
+                       if (fabs (delta) >= _threshold) {
+                               if (_snap_start) {
+                                       delta *= _strength;
+                                       cmd->change ((*i), MidiModel::NoteDiffCommand::StartTime,
+                                                    (*i)->time() + delta);
+                               }
                        }
-               }
 
-               if (fpos > fstart + blocksize) {
-                       fpos -= to_read;
-                       to_read = blocksize;
-               } else {
-                       to_read = fpos - fstart;
-                       fpos = fstart;
-               }
-       };
+                       if (_snap_end) {
+                               delta = new_end - (*i)->end_time();
 
-       ret = finish (region, nsrcs);
+                               if (fabs (delta) >= _threshold) {
+                                       double new_dur = new_end - new_start;
 
-  out:
+                                       if (new_dur == 0.0) {
+                                               new_dur = _end_grid;
+                                       }
 
-       if (buf) {
-               delete [] buf;
-       }
+                                       cmd->change ((*i), MidiModel::NoteDiffCommand::Length, new_dur);
+                               }
+                       }
 
-       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 ();
+                       even = !even;
                }
        }
-       
-       return ret;
-#endif
+
+       return cmd;
 }