mirror of https://github.com/odrling/Aegisub
151 lines
4.1 KiB
C++
151 lines
4.1 KiB
C++
// Copyright (c) 2022, arch1t3cht <arch1t3cht@gmail.com>
|
|
//
|
|
// Permission to use, copy, modify, and distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice appear in all copies.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
//
|
|
// Aegisub Project http://www.aegisub.org/
|
|
|
|
/// @file vapoursynth_wrap.cpp
|
|
/// @brief Wrapper-layer for VapourSynth
|
|
/// @ingroup video_input audio_input
|
|
///
|
|
|
|
#ifdef WITH_VAPOURSYNTH
|
|
#include "vapoursynth_wrap.h"
|
|
|
|
#include "VSScript4.h"
|
|
|
|
#include "options.h"
|
|
|
|
#ifndef _WIN32
|
|
#include <dlfcn.h>
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
#define VSSCRIPT_SO "vsscript.dll"
|
|
|
|
#ifdef _WIN64
|
|
#define VS_INSTALL_REGKEY L"Software\\VapourSynth"
|
|
#else
|
|
#define VS_INSTALL_REGKEY L"Software\\VapourSynth-32"
|
|
#endif
|
|
|
|
#else
|
|
#ifdef __APPLE__
|
|
#define VSSCRIPT_SO "libvapoursynth-script.dylib"
|
|
#define DLOPEN_FLAGS RTLD_LAZY | RTLD_GLOBAL
|
|
#else
|
|
#define VSSCRIPT_SO "libvapoursynth-script.so"
|
|
#define DLOPEN_FLAGS RTLD_LAZY | RTLD_GLOBAL | RTLD_DEEPBIND
|
|
#endif
|
|
#endif
|
|
|
|
// Allocate storage for and initialise static members
|
|
namespace {
|
|
bool vs_loaded = false;
|
|
#ifdef _WIN32
|
|
HINSTANCE hLib = nullptr;
|
|
#else
|
|
void* hLib = nullptr;
|
|
#endif
|
|
const VSAPI *api = nullptr;
|
|
VSSCRIPTAPI *scriptapi = nullptr;
|
|
std::mutex VapourSynthMutex;
|
|
}
|
|
|
|
typedef VSSCRIPTAPI* VS_CC FUNC(int);
|
|
|
|
VapourSynthWrapper::VapourSynthWrapper() {
|
|
// VSScript assumes it's only loaded once, so unlike AVS we can't unload it when the refcount reaches zero
|
|
if (!vs_loaded) {
|
|
#ifdef _WIN32
|
|
|
|
std::wstring vsscriptDLLpath = L"";
|
|
|
|
HKEY hKey;
|
|
LONG lRes = RegOpenKeyEx(HKEY_CURRENT_USER, VS_INSTALL_REGKEY, 0, KEY_READ, &hKey);
|
|
|
|
if (lRes != ERROR_SUCCESS) {
|
|
lRes = RegOpenKeyEx(HKEY_LOCAL_MACHINE, VS_INSTALL_REGKEY, 0, KEY_READ, &hKey);
|
|
}
|
|
|
|
if (lRes == ERROR_SUCCESS) {
|
|
WCHAR szBuffer[512];
|
|
DWORD dwBufferSize = sizeof(szBuffer);
|
|
ULONG nError;
|
|
|
|
nError = RegQueryValueEx(hKey, L"VSScriptDLL", 0, nullptr, (LPBYTE)szBuffer, &dwBufferSize);
|
|
RegCloseKey(hKey);
|
|
|
|
if (nError == ERROR_SUCCESS)
|
|
vsscriptDLLpath = szBuffer;
|
|
}
|
|
|
|
if (vsscriptDLLpath.length()) {
|
|
hLib = LoadLibraryW(vsscriptDLLpath.c_str());
|
|
}
|
|
|
|
if (!hLib) {
|
|
#define CONCATENATE(x, y) x ## y
|
|
#define _Lstr(x) CONCATENATE(L, x)
|
|
hLib = LoadLibraryW(_Lstr(VSSCRIPT_SO));
|
|
#undef _Lstr
|
|
#undef CONCATENATE
|
|
}
|
|
#else
|
|
hLib = dlopen(VSSCRIPT_SO, DLOPEN_FLAGS);
|
|
#endif
|
|
|
|
if (!hLib)
|
|
throw VapourSynthError("Could not load " VSSCRIPT_SO ". Make sure VapourSynth is installed correctly.");
|
|
|
|
#ifdef _WIN32
|
|
FUNC* getVSScriptAPI = (FUNC*)GetProcAddress(hLib, "getVSScriptAPI");
|
|
#else
|
|
FUNC* getVSScriptAPI = (FUNC*)dlsym(hLib, "getVSScriptAPI");
|
|
#endif
|
|
if (!getVSScriptAPI)
|
|
throw VapourSynthError("Failed to get address of getVSScriptAPI from " VSSCRIPT_SO);
|
|
|
|
// Python will set the program's locale to the user's default locale, which will break
|
|
// half of wxwidgets on some operating systems due to locale mismatches. There's not really anything
|
|
// we can do to fix it except for saving it and setting it back to its original value afterwards.
|
|
std::string oldlocale(setlocale(LC_ALL, NULL));
|
|
scriptapi = getVSScriptAPI(VSSCRIPT_API_VERSION);
|
|
setlocale(LC_ALL, oldlocale.c_str());
|
|
|
|
if (!scriptapi)
|
|
throw VapourSynthError("Failed to get VapourSynth ScriptAPI. Make sure VapourSynth is installed correctly.");
|
|
|
|
api = scriptapi->getVSAPI(VAPOURSYNTH_API_VERSION);
|
|
|
|
if (!api)
|
|
throw VapourSynthError("Failed to get VapourSynth API");
|
|
|
|
vs_loaded = true;
|
|
}
|
|
}
|
|
|
|
std::mutex& VapourSynthWrapper::GetMutex() const {
|
|
return VapourSynthMutex;
|
|
}
|
|
|
|
const VSAPI *VapourSynthWrapper::GetAPI() const {
|
|
return api;
|
|
}
|
|
|
|
const VSSCRIPTAPI *VapourSynthWrapper::GetScriptAPI() const {
|
|
return scriptapi;
|
|
}
|
|
|
|
#endif
|