mirror of https://github.com/odrling/Aegisub
91 lines
3.3 KiB
C++
91 lines
3.3 KiB
C++
|
#include "ffmpegsource.h"
|
||
|
|
|
||
|
|
int GetPPCPUFlags(IScriptEnvironment *Env) {
|
||
|
|
int Flags = 0;
|
||
|
|
long CPUFlags = Env->GetCPUFlags();
|
||
|
|
|
||
|
|
if (CPUFlags & CPUF_MMX)
|
||
|
|
CPUFlags |= PP_CPU_CAPS_MMX;
|
||
|
|
if (CPUFlags & CPUF_INTEGER_SSE)
|
||
|
|
CPUFlags |= PP_CPU_CAPS_MMX2;
|
||
|
|
if (CPUFlags & CPUF_3DNOW)
|
||
|
|
CPUFlags |= PP_CPU_CAPS_3DNOW;
|
||
|
|
|
||
|
|
return Flags;
|
||
|
|
}
|
||
|
|
|
||
|
|
FFPP::FFPP(PClip AChild, const char *APPString, int AQuality, IScriptEnvironment *Env) : GenericVideoFilter(AChild) {
|
||
|
|
if (!strcmp(APPString, ""))
|
||
|
|
Env->ThrowError("FFPP: PP argument is empty");
|
||
|
|
if (AQuality < 0 || AQuality > PP_QUALITY_MAX)
|
||
|
|
Env->ThrowError("FFPP: Quality is out of range");
|
||
|
|
|
||
|
|
PPContext = NULL;
|
||
|
|
PPMode = NULL;
|
||
|
|
SWSTo422P = NULL;
|
||
|
|
SWSFrom422P = NULL;
|
||
|
|
|
||
|
|
memset(&InputPicture, 0, sizeof(InputPicture));
|
||
|
|
memset(&OutputPicture, 0, sizeof(OutputPicture));
|
||
|
|
|
||
|
|
PPMode = pp_get_mode_by_name_and_quality((char *)APPString, AQuality);
|
||
|
|
if (!PPMode)
|
||
|
|
Env->ThrowError("FFPP: Invalid postprocesing settings");
|
||
|
|
|
||
|
|
int Flags = GetPPCPUFlags(Env);
|
||
|
|
|
||
|
|
if (vi.IsYV12()) {
|
||
|
|
Flags |= PP_FORMAT_420;
|
||
|
|
} else if (vi.IsYUY2()) {
|
||
|
|
Flags |= PP_FORMAT_422;
|
||
|
|
SWSTo422P = sws_getContext(vi.width, vi.height, PIX_FMT_YUV422, vi.width, vi.height, PIX_FMT_YUV422P, GetSWSCPUFlags(Env) | SWS_BICUBIC, NULL, NULL, NULL);
|
||
|
|
SWSFrom422P = sws_getContext(vi.width, vi.height, PIX_FMT_YUV422P, vi.width, vi.height, PIX_FMT_YUV422, GetSWSCPUFlags(Env) | SWS_BICUBIC, NULL, NULL, NULL);
|
||
|
|
avpicture_alloc(&InputPicture, PIX_FMT_YUV422P, vi.width, vi.height);
|
||
|
|
avpicture_alloc(&OutputPicture, PIX_FMT_YUV422P, vi.width, vi.height);
|
||
|
|
} else {
|
||
|
|
Env->ThrowError("FFPP: Only YV12 and YUY2 video supported");
|
||
|
|
}
|
||
|
|
|
||
|
|
PPContext = pp_get_context(vi.width, vi.height, Flags);
|
||
|
|
if (!PPContext)
|
||
|
|
Env->ThrowError("FFPP: Failed to create context");
|
||
|
|
}
|
||
|
|
|
||
|
|
FFPP::~FFPP() {
|
||
|
|
if (PPMode)
|
||
|
|
pp_free_mode(PPMode);
|
||
|
|
if (PPContext)
|
||
|
|
pp_free_context(PPContext);
|
||
|
|
if (SWSTo422P)
|
||
|
|
sws_freeContext(SWSTo422P);
|
||
|
|
if (SWSFrom422P)
|
||
|
|
sws_freeContext(SWSFrom422P);
|
||
|
|
avpicture_free(&InputPicture);
|
||
|
|
avpicture_free(&OutputPicture);
|
||
|
|
}
|
||
|
|
|
||
|
|
PVideoFrame __stdcall FFPP::GetFrame(int n, IScriptEnvironment* Env) {
|
||
|
|
PVideoFrame Src = child->GetFrame(n, Env);
|
||
|
|
PVideoFrame Dst = Env->NewVideoFrame(vi);
|
||
|
|
|
||
|
|
if (vi.IsYV12()) {
|
||
|
|
uint8_t *SrcData[3] = {(uint8_t *)Src->GetReadPtr(PLANAR_Y), (uint8_t *)Src->GetReadPtr(PLANAR_U), (uint8_t *)Src->GetReadPtr(PLANAR_V)};
|
||
|
|
int SrcStride[3] = {Src->GetPitch(PLANAR_Y), Src->GetPitch(PLANAR_U), Src->GetPitch(PLANAR_V)};
|
||
|
|
uint8_t *DstData[3] = {Dst->GetWritePtr(PLANAR_Y), Dst->GetWritePtr(PLANAR_U), Dst->GetWritePtr(PLANAR_V)};
|
||
|
|
int DstStride[3] = {Dst->GetPitch(PLANAR_Y), Dst->GetPitch(PLANAR_U), Dst->GetPitch(PLANAR_V)};
|
||
|
|
|
||
|
|
pp_postprocess(SrcData, SrcStride, DstData, DstStride, vi.width, vi.height, NULL, 0, PPMode, PPContext, 0);
|
||
|
|
} else if (vi.IsYUY2()) {
|
||
|
|
uint8_t *SrcData[1] = {(uint8_t *)Src->GetReadPtr()};
|
||
|
|
int SrcStride[1] = {Src->GetPitch()};
|
||
|
|
sws_scale(SWSTo422P, SrcData, SrcStride, 0, vi.height, InputPicture.data, InputPicture.linesize);
|
||
|
|
|
||
|
|
pp_postprocess(InputPicture.data, InputPicture.linesize, OutputPicture.data, OutputPicture.linesize, vi.width, vi.height, NULL, 0, PPMode, PPContext, 0);
|
||
|
|
|
||
|
|
uint8_t *DstData[1] = {Dst->GetWritePtr()};
|
||
|
|
int DstStride[1] = {Dst->GetPitch()};
|
||
|
|
sws_scale(SWSFrom422P, OutputPicture.data, OutputPicture.linesize, 0, vi.height, DstData, DstStride);
|
||
|
|
}
|
||
|
|
|
||
|
|
return Dst;
|
||
|
|
}
|