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.
31 #include <samplerate.h>
35 #include <boost/scoped_array.hpp>
36 #include <boost/shared_array.hpp>
38 #include "pbd/basename.h"
39 #include "pbd/convert.h"
41 #include "evoral/SMF.hpp"
43 #include "ardour/ardour.h"
44 #include "ardour/session.h"
45 #include "ardour/session_directory.h"
46 #include "ardour/audio_diskstream.h"
47 #include "ardour/audioengine.h"
48 #include "ardour/sndfilesource.h"
49 #include "ardour/sndfile_helpers.h"
50 #include "ardour/audioregion.h"
51 #include "ardour/region_factory.h"
52 #include "ardour/source_factory.h"
53 #include "ardour/resampled_source.h"
54 #include "ardour/sndfileimportable.h"
55 #include "ardour/analyser.h"
56 #include "ardour/smf_source.h"
57 #include "ardour/tempo.h"
60 #ifdef USE_COREAUDIO_FOR_FILE_IO
61 #include "ardour/caimportable.h"
68 using namespace ARDOUR;
71 static boost::shared_ptr<ImportableSource>
72 open_importable_source (const string& path, nframes_t samplerate, ARDOUR::SrcQuality quality)
75 #ifdef USE_COREAUDIO_FOR_FILE_IO
77 /* see if we can use CoreAudio to handle the IO */
80 CAImportableSource* src = new CAImportableSource(path);
81 boost::shared_ptr<CAImportableSource> source (src);
83 if (source->samplerate() == samplerate) {
87 /* rewrap as a resampled source */
89 return boost::shared_ptr<ImportableSource>(new ResampledImportableSource(source, samplerate, quality));
94 /* fall back to SndFile */
99 boost::shared_ptr<SndFileImportableSource> source(new SndFileImportableSource(path));
101 if (source->samplerate() == samplerate) {
105 /* rewrap as a resampled source */
107 return boost::shared_ptr<ImportableSource>(new ResampledImportableSource(source, samplerate, quality));
114 #ifdef HAVE_COREAUDIO
115 #ifdef USE_COREAUDIO_FOR_FILE_IO
122 get_non_existent_filename (DataType type, const bool allow_replacing, const std::string destdir, const std::string& basename, uint channel, uint channels)
124 char buf[PATH_MAX+1];
125 bool goodfile = false;
126 string base(basename);
127 const char* ext = (type == DataType::AUDIO) ? "wav" : "mid";
131 if (type == DataType::AUDIO && channels == 2) {
133 snprintf (buf, sizeof(buf), "%s-L.wav", base.c_str());
135 snprintf (buf, sizeof(buf), "%s-R.wav", base.c_str());
137 } else if (channels > 1) {
138 snprintf (buf, sizeof(buf), "%s-c%d.%s", base.c_str(), channel, ext);
140 snprintf (buf, sizeof(buf), "%s.%s", base.c_str(), ext);
144 string tempname = destdir + "/" + buf;
145 if (!allow_replacing && Glib::file_test (tempname, Glib::FILE_TEST_EXISTS)) {
147 /* if the file already exists, we must come up with
148 * a new name for it. for now we just keep appending
159 } while ( !goodfile);
164 static vector<string>
165 get_paths_for_new_sources (const bool allow_replacing, const string& import_file_path, const string& session_dir, uint channels)
167 vector<string> new_paths;
168 const string basename = basename_nosuffix (import_file_path);
170 SessionDirectory sdir(session_dir);
172 for (uint n = 0; n < channels; ++n) {
174 const DataType type = (import_file_path.rfind(".mid") != string::npos)
175 ? DataType::MIDI : DataType::AUDIO;
177 std::string filepath = (type == DataType::MIDI)
178 ? sdir.midi_path().to_string() : sdir.sound_path().to_string();
181 filepath += get_non_existent_filename (type, allow_replacing, filepath, basename, n, channels);
182 new_paths.push_back (filepath);
189 map_existing_mono_sources (const vector<string>& new_paths, Session& sess,
190 uint samplerate, vector<boost::shared_ptr<Source> >& newfiles, Session *session)
192 for (vector<string>::const_iterator i = new_paths.begin();
193 i != new_paths.end(); ++i)
195 boost::shared_ptr<Source> source = session->source_by_path_and_channel(*i, 0);
198 error << string_compose(_("Could not find a source for %1 even though we are updating this file!"), (*i)) << endl;
202 newfiles.push_back(boost::dynamic_pointer_cast<Source>(source));
208 create_mono_sources_for_writing (const vector<string>& new_paths, Session& sess,
209 uint samplerate, vector<boost::shared_ptr<Source> >& newfiles)
211 for (vector<string>::const_iterator i = new_paths.begin();
212 i != new_paths.end(); ++i)
214 boost::shared_ptr<Source> source;
218 const DataType type = ((*i).rfind(".mid") != string::npos)
219 ? DataType::MIDI : DataType::AUDIO;
221 source = SourceFactory::createWritable (type, sess,
223 false, // destructive
226 catch (const failed_constructor& err)
228 error << string_compose (_("Unable to create file %1 during import"), *i) << endmsg;
232 newfiles.push_back(boost::dynamic_pointer_cast<Source>(source));
238 compose_status_message (const string& path,
239 uint file_samplerate,
240 uint session_samplerate,
244 if (file_samplerate != session_samplerate) {
245 return string_compose (_("converting %1\n(resample from %2KHz to %3KHz)\n(%4 of %5)"),
246 Glib::path_get_basename (path),
247 file_samplerate/1000.0f,
248 session_samplerate/1000.0f,
249 current_file, total_files);
252 return string_compose (_("converting %1\n(%2 of %3)"),
253 Glib::path_get_basename (path),
254 current_file, total_files);
258 write_audio_data_to_new_files (ImportableSource* source, Session::ImportStatus& status,
259 vector<boost::shared_ptr<Source> >& newfiles)
261 const nframes_t nframes = ResampledImportableSource::blocksize;
262 boost::shared_ptr<AudioFileSource> afs;
263 uint channels = source->channels();
265 boost::scoped_array<float> data(new float[nframes * channels]);
266 vector<boost::shared_array<Sample> > channel_data;
268 for (uint n = 0; n < channels; ++n) {
269 channel_data.push_back(boost::shared_array<Sample>(new Sample[nframes]));
273 status.progress = 0.0f;
275 while (!status.cancel) {
277 nframes_t nread, nfread;
281 if ((nread = source->read (data.get(), nframes)) == 0) {
284 nfread = nread / channels;
288 for (chn = 0; chn < channels; ++chn) {
291 for (x = chn, n = 0; n < nfread; x += channels, ++n) {
292 channel_data[chn][n] = (Sample) data[x];
298 for (chn = 0; chn < channels; ++chn) {
299 if ((afs = boost::dynamic_pointer_cast<AudioFileSource>(newfiles[chn])) != 0) {
300 afs->write (channel_data[chn].get(), nfread);
305 status.progress = read_count / (source->ratio () * source->length() * channels);
310 write_midi_data_to_new_files (Evoral::SMF* source, Session::ImportStatus& status,
311 vector<boost::shared_ptr<Source> >& newfiles)
313 uint32_t buf_size = 4;
314 uint8_t* buf = (uint8_t*)malloc(buf_size);
316 status.progress = 0.0f;
320 for (unsigned i = 1; i <= source->num_tracks(); ++i) {
321 boost::shared_ptr<SMFSource> smfs = boost::dynamic_pointer_cast<SMFSource>(newfiles[i-1]);
324 source->seek_to_track(i);
327 uint32_t delta_t = 0;
330 while (!status.cancel) {
333 int ret = source->read_event(&delta_t, &size, &buf);
337 if (ret < 0) { // EOT
343 if (ret == 0) { // Meta
347 smfs->append_event_unlocked_beats(Evoral::Event<double>(0,
348 (double)t / (double)source->ppqn(),
352 if (status.progress < 0.99)
353 status.progress += 0.01;
356 const nframes64_t pos = 0;
357 const double length_beats = ceil(t / (double)source->ppqn());
358 BeatsFramesConverter converter(smfs->session(), pos);
359 smfs->update_length(pos, converter.to(length_beats));
368 error << "Corrupt MIDI file " << source->file_path() << endl;
373 remove_file_source (boost::shared_ptr<Source> source)
375 ::unlink (source->path().c_str());
378 // This function is still unable to cleanly update an existing source, even though
379 // it is possible to set the ImportStatus flag accordingly. The functinality
380 // is disabled at the GUI until the Source implementations are able to provide
381 // the necessary API.
383 Session::import_audiofiles (ImportStatus& status)
386 typedef vector<boost::shared_ptr<Source> > Sources;
387 Sources all_new_sources;
388 boost::shared_ptr<AudioFileSource> afs;
389 boost::shared_ptr<SMFSource> smfs;
392 status.sources.clear ();
394 for (vector<Glib::ustring>::iterator p = status.paths.begin();
395 p != status.paths.end() && !status.cancel;
398 boost::shared_ptr<ImportableSource> source;
399 std::auto_ptr<Evoral::SMF> smf_reader;
400 const DataType type = ((*p).rfind(".mid") != string::npos) ?
401 DataType::MIDI : DataType::AUDIO;
403 if (type == DataType::AUDIO) {
405 source = open_importable_source (*p, frame_rate(), status.quality);
406 channels = source->channels();
407 } catch (const failed_constructor& err) {
408 error << string_compose(_("Import: cannot open input sound file \"%1\""), (*p)) << endmsg;
409 status.done = status.cancel = true;
415 smf_reader = std::auto_ptr<Evoral::SMF>(new Evoral::SMF());
416 smf_reader->open(*p);
417 channels = smf_reader->num_tracks();
419 error << _("Import: error opening MIDI file") << endmsg;
420 status.done = status.cancel = true;
425 vector<string> new_paths = get_paths_for_new_sources (status.replace_existing_source, *p,
426 get_best_session_directory_for_new_source (),
430 if (status.replace_existing_source) {
431 fatal << "THIS IS NOT IMPLEMENTED YET, IT SHOULD NEVER GET CALLED!!! DYING!" << endl;
432 status.cancel = !map_existing_mono_sources (new_paths, *this, frame_rate(), newfiles, this);
434 status.cancel = !create_mono_sources_for_writing (new_paths, *this, frame_rate(), newfiles);
437 // copy on cancel/failure so that any files that were created will be removed below
438 std::copy (newfiles.begin(), newfiles.end(), std::back_inserter(all_new_sources));
440 if (status.cancel) break;
442 for (Sources::iterator i = newfiles.begin(); i != newfiles.end(); ++i) {
443 if ((afs = boost::dynamic_pointer_cast<AudioFileSource>(*i)) != 0) {
444 afs->prepare_for_peakfile_writes ();
448 if (source) { // audio
449 status.doing_what = compose_status_message (*p, source->samplerate(),
450 frame_rate(), cnt, status.total);
451 write_audio_data_to_new_files (source.get(), status, newfiles);
452 } else if (smf_reader.get()) { // midi
453 status.doing_what = string_compose(_("Loading MIDI file %1"), *p);
454 write_midi_data_to_new_files (smf_reader.get(), status, newfiles);
458 if (!status.cancel) {
462 now = localtime (&xnow);
463 status.freeze = true;
465 /* flush the final length(s) to the header(s) */
467 for (Sources::iterator x = all_new_sources.begin(); x != all_new_sources.end(); ) {
468 if ((afs = boost::dynamic_pointer_cast<AudioFileSource>(*x)) != 0) {
469 afs->update_header(0, *now, xnow);
470 afs->done_with_peakfile_writes ();
472 /* now that there is data there, requeue the file for analysis */
474 if (Config->get_auto_analyse_audio()) {
475 Analyser::queue_source_for_analysis (boost::static_pointer_cast<Source>(*x), false);
479 /* don't create tracks for empty MIDI sources (channels) */
481 if ((smfs = boost::dynamic_pointer_cast<SMFSource>(*x)) != 0 && smfs->is_empty()) {
482 x = all_new_sources.erase(x);
488 /* save state so that we don't lose these new Sources */
492 std::copy (all_new_sources.begin(), all_new_sources.end(), std::back_inserter(status.sources));
494 // this can throw...but it seems very unlikely
495 std::for_each (all_new_sources.begin(), all_new_sources.end(), remove_file_source);