src/lib/util.h

   1 /*
   2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   3     Copyright (C) 2000-2007 Paul Davis
   4
   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.
   9
  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.
  14
  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.
  18
  19 */
  20
  21 /** @file src/util.h
  22  *  @brief Some utility functions and classes.
  23  */
  24
  25 #ifndef DVDOMATIC_UTIL_H
  26 #define DVDOMATIC_UTIL_H
  27
  28 #include <string>
  29 #include <vector>
  30 #include <boost/shared_ptr.hpp>
  31 #include <boost/asio.hpp>
  32 extern "C" {
  33 #include <libavcodec/avcodec.h>
  34 #include <libavfilter/avfilter.h>
  35 }
  36 #include "compose.hpp"
  37
  38 #ifdef DVDOMATIC_DEBUG
  39 #define TIMING(...) _film->log()->microsecond_log (String::compose (__VA_ARGS__), Log::TIMING);
  40 #else
  41 #define TIMING(...)
  42 #endif
  43
  44 /** The maximum number of audio channels that we can cope with */
  45 #define MAX_AUDIO_CHANNELS 6
  46
  47 class Scaler;
  48
  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 ();
  59
  60 typedef int SourceFrame;
  61
  62 struct DCPFrameRate
  63 {
  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.
  68          */
  69         int skip;
  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)
  72          */
  73         bool run_fast;
  74 };
  75
  76 enum ContentType {
  77         STILL, ///< content is still images
  78         VIDEO  ///< content is a video
  79 };
  80
  81 /** @class Size
  82  *  @brief Representation of the size of something */
  83 struct Size
  84 {
  85         /** Construct a zero Size */
  86         Size ()
  87                 : width (0)
  88                 , height (0)
  89         {}
  90
  91         /** @param w Width.
  92          *  @param h Height.
  93          */
  94         Size (int w, int h)
  95                 : width (w)
  96                 , height (h)
  97         {}
  98
  99         /** width */
 100         int width;
 101         /** height */
 102         int height;
 103 };
 104
 105 extern bool operator== (Size const & a, Size const & b);
 106
 107 /** @struct Crop
 108  *  @brief A description of the crop of an image or video.
 109  */
 110 struct Crop
 111 {
 112         Crop () : left (0), right (0), top (0), bottom (0) {}
 113
 114         /** Number of pixels to remove from the left-hand side */
 115         int left;
 116         /** Number of pixels to remove from the right-hand side */
 117         int right;
 118         /** Number of pixels to remove from the top */
 119         int top;
 120         /** Number of pixels to remove from the bottom */
 121         int bottom;
 122 };
 123
 124 extern bool operator== (Crop const & a, Crop const & b);
 125 extern bool operator!= (Crop const & a, Crop const & b);
 126
 127 /** @struct Position
 128  *  @brief A position.
 129  */
 130 struct Position
 131 {
 132         Position ()
 133                 : x (0)
 134                 , y (0)
 135         {}
 136
 137         Position (int x_, int y_)
 138                 : x (x_)
 139                 , y (y_)
 140         {}
 141
 142         /** x coordinate */
 143         int x;
 144         /** y coordinate */
 145         int y;
 146 };
 147
 148 /** @struct Rect
 149  *  @brief A rectangle.
 150  */
 151 struct Rect
 152 {
 153         Rect ()
 154                 : x (0)
 155                 , y (0)
 156                 , width (0)
 157                 , height (0)
 158         {}
 159
 160         Rect (int x_, int y_, int w_, int h_)
 161                 : x (x_)
 162                 , y (y_)
 163                 , width (w_)
 164                 , height (h_)
 165         {}
 166
 167         int x;
 168         int y;
 169         int width;
 170         int height;
 171
 172         Position position () const {
 173                 return Position (x, y);
 174         }
 175
 176         Size size () const {
 177                 return Size (width, height);
 178         }
 179
 180         Rect intersection (Rect const & other) const;
 181 };
 182
 183 extern std::string crop_string (Position, Size);
 184 extern int dcp_audio_sample_rate (int);
 185 extern DCPFrameRate dcp_frame_rate (float);
 186 extern std::string colour_lut_index_to_name (int index);
 187 extern int stride_round_up (int, int const *, int);
 188 extern int stride_lookup (int c, int const * stride);
 189 extern std::multimap<std::string, std::string> read_key_value (std::istream& s);
 190 extern int get_required_int (std::multimap<std::string, std::string> const & kv, std::string k);
 191 extern float get_required_float (std::multimap<std::string, std::string> const & kv, std::string k);
 192 extern std::string get_required_string (std::multimap<std::string, std::string> const & kv, std::string k);
 193 extern int get_optional_int (std::multimap<std::string, std::string> const & kv, std::string k);
 194 extern std::string get_optional_string (std::multimap<std::string, std::string> const & kv, std::string k);
 195
 196 /** @class Socket
 197  *  @brief A class to wrap a boost::asio::ip::tcp::socket with some things
 198  *  that are useful for DVD-o-matic.
 199  *
 200  *  This class wraps some things that I could not work out how to do with boost;
 201  *  most notably, sync read/write calls with timeouts, and the ability to peek into
 202  *  data being read.
 203  */
 204 class Socket
 205 {
 206 public:
 207         Socket ();
 208
 209         /** @return Our underlying socket */
 210         boost::asio::ip::tcp::socket& socket () {
 211                 return _socket;
 212         }
 213
 214         void connect (boost::asio::ip::basic_resolver_entry<boost::asio::ip::tcp> const & endpoint, int timeout);
 215         void write (uint8_t const * data, int size, int timeout);
 216
 217         void read_definite_and_consume (uint8_t* data, int size, int timeout);
 218         void read_indefinite (uint8_t* data, int size, int timeout);
 219         void consume (int amount);
 220
 221 private:
 222         void check ();
 223         int read (uint8_t* data, int size, int timeout);
 224
 225         Socket (Socket const &);
 226
 227         boost::asio::io_service _io_service;
 228         boost::asio::deadline_timer _deadline;
 229         boost::asio::ip::tcp::socket _socket;
 230         /** a buffer for small reads */
 231         uint8_t _buffer[1024];
 232         /** amount of valid data in the buffer */
 233         int _buffer_data;
 234 };
 235
 236 /** @class AudioBuffers
 237  *  @brief A class to hold multi-channel audio data in float format.
 238  */
 239 class AudioBuffers
 240 {
 241 public:
 242         AudioBuffers (int channels, int frames);
 243         AudioBuffers (AudioBuffers const &);
 244         ~AudioBuffers ();
 245
 246         float** data () const {
 247                 return _data;
 248         }
 249
 250         float* data (int) const;
 251
 252         int channels () const {
 253                 return _channels;
 254         }
 255
 256         int frames () const {
 257                 return _frames;
 258         }
 259
 260         void set_frames (int f);
 261
 262         void make_silent ();
 263         void make_silent (int c);
 264
 265         void copy_from (AudioBuffers* from, int frames_to_copy, int read_offset, int write_offset);
 266         void move (int from, int to, int frames);
 267
 268 private:
 269         /** Number of channels */
 270         int _channels;
 271         /** Number of frames (where a frame is one sample across all channels) */
 272         int _frames;
 273         /** Number of frames that _data can hold */
 274         int _allocated_frames;
 275         /** Audio data (so that, e.g. _data[2][6] is channel 2, sample 6) */
 276         float** _data;
 277 };
 278
 279 extern int64_t video_frames_to_audio_frames (SourceFrame v, float audio_sample_rate, float frames_per_second);
 280 extern bool still_image_file (std::string);
 281
 282 #endif
 283