2 Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
3 Copyright (C) 2000-2007 Paul Davis
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software
17 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 * @brief Some utility functions and classes.
25 #ifndef DVDOMATIC_UTIL_H
26 #define DVDOMATIC_UTIL_H
30 #include <boost/shared_ptr.hpp>
31 #include <boost/asio.hpp>
33 #include <libavcodec/avcodec.h>
34 #include <libavfilter/avfilter.h>
36 #include "compose.hpp"
38 #ifdef DVDOMATIC_DEBUG
39 #define TIMING(...) _film->log()->microsecond_log (String::compose (__VA_ARGS__), Log::TIMING);
44 /** The maximum number of audio channels that we can cope with */
45 #define MAX_AUDIO_CHANNELS 6
49 extern std::string seconds_to_hms (int);
50 extern std::string seconds_to_approximate_hms (int);
51 extern void stacktrace (std::ostream &, int);
52 extern std::string dependency_version_summary ();
53 extern double seconds (struct timeval);
54 extern void dvdomatic_setup ();
55 extern std::vector<std::string> split_at_spaces_considering_quotes (std::string);
56 extern std::string md5_digest (std::string);
57 extern std::string md5_digest (void const *, int);
58 extern void ensure_ui_thread ();
60 typedef int SourceFrame;
64 /** frames per second for the DCP */
65 int frames_per_second;
66 /** Skip every `skip' frames. e.g. if this is 1, we skip nothing;
67 * if it's 2, we skip every other frame.
70 /** true if this DCP will run its video faster than the source
71 * (e.g. if the source is 29.97fps and we will run the DCP at 30fps)
77 STILL, ///< content is still images
78 VIDEO ///< content is a video
82 * @brief Representation of the size of something */
85 /** Construct a zero Size */
105 extern bool operator== (Size const & a, Size const & b);
106 extern bool operator!= (Size const & a, Size const & b);
109 * @brief A description of the crop of an image or video.
113 Crop () : left (0), right (0), top (0), bottom (0) {}
115 /** Number of pixels to remove from the left-hand side */
117 /** Number of pixels to remove from the right-hand side */
119 /** Number of pixels to remove from the top */
121 /** Number of pixels to remove from the bottom */
125 extern bool operator== (Crop const & a, Crop const & b);
126 extern bool operator!= (Crop const & a, Crop const & b);
138 Position (int x_, int y_)
150 * @brief A rectangle.
161 Rect (int x_, int y_, int w_, int h_)
173 Position position () const {
174 return Position (x, y);
178 return Size (width, height);
181 Rect intersection (Rect const & other) const;
184 extern std::string crop_string (Position, Size);
185 extern int dcp_audio_sample_rate (int);
186 extern DCPFrameRate dcp_frame_rate (float);
187 extern std::string colour_lut_index_to_name (int index);
188 extern int stride_round_up (int, int const *, int);
189 extern int stride_lookup (int c, int const * stride);
190 extern std::multimap<std::string, std::string> read_key_value (std::istream& s);
191 extern int get_required_int (std::multimap<std::string, std::string> const & kv, std::string k);
192 extern float get_required_float (std::multimap<std::string, std::string> const & kv, std::string k);
193 extern std::string get_required_string (std::multimap<std::string, std::string> const & kv, std::string k);
194 extern int get_optional_int (std::multimap<std::string, std::string> const & kv, std::string k);
195 extern std::string get_optional_string (std::multimap<std::string, std::string> const & kv, std::string k);
198 * @brief A class to wrap a boost::asio::ip::tcp::socket with some things
199 * that are useful for DVD-o-matic.
201 * This class wraps some things that I could not work out how to do with boost;
202 * most notably, sync read/write calls with timeouts, and the ability to peek into
210 /** @return Our underlying socket */
211 boost::asio::ip::tcp::socket& socket () {
215 void connect (boost::asio::ip::basic_resolver_entry<boost::asio::ip::tcp> const & endpoint, int timeout);
216 void write (uint8_t const * data, int size, int timeout);
218 void read_definite_and_consume (uint8_t* data, int size, int timeout);
219 void read_indefinite (uint8_t* data, int size, int timeout);
220 void consume (int amount);
224 int read (uint8_t* data, int size, int timeout);
226 Socket (Socket const &);
228 boost::asio::io_service _io_service;
229 boost::asio::deadline_timer _deadline;
230 boost::asio::ip::tcp::socket _socket;
231 /** a buffer for small reads */
232 uint8_t _buffer[1024];
233 /** amount of valid data in the buffer */
237 /** @class AudioBuffers
238 * @brief A class to hold multi-channel audio data in float format.
243 AudioBuffers (int channels, int frames);
244 AudioBuffers (AudioBuffers const &);
247 float** data () const {
251 float* data (int) const;
253 int channels () const {
257 int frames () const {
261 void set_frames (int f);
264 void make_silent (int c);
266 void copy_from (AudioBuffers* from, int frames_to_copy, int read_offset, int write_offset);
267 void move (int from, int to, int frames);
270 /** Number of channels */
272 /** Number of frames (where a frame is one sample across all channels) */
274 /** Number of frames that _data can hold */
275 int _allocated_frames;
276 /** Audio data (so that, e.g. _data[2][6] is channel 2, sample 6) */
280 extern int64_t video_frames_to_audio_frames (SourceFrame v, float audio_sample_rate, float frames_per_second);
281 extern bool still_image_file (std::string);