#include <sstream>
#include <iomanip>
#include <iostream>
+#include <fstream>
+#ifdef DVDOMATIC_POSIX
#include <execinfo.h>
#include <cxxabi.h>
+#endif
+#include <libssh/libssh.h>
#include <signal.h>
-#include <sys/types.h>
-#include <sys/socket.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 <boost/filesystem.hpp>
#include <openjpeg.h>
+#include <openssl/md5.h>
#include <magick/MagickCore.h>
#include <magick/version.h>
-#include <libssh/libssh.h>
+#include <libdcp/version.h>
extern "C" {
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libswscale/swscale.h>
-#include <libswresample/swresample.h>
#include <libavfilter/avfiltergraph.h>
-#include <libavfilter/avcodec.h>
-#include <libavfilter/buffersink.h>
#include <libpostproc/postprocess.h>
#include <libavutil/pixfmt.h>
}
#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"
-
-#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.
*
return ap.str ();
}
+#ifdef DVDOMATIC_POSIX
/** @param l Mangled C++ identifier.
* @return Demangled version.
*/
free (strings);
}
}
-
-/** @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 OpenDCP that is on the path (and hence that we will use) */
-static string
-opendcp_version ()
-{
- FILE* f = popen ("opendcp_xml", "r");
- if (f == 0) {
- throw EncodeError ("could not run opendcp_xml to check version");
- }
-
- string version = "unknown";
-
- while (!feof (f)) {
- char* buf = 0;
- size_t n = 0;
- ssize_t const r = getline (&buf, &n, f);
- if (r > 0) {
- string s (buf);
- vector<string> b;
- split (b, s, is_any_of (" "));
- if (b.size() >= 3 && b[0] == "OpenDCP" && b[1] == "version") {
- version = b[2];
- }
- free (buf);
- }
- }
-
- pclose (f);
-
- return version;
-}
+#endif
/** @return Version of vobcopy that is on the path (and hence that we will use) */
static string
string version = "unknown";
while (!feof (f)) {
- char* buf = 0;
- size_t n = 0;
- ssize_t const r = getline (&buf, &n, f);
- if (r > 0) {
+ char buf[256];
+ if (fgets (buf, sizeof (buf), f)) {
string s (buf);
vector<string> b;
split (b, s, is_any_of (" "));
if (b.size() >= 2 && b[0] == "Vobcopy") {
version = b[1];
}
- free (buf);
}
}
{
stringstream s;
s << "libopenjpeg " << opj_version () << ", "
- << "opendcp " << opendcp_version () << ", "
<< "vobcopy " << vobcopy_version() << ", "
- << "libswresample " << ffmpeg_version_to_string (swresample_version()) << ", "
<< "libavcodec " << ffmpeg_version_to_string (avcodec_version()) << ", "
<< "libavfilter " << ffmpeg_version_to_string (avfilter_version()) << ", "
<< "libavformat " << ffmpeg_version_to_string (avformat_version()) << ", "
<< "libpostproc " << ffmpeg_version_to_string (postproc_version()) << ", "
<< "libswscale " << ffmpeg_version_to_string (swscale_version()) << ", "
<< MagickVersion << ", "
- << "libssh " << ssh_version (0);
+ << "libssh " << ssh_version (0) << ", "
+ << "libdcp " << libdcp::version << " git " << libdcp::git_commit;
return s.str ();
}
-/** Write some data to a socket.
- * @param fd Socket file descriptor.
- * @param data Data.
- * @param size Amount to write, in bytes.
+double
+seconds (struct timeval t)
+{
+ return t.tv_sec + (double (t.tv_usec) / 1e6);
+}
+
+
+#ifdef DVDOMATIC_POSIX
+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 ();
}
-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 ();
+}
+
+DCPFrameRate
+dcp_frame_rate (float fps)
+{
+ DCPFrameRate dfr;
+
+ dfr.run_fast = (fps != rint (fps));
+ dfr.frames_per_second = rint (fps);
+ dfr.skip = 1;
+
+ /* XXX: somewhat arbitrary */
+ if (fps == 50) {
+ dfr.frames_per_second = 25;
+ dfr.skip = 2;
+ }
+
+ return dfr;
+}
+
+/** @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== (Size const & a, Size const & b)
+{
+ return (a.width == b.width && a.height == b.height);
+}
+
+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);
}
-/** @param fd File descriptor to read from */
-SocketReader::SocketReader (int fd)
- : _fd (fd)
+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
* @param size Amount of data to consume, in bytes.
*/
void
-SocketReader::consume (int size)
+Socket::consume (int size)
{
assert (_buffer_data >= 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) {
/* 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");
}
* @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)));
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");
}
memcpy (data, _buffer, size);
}
-void
-sigchld_handler (int, siginfo_t* info, void *)
+Rect
+Rect::intersection (Rect const & other) const
{
- PlayerManager::instance()->child_exited (info->si_pid);
+ 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
+ );
}
-/** Call the required functions to set up DVD-o-matic's static arrays, etc. */
-void
-dvdomatic_setup ()
+/** 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.
+ */
+
+int
+stride_round_up (int c, int const * stride, int t)
{
- Format::setup_formats ();
- DCPContentType::setup_dcp_content_types ();
- Scaler::setup_scalers ();
- Filter::setup_filters ();
+ int const a = stride[c] + (t - 1);
+ return a - (a % t);
+}
- struct sigaction sa;
- sa.sa_flags = SA_SIGINFO;
- sigemptyset (&sa.sa_mask);
- sa.sa_sigaction = sigchld_handler;
- sigaction (SIGCHLD, &sa, 0);
+int
+stride_lookup (int c, int const * stride)
+{
+ return stride[c];
+}
+
+/** 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)
+{
+ 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)));
+ }
+
+ return kv;
}
string
-crop_string (Position start, Size size)
+get_required_string (multimap<string, string> const & kv, string k)
{
- stringstream s;
- s << "crop=" << size.width << ":" << size.height << ":" << start.x << ":" << start.y;
- return s.str ();
+ if (kv.count (k) > 1) {
+ throw StringError ("unexpected multiple keys in key-value set");
+ }
+
+ multimap<string, string>::const_iterator i = kv.find (k);
+
+ if (i == kv.end ()) {
+ throw StringError (String::compose ("missing key %1 in key-value set", k));
+ }
+
+ return i->second;
}
-vector<string>
-split_at_spaces_considering_quotes (string s)
+int
+get_required_int (multimap<string, string> const & kv, string k)
{
- 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];
- }
+ 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);
+}
+
+string
+get_optional_string (multimap<string, string> const & kv, string k)
+{
+ if (kv.count (k) > 1) {
+ throw StringError ("unexpected multiple keys in key-value set");
}
- out.push_back (c);
- return out;
+ multimap<string, string>::const_iterator i = kv.find (k);
+ if (i == kv.end ()) {
+ return "";
+ }
+
+ 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");
+ }
+
+ multimap<string, string>::const_iterator i = kv.find (k);
+ if (i == kv.end ()) {
+ return 0;
+ }
+
+ return lexical_cast<int> (i->second);
+}
+
+/** Construct an AudioBuffers. Audio data is undefined after this constructor.
+ * @param channels Number of channels.
+ * @param frames Number of frames to reserve space for.
+ */
+AudioBuffers::AudioBuffers (int channels, int frames)
+ : _channels (channels)
+ , _frames (frames)
+ , _allocated_frames (frames)
+{
+ _data = new float*[_channels];
+ for (int i = 0; i < _channels; ++i) {
+ _data[i] = new float[frames];
+ }
+}
+
+/** Copy constructor.
+ * @param other Other AudioBuffers; data is copied.
+ */
+AudioBuffers::AudioBuffers (AudioBuffers const & other)
+ : _channels (other._channels)
+ , _frames (other._frames)
+ , _allocated_frames (other._frames)
+{
+ _data = new float*[_channels];
+ for (int i = 0; i < _channels; ++i) {
+ _data[i] = new float[_frames];
+ memcpy (_data[i], other._data[i], _frames * sizeof (float));
+ }
+}
+
+AudioBuffers::~AudioBuffers ()
+{
+ for (int i = 0; i < _channels; ++i) {
+ delete[] _data[i];
+ }
+
+ delete[] _data;
+}
+
+/** @param c Channel index.
+ * @return Buffer for this channel.
+ */
+float*
+AudioBuffers::data (int c) const
+{
+ assert (c >= 0 && c < _channels);
+ return _data[c];
}
-#ifdef DEBUG_HASH
+/** Set the number of frames that these AudioBuffers will report themselves
+ * as having.
+ * @param f Frames; must be less than or equal to the number of allocated frames.
+ */
+void
+AudioBuffers::set_frames (int f)
+{
+ assert (f <= _allocated_frames);
+ _frames = f;
+}
+
+/** Make all samples on all channels silent */
void
-md5_data (string title, void const * data, int size)
+AudioBuffers::make_silent ()
{
- MHASH ht = mhash_init (MHASH_MD5);
- if (ht == MHASH_FAILED) {
- throw EncodeError ("could not create hash thread");
+ for (int i = 0; i < _channels; ++i) {
+ make_silent (i);
+ }
+}
+
+/** Make all samples on a given channel silent.
+ * @param c Channel.
+ */
+void
+AudioBuffers::make_silent (int c)
+{
+ assert (c >= 0 && c < _channels);
+
+ for (int i = 0; i < _frames; ++i) {
+ _data[c][i] = 0;
+ }
+}
+
+/** Copy data from another AudioBuffers to this one. All channels are copied.
+ * @param from AudioBuffers to copy from; must have the same number of channels as this.
+ * @param frames_to_copy Number of frames to copy.
+ * @param read_offset Offset to read from in `from'.
+ * @param write_offset Offset to write to in `to'.
+ */
+void
+AudioBuffers::copy_from (AudioBuffers* from, int frames_to_copy, int read_offset, int write_offset)
+{
+ assert (from->channels() == channels());
+
+ assert (from);
+ assert (read_offset >= 0 && (read_offset + frames_to_copy) <= from->_allocated_frames);
+ assert (write_offset >= 0 && (write_offset + frames_to_copy) <= _allocated_frames);
+
+ for (int i = 0; i < _channels; ++i) {
+ memcpy (_data[i] + write_offset, from->_data[i] + read_offset, frames_to_copy * sizeof(float));
}
+}
- mhash (ht, data, size);
+/** Move audio data around.
+ * @param from Offset to move from.
+ * @param to Offset to move to.
+ * @param frames Number of frames to move.
+ */
+
+void
+AudioBuffers::move (int from, int to, int frames)
+{
+ if (frames == 0) {
+ return;
+ }
- uint8_t hash[16];
- mhash_deinit (ht, hash);
+ assert (from >= 0);
+ assert (from < _frames);
+ assert (to >= 0);
+ assert (to < _frames);
+ assert (frames > 0);
+ assert (frames <= _frames);
+ assert ((from + frames) <= _frames);
+ assert ((to + frames) <= _frames);
- printf ("%s [%d]: ", title.c_str (), size);
- for (int i = 0; i < int (mhash_get_block_size (MHASH_MD5)); ++i) {
- printf ("%.2x", hash[i]);
+ for (int i = 0; i < _channels; ++i) {
+ memmove (_data[i] + to, _data[i] + from, frames * sizeof(float));
}
- printf ("\n");
}
+
+/** Trip an assert if the caller is not in the UI thread */
+void
+ensure_ui_thread ()
+{
+ assert (this_thread::get_id() == ui_thread);
+}
+
+/** @param v Source video frame.
+ * @param audio_sample_rate Source audio sample rate.
+ * @param frames_per_second Number of video frames per second.
+ * @return Equivalent number of audio frames for `v'.
+ */
+int64_t
+video_frames_to_audio_frames (SourceFrame v, float audio_sample_rate, float frames_per_second)
+{
+ return ((int64_t) v * audio_sample_rate / frames_per_second);
+}
+
+/** @param f Filename.
+ * @return true if this file is a still image, false if it is something else.
+ */
+bool
+still_image_file (string f)
+{
+#if BOOST_FILESYSTEM_VERSION == 3
+ string ext = boost::filesystem::path(f).extension().string();
+#else
+ string ext = boost::filesystem::path(f).extension();
#endif
+ transform (ext.begin(), ext.end(), ext.begin(), ::tolower);
+
+ return (ext == ".tif" || ext == ".tiff" || ext == ".jpg" || ext == ".jpeg" || ext == ".png");
+}