X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=libs%2Fardour%2Fimport.cc;h=67ed6da941c2e683042cb5d4304b9a9154262eba;hb=a2897ecef6da6a458aa1de8c2d9973a1e809dca2;hp=56b2aec6485da39b7226e8df54d152388a8c7e6a;hpb=9e8afdd306e373028f17ac8c244633664786216b;p=ardour.git diff --git a/libs/ardour/import.cc b/libs/ardour/import.cc index 56b2aec648..67ed6da941 100644 --- a/libs/ardour/import.cc +++ b/libs/ardour/import.cc @@ -1,22 +1,26 @@ /* - Copyright (C) 2000 Paul Davis + Copyright (C) 2000 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 - the Free Software Foundation; either version 2 of the License, or - (at your option) any later version. + 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 + the Free Software Foundation; either version 2 of the License, or + (at your option) any later version. - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU General Public License for more details. + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. - You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ +#ifdef WAF_BUILD +#include "libardour-config.h" +#endif + #include #include #include @@ -25,6 +29,7 @@ #include #include #include +#include #include #include @@ -34,69 +39,242 @@ #include #include -#include -#include - -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include +#include "pbd/basename.h" +#include "pbd/convert.h" + +#include "evoral/SMF.hpp" + +#include "ardour/analyser.h" +#include "ardour/ardour.h" +#include "ardour/audio_diskstream.h" +#include "ardour/audioengine.h" +#include "ardour/audioregion.h" +#include "ardour/import_status.h" +#include "ardour/region_factory.h" +#include "ardour/resampled_source.h" +#include "ardour/session.h" +#include "ardour/session_directory.h" +#include "ardour/smf_source.h" +#include "ardour/sndfile_helpers.h" +#include "ardour/sndfileimportable.h" +#include "ardour/sndfilesource.h" +#include "ardour/source_factory.h" +#include "ardour/tempo.h" + +#ifdef HAVE_COREAUDIO +#include "ardour/caimportable.h" +#endif #include "i18n.h" +using namespace std; using namespace ARDOUR; using namespace PBD; -std::string -get_non_existent_filename (const std::string& basename, uint channel, uint channels) +static boost::shared_ptr +open_importable_source (const string& path, framecnt_t samplerate, ARDOUR::SrcQuality quality) +{ + /* try libsndfile first, because it can get BWF info from .wav, which ExtAudioFile cannot. + We don't necessarily need that information in an ImportableSource, but it keeps the + logic the same as in SourceFactory::create() + */ + + try { + boost::shared_ptr source(new SndFileImportableSource(path)); + + if (source->samplerate() == samplerate) { + return source; + } + + /* rewrap as a resampled source */ + + return boost::shared_ptr(new ResampledImportableSource(source, samplerate, quality)); + } + + catch (...) { + +#ifdef HAVE_COREAUDIO + + /* libsndfile failed, see if we can use CoreAudio to handle the IO */ + + CAImportableSource* src = new CAImportableSource(path); + boost::shared_ptr source (src); + + if (source->samplerate() == samplerate) { + return source; + } + + /* rewrap as a resampled source */ + + return boost::shared_ptr(new ResampledImportableSource(source, samplerate, quality)); + +#else + throw; // rethrow +#endif + + } +} + +static std::string +get_non_existent_filename (HeaderFormat hf, DataType type, const bool allow_replacing, const std::string& destdir, const std::string& basename, uint channel, uint channels) { char buf[PATH_MAX+1]; bool goodfile = false; - string base(basename); + string base = basename; + string ext = native_header_format_extension (hf, type); + uint32_t cnt = 1; do { - if (channels == 2) { + + if (type == DataType::AUDIO && channels == 2) { if (channel == 0) { - snprintf (buf, sizeof(buf), "%s-L.wav", base.c_str()); + if (cnt == 1) { + snprintf (buf, sizeof(buf), "%s-L%s", base.c_str(), ext.c_str()); + } else { + snprintf (buf, sizeof(buf), "%s-%d-L%s", base.c_str(), cnt, ext.c_str()); + } } else { - snprintf (buf, sizeof(buf), "%s-R.wav", base.c_str()); + if (cnt == 1) { + snprintf (buf, sizeof(buf), "%s-R%s", base.c_str(), ext.c_str()); + } else { + snprintf (buf, sizeof(buf), "%s-%d-R%s", base.c_str(), cnt, ext.c_str()); + } } } else if (channels > 1) { - snprintf (buf, sizeof(buf), "%s-c%d.wav", base.c_str(), channel+1); + if (cnt == 1) { + snprintf (buf, sizeof(buf), "%s-c%d%s", base.c_str(), channel, ext.c_str()); + } else { + snprintf (buf, sizeof(buf), "%s-%d-c%d%s", base.c_str(), cnt, channel, ext.c_str()); + } } else { - snprintf (buf, sizeof(buf), "%s.wav", base.c_str()); + if (cnt == 1) { + snprintf (buf, sizeof(buf), "%s%s", base.c_str(), ext.c_str()); + } else { + snprintf (buf, sizeof(buf), "%s-%d%s", base.c_str(), cnt, ext.c_str()); + } } - if (sys::exists (buf)) { + string tempname = destdir + "/" + buf; - /* if the file already exists, we must come up with - * a new name for it. for now we just keep appending - * _ to basename - */ + if (!allow_replacing && Glib::file_test (tempname, Glib::FILE_TEST_EXISTS)) { - base += "_"; + cnt++; } else { goodfile = true; } - } while ( !goodfile); + } while (!goodfile); return buf; } -void -write_audio_data_to_new_files (ImportableSource* source, Session::import_status& status, - vector >& newfiles) +static vector +get_paths_for_new_sources (HeaderFormat hf, const bool allow_replacing, const string& import_file_path, const string& session_dir, uint channels) +{ + vector new_paths; + const string basename = basename_nosuffix (import_file_path); + + SessionDirectory sdir(session_dir); + + for (uint n = 0; n < channels; ++n) { + + const DataType type = SMFSource::safe_midi_file_extension (import_file_path) ? DataType::MIDI : DataType::AUDIO; + + std::string filepath = (type == DataType::MIDI) + ? sdir.midi_path().to_string() : sdir.sound_path().to_string(); + + filepath = Glib::build_filename (filepath, + get_non_existent_filename (hf, type, allow_replacing, filepath, basename, n, channels)); + new_paths.push_back (filepath); + } + + return new_paths; +} + +static bool +map_existing_mono_sources (const vector& new_paths, Session& /*sess*/, + uint /*samplerate*/, vector >& newfiles, Session *session) +{ + for (vector::const_iterator i = new_paths.begin(); + i != new_paths.end(); ++i) + { + boost::shared_ptr source = session->source_by_path_and_channel(*i, 0); + + if (source == 0) { + error << string_compose(_("Could not find a source for %1 even though we are updating this file!"), (*i)) << endl; + return false; + } + + newfiles.push_back(boost::dynamic_pointer_cast(source)); + } + return true; +} + +static bool +create_mono_sources_for_writing (const string& origin, + const vector& new_paths, + Session& sess, uint samplerate, + vector >& newfiles, + framepos_t timeline_position) +{ + for (vector::const_iterator i = new_paths.begin(); i != new_paths.end(); ++i) { + + boost::shared_ptr source; + + try { + const DataType type = SMFSource::safe_midi_file_extension (*i) ? DataType::MIDI : DataType::AUDIO; + + source = SourceFactory::createWritable (type, sess, + i->c_str(), + origin, + false, // destructive + samplerate); + } + + catch (const failed_constructor& err) { + error << string_compose (_("Unable to create file %1 during import"), *i) << endmsg; + return false; + } + + newfiles.push_back(boost::dynamic_pointer_cast(source)); + + /* for audio files, reset the timeline position so that any BWF-ish + information in the original files we are importing from is maintained. + */ + + boost::shared_ptr afs; + if ((afs = boost::dynamic_pointer_cast(source)) != 0) { + afs->set_timeline_position(timeline_position); + } + } + return true; +} + +static string +compose_status_message (const string& path, + uint file_samplerate, + uint session_samplerate, + uint /* current_file */, + uint /* total_files */) +{ + if (file_samplerate != session_samplerate) { + return string_compose (_("Resampling %1 from %2kHz to %3kHz"), + Glib::path_get_basename (path), + file_samplerate/1000.0f, + session_samplerate/1000.0f); + } + + return string_compose (_("Copying %1"), Glib::path_get_basename (path)); +} + +static void +write_audio_data_to_new_files (ImportableSource* source, ImportStatus& status, + vector >& newfiles) { - const nframes_t nframes = ResampledImportableSource::blocksize; + const framecnt_t nframes = ResampledImportableSource::blocksize; + boost::shared_ptr afs; uint channels = source->channels(); boost::scoped_array data(new float[nframes * channels]); @@ -105,26 +283,72 @@ write_audio_data_to_new_files (ImportableSource* source, Session::import_status& for (uint n = 0; n < channels; ++n) { channel_data.push_back(boost::shared_array(new Sample[nframes])); } - - uint read_count = 0; + + float gain = 1; + + boost::shared_ptr s = boost::dynamic_pointer_cast (newfiles[0]); + assert (s); + status.progress = 0.0f; + float progress_multiplier = 1; + float progress_base = 0; + + if (!source->clamped_at_unity() && s->clamped_at_unity()) { + + /* The source we are importing from can return sample values with a magnitude greater than 1, + and the file we are writing the imported data to cannot handle such values. Compute the gain + factor required to normalize the input sources to have a magnitude of less than 1. + */ + + float peak = 0; + uint read_count = 0; + + while (!status.cancel) { + framecnt_t const nread = source->read (data.get(), nframes); + if (nread == 0) { + break; + } + + peak = compute_peak (data.get(), nread, peak); + + read_count += nread; + status.progress = 0.5 * read_count / (source->ratio() * source->length() * channels); + } + + if (peak >= 1) { + /* we are out of range: compute a gain to fix it */ + gain = (1 - FLT_EPSILON) / peak; + } + + source->seek (0); + progress_multiplier = 0.5; + progress_base = 0.5; + } + + uint read_count = 0; while (!status.cancel) { - nframes_t nread, nfread; + framecnt_t nread, nfread; uint x; uint chn; if ((nread = source->read (data.get(), nframes)) == 0) { break; } + + if (gain != 1) { + /* here is the gain fix for out-of-range sample values that we computed earlier */ + apply_gain_to_buffer (data.get(), nread, gain); + } + nfread = nread / channels; /* de-interleave */ for (chn = 0; chn < channels; ++chn) { - nframes_t n; + framecnt_t n; for (x = chn, n = 0; n < nfread; x += channels, ++n) { channel_data[chn][n] = (Sample) data[x]; } @@ -133,142 +357,243 @@ write_audio_data_to_new_files (ImportableSource* source, Session::import_status& /* flush to disk */ for (chn = 0; chn < channels; ++chn) { - newfiles[chn]->write (channel_data[chn].get(), nfread); + if ((afs = boost::dynamic_pointer_cast(newfiles[chn])) != 0) { + afs->write (channel_data[chn].get(), nfread); + } } read_count += nread; - status.progress = read_count / (source->ratio () * source->length() * channels); + status.progress = progress_base + progress_multiplier * read_count / (source->ratio () * source->length() * channels); } } -int -Session::import_audiofile (import_status& status) +static void +write_midi_data_to_new_files (Evoral::SMF* source, ImportStatus& status, + vector >& newfiles) { - vector > newfiles; - SF_INFO info; - int nfiles = 0; - string basepath; - int ret = -1; - vector new_paths; - struct tm* now; - ImportableSource* importable = 0; - uint32_t cnt = 1; + uint32_t buf_size = 4; + uint8_t* buf = (uint8_t*) malloc (buf_size); - status.sources.clear (); - - for (vector::iterator p = status.paths.begin(); p != status.paths.end(); ++p, ++cnt) { + status.progress = 0.0f; - boost::shared_ptr in (sf_open (p->c_str(), SFM_READ, &info), sf_close); + assert (newfiles.size() == source->num_tracks()); - if (!in) { - error << string_compose(_("Import: cannot open input sound file \"%1\""), (*p)) << endmsg; - status.done = 1; - status.cancel = 1; - return -1; - } - - if ((nframes_t) info.samplerate != frame_rate()) { - importable = new ResampledImportableSource (in.get(), &info, frame_rate(), status.quality); - } else { - importable = new ImportableSource (in.get(), &info); - } - - newfiles.clear (); + try { + vector >::iterator s = newfiles.begin(); - for (int n = 0; n < info.channels; ++n) { - newfiles.push_back (boost::shared_ptr()); - } - - SessionDirectory sdir(get_best_session_directory_for_new_source ()); + for (unsigned i = 1; i <= source->num_tracks(); ++i) { + + boost::shared_ptr smfs = boost::dynamic_pointer_cast (*s); + + smfs->drop_model (); + source->seek_to_track (i); + + uint64_t t = 0; + uint32_t delta_t = 0; + uint32_t size = 0; + bool first = true; + + while (!status.cancel) { + gint note_id_ignored; // imported files either don't have NoteID's or we ignore them. + + size = buf_size; + + int ret = source->read_event (&delta_t, &size, &buf, ¬e_id_ignored); + + if (size > buf_size) { + buf_size = size; + } + + if (ret < 0) { // EOT + break; + } + + t += delta_t; - basepath = PBD::basename_nosuffix ((*p)); - - for (int n = 0; n < info.channels; ++n) { + if (ret == 0) { // Meta + continue; + } - std::string filename = get_non_existent_filename (basepath, n, info.channels); + if (first) { + smfs->mark_streaming_write_started (); + first = false; + } - sys::path filepath = sdir.sound_path() / filename; + smfs->append_event_unlocked_beats(Evoral::Event(0, + (double)t / (double)source->ppqn(), + size, + buf)); - try { - newfiles[n] = boost::dynamic_pointer_cast (SourceFactory::createWritable (DataType::AUDIO, *this, filepath.to_string().c_str(), false, frame_rate())); + if (status.progress < 0.99) { + status.progress += 0.01; + } } - catch (failed_constructor& err) { - error << string_compose(_("Session::import_audiofile: cannot open new file source for channel %1"), n+1) << endmsg; - goto out; + if (!first) { + + /* we wrote something */ + + const framepos_t pos = 0; + const double length_beats = ceil(t / (double)source->ppqn()); + BeatsFramesConverter converter(smfs->session().tempo_map(), pos); + smfs->update_length(pos + converter.to(length_beats)); + smfs->mark_streaming_write_completed (); + + if (status.cancel) { + break; + } + } else { + warning << string_compose (_("Track %1 of %2 contained no usable MIDI data"), i, source->file_path()) << endmsg; } - new_paths.push_back (filepath.to_string()); - newfiles[n]->prepare_for_peakfile_writes (); - nfiles++; + ++s; // next source } - if ((nframes_t) info.samplerate != frame_rate()) { - status.doing_what = string_compose (_("converting %1\n(resample from %2KHz to %3KHz)\n(%4 of %5)"), - basepath, - info.samplerate/1000.0f, - frame_rate()/1000.0f, - cnt, status.paths.size()); - - } else { - status.doing_what = string_compose (_("converting %1\n(%2 of %3)"), - basepath, - cnt, status.paths.size()); + } catch (...) { + error << string_compose (_("MIDI file %1 was not readable (no reason available"), source->file_path()) << endmsg; + } - } + if (buf) { + free (buf); + } +} - write_audio_data_to_new_files (importable, status, newfiles); +static void +remove_file_source (boost::shared_ptr source) +{ + boost::shared_ptr fs = boost::dynamic_pointer_cast (source); - if (status.cancel) { - goto out; + if (fs) { + ::unlink (fs->path().c_str()); + } +} + +// This function is still unable to cleanly update an existing source, even though +// it is possible to set the ImportStatus flag accordingly. The functinality +// is disabled at the GUI until the Source implementations are able to provide +// the necessary API. +void +Session::import_audiofiles (ImportStatus& status) +{ + typedef vector > Sources; + Sources all_new_sources; + boost::shared_ptr afs; + boost::shared_ptr smfs; + uint channels = 0; + + status.sources.clear (); + + for (vector::iterator p = status.paths.begin(); + p != status.paths.end() && !status.cancel; + ++p) + { + boost::shared_ptr source; + std::auto_ptr smf_reader; + const DataType type = SMFSource::safe_midi_file_extension (*p) ? DataType::MIDI : DataType::AUDIO; + + if (type == DataType::AUDIO) { + try { + source = open_importable_source (*p, frame_rate(), status.quality); + channels = source->channels(); + } catch (const failed_constructor& err) { + error << string_compose(_("Import: cannot open input sound file \"%1\""), (*p)) << endmsg; + status.done = status.cancel = true; + return; + } + + } else { + try { + smf_reader = std::auto_ptr(new Evoral::SMF()); + smf_reader->open(*p); + channels = smf_reader->num_tracks(); + } catch (...) { + error << _("Import: error opening MIDI file") << endmsg; + status.done = status.cancel = true; + return; + } } - - for (int n = 0; n < info.channels; ++n) { - status.sources.push_back (newfiles[n]); + + vector new_paths = get_paths_for_new_sources (config.get_native_file_header_format(), + status.replace_existing_source, *p, + get_best_session_directory_for_new_source (), + channels); + Sources newfiles; + framepos_t natural_position = source ? source->natural_position() : 0; + + + if (status.replace_existing_source) { + fatal << "THIS IS NOT IMPLEMENTED YET, IT SHOULD NEVER GET CALLED!!! DYING!" << endmsg; + status.cancel = !map_existing_mono_sources (new_paths, *this, frame_rate(), newfiles, this); + } else { + status.cancel = !create_mono_sources_for_writing (*p, new_paths, *this, frame_rate(), newfiles, natural_position); } + // copy on cancel/failure so that any files that were created will be removed below + std::copy (newfiles.begin(), newfiles.end(), std::back_inserter(all_new_sources)); + if (status.cancel) { - goto out; + break; } - } - - status.freeze = true; - time_t xnow; - time (&xnow); - now = localtime (&xnow); + for (Sources::iterator i = newfiles.begin(); i != newfiles.end(); ++i) { + if ((afs = boost::dynamic_pointer_cast(*i)) != 0) { + afs->prepare_for_peakfile_writes (); + } + } - /* flush the final length(s) to the header(s) */ + if (source) { // audio + status.doing_what = compose_status_message (*p, source->samplerate(), + frame_rate(), status.current, status.total); + write_audio_data_to_new_files (source.get(), status, newfiles); + } else if (smf_reader.get()) { // midi + status.doing_what = string_compose(_("Loading MIDI file %1"), *p); + write_midi_data_to_new_files (smf_reader.get(), status, newfiles); + } - for (SourceList::iterator x = status.sources.begin(); x != status.sources.end() && !status.cancel; ++x) { - boost::dynamic_pointer_cast(*x)->update_header(0, *now, xnow); - boost::dynamic_pointer_cast(*x)->done_with_peakfile_writes (); + ++status.current; + status.progress = 0; } - /* save state so that we don't lose these new Sources */ - if (!status.cancel) { - save_state (_name); - } + struct tm* now; + time_t xnow; + time (&xnow); + now = localtime (&xnow); + status.freeze = true; + + /* flush the final length(s) to the header(s) */ - ret = 0; + for (Sources::iterator x = all_new_sources.begin(); x != all_new_sources.end(); ) { + if ((afs = boost::dynamic_pointer_cast(*x)) != 0) { + afs->update_header((*x)->natural_position(), *now, xnow); + afs->done_with_peakfile_writes (); - out: + /* now that there is data there, requeue the file for analysis */ - if (status.cancel) { + if (Config->get_auto_analyse_audio()) { + Analyser::queue_source_for_analysis (boost::static_pointer_cast(*x), false); + } + } - status.sources.clear (); + /* don't create tracks for empty MIDI sources (channels) */ - for (vector::iterator i = new_paths.begin(); i != new_paths.end(); ++i) { - unlink ((*i).c_str()); + if ((smfs = boost::dynamic_pointer_cast(*x)) != 0 && smfs->is_empty()) { + x = all_new_sources.erase(x); + } else { + ++x; + } } - } - if (importable) { - delete importable; + /* save state so that we don't lose these new Sources */ + + save_state (_name); + + std::copy (all_new_sources.begin(), all_new_sources.end(), std::back_inserter(status.sources)); + } else { + // this can throw...but it seems very unlikely + std::for_each (all_new_sources.begin(), all_new_sources.end(), remove_file_source); } status.done = true; - - return ret; } +