src/subtitles/GFN.cpp

   1 /*
   2  *      Copyright (C) 2003-2006 Gabest
   3  *      http://www.gabest.org
   4  *
   5  *  This Program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2, or (at your option)
   8  *  any later version.
   9  *
  10  *  This Program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with GNU Make; see the file COPYING.  If not, write to
  17  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  18  *  http://www.gnu.org/copyleft/gpl.html
  19  *
  20  */
  21
  22 #include "stdafx.h"
  23 #include <io.h>
  24 #include "TextFile.h"
  25 #include "GFN.h"
  26
  27 TCHAR* exttypestr[] =
  28 {
  29         _T("srt"), _T("sub"), _T("smi"), _T("psb"),
  30         _T("ssa"), _T("ass"), _T("idx"), _T("usf"),
  31         _T("xss"), _T("txt"), _T("ssf"), _T("rt"), _T("sup")
  32 };
  33
  34 static TCHAR* ext[2][countof(exttypestr)] =
  35 {
  36         {
  37                 _T(".srt"), _T(".sub"), _T(".smi"), _T(".psb"),
  38                 _T(".ssa"), _T(".ass"), _T(".idx"), _T(".usf"),
  39                 _T(".xss"), _T(".txt"), _T(".ssf"), _T(".rt"), _T(".sup")
  40         },
  41         {
  42                 _T(".*.srt"), _T(".*.sub"), _T(".*.smi"), _T(".*.psb"),
  43                 _T(".*.ssa"), _T(".*.ass"), _T(".*.dummyidx"), _T(".*.usf"),
  44                 _T(".*.xss"), _T(".*.txt"), _T(".*.ssf"), _T(".*.rt"), _T(".*.sup")
  45         },
  46 };
  47
  48 #define WEBSUBEXT _T(".wse")
  49
  50 static int SubFileCompare(const void* elem1, const void* elem2)
  51 {
  52         return(((SubFile*)elem1)->fn.CompareNoCase(((SubFile*)elem2)->fn));
  53 }
  54
  55 void GetSubFileNames(CString fn, CAtlArray<CString>& paths, CAtlArray<SubFile>& ret)
  56 {
  57         ret.RemoveAll();
  58
  59         int extlistnum = countof(ext);
  60         int extsubnum = countof(ext[0]);
  61
  62         fn.Replace('\\', '/');
  63
  64         bool fWeb = false;
  65         {
  66 //              int i = fn.Find(_T("://"));
  67                 int i = fn.Find(_T("http://"));
  68                 if(i > 0) {fn = _T("http") + fn.Mid(i); fWeb = true;}
  69         }
  70
  71         int     l = fn.GetLength(), l2 = l;
  72         l2 = fn.ReverseFind('.');
  73         l = fn.ReverseFind('/') + 1;
  74         if(l2 < l) l2 = l;
  75
  76         CString orgpath = fn.Left(l);
  77         CString title = fn.Mid(l, l2-l);
  78         CString filename = title + _T(".nooneexpectsthespanishinquisition");
  79
  80         if(!fWeb)
  81         {
  82                 // struct _tfinddata_t file, file2;
  83                 // long hFile, hFile2 = 0;
  84
  85                 WIN32_FIND_DATA wfd, wfd2;
  86                 HANDLE hFile, hFile2;
  87
  88                 for(int k = 0; k < paths.GetCount(); k++)
  89                 {
  90                         CString path = paths[k];
  91                         path.Replace('\\', '/');
  92
  93                         l = path.GetLength();
  94                         if(l > 0 && path[l-1] != '/') path += '/';
  95
  96                         if(path.Find(':') == -1 && path.Find(_T("\\\\")) != 0) path = orgpath + path;
  97
  98                         path.Replace(_T("/./"), _T("/"));
  99                         path.Replace('/', '\\');
 100
 101                         // CAtlList<CString> sl;
 102
 103                         bool fEmpty = true;
 104
 105                         if((hFile = FindFirstFile(path + title + _T("*"), &wfd)) != INVALID_HANDLE_VALUE)
 106                         {
 107                                 do
 108                                 {
 109                                         if(filename.CompareNoCase(wfd.cFileName) != 0)
 110                                         {
 111                                                 fEmpty = false;
 112                                                 // sl.AddTail(path + file.name);
 113                                         }
 114                                 }
 115                                 while(FindNextFile(hFile, &wfd));
 116
 117                                 FindClose(hFile);
 118                         }
 119
 120                         // TODO: use 'sl' in the next step to find files (already a nice speedup as it is now...)
 121                         if(fEmpty) continue;
 122
 123                         for(int j = 0; j < extlistnum; j++)
 124                         {
 125                                 for(int i = 0; i < extsubnum; i++)
 126                                 {
 127                                         if((hFile = FindFirstFile(path + title + ext[j][i], &wfd)) != INVALID_HANDLE_VALUE)
 128                                         {
 129                                                 do
 130                                                 {
 131                                                         CString fn = path + wfd.cFileName;
 132
 133                                                         hFile2 = INVALID_HANDLE_VALUE;
 134
 135                                                         if(j == 0 || (hFile2 = FindFirstFile(fn.Left(fn.ReverseFind('.')) + _T(".avi"), &wfd2)) == INVALID_HANDLE_VALUE)
 136                                                         {
 137                                                                 SubFile f;
 138                                                                 f.fn = fn;
 139                                                                 ret.Add(f);
 140                                                         }
 141
 142                                                         if(hFile2 != INVALID_HANDLE_VALUE)
 143                                                         {
 144                                                                 FindClose(hFile2);
 145                                                         }
 146                                                 }
 147                                                 while(FindNextFile(hFile, &wfd));
 148
 149                                                 FindClose(hFile);
 150                                         }
 151                                 }
 152                         }
 153                 }
 154         }
 155         else if(l > 7)
 156         {
 157                 CWebTextFile wtf; // :)
 158                 if(wtf.Open(orgpath + title + WEBSUBEXT))
 159                 {
 160                         CString fn;
 161                         while(wtf.ReadString(fn) && fn.Find(_T("://")) >= 0)
 162                         {
 163                                 SubFile f;
 164                                 f.fn = fn;
 165                                 ret.Add(f);
 166                         }
 167                 }
 168         }
 169
 170         // sort files, this way the user can define the order (movie.00.English.srt, movie.01.Hungarian.srt, etc)
 171
 172         qsort(ret.GetData(), ret.GetCount(), sizeof(SubFile), SubFileCompare);
 173 }