Add support for reading <font> tags in SubRip.

5 files changed, 46 insertions, 16 deletions

diff --git a/src/colour.cc b/src/colour.cc
index f163424..a47e633 100644
--- a/src/colour.cc
+++ b/src/colour.cc
@@ -25,16 +25,26 @@
 using std::string;
 using namespace sub;
 
-Colour::Colour (string argb_hex)
+Colour
+Colour::from_argb_hex (string argb_hex)
 {
 	int alpha, ir, ig, ib;
 	if (sscanf (argb_hex.c_str(), "%2x%2x%2x%2x", &alpha, &ir, &ig, &ib) < 4) {
 		throw XMLError ("could not parse colour string");
 	}
 
-	r = float (ir) / 255;
-	g = float (ig) / 255;
-	b = float (ib) / 255;
+	return Colour (float (ir) / 255, float (ig) / 255, float (ib) / 255);
+}
+
+Colour
+Colour::from_rgb_hex (string rgb_hex)
+{
+	int ir, ig, ib;
+	if (sscanf (rgb_hex.c_str(), "%2x%2x%2x", &ir, &ig, &ib) < 3) {
+		throw XMLError ("could not parse colour string");
+	}
+
+	return Colour (float (ir) / 255, float (ig) / 255, float (ib) / 255);
 }
 
 bool
diff --git a/src/colour.h b/src/colour.h
index 673bb9f..55d2239 100644
--- a/src/colour.h
+++ b/src/colour.h
@@ -35,14 +35,15 @@ public:
 		, g (0)
 		, b (0)
 	{}
-	
+
 	Colour (float r, float g, float b)
 		: r (r)
 		, g (g)
 		, b (b)
 	{}
 
-	Colour (std::string);
+	static Colour from_argb_hex (std::string);
+	static Colour from_rgb_hex (std::string);
 
 	/** red component (from 0 to 1) */
 	float r;
@@ -54,7 +55,7 @@ public:
 
 bool
 operator== (Colour const & a, Colour const & b);
-	
+
 }
 
 #endif
diff --git a/src/dcp/font.cc b/src/dcp/font.cc
index b2fd128..64787c7 100644
--- a/src/dcp/font.cc
+++ b/src/dcp/font.cc
@@ -35,7 +35,7 @@ dcp::Font::Font (cxml::ConstNodePtr node)
 	italic = node->optional_bool_attribute ("Italic");
 	optional<string> c = node->optional_string_attribute ("Color");
 	if (c) {
-		colour = Colour (c.get ());
+		colour = Colour::from_argb_hex (c.get ());
 	}
 	optional<string> const e = node->optional_string_attribute ("Effect");
 	if (e) {
@@ -43,15 +43,15 @@ dcp::Font::Font (cxml::ConstNodePtr node)
 	}
 	c = node->optional_string_attribute ("EffectColor");
 	if (c) {
-		effect_colour = Colour (c.get ());
+		effect_colour = Colour::from_argb_hex (c.get ());
 	}
 }
 
 dcp::Font::Font (std::list<boost::shared_ptr<Font> > const & font_nodes)
 	: size (0)
 	, italic (false)
-	, colour ("FFFFFFFF")
-	, effect_colour ("FFFFFFFF")
+	, colour (Colour::from_argb_hex ("FFFFFFFF"))
+	, effect_colour (Colour::from_argb_hex ("FFFFFFFF"))
 {
 	for (list<shared_ptr<Font> >::const_iterator i = font_nodes.begin(); i != font_nodes.end(); ++i) {
 		if ((*i)->id) {
diff --git a/src/subrip_reader.cc b/src/subrip_reader.cc
index 3f1c23d..134ca3e 100644
--- a/src/subrip_reader.cc
+++ b/src/subrip_reader.cc
@@ -21,12 +21,17 @@
 #include "exceptions.h"
 #include <boost/algorithm/string.hpp>
 #include <boost/lexical_cast.hpp>
+#include <boost/regex.hpp>
 #include <cstdio>
 #include <vector>
 
 using std::string;
 using std::vector;
+using std::list;
+using std::cout;
+using std::hex;
 using boost::lexical_cast;
+using boost::to_upper;
 using namespace sub;
 
 SubripReader::SubripReader (FILE* f)
@@ -60,7 +65,7 @@ SubripReader::SubripReader (FILE* f)
 			static_cast<unsigned char> (line[1]) == 0xbb &&
 			static_cast<unsigned char> (line[2]) == 0xbf
 			) {
-			
+
 			/* Skip Unicode byte order mark */
 			line = line.substr (3);
 		}
@@ -88,7 +93,7 @@ SubripReader::SubripReader (FILE* f)
 			to = convert_time (p[2]);
 
 			/* XXX: should not ignore coordinate specifications */
-			
+
 			state = CONTENT;
 			break;
 		}
@@ -132,7 +137,7 @@ SubripReader::convert_line (string t, int line_number, Time from, Time to)
 		TEXT,
 		TAG
 	} state = TEXT;
-	
+
 	string tag;
 
 	RawSubtitle p;
@@ -144,7 +149,10 @@ SubripReader::convert_line (string t, int line_number, Time from, Time to)
 	/* XXX: arbitrary */
 	p.vertical_position.lines = 32;
 	p.vertical_position.reference = TOP_OF_SUBTITLE;
-	
+
+	list<Colour> colours;
+	colours.push_back (Colour (1, 1, 1));
+
 	/* XXX: missing <font> support */
 	/* XXX: nesting of tags e.g. <b>foo<i>bar<b>baz</b>fred</i>jim</b> might
 	   not work, I think.
@@ -179,6 +187,17 @@ SubripReader::convert_line (string t, int line_number, Time from, Time to)
 				} else if (tag == "/u") {
 					maybe_content (p);
 					p.underline = false;
+				} else if (boost::starts_with (tag, "font")) {
+					maybe_content (p);
+					boost::regex re (".*color=\"#([0123456789abcdef]+)\"");
+					boost::smatch match;
+					if (boost::regex_search (tag, match, re) && string (match[1]).size() == 6) {
+						p.colour = Colour::from_rgb_hex (match[1]);
+						colours.push_back (p.colour);
+					}
+				} else if (tag == "/font") {
+					colours.pop_back ();
+					p.colour = colours.back ();
 				}
 				tag.clear ();
 				state = TEXT;
diff --git a/src/wscript b/src/wscript
index 89c6e89..6afe1d9 100644
--- a/src/wscript
+++ b/src/wscript
@@ -8,7 +8,7 @@ def build(bld):
 
     obj.name = 'libsub%s' % bld.env.API_VERSION
     obj.target = 'sub%s' % bld.env.API_VERSION
-    obj.uselib = 'CXML DCP BOOST_FILESYSTEM BOOST_LOCALE'
+    obj.uselib = 'CXML DCP BOOST_FILESYSTEM BOOST_LOCALE BOOST_REGEX'
     obj.use = 'libkumu-libsub%s libasdcp-libsub%s' % (bld.env.API_VERSION, bld.env.API_VERSION)
     obj.export_includes = ['.']
     obj.source = """