[vlc-devel] [PATCHv2 06/13] input/subtitles: refactor subtitles_Detect
Thomas Guillem
thomas at gllm.fr
Tue May 3 19:02:33 CEST 2016
From: Benjamin Adolphi <b.adolphi at gmail.com>
This function fills now a list of input_item_slave. Subtitles are not sorted in
input.c.
Signed-off-by: Thomas Guillem <thomas at gllm.fr>
---
src/input/input.c | 65 ++++++++++++---
src/input/input_internal.h | 2 +-
src/input/subtitles.c | 192 ++++++++++++++++++---------------------------
3 files changed, 132 insertions(+), 127 deletions(-)
diff --git a/src/input/input.c b/src/input/input.c
index 5591a4e..9542985 100644
--- a/src/input/input.c
+++ b/src/input/input.c
@@ -923,6 +923,30 @@ static void StartTitle( input_thread_t * p_input )
p_input->p->b_fast_seek = var_GetBool( p_input, "input-fast-seek" );
}
+static int SlaveCompare(const void *a, const void *b)
+{
+ const input_item_slave *p_slave0 = *((const input_item_slave **) a);
+ const input_item_slave *p_slave1 = *((const input_item_slave **) b);
+
+ if( p_slave0 == NULL || p_slave1 == NULL )
+ {
+ /* Put NULL (or rejected) subs at the end */
+ return p_slave0 == NULL ? 1 : p_slave1 == NULL ? -1 : 0;
+ }
+
+ if( p_slave0->i_priority > p_slave1->i_priority )
+ return -1;
+
+ if( p_slave0->i_priority < p_slave1->i_priority )
+ return 1;
+
+#ifdef HAVE_STRCOLL
+ return strcoll( p_slave0->psz_uri, p_slave1->psz_uri );
+#else
+ return strcmp( p_slave0->psz_uri, p_slave1->psz_uri );
+#endif
+}
+
static void LoadSubtitles( input_thread_t *p_input )
{
/* Load subtitles */
@@ -959,23 +983,44 @@ static void LoadSubtitles( input_thread_t *p_input )
if( var_GetBool( p_input, "sub-autodetect-file" ) )
{
+ /* Add local subtitles */
+ input_item_slave **pp_slaves;
+ int i_slaves;
+ TAB_INIT( i_slaves, pp_slaves );
char *psz_autopath = var_GetNonEmptyString( p_input, "sub-autodetect-path" );
- char **ppsz_subs = subtitles_Detect( p_input, psz_autopath,
- p_input->p->p_item->psz_uri );
- free( psz_autopath );
- for( int i = 0; ppsz_subs && ppsz_subs[i]; i++ )
+ if( subtitles_Detect( p_input, psz_autopath, p_input->p->p_item->psz_uri,
+ &pp_slaves, &i_slaves ) == VLC_SUCCESS )
{
- if( !psz_subtitle || strcmp( psz_subtitle, ppsz_subs[i] ) )
+ /* check that we did not add the subtitle through sub-file */
+ for( int i = 0; i < i_slaves; i++ )
{
- i_flags |= SUB_CANFAIL;
- input_SubtitleFileAdd( p_input, ppsz_subs[i], i_flags, false );
- i_flags = SUB_NOFLAG;
+ input_item_slave *p_curr = pp_slaves[i];
+ if( p_curr != NULL && psz_subtitle != NULL
+ && !strcmp( psz_subtitle, p_curr->psz_uri ) )
+ {
+ /* reject current sub */
+ input_item_slave_Delete( p_curr );
+ pp_slaves[i] = NULL;
+ }
}
+ }
+ free( psz_autopath );
- free( ppsz_subs[i] );
+ qsort( pp_slaves, i_slaves, sizeof(input_item_slave*), SlaveCompare );
+
+ /* add all detected subtitles */
+ for( int i = 0; i < i_slaves && pp_slaves[i] != NULL; i++ )
+ {
+ const char *psz_uri = pp_slaves[i]->psz_uri;
+ i_flags |= SUB_CANFAIL;
+ input_SubtitleAdd( p_input, psz_uri, i_flags );
+ i_flags = SUB_NOFLAG;
+
+ input_item_slave_Delete( pp_slaves[i] );
}
- free( ppsz_subs );
+
+ TAB_CLEAN( i_slaves, pp_slaves );
}
free( psz_subtitle );
diff --git a/src/input/input_internal.h b/src/input/input_internal.h
index b68bec7..e15814b 100644
--- a/src/input/input_internal.h
+++ b/src/input/input_internal.h
@@ -248,7 +248,7 @@ void input_ControlVarTitle( input_thread_t *, int i_title );
void input_ConfigVarInit ( input_thread_t * );
/* Subtitles */
-char **subtitles_Detect( input_thread_t *, char* path, const char *fname );
+int subtitles_Detect( input_thread_t *, char *, const char *, input_item_slave ***, int * );
int subtitles_Filter( const char *);
/* input.c */
diff --git a/src/input/subtitles.c b/src/input/subtitles.c
index f1d5cdb..0803f5b 100644
--- a/src/input/subtitles.c
+++ b/src/input/subtitles.c
@@ -42,23 +42,9 @@
#include "input_internal.h"
/**
- * We are not going to autodetect more subtitle files than this.
- */
-#define MAX_SUBTITLE_FILES 128
-
-/**
* The possible extensions for subtitle files we support
*/
-static const char sub_exts[][6] = {
- "idx", "sub", "srt",
- "ssa", "ass", "smi",
- "utf", "utf8", "utf-8",
- "rt", "aqt", "txt",
- "usf", "jss", "cdg",
- "psb", "mpsub","mpl2",
- "pjs", "dks", "stl",
- "vtt", "sbv", ""
-};
+static const char *const sub_exts[] = { SLAVE_SPU_EXTENSIONS, "" };
static void strcpy_trim( char *d, const char *s )
{
@@ -130,39 +116,6 @@ static int whiteonly( const char *s )
return 1;
}
-enum
-{
- SUB_PRIORITY_NONE = 0,
- SUB_PRIORITY_MATCH_NONE = 1,
- SUB_PRIORITY_MATCH_RIGHT = 2,
- SUB_PRIORITY_MATCH_LEFT = 3,
- SUB_PRIORITY_MATCH_ALL = 4,
-};
-typedef struct
-{
- int priority;
- char *psz_fname;
- char *psz_ext;
-} vlc_subfn_t;
-
-static int compare_sub_priority( const void *a, const void *b )
-{
- const vlc_subfn_t *p0 = a;
- const vlc_subfn_t *p1 = b;
-
- if( p0->priority > p1->priority )
- return -1;
-
- if( p0->priority < p1->priority )
- return 1;
-
-#ifdef HAVE_STRCOLL
- return strcoll( p0->psz_fname, p1->psz_fname);
-#else
- return strcmp( p0->psz_fname, p1->psz_fname);
-#endif
-}
-
/*
* Check if a file ends with a subtitle extension
*/
@@ -226,7 +179,6 @@ static char **paths_to_list( const char *psz_dir, char *psz_path )
return subdirs;
}
-
/**
* Detect subtitle files.
*
@@ -238,37 +190,36 @@ static char **paths_to_list( const char *psz_dir, char *psz_path )
* \ingroup Demux
* \param p_this the calling \ref input_thread_t
* \param psz_path a list of subdirectories (separated by a ',') to look in.
- * \param psz_name the complete filename to base the search on.
- * \return a NULL terminated array of filenames with detected possible subtitles.
- * The array contains max MAX_SUBTITLE_FILES items and you need to free it after use.
+ * \param psz_name_org the complete filename to base the search on.
+ * \param pp_slaves an initialized input item slave list to append detected subtitles to
+ * \param p_slaves pointer to the size of the slave list
+ * \return VLC_SUCCESS if ok
*/
-char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
- const char *psz_name_org )
+int subtitles_Detect( input_thread_t *p_this, char *psz_path, const char *psz_name_org,
+ input_item_slave ***ppp_slaves, int *p_slaves )
{
int i_fuzzy = var_GetInteger( p_this, "sub-autodetect-fuzzy" );
if ( i_fuzzy == 0 )
- return NULL;
- int j, i_result2, i_sub_count, i_fname_len;
+ return VLC_EGENERIC;
+ int j, i_fname_len;
+ input_item_slave **pp_slaves = *ppp_slaves;
+ int i_slaves = *p_slaves;
char *f_fname_noext = NULL, *f_fname_trim = NULL;
-
char **subdirs; /* list of subdirectories to look in */
- vlc_subfn_t *result = NULL; /* unsorted results */
- char **result2; /* sorted results */
-
if( !psz_name_org )
- return NULL;
+ return VLC_EGENERIC;
char *psz_fname = vlc_uri2path( psz_name_org );
if( !psz_fname )
- return NULL;
+ return VLC_EGENERIC;
/* extract filename & dirname from psz_fname */
char *f_dir = strdup( psz_fname );
- if( f_dir == NULL )
+ if( f_dir == 0 )
{
free( psz_fname );
- return NULL;
+ return VLC_ENOMEM;
}
const char *f_fname = strrchr( psz_fname, DIR_SEP_CHAR );
@@ -276,7 +227,7 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
{
free( f_dir );
free( psz_fname );
- return NULL;
+ return VLC_EGENERIC;
}
f_fname++; /* Skip the '/' */
f_dir[f_fname - psz_fname] = 0; /* keep dir separator in f_dir */
@@ -291,15 +242,14 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
free( f_fname_noext );
free( f_fname_trim );
free( psz_fname );
- return NULL;
+ return VLC_ENOMEM;
}
strcpy_strip_ext( f_fname_noext, f_fname );
strcpy_trim( f_fname_trim, f_fname_noext );
- result = calloc( MAX_SUBTITLE_FILES+1, sizeof(vlc_subfn_t) ); /* We check it later (simplify code) */
subdirs = paths_to_list( f_dir, psz_path );
- for( j = -1, i_sub_count = 0; (j == -1) || ( j >= 0 && subdirs != NULL && subdirs[j] != NULL ); j++ )
+ for( j = -1; (j == -1) || ( j >= 0 && subdirs != NULL && subdirs[j] != NULL ); j++ )
{
const char *psz_dir = (j < 0) ? f_dir : subdirs[j];
if( psz_dir == NULL || ( j >= 0 && !strcmp( psz_dir, f_dir ) ) )
@@ -313,7 +263,7 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
msg_Dbg( p_this, "looking for a subtitle file in %s", psz_dir );
const char *psz_name;
- while( (psz_name = vlc_readdir( dir )) && i_sub_count < MAX_SUBTITLE_FILES )
+ while( (psz_name = vlc_readdir( dir )) )
{
if( psz_name[0] == '.' || !subtitles_Filter( psz_name ) )
continue;
@@ -322,7 +272,7 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
char tmp_fname_trim[strlen( psz_name ) + 1];
char tmp_fname_ext[strlen( psz_name ) + 1];
const char *tmp;
- int i_prio = SUB_PRIORITY_NONE;
+ int i_prio = SLAVE_PRIORITY_NONE;
/* retrieve various parts of the filename */
strcpy_strip_ext( tmp_fname_noext, psz_name );
@@ -332,7 +282,7 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
if( !strcmp( tmp_fname_trim, f_fname_trim ) )
{
/* matches the movie name exactly */
- i_prio = SUB_PRIORITY_MATCH_ALL;
+ i_prio = SLAVE_PRIORITY_MATCH_ALL;
}
else if( (tmp = strstr( tmp_fname_trim, f_fname_trim )) )
{
@@ -341,40 +291,49 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
if( whiteonly( tmp ) )
{
/* chars in front of the movie name */
- i_prio = SUB_PRIORITY_MATCH_RIGHT;
+ i_prio = SLAVE_PRIORITY_MATCH_RIGHT;
}
else
{
/* chars after (and possibly in front of)
* the movie name */
- i_prio = SUB_PRIORITY_MATCH_LEFT;
+ i_prio = SLAVE_PRIORITY_MATCH_LEFT;
}
}
else if( j == -1 )
{
/* doesn't contain the movie name, prefer files in f_dir over subdirs */
- i_prio = SUB_PRIORITY_MATCH_NONE;
+ i_prio = SLAVE_PRIORITY_MATCH_NONE;
}
if( i_prio >= i_fuzzy )
{
struct stat st;
char *path;
- if( asprintf( &path, "%s"DIR_SEP"%s", psz_dir, psz_name ) < 0 )
+ size_t i_len = strlen( psz_dir );
+ const char *psz_format;
+ if( psz_dir[i_len - 1] == DIR_SEP_CHAR )
+ psz_format = "%s%s";
+ else
+ psz_format = "%s"DIR_SEP"%s";
+
+ if( asprintf( &path, psz_format, psz_dir, psz_name ) < 0 )
continue;
if( strcmp( path, psz_fname )
&& vlc_stat( path, &st ) == 0
- && S_ISREG( st.st_mode ) && result )
+ && S_ISREG( st.st_mode ) )
{
msg_Dbg( p_this,
"autodetected subtitle: %s with priority %d",
path, i_prio );
- result[i_sub_count].priority = i_prio;
- result[i_sub_count].psz_fname = path;
- path = NULL;
- result[i_sub_count].psz_ext = strdup(tmp_fname_ext);
- i_sub_count++;
+ char *psz_uri = vlc_path2uri( path, NULL );
+ input_item_slave *p_sub = psz_uri != NULL ?
+ input_item_slave_New( psz_uri, SLAVE_TYPE_SPU, i_prio )
+ : NULL;
+ if( p_sub )
+ INSERT_ELEM( pp_slaves, i_slaves, i_slaves, p_sub );
+ free( psz_uri );
}
free( path );
}
@@ -392,52 +351,53 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
free( f_fname_noext );
free( psz_fname );
- if( !result )
- return NULL;
-
- qsort( result, i_sub_count, sizeof(vlc_subfn_t), compare_sub_priority );
-
- result2 = calloc( i_sub_count + 1, sizeof(char*) );
-
- for( j = 0, i_result2 = 0; j < i_sub_count && result2 != NULL; j++ )
+ for( int i = 0; i < i_slaves; i++ )
{
- bool b_reject = false;
+ input_item_slave *p_sub = pp_slaves[i];
- if( !result[j].psz_fname || !result[j].psz_ext ) /* memory out */
- break;
+ bool b_reject = false;
+ char *psz_ext = strrchr( p_sub->psz_uri, '.' );
+ if( !psz_ext )
+ continue;
+ psz_ext++;
- if( !strcasecmp( result[j].psz_ext, "sub" ) )
+ if( !strcasecmp( psz_ext, "sub" ) )
{
- int i;
- for( i = 0; i < i_sub_count; i++ )
+ for( int j = 0; j < i_slaves; j++ )
{
- if( result[i].psz_fname && result[i].psz_ext &&
- !strncasecmp( result[j].psz_fname, result[i].psz_fname,
- strlen( result[j].psz_fname) - 3 ) &&
- !strcasecmp( result[i].psz_ext, "idx" ) )
+ input_item_slave *p_sub_inner = pp_slaves[j];
+
+ /* check that the filenames without extension match */
+ if( strncasecmp( p_sub->psz_uri, p_sub_inner->psz_uri,
+ strlen( p_sub->psz_uri ) - 3 ) )
+ continue;
+
+ char *psz_ext_inner = strrchr( p_sub_inner->psz_uri, '.' );
+ if( !psz_ext_inner )
+ continue;
+ psz_ext_inner++;
+
+ /* check that we have an idx file */
+ if( !strcasecmp( psz_ext_inner, "idx" ) )
+ {
+ b_reject = true;
break;
+ }
}
- if( i < i_sub_count )
- b_reject = true;
}
- else if( !strcasecmp( result[j].psz_ext, "cdg" ) )
+ else if( !strcasecmp( psz_ext, "cdg" ) )
{
- if( result[j].priority < SUB_PRIORITY_MATCH_ALL )
+ if( p_sub->i_priority < SLAVE_PRIORITY_MATCH_ALL )
b_reject = true;
}
-
- /* */
- if( !b_reject )
- result2[i_result2++] = strdup( result[j].psz_fname );
- }
-
- for( j = 0; j < i_sub_count; j++ )
- {
- free( result[j].psz_fname );
- free( result[j].psz_ext );
+ if( b_reject )
+ {
+ pp_slaves[i] = NULL;
+ input_item_slave_Delete( p_sub );
+ }
}
- free( result );
- return result2;
+ *ppp_slaves = pp_slaves; /* in case of realloc */
+ *p_slaves = i_slaves;
+ return VLC_SUCCESS;
}
-
--
2.8.0.rc3
More information about the vlc-devel
mailing list