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