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

Go to file

kageru 146743d2c7 Use fixed array for lut generation The old version that was generated at runtime didn’t actually contain 1.0 as a value, so getting full white output for int input was impossible.		2020-09-11 15:51:41 +02:00
benches	No longer use LUT for 32bit input	2020-05-26 18:12:39 +02:00
src	Use fixed array for lut generation	2020-09-11 15:51:41 +02:00
.gitignore	Add Cargo.lock	2020-02-26 22:35:19 +01:00
build_release.sh	update build script	2020-06-22 13:52:30 +02:00
Cargo.lock	Use fixed array for lut generation	2020-09-11 15:51:41 +02:00
Cargo.toml	Use fixed array for lut generation	2020-09-11 15:51:41 +02:00
README.md	Add an explanation for the two dlls to the readme	2020-06-24 21:17:39 +02: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

What’s the no-fma dll? Which one do I need?

There are two Windows builds of the plugin, one for CPUs that support FMA instructions and one for those that don’t.
If your CPU is a Haswell (for Intel) or Piledriver (for AMD) or newer, you can use the regular version (which is about 20% faster). Otherwise, grab no-fma.
The Linux build uses fma instructions. I trust that if you’re a Linux user on older hardware, you know how to compile your own binaries.

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 I didn’t want to reimplement grain when we already have a working grain filter.