API tools faq
paste
Advertisement
mawen1250

Contra-Sharpen mod 3.2

mawen1250
Sep 18th, 2012
130
0
Never
Add comment
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Avisynth 24.20 KB | None | 0 0
raw download clone embed print report
  1. ###### Contra-Sharpen mod 3.2 ###### by mawen1250 ###### 2012.09.18 ######
  2. ###### Requirements: masktools v2.0a48, mvtools v2.5.11.3/v2.6.0.5, RemoveGrain + Repair v1.0pre, RemoveGrainHD v0.5 ######
  3. ###### FluxSmooth v1.1a, nnedi3 v0.9.4, SangNom 2004.01.18(filter_ss="AA"), TEdgeMask v0.9, tcanny v1.0, MSharpen v1.10 ######
  4.  
  5. Function CSmod(clip filtered, clip "source", clip "pclip", bool "chroma",
  6. \ int "edgemode", val "edgemask", float "edgethr", float "tcannysigma",
  7. \ float "ss_w", float "ss_h", bool "ss_hq", string "ssmethod", string "filter_ss", bool "ssrep", bool "ssout", bool "ssoutc",
  8. \ int "preblur", bool "prec", int "preR", bool "usepasf",
  9. \ int "Smethod", val "kernel", bool "secure",
  10. \ int "Smode", float "strength", float "divisor", float "index", float "Szrp", float "Spwr", float "SdmpLo", float "SdmpHi",
  11. \ bool "limit", int "Repmode", int "RepmodeU", int "thr", bool "choversh",
  12. \ bool "Slimit", bool "SlimitY", bool "Tlimit", int "Sovershoot", int "Sundershoot", int "Tovershoot", int "Tundershoot",
  13. \ float "Soft", int "Soothe",
  14. \ int "blksize", int "overlap", int "thSAD", int "thSCD1", int "thSCD2",
  15. \ bool "truemotion", bool "MVglobal", int "pel", int "pelsearch", int "search", int "searchparam", int "MVsharp", int "DCT")
  16. {
  17.  
  18. defsrc   = Defined(source   )
  19. defpclp  = Defined(pclip    )
  20. deffss   = Defined(filter_ss)
  21. AA       = deffss ? filter_ss=="AA" : false
  22. sw       = filtered.width()
  23. sh       = filtered.height()
  24. HD       = (sw > 1024 || sh > 576) ? true : false
  25.  
  26. chroma   = Default(chroma,  false  )
  27.  
  28. edgemode    = Default(edgemode,    AA ? 1 : 0  )
  29. # 0 = Sharpening all, 1 = Sharpening only edge, 2 = Sharpening only non-edge.
  30. # By default, edgemode=2 is tuned for enhancing noise/small detail.
  31. # It will automatically set [edgemask=-2, ss_w=1, ss_h=1, preblur=0, kernel=5, Slimit=false, Tlimit=true].
  32. edgemask    = Default(edgemask,    edgemode==2?-2:AA?7:2)
  33. # -1: Similar to 1 and tweaking for edgemode=2,   -2: Similar to 2 and tweaking for edgemode=2,
  34. #  1: Same as edgemaskHQ=false in LSFmod(min/max), 2: Same as edgemaskHQ=true in LSFmod,
  35. #  3: Same as sharpening mask in MCTD(prewitt),    4: Same as mtype=1 in TAA(sobel),
  36. #  5: Same as mtype=2 in TAA(roberts),             6: Same as mtype=3 in TAA(prewitt),
  37. #  7: Same as mtype=4 in TAA(TEdgeMask),           8: Same as mtype=5 in TAA(tcanny),
  38. #  9: Same as mtype=6 in TAA(MSharpen),           10: (or any other numbers) My own method of tcanny usage.
  39. #     Otherwise define a custom edge mask clip, only luma is taken to merge all planes.
  40. edgethr     = Default(edgethr,     32.0        )   #Tweak edge mask thr in edgemask mode -2/-1/1/2/6/7/8/9/10.
  41. tcannysigma = Default(tcannysigma, 1.2         )   #Tweak tcanny's sigma in edgemask mode 8/10.
  42.  
  43. ss_w     = Default(ss_w,    edgemode>=2?1.00:1.50)   #Super sampling multiplier of width.
  44. ss_h     = Default(ss_h,    edgemode>=2?1.00:1.50)   #Super sampling multiplier of height.
  45. ss_w     = max    (ss_w,    1.00   )
  46. ss_h     = max    (ss_h,    1.00   )
  47. ss_hq    = Default(ss_hq,   ss_w*ss_h>=3.0625)   #True using nnedi3_rpow2 in super sampling, false using non-ringing Spline64Resize.
  48. ssout    = Default(ssout,   false  )   #Output in super sampling resolution.
  49. ssoutc   = Default(ssoutc,  true   )   #Whether to output correct chroma when (chroma==false && ss_hq==false && ssout==true).
  50. ssrep    = Default(ssrep,   false  )   #When limiting sharpening to source clip, whether to Repair in super sampling resolution.
  51. ssout    = (ss_w > 1.0 || ss_h > 1.0) ? ssout : false
  52. ssrep    = (ss_w > 1.0 || ss_h > 1.0) ? ssrep : false
  53. ssoutc   = (!chroma && ssout) ? ssoutc : false
  54. ssrep    = ssout ? true : ssrep
  55. filter_ss= deffss ? AA ? "SangNom(aa=48).TurnRight().SangNom(aa=48).TurnLeft()" : filter_ss : NOP()
  56. # When defining filter_ss="AA", it will automatically use SangNom for anti-aliasing in super sampling clip, and it will automatically set [Repmode=13, edgemode=1, edgemask=7].
  57.  
  58. preblur  = Default(preblur, edgemode>=2?0:ss_w*ss_h==1?1:0)
  59. # Pre-filtering, 0 = disable, 1 = MinBlur, 2 = MinBlur(Uses SBR by default) mixed with MinBlur+FluxSmoothT,
  60. # 3 = MinBlur(Uses SBR by default)+FluxSmoothT, ignored when pclip is defined.
  61. prec     = Default(prec,    preblur>=1)   #Whether to process chroma plane in preblur.
  62. preR     = Default(preR,    preblur>=2?0:ss_w*ss_h>6.25?3:ss_w*ss_h>2.25?2:1)
  63. # Preblur setting, 1-3 sets radius of MinBlur(Gaussian|Median), 0 uses SBR instead of normal Gaussian Blur in MinBlur.
  64. usepasf  = Default(usepasf, false  )
  65. # Whether to use pre-filtered clip as input(filtered) clip, which will reduce noise/halo/ring from input clip.
  66.  
  67. Smethod  = Default(Smethod, ss_w*ss_h>1?3:1)   #Sharpen Method - 1: 3x3 kernel, 2: Min/Max, 3: Min/Max + 3x3 kernel.
  68. kernel   = Default(kernel,  edgemode>=2 ? 5 : Smethod>=3 ? ss_w*ss_h>=9?4:ss_w*ss_h>=4?3:ss_w*ss_h>=2.25?2:1
  69. \                                                        : ss_w*ss_h>=4?4:ss_w*ss_h>=2.25?3:ss_w*ss_h> 1?2:1)
  70. # 1: Gaussian Blur radius=1(RG11), 2: Average Blur(RG20), 3: Gaussian Blur radius=2(RG11+RG20),
  71. # 4: Gaussian Blur radius=3(RG11+RG20+RG20), 5: for noise/detail enhance(RG19+RG4), default for edgemode==2,
  72. # Otherwise define a custom kernel in string such as kernel="RemoveGrain(20, 11)".
  73. secure   = Default(secure,  true   )   #Mode to avoid banding & oil painting (or face wax) effect of sharpening.(from LSFmod)
  74. Smode    = Default(Smode,   3      )   #Sharpen Mode - 1: Linear, 2: Non-linear 1, 3:Non-linear 2(from LSFmod Smode=5).
  75. divisor  = Default(divisor, 1.5    )
  76. index    = Default(index,   0.8    )
  77. strength = Default(strength,120.0  )
  78. Szrp     = Default(Szrp,    16     )
  79. Spwr     = Default(Spwr,    4      )
  80. SdmpLo   = Default(SdmpLo,  4      )
  81. SdmpHi   = Default(SdmpHi,  48     )
  82. miSpwr   = 1.0/Spwr
  83.  
  84. limit    = Default(limit,   true   )
  85. # Whether to limit sharpening to source clip, only takes effect when (Defined(source) || Defined(filter_ss) || usepasf) == true.
  86. thr      = Default(thr,     0      )   #When limiting sharpening to source clip, allow sharpening more than the source in [thr].
  87. choversh = Default(choversh,true   )
  88. # When (chroma==true && limit==true && thr>=1 && choversh==false), luma will sharpen more than source, chroma will not.
  89. Repmode  = Default(Repmode, AA?13:1)
  90. RepmodeU = Default(RepmodeU,Repmode)
  91. limit    = defsrc || deffss || usepasf ? limit : false
  92.  
  93. Slimit      = Default(Slimit,      edgemode>=2?false:!limit)
  94. # Spatial limit with overshoot and undershoot. Disabled when (limit==true && thr==0).
  95. SlimitY     = Default(SlimitY,     Slimit      )   #When Slimit=true, SlimitY controls whether to use Slimit on luma.
  96. Sovershoot  = Default(Sovershoot,  Round(strength/150))
  97. Sundershoot = Default(Sundershoot, Sovershoot  )
  98. Tlimit      = Default(Tlimit,      edgemode>=2?true:false)
  99. # Use MC Temporal limit at the end of sharpening(only process luma).(from MCTD)
  100. Tovershoot  = Default(Tovershoot,  Round(strength/60 ))
  101. Tundershoot = Default(Tundershoot, Tovershoot  )
  102. Soft        = Default(Soft,        limit?  0:-2)
  103. # Soft the sharpening effect (-1 = old autocalculate, -2 = new autocalculate, 0 = disable, (0, 100] = enable). Disabled when (limit==true && thr==0).(from LSFmod)
  104. Soothe      = Default(Soothe,      limit? -1:24)
  105. # Soothe temporal stabilization, 0-100 sets minimum percent of the original sharpening to keep, -1 disables Soothe. Disabled when (chroma==true && Tlimit=true).
  106. Sovershoot  = max(Sovershoot,  0)
  107. Sundershoot = max(Sundershoot, 0)
  108. Tovershoot  = max(Tovershoot,  0)
  109. Tundershoot = max(Tundershoot, 0)
  110.  
  111. blksize     = Default(blksize,     HD ? 16 : 8 )
  112. overlap     = Default(overlap,     blksize/2   )
  113. thSAD       = Default(thSAD,       300         )
  114. thSCD1      = Default(thSCD1,      300         )
  115. thSCD2      = Default(thSCD2,      100         )
  116. truemotion  = Default(truemotion,  false       )
  117. MVglobal    = Default(MVglobal,    true        )
  118. pel         = Default(pel,         2           )
  119. pelsearch   = Default(pelsearch,   2           )
  120. search      = Default(search,      2           )
  121. searchparam = Default(searchparam, 2           )
  122. MVsharp     = Default(MVsharp,     2           )
  123. DCT         = Default(DCT,         0           )
  124.  
  125.  
  126. wss      = Round(sw*ss_w/8)*8
  127. hss      = Round(sh*ss_h/8)*8
  128. rfactor  = max(Round(sqrt(ss_w*ss_h)/2)*2, 2)
  129. ssmethod = Defined(ssmethod) ? ssmethod
  130. \        : ss_hq ? """nnedi3_rpow2(rfactor=rfactor, qual=2, cshift="Spline64Resize", fwidth=wss, fheight=hss)"""
  131. \                : "CSmod_nrSpline64Resize(wss, hss, chroma=chroma||ssoutc)"
  132. # You can define your own super sampling method with ssmethod.
  133.  
  134. ch21     = chroma ? 2 : 1
  135. ch31     = chroma ? 3 : 1
  136. ch41     = chroma ? 4 : 1
  137. ch32     = chroma ? 3 : 2
  138. ch34     = chroma ? 3 : 4
  139. Slimit   = limit && thr==0 ? false : Slimit
  140. Soft     = limit && thr==0 ? 0
  141. \        : Soft <= -2  ? (1.0+(2.0/(ss_w+ss_h))) * sqrt(strength)
  142. \        : Soft == -1  ? sqrt( (((ss_w+ss_h)/2.0-1.0)*100.0) ) * 10
  143. \        : Soft <= 100 ? Soft : 100
  144. Soothe   = Tlimit ? -1 : Soothe
  145. strength = Smode == 1 ? strength/50.0 : Smode == 2 ? strength/20.0 : strength/100.0
  146.  
  147.  
  148. # super sampling with filtering
  149. source      = defsrc ? source : filtered
  150. filtered_os = filtered
  151. filtered_ss = (ss_w > 1.0 || ss_h > 1.0) ? Eval("filtered."+ssmethod) : filtered
  152. source_ss   = (ss_w > 1.0 || ss_h > 1.0) ? defsrc ? Eval("source."+ssmethod) : filtered_ss : source
  153. filtered    = (ss_w > 1.0 || ss_h > 1.0) && deffss ? Eval("filtered_ss."+filter_ss) : filtered_ss
  154. filtered_ds = (ss_w > 1.0 || ss_h > 1.0) && deffss ? filtered.CSmod_nrSpline64Resize(sw, sh, chroma=chroma) : filtered_os
  155.  
  156. Assert( (isFrameBased(filtered_os)),                            """CSmod: input clip "filtered" must be Frame Based!""" )
  157. Assert( (isFrameBased(source)),                                   """CSmod: input clip "source" must be Frame Based!""" )
  158. Assert( (source.width() == sw && source.height() == sh), """CSmod: resolution of "filtered" and "source" must match!""" )
  159.  
  160.  
  161. # pre-filtering before sharpening
  162. spatial     = filtered.MinBlur(preR, chroma ? (prec ? 3 : 2) : 1)
  163. temporal    = spatial .FluxsmoothT(7).Repair(spatial, 1, prec ? 1 : -1)
  164. temporal    = chroma&&!prec ? temporal.MergeChroma(filtered) : temporal
  165. mixed       = temporal.merge(spatial,0.251)
  166. pre         = defpclp    ? pclip
  167. \           : preblur==1 ? spatial
  168. \           : preblur==2 ? mixed
  169. \           : preblur>=3 ? temporal
  170. \           : filtered
  171. # You can define your own pre-filter clip with pclip.
  172. pre_ds      = (pre.width() == sw && pre.height() == sh) ? pre : pre.CSmod_nrSpline64Resize(sw, sh, chroma=chroma)
  173. pre         = (ss_w > 1.0 || ss_h > 1.0) ? (pre.width() == wss && pre.height() == hss) ? pre : Eval("pre."+ssmethod) : pre_ds
  174.  
  175. #whether to use pre-filtered clip as main clip
  176. filtered    = usepasf ? pre    : filtered
  177. filtered_ds = usepasf ? pre_ds : filtered_ds
  178.  
  179.  
  180. # generate edge mask
  181. edgeclip  = ssout ? pre : pre_ds
  182. edgeclip2 = ssout ? source_ss : source
  183. edgemask  = IsClip(edgemask) ? edgemask
  184. \         : IsInt (edgemask) ?
  185. \           edgemask <= -2   ? mt_logic(edgeclip.mt_edge(thY1=0, thY2=255, "8 16 8 0 0 0 -8 -16 -8 4", U=1, V=1),
  186. \                                       edgeclip.mt_edge(thY1=0, thY2=255, "8 0 -8 16 0 -16 8 0 -8 4", U=1, V=1),
  187. \                                       "max", U=1, V=1).mt_lut("x " + string(edgethr*4) + " / 0.86 ^ 255 *", U=1, V=1)
  188. \                             .Removegrain(11, -1)
  189. \         : edgemask == -1   ? edgeclip.mt_edge(thY1=0, thY2=255, mode="min/max", U=1, V=1)
  190. \                             .mt_lut("x " + string(edgethr) + " / 0.86 ^ 255 *", U=1, V=1)
  191. \                             .Removegrain(11, -1)
  192. \         : edgemask == 1    ? edgeclip.mt_edge(thY1=0, thY2=255, mode="min/max", U=1, V=1)
  193. \                             .mt_lut("x " + string(edgethr) + " / 0.86 ^ 255 *", U=1, V=1)
  194. \                             .mt_inflate(U=1, V=1).mt_inflate(U=1, V=1).Removegrain(11, -1)
  195. \         : edgemask == 2    ? mt_logic(edgeclip.mt_edge(thY1=0, thY2=255, "8 16 8 0 0 0 -8 -16 -8 4", U=1, V=1),
  196. \                                       edgeclip.mt_edge(thY1=0, thY2=255, "8 0 -8 16 0 -16 8 0 -8 4", U=1, V=1),
  197. \                                       "max", U=1, V=1).mt_lut("x " + string(edgethr*4) + " / 0.86 ^ 255 *", U=1, V=1)
  198. \                             .mt_inflate(U=1, V=1).mt_inflate(U=1, V=1).RemoveGrain(11, -1)
  199. \         : edgemask == 3    ? mt_edge(edgeclip, "prewitt", 0, 255, 0, 0, V=1, U=1)
  200. \                             .mt_lut("x 1.8 ^", U=1, V=1).RemoveGrain(4, -1).mt_inflate(U=1, V=1).RemoveGrain(20,-1)
  201. \         : edgemask == 4    ? mt_edge(edgeclip, "sobel", 7, 7, 5, 5, U=1, V=1).mt_inflate(U=1, V=1)
  202. \         : edgemask == 5    ? mt_edge(edgeclip, "roberts", 0, 4, 0, 4, U=1, V=1).mt_inflate(U=1, V=1)
  203. \         : edgemask == 6    ? mt_edge(edgeclip, "prewitt", 0, 255, 0, 0, U=1, V=1)
  204. \                             .mt_lut("x " + string(edgethr) + " <= x 1 >> x 1.4 ^ ?", U=1, V=1)
  205. \                             .RemoveGrain(4, -1).mt_inflate(U=1, V=1)
  206. \         : edgemask == 7    ? TEdgeMask(edgeclip, link=2, preblur=false, valon=-1, U=0, V=0)
  207. \                             .mt_lut("x " + string(edgethr/5) + " <= x 1 >> x 4 << ?", U=1, V=1)
  208. \                             .mt_deflate(U=1, V=1).RemoveGrain(HD ? 20 : 11, -1)
  209. \         : edgemask == 8    ? tcanny(edgeclip2, sigma=tcannysigma, mode=1, plane=1)
  210. \                             .mt_lut("x " + string(edgethr) + " <= x 1 >> x 1 << ?", U=1, V=1)
  211. \                             .RemoveGrain(HD ? 20 : 11, -1).mt_inflate(U=1, V=1)
  212. \         : edgemask == 9    ? MSharpen(edgeclip, threshold=Round(edgethr/5), strength=0, mask=true, highq=false)
  213. \         :                    tcanny(edgeclip2, sigma=tcannysigma, mode=1, plane=1)
  214. \                             .mt_lut("x " + string(edgethr) + " <= 0 x " + string(edgethr) + " - 6 << ?", U=1, V=1)
  215. \                             .RemoveGrain(HD ? 20 : 11, -1).mt_inflate(U=1, V=1)
  216. \         :                    Assert(false, """CSmod: "edgemask" should be int or clip!""")
  217.  
  218.  
  219. # unsharp
  220. dark_limit   = pre.mt_inpand(U=ch31, V=ch31)
  221. bright_limit = pre.mt_expand(U=ch31, V=ch31)
  222. minmaxavg    = mt_average(dark_limit, bright_limit, U=ch31, V=ch31)
  223.  
  224. method     = Smethod <= 1     ? pre : minmaxavg
  225.  
  226. method     = Smethod == 2     ? method
  227. \          : IsString(kernel) ? Eval("method."+kernel)
  228. \          : IsInt   (kernel) ?
  229. \            kernel <= 1      ? method.RemoveGrain(11, chroma?11:-1)
  230. \          : kernel == 2      ? method.RemoveGrain(20, chroma?20:-1)
  231. \          : kernel == 3      ? method.RemoveGrain(11, chroma?11:-1).RemoveGrain(20, chroma?20:-1)
  232. \          : kernel == 4      ? method.RemoveGrain(11, chroma?11:-1).RemoveGrain(20, chroma?20:-1).RemoveGrain(20, chroma?20:-1)
  233. \          :                    method.RemoveGrain(19, chroma?19:-1).RemoveGrain(4 , chroma?4 :-1)
  234. \          :                    Assert(false, """CSmod: "kernel" should be int or string!""")
  235.  
  236. method     = secure           ? method.mt_lutxy(pre, "x y < x 1 + x y > x 1 - x ? ?", U=ch31, V=ch31) : method
  237.  
  238.  
  239. # making difference clip for sharpening
  240. sharpdiff  = mt_makediff(pre, method, U=ch31,V=ch31)
  241.  
  242.  
  243. # sharpening diff generate mode
  244. sharpdiff  = Smode == 1 ? sharpdiff.mt_lut("x 128 - "+string(strength)+" * 128 +", U=ch31, V=ch31)
  245. \          : Smode == 2 ? sharpdiff.mt_lut("x 128 == x x 128 - "+String(divisor)+" / Abs "+String(index)+" ^ "+String(strength)+" * x 128 > 1 -1 ? * 128 + ?", U=ch31, V=ch31)
  246. \          : Smode == 3 ? sharpdiff.mt_lut("x 128 == x x 128 - abs "+string(Szrp)+" / "+string(miSpwr)+" ^ "+string(Szrp)+" * "+string(strength)+" * x 128 > 1 -1 ? * x 128 - 2 ^ "+string(Szrp)+" 2 ^ "+string(SdmpLo)+" + * x 128 - 2 ^ "+string(SdmpLo)+" + "+string(Szrp)+" 2 ^ * / * 1 "+string(SdmpHi)+" 0 == 0 "+string(Szrp)+" "+string(SdmpHi)+" / 4 ^ ? + 1 "+string(SdmpHi)+" 0 == 0 x 128 - abs "+string(SdmpHi)+" / 4 ^ ? + / * 128 + ?", U=ch31, V=ch31)
  247. \          : sharpdiff
  248. # x==128 ? x : (abs(x-128)/Szrp)^(miSpwr)*Szrp*strength*(x>128 ? 1 : -1) * ((x-128)^2*(Szrp^2+SdmpLo) / ((x-128)^2+SdmpLo)*Szrp^2) * ((1+ SdmpHi==0 ? 0 : (Szrp/SdmpHi)^4) / (1+ SdmpHi==0 ? 0 : (abs(x-128)/SdmpHi)^4)) + 128
  249.  
  250. # spatial limit
  251. DlimitF    = usepasf || (preblur<=0 && !defpclp) ?   dark_limit : filtered.mt_inpand(U=ch31, V=ch31)
  252. BlimitF    = usepasf || (preblur<=0 && !defpclp) ? bright_limit : filtered.mt_expand(U=ch31, V=ch31)
  253. sclp       = Slimit ? mt_adddiff(filtered, sharpdiff, Y=SlimitY?3:4, U=ch31, V=ch31) : NOP()
  254. sclp       = Slimit ? sclp.mt_clamp(BlimitF, DlimitF, Sovershoot, Sundershoot, Y=SlimitY?3:2, U=ch31, V=ch31) : NOP()
  255.  
  256.  
  257. # Soft
  258. sharpdiff  = Slimit ? mt_makediff(sclp, filtered, Y=SlimitY?3:2, U=ch31, V=ch31) : sharpdiff
  259. sharpdiff  = Soft == 0 ? sharpdiff : mt_lutxy(sharpdiff, sharpdiff.RemoveGrain(19, chroma ? 19 : -1), "x 128 - abs y 128 - abs > y "+string(Soft)+" * x "+string(100-Soft)+" * + 100 / x ?", U=ch31, V=ch31)
  260.  
  261.  
  262. # Soothe
  263. sharpdiff2 = (Soothe >= 0 && Soothe <= 100) ? sharpdiff.temporalsoften(1,255,chroma ? 255 : 0,32,2) : NOP()
  264. sharpdiff  = (Soothe >= 0 && Soothe <= 100) ? mt_lutxy(sharpdiff, sharpdiff2, "x 128 - y 128 - * 0 < x 128 - 100 / "+string(Soothe)+" * 128 + x 128 - abs y 128 - abs > x "+string(Soothe)+" * y 100 "+string(Soothe)+" - * + 100 / x ? ?", U=ch31, V=ch31) : sharpdiff
  265.  
  266.  
  267. # the difference achieved by filtering
  268. allD       = limit  ? ssrep && (ss_w > 1.0 || ss_h > 1.0) ? mt_makediff(source_ss, filtered, U=ch31, V=ch31)
  269. \                                                         : mt_makediff(source, filtered_ds, U=ch31, V=ch31) : NOP()
  270.  
  271. # limiting sharpening to source clip
  272. sharpdiff  = !ssrep && (ss_w > 1.0 || ss_h > 1.0) ? sharpdiff.CSmod_nrSpline64Resize(sw, sh, chroma=chroma) : sharpdiff
  273. ssDD       = limit  ? sharpdiff.Repair(allD, mode=Repmode, modeU=chroma ? RepmodeU : -1) : sharpdiff
  274. ssDD       = limit && thr >= 1 ? sharpdiff.mt_lutxy(ssDD,"x 128 - abs y 128 - abs " + string(thr) + " + <= x y 128 < y " + string(thr) + " - y " + string(thr) + " + ? ?", U=choversh?ch31:ch41, V=choversh?ch31:ch41) : ssDD
  275. ssDD       = limit && thr == 0 || (limit && !choversh) ? ssDD.mt_lutxy(sharpdiff, "x 128 - abs y 128 - abs < x y ?", Y=(limit && thr >= 1 && !choversh)?2:3, U=ch31, V=ch31) : ssDD
  276. ssDD       = ssrep && (ss_w > 1.0 || ss_h > 1.0) && !ssout ? ssDD.CSmod_nrSpline64Resize(sw, sh, chroma=chroma) : ssDD
  277.  
  278. # add difference clip to clip "filtered" of ss/original resolution
  279. sclp       = ssout ? mt_adddiff(filtered, ssDD, U=ch31, V=ch31) : mt_adddiff(filtered_ds, ssDD, U=ch31, V=ch31)
  280.  
  281.  
  282. # temporal limit
  283. pre_ds     = ssout  ? pre       : pre_ds
  284. source     = ssout  ? source_ss : source
  285.  
  286. pMVS       = Tlimit ? pre_ds.MSuper(hpad=0, vpad=0, pel=pel, sharp=MVsharp, chroma=false) : NOP()
  287. f1v        = Tlimit ? MAnalyse(pMVS, isb=false, delta=1, truemotion=truemotion, blksize=blksize, overlap=overlap, pelsearch=pelsearch, search=search, searchparam=searchparam, DCT=DCT, global=MVglobal, chroma=false) : NOP()
  288. b1v        = Tlimit ? MAnalyse(pMVS, isb=true,  delta=1, truemotion=truemotion, blksize=blksize, overlap=overlap, pelsearch=pelsearch, search=search, searchparam=searchparam, DCT=DCT, global=MVglobal, chroma=false) : NOP()
  289. f1c        = Tlimit ? MCompensate(source, pMVS, f1v, thSAD=thSAD, thSCD1=thSCD1, thSCD2=thSCD2) : NOP()
  290. b1c        = Tlimit ? MCompensate(source, pMVS, b1v, thSAD=thSAD, thSCD1=thSCD1, thSCD2=thSCD2) : NOP()
  291. Tmax       = Tlimit ? source.mt_logic(f1c, "max", U=1, V=1).mt_logic(b1c, "max", U=1, V=1) : NOP()
  292. Tmin       = Tlimit ? source.mt_logic(f1c, "min", U=1, V=1).mt_logic(b1c, "min", U=1, V=1) : NOP()
  293. sclp       = Tlimit ? sclp.mt_clamp(Tmax, Tmin, Tovershoot, Tundershoot, U=2, V=2) : sclp
  294.  
  295.  
  296. # merge with edge mask and output correct chroma
  297. end        = edgemode <= 0 ? chroma ? sclp
  298. \                                   : ssout ? ssoutc ? sclp.MergeChroma(filtered_ss)
  299. \                                                    : sclp
  300. \                                           : sclp.MergeChroma(filtered_os)
  301. \          : edgemode == 1 ? ssout ? mt_merge(filtered_ss, sclp, edgemask, luma=chroma, Y=3, U=ssoutc?ch32:ch31, V=ssoutc?ch32:ch31)
  302. \                                  : mt_merge(filtered_os, sclp, edgemask, luma=chroma, Y=3, U=ch32,             V=ch32            )
  303. \                          : ssout ? mt_merge(sclp, filtered_ss, edgemask, luma=chroma, Y=3, U=ssoutc?ch34:ch31, V=ssoutc?ch34:ch31)
  304. \                                  : mt_merge(sclp, filtered_os, edgemask, luma=chroma, Y=3, U=ch34,             V=ch34            )
  305.  
  306. return end
  307. }
  308.  
  309.  
  310. Function MinBlur(clip clp, int "r", int "uv"){
  311.  
  312. r     = Default(r,  1)
  313. uv    = Default(uv, 3)
  314.  
  315. uv2   = (uv==2) ? 1  : uv
  316. rg4   = (uv==3) ? 4  : -1
  317. rg11  = (uv==3) ? 11 : -1
  318. rg20  = (uv==3) ? 20 : -1
  319. medf  = (uv==3) ? 1  : -200
  320. uvm2  = (r==2)  ? (uv==3?3:uv==2?0:-1) : nop()
  321. uvm3  = (r==3)  ? (uv==3?3:uv==2?0:-1) : nop()
  322.  
  323. RG11D = (r<=0) ? mt_makediff(clp, clp.sbr(uv=uv2), U=uv2, V=uv2)
  324. \     : (r==1) ? mt_makediff(clp, clp.RemoveGrain(11, rg11), U=uv2, V=uv2)
  325. \     : (r==2) ? mt_makediff(clp, clp.RemoveGrain(11, rg11).RemoveGrain(20, rg20), U=uv2, V=uv2)
  326. \     :          mt_makediff(clp, clp.RemoveGrain(11, rg11).RemoveGrain(20, rg20).RemoveGrain(20, rg20), U=uv2, V=uv2)
  327. RG4D  = (r<=1) ? mt_makediff(clp, clp.RemoveGrain(4,  rg4 ), U=uv2, V=uv2)
  328. \     : (r==2) ? mt_makediff(clp, clp.Quantile(radius_y=2, radius_u=uvm2, radius_v=uvm2), U=uv2, V=uv2)
  329. \     :          mt_makediff(clp, clp.Quantile(radius_y=3, radius_u=uvm3, radius_v=uvm3), U=uv2, V=uv2)
  330.  
  331. DD    = mt_lutxy(RG11D, RG4D, "x 128 - y 128 - * 0 < 128 x 128 - abs y 128 - abs < x y ? ?", U=uv2, V=uv2)
  332.  
  333. return clp.mt_makediff(DD, U=uv, V=uv)
  334. }
  335.  
  336.  
  337. Function sbr(clip c, int "uv") {
  338. uv     = Default(uv, 1)
  339. uv2    = (uv==2) ? 1  : uv
  340. rg11   = (uv==3) ? 11 : -1
  341. rg11D  = mt_makediff(c, c.RemoveGrain(11, rg11), U=uv2, V=uv2)
  342. rg11DD = mt_lutxy(rg11D, rg11D.RemoveGrain(11, rg11), "x y - x 128 - * 0 < 128 x y - abs x 128 - abs < x y - 128 + x ? ?", U=uv2, V=uv2)
  343.  
  344. return c.mt_makediff(rg11DD, U=uv, V=uv)
  345. }
  346.  
  347.  
  348. Function CSmod_nrSpline64Resize(clip input, int "target_width", int "target_height", float "src_left", float "src_top", float "src_width", float "src_height", bool "chroma", val "nr") {
  349.  
  350.   w             = input.Width
  351.   h             = input.Height
  352.  
  353.   target_width  = Default(target_width,      w)
  354.   target_height = Default(target_height,     h)
  355.   src_left      = Default(src_left,          0)
  356.   src_top       = Default(src_top,           0)
  357.   src_width     = Default(src_width,         w)
  358.   src_height    = Default(src_height,        h)
  359.   chroma        = Default(chroma,         true)
  360.   nr            = Default(nr,             true)
  361.  
  362.   Assert( IsFloat(nr) || IsBool(nr), """CSmod_nrSpline64Resize: "nr" should be either bool or float!""" )
  363.  
  364.   res_mul   = Float( target_width * target_height ) / Float( w * h )
  365.   res_mul   = min( max( res_mul, 1 ), 2.25 )
  366.   nr_weight = IsFloat(nr)       ? nr
  367.   \         : nr /* == True  */ ? Spline( res_mul, 1, 0, 2.25, 1, 3.5, 0, true )
  368.   \         :    /* == False */   0
  369.   nr_weight = min( max( nr_weight, 0 ), 1 )
  370.  
  371.   forceYV12 = input.IsYV12 == false && nr_weight != 0
  372.  
  373.   try {
  374.     inputp = chroma ? forceYV12 ? input .ConvertToYV12
  375.     \                           : input
  376.     \               : input.ConvertToY8
  377.   } catch ( error_msg ) {
  378.     inputp = forceYV12 ? input.ConvertToYV12
  379.     \                  : input
  380.   }
  381.  
  382.   resize = inputp.Spline64Resize(target_width, target_height, src_left, src_top, src_width, src_height)
  383.   nrres  = inputp.GaussResize(target_width, target_height, src_left, src_top, src_width, src_height, p=100)
  384.   resize = resize.CSmod_GetCSP == "Y8" ? resize.ConvertToYV12 : resize
  385.   nrres  = nrres .CSmod_GetCSP == "Y8" ? nrres .ConvertToYV12 : nrres
  386.  
  387.   resize = nr_weight == 0 ? resize
  388.   \      : nr_weight == 1 ? resize.Repair(nrres, 1, chroma ? 1 : -1)
  389.   \      : chroma         ? Merge    (resize, resize.Repair(nrres, 1,  1), nr_weight)
  390.   \                       : MergeLuma(resize, resize.Repair(nrres, 1, -1), nr_weight)
  391.  
  392.   return resize
  393. }
  394.  
  395.  
  396. Function CSmod_GetCSP(clip c) {
  397.   return c.IsPlanar ? c.IsYV12 ? "YV12" :
  398.   \                   c.IsYV16 ? "YV16" :
  399.   \                   c.IsYV24 ? "YV24" : c.CSmod_Y8_YV411 :
  400.   \      c.IsYUY2   ? "YUY2"   :
  401.   \      c.IsRGB32  ? "RGB32"  :
  402.   \      c.IsRGB24  ? "RGB24"  : "Unknown"
  403. }
  404.  
  405.  
  406. Function CSmod_Y8_YV411(clip c) {
  407.   try {
  408.     c.UtoY
  409.     csp = "YV411"
  410.   } catch ( error_msg ) {
  411.     csp = "Y8"
  412.   }
  413.   return csp
  414. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
Public Pastes
Advertisement
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy.  OK, I Understand
Not a member of Pastebin yet?
Sign Up, it unlocks many cool features!