Merged with trunk R992.
[ardour.git] / gtk2_ardour / export_range_markers_dialog.cc
1 /*
2     Copyright (C) 2006 Paul Davis
3     Author: Andre Raue
4
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License
16     along with this program; if not, write to the Free Software
17     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18
19 */
20
21 #include <sys/stat.h>
22
23 #include <sstream>
24
25 #include <ardour/audioengine.h>
26 #include <ardour/sndfile_helpers.h>
27
28 #include "ardour_ui.h"
29 #include "export_range_markers_dialog.h"
30
31 #include "i18n.h"
32
33 using namespace Gtk;
34 using namespace ARDOUR;
35 using namespace PBD;
36 using namespace std;
37
38 ExportRangeMarkersDialog::ExportRangeMarkersDialog (PublicEditor& editor) 
39         : ExportDialog(editor)
40
41         do_not_allow_export_cd_markers();
42         
43         total_duration = 0;
44         current_range_marker_index = 0;
45 }
46         
47         
48 void 
49 ExportRangeMarkersDialog::export_audio_data ()
50 {
51         getSession().locations()->apply(*this, &ExportRangeMarkersDialog::process_range_markers_export);
52 }
53
54 void
55 ExportRangeMarkersDialog::process_range_markers_export(Locations::LocationList& locations)
56 {
57         Locations::LocationList::iterator locationIter;
58         current_range_marker_index = 0;
59         init_progress_computing(locations);
60         
61         for (locationIter = locations.begin(); locationIter != locations.end(); ++locationIter) {
62                 Location *currentLocation = (*locationIter);
63
64                 if(currentLocation->is_range_marker()){
65                         // init filename
66                         string filepath = get_target_filepath(
67                                 get_selected_file_name(),
68                                 currentLocation->name(),
69                                 sndfile_file_ending_from_string(get_selected_header_format()));
70                         
71                         initSpec(filepath);
72                         
73                         spec.start_frame = currentLocation->start();
74                         spec.end_frame = currentLocation->end();
75
76                         getSession().request_locate(spec.start_frame, false);
77
78                         if (getSession().start_audio_export(spec)){
79                                 // if export fails                      
80                                 return;
81                         }
82
83                         // wait until export of this range finished
84                         gtk_main_iteration();
85                         while(spec.running){
86                                 if(gtk_events_pending()){
87                                         gtk_main_iteration();
88                                 }else {
89                                         usleep(10000);
90                                 }
91                         }
92                         
93                         getSession().engine().freewheel (false);
94                         current_range_marker_index++;
95                 }
96         }
97         
98         spec.running = false;
99 }
100
101
102 string
103 ExportRangeMarkersDialog::get_target_filepath(string path, string filename, string postfix)
104 {
105         string target_path = path;
106         if ((target_path.find_last_of ('/')) != string::npos) {
107                 target_path += '/';
108         }
109
110         string target_filepath = target_path + filename + postfix;
111         struct stat statbuf;
112         
113         for(int counter=1; (stat (target_filepath.c_str(), &statbuf) == 0); counter++){
114                 // while file exists    
115                 ostringstream scounter;
116                 scounter.flush();
117                 scounter << counter;
118                 
119                 target_filepath = 
120                         target_path + filename + "_" + scounter.str() + postfix;
121         }
122         
123         return target_filepath;
124 }
125
126
127 bool
128 ExportRangeMarkersDialog::is_filepath_valid(string &filepath)
129 {
130         // sanity check file name first
131         struct stat statbuf;
132   
133         if (filepath.empty()) {
134                 // warning dialog
135                 string txt = _("Please enter a valid target directory.");
136                 MessageDialog msg (*this, txt, false, MESSAGE_ERROR, BUTTONS_OK, true);
137                 msg.run();
138                 return false;
139         }
140         
141         if ( (stat (filepath.c_str(), &statbuf) != 0) || 
142                 (!S_ISDIR (statbuf.st_mode)) ) {
143                         string txt = _("Please select an existing target directory. Files\n"
144                                 "are not allowed!");
145                         MessageDialog msg (*this, txt, false, MESSAGE_ERROR, BUTTONS_OK, true);
146                         msg.run();
147                         return false;
148         }
149         
150         // directory needs to exist and be writable
151         string dirpath = Glib::path_get_dirname (filepath);
152         if (::access (dirpath.c_str(), W_OK) != 0) {
153                 string txt = _("Cannot write file in: ") + dirpath;
154                 MessageDialog msg (*this, txt, false, MESSAGE_ERROR, BUTTONS_OK, true);
155                 msg.run();
156                 return false;
157         }
158         
159         return true;
160 }
161
162
163 void
164 ExportRangeMarkersDialog::init_progress_computing(Locations::LocationList& locations)
165 {
166         // flush vector
167         range_markers_durations_aggregated.resize(0);
168         
169         nframes_t duration_before_current_location = 0;
170         Locations::LocationList::iterator locationIter;
171                 
172         for (locationIter = locations.begin(); locationIter != locations.end(); ++locationIter) {
173                 Location *currentLocation = (*locationIter);
174                 
175                 if(currentLocation->is_range_marker()){
176                         range_markers_durations_aggregated.push_back(
177                                 duration_before_current_location);
178                         
179                         nframes_t duration = 
180                                 currentLocation->end() - currentLocation->start();
181                         
182                         range_markers_durations.push_back(duration);
183                         duration_before_current_location += duration;   
184                 }
185         }
186
187         total_duration = duration_before_current_location;      
188 }
189
190
191 gint
192 ExportRangeMarkersDialog::progress_timeout ()
193 {
194         double progress = 0.0;
195         
196         if(current_range_marker_index >= range_markers_durations.size()){
197                 progress = 1.0;
198         }
199         else{
200                 progress = 
201                         ((double) range_markers_durations_aggregated[current_range_marker_index] +
202                         (spec.progress * (double) range_markers_durations[current_range_marker_index])) /
203                         (double) total_duration;
204         }
205         
206         set_progress_fraction( progress );
207         return TRUE;
208 }