Merge FilmState / Film.
[dcpomatic.git] / src / lib / util.cc
index 9896bff602f073e89caf50d29606c5bbf1424793..90b27f39a6b08b6aede9fb68de0eb71b75cdaf1a 100644 (file)
 #ifdef DVDOMATIC_POSIX
 #include <execinfo.h>
 #include <cxxabi.h>
-#include <sys/types.h> 
-#include <sys/socket.h>
 #endif
 #include <libssh/libssh.h>
 #include <signal.h>
 #include <boost/algorithm/string.hpp>
+#include <boost/bind.hpp>
+#include <boost/lambda/lambda.hpp>
+#include <boost/lexical_cast.hpp>
+#include <boost/thread.hpp>
 #include <openjpeg.h>
 #include <openssl/md5.h>
 #include <magick/MagickCore.h>
@@ -55,18 +57,16 @@ extern "C" {
 #include "dcp_content_type.h"
 #include "filter.h"
 #include "screen.h"
-#include "film_state.h"
+#include "sound_processor.h"
 #ifndef DVDOMATIC_DISABLE_PLAYER
 #include "player_manager.h"
 #endif
 
-#ifdef DEBUG_HASH
-#include <mhash.h>
-#endif
-
 using namespace std;
 using namespace boost;
 
+thread::id ui_thread;
+
 /** Convert some number of seconds to a string representation
  *  in hours, minutes and seconds.
  *
@@ -191,37 +191,6 @@ stacktrace (ostream& out, int levels)
 }
 #endif
 
-/** @param s Sample format.
- *  @return String representation.
- */
-string
-audio_sample_format_to_string (AVSampleFormat s)
-{
-       /* Our sample format handling is not exactly complete */
-       
-       switch (s) {
-       case AV_SAMPLE_FMT_S16:
-               return "S16";
-       default:
-               break;
-       }
-
-       return "Unknown";
-}
-
-/** @param s String representation of a sample format, as returned from audio_sample_format_to_string().
- *  @return Sample format.
- */
-AVSampleFormat
-audio_sample_format_from_string (string s)
-{
-       if (s == "S16") {
-               return AV_SAMPLE_FMT_S16;
-       }
-
-       return AV_SAMPLE_FMT_NONE;
-}
-
 /** @return Version of vobcopy that is on the path (and hence that we will use) */
 static string
 vobcopy_version ()
@@ -281,43 +250,256 @@ dependency_version_summary ()
        return s.str ();
 }
 
+double
+seconds (struct timeval t)
+{
+       return t.tv_sec + (double (t.tv_usec) / 1e6);
+}
+
+
 #ifdef DVDOMATIC_POSIX
-/** Write some data to a socket.
- *  @param fd Socket file descriptor.
- *  @param data Data.
- *  @param size Amount to write, in bytes.
+void
+sigchld_handler (int, siginfo_t* info, void *)
+{
+#ifndef DVDOMATIC_DISABLE_PLAYER       
+       PlayerManager::instance()->child_exited (info->si_pid);
+#endif 
+}
+#endif
+
+/** Call the required functions to set up DVD-o-matic's static arrays, etc.
+ *  Must be called from the UI thread, if there is one.
  */
 void
-socket_write (int fd, uint8_t const * data, int size)
-{
-       uint8_t const * p = data;
-       while (size) {
-               int const n = send (fd, p, size, MSG_NOSIGNAL);
-               if (n < 0) {
-                       stringstream s;
-                       s << "could not write (" << strerror (errno) << ")";
-                       throw NetworkError (s.str ());
+dvdomatic_setup ()
+{
+       Format::setup_formats ();
+       DCPContentType::setup_dcp_content_types ();
+       Scaler::setup_scalers ();
+       Filter::setup_filters ();
+       SoundProcessor::setup_sound_processors ();
+
+       ui_thread = this_thread::get_id ();
+
+#ifdef DVDOMATIC_POSIX 
+       struct sigaction sa;
+       sa.sa_flags = SA_SIGINFO;
+       sigemptyset (&sa.sa_mask);
+       sa.sa_sigaction = sigchld_handler;
+       sigaction (SIGCHLD, &sa, 0);
+#endif 
+}
+
+string
+crop_string (Position start, Size size)
+{
+       stringstream s;
+       s << "crop=" << size.width << ":" << size.height << ":" << start.x << ":" << start.y;
+       return s.str ();
+}
+
+vector<string>
+split_at_spaces_considering_quotes (string s)
+{
+       vector<string> out;
+       bool in_quotes = false;
+       string c;
+       for (string::size_type i = 0; i < s.length(); ++i) {
+               if (s[i] == ' ' && !in_quotes) {
+                       out.push_back (c);
+                       c = "";
+               } else if (s[i] == '"') {
+                       in_quotes = !in_quotes;
+               } else {
+                       c += s[i];
                }
+       }
 
-               size -= n;
-               p += n;
+       out.push_back (c);
+       return out;
+}
+
+string
+md5_digest (void const * data, int size)
+{
+       MD5_CTX md5_context;
+       MD5_Init (&md5_context);
+       MD5_Update (&md5_context, data, size);
+       unsigned char digest[MD5_DIGEST_LENGTH];
+       MD5_Final (digest, &md5_context);
+       
+       stringstream s;
+       for (int i = 0; i < MD5_DIGEST_LENGTH; ++i) {
+               s << hex << setfill('0') << setw(2) << ((int) digest[i]);
        }
+
+       return s.str ();
 }
-#endif
 
-double
-seconds (struct timeval t)
+/** @param file File name.
+ *  @return MD5 digest of file's contents.
+ */
+string
+md5_digest (string file)
 {
-       return t.tv_sec + (double (t.tv_usec) / 1e6);
+       ifstream f (file.c_str(), ios::binary);
+       if (!f.good ()) {
+               throw OpenFileError (file);
+       }
+       
+       f.seekg (0, ios::end);
+       int bytes = f.tellg ();
+       f.seekg (0, ios::beg);
+
+       int const buffer_size = 64 * 1024;
+       char buffer[buffer_size];
+
+       MD5_CTX md5_context;
+       MD5_Init (&md5_context);
+       while (bytes > 0) {
+               int const t = min (bytes, buffer_size);
+               f.read (buffer, t);
+               MD5_Update (&md5_context, buffer, t);
+               bytes -= t;
+       }
+
+       unsigned char digest[MD5_DIGEST_LENGTH];
+       MD5_Final (digest, &md5_context);
+
+       stringstream s;
+       for (int i = 0; i < MD5_DIGEST_LENGTH; ++i) {
+               s << hex << setfill('0') << setw(2) << ((int) digest[i]);
+       }
+
+       return s.str ();
 }
 
-#ifdef DVDOMATIC_POSIX
-/** @param fd File descriptor to read from */
-SocketReader::SocketReader (int fd)
-       : _fd (fd)
+/** @param An arbitrary sampling rate.
+ *  @return The appropriate DCP-approved sampling rate (48kHz or 96kHz).
+ */
+int
+dcp_audio_sample_rate (int fs)
+{
+       if (fs <= 48000) {
+               return 48000;
+       }
+
+       return 96000;
+}
+
+bool operator== (Crop const & a, Crop const & b)
+{
+       return (a.left == b.left && a.right == b.right && a.top == b.top && a.bottom == b.bottom);
+}
+
+bool operator!= (Crop const & a, Crop const & b)
+{
+       return !(a == b);
+}
+
+/** @param index Colour LUT index.
+ *  @return Human-readable name.
+ */
+string
+colour_lut_index_to_name (int index)
+{
+       switch (index) {
+       case 0:
+               return "sRGB";
+       case 1:
+               return "Rec 709";
+       }
+
+       assert (false);
+       return "";
+}
+
+Socket::Socket ()
+       : _deadline (_io_service)
+       , _socket (_io_service)
        , _buffer_data (0)
 {
+       _deadline.expires_at (posix_time::pos_infin);
+       check ();
+}
+
+void
+Socket::check ()
+{
+       if (_deadline.expires_at() <= asio::deadline_timer::traits_type::now ()) {
+               _socket.close ();
+               _deadline.expires_at (posix_time::pos_infin);
+       }
+
+       _deadline.async_wait (boost::bind (&Socket::check, this));
+}
+
+/** Blocking connect with timeout.
+ *  @param endpoint End-point to connect to.
+ *  @param timeout Time-out in seconds.
+ */
+void
+Socket::connect (asio::ip::basic_resolver_entry<asio::ip::tcp> const & endpoint, int timeout)
+{
+       system::error_code ec = asio::error::would_block;
+       _socket.async_connect (endpoint, lambda::var(ec) = lambda::_1);
+       do {
+               _io_service.run_one();
+       } while (ec == asio::error::would_block);
+
+       if (ec || !_socket.is_open ()) {
+               throw NetworkError ("connect timed out");
+       }
+}
+
+/** Blocking write with timeout.
+ *  @param data Buffer to write.
+ *  @param size Number of bytes to write.
+ *  @param timeout Time-out, in seconds.
+ */
+void
+Socket::write (uint8_t const * data, int size, int timeout)
+{
+       _deadline.expires_from_now (posix_time::seconds (timeout));
+       system::error_code ec = asio::error::would_block;
+
+       asio::async_write (_socket, asio::buffer (data, size), lambda::var(ec) = lambda::_1);
+       do {
+               _io_service.run_one ();
+       } while (ec == asio::error::would_block);
+
+       if (ec) {
+               throw NetworkError ("write timed out");
+       }
+}
+
+/** Blocking read with timeout.
+ *  @param data Buffer to read to.
+ *  @param size Number of bytes to read.
+ *  @param timeout Time-out, in seconds.
+ */
+int
+Socket::read (uint8_t* data, int size, int timeout)
+{
+       _deadline.expires_from_now (posix_time::seconds (timeout));
+       system::error_code ec = asio::error::would_block;
+
+       int amount_read = 0;
 
+       _socket.async_read_some (
+               asio::buffer (data, size),
+               (lambda::var(ec) = lambda::_1, lambda::var(amount_read) = lambda::_2)
+               );
+
+       do {
+               _io_service.run_one ();
+       } while (ec == asio::error::would_block);
+       
+       if (ec) {
+               amount_read = 0;
+       }
+
+       return amount_read;
 }
 
 /** Mark some data as being `consumed', so that it will not be returned
@@ -325,7 +507,7 @@ SocketReader::SocketReader (int fd)
  *  @param size Amount of data to consume, in bytes.
  */
 void
-SocketReader::consume (int size)
+Socket::consume (int size)
 {
        assert (_buffer_data >= size);
        
@@ -342,7 +524,7 @@ SocketReader::consume (int size)
  *  @param size Number of bytes to read.
  */
 void
-SocketReader::read_definite_and_consume (uint8_t* data, int size)
+Socket::read_definite_and_consume (uint8_t* data, int size, int timeout)
 {
        int const from_buffer = min (_buffer_data, size);
        if (from_buffer > 0) {
@@ -356,7 +538,7 @@ SocketReader::read_definite_and_consume (uint8_t* data, int size)
 
        /* read() the rest */
        while (size > 0) {
-               int const n = ::read (_fd, data, size);
+               int const n = read (data, size, timeout);
                if (n <= 0) {
                        throw NetworkError ("could not read");
                }
@@ -371,7 +553,7 @@ SocketReader::read_definite_and_consume (uint8_t* data, int size)
  *  @param size Maximum amount of data to read.
  */
 void
-SocketReader::read_indefinite (uint8_t* data, int size)
+Socket::read_indefinite (uint8_t* data, int size, int timeout)
 {
        assert (size < int (sizeof (_buffer)));
 
@@ -379,7 +561,7 @@ SocketReader::read_indefinite (uint8_t* data, int size)
        int to_read = size - _buffer_data;
        while (to_read > 0) {
                /* read as much of it as we can (into our buffer) */
-               int const n = ::read (_fd, _buffer + _buffer_data, to_read);
+               int const n = read (_buffer + _buffer_data, to_read, timeout);
                if (n <= 0) {
                        throw NetworkError ("could not read");
                }
@@ -394,128 +576,175 @@ SocketReader::read_indefinite (uint8_t* data, int size)
        assert (size >= _buffer_data);
        memcpy (data, _buffer, size);
 }
-#endif
 
-#ifdef DVDOMATIC_POSIX
-void
-sigchld_handler (int, siginfo_t* info, void *)
+Rect
+Rect::intersection (Rect const & other) const
 {
-#ifndef DVDOMATIC_DISABLE_PLAYER       
-       PlayerManager::instance()->child_exited (info->si_pid);
-#endif 
+       int const tx = max (x, other.x);
+       int const ty = max (y, other.y);
+       
+       return Rect (
+               tx, ty,
+               min (x + width, other.x + other.width) - tx,
+               min (y + height, other.y + other.height) - ty
+               );
 }
-#endif
-
-/** Call the required functions to set up DVD-o-matic's static arrays, etc. */
-void
-dvdomatic_setup ()
-{
-       Format::setup_formats ();
-       DCPContentType::setup_dcp_content_types ();
-       Scaler::setup_scalers ();
-       Filter::setup_filters ();
 
-#ifdef DVDOMATIC_POSIX 
-       struct sigaction sa;
-       sa.sa_flags = SA_SIGINFO;
-       sigemptyset (&sa.sa_mask);
-       sa.sa_sigaction = sigchld_handler;
-       sigaction (SIGCHLD, &sa, 0);
-#endif 
-}
+/** Round a number up to the nearest multiple of another number.
+ *  @param a Number to round.
+ *  @param t Multiple to round to.
+ *  @return Rounded number.
+ */
 
-string
-crop_string (Position start, Size size)
+int
+round_up (int a, int t)
 {
-       stringstream s;
-       s << "crop=" << size.width << ":" << size.height << ":" << start.x << ":" << start.y;
-       return s.str ();
+       a += (t - 1);
+       return a - (a % t);
 }
 
-vector<string>
-split_at_spaces_considering_quotes (string s)
+/** Read a sequence of key / value pairs from a text stream;
+ *  the keys are the first words on the line, and the values are
+ *  the remainder of the line following the key.  Lines beginning
+ *  with # are ignored.
+ *  @param s Stream to read.
+ *  @return key/value pairs.
+ */
+multimap<string, string>
+read_key_value (istream &s) 
 {
-       vector<string> out;
-       bool in_quotes = false;
-       string c;
-       for (string::size_type i = 0; i < s.length(); ++i) {
-               if (s[i] == ' ' && !in_quotes) {
-                       out.push_back (c);
-                       c = "";
-               } else if (s[i] == '"') {
-                       in_quotes = !in_quotes;
-               } else {
-                       c += s[i];
+       multimap<string, string> kv;
+       
+       string line;
+       while (getline (s, line)) {
+               if (line.empty ()) {
+                       continue;
+               }
+               
+               if (line[0] == '#') {
+                       continue;
+               }
+
+               if (line[line.size() - 1] == '\r') {
+                       line = line.substr (0, line.size() - 1);
+               }
+
+               size_t const s = line.find (' ');
+               if (s == string::npos) {
+                       continue;
                }
+
+               kv.insert (make_pair (line.substr (0, s), line.substr (s + 1)));
        }
 
-       out.push_back (c);
-       return out;
+       return kv;
 }
 
-#ifdef DEBUG_HASH
-void
-md5_data (string title, void const * data, int size)
+string
+get_required_string (multimap<string, string> const & kv, string k)
 {
-       MHASH ht = mhash_init (MHASH_MD5);
-       if (ht == MHASH_FAILED) {
-               throw EncodeError ("could not create hash thread");
+       if (kv.count (k) > 1) {
+               throw StringError ("unexpected multiple keys in key-value set");
        }
 
-       mhash (ht, data, size);
+       multimap<string, string>::const_iterator i = kv.find (k);
        
-       uint8_t hash[16];
-       mhash_deinit (ht, hash);
-       
-       printf ("%s [%d]: ", title.c_str (), size);
-       for (int i = 0; i < int (mhash_get_block_size (MHASH_MD5)); ++i) {
-               printf ("%.2x", hash[i]);
+       if (i == kv.end ()) {
+               throw StringError (String::compose ("missing key %1 in key-value set", k));
        }
-       printf ("\n");
+
+       return i->second;
+}
+
+int
+get_required_int (multimap<string, string> const & kv, string k)
+{
+       string const v = get_required_string (kv, k);
+       return lexical_cast<int> (v);
+}
+
+float
+get_required_float (multimap<string, string> const & kv, string k)
+{
+       string const v = get_required_string (kv, k);
+       return lexical_cast<float> (v);
 }
-#endif
 
 string
-md5_digest (string file)
+get_optional_string (multimap<string, string> const & kv, string k)
 {
-       ifstream f (file.c_str(), ios::binary);
-       if (!f.good ()) {
-               throw OpenFileError (file);
+       if (kv.count (k) > 1) {
+               throw StringError ("unexpected multiple keys in key-value set");
        }
-       
-       f.seekg (0, ios::end);
-       int bytes = f.tellg ();
-       f.seekg (0, ios::beg);
 
-       int const buffer_size = 64 * 1024;
-       char buffer[buffer_size];
+       multimap<string, string>::const_iterator i = kv.find (k);
+       if (i == kv.end ()) {
+               return "";
+       }
 
-       MD5_CTX md5_context;
-       MD5_Init (&md5_context);
-       while (bytes > 0) {
-               int const t = min (bytes, buffer_size);
-               f.read (buffer, t);
-               MD5_Update (&md5_context, buffer, t);
-               bytes -= t;
+       return i->second;
+}
+
+int
+get_optional_int (multimap<string, string> const & kv, string k)
+{
+       if (kv.count (k) > 1) {
+               throw StringError ("unexpected multiple keys in key-value set");
        }
 
-       unsigned char digest[MD5_DIGEST_LENGTH];
-       MD5_Final (digest, &md5_context);
+       multimap<string, string>::const_iterator i = kv.find (k);
+       if (i == kv.end ()) {
+               return 0;
+       }
 
-       stringstream s;
-       for (int i = 0; i < MD5_DIGEST_LENGTH; ++i) {
-               s << hex << setfill('0') << setw(2) << ((int) digest[i]);
+       return lexical_cast<int> (i->second);
+}
+
+AudioBuffers::AudioBuffers (int channels, int frames)
+       : _channels (channels)
+       , _frames (frames)
+{
+       _data = new float*[_channels];
+       for (int i = 0; i < _channels; ++i) {
+               _data[i] = new float[frames];
        }
+}
 
-       return s.str ();
+AudioBuffers::~AudioBuffers ()
+{
+       for (int i = 0; i < _channels; ++i) {
+               delete[] _data[i];
+       }
+
+       delete[] _data;
 }
 
-int
-dcp_audio_sample_rate (int fs)
+float*
+AudioBuffers::data (int c) const
 {
-       if (fs <= 48000) {
-               return 48000;
+       assert (c >= 0 && c < _channels);
+       return _data[c];
+}
+       
+void
+AudioBuffers::set_frames (int f)
+{
+       assert (f <= _frames);
+       _frames = f;
+}
+
+void
+AudioBuffers::make_silent ()
+{
+       for (int i = 0; i < _channels; ++i) {
+               for (int j = 0; j < _frames; ++j) {
+                       _data[i][j] = 0;
+               }
        }
+}
 
-       return 96000;
+void
+ensure_ui_thread ()
+{
+       assert (this_thread::get_id() == ui_thread);
 }