Aegisub/vsfilter/subtitles/GFN.cpp
2008-03-08 23:44:51 +00:00

173 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 + _T("*"), &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);
}