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 DCPOMATIC_UTIL_H
  26 #define DCPOMATIC_UTIL_H
  27
  28 #include <string>
  29 #include <vector>
  30 #include <boost/shared_ptr.hpp>
  31 #include <boost/asio.hpp>
  32 #include <boost/optional.hpp>
  33 #include <boost/filesystem.hpp>
  34 #include <dcp/util.h>
  35 #include <dcp/signer.h>
  36 extern "C" {
  37 #include <libavcodec/avcodec.h>
  38 #include <libavfilter/avfilter.h>
  39 }
  40 #include "compose.hpp"
  41 #include "types.h"
  42 #include "video_content.h"
  43
  44 #ifdef DCPOMATIC_DEBUG
  45 #define TIMING(...) _film->log()->microsecond_log (String::compose (__VA_ARGS__), Log::TIMING);
  46 #else
  47 #define TIMING(...)
  48 #endif
  49
  50 #undef check
  51
  52 /** The maximum number of audio channels that we can cope with */
  53 #define MAX_AUDIO_CHANNELS 12
  54
  55 #define DCPOMATIC_HELLO "Boys, you gotta learn not to talk to nuns that way"
  56
  57 namespace libdcp {
  58         class Signer;
  59 }
  60
  61 class Job;
  62
  63 extern std::string seconds_to_hms (int);
  64 extern std::string seconds_to_approximate_hms (int);
  65 extern void stacktrace (std::ostream &, int);
  66 extern std::string dependency_version_summary ();
  67 extern double seconds (struct timeval);
  68 extern void dcpomatic_setup ();
  69 extern void dcpomatic_setup_gettext_i18n (std::string);
  70 extern std::vector<std::string> split_at_spaces_considering_quotes (std::string);
  71 extern std::string md5_digest (std::vector<boost::filesystem::path>, boost::shared_ptr<Job>);
  72 extern std::string md5_digest (void const *, int);
  73 extern void ensure_ui_thread ();
  74 extern std::string audio_channel_name (int);
  75 extern bool valid_image_file (boost::filesystem::path);
  76 #ifdef DCPOMATIC_WINDOWS
  77 extern boost::filesystem::path mo_path ();
  78 #endif
  79 extern std::string tidy_for_filename (std::string);
  80 extern boost::shared_ptr<const dcp::Signer> make_signer ();
  81 extern dcp::Size fit_ratio_within (float ratio, dcp::Size);
  82 extern std::string entities_to_text (std::string e);
  83 extern std::map<std::string, std::string> split_get_request (std::string url);
  84 extern int dcp_audio_frame_rate (int);
  85 extern int stride_round_up (int, int const *, int);
  86 extern std::multimap<std::string, std::string> read_key_value (std::istream& s);
  87 extern int get_required_int (std::multimap<std::string, std::string> const & kv, std::string k);
  88 extern float get_required_float (std::multimap<std::string, std::string> const & kv, std::string k);
  89 extern std::string get_required_string (std::multimap<std::string, std::string> const & kv, std::string k);
  90 extern int get_optional_int (std::multimap<std::string, std::string> const & kv, std::string k);
  91 extern std::string get_optional_string (std::multimap<std::string, std::string> const & kv, std::string k);
  92 extern void* wrapped_av_malloc (size_t);
  93 extern int64_t divide_with_round (int64_t a, int64_t b);
  94
  95 /** @class Socket
  96  *  @brief A class to wrap a boost::asio::ip::tcp::socket with some things
  97  *  that are useful for DCP-o-matic.
  98  *
  99  *  This class wraps some things that I could not work out how to do with boost;
 100  *  most notably, sync read/write calls with timeouts.
 101  */
 102 class Socket
 103 {
 104 public:
 105         Socket (int timeout = 30);
 106         ~Socket ();
 107
 108         /** @return Our underlying socket */
 109         boost::asio::ip::tcp::socket& socket () {
 110                 return _socket;
 111         }
 112
 113         void connect (boost::asio::ip::tcp::endpoint);
 114         void accept (int);
 115
 116         void write (uint32_t n);
 117         void write (uint8_t const * data, int size);
 118
 119         void read (uint8_t* data, int size);
 120         uint32_t read_uint32 ();
 121
 122 private:
 123         void check ();
 124
 125         Socket (Socket const &);
 126
 127         boost::asio::io_service _io_service;
 128         boost::asio::deadline_timer _deadline;
 129         boost::asio::ip::tcp::socket _socket;
 130         boost::asio::ip::tcp::acceptor* _acceptor;
 131         int _timeout;
 132 };
 133
 134 class LocaleGuard
 135 {
 136 public:
 137         LocaleGuard ();
 138         ~LocaleGuard ();
 139
 140 private:
 141         char* _old;
 142 };
 143
 144 class ScopedTemporary
 145 {
 146 public:
 147         ScopedTemporary ();
 148         ~ScopedTemporary ();
 149
 150         boost::filesystem::path file () const {
 151                 return _file;
 152         }
 153
 154         char const * c_str () const;
 155         FILE* open (char const *);
 156         void close ();
 157
 158 private:
 159         boost::filesystem::path _file;
 160         FILE* _open;
 161 };
 162
 163 #endif
 164