Untested merge of master.
[dcpomatic.git] / src / lib / imagemagick_decoder.cc
1 /*
2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include <iostream>
21 #include <boost/filesystem.hpp>
22 #include <Magick++.h>
23 #include "imagemagick_content.h"
24 #include "imagemagick_decoder.h"
25 #include "image.h"
26 #include "film.h"
27 #include "exceptions.h"
28
29 #include "i18n.h"
30
31 using std::cout;
32 using boost::shared_ptr;
33 using libdcp::Size;
34
35 ImageMagickDecoder::ImageMagickDecoder (shared_ptr<const Film> f, shared_ptr<const ImageMagickContent> c)
36         : Decoder (f)
37         , VideoDecoder (f)
38         , _imagemagick_content (c)
39         , _position (0)
40 {
41         
42 }
43
44 libdcp::Size
45 ImageMagickDecoder::native_size () const
46 {
47         using namespace MagickCore;
48         Magick::Image* image = new Magick::Image (_imagemagick_content->file().string());
49         libdcp::Size const s = libdcp::Size (image->columns(), image->rows());
50         delete image;
51
52         return s;
53 }
54
55 int
56 ImageMagickDecoder::video_length () const
57 {
58         return _imagemagick_content->video_length ();
59 }
60
61 bool
62 ImageMagickDecoder::pass ()
63 {
64         if (_position < 0 || _position >= _imagemagick_content->video_length ()) {
65                 return true;
66         }
67
68         if (_image) {
69                 emit_video (_image, true, double (_position) / 24);
70                 _position++;
71                 return false;
72         }
73         
74         Magick::Image* magick_image = new Magick::Image (_imagemagick_content->file().string ());
75         
76         libdcp::Size size = native_size ();
77         _image.reset (new SimpleImage (PIX_FMT_RGB24, size, false));
78
79         using namespace MagickCore;
80         
81         uint8_t* p = _image->data()[0];
82         for (int y = 0; y < size.height; ++y) {
83                 for (int x = 0; x < size.width; ++x) {
84                         Magick::Color c = magick_image->pixelColor (x, y);
85                         *p++ = c.redQuantum() * 255 / QuantumRange;
86                         *p++ = c.greenQuantum() * 255 / QuantumRange;
87                         *p++ = c.blueQuantum() * 255 / QuantumRange;
88                 }
89         }
90
91         delete magick_image;
92
93         _image = _image->crop (_film->crop(), true);
94         emit_video (_image, false, double (_position) / 24);
95
96         ++_position;
97         return false;
98 }
99
100 PixelFormat
101 ImageMagickDecoder::pixel_format () const
102 {
103         /* XXX: always true? */
104         return PIX_FMT_RGB24;
105 }
106
107 bool
108 ImageMagickDecoder::seek (double t)
109 {
110         int const f = t * _imagemagick_content->video_frame_rate ();
111
112         if (f >= _imagemagick_content->video_length()) {
113                 _position = 0;
114                 return true;
115         }
116
117         _position = f;
118         return false;
119 }