X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Futil.cc;h=45965385c6c9b4f5e4bdb9202be0671b2335f720;hb=21d3f3889ede9bd4b2c48424715e44ad2c405500;hp=7388539cdf68080ead8e43a31bddea226acb313e;hpb=c137bfe8c525d4bfa48faac098c5824d57c68422;p=dcpomatic.git diff --git a/src/lib/util.cc b/src/lib/util.cc index 7388539cd..45965385c 100644 --- a/src/lib/util.cc +++ b/src/lib/util.cc @@ -25,16 +25,21 @@ #include #include #include +#include #ifdef DVDOMATIC_POSIX #include #include -#include -#include #endif #include #include #include +#include +#include +#include +#include +#include #include +#include #include #include #include @@ -52,17 +57,13 @@ extern "C" { #include "format.h" #include "dcp_content_type.h" #include "filter.h" -#include "screen.h" -#include "film_state.h" -#include "player_manager.h" - -#ifdef DEBUG_HASH -#include -#endif +#include "sound_processor.h" using namespace std; using namespace boost; +thread::id ui_thread; + /** Convert some number of seconds to a string representation * in hours, minutes and seconds. * @@ -187,65 +188,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 () -{ - FILE* f = popen ("vobcopy -V 2>&1", "r"); - if (f == 0) { - throw EncodeError ("could not run vobcopy to check version"); - } - - string version = "unknown"; - - while (!feof (f)) { - char buf[256]; - if (fgets (buf, sizeof (buf), f)) { - string s (buf); - vector b; - split (b, s, is_any_of (" ")); - if (b.size() >= 2 && b[0] == "Vobcopy") { - version = b[1]; - } - } - } - - pclose (f); - - return version; -} - /** @param v Version as used by FFmpeg. * @return A string representation of v. */ @@ -263,7 +205,6 @@ dependency_version_summary () { stringstream s; s << "libopenjpeg " << opj_version () << ", " - << "vobcopy " << vobcopy_version() << ", " << "libavcodec " << ffmpeg_version_to_string (avcodec_version()) << ", " << "libavfilter " << ffmpeg_version_to_string (avfilter_version()) << ", " << "libavformat " << ffmpeg_version_to_string (avformat_version()) << ", " @@ -277,43 +218,276 @@ dependency_version_summary () return s.str (); } -#ifdef DVDOMATIC_POSIX -/** 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); +} + +/** 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 (); +} + +/** @param start Start position for the crop within the image. + * @param size Size of the cropped area. + * @return FFmpeg crop filter string. + */ +string +crop_string (Position start, Size size) +{ + stringstream s; + s << "crop=" << size.width << ":" << size.height << ":" << start.x << ":" << start.y; + return s.str (); +} + +/** @param s A string. + * @return Parts of the string split at spaces, except when a space is within quotation marks. + */ +vector +split_at_spaces_considering_quotes (string s) +{ + vector 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 fps Arbitrary frames-per-second value. + * @return DCPFrameRate for this frames-per-second. + */ +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!= (Size const & a, Size const & b) +{ + return !(a == b); +} + +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 const & endpoint, int timeout) +{ + _deadline.expires_from_now (posix_time::seconds (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 @@ -321,7 +495,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); @@ -338,7 +512,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) { @@ -352,7 +526,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"); } @@ -367,7 +541,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))); @@ -375,7 +549,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"); } @@ -390,82 +564,300 @@ 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 *) +/** @param other A Rect. + * @return The intersection of this with `other'. + */ +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 + ); } -#endif -/** 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 c Index. + * @param s Array of numbers to round, indexed by c. + * @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); +} + +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 +read_key_value (istream &s) +{ + multimap 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))); + } -#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 + return kv; } string -crop_string (Position start, Size size) +get_required_string (multimap 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::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 -split_at_spaces_considering_quotes (string s) +int +get_required_int (multimap const & kv, string k) { - vector 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 (v); +} + +float +get_required_float (multimap const & kv, string k) +{ + string const v = get_required_string (kv, k); + return lexical_cast (v); +} + +string +get_optional_string (multimap 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::const_iterator i = kv.find (k); + if (i == kv.end ()) { + return ""; + } + + return i->second; +} + +int +get_optional_int (multimap const & kv, string k) +{ + if (kv.count (k) > 1) { + throw StringError ("unexpected multiple keys in key-value set"); + } + + multimap::const_iterator i = kv.find (k); + if (i == kv.end ()) { + return 0; + } + + return lexical_cast (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 destructor */ +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]; +} + +/** 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 +AudioBuffers::make_silent () +{ + 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; + } } -#ifdef DEBUG_HASH +/** 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 -md5_data (string title, void const * data, int size) +AudioBuffers::copy_from (AudioBuffers* from, int frames_to_copy, int read_offset, int write_offset) { - MHASH ht = mhash_init (MHASH_MD5); - if (ht == MHASH_FAILED) { - throw EncodeError ("could not create hash thread"); + 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"); +}