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, c)
  38         , _imagemagick_content (c)
  39 {
  40
  41 }
  42
  43 libdcp::Size
  44 ImageMagickDecoder::video_size () const
  45 {
  46         if (!_video_size) {
  47                 using namespace MagickCore;
  48                 Magick::Image* image = new Magick::Image (_imagemagick_content->file().string());
  49                 _video_size = libdcp::Size (image->columns(), image->rows());
  50                 delete image;
  51         }
  52
  53         return _video_size.get ();
  54 }
  55
  56 int
  57 ImageMagickDecoder::video_length () const
  58 {
  59         return _imagemagick_content->video_length ();
  60 }
  61
  62 float
  63 ImageMagickDecoder::video_frame_rate () const
  64 {
  65         boost::shared_ptr<const Film> f = _film.lock ();
  66         if (!f) {
  67                 return 24;
  68         }
  69
  70         return f->dcp_video_frame_rate ();
  71 }
  72
  73 void
  74 ImageMagickDecoder::pass ()
  75 {
  76         if (_position < 0 || _position >= _imagemagick_content->video_length ()) {
  77                 return;
  78         }
  79
  80         if (_image) {
  81                 video (_image, true, double (_position) / video_frame_rate());
  82                 _position++;
  83                 return;
  84         }
  85
  86         Magick::Image* magick_image = new Magick::Image (_imagemagick_content->file().string ());
  87         _video_size = libdcp::Size (magick_image->columns(), magick_image->rows());
  88
  89         _image.reset (new SimpleImage (PIX_FMT_RGB24, _video_size.get(), false));
  90
  91         using namespace MagickCore;
  92
  93         uint8_t* p = _image->data()[0];
  94         for (int y = 0; y < _video_size->height; ++y) {
  95                 for (int x = 0; x < _video_size->width; ++x) {
  96                         Magick::Color c = magick_image->pixelColor (x, y);
  97                         *p++ = c.redQuantum() * 255 / QuantumRange;
  98                         *p++ = c.greenQuantum() * 255 / QuantumRange;
  99                         *p++ = c.blueQuantum() * 255 / QuantumRange;
 100                 }
 101         }
 102
 103         delete magick_image;
 104
 105         _image = _image->crop (_imagemagick_content->crop(), true);
 106         video (_image, false, double (_position) / 24);
 107
 108         ++_position;
 109         return false;
 110 }
 111
 112 PixelFormat
 113 ImageMagickDecoder::pixel_format () const
 114 {
 115         /* XXX: always true? */
 116         return PIX_FMT_RGB24;
 117 }
 118
 119 bool
 120 ImageMagickDecoder::seek (double t)
 121 {
 122         int const f = t * _imagemagick_content->video_frame_rate ();
 123
 124         if (f >= _imagemagick_content->video_length()) {
 125                 _position = 0;
 126                 return true;
 127         }
 128
 129         _position = f;
 130         return false;
 131 }
 132
 133 Time
 134 ImageMagickDecoder::next () const
 135 {
 136         return _next_video;
 137 }