libs/ardour/quantize.cc

   1 /*
   2     Copyright (C) 2004 Paul Davis
   3
   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.
   8
   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.
  13
  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.
  17
  18 */
  19 #include <cmath>
  20
  21 #include "pbd/basename.h"
  22
  23 #include "ardour/types.h"
  24 #include "ardour/quantize.h"
  25 #include "ardour/session.h"
  26 #include "ardour/smf_source.h"
  27 #include "ardour/midi_model.h"
  28 #include "ardour/midi_region.h"
  29 #include "ardour/tempo.h"
  30
  31 #include "i18n.h"
  32
  33 using namespace std;
  34 using namespace PBD;
  35 using namespace ARDOUR;
  36
  37 /** Quantize notes
  38  *
  39  * grid parameters are the quantize value in beats, ie 1.0 = quantize to beats,
  40  * 0.25 = quantize to beats/4, etc.
  41  */
  42
  43 Quantize::Quantize (Session& s, QuantizeType /* type */,
  44                     bool snap_start, bool snap_end,
  45                     double start_grid, double end_grid,
  46                     float strength, float swing, float threshold)
  47         : session (s)
  48         , _snap_start (snap_start)
  49         , _snap_end (snap_end)
  50         , _start_grid(start_grid)
  51         , _end_grid(end_grid)
  52         , _strength (strength/100.0)
  53         , _swing (swing/100.0)
  54         , _threshold (threshold)
  55 {
  56 }
  57
  58 Quantize::~Quantize ()
  59 {
  60 }
  61
  62 Command*
  63 Quantize::operator () (boost::shared_ptr<MidiModel> model, std::vector<Evoral::Sequence<Evoral::MusicalTime>::Notes>& seqs)
  64 {
  65         bool even;
  66         MidiModel::DiffCommand* cmd = new MidiModel::DiffCommand (model, "quantize");
  67
  68         for (std::vector<Evoral::Sequence<Evoral::MusicalTime>::Notes>::iterator s = seqs.begin(); s != seqs.end(); ++s) {
  69
  70                 even = false;
  71
  72                 for (Evoral::Sequence<MidiModel::TimeType>::Notes::iterator i = (*s).begin(); i != (*s).end(); ++i) {
  73
  74                         double new_start = round ((*i)->time() / _start_grid) * _start_grid;
  75                         double new_end = round ((*i)->end_time() / _end_grid) * _end_grid;
  76                         double delta;
  77
  78                         if (_swing > 0.0 && !even) {
  79
  80                                 double next_grid = new_start + _start_grid;
  81
  82                                 /* find a spot 2/3 (* swing factor) of the way between the grid point
  83                                    we would put this note at, and the nominal position of the next note.
  84                                 */
  85
  86                                 new_start = new_start + (2.0/3.0 * _swing * (next_grid - new_start));
  87
  88                         } else if (_swing < 0.0 && !even) {
  89
  90                                 double prev_grid = new_start - _start_grid;
  91
  92                                 /* find a spot 2/3 (* swing factor) of the way between the grid point
  93                                    we would put this note at, and the nominal position of the previous note.
  94                                 */
  95
  96                                 new_start = new_start - (2.0/3.0 * _swing * (new_start - prev_grid));
  97
  98                         }
  99
 100                         delta = new_start - (*i)->time();
 101
 102                         if (fabs (delta) >= _threshold) {
 103                                 if (_snap_start) {
 104                                         delta *= _strength;
 105                                         cmd->change ((*i), MidiModel::DiffCommand::StartTime,
 106                                                      (*i)->time() + delta);
 107                                 }
 108                         }
 109
 110                         if (_snap_end) {
 111                                 delta = new_end - (*i)->end_time();
 112
 113                                 if (fabs (delta) >= _threshold) {
 114                                         double new_dur = new_end - new_start;
 115
 116                                         if (new_dur == 0.0) {
 117                                                 new_dur = _end_grid;
 118                                         }
 119
 120                                         cmd->change ((*i), MidiModel::DiffCommand::Length, new_dur);
 121                                 }
 122                         }
 123
 124                         even = !even;
 125                 }
 126         }
 127
 128         return cmd;
 129 }