Fix windows build when using pthreads-win32 library
[ardour.git] / libs / ardour / export_graph_builder.cc
index 2be9044415f339690e368885dd3d7b2b7bdacd30..c054d85242d1f619fc46779101fc841ca1223587 100644 (file)
@@ -1,5 +1,29 @@
+/*
+  Copyright (C) 2008-2012 Paul Davis
+  Author: Sakari Bergen
+
+  This program is free software; you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation; either version 2 of the License, or
+  (at your option) any later version.
+
+  This program is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with this program; if not, write to the Free Software
+  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+*/
+
 #include "ardour/export_graph_builder.h"
 
+#include <vector>
+
+#include <glibmm/miscutils.h>
+
 #include "audiographer/process_context.h"
 #include "audiographer/general/chunker.h"
 #include "audiographer/general/interleaver.h"
@@ -17,6 +41,7 @@
 #include "ardour/export_filename.h"
 #include "ardour/export_format_specification.h"
 #include "ardour/export_timespan.h"
+#include "ardour/session_directory.h"
 #include "ardour/sndfile_helpers.h"
 
 #include "pbd/file_utils.h"
@@ -28,10 +53,10 @@ using std::string;
 namespace ARDOUR {
 
 ExportGraphBuilder::ExportGraphBuilder (Session const & session)
-  : session (session)
-  , thread_pool (hardware_concurrency())
+       : session (session)
+       , thread_pool (hardware_concurrency())
 {
-       process_buffer_frames = session.engine().frames_per_cycle();
+       process_buffer_frames = session.engine().samples_per_cycle();
 }
 
 ExportGraphBuilder::~ExportGraphBuilder ()
@@ -87,6 +112,17 @@ ExportGraphBuilder::reset ()
        normalizers.clear ();
 }
 
+void
+ExportGraphBuilder::cleanup (bool remove_out_files/*=false*/)
+{
+       ChannelConfigList::iterator iter = channel_configs.begin();
+
+       while (iter != channel_configs.end() ) {
+               iter->remove_children(remove_out_files);
+               iter = channel_configs.erase(iter);
+       }
+}
+
 void
 ExportGraphBuilder::set_current_timespan (boost::shared_ptr<ExportTimespan> span)
 {
@@ -106,7 +142,7 @@ ExportGraphBuilder::add_config (FileSpec const & config)
        // If the sample rate is "session rate", change it to the real value.
        // However, we need to copy it to not change the config which is saved...
        FileSpec new_config (config);
-       new_config.format.reset(new ExportFormatSpecification(*new_config.format));
+       new_config.format.reset(new ExportFormatSpecification(*new_config.format, false));
        if(new_config.format->sample_rate() == ExportFormatBase::SR_Session) {
                framecnt_t session_rate = session.nominal_frame_rate();
                new_config.format->set_sample_rate(ExportFormatBase::nearest_sample_rate(session_rate));
@@ -186,6 +222,33 @@ ExportGraphBuilder::Encoder::add_child (FileSpec const & new_config)
        filenames.push_back (new_config.filename);
 }
 
+void
+ExportGraphBuilder::Encoder::destroy_writer (bool delete_out_file)
+{
+       if (delete_out_file ) {
+
+               if (float_writer) {
+                       float_writer->close ();
+               }
+
+               if (int_writer) {
+                       int_writer->close ();
+               }
+
+               if (short_writer) {
+                       short_writer->close ();
+               }
+
+               if (std::remove(writer_filename.c_str() ) != 0) {
+                       std::cout << "Encoder::destroy_writer () : Error removing file: " << strerror(errno) << std::endl;
+               }
+       }
+
+       float_writer.reset ();
+       int_writer.reset ();
+       short_writer.reset ();
+}
+
 bool
 ExportGraphBuilder::Encoder::operator== (FileSpec const & other_config) const
 {
@@ -206,9 +269,9 @@ ExportGraphBuilder::Encoder::init_writer (boost::shared_ptr<AudioGrapher::Sndfil
        unsigned channels = config.channel_config->get_n_chans();
        int format = get_real_format (config);
        config.filename->set_channel_config(config.channel_config);
-       string filename = config.filename->get_path (config.format);
+       writer_filename = config.filename->get_path (config.format);
 
-       writer.reset (new AudioGrapher::SndfileWriter<T> (filename, format, channels, config.format->sample_rate(), config.broadcast_info));
+       writer.reset (new AudioGrapher::SndfileWriter<T> (writer_filename, format, channels, config.format->sample_rate(), config.broadcast_info));
        writer->FileWritten.connect_same_thread (copy_files_connection, boost::bind (&ExportGraphBuilder::Encoder::copy_files, this, _1));
 }
 
@@ -225,7 +288,7 @@ ExportGraphBuilder::Encoder::copy_files (std::string orig_path)
 /* SFC */
 
 ExportGraphBuilder::SFC::SFC (ExportGraphBuilder &, FileSpec const & new_config, framecnt_t max_frames)
-  : data_width(0)
+       : data_width(0)
 {
        config = new_config;
        data_width = sndfile_data_width (Encoder::get_real_format (config));
@@ -281,6 +344,20 @@ ExportGraphBuilder::SFC::add_child (FileSpec const & new_config)
        }
 }
 
+void
+ExportGraphBuilder::SFC::remove_children (bool remove_out_files)
+{
+       boost::ptr_list<Encoder>::iterator iter = children.begin ();
+
+       while (iter != children.end() ) {
+
+               if (remove_out_files) {
+                       iter->destroy_writer(remove_out_files);
+               }
+               iter = children.erase (iter);
+       }
+}
+
 bool
 ExportGraphBuilder::SFC::operator== (FileSpec const & other_config) const
 {
@@ -290,12 +367,18 @@ ExportGraphBuilder::SFC::operator== (FileSpec const & other_config) const
 /* Normalizer */
 
 ExportGraphBuilder::Normalizer::Normalizer (ExportGraphBuilder & parent, FileSpec const & new_config, framecnt_t /*max_frames*/)
-  : parent (parent)
+       : parent (parent)
 {
+       std::string tmpfile_path = parent.session.session_directory().export_path();
+       tmpfile_path = Glib::build_filename(tmpfile_path, "XXXXXX");
+       std::vector<char> tmpfile_path_buf(tmpfile_path.size() + 1);
+       std::copy(tmpfile_path.begin(), tmpfile_path.end(), tmpfile_path_buf.begin());
+       tmpfile_path_buf[tmpfile_path.size()] = '\0';
+
        config = new_config;
        uint32_t const channels = config.channel_config->get_n_chans();
        max_frames_out = 4086 - (4086 % channels); // TODO good chunk size
-       
+
        buffer.reset (new AllocatingProcessContext<Sample> (max_frames_out, channels));
        peak_reader.reset (new PeakReader ());
        normalizer.reset (new AudioGrapher::Normalizer (config.format->normalize_target()));
@@ -305,7 +388,7 @@ ExportGraphBuilder::Normalizer::Normalizer (ExportGraphBuilder & parent, FileSpe
        normalizer->add_output (threader);
 
        int format = ExportFormatBase::F_RAW | ExportFormatBase::SF_Float;
-       tmp_file.reset (new TmpFile<float> (format, channels, config.format->sample_rate()));
+       tmp_file.reset (new TmpFile<float> (&tmpfile_path_buf[0], format, channels, config.format->sample_rate()));
        tmp_file->FileWritten.connect_same_thread (post_processing_connection,
                                                   boost::bind (&Normalizer::start_post_processing, this));
 
@@ -334,11 +417,22 @@ ExportGraphBuilder::Normalizer::add_child (FileSpec const & new_config)
        threader->add_output (children.back().sink());
 }
 
+void
+ExportGraphBuilder::Normalizer::remove_children (bool remove_out_files)
+{
+       boost::ptr_list<SFC>::iterator iter = children.begin ();
+
+       while (iter != children.end() ) {
+               iter->remove_children (remove_out_files);
+               iter = children.erase (iter);
+       }
+}
+
 bool
 ExportGraphBuilder::Normalizer::operator== (FileSpec const & other_config) const
 {
        return config.format->normalize() == other_config.format->normalize() &&
-              config.format->normalize_target() == other_config.format->normalize_target();
+               config.format->normalize_target() == other_config.format->normalize_target();
 }
 
 unsigned
@@ -367,7 +461,7 @@ ExportGraphBuilder::Normalizer::start_post_processing()
 /* SRC */
 
 ExportGraphBuilder::SRC::SRC (ExportGraphBuilder & parent, FileSpec const & new_config, framecnt_t max_frames)
-  : parent (parent)
+       : parent (parent)
 {
        config = new_config;
        converter.reset (new SampleRateConverter (new_config.channel_config->get_n_chans()));
@@ -394,6 +488,27 @@ ExportGraphBuilder::SRC::add_child (FileSpec const & new_config)
        }
 }
 
+void
+ExportGraphBuilder::SRC::remove_children (bool remove_out_files)
+{
+       boost::ptr_list<SFC>::iterator sfc_iter = children.begin();
+
+       while (sfc_iter != children.end() ) {
+               converter->remove_output (sfc_iter->sink() );
+               sfc_iter->remove_children (remove_out_files);
+               sfc_iter = children.erase (sfc_iter);
+       }
+
+       boost::ptr_list<Normalizer>::iterator norm_iter = normalized_children.begin();
+
+       while (norm_iter != normalized_children.end() ) {
+               converter->remove_output (norm_iter->sink() );
+               norm_iter->remove_children (remove_out_files);
+               norm_iter = normalized_children.erase (norm_iter);
+       }
+
+}
+
 template<typename T>
 void
 ExportGraphBuilder::SRC::add_child_to_list (FileSpec const & new_config, boost::ptr_list<T> & list)
@@ -417,7 +532,7 @@ ExportGraphBuilder::SRC::operator== (FileSpec const & other_config) const
 
 /* SilenceHandler */
 ExportGraphBuilder::SilenceHandler::SilenceHandler (ExportGraphBuilder & parent, FileSpec const & new_config, framecnt_t max_frames)
-  : parent (parent)
+       : parent (parent)
 {
        config = new_config;
        max_frames_in = max_frames;
@@ -456,34 +571,48 @@ ExportGraphBuilder::SilenceHandler::add_child (FileSpec const & new_config)
        silence_trimmer->add_output (children.back().sink());
 }
 
+void
+ExportGraphBuilder::SilenceHandler::remove_children (bool remove_out_files)
+{
+       boost::ptr_list<SRC>::iterator iter = children.begin();
+
+       while (iter != children.end() ) {
+               silence_trimmer->remove_output (iter->sink() );
+               iter->remove_children (remove_out_files);
+               iter = children.erase (iter);
+       }
+}
+
 bool
 ExportGraphBuilder::SilenceHandler::operator== (FileSpec const & other_config) const
 {
        ExportFormatSpecification & format = *config.format;
        ExportFormatSpecification & other_format = *other_config.format;
        return (format.trim_beginning() == other_format.trim_beginning()) &&
-              (format.trim_end() == other_format.trim_end()) &&
-              (format.silence_beginning_time() == other_format.silence_beginning_time()) &&
-              (format.silence_end_time() == other_format.silence_end_time());
+               (format.trim_end() == other_format.trim_end()) &&
+               (format.silence_beginning_time() == other_format.silence_beginning_time()) &&
+               (format.silence_end_time() == other_format.silence_end_time());
 }
 
 /* ChannelConfig */
 
 ExportGraphBuilder::ChannelConfig::ChannelConfig (ExportGraphBuilder & parent, FileSpec const & new_config, ChannelMap & channel_map)
-  : parent (parent)
+       : parent (parent)
 {
        typedef ExportChannelConfiguration::ChannelList ChannelList;
 
        config = new_config;
 
-       framecnt_t max_frames = parent.session.engine().frames_per_cycle();
+       framecnt_t max_frames = parent.session.engine().samples_per_cycle();
        interleaver.reset (new Interleaver<Sample> ());
        interleaver->init (new_config.channel_config->get_n_chans(), max_frames);
 
        // Make the chunk size divisible by the channel count
        int chan_count = new_config.channel_config->get_n_chans();
        max_frames_out = 8192;
-       max_frames_out -= max_frames_out % chan_count;
+       if (chan_count > 0) {
+               max_frames_out -= max_frames_out % chan_count;
+       }
        chunker.reset (new Chunker<Sample> (max_frames_out));
        interleaver->add_output(chunker);
 
@@ -519,6 +648,19 @@ ExportGraphBuilder::ChannelConfig::add_child (FileSpec const & new_config)
        chunker->add_output (children.back().sink ());
 }
 
+void
+ExportGraphBuilder::ChannelConfig::remove_children (bool remove_out_files)
+{
+       boost::ptr_list<SilenceHandler>::iterator iter = children.begin();
+
+       while(iter != children.end() ) {
+
+               chunker->remove_output (iter->sink ());
+               iter->remove_children (remove_out_files);
+               iter = children.erase(iter);
+       }
+}
+
 bool
 ExportGraphBuilder::ChannelConfig::operator== (FileSpec const & other_config) const
 {