Rename TYPE_DEBUG_PLAYER to TYPE_DEBUG_VIDEO_VIEW.
[dcpomatic.git] / src / lib / render_text.cc
index 8b9d93423d390062e53b710fba9feaf06b098a92..de33f9380b1b09dfd4a9cde380874aba93e9251f 100644 (file)
@@ -28,6 +28,7 @@
 #include <fontconfig/fontconfig.h>
 #include <cairomm/cairomm.h>
 #include <pangomm.h>
+#include <pango/pangocairo.h>
 #ifndef DCPOMATIC_HAVE_SHOW_IN_CAIRO_CONTEXT
 #include <pango/pangocairo.h>
 #endif
@@ -88,52 +89,20 @@ set_source_rgba (Cairo::RefPtr<Cairo::Context> context, dcp::Colour colour, floa
        context->set_source_rgba (float(colour.r) / 255, float(colour.g) / 255, float(colour.b) / 255, fade_factor);
 }
 
-/** @param subtitles A list of subtitles that are all on the same line,
- *  at the same time and with the same fade in/out.
- */
-static PositionImage
-render_line (list<StringText> subtitles, list<shared_ptr<Font> > fonts, dcp::Size target, DCPTime time, int frame_rate)
-{
-       /* XXX: this method can only handle italic / bold changes mid-line,
-          nothing else yet.
-       */
-
-       DCPOMATIC_ASSERT (!subtitles.empty ());
-
-       /* Calculate x and y scale factors.  These are only used to stretch
-          the font away from its normal aspect ratio.
-       */
-       float xscale = 1;
-       float yscale = 1;
-       if (fabs (subtitles.front().aspect_adjust() - 1.0) > dcp::ASPECT_ADJUST_EPSILON) {
-               if (subtitles.front().aspect_adjust() < 1) {
-                       xscale = max (0.25f, subtitles.front().aspect_adjust ());
-                       yscale = 1;
-               } else {
-                       xscale = 1;
-                       yscale = 1 / min (4.0f, subtitles.front().aspect_adjust ());
-               }
-       }
-
-       /* Make an empty bitmap as wide as target and at
-          least tall enough for this subtitle.
-       */
-
-       int largest = 0;
-       BOOST_FOREACH (dcp::SubtitleString const & i, subtitles) {
-               largest = max (largest, i.size());
-       }
-       /* Basic guess on height... */
-       int height = largest * target.height / (11 * 72);
-       /* ...scaled... */
-       height *= yscale;
-       /* ...and add a bit more for luck */
-       height += target.height / 11;
 
+static shared_ptr<Image>
+create_image (dcp::Size size)
+{
        /* FFmpeg BGRA means first byte blue, second byte green, third byte red, fourth byte alpha */
-       shared_ptr<Image> image (new Image (AV_PIX_FMT_BGRA, dcp::Size (target.width, height), false));
+       shared_ptr<Image> image (new Image(AV_PIX_FMT_BGRA, size, false));
        image->make_black ();
+       return image;
+}
 
+
+static Cairo::RefPtr<Cairo::ImageSurface>
+create_surface (shared_ptr<Image> image)
+{
 #ifdef DCPOMATIC_HAVE_FORMAT_STRIDE_FOR_WIDTH
        Cairo::RefPtr<Cairo::ImageSurface> surface = Cairo::ImageSurface::create (
                image->data()[0],
@@ -156,8 +125,13 @@ render_line (list<StringText> subtitles, list<shared_ptr<Font> > fonts, dcp::Siz
                );
 #endif
 
-       Cairo::RefPtr<Cairo::Context> context = Cairo::Context::create (surface);
+       return surface;
+}
+
 
+static string
+setup_font (StringText const& subtitle, list<shared_ptr<Font> > const& fonts)
+{
        if (!fc_config) {
                fc_config = FcInitLoadConfig ();
        }
@@ -177,7 +151,7 @@ render_line (list<StringText> subtitles, list<shared_ptr<Font> > fonts, dcp::Siz
        }
 
        BOOST_FOREACH (shared_ptr<Font> i, fonts) {
-               if (i->id() == subtitles.front().font() && i->file()) {
+               if (i->id() == subtitle.font() && i->file()) {
                        font_file = i->file ();
                }
        }
@@ -223,23 +197,22 @@ render_line (list<StringText> subtitles, list<shared_ptr<Font> > fonts, dcp::Siz
        }
 
        FcConfigSetCurrent (fc_config);
+       return font_name;
+}
 
-       Glib::RefPtr<Pango::Layout> layout = Pango::Layout::create (context);
-
-       layout->set_alignment (Pango::ALIGN_LEFT);
-
-       context->set_line_width (1);
 
-       /* Compute fade factor */
+static float
+calculate_fade_factor (StringText const& first, DCPTime time, int frame_rate)
+{
        float fade_factor = 1;
 
        /* Round the fade start/end to the nearest frame start.  Otherwise if a subtitle starts just after
           the start of a frame it will be faded out.
        */
-       DCPTime const fade_in_start = DCPTime::from_seconds(subtitles.front().in().as_seconds()).round(frame_rate);
-       DCPTime const fade_in_end = fade_in_start + DCPTime::from_seconds (subtitles.front().fade_up_time().as_seconds ());
-       DCPTime const fade_out_end =  DCPTime::from_seconds (subtitles.front().out().as_seconds()).round(frame_rate);
-       DCPTime const fade_out_start = fade_out_end - DCPTime::from_seconds (subtitles.front().fade_down_time().as_seconds ());
+       DCPTime const fade_in_start = DCPTime::from_seconds(first.in().as_seconds()).round(frame_rate);
+       DCPTime const fade_in_end = fade_in_start + DCPTime::from_seconds (first.fade_up_time().as_seconds ());
+       DCPTime const fade_out_end =  DCPTime::from_seconds (first.out().as_seconds()).round(frame_rate);
+       DCPTime const fade_out_start = fade_out_end - DCPTime::from_seconds (first.fade_down_time().as_seconds ());
 
        if (fade_in_start <= time && time <= fade_in_end && fade_in_start != fade_in_end) {
                fade_factor *= DCPTime(time - fade_in_start).seconds() / DCPTime(fade_in_end - fade_in_start).seconds();
@@ -251,34 +224,155 @@ render_line (list<StringText> subtitles, list<shared_ptr<Font> > fonts, dcp::Siz
                fade_factor = 0;
        }
 
-       /* Render the subtitle at the top left-hand corner of image */
+       return fade_factor;
+}
+
+
+static int
+x_position (StringText const& first, int target_width, int layout_width)
+{
+       int x = 0;
+       switch (first.h_align ()) {
+       case dcp::HALIGN_LEFT:
+               /* h_position is distance between left of frame and left of subtitle */
+               x = first.h_position() * target_width;
+               break;
+       case dcp::HALIGN_CENTER:
+               /* h_position is distance between centre of frame and centre of subtitle */
+               x = (0.5 + first.h_position()) * target_width - layout_width / 2;
+               break;
+       case dcp::HALIGN_RIGHT:
+               /* h_position is distance between right of frame and right of subtitle */
+               x = (1.0 - first.h_position()) * target_width - layout_width;
+               break;
+       }
+
+       return x;
+}
+
+
+
+static int
+y_position (StringText const& first, int target_height, int layout_height)
+{
+       int y = 0;
+       switch (first.v_align ()) {
+       case dcp::VALIGN_TOP:
+               /* SMPTE says that v_position is the distance between top
+                  of frame and top of subtitle, but this doesn't always seem to be
+                  the case in practice; Gunnar Ásgeirsson's Dolby server appears
+                  to put VALIGN_TOP subs with v_position as the distance between top
+                  of frame and bottom of subtitle.
+               */
+               y = first.v_position() * target_height - layout_height;
+               break;
+       case dcp::VALIGN_CENTER:
+               /* v_position is distance between centre of frame and centre of subtitle */
+               y = (0.5 + first.v_position()) * target_height - layout_height / 2;
+               break;
+       case dcp::VALIGN_BOTTOM:
+               /* v_position is distance between bottom of frame and bottom of subtitle */
+               y = (1.0 - first.v_position()) * target_height - layout_height;
+               break;
+       }
+
+       return y;
+}
+
 
+static void
+setup_layout (Glib::RefPtr<Pango::Layout> layout, string font_name, string markup)
+{
+       layout->set_alignment (Pango::ALIGN_LEFT);
        Pango::FontDescription font (font_name);
        layout->set_font_description (font);
-       layout->set_markup (marked_up (subtitles, target.height, fade_factor));
+       layout->set_markup (markup);
+}
 
-       context->scale (xscale, yscale);
-       layout->update_from_cairo_context (context);
+/** Create a Pango layout using a dummy context which we can use to calculate the size
+ *  of the text we will render.  Then we can transfer the layout over to the real context
+ *  for the actual render.
+ */
+static Glib::RefPtr<Pango::Layout>
+create_layout()
+{
+       PangoFontMap* c_font_map = pango_cairo_font_map_new ();
+       Glib::RefPtr<Pango::FontMap> font_map = Glib::wrap (c_font_map);
+       PangoContext* c_context = pango_font_map_create_context (c_font_map);
+       Glib::RefPtr<Pango::Context> context = Glib::wrap (c_context);
+       return Pango::Layout::create (context);
+}
+
+
+/** @param subtitles A list of subtitles that are all on the same line,
+ *  at the same time and with the same fade in/out.
+ */
+static PositionImage
+render_line (list<StringText> subtitles, list<shared_ptr<Font> > fonts, dcp::Size target, DCPTime time, int frame_rate)
+{
+       /* XXX: this method can only handle italic / bold changes mid-line,
+          nothing else yet.
+       */
+
+       DCPOMATIC_ASSERT (!subtitles.empty ());
+       StringText const& first = subtitles.front ();
+
+       string const font_name = setup_font (first, fonts);
+       float const fade_factor = calculate_fade_factor (first, time, frame_rate);
+       string const markup = marked_up (subtitles, target.height, fade_factor);
+       Glib::RefPtr<Pango::Layout> layout = create_layout ();
+       setup_layout (layout, font_name, markup);
+       dcp::Size size;
+       layout->get_pixel_size (size.width, size.height);
+
+       /* Calculate x and y scale factors.  These are only used to stretch
+          the font away from its normal aspect ratio.
+       */
+       float x_scale = 1;
+       float y_scale = 1;
+       if (fabs (first.aspect_adjust() - 1.0) > dcp::ASPECT_ADJUST_EPSILON) {
+               if (first.aspect_adjust() < 1) {
+                       x_scale = max (0.25f, first.aspect_adjust ());
+                       y_scale = 1;
+               } else {
+                       x_scale = 1;
+                       y_scale = 1 / min (4.0f, first.aspect_adjust ());
+               }
+       }
+
+       size.width *= x_scale;
+       size.height *= y_scale;
 
        /* Shuffle the subtitle over very slightly if it has a border so that the left-hand
           side of the first character's border is not cut off.
        */
-       int const x_offset = subtitles.front().effect() == dcp::BORDER ? (target.width / 600.0) : 0;
+       int const x_offset = first.effect() == dcp::BORDER ? (target.width / 600.0) : 0;
        /* Move down a bit so that accents on capital letters can be seen */
        int const y_offset = target.height / 100.0;
 
-       if (subtitles.front().effect() == dcp::SHADOW) {
+       size.width += x_offset;
+       size.height += y_offset;
+
+       shared_ptr<Image> image = create_image (size);
+       Cairo::RefPtr<Cairo::Surface> surface = create_surface (image);
+       Cairo::RefPtr<Cairo::Context> context = Cairo::Context::create (surface);
+
+       context->set_line_width (1);
+       context->scale (x_scale, y_scale);
+       layout->update_from_cairo_context (context);
+
+       if (first.effect() == dcp::SHADOW) {
                /* Drop-shadow effect */
-               set_source_rgba (context, subtitles.front().effect_colour(), fade_factor);
+               set_source_rgba (context, first.effect_colour(), fade_factor);
                context->move_to (x_offset + 4, y_offset + 4);
                layout->add_to_cairo_context (context);
                context->fill ();
        }
 
-       if (subtitles.front().effect() == dcp::BORDER) {
+       if (first.effect() == dcp::BORDER) {
                /* Border effect; stroke the subtitle with a large (arbitrarily chosen) line width */
-               set_source_rgba (context, subtitles.front().effect_colour(), fade_factor);
-               context->set_line_width (subtitles.front().outline_width * target.width / 2048.0);
+               set_source_rgba (context, first.effect_colour(), fade_factor);
+               context->set_line_width (first.outline_width * target.width / 2048.0);
                context->set_line_join (Cairo::LINE_JOIN_ROUND);
                context->move_to (x_offset, y_offset);
                layout->add_to_cairo_context (context);
@@ -295,52 +389,12 @@ render_line (list<StringText> subtitles, list<shared_ptr<Font> > fonts, dcp::Siz
        pango_cairo_show_layout (context->cobj(), layout->gobj());
 #endif
 
-       int layout_width;
-       int layout_height;
-       layout->get_pixel_size (layout_width, layout_height);
-       layout_width *= xscale;
-       layout_height *= yscale;
-
-       int x = 0;
-       switch (subtitles.front().h_align ()) {
-       case dcp::HALIGN_LEFT:
-               /* h_position is distance between left of frame and left of subtitle */
-               x = subtitles.front().h_position() * target.width;
-               break;
-       case dcp::HALIGN_CENTER:
-               /* h_position is distance between centre of frame and centre of subtitle */
-               x = (0.5 + subtitles.front().h_position()) * target.width - layout_width / 2;
-               break;
-       case dcp::HALIGN_RIGHT:
-               /* h_position is distance between right of frame and right of subtitle */
-               x = (1.0 - subtitles.front().h_position()) * target.width - layout_width;
-               break;
-       }
-
-       int y = 0;
-       switch (subtitles.front().v_align ()) {
-       case dcp::VALIGN_TOP:
-               /* SMPTE says that v_position is the distance between top
-                  of frame and top of subtitle, but this doesn't always seem to be
-                  the case in practice; Gunnar Ásgeirsson's Dolby server appears
-                  to put VALIGN_TOP subs with v_position as the distance between top
-                  of frame and bottom of subtitle.
-               */
-               y = subtitles.front().v_position() * target.height - layout_height;
-               break;
-       case dcp::VALIGN_CENTER:
-               /* v_position is distance between centre of frame and centre of subtitle */
-               y = (0.5 + subtitles.front().v_position()) * target.height - layout_height / 2;
-               break;
-       case dcp::VALIGN_BOTTOM:
-               /* v_position is distance between bottom of frame and bottom of subtitle */
-               y = (1.0 - subtitles.front().v_position()) * target.height - layout_height;
-               break;
-       }
-
-       return PositionImage (image, Position<int> (max (0, x), max (0, y)));
+       int const x = x_position (first, target.width, size.width);
+       int const y = y_position (first, target.height, size.height);
+       return PositionImage (image, Position<int>(max (0, x), max(0, y)));
 }
 
+
 /** @param time Time of the frame that these subtitles are going on.
  *  @param frame_rate DCP frame rate.
  */