Move a few declarations to first use.

[ardour.git] / libs / ardour / quantize.cc

diff --git a/libs/ardour/quantize.cc b/libs/ardour/quantize.cc
index becb3bd491a224f75579ad97552e53aead5a5dde..7ede60b578540e73472f4f342819804108810a67 100644 (file)
--- a/libs/ardour/quantize.cc
+++ b/libs/ardour/quantize.cc
@@ -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
@@ -24,12 +24,14 @@
 #include "ardour/quantize.h"
 #include "ardour/session.h"
 #include "ardour/smf_source.h"
+#include "ardour/midi_model.h"
 #include "ardour/midi_region.h"
 #include "ardour/tempo.h"
 
 #include "i18n.h"
 
 using namespace std;
+using namespace PBD;
 using namespace ARDOUR;
 
 /** Quantize notes
@@ -38,9 +40,9 @@ using namespace ARDOUR;
  * 0.25 = quantize to beats/4, etc.
  */
 
-Quantize::Quantize (Session& s, QuantizeType /* type */, 
+Quantize::Quantize (Session& s, QuantizeType /* type */,
                    bool snap_start, bool snap_end,
-                   double start_grid, double end_grid, 
+                   double start_grid, double end_grid,
                    float strength, float swing, float threshold)
        : session (s)
        , _snap_start (snap_start)
@@ -57,71 +59,79 @@ Quantize::~Quantize ()
 {
 }
 
-int
-Quantize::operator () (std::vector<Evoral::Sequence<Evoral::MusicalTime>::Notes>& seqs)
+Command*
+Quantize::operator () (boost::shared_ptr<MidiModel> model,
+                       double position,
+                       std::vector<Evoral::Sequence<Evoral::MusicalTime>::Notes>& seqs)
 {
+       /* 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;
+
        bool even;
+       MidiModel::NoteDiffCommand* cmd = new MidiModel::NoteDiffCommand (model, "quantize");
 
-       for (std::vector<Evoral::Sequence<Evoral::MusicalTime>::Notes>::iterator s = seqs.begin();
-            s != seqs.end(); ++s) {
+       for (std::vector<Evoral::Sequence<Evoral::MusicalTime>::Notes>::iterator s = seqs.begin(); s != seqs.end(); ++s) {
 
                even = false;
 
-               for (Evoral::Sequence<MidiModel::TimeType>::Notes::iterator i = (*s).begin();
-                    i != (*s).end(); ++i) {
-                       
-                       double new_start = round ((*i)->time() / _start_grid) * _start_grid; 
-                       double new_end = round ((*i)->end_time() / _end_grid) * _end_grid;
-                       double delta;
-                       
+               for (Evoral::Sequence<MidiModel::TimeType>::Notes::iterator i = (*s).begin(); i != (*s).end(); ++i) {
+
+                       double new_start = round ((*i)->time() / _start_grid) * _start_grid + offset;
+                       double new_end = round ((*i)->end_time() / _end_grid) * _end_grid + offset;
+
                        if (_swing > 0.0 && !even) {
-                               
+
                                double next_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 next note.
                                */
-                               
+
                                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));
-                               
+
                        }
-                       
-                       delta = new_start - (*i)->time();
-                       
+
+                       double delta = new_start - (*i)->time();
+
                        if (fabs (delta) >= _threshold) {
                                if (_snap_start) {
                                        delta *= _strength;
-                                       (*i)->set_time ((*i)->time() + delta);
+                                       cmd->change ((*i), MidiModel::NoteDiffCommand::StartTime,
+                                                    (*i)->time() + delta);
                                }
                        }
-                       
+
                        if (_snap_end) {
                                delta = new_end - (*i)->end_time();
-                               
+
                                if (fabs (delta) >= _threshold) {
                                        double new_dur = new_end - new_start;
-                                       
+
                                        if (new_dur == 0.0) {
                                                new_dur = _end_grid;
                                        }
-                                       
-                                       (*i)->set_length (new_dur);
+
+                                       cmd->change ((*i), MidiModel::NoteDiffCommand::Length, new_dur);
                                }
                        }
-                       
+
                        even = !even;
                }
        }
 
-       return 0;
+       return cmd;
 }