X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=src%2Flib%2Frender_subtitles.cc;h=94a6b837b2f5981e583b0a93bc4d53f9446e7491;hp=5364b8dfe3eeb9a3e0f0c1655c2b1ff4a425ca3a;hb=5a5324ed3a381a86dfe0a6e3932c1d58fdcd596f;hpb=05654d0e1799746a9df3ccab040c92e0ed825cac diff --git a/src/lib/render_subtitles.cc b/src/lib/render_subtitles.cc index 5364b8dfe..94a6b837b 100644 --- a/src/lib/render_subtitles.cc +++ b/src/lib/render_subtitles.cc @@ -1,27 +1,35 @@ /* - Copyright (C) 2014 Carl Hetherington + Copyright (C) 2014-2016 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic 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, + DCP-o-matic 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. + along with DCP-o-matic. If not, see . */ -#include -#include #include "render_subtitles.h" #include "types.h" #include "image.h" +#include "cross.h" +#include "font.h" +#include "dcpomatic_assert.h" +#include +#include +#include +#include +#include +#include using std::list; using std::cout; @@ -29,48 +37,55 @@ using std::string; using std::min; using std::max; using std::pair; +using std::cerr; +using std::make_pair; using boost::shared_ptr; +using boost::make_shared; using boost::optional; -static int -calculate_position (dcp::VAlign v_align, double v_position, int target_height, int offset) -{ - switch (v_align) { - case dcp::TOP: - return (v_position / 100) * target_height - offset; - case dcp::CENTER: - return (0.5 + v_position / 100) * target_height - offset; - case dcp::BOTTOM: - return (1.0 - v_position / 100) * target_height - offset; - } +static FcConfig* fc_config = 0; +static list > fc_config_fonts; - return 0; -} - -PositionImage -render_subtitles (list subtitles, dcp::Size target) +/** @param subtitles A list of subtitles that are all on the same line */ +static PositionImage +render_line (list subtitles, list > fonts, dcp::Size target) { - if (subtitles.empty ()) { - return PositionImage (); - } + /* XXX: this method can only handle italic / bold changes mid-line, + nothing else yet. + */ - /* Estimate height that the subtitle image needs to be */ - optional top; - optional bottom; - for (list::const_iterator i = subtitles.begin(); i != subtitles.end(); ++i) { - int const b = calculate_position (i->v_align(), i->v_position(), target.height, 0); - int const t = b - i->size() * target.height / (11 * 72); + DCPOMATIC_ASSERT (!subtitles.empty ()); - top = min (top.get_value_or (t), t); - bottom = max (bottom.get_value_or (b), b); + /* 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 ()); + } } - top = top.get() - 32; - bottom = bottom.get() + 32; + /* Make an empty bitmap as wide as target and at + least tall enough for this subtitle. + */ + + /* Basic guess on height... */ + int height = subtitles.front().size() * target.height / (11 * 72); + /* ...scaled... */ + height *= yscale; + /* ...and add a bit more for luck */ + height += target.height / 11; - shared_ptr image (new Image (PIX_FMT_RGBA, dcp::Size (target.width, bottom.get() - top.get ()), false)); + shared_ptr image = make_shared (AV_PIX_FMT_RGBA, dcp::Size (target.width, height), false); image->make_black (); +#ifdef DCPOMATIC_HAVE_FORMAT_STRIDE_FOR_WIDTH Cairo::RefPtr surface = Cairo::ImageSurface::create ( image->data()[0], Cairo::FORMAT_ARGB32, @@ -78,75 +93,244 @@ render_subtitles (list subtitles, dcp::Size target) image->size().height, Cairo::ImageSurface::format_stride_for_width (Cairo::FORMAT_ARGB32, image->size().width) ); - +#else + /* Centos 5 does not have Cairo::ImageSurface::format_stride_for_width, so just use width * 4 + which I hope is safe (if slow) + */ + Cairo::RefPtr surface = Cairo::ImageSurface::create ( + image->data()[0], + Cairo::FORMAT_ARGB32, + image->size().width, + image->size().height, + image->size().width * 4 + ); +#endif + Cairo::RefPtr context = Cairo::Context::create (surface); - Glib::RefPtr layout = Pango::Layout::create (context); - layout->set_width (image->size().width * PANGO_SCALE); - layout->set_alignment (Pango::ALIGN_CENTER); + if (!fc_config) { + fc_config = FcConfigCreate (); + } - context->set_line_width (1); + FontFiles font_files; + + try { + font_files.set (FontFiles::NORMAL, shared_path () / "LiberationSans-Regular.ttf"); + font_files.set (FontFiles::ITALIC, shared_path () / "LiberationSans-Italic.ttf"); + font_files.set (FontFiles::BOLD, shared_path () / "LiberationSans-Bold.ttf"); + } catch (boost::filesystem::filesystem_error& e) { + + } + + /* Hack: try the debian/ubuntu locations if getting the shared path failed */ + + if (!font_files.get(FontFiles::NORMAL) || !boost::filesystem::exists(font_files.get(FontFiles::NORMAL).get())) { + font_files.set (FontFiles::NORMAL, "/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf"); + } + if (!font_files.get(FontFiles::ITALIC) || !boost::filesystem::exists(font_files.get(FontFiles::ITALIC).get())) { + font_files.set (FontFiles::ITALIC, "/usr/share/fonts/truetype/liberation/LiberationSans-Italic.ttf"); + } + if (!font_files.get(FontFiles::BOLD) || !boost::filesystem::exists(font_files.get(FontFiles::BOLD).get())) { + font_files.set (FontFiles::BOLD, "/usr/share/fonts/truetype/liberation/LiberationSans-Bold.ttf"); + } - for (list::const_iterator i = subtitles.begin(); i != subtitles.end(); ++i) { - string f = i->font (); - if (f.empty ()) { - f = "Arial"; + BOOST_FOREACH (shared_ptr i, fonts) { + if (i->id() == subtitles.front().font() && i->file(FontFiles::NORMAL)) { + font_files = i->files (); } - Pango::FontDescription font (f); - font.set_absolute_size (i->size_in_pixels (target.height) * PANGO_SCALE); - if (i->italic ()) { - font.set_style (Pango::STYLE_ITALIC); + } + + list >::const_iterator existing = fc_config_fonts.begin (); + while (existing != fc_config_fonts.end() && existing->first != font_files) { + ++existing; + } + + string font_name; + if (existing != fc_config_fonts.end ()) { + font_name = existing->second; + } else { + /* Make this font available to DCP-o-matic */ + for (int i = 0; i < FontFiles::VARIANTS; ++i) { + if (font_files.get(static_cast(i))) { + FcConfigAppFontAddFile ( + fc_config, + reinterpret_cast (font_files.get(static_cast(i)).get().string().c_str()) + ); + } } - layout->set_font_description (font); - layout->set_text (i->text ()); - - /* Compute fade factor */ - /* XXX */ - float fade_factor = 1; -#if 0 - dcp::Time now (time * 1000 / (4 * TIME_HZ)); - dcp::Time end_fade_up = i->in() + i->fade_up_time (); - dcp::Time start_fade_down = i->out() - i->fade_down_time (); - if (now < end_fade_up) { - fade_factor = (now - i->in()) / i->fade_up_time(); - } else if (now > start_fade_down) { - fade_factor = 1.0 - ((now - start_fade_down) / i->fade_down_time ()); + + FcPattern* pattern = FcPatternBuild (0, FC_FILE, FcTypeString, font_files.get(FontFiles::NORMAL).get().string().c_str(), static_cast (0)); + FcObjectSet* object_set = FcObjectSetBuild (FC_FAMILY, FC_STYLE, FC_LANG, FC_FILE, static_cast (0)); + FcFontSet* font_set = FcFontList (fc_config, pattern, object_set); + if (font_set) { + for (int i = 0; i < font_set->nfont; ++i) { + FcPattern* font = font_set->fonts[i]; + FcChar8* file; + FcChar8* family; + FcChar8* style; + if ( + FcPatternGetString (font, FC_FILE, 0, &file) == FcResultMatch && + FcPatternGetString (font, FC_FAMILY, 0, &family) == FcResultMatch && + FcPatternGetString (font, FC_STYLE, 0, &style) == FcResultMatch + ) { + font_name = reinterpret_cast (family); + } + } + + FcFontSetDestroy (font_set); } -#endif - - layout->update_from_cairo_context (context); - - /* Work out position */ - - int const x = 0; - int const y = calculate_position (i->v_align (), i->v_position (), target.height, (layout->get_baseline() / PANGO_SCALE) + top.get ()); - - if (i->effect() == dcp::SHADOW) { - /* Drop-shadow effect */ - dcp::Color const ec = i->effect_color (); - context->set_source_rgba (float(ec.r) / 255, float(ec.g) / 255, float(ec.b) / 255, fade_factor); - context->move_to (x + 4, y + 4); - layout->add_to_cairo_context (context); - context->fill (); + + FcObjectSetDestroy (object_set); + FcPatternDestroy (pattern); + + fc_config_fonts.push_back (make_pair (font_files, font_name)); + } + + FcConfigSetCurrent (fc_config); + + Glib::RefPtr layout = Pango::Layout::create (context); + + layout->set_alignment (Pango::ALIGN_LEFT); + + context->set_line_width (1); + + /* Render the subtitle at the top left-hand corner of image */ + + Pango::FontDescription font (font_name); + font.set_absolute_size (subtitles.front().size_in_pixels (target.height) * PANGO_SCALE); + layout->set_font_description (font); + + string marked_up; + bool italic = false; + bool bold = false; + BOOST_FOREACH (dcp::SubtitleString const & i, subtitles) { + if (i.italic() != italic) { + if (i.italic()) { + marked_up += ""; + } else { + marked_up += ""; + } + italic = i.italic (); } - /* The actual subtitle */ - context->move_to (x, y); - dcp::Color const c = i->color (); - context->set_source_rgba (float(c.r) / 255, float(c.g) / 255, float(c.b) / 255, fade_factor); + if (i.bold() != bold) { + if (i.bold()) { + marked_up += ""; + } else { + marked_up += ""; + } + bold = i.bold (); + } + + marked_up += i.text (); + } + + if (italic) { + marked_up += ""; + } + + if (bold) { + marked_up += ""; + } + + layout->set_markup (marked_up); + + /* Compute fade factor */ + /* XXX */ + float fade_factor = 1; + + context->scale (xscale, yscale); + layout->update_from_cairo_context (context); + + if (subtitles.front().effect() == dcp::SHADOW) { + /* Drop-shadow effect */ + dcp::Colour const ec = subtitles.front().effect_colour (); + context->set_source_rgba (float(ec.r) / 255, float(ec.g) / 255, float(ec.b) / 255, fade_factor); + context->move_to (4, 4); layout->add_to_cairo_context (context); context->fill (); + } - if (i->effect() == dcp::BORDER) { - /* Border effect */ - context->move_to (x, y); - dcp::Color ec = i->effect_color (); - context->set_source_rgba (float(ec.r) / 255, float(ec.g) / 255, float(ec.b) / 255, fade_factor); - layout->add_to_cairo_context (context); - context->stroke (); - } + /* The actual subtitle */ + + dcp::Colour const c = subtitles.front().colour (); + context->set_source_rgba (float(c.r) / 255, float(c.g) / 255, float(c.b) / 255, fade_factor); + context->move_to (0, 0); + layout->add_to_cairo_context (context); + context->fill (); + + if (subtitles.front().effect() == dcp::BORDER) { + /* Border effect */ + dcp::Colour ec = subtitles.front().effect_colour (); + context->set_source_rgba (float(ec.r) / 255, float(ec.g) / 255, float(ec.b) / 255, fade_factor); + context->move_to (0, 0); + layout->add_to_cairo_context (context); + context->stroke (); } - return PositionImage (image, Position (0, top.get ())); + 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 (max (0, x), max (0, y))); } +list +render_subtitles (list subtitles, list > fonts, dcp::Size target) +{ + list pending; + list images; + + BOOST_FOREACH (dcp::SubtitleString const & i, subtitles) { + if (!pending.empty() && fabs (i.v_position() - pending.back().v_position()) > 1e-4) { + images.push_back (render_line (pending, fonts, target)); + pending.clear (); + } + pending.push_back (i); + } + + if (!pending.empty ()) { + images.push_back (render_line (pending, fonts, target)); + } + + return images; +}