demux/demux_textsub: allow configuring probe size

Author: afishhh

demux/demux_textsub.c

@@ -15,16 +15,36 @@
  * License along with mpv.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+#include <limits.h>
 #include <math.h>
 
 #include <subrandr/subrandr.h>
 
 #include "common/common.h"
 #include "demux/packet.h"
 #include "misc/bstr.h"
+#include "options/m_config.h"
+#include "options/m_option.h"
 #include "stream/stream.h"
 #include "demux.h"
 
+#define OPT_BASE_STRUCT struct demux_textsub_opts
+struct demux_textsub_opts {
+    int probesize;
+};
+
+const struct m_sub_options demux_textsub_conf = {
+    .opts = (const m_option_t[]) {
+        {"demuxer-textsub-probesize", OPT_INT(probesize), M_RANGE(32, INT_MAX)},
+        {0}
+    },
+    .size = sizeof(struct demux_textsub_opts),
+    .defaults = &(const struct demux_textsub_opts){
+        .probesize = 128,
+    },
+    .change_flags = UPDATE_DEMUXER,
+};
+
 struct format_codec_info {
     const char *codec;
     const char *codec_desc;
@@ -48,8 +68,6 @@ static const struct textsub_ext TEXT_FORMAT_EXTS[] = {
     {NULL}
 };
 
-static const int SUBRANDR_PROBE_SIZE = 128;
-
 struct demux_textsub_priv {
     bstr content;
     bool exhausted;
@@ -60,13 +78,15 @@ static int demux_open_textsub(struct demuxer *demuxer, enum demux_check check)
     bstr filename = bstr0(demuxer->filename);
     struct format_codec_info codec_info = {NULL, NULL};
 
+    struct demux_textsub_opts *opts = mp_get_config_group(demuxer, demuxer->global, &demux_textsub_conf);
+
     for (const struct textsub_ext *ext = TEXT_FORMAT_EXTS; ext->ext; ++ext) {
         if (bstr_endswith0(filename, ext->ext))
             codec_info = ext->codec_info;
     }
 
     if (!codec_info.codec) {
-        int probe_size = stream_peek(demuxer->stream, SUBRANDR_PROBE_SIZE);
+        int probe_size = stream_peek(demuxer->stream, opts->probesize);
         uint8_t *probe_buffer = demuxer->stream->buffer;
 
         sbr_subtitle_format fmt = sbr_probe_text((const char *)probe_buffer, (size_t)probe_size);

options/options.c

@@ -70,6 +70,9 @@ extern const struct m_sub_options demux_rawvideo_conf;
 extern const struct m_sub_options demux_playlist_conf;
 extern const struct m_sub_options demux_lavf_conf;
 extern const struct m_sub_options demux_mkv_conf;
+#if HAVE_SUBRANDR
+extern const struct m_sub_options demux_textsub_conf;
+#endif
 extern const struct m_sub_options vd_lavc_conf;
 extern const struct m_sub_options ad_lavc_conf;
 extern const struct m_sub_options hwdec_conf;
@@ -693,6 +696,9 @@ static const m_option_t mp_opts[] = {
     {"demuxer-rawvideo", OPT_SUBSTRUCT(demux_rawvideo, demux_rawvideo_conf)},
     {"", OPT_SUBSTRUCT(demux_playlist, demux_playlist_conf)},
     {"demuxer-mkv", OPT_SUBSTRUCT(demux_mkv, demux_mkv_conf)},
+#if HAVE_SUBRANDR
+    {"", OPT_SUBSTRUCT(demux_textsub, demux_textsub_conf)},
+#endif
 
 // ------------------------- subtitles options --------------------
 

options/options.h

@@ -365,6 +365,9 @@ typedef struct MPOpts {
     struct demux_playlist_opts *demux_playlist;
     struct demux_lavf_opts *demux_lavf;
     struct demux_mkv_opts *demux_mkv;
+#if HAVE_SUBRANDR
+    struct demux_textsub_opts *demux_textsub;
+#endif
 
     struct demux_opts *demux_opts;
     struct demux_cache_opts *demux_cache_opts;