Support binary STL subtitle files.

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

diff --git a/src/lib/string_text_file.cc b/src/lib/string_text_file.cc
index c7eadbbb1a9332ba1f7312335bd7ad8acd9e26fd..c5b297a8b8e73ab1089b9ce1bcda831f56edfc41 100644 (file)
--- a/src/lib/string_text_file.cc
+++ b/src/lib/string_text_file.cc
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2014-2018 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2014-2020 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
@@ -24,6 +24,7 @@
 #include "string_text_file_content.h"
 #include <sub/subrip_reader.h>
 #include <sub/ssa_reader.h>
+#include <sub/stl_binary_reader.h>
 #include <sub/collect.h>
 #include <unicode/ucsdet.h>
 #include <unicode/ucnv.h>
@@ -41,50 +42,67 @@ using dcp::Data;
 
 StringTextFile::StringTextFile (shared_ptr<const StringTextFileContent> content)
 {
-       Data in (content->path (0));
-
-       UErrorCode status = U_ZERO_ERROR;
-       UCharsetDetector* detector = ucsdet_open (&status);
-       ucsdet_setText (detector, reinterpret_cast<const char *> (in.data().get()), in.size(), &status);
-
-       UCharsetMatch const * match = ucsdet_detect (detector, &status);
-       char const * in_charset = ucsdet_getName (match, &status);
-
-       UConverter* to_utf16 = ucnv_open (in_charset, &status);
-       /* This is a guess; I think we should be able to encode any input in 4 times its input size */
-       scoped_array<uint16_t> utf16 (new uint16_t[in.size() * 2]);
-       int const utf16_len = ucnv_toUChars (
-               to_utf16, reinterpret_cast<UChar*>(utf16.get()), in.size() * 2,
-               reinterpret_cast<const char *> (in.data().get()), in.size(),
-               &status
-               );
-
-       UConverter* to_utf8 = ucnv_open ("UTF-8", &status);
-       /* Another guess */
-       scoped_array<char> utf8 (new char[utf16_len * 2]);
-       ucnv_fromUChars (to_utf8, utf8.get(), utf16_len * 2, reinterpret_cast<UChar*>(utf16.get()), utf16_len, &status);
-
-       /* Fix OS X line endings */
-       size_t utf8_len = strlen (utf8.get ());
-       for (size_t i = 0; i < utf8_len; ++i) {
-               if (utf8[i] == '\r' && ((i == utf8_len - 1) || utf8[i + 1] != '\n')) {
-                       utf8[i] = '\n';
-               }
-       }
-
-       ucsdet_close (detector);
-       ucnv_close (to_utf16);
-       ucnv_close (to_utf8);
+       string ext = content->path(0).extension().string();
+       transform (ext.begin(), ext.end(), ext.begin(), ::tolower);
 
        sub::Reader* reader = 0;
 
-       string ext = content->path(0).extension().string();
-       transform (ext.begin(), ext.end(), ext.begin(), ::tolower);
+       if (ext == ".stl") {
+               FILE* f = fopen_boost (content->path(0), "rb");
+               if (!f) {
+                       throw OpenFileError (content->path(0), errno, OpenFileError::READ);
+               }
+               try {
+                       reader = new sub::STLBinaryReader (f);
+               } catch (...) {
+                       fclose (f);
+                       throw;
+               }
+               fclose (f);
+
+       } else {
+               /* Text-based file; sort out its character encoding before we try to parse it */
+
+               Data in (content->path (0));
+
+               UErrorCode status = U_ZERO_ERROR;
+               UCharsetDetector* detector = ucsdet_open (&status);
+               ucsdet_setText (detector, reinterpret_cast<const char *> (in.data().get()), in.size(), &status);
+
+               UCharsetMatch const * match = ucsdet_detect (detector, &status);
+               char const * in_charset = ucsdet_getName (match, &status);
+
+               UConverter* to_utf16 = ucnv_open (in_charset, &status);
+               /* This is a guess; I think we should be able to encode any input in 4 times its input size */
+               scoped_array<uint16_t> utf16 (new uint16_t[in.size() * 2]);
+               int const utf16_len = ucnv_toUChars (
+                               to_utf16, reinterpret_cast<UChar*>(utf16.get()), in.size() * 2,
+                               reinterpret_cast<const char *> (in.data().get()), in.size(),
+                               &status
+                               );
+
+               UConverter* to_utf8 = ucnv_open ("UTF-8", &status);
+               /* Another guess */
+               scoped_array<char> utf8 (new char[utf16_len * 2]);
+               ucnv_fromUChars (to_utf8, utf8.get(), utf16_len * 2, reinterpret_cast<UChar*>(utf16.get()), utf16_len, &status);
+
+               /* Fix OS X line endings */
+               size_t utf8_len = strlen (utf8.get ());
+               for (size_t i = 0; i < utf8_len; ++i) {
+                       if (utf8[i] == '\r' && ((i == utf8_len - 1) || utf8[i + 1] != '\n')) {
+                               utf8[i] = '\n';
+                       }
+               }
+
+               ucsdet_close (detector);
+               ucnv_close (to_utf16);
+               ucnv_close (to_utf8);
 
-       if (ext == ".srt") {
-               reader = new sub::SubripReader (utf8.get());
-       } else if (ext == ".ssa" || ext == ".ass") {
-               reader = new sub::SSAReader (utf8.get());
+               if (ext == ".srt") {
+                       reader = new sub::SubripReader (utf8.get());
+               } else if (ext == ".ssa" || ext == ".ass") {
+                       reader = new sub::SSAReader (utf8.get());
+               }
        }
 
        if (reader) {