Welcome to Doom9's Forum, THE in-place to be for everyone interested in DVD conversion.

Before you start posting please read the forum rules. By posting to this forum you agree to abide by the rules.

 

Go Back   Doom9's Forum > Capturing and Editing Video > VapourSynth

Reply
 
Thread Tools Search this Thread Display Modes
Old 4th October 2015, 17:28   #1  |  Link
Overdrive80
Anime addict
 
Overdrive80's Avatar
 
Join Date: Feb 2009
Location: Spain
Posts: 646
Hysteria - Line Darkening

Port of script of Scintilla, all merits to him.

Code:
#####################################################
#                                                   #
# Hysteria, a line darkening script by Scintilla    #
# Last updated 9/11/10                              #
#                                                   #
#####################################################
#
# Syntax:
# Hysteria(clip, strength= 1.0, usemask=True, lowthresh=6, highthresh=20, luma_cap=191, maxchg=255, minchg=0, planes = [0], luma=True, showmask=False)
#
# Requires YV12 input, frame-based only.  Is reasonably fast.
# Suggestions for improvement welcome: scintilla@aquilinestudios.org
#
# Required plugins:
# MaskTools 2.0 (MT_MaskTools)
#
# Arguments:
#
# strength (default=1.0) - This is a multiplicative factor for the amounts
#	by which the pixels are darkened.  Ordinarily, each pixel is
#	darkened by the difference between its luma value and the average
#	luma value of its brighter spatial neighbours.  So if you want more
#	darkening, increase this value.
#
# usemask (default=true) - Whether or not to apply the mask.  If false,
#	the entire image will have its edges darkened instead of just the
#	edges detected in the mask.  Could be useful on some sources
#	(specifically, it will often make dark lines look thicker), but
#	you will probably want to stick to lower values of strength if you
#	choose to go that route.
#
# lowthresh (default=6) - This is the threshold used for the noisy mask.
#	Increase this value if your mask is picking up too much noise
#	around the edges, or decrease it if the mask is not being grown
#	thick enough to cover all of each edge.
#
# highthresh (default=20) - This is the threshold used for the clean mask.
#	Increase this value if your mask is picking up too many weaker
#	edges, or decrease it if the mask is not picking up enough.
#
# luma_cap (default=191) - An idea swiped from FLD/VMToon.  Any pixels
#	brighter than this value will not be darkened at all, no matter
#	what the rest of the parameters are.  This is useful if you have
#	lighter edges that you do not want darkened.  0 will result in
#	no darkening at all, while 255 will turn off the cap.
#
# maxchg (default=255) - No pixel will be darkened by more than this
#	amount, no matter how high you set the strength parameter.
#	This can be useful if you want to darken your weaker lines more
#	without going overboard on the stronger ones.  0 will result in
#	no darkening at all, while 255 (the default) will turn off the
#	limiting.
#
# minchg (default=0) - Another idea swiped from FLD/VMToon (though in
#	those functions it was called "threshold").  Any pixels that
#	would have been darkened by less than this amount will instead
#	not be darkened at all.  This can be useful if you have noise
#	that is getting darkened slightly.  0 (the default) will turn
#	off the thresholding, while 255 will result in no darkening at all.
# 
# planes (default=0) - Luma plane
#
# luma (default=True) - Use luma plane for masking
#    
# showmask (default=false) - When true, the function will display the
#	current edge mask plus the chroma from the original image.
#	Use this to find the optimal values of lowthresh and highthresh.
#
###################
#
# Changelog:
#
# 9/11/10: Is this thing on?
#
###################


import vapoursynth as vs

def scale(old_value, new_bd=16):
    return int((old_value * ((1 << new_bd) - 1)) / 255)
	
def hysteria(clip, strength= 1.0, usemask=True, lowthresh=6, highthresh=20, luma_cap=191, maxchg=255, minchg=0, planes = [0], luma=True, showmask=False):
				
    core = vs.get_core()
	
    if not isinstance(clip, vs.VideoNode):
        raise ValueError('This is not a clip')
	
    # This scales the values parameters of Levels
    if clip.format.bits_per_sample != 8:
        max_in = (1 << clip.format.bits_per_sample) - 1
        max_out = (1 << clip.format.bits_per_sample) - 1
        min_out = scale(80, clip.format.bits_per_sample)
		
    # Medium value
    mid =  (2 ** clip.format.bits_per_sample) // 2

		
    noisymask = core.std.Sobel(clip, min=lowthresh, max=lowthresh, planes=planes, rshift=0)
    cleanmask = core.std.Sobel(clip, min=highthresh, max=highthresh, planes=planes, rshift=0)
	
    themask = core.generic.Hysteresis(cleanmask,noisymask)
    themask = core.std.Inflate(themask)
    themask = core.generic.Blur(themask, ratio_h=1.0)
    themask = core.generic.Blur(themask, ratio_h=1.0)
    themask = core.std.Deflate(themask)
	
    clipa = core.std.Inflate(clip, planes=[0])
    diffs = core.std.MakeDiff(clipa,clip)
    diffs = core.std.Expr([diffs], ['x {mid} - {strength} *'.format(strength=strength, mid=mid)])

    darkened = core.std.Expr([clip, diffs], ['x x {luma_cap} > 0 y {maxchg} > {maxchg} y {minchg} < 0 y ? ? ? -'.format(luma_cap=luma_cap,maxchg=maxchg,minchg=minchg)])

    if usemask:
        themask = core.std.ShufflePlanes(themask, planes=[0], colorfamily=vs.GRAY)
        final = core.std.MaskedMerge(clip,darkened,themask, planes=planes, first_plane=luma)
    else:
        final = core.std.ShufflePlanes(clips=[darkened,clip], planes=[0, 1, 2], colorfamily=vs.YUV)

    if showmask:
        mascara = core.std.Levels(themask, min_in=0, max_in=max_in, gamma=1.0, min_out=min_out, max_out=max_out)
        mascara = core.std.ShufflePlanes([mascara,clip], planes=[0,1,2], colorfamily=vs.YUV)
        return mascara

    return final
EDIT: Thanks to Are_ for his suggestions.
__________________
Intel i7-6700K + Noctua NH-D15 + Z170A XPower G. Titanium + Kingston HyperX Savage DDR4 2x8GB + Nvidia GTX750 2GB DDR5 + SSD Vertex 4 256 GB + Antec EDG750 80 Plus Gold Mod + Corsair 780T Graphite

Last edited by Overdrive80; 5th October 2015 at 19:25.
Overdrive80 is offline   Reply With Quote
Old 5th October 2015, 17:50   #2  |  Link
mastrboy
Registered User
 
Join Date: Sep 2008
Posts: 287
Thanks, I often use this in Avisynth. Will be useful when I decide to fully convert to Vapoursynth.
__________________
(i have a tendency to drunk post)
mastrboy is offline   Reply With Quote
Old 13th June 2017, 21:23   #3  |  Link
BluBb_mADe
Registered User
 
Join Date: Apr 2013
Posts: 1
fix and improvement

Since everything that's necessary is now part of the VapourSynth std and I found problems with the parameter scaling and the masking,
I decided to post a fixed version without any dependencies.

Code:
#####################################################
#                                                   #
# Hysteria, a line darkening script by Scintilla    #
# Last updated 6/13/17                              #
#                                                   #
#####################################################
#
# Syntax:
# Hysteria(clip, strength= 1.0, usemask=True, lowthresh=6, highthresh=20, luma_cap=191, maxchg=255, minchg=0, planes = [0], luma=True, showmask=False)
#
# Requires YV12 input, frame-based only.  Is reasonably fast.
# Suggestions for improvement welcome: scintilla@aquilinestudios.org
#
#
# Arguments:
#
# strength (default=1.0) - This is a multiplicative factor for the amounts
#    by which the pixels are darkened.  Ordinarily, each pixel is
#    darkened by the difference between its luma value and the average
#    luma value of its brighter spatial neighbours.  So if you want more
#    darkening, increase this value.
#
# usemask (default=True) - Whether or not to apply the mask.  If False,
#    the entire image will have its edges darkened instead of just the
#    edges detected in the mask.  Could be useful on some sources
#    (specifically, it will often make dark lines look thicker), but
#    you will probably want to stick to lower values of strength if you
#    choose to go that route.
#
# lowthresh (default=6) - This is the threshold used for the noisy mask.
#    Increase this value if your mask is picking up too much noise
#    around the edges, or decrease it if the mask is not being grown
#    thick enough to cover all of each edge.
#
# highthresh (default=20) - This is the threshold used for the clean mask.
#    Increase this value if your mask is picking up too many weaker
#    edges, or decrease it if the mask is not picking up enough.
#
# luma_cap (default=191) - An idea swiped from FLD/VMToon.  Any pixels
#    brighter than this value will not be darkened at all, no matter
#    what the rest of the parameters are.  This is useful if you have
#    lighter edges that you do not want darkened.  0 will result in
#    no darkening at all, while 255 will turn off the cap.
#
# maxchg (default=255) - No pixel will be darkened by more than this
#    amount, no matter how high you set the strength parameter.
#    This can be useful if you want to darken your weaker lines more
#    without going overboard on the stronger ones.  0 will result in
#    no darkening at all, while 255 (the default) will turn off the
#    limiting.
#
# minchg (default=0) - Another idea swiped from FLD/VMToon (though in
#    those functions it was called "threshold").  Any pixels that
#    would have been darkened by less than this amount will instead
#    not be darkened at all.  This can be useful if you have noise
#    that is getting darkened slightly.  0 (the default) will turn
#    off the thresholding, while 255 will result in no darkening at all.
# 
# planes (default=0) - Luma plane
#
# luma (default=True) - Use luma plane for masking
#    
# showmask (default=False) - When True, the function will display the
#    current edge mask plus the chroma from the original image.
#    Use this to find the optimal values of lowthresh and highthresh.
#
###################
#
# Changelog:
#
# 9/11/10: Is this thing on?
# 10/4/15: Port to VapourSynth by Overdrive80
# 6/13/17: removed dependency, fixed parameter scaling and mask mode is now "cartoon" by BluBb_mADe
#
###################


import vapoursynth as vs


def Hysteria(clip, strength= 1.0, usemask=True, lowthresh=6, highthresh=20, luma_cap=191, maxchg=255, minchg=0, planes = [0], luma=True, showmask=False):

    core = vs.get_core()
    if not isinstance(clip, vs.VideoNode):
        raise ValueError('This is not a clip')

    max_bitval = (1<<clip.format.bits_per_sample) - 1
    
    def scale(old_value):
        return int((old_value * (max_bitval)) / 255)
    
    # This scales the colordepth dependant parameters
    if clip.format.bits_per_sample != 8:
        lowthresh = scale(lowthresh)
        highthresh = scale(highthresh)
        
        luma_cap = scale(luma_cap)
        
        maxchg = scale(maxchg)
        minchg = scale(minchg)


    # Medium value
    mid =  (2 ** clip.format.bits_per_sample) // 2
    
    # imitate mt_edge(mode=cartoon) (stolen from Frechdachs)
    noisymask = core.std.Convolution(clip,matrix=[0,-2,1,0,1,0,0,0,0], planes=planes, saturate=True)
    noisymask = core.std.Expr(noisymask, ['x {high} >= {maxvalue} x {low} <= 0 x ? ?'
                                           .format(low=lowthresh, high=lowthresh, maxvalue=max_bitval), ''])
    
    cleanmask = core.std.Convolution(clip,matrix=[0,-2,1,0,1,0,0,0,0], planes=planes, saturate=True)
    cleanmask = core.std.Expr(cleanmask, ['x {high} >= {maxvalue} x {low} <= 0 x ? ?'
                                           .format(low=highthresh, high=highthresh, maxvalue=max_bitval), ''])

    themask = core.misc.Hysteresis(cleanmask,noisymask)
    themask = core.std.Inflate(themask)
    
    # blur replacement
    themask = core.std.Convolution(themask,matrix=[1, 2, 1, 2, 4, 2, 1, 2, 1])
    themask = core.std.Convolution(themask,matrix=[1, 2, 1, 2, 4, 2, 1, 2, 1])
    
    themask = core.std.Deflate(themask)

    clipa = core.std.Inflate(clip, planes=[0])
    diffs = core.std.MakeDiff(clipa,clip)
    diffs = core.std.Expr([diffs], ['x {mid} - {strength} *'.format(strength=strength, mid=mid)])

    darkened = core.std.Expr([clip, diffs], ['x x {luma_cap} > 0 y {maxchg} > {maxchg} y {minchg} < 0 y ? ? ? -'.format(luma_cap=luma_cap,maxchg=maxchg,minchg=minchg)])

    if usemask:
        themask = core.std.ShufflePlanes(themask, planes=[0], colorfamily=vs.GRAY)
        final = core.std.MaskedMerge(clip,darkened,themask, planes=planes, first_plane=luma)
    else:
        final = core.std.ShufflePlanes(clips=[darkened,clip], planes=[0, 1, 2], colorfamily=vs.YUV)

    if not showmask:
        return final
    
    mascara = core.std.Levels(themask, min_in=0, max_in=max_bitval, min_out=scale(80), max_out=max_bitval)
    mascara = core.std.ShufflePlanes([mascara,clip], planes=[0,1,2], colorfamily=vs.YUV)
    return mascara
Edit: I posted the script on GitHub. I am not sure about any license yet.

Last edited by BluBb_mADe; 14th June 2017 at 01:11.
BluBb_mADe is offline   Reply With Quote
Reply

Thread Tools Search this Thread
Search this Thread:

Advanced Search
Display Modes

Posting Rules
You may not post new threads
You may not post replies
You may not post attachments
You may not edit your posts

BB code is On
Smilies are On
[IMG] code is On
HTML code is Off

Forum Jump


All times are GMT +1. The time now is 06:55.


Powered by vBulletin® Version 3.8.11
Copyright ©2000 - 2019, vBulletin Solutions Inc.