Try to fix the filter / AVFrame ownership.

[dcpomatic.git] / src / lib / imagemagick_decoder.cc

diff --git a/src/lib/imagemagick_decoder.cc b/src/lib/imagemagick_decoder.cc
index 7cee01ec56c52ef1eb0dd637e5e71a5bee5a570f..5ce22c29622a403c33425a5ce6e8e9a5f3e26de5 100644 (file)
--- a/src/lib/imagemagick_decoder.cc
+++ b/src/lib/imagemagick_decoder.cc
@@ -1,55 +1,158 @@
+/*
+    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, write to the Free Software
+    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+*/
+
 #include <iostream>
-#include <Magick++/Image.h>
+#include <boost/filesystem.hpp>
+#include <Magick++.h>
 #include "imagemagick_decoder.h"
-#include "film_state.h"
 #include "image.h"
+#include "film.h"
+#include "exceptions.h"
+
+#include "i18n.h"
 
-using namespace std;
+using std::cout;
+using boost::shared_ptr;
+using libdcp::Size;
 
 ImageMagickDecoder::ImageMagickDecoder (
-       boost::shared_ptr<const FilmState> s, boost::shared_ptr<const Options> o, Job* j, Log* l, bool minimal, bool ignore_length)
-       : Decoder (s, o, j, l, minimal, ignore_length)
-       , _done (false)
+       boost::shared_ptr<Film> f, DecodeOptions o)
+       : Decoder (f, o)
+       , VideoDecoder (f, o)
 {
-       _magick_image = new Magick::Image (_fs->content_path ());
+       if (boost::filesystem::is_directory (_film->content_path())) {
+               for (
+                       boost::filesystem::directory_iterator i = boost::filesystem::directory_iterator (_film->content_path());
+                       i != boost::filesystem::directory_iterator();
+                       ++i) {
+
+                       if (still_image_file (i->path().string())) {
+                               _files.push_back (i->path().string());
+                       }
+               }
+       } else {
+               _files.push_back (_film->content_path ());
+       }
+
+       _iter = _files.begin ();
 }
 
-Size
+libdcp::Size
 ImageMagickDecoder::native_size () const
 {
-       return Size (_magick_image->columns(), _magick_image->rows());
+       if (_files.empty ()) {
+               throw DecodeError (_("no still image files found"));
+       }
+
+       /* Look at the first file and assume its size holds for all */
+       using namespace MagickCore;
+       Magick::Image* image = new Magick::Image (_film->content_path ());
+       libdcp::Size const s = libdcp::Size (image->columns(), image->rows());
+       delete image;
+
+       return s;
 }
 
 bool
-ImageMagickDecoder::do_pass ()
+ImageMagickDecoder::pass ()
 {
-       if (_done) {
-               return true;
-       }
+       if (_iter == _files.end()) {
+               if (video_frame() >= _film->still_duration_in_frames()) {
+                       return true;
+               }
 
-       Size size = native_size ();
-       RGBFrameImage image (size);
+               emit_video (_image, true, double (video_frame()) / frames_per_second());
+               return false;
+       }
+       
+       Magick::Image* magick_image = new Magick::Image (_film->content_path ());
+       
+       libdcp::Size size = native_size ();
+       shared_ptr<Image> image (new SimpleImage (PIX_FMT_RGB24, size, false));
 
-       uint8_t* p = image.data()[0];
+       using namespace MagickCore;
+       
+       uint8_t* p = image->data()[0];
        for (int y = 0; y < size.height; ++y) {
                for (int x = 0; x < size.width; ++x) {
-                       Magick::Color c = _magick_image->pixelColor (x, y);
-                       *p++ = c.redQuantum() * 255 / MaxRGB;
-                       *p++ = c.greenQuantum() * 255 / MaxRGB;
-                       *p++ = c.blueQuantum() * 255 / MaxRGB;
+                       Magick::Color c = magick_image->pixelColor (x, y);
+                       *p++ = c.redQuantum() * 255 / QuantumRange;
+                       *p++ = c.greenQuantum() * 255 / QuantumRange;
+                       *p++ = c.blueQuantum() * 255 / QuantumRange;
                }
-
        }
-       
-       process_video (image.frame ());
 
-       _done = true;
+       delete magick_image;
+
+       _image = image->crop (_film->crop(), true);
+
+       emit_video (_image, false, double (video_frame()) / frames_per_second());
+
+       ++_iter;
        return false;
 }
 
 PixelFormat
 ImageMagickDecoder::pixel_format () const
 {
+       /* XXX: always true? */
        return PIX_FMT_RGB24;
 }
 
+bool
+ImageMagickDecoder::seek_to_last ()
+{
+       if (_iter == _files.end()) {
+               _iter = _files.begin();
+       } else {
+               --_iter;
+       }
+
+       return false;
+}
+
+bool
+ImageMagickDecoder::seek (double t)
+{
+       int const f = t * frames_per_second();
+       
+       _iter = _files.begin ();
+       for (int i = 0; i < f; ++i) {
+               if (_iter == _files.end()) {
+                       return true;
+               }
+               ++_iter;
+       }
+       
+       return false;
+}
+
+void
+ImageMagickDecoder::film_changed (Film::Property p)
+{
+       if (p == Film::CROP) {
+               OutputChanged ();
+       }
+}
+
+float
+ImageMagickDecoder::frames_per_second () const
+{
+       return _film->source_frame_rate ();
+}