KSMDegrain.avs

function KSMDegrain_MAnalyze( clip srchSuper, clip srchSuperP, bool isb, int delta, int blksize, int overlap, \
					int search, bool truemotion, bool chroma, int batch)
{
	# partial mv on CPU
	pmv = srchSuperP.KMAnalyse(blksize=blksize, overlap=overlap, search=search, isb=isb, chroma=chroma, \
					delta=delta, meander=false, batch=1).OnCPU(2)

	# main mv on CUDA
	return srchSuper.KMAnalyse(blksize=blksize, overlap=overlap, search=search, isb=isb, chroma=chroma, \
					delta=delta, meander=false, partial=pmv, batch=batch)
}

function KSMDegrain(clip c, bool "binomial")
{
    super = c.KTGMC_ToFullRange(U=3,V=3).KMSuper(pel=1,chroma=True)
    psuper = super.KMPartialSuper().OnCUDA(2)
    dsuper = c.KMSuper(pel=1,chroma=True,levels=1)
    bvec2 = super.KSMDegrain_MAnalyze(psuper, True,2,16,8,4,False,True,4)
    bvec1 = super.KSMDegrain_MAnalyze(psuper, True,1,16,8,4,False,True,4)
    fvec1 = super.KSMDegrain_MAnalyze(psuper, False,1,16,8,4,False,True,4)
    fvec2 = super.KSMDegrain_MAnalyze(psuper, False,2,16,8,4,False,True,4)
    return c.KMDegrain2(dsuper, bvec1, fvec1, bvec2, fvec2,thSAD=300,thSADC=150,thSCD1=1600,thSCD2=130,limit=255,limitc=255, binomial=binomial)
}