mirror of
https://github.com/staxrip/staxrip.git
synced 2026-10-22 09:23:18 +08:00
Frank Skare
264 lines
8.6 KiB
C++
264 lines
8.6 KiB
C++
|
|
#include "VapourSynthServer.h"
|
|
|
|
|
|
// IUnknown
|
|
|
|
HRESULT __stdcall VapourSynthServer::QueryInterface(const IID& iid, void** ppv)
|
|
{
|
|
if (!ppv)
|
|
return E_POINTER;
|
|
|
|
if (iid == IID_IUnknown || iid == IID_IFrameServer)
|
|
{
|
|
*ppv = this;
|
|
AddRef();
|
|
return S_OK;
|
|
}
|
|
|
|
*ppv = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
|
|
ULONG __stdcall VapourSynthServer::AddRef()
|
|
{
|
|
return ++m_References;
|
|
}
|
|
|
|
|
|
ULONG __stdcall VapourSynthServer::Release() {
|
|
int refs = --m_References;
|
|
|
|
if (!refs)
|
|
delete this;
|
|
|
|
return refs;
|
|
}
|
|
|
|
|
|
// IFrameServer
|
|
|
|
HRESULT __stdcall VapourSynthServer::OpenFile(WCHAR* file)
|
|
{
|
|
try
|
|
{
|
|
static bool wasResolved = false;
|
|
|
|
if (!wasResolved)
|
|
{
|
|
HMODULE dll = LoadLibrary(L"VSScript.dll");
|
|
|
|
if (!dll)
|
|
{
|
|
std::string msg = GetWinErrorMessage(GetLastError());
|
|
throw std::runtime_error("Failed to load VapourSynth:\r\n\r\n" + msg);
|
|
}
|
|
|
|
bool x64 = sizeof(void*) == 8;
|
|
|
|
std::array resolvePairs = {
|
|
std::pair((void**)&vs_init, x64 ? "vsscript_init" : "_vsscript_init@0"),
|
|
std::pair((void**)&vs_finalize, x64 ? "vsscript_finalize" : "_vsscript_finalize@0"),
|
|
std::pair((void**)&vs_evaluateScript, x64 ? "vsscript_evaluateScript" : "_vsscript_evaluateScript@16"),
|
|
std::pair((void**)&vs_evaluateFile, x64 ? "vsscript_evaluateFile" : "_vsscript_evaluateFile@12"),
|
|
std::pair((void**)&vs_freeScript, x64 ? "vsscript_freeScript" : "_vsscript_freeScript@4"),
|
|
std::pair((void**)&vs_getError, x64 ? "vsscript_getError" : "_vsscript_getError@4"),
|
|
std::pair((void**)&vs_getOutput, x64 ? "vsscript_getOutput" : "_vsscript_getOutput@8"),
|
|
std::pair((void**)&vs_clearOutput, x64 ? "vsscript_clearOutput" : "_vsscript_clearOutput@8"),
|
|
std::pair((void**)&vs_getCore, x64 ? "vsscript_getCore" : "_vsscript_getCore@4"),
|
|
std::pair((void**)&vs_getVSApi, x64 ? "vsscript_getVSApi" : "_vsscript_getVSApi@0")
|
|
};
|
|
|
|
for (auto& i : resolvePairs)
|
|
if (NULL == (*(i.first) = GetProcAddress(dll, i.second)))
|
|
throw std::exception("Failed to resolve VapourSynth vsscript functions");
|
|
}
|
|
|
|
wasResolved = true;
|
|
|
|
if (!(m_vsInit = vs_init()))
|
|
throw std::exception("Failed to initialize VapourSynth");
|
|
|
|
m_vsAPI = vs_getVSApi();
|
|
|
|
if (!m_vsAPI)
|
|
throw std::exception("Failed to call VapourSynth vsscript_getVSApi");
|
|
|
|
std::string utf8file = ConvertWideToUtf8(file);
|
|
|
|
if (vs_evaluateFile(&m_vsScript, utf8file.c_str(), 0))
|
|
{
|
|
m_Error = ConvertUtf8ToWide(vs_getError(m_vsScript));
|
|
Free();
|
|
return E_FAIL;
|
|
}
|
|
|
|
m_vsNode = vs_getOutput(m_vsScript, 0);
|
|
|
|
if (!m_vsNode)
|
|
throw std::exception("Failed to get VapourSynth output");
|
|
|
|
m_vsInfo = m_vsAPI->getVideoInfo(m_vsNode);
|
|
|
|
if (!m_vsInfo)
|
|
throw std::exception("Failed to get VapourSynth info");
|
|
|
|
m_Info.Width = m_vsInfo->width;
|
|
m_Info.Height = m_vsInfo->height;
|
|
m_Info.FrameCount = m_vsInfo->numFrames;
|
|
m_Info.FrameRateNum = m_vsInfo->fpsNum;
|
|
m_Info.FrameRateDen = m_vsInfo->fpsDen;
|
|
|
|
const VSFormat* format = m_vsInfo->format;
|
|
int id = format->id;
|
|
|
|
if (id == pfCompatBGR32) m_Info.ColorSpace = VideoInfo::CS_BGR32;
|
|
else if (id == pfCompatYUY2) m_Info.ColorSpace = VideoInfo::CS_YUY2;
|
|
else if (id == pfYUV444P8) m_Info.ColorSpace = VideoInfo::CS_YV24;
|
|
else if (id == pfYUV422P8) m_Info.ColorSpace = VideoInfo::CS_YV16;
|
|
else if (id == pfYUV420P8) m_Info.ColorSpace = VideoInfo::CS_YV12;
|
|
else if (id == pfYUV410P8) m_Info.ColorSpace = VideoInfo::CS_YUV9;
|
|
else if (id == pfYUV411P8) m_Info.ColorSpace = VideoInfo::CS_YV411;
|
|
else if (id == pfGray8) m_Info.ColorSpace = VideoInfo::CS_Y8;
|
|
else if (id == pfYUV444P10) m_Info.ColorSpace = VideoInfo::CS_YUV444P10;
|
|
else if (id == pfYUV422P10) m_Info.ColorSpace = VideoInfo::CS_YUV422P10;
|
|
else if (id == pfYUV420P10) m_Info.ColorSpace = VideoInfo::CS_YUV420P10;
|
|
else if (id == pfYUV444P12) m_Info.ColorSpace = VideoInfo::CS_YUV444P12;
|
|
else if (id == pfYUV422P12) m_Info.ColorSpace = VideoInfo::CS_YUV422P12;
|
|
else if (id == pfYUV420P12) m_Info.ColorSpace = VideoInfo::CS_YUV420P12;
|
|
else if (id == pfYUV444P14) m_Info.ColorSpace = VideoInfo::CS_YUV444P14;
|
|
else if (id == pfYUV422P14) m_Info.ColorSpace = VideoInfo::CS_YUV422P14;
|
|
else if (id == pfYUV420P14) m_Info.ColorSpace = VideoInfo::CS_YUV420P14;
|
|
else if (id == pfYUV444P16) m_Info.ColorSpace = VideoInfo::CS_YUV444P16;
|
|
else if (id == pfYUV422P16) m_Info.ColorSpace = VideoInfo::CS_YUV422P16;
|
|
else if (id == pfYUV420P16) m_Info.ColorSpace = VideoInfo::CS_YUV420P16;
|
|
else if (id == pfGray16) m_Info.ColorSpace = VideoInfo::CS_Y16;
|
|
else if (id == pfYUV444PS) m_Info.ColorSpace = VideoInfo::CS_YUV444PS;
|
|
else if (id == pfGrayS) m_Info.ColorSpace = VideoInfo::CS_Y32;
|
|
else if (id == pfRGB24) m_Info.ColorSpace = VideoInfo::CS_RGBP;
|
|
else if (id == pfRGB30) m_Info.ColorSpace = VideoInfo::CS_RGBP10;
|
|
else if (id == pfRGB48) m_Info.ColorSpace = VideoInfo::CS_RGBP16;
|
|
else if (id == pfGrayS) m_Info.ColorSpace = VideoInfo::CS_Y32;
|
|
else if (format->bitsPerSample == 12 && format->colorFamily == cmRGB)
|
|
m_Info.ColorSpace = VideoInfo::CS_RGBP12;
|
|
else if (format->bitsPerSample == 14 && format->colorFamily == cmRGB)
|
|
m_Info.ColorSpace = VideoInfo::CS_RGBP14;
|
|
else if (format->bitsPerSample == 32 && format->colorFamily == cmYUV && format->sampleType == stFloat && format->subSamplingH == 0 && format->subSamplingW == 1)
|
|
m_Info.ColorSpace = VideoInfo::CS_YUV422PS;
|
|
else if (format->bitsPerSample == 32 && format->colorFamily == cmYUV && format->sampleType == stFloat && format->subSamplingH == 1 && format->subSamplingW == 1)
|
|
m_Info.ColorSpace = VideoInfo::CS_YUV420PS;
|
|
else if (format->bitsPerSample == 12 && format->colorFamily == cmGray && format->sampleType == stInteger)
|
|
m_Info.ColorSpace = VideoInfo::CS_Y12;
|
|
else if (format->bitsPerSample == 10 && format->colorFamily == cmGray && format->sampleType == stInteger)
|
|
m_Info.ColorSpace = VideoInfo::CS_Y10;
|
|
else if (format->bitsPerSample == 14 && format->colorFamily == cmGray && format->sampleType == stInteger)
|
|
m_Info.ColorSpace = VideoInfo::CS_Y14;
|
|
|
|
return S_OK;
|
|
}
|
|
catch (std::exception& e)
|
|
{
|
|
m_Error = ConvertAnsiToWide(e.what());
|
|
}
|
|
|
|
Free();
|
|
return E_FAIL;
|
|
}
|
|
|
|
|
|
HRESULT __stdcall VapourSynthServer::GetFrame(int position, void** data, int& pitch)
|
|
{
|
|
if (!data)
|
|
return E_POINTER;
|
|
|
|
if (!m_vsAPI || !m_vsNode)
|
|
return E_FAIL;
|
|
|
|
const VSFrameRef* frame = m_vsAPI->getFrame(position, m_vsNode, m_vsErrorMessage, sizeof(m_vsErrorMessage));
|
|
|
|
if (!frame)
|
|
{
|
|
m_Error = ConvertUtf8ToWide(m_vsErrorMessage);
|
|
return E_FAIL;
|
|
}
|
|
|
|
*data = (void*)m_vsAPI->getReadPtr(frame, 0);
|
|
|
|
if (!(*data))
|
|
{
|
|
m_Error = L"VapourSynthServer m_vsAPI->getReadPtr returned NULL";
|
|
return E_FAIL;
|
|
}
|
|
|
|
pitch = m_vsAPI->getStride(frame, 0);
|
|
|
|
if (m_vsFrame)
|
|
m_vsAPI->freeFrame(m_vsFrame);
|
|
|
|
m_vsFrame = frame;
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
ServerInfo* __stdcall VapourSynthServer::GetInfo()
|
|
{
|
|
return &m_Info;
|
|
}
|
|
|
|
|
|
WCHAR* __stdcall VapourSynthServer::GetError()
|
|
{
|
|
return (WCHAR*)m_Error.c_str();
|
|
}
|
|
|
|
|
|
/////////// local
|
|
|
|
VapourSynthServer::~VapourSynthServer()
|
|
{
|
|
Free();
|
|
}
|
|
|
|
|
|
void VapourSynthServer::Free()
|
|
{
|
|
if (m_vsAPI)
|
|
{
|
|
if (m_vsFrame)
|
|
{
|
|
m_vsAPI->freeFrame(m_vsFrame);
|
|
m_vsFrame = NULL;
|
|
}
|
|
|
|
if (m_vsNode)
|
|
{
|
|
m_vsAPI->freeNode(m_vsNode);
|
|
m_vsNode = NULL;
|
|
}
|
|
}
|
|
|
|
if (m_vsScript)
|
|
{
|
|
vs_freeScript(m_vsScript);
|
|
m_vsScript = NULL;
|
|
}
|
|
|
|
if (m_vsInit)
|
|
{
|
|
vs_finalize();
|
|
m_vsInit = 0;
|
|
}
|
|
}
|
|
|
|
|
|
// Extern
|
|
|
|
extern "C" __declspec(dllexport) VapourSynthServer* __stdcall
|
|
CreateVapourSynthServer()
|
|
{
|
|
VapourSynthServer* server = new VapourSynthServer();
|
|
server->AddRef();
|
|
return server;
|
|
}
|