harden TempoMap::next_tempo_section () a bit.
[ardour.git] / libs / ardour / sndfilesource.cc
index 5acfe7b119246983cce6e02fe7856333853a29a4..89b497a2483470178a4e42872aae5a307c4f7fd9 100644 (file)
 #include <cerrno>
 #include <climits>
 #include <cstdarg>
+#include <fcntl.h>
 
 #include <sys/stat.h>
 
-#ifdef PLATFORM_WINDOWS
+#include <glib.h>
+#include "pbd/gstdio_compat.h"
+
 #include <glibmm/convert.h>
-#endif
 #include <glibmm/fileutils.h>
 #include <glibmm/miscutils.h>
 
+#include "ardour/runtime_functions.h"
 #include "ardour/sndfilesource.h"
 #include "ardour/sndfile_helpers.h"
 #include "ardour/utils.h"
 #include "ardour/session.h"
 
-#include "i18n.h"
+#include "pbd/i18n.h"
 
 using namespace std;
 using namespace ARDOUR;
@@ -76,7 +79,7 @@ SndFileSource::SndFileSource (Session& s, const XMLNode& node)
 }
 
 /** Constructor for existing external-to-session files.
-    Files created this way are never writable or removable 
+    Files created this way are never writable or removable
 */
 SndFileSource::SndFileSource (Session& s, const string& path, int chn, Flag flags)
        : Source(s, DataType::AUDIO, path, flags)
@@ -101,8 +104,8 @@ SndFileSource::SndFileSource (Session& s, const string& path, int chn, Flag flag
        }
 }
 
-/** This constructor is used to construct new internal-to-session files, 
-    not open existing ones. 
+/** This constructor is used to construct new internal-to-session files,
+    not open existing ones.
 */
 SndFileSource::SndFileSource (Session& s, const string& path, const string& origin,
                               SampleFormat sfmt, HeaderFormat hf, framecnt_t rate, Flag flags)
@@ -150,6 +153,23 @@ SndFileSource::SndFileSource (Session& s, const string& path, const string& orig
                _flags = Flag (_flags & ~Broadcast);
                break;
 
+       case RF64_WAV:
+               fmt = SF_FORMAT_RF64;
+               _flags = Flag (_flags & ~Broadcast);
+               _flags = Flag (_flags | RF64_RIFF);
+               break;
+
+       case MBWF:
+               fmt = SF_FORMAT_RF64;
+               _flags = Flag (_flags | Broadcast);
+               _flags = Flag (_flags | RF64_RIFF);
+               break;
+
+       case RF64:
+               fmt = SF_FORMAT_RF64;
+               _flags = Flag (_flags & ~Broadcast);
+               break;
+
        default:
                fatal << string_compose (_("programming error: %1"), X_("unsupported audio header format requested")) << endmsg;
                abort(); /*NOTREACHED*/
@@ -213,6 +233,88 @@ SndFileSource::SndFileSource (Session& s, const string& path, int chn)
        }
 }
 
+/** Constructor to losslessly compress existing source to flac */
+SndFileSource::SndFileSource (Session& s, const AudioFileSource& other, const string& path, bool use16bits, Progress* progress)
+       : Source(s, DataType::AUDIO, path, Flag ((other.flags () | default_writable_flags | NoPeakFile) & ~RF64_RIFF))
+       , AudioFileSource (s, path, "", Flag ((other.flags () | default_writable_flags | NoPeakFile) & ~RF64_RIFF), /*unused*/ FormatFloat, /*unused*/ WAVE64)
+       , _sndfile (0)
+       , _broadcast_info (0)
+       , _capture_start (false)
+       , _capture_end (false)
+       , file_pos (0)
+       , xfade_buf (0)
+{
+       if (other.readable_length () == 0) {
+               throw failed_constructor();
+       }
+
+       assert (!Glib::file_test (_path, Glib::FILE_TEST_EXISTS));
+
+       _channel = other.channel ();
+       init_sndfile ();
+
+       _file_is_new = true;
+
+       _info.channels = other.n_channels();
+       _info.samplerate = other.sample_rate ();
+       _info.format = SF_FORMAT_FLAC | (use16bits ? SF_FORMAT_PCM_16 : SF_FORMAT_PCM_24);
+
+       /* flac is either read or write -- never both,
+        * so we need to special-case ::open () */
+#ifdef PLATFORM_WINDOWS
+       int fd = g_open (_path.c_str(), O_CREAT | O_RDWR, 0644);
+#else
+       int fd = ::open (_path.c_str(), O_CREAT | O_RDWR, 0644);
+#endif
+       if (fd == -1) {
+               throw failed_constructor();
+       }
+
+       _sndfile = sf_open_fd (fd, SFM_WRITE, &_info, true);
+
+       if (_sndfile == 0) {
+               throw failed_constructor();
+       }
+
+       Sample buf[8192];
+       framecnt_t off = 0;
+       float peak = 0;
+       float norm = 1.f;
+
+       /* normalize before converting to fixed point, calc gain factor */
+       framecnt_t len = other.read (buf, off, 8192, /*channel*/0);
+       while (len > 0) {
+               peak = compute_peak (buf, len, peak);
+               off += len;
+               len = other.read (buf, off, 8192, /*channel*/0);
+               if (progress) {
+                       progress->set_progress (0.5f * (float) off / other.readable_length ());
+               }
+       }
+
+       if (peak > 0) {
+               _gain *= peak;
+               norm = 1.f / peak;
+       }
+
+       /* copy file */
+       off = 0;
+       len = other.read (buf, off, 8192, /*channel*/0);
+       while (len > 0) {
+               if (norm != 1.f) {
+                       for (framecnt_t i = 0; i < len; ++i) {
+                               buf[i] *= norm;
+                       }
+               }
+               write (buf, len);
+               off += len;
+               len = other.read (buf, off, 8192, /*channel*/0);
+               if (progress) {
+                       progress->set_progress (0.5f + 0.5f * (float) off / other.readable_length ());
+               }
+       }
+}
+
 void
 SndFileSource::init_sndfile ()
 {
@@ -236,25 +338,42 @@ SndFileSource::close ()
        if (_sndfile) {
                sf_close (_sndfile);
                _sndfile = 0;
+               file_closed ();
        }
 }
 
 int
 SndFileSource::open ()
 {
-       string path_to_open;
-
        if (_sndfile) {
                return 0;
        }
-       
+
+// We really only want to use g_open for all platforms but because of this
+// method(SndfileSource::open), the compiler(or at least GCC) is confused
+// because g_open will expand to "open" on non-POSIX systems and needs the
+// global namespace qualifer. The problem is since since C99 ::g_open will
+// apparently expand to ":: open"
 #ifdef PLATFORM_WINDOWS
-       path_to_open = Glib::locale_from_utf8(_path);
+       int fd = g_open (_path.c_str(), writable() ? O_CREAT | O_RDWR : O_RDONLY, writable() ? 0644 : 0444);
 #else
-       path_to_open = _path;
+       int fd = ::open (_path.c_str(), writable() ? O_CREAT | O_RDWR : O_RDONLY, writable() ? 0644 : 0444);
 #endif
 
-       _sndfile = sf_open (path_to_open.c_str(), writable() ? SFM_RDWR : SFM_READ, &_info);
+       if (fd == -1) {
+               error << string_compose (
+                            _ ("SndFileSource: cannot open file \"%1\" for %2"),
+                            _path,
+                            (writable () ? "read+write" : "reading")) << endmsg;
+               return -1;
+       }
+
+       if ((_info.format & SF_FORMAT_TYPEMASK ) == SF_FORMAT_FLAC) {
+               assert (!writable());
+               _sndfile = sf_open_fd (fd, SFM_READ, &_info, true);
+       } else {
+               _sndfile = sf_open_fd (fd, writable() ? SFM_RDWR : SFM_READ, &_info, true);
+       }
 
        if (_sndfile == 0) {
                char errbuf[1024];
@@ -264,10 +383,10 @@ SndFileSource::open ()
                   so we don't want to see this message.
                */
 
-                cerr << "failed to open " << path_to_open << " with name " << _name << endl;
+                cerr << "failed to open " << _path << " with name " << _name << endl;
 
                error << string_compose(_("SndFileSource: cannot open file \"%1\" for %2 (%3)"),
-                                       path_to_open, (writable() ? "read+write" : "reading"), errbuf) << endmsg;
+                                       _path, (writable() ? "read+write" : "reading"), errbuf) << endmsg;
 #endif
                return -1;
        }
@@ -283,6 +402,19 @@ SndFileSource::open ()
 
        _length = _info.frames;
 
+#ifdef HAVE_RF64_RIFF
+       if (_file_is_new && _length == 0 && writable()) {
+               if (_flags & RF64_RIFF) {
+                       if (sf_command (_sndfile, SFC_RF64_AUTO_DOWNGRADE, 0, 0) != SF_TRUE) {
+                               char errbuf[256];
+                               sf_error_str (_sndfile, errbuf, sizeof (errbuf) - 1);
+                               error << string_compose (_("Cannot mark RF64 audio file for automatic downgrade to WAV: %1"), errbuf)
+                                     << endmsg;
+                       }
+               }
+       }
+#endif
+
        if (!_broadcast_info) {
                _broadcast_info = new BroadcastInfo;
        }
@@ -305,12 +437,12 @@ SndFileSource::open ()
                delete _broadcast_info;
                _broadcast_info = 0;
                _flags = Flag (_flags & ~Broadcast);
-       } 
+       }
 
        /* Set the broadcast flag if the BWF info is already there. We need
         * this when recovering or using existing files.
         */
-       
+
        if (bwf_info_exists) {
                _flags = Flag (_flags | Broadcast);
        }
@@ -329,7 +461,7 @@ SndFileSource::open ()
 
                         if (!_broadcast_info->write_to_file (_sndfile)) {
                                 error << string_compose (_("cannot set broadcast info for audio file %1 (%2); dropping broadcast info for this file"),
-                                                         path_to_open, _broadcast_info->get_error())
+                                                         _path, _broadcast_info->get_error())
                                       << endmsg;
                                 _flags = Flag (_flags & ~Broadcast);
                                 delete _broadcast_info;
@@ -337,7 +469,7 @@ SndFileSource::open ()
                         }
                 }
         }
-       
+
        return 0;
 }
 
@@ -358,7 +490,7 @@ framecnt_t
 SndFileSource::read_unlocked (Sample *dst, framepos_t start, framecnt_t cnt) const
 {
        assert (cnt >= 0);
-       
+
        framecnt_t nread;
        float *ptr;
        framecnt_t real_cnt;
@@ -417,6 +549,11 @@ SndFileSource::read_unlocked (Sample *dst, framepos_t start, framecnt_t cnt) con
                                sf_error_str (0, errbuf, sizeof (errbuf) - 1);
                                error << string_compose(_("SndFileSource: @ %1 could not read %2 within %3 (%4) (len = %5, ret was %6)"), start, file_cnt, _name.val().substr (1), errbuf, _length, ret) << endl;
                        }
+                       if (_gain != 1.f) {
+                               for (framecnt_t i = 0; i < ret; ++i) {
+                                       dst[i] *= _gain;
+                               }
+                       }
                        return ret;
                }
        }
@@ -431,9 +568,16 @@ SndFileSource::read_unlocked (Sample *dst, framepos_t start, framecnt_t cnt) con
 
        /* stride through the interleaved data */
 
-       for (framecnt_t n = 0; n < nread; ++n) {
-               dst[n] = *ptr;
-               ptr += _info.channels;
+       if (_gain != 1.f) {
+               for (framecnt_t n = 0; n < nread; ++n) {
+                       dst[n] = *ptr * _gain;
+                       ptr += _info.channels;
+               }
+       } else {
+               for (framecnt_t n = 0; n < nread; ++n) {
+                       dst[n] = *ptr;
+                       ptr += _info.channels;
+               }
        }
 
        return nread;
@@ -670,7 +814,10 @@ SndFileSource::set_header_timeline_position ()
 framecnt_t
 SndFileSource::write_float (Sample* data, framepos_t frame_pos, framecnt_t cnt)
 {
-       if (_sndfile == 0 || sf_seek (_sndfile, frame_pos, SEEK_SET|SFM_WRITE) < 0) {
+       if ((_info.format & SF_FORMAT_TYPEMASK ) == SF_FORMAT_FLAC) {
+               assert (_length == frame_pos);
+       }
+       else if (_sndfile == 0 || sf_seek (_sndfile, frame_pos, SEEK_SET|SFM_WRITE) < 0) {
                char errbuf[256];
                sf_error_str (0, errbuf, sizeof (errbuf) - 1);
                error << string_compose (_("%1: cannot seek to %2 (libsndfile error: %3)"), _path, frame_pos, errbuf) << endmsg;
@@ -690,6 +837,7 @@ SndFileSource::natural_position() const
        return _timeline_position;
 }
 
+#ifdef XXX_OLD_DESTRUCTIVE_API_XXX
 bool
 SndFileSource::set_destructive (bool yn)
 {
@@ -708,6 +856,7 @@ SndFileSource::set_destructive (bool yn)
 
        return true;
 }
+#endif
 
 void
 SndFileSource::clear_capture_marks ()
@@ -924,8 +1073,23 @@ SndFileSource::get_soundfile_info (const string& path, SoundFileInfo& info, stri
 
        sf_info.format = 0; // libsndfile says to clear this before sf_open().
 
-       if ((sf = sf_open (const_cast<char*>(path.c_str()), SFM_READ, &sf_info)) == 0) {
-               char errbuf[256];
+       if (path.empty() || Glib::file_test(path, Glib::FILE_TEST_IS_DIR)) {
+               return false;
+       }
+
+#ifdef PLATFORM_WINDOWS
+       int fd = g_open (path.c_str(), O_RDONLY, 0444);
+#else
+       int fd = ::open (path.c_str(), O_RDONLY, 0444);
+#endif
+
+       if (fd == -1) {
+               error << string_compose ( _("SndFileSource: cannot open file \"%1\" for reading"), path)
+                     << endmsg;
+               return false;
+       }
+       if ((sf = sf_open_fd (fd, SFM_READ, &sf_info, true)) == 0) {
+               char errbuf[1024];
                error_msg = sf_error_str (0, errbuf, sizeof (errbuf) - 1);
                return false;
        }