/*
- 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);
-
- for (unsigned int i = 0; i < sub.num_rects; ++i) {
- _images.push_back (shared_ptr<SubtitleImage> (new SubtitleImage (sub.rects[i])));
- }
+ update (film, video_container_size);
}
-/** @param t Time in seconds from the start of the film */
-bool
-Subtitle::displayed_at (double t)
+void
+Subtitle::update (shared_ptr<const Film> film, libdcp::Size video_container_size)
{
- return t >= _from && t <= _to;
-}
-
-SubtitleImage::SubtitleImage (AVSubtitleRect const * rect)
- : _position (rect->x, rect->y)
- , _image (new AlignedImage (PIX_FMT_RGBA, Size (rect->w, rect->h)))
-{
- if (rect->type != SUBTITLE_BITMAP) {
- throw DecodeError ("non-bitmap subtitles not yet supported");
+ shared_ptr<Piece> piece = _piece.lock ();
+ if (!piece) {
+ return;
}
- /* 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);
+ if (!_in_image) {
+ _out_image.reset ();
+ return;
}
-}
-Rectangle
-transformed_subtitle_area (
- float target_x_scale, float target_y_scale,
- Rectangle sub_area, int subtitle_offset, float subtitle_scale
- )
-{
- Rectangle tx;
+ shared_ptr<SubtitleContent> sc = dynamic_pointer_cast<SubtitleContent> (piece->content);
+ assert (sc);
+
+ 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
* 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 ();
- return tx;
+ check_out_to ();
}
-Rectangle
-SubtitleImage::area () const
+bool
+Subtitle::covers (Time t) const
{
- return Rectangle (_position.x, _position.y, _image->size().width, _image->size().height);
+ return _out_from <= t && t <= _out_to;
+}
+
+void
+Subtitle::check_out_to ()
+{
+ if (_stop && _out_to > _stop.get ()) {
+ _out_to = _stop.get ();
+ }
}