sub/find_subfiles.c

   1 #include <dirent.h>
   2 #include <string.h>
   3 #include <stdlib.h>
   4 #include <ctype.h>
   5
   6 #include "mp_msg.h"
   7 #include "options.h"
   8 #include "path.h"
   9 #include "mpcommon.h"
  10 #include "sub/find_subfiles.h"
  11 #include "sub/sub.h"
  12
  13 static void strcpy_trim(char *d, char *s)
  14 {
  15     // skip leading whitespace
  16     while (*s && isspace(*s)) {
  17         s++;
  18     }
  19     for (;;) {
  20         // copy word
  21         while (*s && !isspace(*s)) {
  22             *d = tolower(*s);
  23             s++; d++;
  24         }
  25         if (*s == 0)
  26             break;
  27         // trim excess whitespace
  28         while (*s && isspace(*s)) {
  29             s++;
  30         }
  31         if (*s == 0)
  32             break;
  33         *d++ = ' ';
  34     }
  35     *d = 0;
  36 }
  37
  38 static void strcpy_strip_ext(char *d, char *s)
  39 {
  40     char *tmp = strrchr(s, '.');
  41     if (!tmp) {
  42         strcpy(d, s);
  43         return;
  44     } else {
  45         strncpy(d, s, tmp-s);
  46         d[tmp-s] = 0;
  47     }
  48     while (*d) {
  49         *d = tolower(*d);
  50         d++;
  51     }
  52 }
  53
  54 static void strcpy_get_ext(char *d, char *s)
  55 {
  56     char *tmp = strrchr(s, '.');
  57     if (!tmp) {
  58         strcpy(d, "");
  59         return;
  60     } else {
  61         strcpy(d, tmp+1);
  62     }
  63 }
  64
  65 static int whiteonly(char *s)
  66 {
  67     while (*s) {
  68         if (!isspace(*s))
  69             return 0;
  70         s++;
  71     }
  72     return 1;
  73 }
  74
  75 typedef struct subfn {
  76     int priority;
  77     char *fname;
  78 } subfn;
  79
  80 static int compare_sub_priority(const void *a, const void *b)
  81 {
  82     if (((const subfn*)a)->priority > ((const subfn*)b)->priority) {
  83         return -1;
  84     } else if (((const subfn*)a)->priority < ((const subfn*)b)->priority) {
  85         return 1;
  86     } else {
  87         return strcoll(((const subfn*)a)->fname, ((const subfn*)b)->fname);
  88     }
  89 }
  90
  91 /**
  92  * @brief Append all the subtitles in the given path matching fname
  93  * @param slist pointer to the subtitles list tallocated
  94  * @param nsub pointer to the number of subtitles
  95  * @param path Look for subtitles in this directory
  96  * @param fname Subtitle filename (pattern)
  97  * @param limit_fuzziness Ignore flag when sub_fuziness == 2
  98  */
  99 static void append_dir_subtitles(struct subfn **slist, int *nsub,
 100                                  struct bstr path, const char *fname,
 101                                  int limit_fuzziness)
 102 {
 103     char *f_fname, *f_fname_noext, *f_fname_trim, *tmp, *tmp_sub_id;
 104     char *tmp_fname_noext, *tmp_fname_trim, *tmp_fname_ext, *tmpresult;
 105
 106     int len, found, i;
 107     char *sub_exts[] = {"utf", "utf8", "utf-8", "sub", "srt", "smi", "rt", "txt", "ssa", "aqt", "jss", "js", "ass", NULL};
 108     FILE *f;
 109
 110     DIR *d;
 111     struct dirent *de;
 112
 113     len =   (strlen(fname) > 256 ? strlen(fname) : 256)
 114           + (path.len      > 256 ? path.len      : 256) + 2;
 115
 116     f_fname       = mp_basename(fname);
 117     f_fname_noext = malloc(len);
 118     f_fname_trim  = malloc(len);
 119
 120     tmp_fname_noext = malloc(len);
 121     tmp_fname_trim  = malloc(len);
 122     tmp_fname_ext   = malloc(len);
 123
 124     tmpresult = malloc(len);
 125
 126     strcpy_strip_ext(f_fname_noext, f_fname);
 127     strcpy_trim(f_fname_trim, f_fname_noext);
 128
 129     /* The code using sub language here is broken - it assumes strict
 130      * "videoname languagename" syntax for the subtitle file, which is
 131      * very unlikely to match especially if language name uses "en,de"
 132      * syntax... */
 133     tmp_sub_id = NULL;
 134 #if 0
 135     if (dvdsub_lang && !whiteonly(dvdsub_lang)) {
 136         tmp_sub_id = malloc(strlen(dvdsub_lang) + 1);
 137         strcpy_trim(tmp_sub_id, dvdsub_lang);
 138     }
 139 #endif
 140
 141     // 0 = nothing
 142     // 1 = any subtitle file
 143     // 2 = any sub file containing movie name
 144     // 3 = sub file containing movie name and the lang extension
 145     char *path0 = bstrdup0(NULL, path);
 146     d = opendir(path0);
 147     talloc_free(path0);
 148     if (d) {
 149         mp_msg(MSGT_SUBREADER, MSGL_INFO, "Load subtitles in %.*s\n", BSTR_P(path));
 150         while ((de = readdir(d))) {
 151             // retrieve various parts of the filename
 152             strcpy_strip_ext(tmp_fname_noext, de->d_name);
 153             strcpy_get_ext(tmp_fname_ext, de->d_name);
 154             strcpy_trim(tmp_fname_trim, tmp_fname_noext);
 155
 156             // does it end with a subtitle extension?
 157             found = 0;
 158 #ifdef CONFIG_ICONV
 159 #ifdef CONFIG_ENCA
 160             for (i = ((sub_cp && strncasecmp(sub_cp, "enca", 4) != 0) ? 3 : 0); sub_exts[i]; i++) {
 161 #else
 162             for (i = (sub_cp ? 3 : 0); sub_exts[i]; i++) {
 163 #endif
 164 #else
 165             for (i = 0; sub_exts[i]; i++) {
 166 #endif
 167                 if (strcasecmp(sub_exts[i], tmp_fname_ext) == 0) {
 168                     found = 1;
 169                     break;
 170                 }
 171             }
 172
 173             // we have a (likely) subtitle file
 174             if (found) {
 175                 int prio = 0;
 176                 if (!prio && tmp_sub_id) {
 177                     sprintf(tmpresult, "%s %s", f_fname_trim, tmp_sub_id);
 178                     if (strcmp(tmp_fname_trim, tmpresult) == 0 && sub_match_fuzziness >= 1) {
 179                         // matches the movie name + lang extension
 180                         prio = 5;
 181                     }
 182                 }
 183                 if (!prio && strcmp(tmp_fname_trim, f_fname_trim) == 0) {
 184                     // matches the movie name
 185                     prio = 4;
 186                 }
 187                 if (!prio && (tmp = strstr(tmp_fname_trim, f_fname_trim)) && sub_match_fuzziness >= 1) {
 188                     // contains the movie name
 189                     tmp += strlen(f_fname_trim);
 190                     if (tmp_sub_id && strstr(tmp, tmp_sub_id)) {
 191                         // with sub_id specified prefer localized subtitles
 192                         prio = 3;
 193                     } else if ((tmp_sub_id == NULL) && whiteonly(tmp)) {
 194                         // without sub_id prefer "plain" name
 195                         prio = 3;
 196                     } else {
 197                         // with no localized subs found, try any else instead
 198                         prio = 2;
 199                     }
 200                 }
 201                 if (!prio) {
 202                     // doesn't contain the movie name
 203                     // don't try in the mplayer subtitle directory
 204                     if (!limit_fuzziness && sub_match_fuzziness >= 2) {
 205                         prio = 1;
 206                     }
 207                 }
 208
 209                 mp_msg(MSGT_SUBREADER, MSGL_DBG2, "Potential sub file: "
 210                        "\"%s\"  Priority: %d\n", de->d_name, prio);
 211                 if (prio) {
 212                     prio += prio;
 213 #ifdef CONFIG_ICONV
 214                     if (i < 3) // prefer UTF-8 coded
 215                         prio++;
 216 #endif
 217                     char *subpath = mp_path_join(*slist, path, BSTR(de->d_name));
 218                     if ((f = fopen(subpath, "rt"))) {
 219                         MP_GROW_ARRAY(*slist, *nsub);
 220                         struct subfn *sub = *slist + (*nsub)++;
 221
 222                         fclose(f);
 223                         sub->priority = prio;
 224                         sub->fname    = subpath;
 225                     } else
 226                         talloc_free(subpath);
 227                 }
 228             }
 229         }
 230         closedir(d);
 231     }
 232
 233     free(tmp_sub_id);
 234
 235     free(f_fname_noext);
 236     free(f_fname_trim);
 237
 238     free(tmp_fname_noext);
 239     free(tmp_fname_trim);
 240     free(tmp_fname_ext);
 241
 242     free(tmpresult);
 243 }
 244
 245 char **find_text_subtitles(struct MPOpts *opts, const char *fname)
 246 {
 247     char **subnames = NULL;
 248     struct subfn *slist = talloc_array_ptrtype(NULL, slist, 1);
 249     int n = 0;
 250
 251     // Load subtitles from current media directory
 252     append_dir_subtitles(&slist, &n, mp_dirname(fname), fname, 0);
 253
 254     // Load subtitles in dirs specified by sub-paths option
 255     if (opts->sub_paths) {
 256         for (int i = 0; opts->sub_paths[i]; i++) {
 257             char *path = mp_path_join(slist, mp_dirname(fname),
 258                                       BSTR(opts->sub_paths[i]));
 259             append_dir_subtitles(&slist, &n, BSTR(path), fname, 0);
 260         }
 261     }
 262
 263     // Load subtitles in ~/.mplayer/sub limiting sub fuzziness
 264     char *mp_subdir = get_path("sub/");
 265     if (mp_subdir)
 266         append_dir_subtitles(&slist, &n, BSTR(mp_subdir), fname, 1);
 267     free(mp_subdir);
 268
 269     // Sort subs by priority and append them
 270     qsort(slist, n, sizeof(*slist), compare_sub_priority);
 271
 272     subnames = talloc_array_ptrtype(NULL, subnames, n);
 273     for (int i = 0; i < n; i++)
 274         subnames[i] = talloc_strdup(subnames, slist[i].fname);
 275
 276     talloc_free(slist);
 277     return subnames;
 278 }
 279
 280 char **find_vob_subtitles(struct MPOpts *opts, const char *fname)
 281 {
 282     char **vobs = talloc_array_ptrtype(NULL, vobs, 1);
 283     int n = 0;
 284
 285     // Potential vobsub in the media directory
 286     struct bstr bname = BSTR(mp_basename(fname));
 287     int pdot = bstrrchr(bname, '.');
 288     if (pdot >= 0)
 289         bname.len = pdot;
 290     vobs[n++] = mp_path_join(vobs, mp_dirname(fname), bname);
 291
 292     // Potential vobsubs in directories specified by sub-paths option
 293     if (opts->sub_paths) {
 294         for (int i = 0; opts->sub_paths[i]; i++) {
 295             char *path = mp_path_join(NULL, mp_dirname(fname),
 296                                       BSTR(opts->sub_paths[i]));
 297             MP_GROW_ARRAY(vobs, n);
 298             vobs[n++] = mp_path_join(vobs, BSTR(path), bname);
 299             talloc_free(path);
 300         }
 301     }
 302
 303     // Potential vobsub in ~/.mplayer/sub
 304     char *mp_subdir = get_path("sub/");
 305     if (mp_subdir) {
 306         MP_GROW_ARRAY(vobs, n);
 307         vobs[n++] = mp_path_join(vobs, BSTR(mp_subdir), bname);
 308     }
 309
 310     free(mp_subdir);
 311     MP_RESIZE_ARRAY(NULL, vobs, n);
 312     return vobs;
 313 }