static AVSValue __cdecl CreateFFIndex(AVSValue Args, void* UserData, IScriptEnvironment* Env) {
FFMS_Init((int)AvisynthToFFCPUFlags(Env->GetCPUFlags()), Args[7].AsBool(false));
char ErrorMsg[1024];
FFMS_ErrorInfo E;
E.Buffer = ErrorMsg;
E.BufferSize = sizeof(ErrorMsg);
if (!Args[0].Defined())
Env->ThrowError("FFIndex: No source specified");
const char *Source = Args[0].AsString();
const char *CacheFile = Args[1].AsString("");
int IndexMask = Args[2].AsInt(-1);
int DumpMask = Args[3].AsInt(0);
const char *AudioFile = Args[4].AsString("%sourcefile%.%trackzn%.w64");
int ErrorHandling = Args[5].AsInt(FFMS_IEH_IGNORE);
bool OverWrite = Args[6].AsBool(false);
const char *DemuxerStr = Args[8].AsString("default");
std::string DefaultCache(Source);
DefaultCache.append(".ffindex");
if (!strcmp(CacheFile, ""))
CacheFile = DefaultCache.c_str();
if (!strcmp(AudioFile, ""))
Env->ThrowError("FFIndex: Specifying an empty audio filename is not allowed");
int Demuxer;
if (!strcmp(DemuxerStr, "default"))
Demuxer = FFMS_SOURCE_DEFAULT;
else if (!strcmp(DemuxerStr, "lavf"))
Demuxer = FFMS_SOURCE_LAVF;
else if (!strcmp(DemuxerStr, "matroska"))
Demuxer = FFMS_SOURCE_MATROSKA;
else if (!strcmp(DemuxerStr, "haalimpeg"))
Demuxer = FFMS_SOURCE_HAALIMPEG;
else if (!strcmp(DemuxerStr, "haaliogg"))
Demuxer = FFMS_SOURCE_HAALIOGG;
else
Env->ThrowError("FFIndex: Invalid demuxer requested");
FFMS_Index *Index = FFMS_ReadIndex(CacheFile, &E);
if (OverWrite || !Index || (Index && FFMS_IndexBelongsToFile(Index, Source, 0) != FFMS_ERROR_SUCCESS)) {
FFMS_Indexer *Indexer = FFMS_CreateIndexerWithDemuxer(Source, Demuxer, &E);
if (!Indexer)
Env->ThrowError("FFIndex: %s", E.Buffer);
if (!(Index = FFMS_DoIndexing(Indexer, IndexMask, DumpMask, FFMS_DefaultAudioFilename, (void *)AudioFile, ErrorHandling, NULL, NULL, &E)))
Env->ThrowError("FFIndex: %s", E.Buffer);
if (FFMS_WriteIndex(CacheFile, Index, &E)) {
FFMS_DestroyIndex(Index);
Env->ThrowError("FFIndex: %s", E.Buffer);
}
FFMS_DestroyIndex(Index);
if (!OverWrite)
return AVSValue(1);
else
return AVSValue(2);
} else {
FFMS_DestroyIndex(Index);
return AVSValue(0);
}
}
FFMS_API(FFMS_Indexer *) FFMS_CreateIndexer(const char *SourceFile, FFMS_ErrorInfo *ErrorInfo) {
return FFMS_CreateIndexerWithDemuxer(SourceFile, FFMS_SOURCE_DEFAULT, ErrorInfo);
}
