Added missing files.
[ardour.git] / 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
20 #include <pbd/basename.h>
21
22 #include <ardour/types.h>
23 #include <ardour/quantize.h>
24 #include <ardour/session.h>
25 #include <ardour/smf_source.h>
26 #include <ardour/midi_region.h>
27 #include <ardour/tempo.h>
28
29 #include "i18n.h"
30
31 using namespace std;
32 using namespace ARDOUR;
33
34
35 /** Quantize notes, valid for MIDI regions only.
36  *
37  * Q is the quantize value in beats, ie 1.0 = quantize to beats,
38  * 0.25 = quantize to beats/4, etc.
39  */
40 Quantize::Quantize (Session& s, double q)
41         : Filter (s)
42         , _q(q)
43 {
44 }
45
46 Quantize::~Quantize ()
47 {
48 }
49
50 int
51 Quantize::run (boost::shared_ptr<Region> r)
52 {
53         boost::shared_ptr<MidiRegion> region = boost::dynamic_pointer_cast<MidiRegion>(r);
54         if (!region)
55                 return -1;
56
57         // FIXME: how to make a whole file region if it isn't?
58         //assert(region->whole_file());
59
60         boost::shared_ptr<MidiSource> src = region->midi_source(0);
61         src->load_model();
62
63         boost::shared_ptr<MidiModel> model = src->model();
64         
65         // FIXME: Model really needs to be switched to beat time (double) ASAP
66         
67         const Tempo& t = session.tempo_map().tempo_at(r->start());
68         const Meter& m = session.tempo_map().meter_at(r->start());
69
70         double q_frames = _q * (m.frames_per_bar(t, session.frame_rate()) / (double)m.beats_per_bar());
71
72         for (MidiModel::Notes::iterator i = model->notes().begin(); i != model->notes().end(); ++i) {
73                 const double new_time = lrint(i->time() / q_frames) * q_frames;
74                 const double new_dur = ((i->time() != 0 && new_dur < (q_frames * 1.5))
75                         ? q_frames
76                         : lrint(i->duration() / q_frames) * q_frames);
77                 
78                 i->set_time(new_time);
79                 i->set_duration(new_dur);
80         }
81
82         return 0;
83 #if 0
84         SourceList nsrcs;
85         SourceList::iterator si;
86         nframes_t blocksize = 256 * 1024;
87         Sample* buf = 0;
88         nframes_t fpos;
89         nframes_t fstart;
90         nframes_t to_read;
91         int ret = -1;
92
93         boost::shared_ptr<AudioRegion> region = boost::dynamic_pointer_cast<AudioRegion>(r);
94         if (!region)
95                 return ret;
96
97         /* create new sources */
98
99         if (make_new_sources (region, nsrcs)) {
100                 goto out;
101         }
102
103         fstart = region->start();
104
105         if (blocksize > region->length()) {
106                 blocksize = region->length();
107         }
108
109         fpos = max (fstart, (fstart + region->length() - blocksize));
110         buf = new Sample[blocksize];
111         to_read = blocksize;
112
113         /* now read it backwards */
114
115         while (to_read) {
116
117                 uint32_t n;
118
119                 for (n = 0, si = nsrcs.begin(); n < region->n_channels(); ++n, ++si) {
120
121                         /* read it in */
122                         
123                         if (region->audio_source (n)->read (buf, fpos, to_read) != to_read) {
124                                 goto out;
125                         }
126
127                         /* swap memory order */
128                         
129                         for (nframes_t i = 0; i < to_read/2; ++i) {
130                                 swap (buf[i],buf[to_read-1-i]);
131                         }
132                         
133                         /* write it out */
134
135                         boost::shared_ptr<AudioSource> asrc(boost::dynamic_pointer_cast<AudioSource>(*si));
136
137                         if (asrc && asrc->write (buf, to_read) != to_read) {
138                                 goto out;
139                         }
140                 }
141
142                 if (fpos > fstart + blocksize) {
143                         fpos -= to_read;
144                         to_read = blocksize;
145                 } else {
146                         to_read = fpos - fstart;
147                         fpos = fstart;
148                 }
149         };
150
151         ret = finish (region, nsrcs);
152
153   out:
154
155         if (buf) {
156                 delete [] buf;
157         }
158
159         if (ret) {
160                 for (si = nsrcs.begin(); si != nsrcs.end(); ++si) {
161                         boost::shared_ptr<AudioSource> asrc(boost::dynamic_pointer_cast<AudioSource>(*si));
162                         asrc->mark_for_remove ();
163                 }
164         }
165         
166         return ret;
167 #endif
168 }