Ticket #2757: test_scrolling.py

#!/usr/bin/env python
# This file is part of Xpra.
# Copyright (C) 2020 Antoine Martin <antoine@xpra.org>
# Xpra is released under the terms of the GNU GPL v2, or, at your option, any
# later version. See the file COPYING for details.

import sys

from PIL import Image
from xpra.util import csv
from xpra.server.window.motion import ScrollData                    #@UnresolvedImport


def main():
    scroll_data = ScrollData()
    for i, x in enumerate(sys.argv[1:]):
        img = Image.open(x)
        img = img.convert("RGBA")
        pixels = img.tobytes("raw", "BGRA")
        w, h = img.size
        Bpp = 4
        stride = w*Bpp
        scroll_data.update(pixels, 0, 0, w, h, stride, Bpp)
        if i==0:
            continue
        scroll_data.calculate(1000)
        scroll, count = scroll_data.get_best_match()
        match_pct = int(100*count/h)
        print("best scroll guess matches %i%% of %i lines: %s" % (match_pct, h, scroll))
        assert match_pct>10
        raw_scroll, non_scroll = scroll_data.get_scroll_values()
        scrolls = []
        for scroll, line_defs in raw_scroll.items():
            if scroll==0:
                continue
            for line, count in line_defs.items():
                scrolls.append((0, 0+line, w, count, 0, scroll))
        print("scrolls:")
        for scroll in scrolls:
            print("%s" % (csv(scroll),))
        #self.encode_scrolling(scroll_data, image, options, match_pct)


if __name__ == "__main__":
    main()