Skip to content

Commit

Permalink
Merge branch 'avisynth' into feature
Browse files Browse the repository at this point in the history
  • Loading branch information
arch1t3cht committed Aug 13, 2022
2 parents 95bd0ed + ad443dd commit 53466be
Show file tree
Hide file tree
Showing 8 changed files with 66 additions and 22 deletions.
8 changes: 6 additions & 2 deletions meson.build
Original file line number Diff line number Diff line change
Expand Up @@ -224,9 +224,9 @@ foreach dep: [
endif
endforeach

if host_machine.system() == 'windows' and get_option('avisynth').enabled()
if get_option('avisynth').enabled()
conf.set('WITH_AVISYNTH', 1) # bundled separately with installer
deps += cc.find_library('avifil32', required: true)
dep_avail += 'AviSynth'

avs_opt = cmake.subproject_options()
avs_opt.add_cmake_defines({
Expand All @@ -235,6 +235,10 @@ if host_machine.system() == 'windows' and get_option('avisynth').enabled()

avs = cmake.subproject('avisynth', options: avs_opt)
deps_inc += avs.include_directories('AviSynth-Headers')

if host_machine.system() == 'windows'
deps += cc.find_library('avifil32', required: true)
endif
endif

if host_machine.system() == 'windows' and not get_option('directsound').disabled()
Expand Down
18 changes: 12 additions & 6 deletions src/audio_provider_avs.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ class AvisynthAudioProvider final : public agi::AudioProvider {
bool NeedsCache() const override { return true; }
};

AvisynthAudioProvider::AvisynthAudioProvider(agi::fs::path const& filename) {
AvisynthAudioProvider::AvisynthAudioProvider(agi::fs::path const& filename) try {
agi::acs::CheckFileRead(filename);

std::lock_guard<std::mutex> lock(avs_wrapper.GetMutex());
Expand Down Expand Up @@ -100,19 +100,25 @@ AvisynthAudioProvider::AvisynthAudioProvider(agi::fs::path const& filename) {
throw agi::AudioProviderError("Avisynth error: " + errmsg);
}
}
catch (AvisynthError& err) {
throw agi::AudioProviderError("Avisynth error: " + std::string(err.msg));
}

void AvisynthAudioProvider::LoadFromClip(AVSValue clip) {
// Check if it has audio
VideoInfo vi = clip.AsClip()->GetVideoInfo();
if (!vi.HasAudio()) throw agi::AudioDataNotFound("No audio found.");

IScriptEnvironment *env = avs_wrapper.GetEnv();
AVSValue script;

// Convert to one channel
AVSValue script = env->Invoke(OPT_GET("Audio/Downmixer")->GetString().c_str(), clip);
std::string downmixtype = OPT_GET("Audio/Downmixer")->GetString();
if (downmixtype == "ConvertToMono" || downmixtype == "GetLeftChannel" || downmixtype == "GetRightChannel")
script = env->Invoke(downmixtype.c_str(), clip);
else
script = clip;

// Convert to 16 bits per sample
script = env->Invoke("ConvertAudioTo16bit", script);
vi = script.AsClip()->GetVideoInfo();

// Convert sample rate
Expand All @@ -132,8 +138,8 @@ void AvisynthAudioProvider::LoadFromClip(AVSValue clip) {
channels = vi.AudioChannels();
decoded_samples = num_samples = vi.num_audio_samples;
sample_rate = vi.SamplesPerSecond();
bytes_per_sample = vi.BytesPerAudioSample();
float_samples = false;
bytes_per_sample = vi.BytesPerChannelSample();
float_samples = vi.IsSampleType(SAMPLE_FLOAT);

this->clip = tempclip;
}
Expand Down
42 changes: 36 additions & 6 deletions src/avisynth_wrap.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -40,10 +40,24 @@

#include <mutex>

#ifndef _WIN32
#include <dlfcn.h>
#endif

#ifdef _WIN32
#define AVISYNTH_SO "avisynth.dll"
#else
#define AVISYNTH_SO "libavisynth.so"
#endif

// Allocate storage for and initialise static members
namespace {
int avs_refcount = 0;
#ifdef _WIN32
HINSTANCE hLib = nullptr;
#else
void* hLib = nullptr;
#endif
IScriptEnvironment *env = nullptr;
std::mutex AviSynthMutex;
}
Expand All @@ -54,14 +68,26 @@ typedef IScriptEnvironment* __stdcall FUNC(int);

AviSynthWrapper::AviSynthWrapper() {
if (!avs_refcount++) {
hLib = LoadLibrary(L"avisynth.dll");
#ifdef _WIN32
#define CONCATENATE(x, y) x ## y
#define _Lstr(x) CONCATENATE(L, x)
hLib = LoadLibraryW(_Lstr(AVISYNTH_SO));
#undef _Lstr
#undef CONCATENATE
#else
hLib = dlopen(AVISYNTH_SO, RTLD_LAZY | RTLD_LOCAL | RTLD_DEEPBIND);
#endif

if (!hLib)
throw AvisynthError("Could not load avisynth.dll");
throw AvisynthError("Could not load " AVISYNTH_SO);

FUNC *CreateScriptEnv = (FUNC*)GetProcAddress(hLib, "CreateScriptEnvironment");
#ifdef _WIN32
FUNC* CreateScriptEnv = (FUNC*)GetProcAddress(hLib, "CreateScriptEnvironment");
#else
FUNC* CreateScriptEnv = (FUNC*)dlsym(hLib, "CreateScriptEnvironment");
#endif
if (!CreateScriptEnv)
throw AvisynthError("Failed to get address of CreateScriptEnv from avisynth.dll");
throw AvisynthError("Failed to get address of CreateScriptEnv from " AVISYNTH_SO);

env = CreateScriptEnv(AVISYNTH_INTERFACE_VERSION);

Expand All @@ -79,9 +105,13 @@ AviSynthWrapper::AviSynthWrapper() {

AviSynthWrapper::~AviSynthWrapper() {
if (!--avs_refcount) {
delete env;
AVS_linkage = nullptr;
env->DeleteScriptEnvironment();
#ifdef _WIN32
FreeLibrary(hLib);
#else
dlclose(hLib);
#endif
AVS_linkage = nullptr;
}
}

Expand Down
2 changes: 1 addition & 1 deletion src/libresrc/default_config.json
Original file line number Diff line number Diff line change
Expand Up @@ -339,7 +339,7 @@
}
},
"Avisynth" : {
"Memory Max" : 128
"Memory Max" : 1024
},
"FFmpegSource" : {
"Cache" : {
Expand Down
2 changes: 1 addition & 1 deletion src/libresrc/osx/default_config.json
Original file line number Diff line number Diff line change
Expand Up @@ -338,7 +338,7 @@
}
},
"Avisynth" : {
"Memory Max" : 128
"Memory Max" : 1024
},
"FFmpegSource" : {
"Cache" : {
Expand Down
7 changes: 4 additions & 3 deletions src/meson.build
Original file line number Diff line number Diff line change
Expand Up @@ -174,11 +174,8 @@ if host_machine.system() == 'darwin'
)
elif host_machine.system() == 'windows'
aegisub_src += files(
'avisynth_wrap.cpp',
'font_file_lister_gdi.cpp',
# 'libass_gdi_fontselect.cpp',
'audio_provider_avs.cpp',
'video_provider_avs.cpp',
)

if cc.has_header('wingdi.h')
Expand Down Expand Up @@ -241,6 +238,10 @@ opt_src = [
'video_provider_ffmpegsource.cpp',
'ffmpegsource_common.cpp']],

['AviSynth', ['avisynth_wrap.cpp',
'audio_provider_avs.cpp',
'video_provider_avs.cpp']],

['Hunspell', 'spellchecker_hunspell.cpp'],
]

Expand Down
4 changes: 2 additions & 2 deletions src/preferences.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -399,8 +399,8 @@ void Advanced_Audio(wxTreebook *book, Preferences *parent) {

#ifdef WITH_AVISYNTH
auto avisynth = p->PageSizer("Avisynth");
const wxString adm_arr[3] = { "ConvertToMono", "GetLeftChannel", "GetRightChannel" };
wxArrayString adm_choice(3, adm_arr);
const wxString adm_arr[4] = { "None", "ConvertToMono", "GetLeftChannel", "GetRightChannel" };
wxArrayString adm_choice(4, adm_arr);
p->OptionChoice(avisynth, _("Avisynth down-mixer"), adm_choice, "Audio/Downmixer");
p->OptionAdd(avisynth, _("Force sample rate"), "Provider/Audio/AVS/Sample Rate");
#endif
Expand Down
5 changes: 4 additions & 1 deletion src/video_provider_avs.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ class AvisynthVideoProvider: public VideoProvider {
bool HasAudio() const override { return has_audio; }
};

AvisynthVideoProvider::AvisynthVideoProvider(agi::fs::path const& filename, std::string const& colormatrix) {
AvisynthVideoProvider::AvisynthVideoProvider(agi::fs::path const& filename, std::string const& colormatrix) try {
agi::acs::CheckFileRead(filename);

std::lock_guard<std::mutex> lock(avs.GetMutex());
Expand Down Expand Up @@ -182,6 +182,9 @@ AvisynthVideoProvider::AvisynthVideoProvider(agi::fs::path const& filename, std:
throw VideoOpenError("Avisynth error: " + std::string(err.msg));
}
}
catch (AvisynthError const& err) {
throw VideoProviderError("Avisynth error: " + std::string(err.msg));
}

void AvisynthVideoProvider::Init(std::string const& colormatrix) {
auto script = source_clip;
Expand Down

0 comments on commit 53466be

Please sign in to comment.