import vapoursynth as vs
import sgvsfunc as sgf

core = vs.get_core()

source = core.dgdecodenv.DGSource(r"00800.dgi")
source = core.std.Crop(source, top=32, bottom=34)

# Global Fix
src = core.fb.FillBorders(source, top=1, mode="fillmargins")
src = sgf.bbmod(src, 2, 0, 0, 0)

# Intro - Dirty Edge
intro = core.std.Crop(src, top=104, bottom=104)
intro = core.fb.FillBorders(intro, top=1, mode="fillmargins")
intro = sgf.bbmod(intro, 2, 0, 0, 0, blur=100)
intro = core.std.AddBorders(intro, top=104, bottom=104)

# IMAX - Deband
imax_db = core.f3kdb.Deband(src, y=60, grainy=45, dynamic_grain=True)

# Non-IMAX - Deband
non_imax_db = core.f3kdb.Deband(src, y=60, grainy=45, dynamic_grain=True)
non_imax_db = core.std.Crop(non_imax_db, top=104, bottom=104)
non_imax_db = core.std.AddBorders(non_imax_db, top=104, bottom=104)

# Merge
output = sgf.ReplaceFrames(src, intro, avs_filename=r'intro.txt')
output = sgf.ReplaceFrames(output, imax_db, avs_filename=r'deband_IMAX.txt')
output = sgf.ReplaceFrames(output, non_imax_db, avs_filename=r'deband.txt')

# Output
output.set_output()