Reimplementation of the masking logic of adaptivegrain in rust. It’s almost safe :^)

Go to file

kageru 8ef0654595 No longer use LUT for 32bit input That means you’ll finally get full 32bit precision, not just 8bit. This comes at a cost (~20% lower FPS), but it’s a lot more accurate, and there’s potentially room for optimization in the future because we actually do math now instead of just hacky lookups in a 256-element vector.		2020-05-26 18:12:39 +02:00
benches	No longer use LUT for 32bit input	2020-05-26 18:12:39 +02:00
src	No longer use LUT for 32bit input	2020-05-26 18:12:39 +02:00
.gitignore	Add Cargo.lock	2020-02-26 22:35:19 +01:00
build_release.sh	add script to build release binaries	2020-04-13 13:10:23 +02:00
Cargo.lock	Add benchmarks	2020-05-24 20:46:23 +02:00
Cargo.toml	Add benchmarks	2020-05-24 20:46:23 +02:00
README.md	Link AUR package in README	2020-02-26 23:09:43 +01:00

README.md

Adaptivegrain-rs

Reimplementation of the adaptive_grain mask as a Vapoursynth plugin. For a description of the math and the general idea, see the article.

Usage

core.adg.Mask(clip, luma_scaling: float)

You must call std.PlaneStats() before this plugin (or fill the PlaneStatsAverage frame property using some other method). Supported formats are YUV with 8-32 bit precision integer or single precision float. Half precision float input is not supported since no one seems to be using that anyway. Since the output is grey and only luma is processed, the subsampling of the input does not matter.

To replicate the original behaviour of adaptivegrain, a wrapper is provided in kagefunc. It behaves exactly like the original implementation (except for the performance, which is about 3x faster on my machine).

Parameters

clip: vapoursynth.VideoNode

the input clip to generate a mask for.

luma_scaling: float = 10.0

the luma_scaling factor as described in the blog post. Lower values will make the mask brighter overall.

Build instructions

If you’re on Arch Linux, there’s an AUR package for this plugin. Otherwise you’ll have to build and install the package manually.

cargo build --release

That’s it. This is Rust, after all. No idea what the minimum version is, but it works with stable rust 1.41. That’s all I know. Binaries for Windows and Linux are in the release tab.

FAQ

Why do I have to call std.PlaneStats() manually?

~~Because I didn’t want to reimplement it. kagefunc.adaptive_grain(clip, show_mask=True) does that for you and then just returns the mask.~~ Because I was too dumb to realize this exists. I’ll fix that at some point.

Why doesn’t this also add grain?

I was going to do that originally, but it just goes back to the same point about not wanting to reimplement something that already exists.