2 Copyright (C) 2000 Paul Davis
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.
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.
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.
30 #include <samplerate.h>
34 #include <boost/scoped_array.hpp>
35 #include <boost/shared_array.hpp>
37 #include <pbd/convert.h>
39 #include <ardour/ardour.h>
40 #include <ardour/session.h>
41 #include <ardour/session_directory.h>
42 #include <ardour/audio_diskstream.h>
43 #include <ardour/sndfilesource.h>
44 #include <ardour/sndfile_helpers.h>
45 #include <ardour/audioregion.h>
46 #include <ardour/region_factory.h>
47 #include <ardour/source_factory.h>
48 #include <ardour/resampled_source.h>
52 using namespace ARDOUR;
55 std::auto_ptr<ImportableSource>
56 open_importable_source (const string& path, nframes_t samplerate,
57 ARDOUR::SrcQuality quality)
59 std::auto_ptr<ImportableSource> source(new ImportableSource(path));
61 if (source->samplerate() == samplerate) {
65 return std::auto_ptr<ImportableSource>(
66 new ResampledImportableSource(path, samplerate, quality)
71 get_non_existent_filename (const std::string& basename, uint channel, uint channels)
74 bool goodfile = false;
75 string base(basename);
80 snprintf (buf, sizeof(buf), "%s-L.wav", base.c_str());
82 snprintf (buf, sizeof(buf), "%s-R.wav", base.c_str());
84 } else if (channels > 1) {
85 snprintf (buf, sizeof(buf), "%s-c%d.wav", base.c_str(), channel+1);
87 snprintf (buf, sizeof(buf), "%s.wav", base.c_str());
90 if (sys::exists (buf)) {
92 /* if the file already exists, we must come up with
93 * a new name for it. for now we just keep appending
104 } while ( !goodfile);
110 get_paths_for_new_sources (const string& import_file_path, const string& session_dir,
113 vector<string> new_paths;
114 const string basename = sys::basename (import_file_path);
115 SessionDirectory sdir(session_dir);
117 for (uint n = 0; n < channels; ++n) {
119 std::string filename = get_non_existent_filename (basename, n, channels);
121 sys::path filepath = sdir.sound_path() / filename;
123 new_paths.push_back (filepath.to_string());
130 create_mono_sources_for_writing (const vector<string>& new_paths, Session& sess,
131 uint samplerate, vector<boost::shared_ptr<AudioFileSource> >& newfiles)
133 for (vector<string>::const_iterator i = new_paths.begin();
134 i != new_paths.end(); ++i)
136 boost::shared_ptr<Source> source;
140 source = SourceFactory::createWritable (
144 false, // destructive
148 catch (const failed_constructor& err)
150 error << string_compose (_("Unable to create file %1 during import"), *i) << endmsg;
154 newfiles.push_back(boost::dynamic_pointer_cast<AudioFileSource>(source));
160 compose_status_message (const string& path,
161 uint file_samplerate,
162 uint session_samplerate,
166 if (file_samplerate != session_samplerate) {
167 return string_compose (_("converting %1\n(resample from %2KHz to %3KHz)\n(%4 of %5)"),
168 sys::path(path).leaf(),
169 file_samplerate/1000.0f,
170 session_samplerate/1000.0f,
171 current_file, total_files);
174 return string_compose (_("converting %1\n(%2 of %3)"),
175 sys::path(path).leaf(),
176 current_file, total_files);
180 write_audio_data_to_new_files (ImportableSource* source, Session::import_status& status,
181 vector<boost::shared_ptr<AudioFileSource> >& newfiles)
183 const nframes_t nframes = ResampledImportableSource::blocksize;
184 uint channels = source->channels();
186 boost::scoped_array<float> data(new float[nframes * channels]);
187 vector<boost::shared_array<Sample> > channel_data;
189 for (uint n = 0; n < channels; ++n) {
190 channel_data.push_back(boost::shared_array<Sample>(new Sample[nframes]));
194 status.progress = 0.0f;
196 while (!status.cancel) {
198 nframes_t nread, nfread;
202 if ((nread = source->read (data.get(), nframes)) == 0) {
205 nfread = nread / channels;
209 for (chn = 0; chn < channels; ++chn) {
212 for (x = chn, n = 0; n < nfread; x += channels, ++n) {
213 channel_data[chn][n] = (Sample) data[x];
219 for (chn = 0; chn < channels; ++chn) {
220 newfiles[chn]->write (channel_data[chn].get(), nfread);
224 status.progress = read_count / (source->ratio () * source->length() * channels);
229 remove_file_source (boost::shared_ptr<AudioFileSource> file_source)
231 sys::remove (std::string(file_source->path()));
235 Session::import_audiofile (import_status& status)
238 typedef vector<boost::shared_ptr<AudioFileSource> > AudioSources;
239 AudioSources all_new_sources;
241 status.sources.clear ();
243 for (vector<Glib::ustring>::iterator p = status.paths.begin();
244 p != status.paths.end() && !status.cancel;
247 std::auto_ptr<ImportableSource> source;
251 source = open_importable_source (*p, frame_rate(), status.quality);
253 catch (const failed_constructor& err)
255 error << string_compose(_("Import: cannot open input sound file \"%1\""), (*p)) << endmsg;
256 status.done = status.cancel = true;
260 vector<string> new_paths = get_paths_for_new_sources (*p,
261 get_best_session_directory_for_new_source (),
264 AudioSources newfiles;
266 status.cancel = !create_mono_sources_for_writing (new_paths, *this, frame_rate(), newfiles);
268 // copy on cancel/failure so that any files that were created will be removed below
269 std::copy (newfiles.begin(), newfiles.end(), std::back_inserter(all_new_sources));
271 if (status.cancel) break;
273 for (AudioSources::iterator i = newfiles.begin(); i != newfiles.end(); ++i) {
274 (*i)->prepare_for_peakfile_writes ();
277 status.doing_what = compose_status_message (*p, source->samplerate(),
278 frame_rate(), cnt, status.paths.size());
280 write_audio_data_to_new_files (source.get(), status, newfiles);
285 if (!status.cancel) {
289 now = localtime (&xnow);
290 status.freeze = true;
292 /* flush the final length(s) to the header(s) */
294 for (AudioSources::iterator x = all_new_sources.begin();
295 x != all_new_sources.end(); ++x)
297 (*x)->update_header(0, *now, xnow);
298 (*x)->done_with_peakfile_writes ();
301 /* save state so that we don't lose these new Sources */
305 std::copy (all_new_sources.begin(), all_new_sources.end(),
306 std::back_inserter(status.sources));
310 // this can throw...but it seems very unlikely
311 std::for_each (all_new_sources.begin(), all_new_sources.end(), remove_file_source);