first vaguely working version using PresentationInfo
[ardour.git] / 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 <gtkmm2ext/choice.h>
33
34 #include "ardour/audio_track.h"
35 #include "ardour/audiofilesource.h"
36 #include "ardour/audioregion.h"
37 #include "ardour/midi_region.h"
38 #include "ardour/midi_track.h"
39 #include "ardour/operations.h"
40 #include "ardour/region_factory.h"
41 #include "ardour/smf_source.h"
42 #include "ardour/source_factory.h"
43 #include "ardour/utils.h"
44 #include "ardour/playlist.h"
45 #include "ardour/session.h"
46 #include "pbd/memento_command.h"
47
48 #include "ptformat/ptfformat.h"
49
50 #include "ardour_ui.h"
51 #include "cursor_context.h"
52 #include "editor.h"
53 #include "sfdb_ui.h"
54 #include "editing.h"
55 #include "audio_time_axis.h"
56 #include "midi_time_axis.h"
57 #include "session_import_dialog.h"
58 #include "gui_thread.h"
59 #include "interthread_progress_window.h"
60 #include "mouse_cursors.h"
61 #include "editor_cursors.h"
62
63 #include "i18n.h"
64
65 using namespace std;
66 using namespace ARDOUR;
67 using namespace PBD;
68 using namespace Gtk;
69 using namespace Gtkmm2ext;
70 using namespace Editing;
71 using std::string;
72
73 /* Functions supporting the incorporation of PT sessions into ardour */
74
75 void
76 Editor::external_pt_dialog ()
77 {
78         std::string ptpath;
79
80         if (_session == 0) {
81                 MessageDialog msg (_("You can't import a PT session until you have a session loaded."));
82                 msg.run ();
83                 return;
84         }
85
86         Gtk::FileChooserDialog dialog(_("Import PT Session"), FILE_CHOOSER_ACTION_OPEN);
87         dialog.add_button(Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);
88         dialog.add_button(Gtk::Stock::OK, Gtk::RESPONSE_OK);
89
90         while (true) {
91                 int result = dialog.run();
92
93                 if (result == Gtk::RESPONSE_OK) {
94                         ptpath = dialog.get_filename ();
95
96                         if (!Glib::file_test (ptpath, Glib::FILE_TEST_IS_DIR|Glib::FILE_TEST_EXISTS)) {
97                                 Gtk::MessageDialog msg (string_compose (_("%1: this is only the directory/folder name, not the filename.\n"), ptpath));
98                                 msg.run ();
99                                 continue;
100                         }
101                 }
102
103                 if (ptpath.length()) {
104                         do_ptimport(ptpath, SrcBest);
105                         break;
106                 }
107
108                 if (result == Gtk::RESPONSE_CANCEL) {
109                         break;
110                 }
111         }
112 }
113
114 void
115 Editor::do_ptimport (std::string ptpath,
116                       SrcQuality  quality)
117 {
118         vector<boost::shared_ptr<Region> > regions;
119         boost::shared_ptr<ARDOUR::Track> track;
120         ARDOUR::PluginInfoPtr instrument;
121         vector<string> to_import;
122         string fullpath;
123         bool ok = false;
124         bool onefailed = false;
125         PTFFormat ptf;
126         framepos_t pos = -1;
127
128         vector<ptflookup_t> ptfwavpair;
129         vector<ptflookup_t> ptfregpair;
130
131         if (ptf.load(ptpath, _session->frame_rate()) == -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 regions\n%5 active regions\n\nContinue..."), (int)ptf.version, ptf.sessionrate, ptf.audiofiles.size(), ptf.regions.size(), ptf.tracks.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) {
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         import_status.sources.clear();
281
282         if (ok) {
283                 _session->save_state ("");
284         }
285         import_status.all_done = true;
286 }