Reimplementation of the masking logic of adaptivegrain in rust. It’s almost safe :^)
Go to file
kageru 1c062e6dd0
Add LICENSE file
Thanks tdeo for pointing this out
2021-05-29 07:55:38 +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 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
LICENSE Add LICENSE file 2021-05-29 07:55:38 +02:00 Add an explanation for the two dlls to the readme 2020-06-24 21:17:39 +02:00


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


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).


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.


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.