77 lines
2.2 KiB
C
77 lines
2.2 KiB
C
|
#include <windows.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <vector>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <errno.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <fcntl.h>
|
||
|
|
#include <io.h>
|
||
|
|
|
||
|
|
#include "MatroskaParser.h"
|
||
|
|
#include "avisynth.h"
|
||
|
|
|
||
|
|
extern "C" {
|
||
|
|
#include <ffmpeg\avformat.h>
|
||
|
|
#include <ffmpeg\avcodec.h>
|
||
|
|
#include <ffmpeg\swscale.h>
|
||
|
|
#include <postproc\postprocess.h>
|
||
|
|
}
|
||
|
|
|
||
|
|
int GetPPCPUFlags(IScriptEnvironment *Env);
|
||
|
|
int GetSWSCPUFlags(IScriptEnvironment *Env);
|
||
|
|
CodecID MatroskaToFFCodecID(TrackInfo *TI);
|
||
|
|
|
||
|
|
class FFPP : public GenericVideoFilter {
|
||
|
|
private:
|
||
|
|
pp_context_t *PPContext;
|
||
|
|
pp_mode_t *PPMode;
|
||
|
|
SwsContext *SWSTo422P;
|
||
|
|
SwsContext *SWSFrom422P;
|
||
|
|
AVPicture InputPicture;
|
||
|
|
AVPicture OutputPicture;
|
||
|
|
public:
|
||
|
|
FFPP(PClip AChild, const char *APPString, int AQuality, IScriptEnvironment *Env);
|
||
|
|
~FFPP();
|
||
|
|
PVideoFrame __stdcall GetFrame(int n, IScriptEnvironment* Env);
|
||
|
|
};
|
||
|
|
|
||
|
|
class FFBase : public IClip{
|
||
|
|
private:
|
||
|
|
pp_context_t *PPContext;
|
||
|
|
pp_mode_t *PPMode;
|
||
|
|
SwsContext *SWS;
|
||
|
|
int ConvertToFormat;
|
||
|
|
AVPicture PPPicture;
|
||
|
|
protected:
|
||
|
|
VideoInfo VI;
|
||
|
|
AVFrame *DecodeFrame;
|
||
|
|
FILE *AudioCache;
|
||
|
|
|
||
|
|
struct FrameInfo {
|
||
|
|
int64_t DTS;
|
||
|
|
bool KeyFrame;
|
||
|
|
FrameInfo(int64_t ADTS, bool AKeyFrame) : DTS(ADTS), KeyFrame(AKeyFrame) {};
|
||
|
|
};
|
||
|
|
|
||
|
|
std::vector<FrameInfo> FrameToDTS;
|
||
|
|
|
||
|
|
int FindClosestKeyFrame(int AFrame);
|
||
|
|
int FrameFromDTS(int64_t ADTS);
|
||
|
|
int ClosestFrameFromDTS(int64_t ADTS);
|
||
|
|
bool LoadFrameInfoFromFile(const char *AVideoCacheFile, const char *ASource, int AVideoTrack);
|
||
|
|
bool SaveFrameInfoToFile(const char *AVideoCacheFile, const char *ASource, int AVideoTrack);
|
||
|
|
bool SaveTimecodesToFile(const char *ATimecodeFile, int64_t ScaleD, int64_t ScaleN);
|
||
|
|
bool PrepareAudioCache(const char *AAudioCacheFile, const char *ASource, int AAudioTrack, IScriptEnvironment *Env);
|
||
|
|
void InitPP(int AWidth, int AHeight, const char *APPString, int AQuality, int APixelFormat, IScriptEnvironment *Env);
|
||
|
|
void SetOutputFormat(int ACurrentFormat, IScriptEnvironment *Env);
|
||
|
|
PVideoFrame OutputFrame(AVFrame *AFrame, IScriptEnvironment *Env);
|
||
|
|
|
||
|
|
public:
|
||
|
|
FFBase();
|
||
|
|
~FFBase();
|
||
|
|
|
||
|
|
bool __stdcall GetParity(int n) { return false; }
|
||
|
|
void __stdcall SetCacheHints(int cachehints, int frame_range) { }
|
||
|
|
const VideoInfo& __stdcall GetVideoInfo() { return VI; }
|
||
|
|
void __stdcall GetAudio(void* Buf, __int64 Start, __int64 Count, IScriptEnvironment* Env);
|
||
|
|
};
|