174 lines
4.1 KiB
C++
174 lines
4.1 KiB
C++
/*
|
|
* Copyright (C) 2003-2006 Gabest
|
|
* http://www.gabest.org
|
|
*
|
|
* This Program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2, or (at your option)
|
|
* any later version.
|
|
*
|
|
* This Program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with GNU Make; see the file COPYING. If not, write to
|
|
* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
* http://www.gnu.org/copyleft/gpl.html
|
|
*
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include <io.h>
|
|
#include "TextFile.h"
|
|
#include "GFN.h"
|
|
|
|
TCHAR* exttypestr[] =
|
|
{
|
|
_T("srt"), _T("sub"), _T("smi"), _T("psb"),
|
|
_T("ssa"), _T("ass"), _T("idx"), _T("usf"),
|
|
_T("xss"), _T("txt"), _T("ssf")
|
|
};
|
|
|
|
static TCHAR* ext[2][countof(exttypestr)] =
|
|
{
|
|
{
|
|
_T(".srt"), _T(".sub"), _T(".smi"), _T(".psb"),
|
|
_T(".ssa"), _T(".ass"), _T(".idx"), _T(".usf"),
|
|
_T(".xss"), _T(".txt"), _T(".ssf")
|
|
},
|
|
{
|
|
_T(".*.srt"), _T(".*.sub"), _T(".*.smi"), _T(".*.psb"),
|
|
_T(".*.ssa"), _T(".*.ass"), _T(".*.dummyidx"), _T(".*.usf"),
|
|
_T(".*.xss"), _T(".*.txt"), _T(".*.ssf")
|
|
},
|
|
};
|
|
|
|
#define WEBSUBEXT _T(".wse")
|
|
|
|
static int SubFileCompare(const void* elem1, const void* elem2)
|
|
{
|
|
return(((SubFile*)elem1)->fn.CompareNoCase(((SubFile*)elem2)->fn));
|
|
}
|
|
|
|
void GetSubFileNames(CString fn, CAtlArray<CString>& paths, CAtlArray<SubFile>& ret)
|
|
{
|
|
ret.RemoveAll();
|
|
|
|
int extlistnum = countof(ext);
|
|
int extsubnum = countof(ext[0]);
|
|
|
|
fn.Replace('\\', '/');
|
|
|
|
bool fWeb = false;
|
|
{
|
|
// int i = fn.Find(_T("://"));
|
|
int i = fn.Find(_T("http://"));
|
|
if(i > 0) {fn = _T("http") + fn.Mid(i); fWeb = true;}
|
|
}
|
|
|
|
int l = fn.GetLength(), l2 = l;
|
|
l2 = fn.ReverseFind('.');
|
|
l = fn.ReverseFind('/') + 1;
|
|
if(l2 < l) l2 = l;
|
|
|
|
CString orgpath = fn.Left(l);
|
|
CString title = fn.Mid(l, l2-l);
|
|
CString filename = title + _T(".nooneexpectsthespanishinquisition");
|
|
|
|
if(!fWeb)
|
|
{
|
|
// struct _tfinddata_t file, file2;
|
|
// long hFile, hFile2 = 0;
|
|
|
|
WIN32_FIND_DATA wfd, wfd2;
|
|
HANDLE hFile, hFile2;
|
|
|
|
for(int k = 0; k < paths.GetCount(); k++)
|
|
{
|
|
CString path = paths[k];
|
|
path.Replace('\\', '/');
|
|
|
|
l = path.GetLength();
|
|
if(l > 0 && path[l-1] != '/') path += '/';
|
|
|
|
if(path.Find(':') == -1 && path.Find(_T("\\\\")) != 0) path = orgpath + path;
|
|
|
|
path.Replace(_T("/./"), _T("/"));
|
|
path.Replace('/', '\\');
|
|
|
|
// CAtlList<CString> sl;
|
|
|
|
bool fEmpty = true;
|
|
|
|
if((hFile = FindFirstFile(path + title + "*", &wfd)) != INVALID_HANDLE_VALUE)
|
|
{
|
|
do
|
|
{
|
|
if(filename.CompareNoCase(wfd.cFileName) != 0)
|
|
{
|
|
fEmpty = false;
|
|
// sl.AddTail(path + file.name);
|
|
}
|
|
}
|
|
while(FindNextFile(hFile, &wfd));
|
|
|
|
FindClose(hFile);
|
|
}
|
|
|
|
// TODO: use 'sl' in the next step to find files (already a nice speedup as it is now...)
|
|
if(fEmpty) continue;
|
|
|
|
for(int j = 0; j < extlistnum; j++)
|
|
{
|
|
for(int i = 0; i < extsubnum; i++)
|
|
{
|
|
if((hFile = FindFirstFile(path + title + ext[j][i], &wfd)) != INVALID_HANDLE_VALUE)
|
|
{
|
|
do
|
|
{
|
|
CString fn = path + wfd.cFileName;
|
|
|
|
hFile2 = INVALID_HANDLE_VALUE;
|
|
|
|
if(j == 0 || (hFile2 = FindFirstFile(fn.Left(fn.ReverseFind('.')) + _T(".avi"), &wfd2)) == INVALID_HANDLE_VALUE)
|
|
{
|
|
SubFile f;
|
|
f.fn = fn;
|
|
ret.Add(f);
|
|
}
|
|
|
|
if(hFile2 != INVALID_HANDLE_VALUE)
|
|
{
|
|
FindClose(hFile2);
|
|
}
|
|
}
|
|
while(FindNextFile(hFile, &wfd));
|
|
|
|
FindClose(hFile);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else if(l > 7)
|
|
{
|
|
CWebTextFile wtf; // :)
|
|
if(wtf.Open(orgpath + title + WEBSUBEXT))
|
|
{
|
|
CString fn;
|
|
while(wtf.ReadString(fn) && fn.Find(_T("://")) >= 0)
|
|
{
|
|
SubFile f;
|
|
f.fn = fn;
|
|
ret.Add(f);
|
|
}
|
|
}
|
|
}
|
|
|
|
// sort files, this way the user can define the order (movie.00.English.srt, movie.01.Hungarian.srt, etc)
|
|
|
|
qsort(ret.GetData(), ret.GetCount(), sizeof(SubFile), SubFileCompare);
|
|
}
|