forked from mia/Aegisub
441 lines
13 KiB
C++
441 lines
13 KiB
C++
// Copyright (c) 2007-2009 Fredrik Mellbin
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
// THE SOFTWARE.
|
|
|
|
#include "indexing.h"
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <algorithm>
|
|
#include <stdio.h>
|
|
|
|
extern "C" {
|
|
#include <libavutil/sha1.h>
|
|
}
|
|
|
|
|
|
|
|
struct IndexHeader {
|
|
uint32_t Id;
|
|
uint32_t Version;
|
|
uint32_t Tracks;
|
|
uint32_t Decoder;
|
|
uint32_t LAVUVersion;
|
|
uint32_t LAVFVersion;
|
|
uint32_t LAVCVersion;
|
|
uint32_t LSWSVersion;
|
|
uint32_t LPPVersion;
|
|
int64_t FileSize;
|
|
uint8_t FileSignature[20];
|
|
};
|
|
|
|
SharedVideoContext::SharedVideoContext(bool FreeCodecContext) {
|
|
CodecContext = NULL;
|
|
Parser = NULL;
|
|
CS = NULL;
|
|
this->FreeCodecContext = FreeCodecContext;
|
|
}
|
|
|
|
SharedVideoContext::~SharedVideoContext() {
|
|
if (CodecContext) {
|
|
avcodec_close(CodecContext);
|
|
if (FreeCodecContext)
|
|
av_freep(&CodecContext);
|
|
}
|
|
|
|
if (Parser)
|
|
av_parser_close(Parser);
|
|
|
|
if (CS)
|
|
cs_Destroy(CS);
|
|
}
|
|
|
|
SharedAudioContext::SharedAudioContext(bool FreeCodecContext) {
|
|
W64Writer = NULL;
|
|
CodecContext = NULL;
|
|
CurrentSample = 0;
|
|
CS = NULL;
|
|
this->FreeCodecContext = FreeCodecContext;
|
|
}
|
|
|
|
SharedAudioContext::~SharedAudioContext() {
|
|
delete W64Writer;
|
|
if (CodecContext) {
|
|
avcodec_close(CodecContext);
|
|
if (FreeCodecContext)
|
|
av_freep(&CodecContext);
|
|
}
|
|
|
|
if (CS)
|
|
cs_Destroy(CS);
|
|
}
|
|
|
|
TFrameInfo::TFrameInfo(int64_t DTS, int64_t SampleStart, int RepeatPict, bool KeyFrame, int64_t FilePos, unsigned int FrameSize) {
|
|
this->DTS = DTS;
|
|
this->RepeatPict = RepeatPict;
|
|
this->KeyFrame = KeyFrame;
|
|
this->SampleStart = SampleStart;
|
|
this->FilePos = FilePos;
|
|
this->FrameSize = FrameSize;
|
|
}
|
|
|
|
TFrameInfo TFrameInfo::VideoFrameInfo(int64_t DTS, int RepeatPict, bool KeyFrame, int64_t FilePos, unsigned int FrameSize) {
|
|
return TFrameInfo(DTS, 0, RepeatPict, KeyFrame, FilePos, FrameSize);
|
|
}
|
|
|
|
TFrameInfo TFrameInfo::AudioFrameInfo(int64_t DTS, int64_t SampleStart, bool KeyFrame, int64_t FilePos, unsigned int FrameSize) {
|
|
return TFrameInfo(DTS, SampleStart, 0, KeyFrame, FilePos, FrameSize);
|
|
}
|
|
|
|
int FFTrack::WriteTimecodes(const char *TimecodeFile, char *ErrorMsg, unsigned MsgSize) {
|
|
ffms_fstream Timecodes(TimecodeFile, std::ios::out | std::ios::trunc);
|
|
|
|
if (!Timecodes.is_open()) {
|
|
snprintf(ErrorMsg, MsgSize, "Failed to open '%s' for writing", TimecodeFile);
|
|
return 1;
|
|
}
|
|
|
|
Timecodes << "# timecode format v2\n";
|
|
|
|
for (iterator Cur=begin(); Cur!=end(); Cur++)
|
|
Timecodes << std::fixed << ((Cur->DTS * TB.Num) / (double)TB.Den) << "\n";
|
|
|
|
return 0;
|
|
}
|
|
|
|
int FFTrack::FrameFromDTS(int64_t DTS) {
|
|
for (int i = 0; i < static_cast<int>(size()); i++)
|
|
if (at(i).DTS == DTS)
|
|
return i;
|
|
return -1;
|
|
}
|
|
|
|
int FFTrack::ClosestFrameFromDTS(int64_t DTS) {
|
|
int Frame = 0;
|
|
int64_t BestDiff = 0xFFFFFFFFFFFFFFLL; // big number
|
|
for (int i = 0; i < static_cast<int>(size()); i++) {
|
|
int64_t CurrentDiff = FFABS(at(i).DTS - DTS);
|
|
if (CurrentDiff < BestDiff) {
|
|
BestDiff = CurrentDiff;
|
|
Frame = i;
|
|
}
|
|
}
|
|
|
|
return Frame;
|
|
}
|
|
|
|
int FFTrack::FindClosestVideoKeyFrame(int Frame) {
|
|
Frame = FFMIN(FFMAX(Frame, 0), static_cast<int>(size()) - 1);
|
|
for (int i = Frame; i > 0; i--)
|
|
if (at(i).KeyFrame)
|
|
return i;
|
|
return 0;
|
|
}
|
|
|
|
int FFTrack::FindClosestAudioKeyFrame(int64_t Sample) {
|
|
for (size_t i = 0; i < size(); i++) {
|
|
if (at(i).SampleStart == Sample && at(i).KeyFrame)
|
|
return i;
|
|
else if (at(i).SampleStart > Sample && at(i).KeyFrame)
|
|
return i - 1;
|
|
}
|
|
return size() - 1;
|
|
}
|
|
|
|
FFTrack::FFTrack() {
|
|
this->TT = FFMS_TYPE_UNKNOWN;
|
|
this->TB.Num = 0;
|
|
this->TB.Den = 0;
|
|
}
|
|
|
|
FFTrack::FFTrack(int64_t Num, int64_t Den, FFMS_TrackType TT) {
|
|
this->TT = TT;
|
|
this->TB.Num = Num;
|
|
this->TB.Den = Den;
|
|
}
|
|
|
|
int FFIndex::CalculateFileSignature(const char *Filename, int64_t *Filesize, uint8_t Digest[20], char *ErrorMsg, unsigned MsgSize) {
|
|
// use cstdio because Microsoft's implementation of std::fstream doesn't support files >4GB.
|
|
// please kill me now.
|
|
FILE *SFile = ffms_fopen(Filename,"rb");
|
|
|
|
if (SFile == NULL) {
|
|
snprintf(ErrorMsg, MsgSize, "Failed to open '%s' for hashing", Filename);
|
|
return 1;
|
|
}
|
|
|
|
const int BlockSize = 2*1024*1024;
|
|
std::vector<uint8_t> FileBuffer(BlockSize);
|
|
std::vector<uint8_t> ctxmem(av_sha1_size);
|
|
AVSHA1 *ctx = (AVSHA1 *)&ctxmem[0];
|
|
av_sha1_init(ctx);
|
|
|
|
memset(&FileBuffer[0], 0, BlockSize);
|
|
fread(&FileBuffer[0], 1, BlockSize, SFile);
|
|
if (ferror(SFile) && !feof(SFile)) {
|
|
snprintf(ErrorMsg, MsgSize, "Failed to read from '%s' for hashing", Filename);
|
|
av_sha1_final(ctx, Digest);
|
|
fclose(SFile);
|
|
return 1;
|
|
}
|
|
av_sha1_update(ctx, &FileBuffer[0], BlockSize);
|
|
|
|
fseeko(SFile, -BlockSize, SEEK_END);
|
|
memset(&FileBuffer[0], 0, BlockSize);
|
|
fread(&FileBuffer[0], 1, BlockSize, SFile);
|
|
if (ferror(SFile) && !feof(SFile)) {
|
|
snprintf(ErrorMsg, MsgSize, "Failed to seek with offset %d from file end in '%s' for hashing", BlockSize, Filename);
|
|
av_sha1_final(ctx, Digest);
|
|
fclose(SFile);
|
|
return 1;
|
|
}
|
|
av_sha1_update(ctx, &FileBuffer[0], BlockSize);
|
|
|
|
fseeko(SFile, 0, SEEK_END);
|
|
if (ferror(SFile)) {
|
|
snprintf(ErrorMsg, MsgSize, "Failed to seek to end of '%s' for hashing", Filename);
|
|
av_sha1_final(ctx, Digest);
|
|
fclose(SFile);
|
|
return 1;
|
|
}
|
|
*Filesize = ftello(SFile);
|
|
fclose(SFile);
|
|
|
|
av_sha1_final(ctx, Digest);
|
|
return 0;
|
|
}
|
|
|
|
static bool DTSComparison(TFrameInfo FI1, TFrameInfo FI2) {
|
|
return FI1.DTS < FI2.DTS;
|
|
}
|
|
|
|
void FFIndex::Sort() {
|
|
for (FFIndex::iterator Cur=begin(); Cur!=end(); Cur++)
|
|
std::sort(Cur->begin(), Cur->end(), DTSComparison);
|
|
}
|
|
|
|
int FFIndex::CompareFileSignature(const char *Filename, char *ErrorMsg, unsigned MsgSize) {
|
|
int64_t CFilesize;
|
|
uint8_t CDigest[20];
|
|
CalculateFileSignature(Filename, &CFilesize, CDigest, ErrorMsg, MsgSize);
|
|
|
|
if (CFilesize != Filesize || memcmp(CDigest, Digest, sizeof(Digest))) {
|
|
snprintf(ErrorMsg, MsgSize, "Index and source file signature mismatch");
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int FFIndex::WriteIndex(const char *IndexFile, char *ErrorMsg, unsigned MsgSize) {
|
|
ffms_fstream IndexStream(IndexFile, std::ios::out | std::ios::binary | std::ios::trunc);
|
|
|
|
if (!IndexStream.is_open()) {
|
|
snprintf(ErrorMsg, MsgSize, "Failed to open '%s' for writing", IndexFile);
|
|
return 1;
|
|
}
|
|
|
|
// Write the index file header
|
|
IndexHeader IH;
|
|
IH.Id = INDEXID;
|
|
IH.Version = INDEXVERSION;
|
|
IH.Tracks = size();
|
|
IH.Decoder = Decoder;
|
|
IH.LAVUVersion = avutil_version();
|
|
IH.LAVFVersion = avformat_version();
|
|
IH.LAVCVersion = avcodec_version();
|
|
IH.LSWSVersion = swscale_version();
|
|
IH.LPPVersion = postproc_version();
|
|
IH.FileSize = Filesize;
|
|
memcpy(IH.FileSignature, Digest, sizeof(Digest));
|
|
|
|
IndexStream.write(reinterpret_cast<char *>(&IH), sizeof(IH));
|
|
|
|
for (unsigned int i = 0; i < IH.Tracks; i++) {
|
|
uint32_t TT = at(i).TT;
|
|
IndexStream.write(reinterpret_cast<char *>(&TT), sizeof(TT));
|
|
int64_t Num = at(i).TB.Num;
|
|
IndexStream.write(reinterpret_cast<char *>(&Num), sizeof(Num));
|
|
int64_t Den = at(i).TB.Den;
|
|
IndexStream.write(reinterpret_cast<char *>(&Den), sizeof(Den));
|
|
int64_t Frames = at(i).size();
|
|
IndexStream.write(reinterpret_cast<char *>(&Frames), sizeof(Frames));
|
|
|
|
for (FFTrack::iterator Cur=at(i).begin(); Cur!=at(i).end(); Cur++)
|
|
IndexStream.write(reinterpret_cast<char *>(&*Cur), sizeof(TFrameInfo));
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int FFIndex::ReadIndex(const char *IndexFile, char *ErrorMsg, unsigned MsgSize) {
|
|
ffms_fstream Index(IndexFile, std::ios::in | std::ios::binary);
|
|
|
|
if (!Index.is_open()) {
|
|
snprintf(ErrorMsg, MsgSize, "Failed to open '%s' for reading", IndexFile);
|
|
return 1;
|
|
}
|
|
|
|
// Read the index file header
|
|
IndexHeader IH;
|
|
Index.read(reinterpret_cast<char *>(&IH), sizeof(IH));
|
|
if (IH.Id != INDEXID) {
|
|
snprintf(ErrorMsg, MsgSize, "'%s' is not a valid index file", IndexFile);
|
|
return 2;
|
|
}
|
|
|
|
if (IH.Version != INDEXVERSION) {
|
|
snprintf(ErrorMsg, MsgSize, "'%s' is not the expected index version", IndexFile);
|
|
return 3;
|
|
}
|
|
|
|
if (IH.LAVUVersion != avutil_version() || IH.LAVFVersion != avformat_version() ||
|
|
IH.LAVCVersion != avcodec_version() || IH.LSWSVersion != swscale_version() ||
|
|
IH.LPPVersion != postproc_version()) {
|
|
snprintf(ErrorMsg, MsgSize, "A different FFmpeg build was used to create '%s'", IndexFile);
|
|
return 4;
|
|
}
|
|
|
|
Decoder = IH.Decoder;
|
|
Filesize = IH.FileSize;
|
|
memcpy(Digest, IH.FileSignature, sizeof(Digest));
|
|
|
|
try {
|
|
|
|
for (unsigned int i = 0; i < IH.Tracks; i++) {
|
|
// Read how many records belong to the current stream
|
|
uint32_t TT;
|
|
Index.read(reinterpret_cast<char *>(&TT), sizeof(TT));
|
|
int64_t Num;
|
|
Index.read(reinterpret_cast<char *>(&Num), sizeof(Num));
|
|
int64_t Den;
|
|
Index.read(reinterpret_cast<char *>(&Den), sizeof(Den));
|
|
int64_t Frames;
|
|
Index.read(reinterpret_cast<char *>(&Frames), sizeof(Frames));
|
|
push_back(FFTrack(Num, Den, static_cast<FFMS_TrackType>(TT)));
|
|
|
|
TFrameInfo FI = TFrameInfo::VideoFrameInfo(0, 0, false);
|
|
for (size_t j = 0; j < Frames; j++) {
|
|
Index.read(reinterpret_cast<char *>(&FI), sizeof(TFrameInfo));
|
|
at(i).push_back(FI);
|
|
}
|
|
}
|
|
|
|
} catch (...) {
|
|
snprintf(ErrorMsg, MsgSize, "Unknown error while reading index information in '%s'", IndexFile);
|
|
return 5;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
FFIndex::FFIndex() {
|
|
// this comment documents nothing
|
|
}
|
|
|
|
FFIndex::FFIndex(int64_t Filesize, uint8_t Digest[20]) {
|
|
this->Filesize = Filesize;
|
|
memcpy(this->Digest, Digest, sizeof(this->Digest));
|
|
}
|
|
|
|
void FFIndexer::SetIndexMask(int IndexMask) {
|
|
this->IndexMask = IndexMask;
|
|
}
|
|
|
|
void FFIndexer::SetDumpMask(int DumpMask) {
|
|
this->DumpMask = DumpMask;
|
|
}
|
|
|
|
void FFIndexer::SetIgnoreDecodeErrors(bool IgnoreDecodeErrors) {
|
|
this->IgnoreDecodeErrors = IgnoreDecodeErrors;
|
|
}
|
|
|
|
void FFIndexer::SetProgressCallback(TIndexCallback IC, void *ICPrivate) {
|
|
this->IC = IC;
|
|
this->ICPrivate = ICPrivate;
|
|
}
|
|
|
|
void FFIndexer::SetAudioNameCallback(TAudioNameCallback ANC, void *ANCPrivate) {
|
|
this->ANC = ANC;
|
|
this->ANCPrivate = ANCPrivate;
|
|
}
|
|
|
|
FFIndexer *FFIndexer::CreateFFIndexer(const char *Filename, char *ErrorMsg, unsigned MsgSize) {
|
|
AVFormatContext *FormatContext = NULL;
|
|
|
|
if (av_open_input_file(&FormatContext, Filename, NULL, 0, NULL) != 0) {
|
|
snprintf(ErrorMsg, MsgSize, "Can't open '%s'", Filename);
|
|
return NULL;
|
|
}
|
|
|
|
// Do matroska indexing instead?
|
|
if (!strcmp(FormatContext->iformat->name, "matroska")) {
|
|
av_close_input_file(FormatContext);
|
|
return new FFMatroskaIndexer(Filename, ErrorMsg, MsgSize);
|
|
}
|
|
|
|
#ifdef HAALISOURCE
|
|
// Do haali ts indexing instead?
|
|
if (!strcmp(FormatContext->iformat->name, "mpeg") || !strcmp(FormatContext->iformat->name, "mpegts")) {
|
|
av_close_input_file(FormatContext);
|
|
return new FFHaaliIndexer(Filename, 0, ErrorMsg, MsgSize);
|
|
}
|
|
|
|
if (!strcmp(FormatContext->iformat->name, "ogg")) {
|
|
av_close_input_file(FormatContext);
|
|
return new FFHaaliIndexer(Filename, 1, ErrorMsg, MsgSize);
|
|
}
|
|
#endif
|
|
|
|
return new FFLAVFIndexer(Filename, FormatContext, ErrorMsg, MsgSize);
|
|
}
|
|
|
|
FFIndexer::FFIndexer(const char *Filename, char *ErrorMsg, unsigned MsgSize) : DecodingBuffer(AVCODEC_MAX_AUDIO_FRAME_SIZE * 5) {
|
|
if (FFIndex::CalculateFileSignature(Filename, &Filesize, Digest, ErrorMsg, MsgSize))
|
|
throw ErrorMsg;
|
|
}
|
|
|
|
FFIndexer::~FFIndexer() {
|
|
|
|
}
|
|
|
|
bool FFIndexer::WriteAudio(SharedAudioContext &AudioContext, FFIndex *Index, int Track, int DBSize, char *ErrorMsg, unsigned MsgSize) {
|
|
// Delay writer creation until after an audio frame has been decoded. This ensures that all parameters are known when writing the headers.
|
|
if (DBSize > 0) {
|
|
if (!AudioContext.W64Writer) {
|
|
FFAudioProperties AP;
|
|
FillAP(AP, AudioContext.CodecContext, (*Index)[Track]);
|
|
int FNSize = (*ANC)(SourceFile, Track, &AP, NULL, 0, ANCPrivate);
|
|
std::vector<char> WName(FNSize);
|
|
(*ANC)(SourceFile, Track, &AP, &WName[0], FNSize, ANCPrivate);
|
|
std::string WN(&WName[0]);
|
|
try {
|
|
AudioContext.W64Writer = new Wave64Writer(WN.c_str(), av_get_bits_per_sample_format(AudioContext.CodecContext->sample_fmt),
|
|
AudioContext.CodecContext->channels, AudioContext.CodecContext->sample_rate, (AudioContext.CodecContext->sample_fmt == SAMPLE_FMT_FLT) || (AudioContext.CodecContext->sample_fmt == SAMPLE_FMT_DBL));
|
|
} catch (...) {
|
|
snprintf(ErrorMsg, MsgSize, "Failed to write wave data");
|
|
return false;
|
|
}
|
|
}
|
|
|
|
AudioContext.W64Writer->WriteData(&DecodingBuffer[0], DBSize);
|
|
}
|
|
|
|
return true;
|
|
}
|