Fix hidden advanced preferences button in some locales (e.g. de_DE).
[dcpomatic.git] / src / lib / subtitle.cc
index 2e9a890880ac9239dcd06f7b3ee12ab742feeaed..0d18861c455f69f2d2649730f93d898d0ed66463 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2013-2014 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
 */
 
 #include "subtitle.h"
+#include "subtitle_content.h"
+#include "piece.h"
 #include "image.h"
-#include "exceptions.h"
-#include "film_state.h"
-
-using namespace std;
-using namespace boost;
-
-Subtitle::Subtitle (AVSubtitle const & sub)
+#include "scaler.h"
+#include "film.h"
+
+using boost::shared_ptr;
+using boost::dynamic_pointer_cast;
+using boost::weak_ptr;
+
+Subtitle::Subtitle (shared_ptr<const Film> film, libdcp::Size video_container_size, weak_ptr<Piece> weak_piece, shared_ptr<Image> image, dcpomatic::Rect<double> rect, Time from, Time to)
+       : _piece (weak_piece)
+       , _in_image (image)
+       , _in_rect (rect)
+       , _in_from (from)
+       , _in_to (to)
 {
-       /* subtitle PTS in seconds */
-       float const packet_time = (sub.pts / AV_TIME_BASE) + float (sub.pts % AV_TIME_BASE) / 1e6;
-       
-       /* hence start time for this sub */
-       _from = packet_time + (double (sub.start_display_time) / 1e3);
-       _to = packet_time + (double (sub.end_display_time) / 1e3);
+       update (film, video_container_size);
+}
 
-       if (sub.num_rects > 1) {
-               throw DecodeError ("multi-part subtitles not yet supported");
+void
+Subtitle::update (shared_ptr<const Film> film, libdcp::Size video_container_size)
+{
+       shared_ptr<Piece> piece = _piece.lock ();
+       if (!piece) {
+               return;
        }
 
-       AVSubtitleRect const * rect = sub.rects[0];
-
-       if (rect->type != SUBTITLE_BITMAP) {
-               throw DecodeError ("non-bitmap subtitles not yet supported");
+       if (!_in_image) {
+               _out_image.reset ();
+               return;
        }
-       
-       _position = Position (rect->x, rect->y);
-       _image.reset (new AlignedImage (PIX_FMT_RGBA, Size (rect->w, rect->h)));
-
-       /* Start of the first line in the subtitle */
-       uint8_t* sub_p = rect->pict.data[0];
-       /* sub_p looks up into a RGB palette which is here */
-       uint32_t const * palette = (uint32_t *) rect->pict.data[1];
-       /* Start of the output data */
-       uint32_t* out_p = (uint32_t *) _image->data()[0];
-       
-       for (int y = 0; y < rect->h; ++y) {
-               uint8_t* sub_line_p = sub_p;
-               uint32_t* out_line_p = out_p;
-               for (int x = 0; x < rect->w; ++x) {
-                       *out_line_p++ = palette[*sub_line_p++];
-               }
-               sub_p += rect->pict.linesize[0];
-               out_p += _image->stride()[0] / sizeof (uint32_t);
-       }
-}
 
-Subtitle::Subtitle (Position p, shared_ptr<Image> i)
-       : _from (0)
-       , _to (0)
-       , _position (p)
-       , _image (i)
-{
-
-}
-       
-/** @param t Time in seconds from the start of the film */
-bool
-Subtitle::displayed_at (double t)
-{
-       return t >= _from && t <= _to;
-}
+       shared_ptr<SubtitleContent> sc = dynamic_pointer_cast<SubtitleContent> (piece->content);
+       assert (sc);
 
-Rectangle
-subtitle_transformed_area (
-       float target_x_scale, float target_y_scale,
-       Rectangle sub_area, int subtitle_offset, float subtitle_scale
-       )
-{
-       Rectangle tx;
+       dcpomatic::Rect<double> in_rect = _in_rect;
+       libdcp::Size scaled_size;
 
-       sub_area.y += subtitle_offset;
+       in_rect.x += sc->subtitle_x_offset ();
+       in_rect.y += sc->subtitle_y_offset ();
 
-       /* We will scale the subtitle by the same amount as the video frame, and also by the additional
-          subtitle_scale
-       */
-       tx.w = sub_area.w * target_x_scale * subtitle_scale;
-       tx.h = sub_area.h * target_y_scale * subtitle_scale;
+       /* We will scale the subtitle up to fit _video_container_size, and also by the additional subtitle_scale */
+       scaled_size.width = in_rect.width * video_container_size.width * sc->subtitle_scale ();
+       scaled_size.height = in_rect.height * video_container_size.height * sc->subtitle_scale ();
 
        /* Then we need a corrective translation, consisting of two parts:
         *
-        * 1.  that which is the result of the scaling of the subtitle by target_x_scale and target_y_scale; this will be
-        *     sub_area.x * target_x_scale and sub_area.y * target_y_scale.
+        * 1.  that which is the result of the scaling of the subtitle by _video_container_size; this will be
+        *     rect.x * _video_container_size.width and rect.y * _video_container_size.height.
         *
         * 2.  that to shift the origin of the scale by subtitle_scale to the centre of the subtitle; this will be
         *     (width_before_subtitle_scale * (1 - subtitle_scale) / 2) and
@@ -109,14 +76,41 @@ subtitle_transformed_area (
         * Combining these two translations gives these expressions.
         */
        
-       tx.x = target_x_scale * (sub_area.x + (sub_area.w * (1 - subtitle_scale) / 2));
-       tx.y = target_y_scale * (sub_area.y + (sub_area.h * (1 - subtitle_scale) / 2));
+       _out_position.x = rint (video_container_size.width * (in_rect.x + (in_rect.width * (1 - sc->subtitle_scale ()) / 2)));
+       _out_position.y = rint (video_container_size.height * (in_rect.y + (in_rect.height * (1 - sc->subtitle_scale ()) / 2)));
+       
+       _out_image = _in_image->scale (
+               scaled_size,
+               Scaler::from_id ("bicubic"),
+               _in_image->pixel_format (),
+               true
+               );
+
+       /* XXX: hack */
+       Time from = _in_from;
+       Time to = _in_to;
+       shared_ptr<VideoContent> vc = dynamic_pointer_cast<VideoContent> (piece->content);
+       if (vc) {
+               from = rint (from * vc->video_frame_rate() / film->video_frame_rate());
+               to = rint (to * vc->video_frame_rate() / film->video_frame_rate());
+       }
+       
+       _out_from = from + piece->content->position ();
+       _out_to = to + piece->content->position ();
+
+       check_out_to ();
+}
 
-       return tx;
+bool
+Subtitle::covers (Time t) const
+{
+       return _out_from <= t && t <= _out_to;
 }
 
-Rectangle
-Subtitle::area () const
+void
+Subtitle::check_out_to ()
 {
-       return Rectangle (_position.x, _position.y, _image->size().width, _image->size().height);
+       if (_stop && _out_to > _stop.get ()) {
+               _out_to = _stop.get ();
+       }
 }