Merged with trunk (painfully)
[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 std;
36
37 ExportRangeMarkersDialog::ExportRangeMarkersDialog (PublicEditor& editor) 
38         : ExportDialog(editor)
39
40         do_not_allow_export_cd_markers();
41         
42         total_duration = 0;
43         current_range_marker_index = 0;
44 }
45         
46         
47 void 
48 ExportRangeMarkersDialog::export_audio_data ()
49 {
50         getSession().locations()->apply(*this, &ExportRangeMarkersDialog::process_range_markers_export);
51 }
52
53 void
54 ExportRangeMarkersDialog::process_range_markers_export(Locations::LocationList& locations)
55 {
56         Locations::LocationList::iterator locationIter;
57         current_range_marker_index = 0;
58         init_progress_computing(locations);
59         
60         for (locationIter = locations.begin(); locationIter != locations.end(); ++locationIter) {
61                 Location *currentLocation = (*locationIter);
62
63                 if(currentLocation->is_range_marker()){
64                         // init filename
65                         string filepath = get_target_filepath(
66                                 get_selected_file_name(),
67                                 currentLocation->name(),
68                                 sndfile_file_ending_from_string(get_selected_header_format()));
69                         
70                         initSpec(filepath);
71                         
72                         spec.start_frame = currentLocation->start();
73                         spec.end_frame = currentLocation->end();
74
75                         getSession().request_locate(spec.start_frame, false);
76
77                         if (getSession().start_audio_export(spec)){
78                                 // if export fails                      
79                                 return;
80                         }
81
82                         // wait until export of this range finished
83                         gtk_main_iteration();
84                         while(spec.running){
85                                 if(gtk_events_pending()){
86                                         gtk_main_iteration();
87                                 }else {
88                                         usleep(10000);
89                                 }
90                         }
91                         
92                         getSession().engine().freewheel (false);
93                         current_range_marker_index++;
94                 }
95         }
96         
97         spec.running = false;
98 }
99
100
101 string
102 ExportRangeMarkersDialog::get_target_filepath(string path, string filename, string postfix)
103 {
104         string target_path = path;
105         if ((target_path.find_last_of ('/')) != string::npos) {
106                 target_path += '/';
107         }
108
109         string target_filepath = target_path + filename + postfix;
110         struct stat statbuf;
111         
112         for(int counter=1; (stat (target_filepath.c_str(), &statbuf) == 0); counter++){
113                 // while file exists    
114                 ostringstream scounter;
115                 scounter.flush();
116                 scounter << counter;
117                 
118                 target_filepath = 
119                         target_path + filename + "_" + scounter.str() + postfix;
120         }
121         
122         return target_filepath;
123 }
124
125
126 bool
127 ExportRangeMarkersDialog::is_filepath_valid(string &filepath)
128 {
129         // sanity check file name first
130         struct stat statbuf;
131   
132         if (filepath.empty()) {
133                 // warning dialog
134                 string txt = _("Please enter a valid target directory.");
135                 MessageDialog msg (*this, txt, false, MESSAGE_ERROR, BUTTONS_OK, true);
136                 msg.run();
137                 return false;
138         }
139         
140         if ( (stat (filepath.c_str(), &statbuf) != 0) || 
141                 (!S_ISDIR (statbuf.st_mode)) ) {
142                         string txt = _("Please select an existing target directory. Files\n"
143                                 "are not allowed!");
144                         MessageDialog msg (*this, txt, false, MESSAGE_ERROR, BUTTONS_OK, true);
145                         msg.run();
146                         return false;
147         }
148         
149         // directory needs to exist and be writable
150         string dirpath = Glib::path_get_dirname (filepath);
151         if (::access (dirpath.c_str(), W_OK) != 0) {
152                 string txt = _("Cannot write file in: ") + dirpath;
153                 MessageDialog msg (*this, txt, false, MESSAGE_ERROR, BUTTONS_OK, true);
154                 msg.run();
155                 return false;
156         }
157         
158         return true;
159 }
160
161
162 void
163 ExportRangeMarkersDialog::init_progress_computing(Locations::LocationList& locations)
164 {
165         // flush vector
166         range_markers_durations_aggregated.resize(0);
167         
168         jack_nframes_t duration_before_current_location = 0;
169         Locations::LocationList::iterator locationIter;
170                 
171         for (locationIter = locations.begin(); locationIter != locations.end(); ++locationIter) {
172                 Location *currentLocation = (*locationIter);
173                 
174                 if(currentLocation->is_range_marker()){
175                         range_markers_durations_aggregated.push_back(
176                                 duration_before_current_location);
177                         
178                         jack_nframes_t duration = 
179                                 currentLocation->end() - currentLocation->start();
180                         
181                         range_markers_durations.push_back(duration);
182                         duration_before_current_location += duration;   
183                 }
184         }
185
186         total_duration = duration_before_current_location;      
187 }
188
189
190 gint
191 ExportRangeMarkersDialog::progress_timeout ()
192 {
193         double progress = 0.0;
194         
195         if(current_range_marker_index >= range_markers_durations.size()){
196                 progress = 1.0;
197         }
198         else{
199                 progress = 
200                         ((double) range_markers_durations_aggregated[current_range_marker_index] +
201                         (spec.progress * (double) range_markers_durations[current_range_marker_index])) /
202                         (double) total_duration;
203         }
204         
205         set_progress_fraction( progress );
206         return TRUE;
207 }