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 (); x != imported.end (); ++x) {
 188                                         if ((*x)->id () == p->id) {
 189                                                 /* Matched an uncreated ptf region to ardour region */
 190                                                 ptflookup_t rp;
 191                                                 PropertyList plist;
 192
 193                                                 plist.add (ARDOUR::Properties::start, a->sampleoffset);
 194                                                 plist.add (ARDOUR::Properties::position, 0);
 195                                                 plist.add (ARDOUR::Properties::length, a->length);
 196                                                 plist.add (ARDOUR::Properties::name, a->name);
 197                                                 plist.add (ARDOUR::Properties::layer, 0);
 198                                                 plist.add (ARDOUR::Properties::whole_file, false);
 199                                                 plist.add (ARDOUR::Properties::external, true);
 200
 201                                                 just_one.clear ();
 202                                                 just_one.push_back (*x);
 203
 204                                                 boost::shared_ptr<Region> r = RegionFactory::create (just_one, plist);
 205                                                 regions.push_back (r);
 206
 207                                                 rp.id = regions.back ()->id ();
 208                                                 rp.index1 = a->index;
 209                                                 ptfregpair.push_back (rp);
 210                                         }
 211                                 }
 212                         }
 213                 }
 214         }
 215
 216         boost::shared_ptr<AudioTrack> existing_track;
 217         uint16_t nth = 0;
 218         vector<ptflookup_t> usedtracks;
 219         ptflookup_t utr;
 220
 221         for (vector<PTFFormat::track_t>::iterator a = ptf.tracks.begin (); a != ptf.tracks.end (); ++a) {
 222                 for (vector<ptflookup_t>::iterator p = ptfregpair.begin ();
 223                                 p != ptfregpair.end (); ++p) {
 224
 225                         if (p->index1 == a->reg.index)  {
 226                                 /* Matched a ptf active region to an ardour region */
 227                                 utr.index1 = a->index;
 228                                 utr.index2 = nth;
 229                                 utr.id = p->id;
 230                                 boost::shared_ptr<Region> r = RegionFactory::region_by_id (p->id);
 231                                 vector<ptflookup_t>::iterator lookuptr = usedtracks.begin ();
 232                                 vector<ptflookup_t>::iterator found;
 233                                 if ((found = std::find (lookuptr, usedtracks.end (), utr)) != usedtracks.end ()) {
 234                                         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));
 235                                         existing_track =  get_nth_selected_audio_track (found->index2);
 236                                         /* Put on existing track */
 237                                         boost::shared_ptr<Playlist> playlist = existing_track->playlist ();
 238                                         boost::shared_ptr<Region> copy (RegionFactory::create (r, true));
 239                                         playlist->clear_changes ();
 240                                         playlist->add_region (copy, a->reg.startpos);
 241                                         //_session->add_command (new StatefulDiffCommand (playlist));
 242                                 } else {
 243                                         /* Put on a new track */
 244                                         DEBUG_TRACE (DEBUG::FileUtils, string_compose ("\twav(%1) reg(%2) new_tr(%3)\n", a->reg.wave.filename.c_str (), a->reg.index, nth));
 245                                         list<boost::shared_ptr<AudioTrack> > at (_session->new_audio_track (1, 2, 0, 1, "", PresentationInfo::max_order, Normal));
 246                                         if (at.empty ()) {
 247                                                 return;
 248                                         }
 249                                         existing_track = at.back ();
 250                                         std::string trackname;
 251                                         try {
 252                                                 trackname = Glib::convert_with_fallback (a->name, "UTF-8", "UTF-8", "_");
 253                                         } catch (Glib::ConvertError& err) {
 254                                                 trackname = string_compose ("Invalid %1", a->index);
 255                                         }
 256                                         /* TODO legalize track name (no slashes, no colons) */
 257 #if 0 /* TODO --  "find_route_name" is currently private */
 258                                         /* generate a unique name by adding a number if needed */
 259                                         uint32_t id = 0;
 260                                         if (!_session->find_route_name (trackname.c_str (), id, trackname, false)) {
 261                                                 fatal << _("PTImport: UINT_MAX routes? impossible!") << endmsg;
 262                                                 abort(); /*NOTREACHED*/
 263                                         }
 264 #endif
 265                                         existing_track->set_name (trackname);
 266                                         boost::shared_ptr<Playlist> playlist = existing_track->playlist();
 267                                         boost::shared_ptr<Region> copy (RegionFactory::create (r, true));
 268                                         playlist->clear_changes ();
 269                                         playlist->add_region (copy, a->reg.startpos);
 270                                         //_session->add_command (new StatefulDiffCommand (playlist));
 271                                         nth++;
 272                                 }
 273                                 usedtracks.push_back (utr);
 274                         }
 275                 }
 276         }
 277
 278         /* MIDI - Find list of unique midi tracks first */
 279         typedef struct midipair {
 280                 uint16_t ptfindex;
 281                 string trname;
 282         } midipair_t;
 283
 284         vector<midipair_t> uniquetr;
 285         uint16_t ith = 0;
 286         bool found;
 287
 288         for (vector<PTFFormat::track_t>::iterator a = ptf.miditracks.begin (); a != ptf.miditracks.end (); ++a) {
 289                 found = false;
 290                 for (vector<midipair_t>::iterator b = uniquetr.begin (); b != uniquetr.end (); ++b) {
 291                         if (b->trname == a->name) {
 292                                 found = true;
 293                                 break;
 294                         }
 295                 }
 296                 if (!found) {
 297                         uniquetr.push_back ({ith, a->name});
 298                         //printf(" : %d : %s\n", ith, a->name.c_str());
 299                         ith++;
 300                 }
 301         }
 302
 303         /* MIDI - Create unique midi tracks and a lookup table for used tracks */
 304         for (vector<midipair_t>::iterator a = uniquetr.begin ();
 305                         a != uniquetr.end (); ++a) {
 306                 ptflookup_t miditr;
 307                 list<boost::shared_ptr<MidiTrack> > mt (_session->new_midi_track (
 308                                 ChanCount (DataType::MIDI, 1),
 309                                 ChanCount (DataType::MIDI, 1),
 310                                 true,
 311                                 instrument, (Plugin::PresetRecord*) 0,
 312                                 (RouteGroup*) 0,
 313                                 1,
 314                                 a->trname,
 315                                 PresentationInfo::max_order,
 316                                 Normal));
 317         }
 318
 319         /* Select all MIDI tracks by selecting all tracks then subsetting to midi */
 320         select_all_tracks ();
 321
 322         TrackViewList miditracks;
 323         MidiTimeAxisView* mtv;
 324         TrackSelection::iterator x;
 325
 326         for (x = selection->tracks.begin (); x != selection->tracks.end (); ++x) {
 327
 328                 mtv = dynamic_cast<MidiTimeAxisView*>(*x);
 329
 330                 if (!mtv) {
 331                         continue;
 332                 } else if (mtv->is_midi_track ()) {
 333                         miditracks.push_back (dynamic_cast<TimeAxisView*> (*x));
 334                 }
 335         }
 336
 337         selection->set (miditracks);
 338
 339         /* MIDI - Add midi regions one-by-one to corresponding midi tracks */
 340         for (vector<PTFFormat::track_t>::iterator a = ptf.miditracks.begin (); a != ptf.miditracks.end (); ++a) {
 341
 342                 boost::shared_ptr<MidiTrack> midi_track = get_nth_selected_midi_track (a->index);
 343                 boost::shared_ptr<Playlist> playlist = midi_track->playlist ();
 344                 samplepos_t f = (samplepos_t)a->reg.startpos * srate / 1920000.;
 345                 samplecnt_t length = (samplecnt_t)a->reg.length * srate / 1920000.;
 346                 MusicSample pos (f, 0);
 347                 boost::shared_ptr<Source> src = _session->create_midi_source_by_stealing_name (midi_track);
 348                 PropertyList plist;
 349                 plist.add (ARDOUR::Properties::start, 0);
 350                 plist.add (ARDOUR::Properties::length, length);
 351                 plist.add (ARDOUR::Properties::name, PBD::basename_nosuffix (src->name ()));
 352                 //printf(" : %d - trackname: (%s)\n", a->index, src->name ().c_str ());
 353                 boost::shared_ptr<Region> region = (RegionFactory::create (src, plist));
 354                 /* sets beat position */
 355                 region->set_position (pos.sample, pos.division);
 356                 midi_track->playlist ()->add_region (region, pos.sample, 1.0, false, pos.division);
 357
 358                 boost::shared_ptr<MidiRegion> mr = boost::dynamic_pointer_cast<MidiRegion>(region);
 359                 boost::shared_ptr<MidiModel> mm = mr->midi_source (0)->model ();
 360                 MidiModel::NoteDiffCommand *midicmd;
 361                 midicmd = mm->new_note_diff_command ("Import ProTools MIDI");
 362
 363                 for (vector<PTFFormat::midi_ev_t>::iterator j = a->reg.midi.begin (); j != a->reg.midi.end (); ++j) {
 364                         //printf(" : MIDI : pos=%f len=%f\n", (float)j->pos / 960000., (float)j->length / 960000.);
 365                         Temporal::Beats start = (Temporal::Beats)(j->pos / 960000.);
 366                         Temporal::Beats len = (Temporal::Beats)(j->length / 960000.);
 367                         /* PT C-2 = 0, Ardour C-1 = 0, subtract twelve to convert ? */
 368                         midicmd->add (boost::shared_ptr<Evoral::Note<Temporal::Beats> > (new Evoral::Note<Temporal::Beats> ((uint8_t)1, start, len, j->note, j->velocity)));
 369                 }
 370                 mm->apply_command (_session, midicmd);
 371                 boost::shared_ptr<Region> copy (RegionFactory::create (mr, true));
 372                 playlist->clear_changes ();
 373                 playlist->add_region (copy, f);
 374         }
 375
 376         import_status.sources.clear ();
 377
 378         if (ok) {
 379                 _session->save_state ("");
 380         }
 381         import_status.all_done = true;
 382 }