Basic SSA support (#128).

[dcpomatic.git] / src / lib / text_subtitle.cc

diff --git a/src/lib/text_subtitle.cc b/src/lib/text_subtitle.cc
index d55ff68d7e46f3bc1702389ac75b2ea72888382a..7faecb459b4f2b6f7f6f20a2007ea4d77da8fcd3 100644 (file)
--- a/src/lib/text_subtitle.cc
+++ b/src/lib/text_subtitle.cc
@@ -22,6 +22,7 @@
 #include "exceptions.h"
 #include "text_subtitle_content.h"
 #include <sub/subrip_reader.h>
+#include <sub/ssa_reader.h>
 #include <sub/collect.h>
 #include <unicode/ucsdet.h>
 #include <unicode/ucnv.h>
@@ -65,8 +66,17 @@ TextSubtitle::TextSubtitle (shared_ptr<const TextSubtitleContent> content)
        ucnv_close (to_utf16);
        ucnv_close (to_utf8);
 
-       sub::SubripReader reader (utf8.get());
-       _subtitles = sub::collect<vector<sub::Subtitle> > (reader.subtitles ());
+       sub::Reader* reader = 0;
+
+       if (content->path(0).extension() == ".srt" || content->path(0).extension() == ".SRT") {
+               reader = new sub::SubripReader (utf8.get());
+       } else if (content->path(0).extension() == ".ssa" || content->path(0).extension() == ".SSA") {
+               reader = new sub::SSAReader (utf8.get());
+       }
+
+       if (reader) {
+               _subtitles = sub::collect<vector<sub::Subtitle> > (reader->subtitles ());
+       }
 }
 
 ContentTime