radically rethink export/bounce/freeze code design. probably not 100% done by freeze...
[ardour.git] / libs / ardour / export_channel.cc
1 /*
2     Copyright (C) 2008 Paul Davis
3     Author: Sakari Bergen
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 "ardour/audio_buffer.h"
22 #include "ardour/audio_port.h"
23 #include "ardour/audio_track.h"
24 #include "ardour/audioengine.h"
25 #include "ardour/audioregion.h"
26 #include "ardour/capturing_processor.h"
27 #include "ardour/export_channel.h"
28 #include "ardour/export_failed.h"
29 #include "ardour/session.h"
30
31 #include "pbd/error.h"
32
33 #include "i18n.h"
34
35 using namespace ARDOUR;
36
37 PortExportChannel::PortExportChannel ()
38         : buffer_size(0)
39 {
40 }
41
42 void PortExportChannel::set_max_buffer_size(framecnt_t frames)
43 {
44         buffer_size = frames;
45         buffer.reset (new Sample[frames]);
46 }
47
48 bool
49 PortExportChannel::operator< (ExportChannel const & other) const
50 {
51         PortExportChannel const * pec;
52         if (!(pec = dynamic_cast<PortExportChannel const *> (&other))) {
53                 return this < &other;
54         }
55         return ports < pec->ports;
56 }
57
58 void
59 PortExportChannel::read (Sample const *& data, framecnt_t frames) const
60 {
61         assert(buffer);
62         assert(frames <= buffer_size);
63
64         if (ports.size() == 1) {
65                 boost::shared_ptr<AudioPort> p = ports.begin()->lock ();
66                 data = p->get_audio_buffer(frames).data();
67                 return;
68         }
69
70         memset (buffer.get(), 0, frames * sizeof (Sample));
71
72         for (PortSet::const_iterator it = ports.begin(); it != ports.end(); ++it) {
73                 boost::shared_ptr<AudioPort> p = it->lock ();
74                 if (p) {
75                         Sample* port_buffer = p->get_audio_buffer(frames).data();
76
77                         for (uint32_t i = 0; i < frames; ++i) {
78                                 buffer[i] += (float) port_buffer[i];
79                         }
80                 }
81         }
82
83         data = buffer.get();
84 }
85
86 void
87 PortExportChannel::get_state (XMLNode * node) const
88 {
89         XMLNode * port_node;
90         for (PortSet::const_iterator it = ports.begin(); it != ports.end(); ++it) {
91                 boost::shared_ptr<Port> p = it->lock ();
92                 if (p && (port_node = node->add_child ("Port"))) {
93                         port_node->add_property ("name", p->name());
94                 }
95         }
96 }
97
98 void
99 PortExportChannel::set_state (XMLNode * node, Session & session)
100 {
101         XMLProperty * prop;
102         XMLNodeList xml_ports = node->children ("Port");
103         for (XMLNodeList::iterator it = xml_ports.begin(); it != xml_ports.end(); ++it) {
104                 if ((prop = (*it)->property ("name"))) {
105                         std::string const & name = prop->value();
106                         boost::shared_ptr<AudioPort> port = boost::dynamic_pointer_cast<AudioPort> (session.engine().get_port_by_name (name));
107                         if (port) {
108                                 ports.insert (port);
109                         } else {
110                                 PBD::warning << string_compose (_("Could not get port for export channel \"%1\", dropping the channel"), name) << endmsg;
111                         }
112                 }
113         }
114 }
115
116 RegionExportChannelFactory::RegionExportChannelFactory (Session * session, AudioRegion const & region, AudioTrack & track, Type type) 
117         : region (region)
118         , track (track)
119         , type (type)
120         , frames_per_cycle (session->engine().frames_per_cycle ())
121         , buffers_up_to_date (false)
122         , region_start (region.position())
123         , position (region_start)
124 {
125         switch (type) {
126           case Raw:
127                 n_channels = region.n_channels();
128                 break;
129           case Fades:
130                 n_channels = region.n_channels();
131
132                 mixdown_buffer.reset (new Sample [frames_per_cycle]);
133                 gain_buffer.reset (new Sample [frames_per_cycle]);
134                 memset (gain_buffer.get(), 1.0, sizeof (Sample) * frames_per_cycle);
135
136                 break;
137           case Processed:
138                 n_channels = track.n_outputs().n_audio();
139                 break;
140           default:
141                 throw ExportFailed ("Unhandled type in ExportChannelFactory constructor");
142         }
143
144         session->ProcessExport.connect_same_thread (export_connection, boost::bind (&RegionExportChannelFactory::new_cycle_started, this, _1));
145
146         buffers.ensure_buffers (DataType::AUDIO, n_channels, frames_per_cycle);
147         buffers.set_count (ChanCount (DataType::AUDIO, n_channels));
148 }
149
150 RegionExportChannelFactory::~RegionExportChannelFactory ()
151 {
152 }
153
154 ExportChannelPtr
155 RegionExportChannelFactory::create (uint32_t channel)
156 {
157         assert (channel < n_channels);
158         return ExportChannelPtr (new RegionExportChannel (*this, channel));
159 }
160
161 void
162 RegionExportChannelFactory::read (uint32_t channel, Sample const *& data, framecnt_t frames_to_read)
163 {
164         assert (channel < n_channels);
165         assert (frames_to_read <= frames_per_cycle);
166
167         if (!buffers_up_to_date) {
168                 update_buffers(frames_to_read);
169                 buffers_up_to_date = true;
170         }
171
172         data = buffers.get_audio (channel).data();
173 }
174
175 void
176 RegionExportChannelFactory::update_buffers (framecnt_t frames)
177 {
178         assert (frames <= frames_per_cycle);
179
180         switch (type) {
181           case Raw:
182                 for (size_t channel = 0; channel < n_channels; ++channel) {
183                         region.read (buffers.get_audio (channel).data(), position - region_start, frames, channel);
184                 }
185                 break;
186           case Fades:
187                 assert (mixdown_buffer && gain_buffer);
188                 for (size_t channel = 0; channel < n_channels; ++channel) {
189                         memset (mixdown_buffer.get(), 0, sizeof (Sample) * frames);
190                         region.read_at (buffers.get_audio (channel).data(), mixdown_buffer.get(), gain_buffer.get(), position, frames, channel);
191                 }
192                 break;
193         case Processed:
194                 track.export_stuff (buffers, position, frames, track.main_outs(), true, true);
195                 break;
196         default:
197                 throw ExportFailed ("Unhandled type in ExportChannelFactory::update_buffers");
198         }
199
200         position += frames;
201 }
202
203
204 RouteExportChannel::RouteExportChannel(boost::shared_ptr<CapturingProcessor> processor, size_t channel,
205                                        boost::shared_ptr<ProcessorRemover> remover)
206   : processor (processor)
207   , channel (channel)
208   , remover (remover)
209 {
210 }
211
212 RouteExportChannel::~RouteExportChannel()
213 {
214 }
215
216 void
217 RouteExportChannel::create_from_route(std::list<ExportChannelPtr> & result, Route & route)
218 {
219         boost::shared_ptr<CapturingProcessor> processor = route.add_export_point();
220         uint32_t channels = processor->input_streams().n_audio();
221
222         boost::shared_ptr<ProcessorRemover> remover (new ProcessorRemover (route, processor));
223         result.clear();
224         for (uint32_t i = 0; i < channels; ++i) {
225                 result.push_back (ExportChannelPtr (new RouteExportChannel (processor, i, remover)));
226         }
227 }
228
229 void
230 RouteExportChannel::set_max_buffer_size(framecnt_t frames)
231 {
232         if (processor) {
233                 processor->set_block_size (frames);
234         }
235 }
236
237 void
238 RouteExportChannel::read (Sample const *& data, framecnt_t frames) const
239 {
240         assert(processor);
241         AudioBuffer const & buffer = processor->get_capture_buffers().get_audio (channel);
242         assert (frames <= (framecnt_t) buffer.size());
243         data = buffer.data();
244 }
245
246 void
247 RouteExportChannel::get_state (XMLNode *) const
248 {
249         // TODO
250 }
251
252 void
253 RouteExportChannel::set_state (XMLNode *, Session &)
254 {
255         // TODO
256 }
257
258 bool
259 RouteExportChannel::operator< (ExportChannel const & other) const
260 {
261         RouteExportChannel const * rec;
262         if ((rec = dynamic_cast<RouteExportChannel const *>(&other)) == 0) {
263                 return this < &other;
264         }
265
266         if (processor.get() == rec->processor.get()) {
267                 return channel < rec->channel;
268         }
269         return processor.get() < rec->processor.get();
270 }
271
272 RouteExportChannel::ProcessorRemover::~ProcessorRemover()
273 {
274         route.remove_processor (processor);
275 }