Merge branch '0.70-support'
[dcpomatic.git] / src / lib / util.cc
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/lib/util.cc
22  *  @brief Some utility functions and classes.
23  */
24
25 #include <sstream>
26 #include <iomanip>
27 #include <iostream>
28 #include <fstream>
29 #ifdef DVDOMATIC_POSIX
30 #include <execinfo.h>
31 #include <cxxabi.h>
32 #endif
33 #include <libssh/libssh.h>
34 #include <signal.h>
35 #include <boost/algorithm/string.hpp>
36 #include <boost/bind.hpp>
37 #include <boost/lambda/lambda.hpp>
38 #include <boost/lexical_cast.hpp>
39 #include <boost/thread.hpp>
40 #include <boost/filesystem.hpp>
41 #include <openjpeg.h>
42 #include <openssl/md5.h>
43 #include <magick/MagickCore.h>
44 #include <magick/version.h>
45 #include <libdcp/version.h>
46 extern "C" {
47 #include <libavcodec/avcodec.h>
48 #include <libavformat/avformat.h>
49 #include <libswscale/swscale.h>
50 #include <libavfilter/avfiltergraph.h>
51 #include <libpostproc/postprocess.h>
52 #include <libavutil/pixfmt.h>
53 }
54 #include "util.h"
55 #include "exceptions.h"
56 #include "scaler.h"
57 #include "format.h"
58 #include "dcp_content_type.h"
59 #include "filter.h"
60 #include "sound_processor.h"
61
62 using namespace std;
63 using namespace boost;
64
65 thread::id ui_thread;
66
67 /** Convert some number of seconds to a string representation
68  *  in hours, minutes and seconds.
69  *
70  *  @param s Seconds.
71  *  @return String of the form H:M:S (where H is hours, M
72  *  is minutes and S is seconds).
73  */
74 string
75 seconds_to_hms (int s)
76 {
77         int m = s / 60;
78         s -= (m * 60);
79         int h = m / 60;
80         m -= (h * 60);
81
82         stringstream hms;
83         hms << h << ":";
84         hms.width (2);
85         hms << setfill ('0') << m << ":";
86         hms.width (2);
87         hms << setfill ('0') << s;
88
89         return hms.str ();
90 }
91
92 /** @param s Number of seconds.
93  *  @return String containing an approximate description of s (e.g. "about 2 hours")
94  */
95 string
96 seconds_to_approximate_hms (int s)
97 {
98         int m = s / 60;
99         s -= (m * 60);
100         int h = m / 60;
101         m -= (h * 60);
102
103         stringstream ap;
104         
105         if (h > 0) {
106                 if (m > 30) {
107                         ap << (h + 1) << " hours";
108                 } else {
109                         if (h == 1) {
110                                 ap << "1 hour";
111                         } else {
112                                 ap << h << " hours";
113                         }
114                 }
115         } else if (m > 0) {
116                 if (m == 1) {
117                         ap << "1 minute";
118                 } else {
119                         ap << m << " minutes";
120                 }
121         } else {
122                 ap << s << " seconds";
123         }
124
125         return ap.str ();
126 }
127
128 #ifdef DVDOMATIC_POSIX
129 /** @param l Mangled C++ identifier.
130  *  @return Demangled version.
131  */
132 static string
133 demangle (string l)
134 {
135         string::size_type const b = l.find_first_of ("(");
136         if (b == string::npos) {
137                 return l;
138         }
139
140         string::size_type const p = l.find_last_of ("+");
141         if (p == string::npos) {
142                 return l;
143         }
144
145         if ((p - b) <= 1) {
146                 return l;
147         }
148         
149         string const fn = l.substr (b + 1, p - b - 1);
150
151         int status;
152         try {
153                 
154                 char* realname = abi::__cxa_demangle (fn.c_str(), 0, 0, &status);
155                 string d (realname);
156                 free (realname);
157                 return d;
158                 
159         } catch (std::exception) {
160                 
161         }
162         
163         return l;
164 }
165
166 /** Write a stacktrace to an ostream.
167  *  @param out Stream to write to.
168  *  @param levels Number of levels to go up the call stack.
169  */
170 void
171 stacktrace (ostream& out, int levels)
172 {
173         void *array[200];
174         size_t size;
175         char **strings;
176         size_t i;
177      
178         size = backtrace (array, 200);
179         strings = backtrace_symbols (array, size);
180      
181         if (strings) {
182                 for (i = 0; i < size && (levels == 0 || i < size_t(levels)); i++) {
183                         out << "  " << demangle (strings[i]) << endl;
184                 }
185                 
186                 free (strings);
187         }
188 }
189 #endif
190
191 /** @param v Version as used by FFmpeg.
192  *  @return A string representation of v.
193  */
194 static string
195 ffmpeg_version_to_string (int v)
196 {
197         stringstream s;
198         s << ((v & 0xff0000) >> 16) << "." << ((v & 0xff00) >> 8) << "." << (v & 0xff);
199         return s.str ();
200 }
201
202 /** Return a user-readable string summarising the versions of our dependencies */
203 string
204 dependency_version_summary ()
205 {
206         stringstream s;
207         s << "libopenjpeg " << opj_version () << ", "
208           << "libavcodec " << ffmpeg_version_to_string (avcodec_version()) << ", "
209           << "libavfilter " << ffmpeg_version_to_string (avfilter_version()) << ", "
210           << "libavformat " << ffmpeg_version_to_string (avformat_version()) << ", "
211           << "libavutil " << ffmpeg_version_to_string (avutil_version()) << ", "
212           << "libpostproc " << ffmpeg_version_to_string (postproc_version()) << ", "
213           << "libswscale " << ffmpeg_version_to_string (swscale_version()) << ", "
214           << MagickVersion << ", "
215           << "libssh " << ssh_version (0) << ", "
216           << "libdcp " << libdcp::version << " git " << libdcp::git_commit;
217
218         return s.str ();
219 }
220
221 double
222 seconds (struct timeval t)
223 {
224         return t.tv_sec + (double (t.tv_usec) / 1e6);
225 }
226
227 /** Call the required functions to set up DVD-o-matic's static arrays, etc.
228  *  Must be called from the UI thread, if there is one.
229  */
230 void
231 dvdomatic_setup ()
232 {
233         Format::setup_formats ();
234         DCPContentType::setup_dcp_content_types ();
235         Scaler::setup_scalers ();
236         Filter::setup_filters ();
237         SoundProcessor::setup_sound_processors ();
238
239         ui_thread = this_thread::get_id ();
240 }
241
242 /** @param start Start position for the crop within the image.
243  *  @param size Size of the cropped area.
244  *  @return FFmpeg crop filter string.
245  */
246 string
247 crop_string (Position start, libdcp::Size size)
248 {
249         stringstream s;
250         s << "crop=" << size.width << ":" << size.height << ":" << start.x << ":" << start.y;
251         return s.str ();
252 }
253
254 /** @param s A string.
255  *  @return Parts of the string split at spaces, except when a space is within quotation marks.
256  */
257 vector<string>
258 split_at_spaces_considering_quotes (string s)
259 {
260         vector<string> out;
261         bool in_quotes = false;
262         string c;
263         for (string::size_type i = 0; i < s.length(); ++i) {
264                 if (s[i] == ' ' && !in_quotes) {
265                         out.push_back (c);
266                         c = "";
267                 } else if (s[i] == '"') {
268                         in_quotes = !in_quotes;
269                 } else {
270                         c += s[i];
271                 }
272         }
273
274         out.push_back (c);
275         return out;
276 }
277
278 string
279 md5_digest (void const * data, int size)
280 {
281         MD5_CTX md5_context;
282         MD5_Init (&md5_context);
283         MD5_Update (&md5_context, data, size);
284         unsigned char digest[MD5_DIGEST_LENGTH];
285         MD5_Final (digest, &md5_context);
286         
287         stringstream s;
288         for (int i = 0; i < MD5_DIGEST_LENGTH; ++i) {
289                 s << hex << setfill('0') << setw(2) << ((int) digest[i]);
290         }
291
292         return s.str ();
293 }
294
295 /** @param file File name.
296  *  @return MD5 digest of file's contents.
297  */
298 string
299 md5_digest (string file)
300 {
301         ifstream f (file.c_str(), ios::binary);
302         if (!f.good ()) {
303                 throw OpenFileError (file);
304         }
305         
306         f.seekg (0, ios::end);
307         int bytes = f.tellg ();
308         f.seekg (0, ios::beg);
309
310         int const buffer_size = 64 * 1024;
311         char buffer[buffer_size];
312
313         MD5_CTX md5_context;
314         MD5_Init (&md5_context);
315         while (bytes > 0) {
316                 int const t = min (bytes, buffer_size);
317                 f.read (buffer, t);
318                 MD5_Update (&md5_context, buffer, t);
319                 bytes -= t;
320         }
321
322         unsigned char digest[MD5_DIGEST_LENGTH];
323         MD5_Final (digest, &md5_context);
324
325         stringstream s;
326         for (int i = 0; i < MD5_DIGEST_LENGTH; ++i) {
327                 s << hex << setfill('0') << setw(2) << ((int) digest[i]);
328         }
329
330         return s.str ();
331 }
332
333 /** @param fps Arbitrary frames-per-second value.
334  *  @return DCPFrameRate for this frames-per-second.
335  */
336 DCPFrameRate
337 dcp_frame_rate (float fps)
338 {
339         DCPFrameRate dfr;
340
341         dfr.run_fast = (fps != rint (fps));
342         dfr.frames_per_second = rint (fps);
343         dfr.skip = 1;
344
345         /* XXX: somewhat arbitrary */
346         if (fps == 50) {
347                 dfr.frames_per_second = 25;
348                 dfr.skip = 2;
349         }
350
351         return dfr;
352 }
353
354 /** @param An arbitrary sampling rate.
355  *  @return The appropriate DCP-approved sampling rate (48kHz or 96kHz).
356  */
357 int
358 dcp_audio_sample_rate (int fs)
359 {
360         if (fs <= 48000) {
361                 return 48000;
362         }
363
364         return 96000;
365 }
366
367 int
368 dcp_audio_channels (int f)
369 {
370         if (f == 1) {
371                 /* The source is mono, so to put the mono channel into
372                    the centre we need to generate a 5.1 soundtrack.
373                 */
374                 return 6;
375         }
376
377         return f;
378 }
379
380 bool operator== (Crop const & a, Crop const & b)
381 {
382         return (a.left == b.left && a.right == b.right && a.top == b.top && a.bottom == b.bottom);
383 }
384
385 bool operator!= (Crop const & a, Crop const & b)
386 {
387         return !(a == b);
388 }
389
390 /** @param index Colour LUT index.
391  *  @return Human-readable name.
392  */
393 string
394 colour_lut_index_to_name (int index)
395 {
396         switch (index) {
397         case 0:
398                 return "sRGB";
399         case 1:
400                 return "Rec 709";
401         }
402
403         assert (false);
404         return "";
405 }
406
407 Socket::Socket ()
408         : _deadline (_io_service)
409         , _socket (_io_service)
410         , _buffer_data (0)
411 {
412         _deadline.expires_at (posix_time::pos_infin);
413         check ();
414 }
415
416 void
417 Socket::check ()
418 {
419         if (_deadline.expires_at() <= asio::deadline_timer::traits_type::now ()) {
420                 _socket.close ();
421                 _deadline.expires_at (posix_time::pos_infin);
422         }
423
424         _deadline.async_wait (boost::bind (&Socket::check, this));
425 }
426
427 /** Blocking connect with timeout.
428  *  @param endpoint End-point to connect to.
429  *  @param timeout Time-out in seconds.
430  */
431 void
432 Socket::connect (asio::ip::basic_resolver_entry<asio::ip::tcp> const & endpoint, int timeout)
433 {
434         _deadline.expires_from_now (posix_time::seconds (timeout));
435         system::error_code ec = asio::error::would_block;
436         _socket.async_connect (endpoint, lambda::var(ec) = lambda::_1);
437         do {
438                 _io_service.run_one();
439         } while (ec == asio::error::would_block);
440
441         if (ec || !_socket.is_open ()) {
442                 throw NetworkError ("connect timed out");
443         }
444 }
445
446 /** Blocking write with timeout.
447  *  @param data Buffer to write.
448  *  @param size Number of bytes to write.
449  *  @param timeout Time-out, in seconds.
450  */
451 void
452 Socket::write (uint8_t const * data, int size, int timeout)
453 {
454         _deadline.expires_from_now (posix_time::seconds (timeout));
455         system::error_code ec = asio::error::would_block;
456
457         asio::async_write (_socket, asio::buffer (data, size), lambda::var(ec) = lambda::_1);
458         do {
459                 _io_service.run_one ();
460         } while (ec == asio::error::would_block);
461
462         if (ec) {
463                 throw NetworkError ("write timed out");
464         }
465 }
466
467 /** Blocking read with timeout.
468  *  @param data Buffer to read to.
469  *  @param size Number of bytes to read.
470  *  @param timeout Time-out, in seconds.
471  */
472 int
473 Socket::read (uint8_t* data, int size, int timeout)
474 {
475         _deadline.expires_from_now (posix_time::seconds (timeout));
476         system::error_code ec = asio::error::would_block;
477
478         int amount_read = 0;
479
480         _socket.async_read_some (
481                 asio::buffer (data, size),
482                 (lambda::var(ec) = lambda::_1, lambda::var(amount_read) = lambda::_2)
483                 );
484
485         do {
486                 _io_service.run_one ();
487         } while (ec == asio::error::would_block);
488         
489         if (ec) {
490                 amount_read = 0;
491         }
492
493         return amount_read;
494 }
495
496 /** Mark some data as being `consumed', so that it will not be returned
497  *  as data again.
498  *  @param size Amount of data to consume, in bytes.
499  */
500 void
501 Socket::consume (int size)
502 {
503         assert (_buffer_data >= size);
504         
505         _buffer_data -= size;
506         if (_buffer_data > 0) {
507                 /* Shift still-valid data to the start of the buffer */
508                 memmove (_buffer, _buffer + size, _buffer_data);
509         }
510 }
511
512 /** Read a definite amount of data from our socket, and mark
513  *  it as consumed.
514  *  @param data Where to put the data.
515  *  @param size Number of bytes to read.
516  */
517 void
518 Socket::read_definite_and_consume (uint8_t* data, int size, int timeout)
519 {
520         int const from_buffer = min (_buffer_data, size);
521         if (from_buffer > 0) {
522                 /* Get data from our buffer */
523                 memcpy (data, _buffer, from_buffer);
524                 consume (from_buffer);
525                 /* Update our output state */
526                 data += from_buffer;
527                 size -= from_buffer;
528         }
529
530         /* read() the rest */
531         while (size > 0) {
532                 int const n = read (data, size, timeout);
533                 if (n <= 0) {
534                         throw NetworkError ("could not read");
535                 }
536
537                 data += n;
538                 size -= n;
539         }
540 }
541
542 /** Read as much data as is available, up to some limit.
543  *  @param data Where to put the data.
544  *  @param size Maximum amount of data to read.
545  */
546 void
547 Socket::read_indefinite (uint8_t* data, int size, int timeout)
548 {
549         assert (size < int (sizeof (_buffer)));
550
551         /* Amount of extra data we need to read () */
552         int to_read = size - _buffer_data;
553         while (to_read > 0) {
554                 /* read as much of it as we can (into our buffer) */
555                 int const n = read (_buffer + _buffer_data, to_read, timeout);
556                 if (n <= 0) {
557                         throw NetworkError ("could not read");
558                 }
559
560                 to_read -= n;
561                 _buffer_data += n;
562         }
563
564         assert (_buffer_data >= size);
565
566         /* copy data into the output buffer */
567         assert (size >= _buffer_data);
568         memcpy (data, _buffer, size);
569 }
570
571 /** @param other A Rect.
572  *  @return The intersection of this with `other'.
573  */
574 Rect
575 Rect::intersection (Rect const & other) const
576 {
577         int const tx = max (x, other.x);
578         int const ty = max (y, other.y);
579         
580         return Rect (
581                 tx, ty,
582                 min (x + width, other.x + other.width) - tx,
583                 min (y + height, other.y + other.height) - ty
584                 );
585 }
586
587 /** Round a number up to the nearest multiple of another number.
588  *  @param c Index.
589  *  @param s Array of numbers to round, indexed by c.
590  *  @param t Multiple to round to.
591  *  @return Rounded number.
592  */
593 int
594 stride_round_up (int c, int const * stride, int t)
595 {
596         int const a = stride[c] + (t - 1);
597         return a - (a % t);
598 }
599
600 int
601 stride_lookup (int c, int const * stride)
602 {
603         return stride[c];
604 }
605
606 /** Read a sequence of key / value pairs from a text stream;
607  *  the keys are the first words on the line, and the values are
608  *  the remainder of the line following the key.  Lines beginning
609  *  with # are ignored.
610  *  @param s Stream to read.
611  *  @return key/value pairs.
612  */
613 multimap<string, string>
614 read_key_value (istream &s) 
615 {
616         multimap<string, string> kv;
617         
618         string line;
619         while (getline (s, line)) {
620                 if (line.empty ()) {
621                         continue;
622                 }
623
624                 if (line[0] == '#') {
625                         continue;
626                 }
627
628                 if (line[line.size() - 1] == '\r') {
629                         line = line.substr (0, line.size() - 1);
630                 }
631
632                 size_t const s = line.find (' ');
633                 if (s == string::npos) {
634                         continue;
635                 }
636
637                 kv.insert (make_pair (line.substr (0, s), line.substr (s + 1)));
638         }
639
640         return kv;
641 }
642
643 string
644 get_required_string (multimap<string, string> const & kv, string k)
645 {
646         if (kv.count (k) > 1) {
647                 throw StringError ("unexpected multiple keys in key-value set");
648         }
649
650         multimap<string, string>::const_iterator i = kv.find (k);
651         
652         if (i == kv.end ()) {
653                 throw StringError (String::compose ("missing key %1 in key-value set", k));
654         }
655
656         return i->second;
657 }
658
659 int
660 get_required_int (multimap<string, string> const & kv, string k)
661 {
662         string const v = get_required_string (kv, k);
663         return lexical_cast<int> (v);
664 }
665
666 float
667 get_required_float (multimap<string, string> const & kv, string k)
668 {
669         string const v = get_required_string (kv, k);
670         return lexical_cast<float> (v);
671 }
672
673 string
674 get_optional_string (multimap<string, string> const & kv, string k)
675 {
676         if (kv.count (k) > 1) {
677                 throw StringError ("unexpected multiple keys in key-value set");
678         }
679
680         multimap<string, string>::const_iterator i = kv.find (k);
681         if (i == kv.end ()) {
682                 return "";
683         }
684
685         return i->second;
686 }
687
688 int
689 get_optional_int (multimap<string, string> const & kv, string k)
690 {
691         if (kv.count (k) > 1) {
692                 throw StringError ("unexpected multiple keys in key-value set");
693         }
694
695         multimap<string, string>::const_iterator i = kv.find (k);
696         if (i == kv.end ()) {
697                 return 0;
698         }
699
700         return lexical_cast<int> (i->second);
701 }
702
703 /** Construct an AudioBuffers.  Audio data is undefined after this constructor.
704  *  @param channels Number of channels.
705  *  @param frames Number of frames to reserve space for.
706  */
707 AudioBuffers::AudioBuffers (int channels, int frames)
708         : _channels (channels)
709         , _frames (frames)
710         , _allocated_frames (frames)
711 {
712         _data = new float*[_channels];
713         for (int i = 0; i < _channels; ++i) {
714                 _data[i] = new float[frames];
715         }
716 }
717
718 /** Copy constructor.
719  *  @param other Other AudioBuffers; data is copied.
720  */
721 AudioBuffers::AudioBuffers (AudioBuffers const & other)
722         : _channels (other._channels)
723         , _frames (other._frames)
724         , _allocated_frames (other._frames)
725 {
726         _data = new float*[_channels];
727         for (int i = 0; i < _channels; ++i) {
728                 _data[i] = new float[_frames];
729                 memcpy (_data[i], other._data[i], _frames * sizeof (float));
730         }
731 }
732
733 /** AudioBuffers destructor */
734 AudioBuffers::~AudioBuffers ()
735 {
736         for (int i = 0; i < _channels; ++i) {
737                 delete[] _data[i];
738         }
739
740         delete[] _data;
741 }
742
743 /** @param c Channel index.
744  *  @return Buffer for this channel.
745  */
746 float*
747 AudioBuffers::data (int c) const
748 {
749         assert (c >= 0 && c < _channels);
750         return _data[c];
751 }
752
753 /** Set the number of frames that these AudioBuffers will report themselves
754  *  as having.
755  *  @param f Frames; must be less than or equal to the number of allocated frames.
756  */
757 void
758 AudioBuffers::set_frames (int f)
759 {
760         assert (f <= _allocated_frames);
761         _frames = f;
762 }
763
764 /** Make all samples on all channels silent */
765 void
766 AudioBuffers::make_silent ()
767 {
768         for (int i = 0; i < _channels; ++i) {
769                 make_silent (i);
770         }
771 }
772
773 /** Make all samples on a given channel silent.
774  *  @param c Channel.
775  */
776 void
777 AudioBuffers::make_silent (int c)
778 {
779         assert (c >= 0 && c < _channels);
780         
781         for (int i = 0; i < _frames; ++i) {
782                 _data[c][i] = 0;
783         }
784 }
785
786 /** Copy data from another AudioBuffers to this one.  All channels are copied.
787  *  @param from AudioBuffers to copy from; must have the same number of channels as this.
788  *  @param frames_to_copy Number of frames to copy.
789  *  @param read_offset Offset to read from in `from'.
790  *  @param write_offset Offset to write to in `to'.
791  */
792 void
793 AudioBuffers::copy_from (AudioBuffers* from, int frames_to_copy, int read_offset, int write_offset)
794 {
795         assert (from->channels() == channels());
796
797         assert (from);
798         assert (read_offset >= 0 && (read_offset + frames_to_copy) <= from->_allocated_frames);
799         assert (write_offset >= 0 && (write_offset + frames_to_copy) <= _allocated_frames);
800
801         for (int i = 0; i < _channels; ++i) {
802                 memcpy (_data[i] + write_offset, from->_data[i] + read_offset, frames_to_copy * sizeof(float));
803         }
804 }
805
806 /** Move audio data around.
807  *  @param from Offset to move from.
808  *  @param to Offset to move to.
809  *  @param frames Number of frames to move.
810  */
811     
812 void
813 AudioBuffers::move (int from, int to, int frames)
814 {
815         if (frames == 0) {
816                 return;
817         }
818         
819         assert (from >= 0);
820         assert (from < _frames);
821         assert (to >= 0);
822         assert (to < _frames);
823         assert (frames > 0);
824         assert (frames <= _frames);
825         assert ((from + frames) <= _frames);
826         assert ((to + frames) <= _frames);
827         
828         for (int i = 0; i < _channels; ++i) {
829                 memmove (_data[i] + to, _data[i] + from, frames * sizeof(float));
830         }
831 }
832
833 /** Trip an assert if the caller is not in the UI thread */
834 void
835 ensure_ui_thread ()
836 {
837         assert (this_thread::get_id() == ui_thread);
838 }
839
840 /** @param v Source video frame.
841  *  @param audio_sample_rate Source audio sample rate.
842  *  @param frames_per_second Number of video frames per second.
843  *  @return Equivalent number of audio frames for `v'.
844  */
845 int64_t
846 video_frames_to_audio_frames (SourceFrame v, float audio_sample_rate, float frames_per_second)
847 {
848         return ((int64_t) v * audio_sample_rate / frames_per_second);
849 }
850
851 /** @param f Filename.
852  *  @return true if this file is a still image, false if it is something else.
853  */
854 bool
855 still_image_file (string f)
856 {
857 #if BOOST_FILESYSTEM_VERSION == 3
858         string ext = boost::filesystem::path(f).extension().string();
859 #else
860         string ext = boost::filesystem::path(f).extension();
861 #endif
862
863         transform (ext.begin(), ext.end(), ext.begin(), ::tolower);
864         
865         return (ext == ".tif" || ext == ".tiff" || ext == ".jpg" || ext == ".jpeg" || ext == ".png");
866 }
867
868 /** @return A pair containing CPU model name and the number of processors */
869 pair<string, int>
870 cpu_info ()
871 {
872         pair<string, int> info;
873         info.second = 0;
874         
875 #ifdef DVDOMATIC_POSIX
876         ifstream f ("/proc/cpuinfo");
877         while (f.good ()) {
878                 string l;
879                 getline (f, l);
880                 if (boost::algorithm::starts_with (l, "model name")) {
881                         string::size_type const c = l.find (':');
882                         if (c != string::npos) {
883                                 info.first = l.substr (c + 2);
884                         }
885                 } else if (boost::algorithm::starts_with (l, "processor")) {
886                         ++info.second;
887                 }
888         }
889 #endif  
890
891         return info;
892 }