gtk2_ardour/editor_pt_import.cc

   1 /*
   2     Copyright (C) 2000-2006 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 <sys/types.h>
  21 #include <sys/stat.h>
  22 #include <sys/time.h>
  23 #include <errno.h>
  24 #include <unistd.h>
  25 #include <algorithm>
  26
  27 #include "pbd/pthread_utils.h"
  28 #include "pbd/basename.h"
  29 #include "pbd/shortpath.h"
  30 #include "pbd/stateful_diff_command.h"
  31
  32 #include "ardour/audio_track.h"
  33 #include "ardour/audiofilesource.h"
  34 #include "ardour/audioregion.h"
  35 #include "ardour/midi_region.h"
  36 #include "ardour/midi_track.h"
  37 #include "ardour/midi_model.h"
  38 #include "ardour/operations.h"
  39 #include "ardour/region_factory.h"
  40 #include "ardour/smf_source.h"
  41 #include "ardour/source_factory.h"
  42 #include "ardour/utils.h"
  43 #include "ardour/playlist.h"
  44 #include "ardour/session.h"
  45 #include "pbd/memento_command.h"
  46
  47 #include "ptformat/ptfformat.h"
  48
  49 #include "ardour_ui.h"
  50 #include "cursor_context.h"
  51 #include "editor.h"
  52 #include "sfdb_ui.h"
  53 #include "editing.h"
  54 #include "audio_time_axis.h"
  55 #include "midi_time_axis.h"
  56 #include "session_import_dialog.h"
  57 #include "gui_thread.h"
  58 #include "interthread_progress_window.h"
  59 #include "mouse_cursors.h"
  60 #include "editor_cursors.h"
  61
  62 #include "pbd/i18n.h"
  63
  64 using namespace std;
  65 using namespace ARDOUR;
  66 using namespace PBD;
  67 using namespace Gtk;
  68 using namespace Gtkmm2ext;
  69 using namespace Editing;
  70 using std::string;
  71
  72 /* Functions supporting the incorporation of PT sessions into ardour */
  73
  74 void
  75 Editor::external_pt_dialog ()
  76 {
  77         std::string ptpath;
  78
  79         if (_session == 0) {
  80                 MessageDialog msg (_("You can't import a PT session until you have a session loaded."));
  81                 msg.run ();
  82                 return;
  83         }
  84
  85         Gtk::FileChooserDialog dialog(_("Import PT Session"), FILE_CHOOSER_ACTION_OPEN);
  86         dialog.add_button(Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);
  87         dialog.add_button(Gtk::Stock::OK, Gtk::RESPONSE_OK);
  88
  89         while (true) {
  90                 int result = dialog.run();
  91
  92                 if (result == Gtk::RESPONSE_OK) {
  93                         ptpath = dialog.get_filename ();
  94
  95                         if (!Glib::file_test (ptpath, Glib::FILE_TEST_IS_DIR|Glib::FILE_TEST_EXISTS)) {
  96                                 Gtk::MessageDialog msg (string_compose (_("%1: this is only the directory/folder name, not the filename.\n"), ptpath));
  97                                 msg.run ();
  98                                 continue;
  99                         }
 100                 }
 101
 102                 if (ptpath.length()) {
 103                         do_ptimport(ptpath, SrcBest);
 104                         break;
 105                 }
 106
 107                 if (result == Gtk::RESPONSE_CANCEL) {
 108                         break;
 109                 }
 110         }
 111 }
 112
 113 void
 114 Editor::do_ptimport (std::string ptpath,
 115                       SrcQuality  quality)
 116 {
 117         vector<boost::shared_ptr<Region> > regions;
 118         boost::shared_ptr<ARDOUR::Track> track;
 119         ARDOUR::PluginInfoPtr instrument;
 120         vector<string> to_import;
 121         string fullpath;
 122         bool ok = false;
 123         bool onefailed = false;
 124         PTFFormat ptf;
 125         samplepos_t pos = -1;
 126         uint32_t srate = _session->sample_rate();
 127
 128         vector<ptflookup_t> ptfwavpair;
 129         vector<ptflookup_t> ptfregpair;
 130
 131         if (ptf.load(ptpath, srate) == -1) {
 132                 MessageDialog msg (_("Doesn't seem to be a valid PT session file"));
 133                 msg.run ();
 134                 return;
 135         } else {
 136                 MessageDialog msg (string_compose (_("PT v%1 Session @ %2Hz\n\n%3 audio files\n%4 audio regions\n%5 active audio regions\n%6 midi regions\n%7 active midi regions\n\nContinue..."), (int)ptf.version, ptf.sessionrate, ptf.audiofiles.size(), ptf.regions.size() - ptf.midiregions.size(), ptf.tracks.size(), ptf.midiregions.size(), ptf.miditracks.size()));
 137                 msg.add_button(Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);
 138
 139                 int result = msg.run ();
 140                 if (result != Gtk::RESPONSE_OK) {
 141                         return;
 142                 }
 143         }
 144         current_interthread_info = &import_status;
 145         import_status.current = 1;
 146         import_status.total = ptf.audiofiles.size ();
 147         import_status.all_done = false;
 148
 149         ImportProgressWindow ipw (&import_status, _("Import"), _("Cancel Import"));
 150
 151         SourceList just_one;
 152         SourceList imported;
 153
 154         for (vector<PTFFormat::wav_t>::iterator a = ptf.audiofiles.begin(); a != ptf.audiofiles.end(); ++a) {
 155                 ptflookup_t p;
 156
 157                 fullpath = Glib::build_filename (Glib::path_get_dirname(ptpath), "Audio Files");
 158                 fullpath = Glib::build_filename (fullpath, a->filename);
 159                 to_import.clear ();
 160                 to_import.push_back (fullpath);
 161                 ipw.show ();
 162                 ok = import_sndfiles (to_import, Editing::ImportDistinctFiles, Editing::ImportAsRegion, quality, pos, 1, -1, track, false, instrument);
 163                 if (!import_status.sources.empty()) {
 164                         p.index1 = a->index;
 165                         p.id = import_status.sources.back()->id();
 166
 167                         ptfwavpair.push_back(p);
 168                         imported.push_back(import_status.sources.back());
 169                 } else {
 170                         onefailed = true;
 171                 }
 172         }
 173
 174         if (onefailed) {
 175                 MessageDialog msg (_("Failed to load one or more of the audio files, but continuing to attempt import."));
 176                 msg.run ();
 177         } else {
 178                 MessageDialog msg (_("Success! Import should complete soon."));
 179                 msg.run ();
 180         }
 181
 182         for (vector<PTFFormat::region_t>::iterator a = ptf.regions.begin();
 183                         a != ptf.regions.end(); ++a) {
 184                 for (vector<ptflookup_t>::iterator p = ptfwavpair.begin();
 185                                 p != ptfwavpair.end(); ++p) {
 186                         if ((p->index1 == a->wave.index) && (strcmp(a->wave.filename.c_str(), "") != 0)) {
 187                                 for (SourceList::iterator x = imported.begin();
 188                                                 x != imported.end(); ++x) {
 189                                         if ((*x)->id() == p->id) {
 190                                                 // Matched an uncreated ptf region to ardour region
 191                                                 ptflookup_t rp;
 192                                                 PropertyList plist;
 193
 194                                                 plist.add (ARDOUR::Properties::start, a->sampleoffset);
 195                                                 plist.add (ARDOUR::Properties::position, 0);
 196                                                 plist.add (ARDOUR::Properties::length, a->length);
 197                                                 plist.add (ARDOUR::Properties::name, a->name);
 198                                                 plist.add (ARDOUR::Properties::layer, 0);
 199                                                 plist.add (ARDOUR::Properties::whole_file, false);
 200                                                 plist.add (ARDOUR::Properties::external, true);
 201
 202                                                 just_one.clear();
 203                                                 just_one.push_back(*x);
 204
 205                                                 boost::shared_ptr<Region> r = RegionFactory::create (just_one, plist);
 206                                                 regions.push_back(r);
 207
 208                                                 rp.id = regions.back()->id();
 209                                                 rp.index1 = a->index;
 210                                                 ptfregpair.push_back(rp);
 211                                         }
 212                                 }
 213                         }
 214                 }
 215         }
 216
 217         boost::shared_ptr<AudioTrack> existing_track;
 218         uint16_t nth = 0;
 219         vector<ptflookup_t> usedtracks;
 220         ptflookup_t utr;
 221
 222         for (vector<PTFFormat::track_t>::iterator a = ptf.tracks.begin();
 223                         a != ptf.tracks.end(); ++a) {
 224                 for (vector<ptflookup_t>::iterator p = ptfregpair.begin();
 225                                 p != ptfregpair.end(); ++p) {
 226
 227                         if (p->index1 == a->reg.index)  {
 228                                 // Matched a ptf active region to an ardour region
 229                                 utr.index1 = a->index;
 230                                 utr.index2 = nth;
 231                                 utr.id = p->id;
 232                                 boost::shared_ptr<Region> r = RegionFactory::region_by_id (p->id);
 233                                 vector<ptflookup_t>::iterator lookuptr = usedtracks.begin();
 234                                 vector<ptflookup_t>::iterator found;
 235                                 if ((found = std::find(lookuptr, usedtracks.end(), utr)) != usedtracks.end()) {
 236                                         DEBUG_TRACE (DEBUG::FileUtils, string_compose ("\twav(%1) reg(%2) ptf_tr(%3) ard_tr(%4)\n", a->reg.wave.filename.c_str(), a->reg.index, found->index1, found->index2));
 237                                         existing_track =  get_nth_selected_audio_track(found->index2);
 238                                         // Put on existing track
 239                                         boost::shared_ptr<Playlist> playlist = existing_track->playlist();
 240                                         boost::shared_ptr<Region> copy (RegionFactory::create (r, true));
 241                                         playlist->clear_changes ();
 242                                         playlist->add_region (copy, a->reg.startpos);
 243                                         //_session->add_command (new StatefulDiffCommand (playlist));
 244                                 } else {
 245                                         // Put on a new track
 246                                         DEBUG_TRACE (DEBUG::FileUtils, string_compose ("\twav(%1) reg(%2) new_tr(%3)\n", a->reg.wave.filename.c_str(), a->reg.index, nth));
 247                                         list<boost::shared_ptr<AudioTrack> > at (_session->new_audio_track (1, 2, 0, 1, string(), PresentationInfo::max_order, Normal));
 248                                         if (at.empty()) {
 249                                                 return;
 250                                         }
 251                                         existing_track = at.back();
 252                                         std::string trackname;
 253                                         try {
 254                                                 trackname = Glib::convert_with_fallback (a->name, "UTF-8", "UTF-8", "_");
 255                                         } catch (Glib::ConvertError& err) {
 256                                                 trackname = string_compose ("Invalid %1", a->index);
 257                                         }
 258                                         // TODO legalize track name (no slashes, no colons)
 259 #if 0 // TODO --  "find_route_name" is currently private
 260                                         /* generate a unique name by adding a number if needed */
 261                                         uint32_t id = 0;
 262                                         if (!_session->find_route_name (trackname.c_str (), id, trackname, false)) {
 263                                                 fatal << _("PTImport: UINT_MAX routes? impossible!") << endmsg;
 264                                                 abort(); /*NOTREACHED*/
 265                                         }
 266 #endif
 267                                         existing_track->set_name (trackname);
 268                                         boost::shared_ptr<Playlist> playlist = existing_track->playlist();
 269                                         boost::shared_ptr<Region> copy (RegionFactory::create (r, true));
 270                                         playlist->clear_changes ();
 271                                         playlist->add_region (copy, a->reg.startpos);
 272                                         //_session->add_command (new StatefulDiffCommand (playlist));
 273                                         nth++;
 274                                 }
 275                                 usedtracks.push_back(utr);
 276                         }
 277                 }
 278         }
 279
 280         // MIDI - Find list of unique midi tracks first
 281         typedef struct midipair {
 282                 uint16_t ptfindex;
 283                 string trname;
 284         } midipair_t;
 285
 286         vector<midipair_t> uniquetr;
 287         uint16_t ith = 0;
 288         bool found;
 289
 290         for (vector<PTFFormat::track_t>::iterator a = ptf.miditracks.begin();
 291                         a != ptf.miditracks.end(); ++a) {
 292                 found = false;
 293                 for (vector<midipair_t>::iterator b = uniquetr.begin();
 294                                 b != uniquetr.end(); ++b) {
 295                         if (b->trname == a->name) {
 296                                 found = true;
 297                                 break;
 298                         }
 299                 }
 300                 if (!found) {
 301                         uniquetr.push_back({ith, a->name});
 302                         //printf(" : %d : %s\n", ith, a->name.c_str());
 303                         ith++;
 304                 }
 305         }
 306
 307         // MIDI - Create unique midi tracks and a lookup table for used tracks
 308         for (vector<midipair_t>::iterator a = uniquetr.begin();
 309                         a != uniquetr.end(); ++a) {
 310                 ptflookup_t miditr;
 311                 list<boost::shared_ptr<MidiTrack> > mt (_session->new_midi_track (
 312                                 ChanCount (DataType::MIDI, 1),
 313                                 ChanCount (DataType::MIDI, 1),
 314                                 true,
 315                                 instrument, (Plugin::PresetRecord*) 0,
 316                                 (RouteGroup*) 0,
 317                                 1,
 318                                 a->trname,
 319                                 PresentationInfo::max_order,
 320                                 Normal));
 321         }
 322
 323         // Select all MIDI tracks by selecting all tracks then subsetting to midi
 324         select_all_tracks();
 325
 326         TrackViewList miditracks;
 327         MidiTimeAxisView* mtv;
 328         TrackSelection::iterator x;
 329
 330         for (x = selection->tracks.begin(); x != selection->tracks.end(); ++x) {
 331
 332                 mtv = dynamic_cast<MidiTimeAxisView*>(*x);
 333
 334                 if (!mtv) {
 335                         continue;
 336                 } else if (mtv->is_midi_track()) {
 337                         miditracks.push_back(dynamic_cast<TimeAxisView*>(*x));
 338                 }
 339         }
 340
 341         selection->set(miditracks);
 342
 343         // MIDI - Add midi regions one-by-one to corresponding midi tracks
 344         for (vector<PTFFormat::track_t>::iterator a = ptf.miditracks.begin();
 345                         a != ptf.miditracks.end(); ++a) {
 346
 347                 boost::shared_ptr<MidiTrack> midi_track = get_nth_selected_midi_track(a->index);
 348                 boost::shared_ptr<Playlist> playlist = midi_track->playlist();
 349                 samplepos_t f = (samplepos_t)a->reg.startpos * srate / 1920000.;
 350                 samplecnt_t length = (samplecnt_t)a->reg.length * srate / 1920000.;
 351                 MusicSample pos (f, 0);
 352                 boost::shared_ptr<Source> src = _session->create_midi_source_by_stealing_name (midi_track);
 353                 PropertyList plist;
 354                 plist.add (ARDOUR::Properties::start, 0);
 355                 plist.add (ARDOUR::Properties::length, length);
 356                 plist.add (ARDOUR::Properties::name, PBD::basename_nosuffix(src->name()));
 357                 //printf(" : %d - trackname: (%s)\n", a->index, src->name().c_str());
 358                 boost::shared_ptr<Region> region = (RegionFactory::create (src, plist));
 359                 // sets beat position
 360                 region->set_position (pos.sample, pos.division);
 361                 midi_track->playlist()->add_region (region, pos.sample, 1.0, false, pos.division);
 362
 363                 boost::shared_ptr<MidiRegion> mr = boost::dynamic_pointer_cast<MidiRegion>(region);
 364                 boost::shared_ptr<MidiModel> mm = mr->midi_source(0)->model();
 365                 MidiModel::NoteDiffCommand *midicmd;
 366                 midicmd = mm->new_note_diff_command ("Import ProTools MIDI");
 367
 368                 for (vector<PTFFormat::midi_ev_t>::iterator
 369                                 j = a->reg.midi.begin();
 370                                 j != a->reg.midi.end(); ++j) {
 371                         //printf(" : MIDI : pos=%f len=%f\n", (float)j->pos / 960000., (float)j->length / 960000.);
 372                         Temporal::Beats start = (Temporal::Beats)(j->pos / 960000.);
 373                         Temporal::Beats len = (Temporal::Beats)(j->length / 960000.);
 374                         // PT C-2 = 0, Ardour C-1 = 0, subtract twelve to convert ?
 375                         midicmd->add(boost::shared_ptr<Evoral::Note<Temporal::Beats> >
 376                                 (new Evoral::Note<Temporal::Beats>( (uint8_t)1, start, len, j->note, j->velocity )));
 377                 }
 378                 mm->apply_command (_session, midicmd);
 379                 boost::shared_ptr<Region> copy (RegionFactory::create (mr, true));
 380                 playlist->clear_changes ();
 381                 playlist->add_region (copy, f);
 382         }
 383
 384         import_status.sources.clear();
 385
 386         if (ok) {
 387                 _session->save_state ("");
 388         }
 389         import_status.all_done = true;
 390 }