X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fsubrip.cc;h=8a7423698d850268c7b927970831ab0a63c3029a;hb=4dbc6ef917aeceb906b1ef1caf6911033e7e2c54;hp=11ad3302d3659abf107eed8b8fa69d33d8109c69;hpb=3e12c68dc0451e73b5bc1a84d1d70f4999f7b4b5;p=dcpomatic.git diff --git a/src/lib/subrip.cc b/src/lib/subrip.cc index 11ad3302d..8a7423698 100644 --- a/src/lib/subrip.cc +++ b/src/lib/subrip.cc @@ -17,23 +17,17 @@ */ -#include -#include #include "subrip.h" -#include "subrip_content.h" -#include "subrip_subtitle.h" #include "cross.h" #include "exceptions.h" +#include "subrip_content.h" +#include +#include #include "i18n.h" -using std::string; -using std::list; using std::vector; -using std::cout; using boost::shared_ptr; -using boost::lexical_cast; -using boost::algorithm::trim; SubRip::SubRip (shared_ptr content) { @@ -42,192 +36,8 @@ SubRip::SubRip (shared_ptr content) throw OpenFileError (content->path (0)); } - enum { - COUNTER, - METADATA, - CONTENT - } state = COUNTER; - - char buffer[256]; - int next_count = 1; - - boost::optional current; - list lines; - - while (!feof (f)) { - fgets (buffer, sizeof (buffer), f); - if (feof (f)) { - break; - } - - string line (buffer); - trim_right_if (line, boost::is_any_of ("\n\r")); - - switch (state) { - case COUNTER: - { - if (line.empty ()) { - /* a blank line at the start is ok */ - break; - } - - int x = 0; - try { - x = lexical_cast (line); - } catch (...) { - - } - - if (x == next_count) { - state = METADATA; - ++next_count; - current = SubRipSubtitle (); - } else { - throw SubRipError (line, _("a subtitle count"), content->path (0)); - } - } - break; - case METADATA: - { - vector p; - boost::algorithm::split (p, line, boost::algorithm::is_any_of (" ")); - if (p.size() != 3 && p.size() != 7) { - throw SubRipError (line, _("a time/position line"), content->path (0)); - } - - current->period = ContentTimePeriod (convert_time (p[0]), convert_time (p[2])); - - if (p.size() > 3) { - current->x1 = convert_coordinate (p[3]); - current->x2 = convert_coordinate (p[4]); - current->y1 = convert_coordinate (p[5]); - current->y2 = convert_coordinate (p[6]); - } - state = CONTENT; - break; - } - case CONTENT: - if (line.empty ()) { - state = COUNTER; - current->pieces = convert_content (lines); - _subtitles.push_back (current.get ()); - current.reset (); - lines.clear (); - } else { - lines.push_back (line); - } - break; - } - } - - if (state == CONTENT) { - current->pieces = convert_content (lines); - _subtitles.push_back (current.get ()); - } - - fclose (f); -} - -ContentTime -SubRip::convert_time (string t) -{ - ContentTime r; - - vector a; - boost::algorithm::split (a, t, boost::is_any_of (":")); - assert (a.size() == 3); - r += ContentTime::from_seconds (lexical_cast (a[0]) * 60 * 60); - r += ContentTime::from_seconds (lexical_cast (a[1]) * 60); - - vector b; - boost::algorithm::split (b, a[2], boost::is_any_of (",")); - r += ContentTime::from_seconds (lexical_cast (b[0])); - r += ContentTime::from_seconds (lexical_cast (b[1]) / 1000); - - return r; -} - -int -SubRip::convert_coordinate (string t) -{ - vector a; - boost::algorithm::split (a, t, boost::is_any_of (":")); - assert (a.size() == 2); - return lexical_cast (a[1]); -} - -void -SubRip::maybe_content (list& pieces, SubRipSubtitlePiece& p) -{ - if (!p.text.empty ()) { - pieces.push_back (p); - p.text.clear (); - } -} - -list -SubRip::convert_content (list t) -{ - list pieces; - - SubRipSubtitlePiece p; - - enum { - TEXT, - TAG - } state = TEXT; - - string tag; - - /* XXX: missing support */ - /* XXX: nesting of tags e.g. foobarbazfredjim might - not work, I think. - */ - - for (list::const_iterator i = t.begin(); i != t.end(); ++i) { - for (size_t j = 0; j < i->size(); ++j) { - switch (state) { - case TEXT: - if ((*i)[j] == '<' || (*i)[j] == '{') { - state = TAG; - } else { - p.text += (*i)[j]; - } - break; - case TAG: - if ((*i)[j] == '>' || (*i)[j] == '}') { - if (tag == "b") { - maybe_content (pieces, p); - p.bold = true; - } else if (tag == "/b") { - maybe_content (pieces, p); - p.bold = false; - } else if (tag == "i") { - maybe_content (pieces, p); - p.italic = true; - } else if (tag == "/i") { - maybe_content (pieces, p); - p.italic = false; - } else if (tag == "u") { - maybe_content (pieces, p); - p.underline = true; - } else if (tag == "/u") { - maybe_content (pieces, p); - p.underline = false; - } - tag.clear (); - state = TEXT; - } else { - tag += (*i)[j]; - } - break; - } - } - } - - maybe_content (pieces, p); - - return pieces; + sub::SubripReader reader (f); + _subtitles = sub::collect > (reader.subtitles ()); } ContentTime @@ -237,5 +47,5 @@ SubRip::length () const return ContentTime (); } - return _subtitles.back().period.to; + return ContentTime::from_seconds (_subtitles.back().to.metric().get().all_as_seconds ()); }