import vapoursynth as vs
import havsfunc as haf
import sgvsfunc as sgf

core = vs.get_core(accept_lowercase = True, threads = 36)

core.max_cache_size = 4000

src = core.dgdecodenv.DGSource("O:\BBCCG\BCCG.dgi").std.Crop(2,0,0,0)
src = haf.QTGMC(src, Preset = "Very Slow" ,TFF=True)
fixdirtyline1 = sgf.FixColumnBrightnessProtect2(src, 3, 4).sgf.FixColumnBrightnessProtect2(2, 20).sgf.FixColumnBrightnessProtect2(1,45).sgf.FixColumnBrightnessProtect2(1914,3).sgf.FixColumnBrightnessProtect2(1915,18).sgf.FixColumnBrightnessProtect2(1916,45).core.fb.FillBorders(1,1,0,0, "fillmargins").sgf.bbmod(0,0,4,4,blur=36)
fixdirtyline2 = core.std.Crop(0,2,0,0).fb.FillBorders(1,0,0,0, "fillmargins").sgf.FixColumnBrightnessProtect2(src, 1912, 5).sgf.FixColumnBrightnessProtect2(1913, 20).sgf.FixColumnBrightnessProtect2(1914, 45).sgf.bbmod(0,0,3,4, blur=36).sgf.AddBorders(0,2,0,0)
fixdirtyline3 = core.fb.FillBorders(1, 1, 0, 0, "fillmargins").sgf.bbmod(0,0,2,2, blur=36)
fixdirtyline4 = sgf.FixColumnBrightnessProtect2(src,5,-2).sgf.FixColumnBrightnessProtect2(3, 4).sgf.FixColumnBrightnessProtect2(2, 20).sgf.FixColumnBrightnessProtect2(1, 45).sgf.FixColumnBrightnessProtect2(1914, 3).sgf.FixColumnBrightnessProtect2(1915, 5).sgf.FixColumnBrightnessProtect2(1916, 30).core.fb.fillborder(1, 1, 0, 0,"fillmargins").sgf.bbmod(0, 0, 4, 4, blur=36)
fixdirtyline5= core.std.Crop(0, 2, 0, 0).fillborder(1, 0, 0, 0, "fillmargins").sgf.FixColumnBrightnessProtect2(src,1912,5).sgf.FixColumnBrightnessProtect2(1913, 20).sgf.FixColumnBrightnessProtect2(1914, 35).sgf.FixColumnBrightnessProtect2(1915, 48).sgf.bbmod(0, 0, 4, 4, blur=36).sgf.AddBorders(0, 2, 0, 0)
fixdirtyline6= sgf.FixColumnBrightnessProtect2(src,2, 5).sgf.FixColumnBrightnessProtect2(1, 20).sgf.FixColumnBrightnessProtect2(5, 2).sgf.FixColumnBrightnessProtect2(1913, 4).sgf.FixColumnBrightnessProtect2(1914, 20).sgf.FixColumnBrightnessProtect2(1915, 45).sgf.FixColumnBrightnessProtect2(1916, 72).core.fb.fillborder(1,1,0,0,"fillmargins").sgf.bbmod(0, 0, 4, 4, blur=36)

fix = core.fvf.ReplaceFrames(src,fixdirtyline1, "[0 113738] [114424 116754] [183104 183774] [225216 225446] [225980 294450] [312236 314154] [315062 433302] [433650 479726] [480006 508240]")
fix = core.fvf.ReplaceFrames(fix,fixdirtyline2, "[113740 114422]")
fix = core.fvf.ReplaceFrames(fix,fixdirtyline3, "[116756 118730]")
fix = core.fvf.ReplaceFrames(fix,fixdirtyline4, "[118732 166404]")
fix = core.fvf.ReplaceFrames(fix,fixdirtyline5, "[166406 166750] [166752 181952] [181954 183102] [183776 184056] [184058 217334] [217336 219648] [223324 225214] [225448 225978] [294452 295346] [295348 312110] [312112 312234] [314156 315060] [433304 433648] [479728 480004]")
fix = core.fvf.ReplaceFrames(fix,fixdirtyline6, "[219650 223322]")


Deband1 = core.f3kdb.Deband(fix,grainY = 10,dynamic_grain=True)
Deband2 = core.f3kdb.Deband(fix,Y=90, grainY=10, grainC=0, dither_algo=3, dynamic_grain=True)
db = core.fvf.ReplaceFrames(fix,Deband1,"[135968 136106] [338534 339534]")
db = core.fvf.ReplaceFrames(fix,Deband2,"[3404 3406] [96946 96956] [96974 96998] [105112 105180] [121512 121524] [121542 121556] [121608 121620] [121666 121676] [124376 124410] [159586 159628]")

db.set_output()