#!/usr/bin/env python
# License: GPL v3 Copyright: 2016, Kovid Goyal <kovid at kovidgoyal.net>

import random
from binascii import hexlify
from functools import partial

from kitty.fast_data_types import (
    CURSOR_BLOCK,
    VT_PARSER_BUFFER_SIZE,
    base64_decode,
    base64_encode,
    has_avx2,
    has_avx512,
    has_sse4_2,
    test_blend_pixels,
    test_find_either_of_two_bytes,
    test_printable_ascii_run_length,
    test_utf8_decode_to_sentinel,
    test_xor64,
)

from .base import BaseTest, parse_bytes


def cnv(x):
    if isinstance(x, memoryview):
        x = str(x, 'utf-8')
    return x


class CmdDump(list):
    def __call__(self, window_id, *a):
        if a and a[0] == 'bytes':
            return
        if a and a[0] == 'error':
            a = a[1:]
        self.append(tuple(map(cnv, a)))

    def get_result(self):
        current = ''
        q = []
        for args in self:
            if args[0] == 'draw':
                current += args[1]
            else:
                if current:
                    q.append(('draw', current))
                    current = ''
                q.append(args)
        if current:
            q.append(('draw', current))
        return tuple(q)


class TestParser(BaseTest):
    def create_write_buffer(self, screen):
        return screen.test_create_write_buffer()

    def write_bytes(self, screen, write_buf, data):
        if isinstance(data, str):
            data = data.encode('utf-8')
        s = screen.test_commit_write_buffer(data, write_buf)
        return data[s:]

    def parse_written_data(self, screen, *cmds):
        cd = CmdDump()
        screen.test_parse_written_data(cd)
        cmds = tuple(('draw', x) if isinstance(x, str) else tuple(map(cnv, x)) for x in cmds)
        self.ae(cmds, cd.get_result())

    def parse_bytes_dump(self, s, x, *cmds):
        cd = CmdDump()
        if isinstance(x, str):
            x = x.encode('utf-8')
        cmds = tuple(('draw', x) if isinstance(x, str) else tuple(map(cnv, x)) for x in cmds)
        parse_bytes(s, x, cd)
        self.ae(cmds, cd.get_result())

    def test_charsets(self):
        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)
        pb(b'\xc3')
        pb(b'\xa1', ('draw', b'\xc3\xa1'.decode('utf-8')))
        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)
        pb('\033)0\x0e/_', ('screen_designate_charset', 1, ord('0')), ('screen_change_charset', 1), '/_')
        self.ae(str(s.line(0)), '/\xa0')
        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)
        pb('\033(0/_', ('screen_designate_charset', 0, ord('0')), '/_')
        self.ae(str(s.line(0)), '/\xa0')

    def test_parser_threading(self):
        s = self.create_screen()

        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), 'a\x1b]2;some title'))
        b = self.create_write_buffer(s)
        self.parse_written_data(s, 'a')
        self.assertFalse(self.write_bytes(s, b, ' full\x1b\\'))
        self.parse_written_data(s, ('set_title', 'some title full'))

        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), 'a\x1b]'))
        b = self.create_write_buffer(s)
        self.parse_written_data(s, 'a')
        self.assertFalse(self.write_bytes(s, b, '2;title\x1b\\'))
        self.parse_written_data(s, ('set_title', 'title'))

        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), 'a\x1b'))
        b = self.create_write_buffer(s)
        self.parse_written_data(s, 'a')
        self.assertFalse(self.write_bytes(s, b, ']2;title\x1b\\'))
        self.parse_written_data(s, ('set_title', 'title'))

        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), 'a\x1b]2;some title\x1b'))
        b = self.create_write_buffer(s)
        self.parse_written_data(s, 'a')
        self.assertFalse(self.write_bytes(s, b, '\\b'))
        self.parse_written_data(s, ('set_title', 'some title'), 'b')

        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), '1\x1b'))
        self.parse_written_data(s, '1')
        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), 'E2'))
        self.parse_written_data(s, ('screen_nel',), ('draw', '2'))

        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), '1\x1b[2'))
        self.parse_written_data(s, '1')
        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), '3mx'))
        self.parse_written_data(s, ('select_graphic_rendition', '23'), 'x')

        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), '1\x1b'))
        self.parse_written_data(s, '1')
        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), '[23mx'))
        self.parse_written_data(s, ('select_graphic_rendition', '23'), 'x')

        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), '1\x1b['))
        self.parse_written_data(s, '1')
        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), '23mx'))
        self.parse_written_data(s, ('select_graphic_rendition', '23'), 'x')

        # test full write
        sz = VT_PARSER_BUFFER_SIZE // 3 + 7
        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), b'a' * sz))
        self.assertFalse(self.write_bytes(s, self.create_write_buffer(s), b'b' * sz))
        left = self.write_bytes(s, self.create_write_buffer(s), b'c' * sz)
        self.assertTrue(len(left), 3 * sz - VT_PARSER_BUFFER_SIZE)
        self.assertFalse(self.create_write_buffer(s))
        s.test_parse_written_data()
        b = self.create_write_buffer(s)
        self.assertTrue(b)
        self.write_bytes(s, b, b'')

    def test_base64(self):
        for src, expected in {
            'bGlnaHQgdw==': 'light w',
            'bGlnaHQgd28=': 'light wo',
            'bGlnaHQgd29y': 'light wor',
        }.items():
            self.ae(base64_decode(src.encode()), expected.encode(), f'Decoding of {src} failed')
            self.ae(base64_decode(src.replace('=', '').encode()), expected.encode(), f'Decoding of {src} failed')
            self.ae(base64_encode(expected.encode()), src.replace('=', '').encode(), f'Encoding of {expected} failed')
            self.ae(base64_decode(src.encode(), True), expected.encode(), f'Strict decoding of {src} failed')
        # in strict mode characters outside the base64 alphabet and missing padding are rejected
        for invalid in ('bGlnaHQgdw=', 'bGlnaHQgdw', 'bGln!!Qgdw==', 'bGlnaHQgdw==\n', '!!!!', 'bGl'):
            self.assertRaises(ValueError, base64_decode, invalid.encode(), True)
        self.ae(base64_decode(b'', True), b'')

    def test_simple_parsing(self):
        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)

        pb('12', '12')
        self.ae(str(s.line(0)), '12')
        self.ae(s.cursor.x, 2)
        pb('3456', '3456')
        self.ae(str(s.line(0)), '12345')
        self.ae(str(s.line(1)), '6')
        pb(b'\n123\n\r45', ('screen_linefeed',), '123', ('screen_linefeed',), ('screen_carriage_return',), '45')
        self.ae(str(s.line(1)), '6')
        self.ae(str(s.line(2)), ' 123')
        self.ae(str(s.line(3)), '45')
        pb(b'\rabcde', ('screen_carriage_return',), 'abcde')
        self.ae(str(s.line(3)), 'abcde')
        pb('\rßxyz1', ('screen_carriage_return',), 'ßxyz1')
        self.ae(str(s.line(3)), 'ßxyz1')
        pb('ニチ ', 'ニチ ')
        self.ae(str(s.line(4)), 'ニチ ')
        s.reset()
        self.assertFalse(str(s.line(1)) + str(s.line(2)) + str(s.line(3)))
        c1_controls = '\x84\x85\x88\x8d\x8e\x8f\x90\x96\x97\x98\x9a\x9b\x9c\x9d\x9e\x9f'
        pb(c1_controls, c1_controls)
        self.assertFalse(str(s.line(1)) + str(s.line(2)) + str(s.line(3)))
        pb('😀'.encode()[:-1])
        pb('\x1b\x1b%a', '\ufffd', ('Unknown char after ESC: 0x1b',), ('draw', '%a'))

    def test_utf8_simd_decode(self):
        def unsupported(which):
            return (which == 2 and not has_sse4_2) or (which == 3 and not has_avx2) or (which == 4 and not has_avx512)

        def reset_state():
            test_utf8_decode_to_sentinel(b'', -1)

        def asbytes(x):
            if isinstance(x, str):
                x = x.encode()
            return x

        def t(*a, which=2):
            if unsupported(which):
                return

            def parse_parts(which):
                total_consumed = 0
                esc_found = False
                parts = []
                for x in a:
                    found_sentinel, x, num_consumed = test_utf8_decode_to_sentinel(asbytes(x), which)
                    total_consumed += num_consumed
                    if found_sentinel:
                        esc_found = found_sentinel
                    parts.append(x)
                return esc_found, ''.join(parts), total_consumed

            reset_state()
            expected = parse_parts(1)
            reset_state()
            actual = parse_parts(which)
            self.ae(expected, actual, msg=f'Failed for {a} with {which=}\n{expected!r} !=\n{actual!r}')
            return actual

        def double_test(x):
            for which in (2, 3, 4):
                t(x, which=which)
            t(x * 2, which=3)
            t(x * 4, which=4)
            reset_state()

        # incomplete trailer at end of vector
        t('a' * 10 + '😸😸' + 'b' * 15)

        x = double_test
        x('2:α3')
        x('2:α\x1b3')
        x('2:α3:≤4:😸|')
        x('abcd1234efgh5678')
        x('abc\x1bd1234efgh5678')
        x('abcd1234efgh5678ijklABCDmnopEFGH')

        for which in (2, 3, 4):
            x = partial(t, which=which)
            x('abcdef', 'ghijk')
            x('2:α3', ':≤4:😸|')
            # trailing incomplete sequence
            for prefix in (b'abcd', '😸'.encode()):
                for suffix in (b'1234', '😸'.encode()):
                    x(prefix + b'\xf0\x9f', b'\x98\xb8' + suffix)
                    x(prefix + b'\xf0\x9f\x9b', b'\xb8' + suffix)
                    x(prefix + b'\xf0', b'\x9f\x98\xb8' + suffix)
                    x(prefix + b'\xc3', b'\xa4' + suffix)
                    x(prefix + b'\xe2', b'\x89\xa4' + suffix)
                    x(prefix + b'\xe2\x89', b'\xa4' + suffix)

        def test_expected(src, expected, which=2):
            if unsupported(which):
                return
            reset_state()
            _, actual, _ = t(b'filler' + asbytes(src), which=which)
            expected = 'filler' + expected
            self.ae(expected, actual, f'Failed for: {src!r} with {which=}')

        for which in (1, 2, 3, 4):
            pb = partial(test_expected, which=which)
            pb('ニチ', 'ニチ')
            pb('\x84\x85', '\x84\x85')
            pb('\x84\x85', '\x84\x85')
            pb('\uf4df', '\uf4df')
            pb('\uffff', '\uffff')
            pb('\0', '\0')
            pb(chr(0x10FFFF), chr(0x10FFFF))
            # Kitty's UTF-8 decoding uses `U+FFFD substitution of maximal subparts
            # <https://www.unicode.org/versions/Unicode16.0.0/core-spec/chapter-3/#G66453>`_,
            # same as in the WHATWG Encoding Standard.
            # This means that ill-formed sequences may be replaced by multiple
            # U+FFFD REPLACEMENT CHARACTERs.
            pb(b'abcd\xf51234', 'abcd\ufffd1234')  # bytes > 0xf4
            pb(b'abcd\xff1234', 'abcd\ufffd1234')  # bytes > 0xf4
            pb(b'"\xbf"', '"\ufffd"')
            pb(b'"\x80"', '"\ufffd"')
            pb(b'"\x80\xbf"', '"\ufffd\ufffd"')
            pb(b'"\x80\xbf\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xc0 "', '"\ufffd "')
            pb(b'"\xfe"', '"\ufffd"')
            pb(b'"\xff"', '"\ufffd"')
            pb(b'"\xff\xfe"', '"\ufffd\ufffd"')
            pb(b'"\xfe\xfe\xff\xff"', '"\ufffd\ufffd\ufffd\ufffd"')
            pb(b'"\xef\xbf"', '"\ufffd"')
            pb(b'"\xe0\xa0"', '"\ufffd"')
            pb(b'"\xf0\x9f\x98"', '"\ufffd"')
            pb(b'"\xef\x93\x94\x95"', '"\uf4d4\ufffd"')

            # Lone continuation bytes with no leading starts
            pb(b'"\xbf"', '"\ufffd"')
            pb(b'"\x80"', '"\ufffd"')

            # Multiple lone continuation bytes
            pb(b'"\x80\xbf"', '"\ufffd\ufffd"')
            pb(b'"\x80\xbf\x80"', '"\ufffd\ufffd\ufffd"')

            # Lone starter byte of 2-byte sequence
            pb(b'"\xc0 "', '"\ufffd "')

            # Single never-valid bytes
            pb(b'"\xfe"', '"\ufffd"')
            pb(b'"\xff"', '"\ufffd"')

            # Multiple never-valid bytes
            pb(b'"\xff\xfe"', '"\ufffd\ufffd"')
            pb(b'"\xfe\xfe\xff\xff"', '"\ufffd\ufffd\ufffd\ufffd"')

            # Truncated 2-byte sequence (only 1 byte)
            pb(b'"\xc2"', '"\ufffd"')

            # Truncated 3-byte sequences (only 2 bytes)
            pb(b'"\xef\xbf"', '"\ufffd"')
            pb(b'"\xe0\xa0"', '"\ufffd"')

            # Truncated 4-byte sequence (only 2 or 3 bytes)
            pb(b'"\xf0\x9f"', '"\ufffd"')
            pb(b'"\xf0\x9f\x98"', '"\ufffd"')

            # Bad continuation byte (restored as ASCII)
            pb(b'"\xe1\x28\xa1"', '"\ufffd(\ufffd"')  # )

            # Overlong 2-byte sequence for U+0000 (should be `0x00`)
            pb(b'"\xc0\x80"', '"\ufffd\ufffd"')

            # Overlong 3-byte sequence for U+0000 (violates boundary)
            pb(b'"\xe0\x80\x80"', '"\ufffd\ufffd\ufffd"')

            # Overlong 4-byte sequence for U+0000 (violates boundary)
            pb(b'"\xf0\x80\x80\x80"', '"\ufffd\ufffd\ufffd\ufffd"')

            # High surrogate code point
            pb(b'"\xed\xa0\x80"', '"\ufffd\ufffd\ufffd"')

            # Low surrogate code point
            pb(b'"\xed\xb0\x80"', '"\ufffd\ufffd\ufffd"')

            # Too large starter byte
            pb(b'"\xff\x80\x80\x80"', '"\ufffd\ufffd\ufffd\ufffd"')

            # The following boundary cases come from the table of well-formed UTF-8 byte sequences
            # <https://www.unicode.org/versions/Unicode16.0.0/core-spec/chapter-3/#G27506>`_.
            # For continuation bytes, both 0xC0 and 0xC2 are tested as values that exceed the valid maximum.
            # This is because 0xC0 is an invalid starter byte, but 0xC2 is also a starter byte for 2-byte sequences.
            # simd-string-impl.h prefers classifying bytes as starter bytes when possible (e.g., in "\xf0\x90\xc2\x80").
            # The tests need to check that simd-string-impl.h correctly detects
            # starter bytes that are actually invalid continution bytes, like 0xC2.

            # Boundary cases: 2-byte sequences
            pb(b'"\xc1\x7f"', '"\ufffd\x7f"')
            pb(b'"\xc1\x80"', '"\ufffd\ufffd"')
            pb(b'"\xc1\xbf"', '"\ufffd\ufffd"')
            pb(b'"\xc1\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xc1\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xc2\x7f"', '"\ufffd\x7f"')
            pb(b'"\xc2\x80"', '"\x80"')
            pb(b'"\xc2\xbf"', '"\xbf"')
            pb(b'"\xc2\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xc2\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xdf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xdf\x80"', '"\u07c0"')
            pb(b'"\xdf\xbf"', '"\u07ff"')
            pb(b'"\xdf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xdf\xc2"', '"\ufffd\ufffd"')

            # Boundary cases: 3-byte sequences starting with 0xE0
            pb(b'"\xe0\x9f\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xe0\xa0\x7f"', '"\ufffd\x7f"')
            pb(b'"\xe0\xa0\x80"', '"\u0800"')
            pb(b'"\xe0\xa0\xbf"', '"\u083f"')
            pb(b'"\xe0\xa0\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xe0\xa0\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xe0\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xe0\xbf\x80"', '"\u0fc0"')
            pb(b'"\xe0\xbf\xbf"', '"\u0fff"')
            pb(b'"\xe0\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xe0\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xe0\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xe0\xc2\x80"', '"\ufffd\x80"')

            # Boundary cases: 3-byte sequences starting with 0xE1..0xEC
            pb(b'"\xe1\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xe1\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xe1\x80\x80"', '"\u1000"')
            pb(b'"\xe1\x80\xbf"', '"\u103f"')
            pb(b'"\xe1\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xe1\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xe1\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xe1\xbf\x80"', '"\u1fc0"')
            pb(b'"\xe1\xbf\xbf"', '"\u1fff"')
            pb(b'"\xe1\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xe1\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xe1\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xe1\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xec\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xec\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xec\x80\x80"', '"\uc000"')
            pb(b'"\xec\x80\xbf"', '"\uc03f"')
            pb(b'"\xec\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xec\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xec\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xec\xbf\x80"', '"\ucfc0"')
            pb(b'"\xec\xbf\xbf"', '"\ucfff"')
            pb(b'"\xec\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xec\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xec\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xec\xc2\x80"', '"\ufffd\x80"')

            # Boundary cases: 3-byte sequences starting with 0xED
            pb(b'"\xed\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xed\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xed\x80\x80"', '"\ud000"')
            pb(b'"\xed\x80\xbf"', '"\ud03f"')
            pb(b'"\xed\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xed\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xed\x9f\x7f"', '"\ufffd\x7f"')
            pb(b'"\xed\x9f\x80"', '"\ud7c0"')
            pb(b'"\xed\x9f\xbf"', '"\ud7ff"')
            pb(b'"\xed\x9f\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xed\x9f\xc2"', '"\ufffd\ufffd"')

            # Boundary cases: 3-byte sequences starting with 0xEE..0xEF
            pb(b'"\xed\xa0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xee\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xee\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xee\x80\x80"', '"\ue000"')
            pb(b'"\xee\x80\xbf"', '"\ue03f"')
            pb(b'"\xee\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xee\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xee\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xee\xbf\x80"', '"\uefc0"')
            pb(b'"\xee\xbf\xbf"', '"\uefff"')
            pb(b'"\xee\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xee\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xee\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xee\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xef\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xef\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xef\x80\x80"', '"\uf000"')
            pb(b'"\xef\x80\xbf"', '"\uf03f"')
            pb(b'"\xef\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xef\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xef\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xef\xbf\x80"', '"\uffc0"')
            pb(b'"\xef\xbf\xbf"', '"\uffff"')
            pb(b'"\xef\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xef\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xef\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xef\xc2\x80"', '"\ufffd\x80"')

            # Boundary cases: 4-byte sequences starting with 0xF0
            pb(b'"\xf0\x8f\x80\x80"', '"\ufffd\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\x90\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xf0\x90\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf0\x90\x80\x80"', '"\U00010000"')
            pb(b'"\xf0\x90\x80\xbf"', '"\U0001003f"')
            pb(b'"\xf0\x90\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf0\x90\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf0\x90\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf0\x90\xbf\x80"', '"\U00010fc0"')
            pb(b'"\xf0\x90\xbf\xbf"', '"\U00010fff"')
            pb(b'"\xf0\x90\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf0\x90\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf0\x90\xc0\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf0\x90\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\x90\xc0\xbf"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\x90\xc0\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\x90\xc0\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\x90\xc2\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf0\x90\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xf0\x90\xc2\xbf"', '"\ufffd\xbf"')
            pb(b'"\xf0\x90\xc2\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\x90\xc2\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\xbf\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xf0\xbf\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf0\xbf\x80\x80"', '"\U0003f000"')
            pb(b'"\xf0\xbf\x80\xbf"', '"\U0003f03f"')
            pb(b'"\xf0\xbf\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf0\xbf\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf0\xbf\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf0\xbf\xbf\x80"', '"\U0003ffc0"')
            pb(b'"\xf0\xbf\xbf\xbf"', '"\U0003ffff"')
            pb(b'"\xf0\xbf\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf0\xbf\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf0\xbf\xc0\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf0\xbf\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\xbf\xc0\xbf"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\xbf\xc0\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\xbf\xc0\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\xbf\xc2\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf0\xbf\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xf0\xbf\xc2\xbf"', '"\ufffd\xbf"')
            pb(b'"\xf0\xbf\xc2\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\xbf\xc2\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf0\xc0\x80\x80"', '"\ufffd\ufffd\ufffd\ufffd"')

            # Boundary cases: 4-byte sequences starting with 0xF1..0xF3
            pb(b'"\xf1\x7f\x80\x80"', '"\ufffd\x7f\ufffd\ufffd"')
            pb(b'"\xf1\x80\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xf1\x80\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf1\x80\x80\x80"', '"\U00040000"')
            pb(b'"\xf1\x80\x80\xbf"', '"\U0004003f"')
            pb(b'"\xf1\x80\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf1\x80\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf1\x80\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf1\x80\xbf\x80"', '"\U00040fc0"')
            pb(b'"\xf1\x80\xbf\xbf"', '"\U00040fff"')
            pb(b'"\xf1\x80\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf1\x80\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf1\x80\xc0\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf1\x80\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\x80\xc0\xbf"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\x80\xc0\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\x80\xc2\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf1\x80\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xf1\x80\xc2\xbf"', '"\ufffd\xbf"')
            pb(b'"\xf1\x80\xc2\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\xbf\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xf1\xbf\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf1\xbf\x80\x80"', '"\U0007f000"')
            pb(b'"\xf1\xbf\x80\xbf"', '"\U0007f03f"')
            pb(b'"\xf1\xbf\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf1\xbf\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf1\xbf\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf1\xbf\xbf\x80"', '"\U0007ffc0"')
            pb(b'"\xf1\xbf\xbf\xbf"', '"\U0007ffff"')
            pb(b'"\xf1\xbf\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf1\xbf\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf1\xbf\xc0\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf1\xbf\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\xbf\xc0\xbf"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\xbf\xc0\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\xbf\xc0\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\xbf\xc2\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf1\xbf\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xf1\xbf\xc2\xbf"', '"\ufffd\xbf"')
            pb(b'"\xf1\xbf\xc2\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\xbf\xc2\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\xc0\x80\x80"', '"\ufffd\ufffd\ufffd\ufffd"')
            pb(b'"\xf1\xc2\x80\x80"', '"\ufffd\x80\ufffd"')
            pb(b'"\xf3\x7f\x80\x80"', '"\ufffd\x7f\ufffd\ufffd"')
            pb(b'"\xf3\x80\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xf3\x80\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf3\x80\x80\x80"', '"\U000c0000"')
            pb(b'"\xf3\x80\x80\xbf"', '"\U000c003f"')
            pb(b'"\xf3\x80\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf3\x80\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf3\x80\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf3\x80\xbf\x80"', '"\U000c0fc0"')
            pb(b'"\xf3\x80\xbf\xbf"', '"\U000c0fff"')
            pb(b'"\xf3\x80\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf3\x80\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf3\x80\xc0\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf3\x80\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\x80\xc0\xbf"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\x80\xc0\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\x80\xc0\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\x80\xc2\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf3\x80\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xf3\x80\xc2\xbf"', '"\ufffd\xbf"')
            pb(b'"\xf3\x80\xc2\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\x80\xc2\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\xbf\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xf3\xbf\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf3\xbf\x80\x80"', '"\U000ff000"')
            pb(b'"\xf3\xbf\x80\xbf"', '"\U000ff03f"')
            pb(b'"\xf3\xbf\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf3\xbf\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf3\xbf\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf3\xbf\xbf\x80"', '"\U000fffc0"')
            pb(b'"\xf3\xbf\xbf\xbf"', '"\U000fffff"')
            pb(b'"\xf3\xbf\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf3\xbf\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf3\xbf\xc0\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf3\xbf\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\xbf\xc0\xbf"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\xbf\xc0\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\xbf\xc0\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\xbf\xc2\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf3\xbf\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xf3\xbf\xc2\xbf"', '"\ufffd\xbf"')
            pb(b'"\xf3\xbf\xc2\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\xbf\xc2\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\xc0\x80\x80"', '"\ufffd\ufffd\ufffd\ufffd"')
            pb(b'"\xf3\xc2\x80\x80"', '"\ufffd\x80\ufffd"')

            # Boundary cases: 4-byte sequences starting with 0xF4
            pb(b'"\xf4\x7f\x80\x80"', '"\ufffd\x7f\ufffd\ufffd"')
            pb(b'"\xf4\x80\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xf4\x80\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf4\x80\x80\x80"', '"\U00100000"')
            pb(b'"\xf4\x80\x80\xbf"', '"\U0010003f"')
            pb(b'"\xf4\x80\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf4\x80\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf4\x80\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf4\x80\xbf\x80"', '"\U00100fc0"')
            pb(b'"\xf4\x80\xbf\xbf"', '"\U00100fff"')
            pb(b'"\xf4\x80\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf4\x80\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf4\x80\xc0\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf4\x80\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x80\xc0\xbf"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x80\xc0\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x80\xc0\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x80\xc2\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf4\x80\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xf4\x80\xc2\xbf"', '"\ufffd\xbf"')
            pb(b'"\xf4\x80\xc2\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x80\xc2\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x8f\x7f\x80"', '"\ufffd\x7f\ufffd"')
            pb(b'"\xf4\x8f\x80\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf4\x8f\x80\x80"', '"\U0010f000"')
            pb(b'"\xf4\x8f\x80\xbf"', '"\U0010f03f"')
            pb(b'"\xf4\x8f\x80\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf4\x8f\x80\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf4\x8f\xbf\x7f"', '"\ufffd\x7f"')
            pb(b'"\xf4\x8f\xbf\x80"', '"\U0010ffc0"')
            pb(b'"\xf4\x8f\xbf\xbf"', '"\U0010ffff"')
            pb(b'"\xf4\x8f\xbf\xc0"', '"\ufffd\ufffd"')
            pb(b'"\xf4\x8f\xbf\xc2"', '"\ufffd\ufffd"')
            pb(b'"\xf4\x8f\xc0\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf4\x8f\xc0\x80"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x8f\xc0\xbf"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x8f\xc0\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x8f\xc0\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x8f\xc2\x7f"', '"\ufffd\ufffd\x7f"')
            pb(b'"\xf4\x8f\xc2\x80"', '"\ufffd\x80"')
            pb(b'"\xf4\x8f\xc2\xbf"', '"\ufffd\xbf"')
            pb(b'"\xf4\x8f\xc2\xc0"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x8f\xc2\xc2"', '"\ufffd\ufffd\ufffd"')
            pb(b'"\xf4\x90\x80\x80"', '"\ufffd\ufffd\ufffd\ufffd"')
            pb(b'"\xf5\x80\x80\x80"', '"\ufffd\ufffd\ufffd\ufffd"')

            # Boundary case: too large codepoint (> U+10FFFF)
            pb(b'"\xf5\x80\x80\x80"', '"\ufffd\ufffd\ufffd\ufffd"')

        # The tests below compare SIMD implementations against the scalar one via t().
        # SIMD chunks are 16 (SSE), 32 (AVX2) or 64 (AVX-512) bytes and implementations
        # may compact bytes within 16-byte lanes, so slide multi-byte sequences across
        # every lane and chunk boundary.
        simd_impls = 2, 3, 4
        multibyte = '\xb5'.encode(), 'α'.encode(), '≤'.encode(), '\U0001f638'.encode()
        for which in simd_impls:
            for mb in multibyte:
                for filler_len in range(70):
                    t(b'a' * filler_len + mb + b'b' * 40, which=which)
                    t(b'a' * filler_len + mb * 3 + b'b' * 40, which=which)

        # ESC at and around lane/chunk boundaries, adjacent to multi-byte sequences
        # and cutting off incomplete sequences
        for which in simd_impls:
            for esc_pos in (0, 1, 14, 15, 16, 17, 30, 31, 32, 33, 47, 48, 63, 64, 65):
                buf = bytearray(b'x' * 80)
                buf[esc_pos] = 0x1B
                t(bytes(buf), which=which)
                t(b'y' * esc_pos + '≤'.encode() + b'\x1b' + '\u03c6'.encode(), which=which)
                t(b'y' * esc_pos + b'\xe2\x89\x1b' + b'z' * 8, which=which)
                t(b'y' * esc_pos + b'\xf0\x9f\x98\x1b', which=which)

        # Invalid/incomplete sequences sliding across lane/chunk boundaries, both
        # followed by more data and truncated at the end of input
        bad_seqs = (
            b'\x80',
            b'\xbf\xbf',
            b'\xc0\x80',
            b'\xc1\xbf',
            b'\xe0\x80\x80',
            b'\xe0\x9f\x80',
            b'\xed\xa0\x80',
            b'\xf0\x8f\x80\x80',
            b'\xf4\x90\x80\x80',
            b'\xf5\x81\x82\x83',
            b'\xfe',
            b'\xff',
            b'\xc2',
            b'\xe2\x89',
            b'\xf0\x9f\x98',
        )
        for which in simd_impls:
            for bad in bad_seqs:
                for pos in (0, 13, 14, 15, 16, 17, 29, 30, 31, 32, 33, 62, 63, 64):
                    t(b'k' * pos + bad + b'm' * 9, which=which)
                    t(b'k' * pos + bad, which=which)

        # Overlapping sequences at the end of a chunk. These make counts[last] > 1,
        # triggering the incomplete-trailing-sequence check, without there being an
        # actual incomplete trailing sequence.
        for which in simd_impls:
            for sz in (16, 32, 64, 128):
                for pattern in (b'\xc0\xefh\x7f', b'\xc2\xefh\x7f', b'\xc0\xefhq', b'\xe0\xa0\xc2\xe2\x89'[:4]):
                    t(b'k' * (sz - 4) + pattern, which=which)
                    t(b'k' * (sz - 4) + pattern + b'more data', which=which)

        # Dense/alternating sequence widths (maximum compaction churn) and exact
        # chunk-multiple sizes
        for which in simd_impls:
            t(b'\x80' * 32, which=which)
            t(b'\xbf' * 64, which=which)
            t(('α' * 32).encode(), which=which)
            t(('aα' * 24).encode(), which=which)
            t(('≤' * 22).encode(), which=which)
            t(('\U0001f638' * 17).encode(), which=which)
            t(('a\U0001f638' * 13).encode(), which=which)
            t(('α≤\U0001f638' * 11).encode(), which=which)
            t(b'a' * 32, which=which)
            t(b'a' * 33, which=which)
            t(b'a' * 64, which=which)
            t(b'a' * 65, which=which)
            t(b'a' * 128, which=which)
            t(b'a' * 129, which=which)
            t(('α' * 16).encode() + b'\x1b[m', which=which)
            t(('α' * 32).encode() + b'\x1b[m', which=which)

        # Seeded random fuzzing against the scalar implementation, including
        # split inputs to exercise cross-call state carry-over
        rng = random.Random(0x1BADF00D)
        interesting = 0x1B, 0x00, 0x7F, 0x80, 0xBF, 0xC0, 0xC1, 0xC2, 0xDF, 0xE0, 0xED, 0xEE, 0xEF, 0xF0, 0xF4, 0xF5, 0xFF
        valid_chars = 'a', '\xb5', 'α', '≤', '\U0001f638', '\u07ff', '\u0800', '\ud7ff', '\ue000', '\uffff', '\U00010000', '\U0010ffff'
        for which in simd_impls:
            for _ in range(300):
                parts = []
                for _ in range(rng.randint(1, 80)):
                    r = rng.random()
                    if r < 0.5:
                        parts.append(rng.choice(valid_chars).encode())
                    elif r < 0.8:
                        parts.append(bytes((rng.choice(interesting),)))
                    else:
                        parts.append(bytes((rng.randint(0, 255),)))
                data = b''.join(parts)
                t(data, which=which)
                if len(data) > 2:
                    cut = rng.randint(1, len(data) - 1)
                    t(data[:cut], data[cut:], which=which)

    def test_find_either_of_two_bytes(self):
        sizes = []
        if has_sse4_2:
            sizes.append(2)
        if has_avx2:
            sizes.append(3)
        if has_avx512:
            sizes.append(4)
        sizes.append(0)

        def test(buf, a, b, align_offset=0):
            a_, b_ = ord(a), ord(b)
            expected = test_find_either_of_two_bytes(buf, a_, b_, 1, 0)
            for sz in sizes:
                actual = test_find_either_of_two_bytes(buf, a_, b_, sz, align_offset)
                self.ae(expected, actual, f'Failed for: {buf!r} {a=} {b=} at {sz=} and {align_offset=}')

        q = 'abc'
        for off in range(64):
            test(q, '<', '>', off)
            test(q, ' ', 'b', off)
            test(q, '<', 'a', off)
            test(q, '<', 'b', off)
            test(q, 'c', '>', off)

        def tests(buf, a, b):
            for sz in (0, 16, 32, 64, 79, 128, 133):
                buf = (' ' * sz) + buf
                for align_offset in range(64):
                    test(buf, a, b, align_offset)

        tests('', '<', '>')
        tests('a', '\0', '\0')
        tests('a', '<', '>')
        tests('dsdfsfa', '1', 'a')
        tests('xa', 'a', 'a')
        tests('bbb', 'a', '1')
        tests('bba', 'a', '<')
        tests('baa', '>', 'a')

    def test_printable_ascii_run_length(self):
        impls = [1]
        if has_sse4_2:
            impls.append(2)
        if has_avx2:
            impls.append(3)
        if has_avx512:
            impls.append(4)
        impls.append(0)

        def test(text):
            expected = 0
            while expected < len(text) and ' ' <= text[expected] <= '~':
                expected += 1
            for impl in impls:
                actual = test_printable_ascii_run_length(text, impl)
                self.ae(expected, actual, f'Failed for: {text!r} at {impl=}')

        for prefix_len in (0, 1, 3, 4, 7, 8, 15, 16, 17, 31, 32, 33, 63, 64, 100):
            prefix = 'a' * prefix_len
            test(prefix)
            for bad in ('\x1b', '\n', '\x00', '\x7f', '\x1f', 'ü', '中', '😀', '\U0010ffff'):
                test(prefix + bad)
                test(prefix + bad + 'xyz')
                test(prefix + bad * 3 + prefix)

    def test_xor_data64(self):
        # varying bytes in the key and data so that alignment/key rotation bugs are caught
        base_data = bytes(range(64))
        key = bytes(range(101, 165))
        sizes = []
        if has_sse4_2:
            sizes.append(2)
        if has_avx2:
            sizes.append(3)
        if has_avx512:
            sizes.append(4)
        sizes.append(0)

        def t(key, data, align_offset=0):
            expected = test_xor64(key, data, 1, 0)
            for which_function in sizes:
                actual = test_xor64(key, data, which_function, align_offset)
                self.ae(expected, actual, f'{align_offset=} {len(data)=}')

        t(key, b'')

        for base in (b'abc', base_data, base_data * 2):
            for extra in range(len(base_data)):
                for align_offset in range(64):
                    data = base + base_data[:extra]
                    t(key, data, align_offset)

    def test_blend_pixels(self):
        rng = random.Random(0x1B7F5)
        impls = []
        if has_sse4_2:
            impls.append(2)
        if has_avx2:
            impls.append(3)
        if has_avx512:
            impls.append(4)
        impls.append(0)

        def t(kind, dst, src, color=0xFFFFFF, align_offsets=(0, 1, 2, 3, 7, 13)):
            expected = test_blend_pixels(kind, dst, src, 1, color, 0)
            for which_function in impls:
                for align_offset in align_offsets:
                    actual = test_blend_pixels(kind, dst, src, which_function, color, align_offset)
                    self.ae(expected, actual, f'{kind=} {which_function=} {align_offset=} num_pixels={len(src) // 4}')

        for num_pixels in (0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 15, 16, 17, 31, 32, 33, 63, 64, 65, 255, 256, 257):
            src = bytearray(rng.getrandbits(8) for _ in range(4 * num_pixels))
            # ensure fully transparent and fully opaque source pixels are well represented
            for i in range(num_pixels):
                if i % 5 == 0:
                    src[4 * i + 3] = 0
                elif i % 5 == 1:
                    src[4 * i + 3] = 255
            src = bytes(src)
            dst4 = bytes(rng.getrandbits(8) for _ in range(4 * num_pixels))
            dst3 = bytes(rng.getrandbits(8) for _ in range(3 * num_pixels))
            mask = bytes(rng.getrandbits(8) for _ in range(num_pixels))
            t('straight', dst4, src)
            t('opaque', dst4, src)
            t('opaque3', dst3, src)
            # keep the destination pixels 4-byte aligned for the mask variant
            t('mask', dst4, mask, color=0x123456, align_offsets=(0, 4, 12, 36))

    def test_esc_codes(self):
        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)
        pb('12\033Da', '12', ('screen_index',), 'a')
        self.ae(str(s.line(0)), '12')
        self.ae(str(s.line(1)), '  a')
        pb('\033xa', ('Unknown char after ESC: 0x%x' % ord('x'),), 'a')
        pb('\033c123', ('screen_reset',), '123')
        self.ae(str(s.line(0)), '123')
        pb('\033.\033a', ('Unhandled charset related escape code: 0x2e 0x1b',), 'a')

    def test_csi_codes(self):
        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)
        pb('abcde', 'abcde')
        s.cursor_move(5)
        pb('x\033[2@y', 'x', ('screen_insert_characters', 2), 'y')
        self.ae(str(s.line(0)), 'xy bc')
        pb('x\033[2;7@y', 'x', ('CSI code @ has 2 > 1 parameters',), 'y')
        pb('x\033[2;-7@y', 'x', ('CSI code @ has 2 > 1 parameters',), 'y')
        pb('x\033[-0001234567890@y', 'x', ('CSI code @ is not allowed to have negative parameter (-1234567890)',), 'y')
        pb('x\033[2-3@y', 'x', ('Invalid character in CSI: 3 (0x33), ignoring the sequence',), '@y')
        pb('x\033[@y', 'x', ('screen_insert_characters', 1), 'y')
        pb('x\033[345@y', 'x', ('screen_insert_characters', 345), 'y')
        pb('x\033[345;@y', 'x', ('screen_insert_characters', 345), 'y')
        pb('\033[H', ('screen_cursor_position', 1, 1))
        self.ae(s.cursor.x, 0), self.ae(s.cursor.y, 0)
        pb('\033[4H', ('screen_cursor_position', 4, 1))
        pb('\033[4;0H', ('screen_cursor_position', 4, 0))
        pb('\033[3;2H', ('screen_cursor_position', 3, 2))
        pb('\033[3;2;H', ('screen_cursor_position', 3, 2))
        pb('\033[00000000003;0000000000000002H', ('screen_cursor_position', 3, 2))
        self.ae(s.cursor.x, 1), self.ae(s.cursor.y, 2)
        pb('\033[0001234567890H', ('screen_cursor_position', 1234567890, 1))
        pb('\033[J', ('screen_erase_in_display', 0, 0))
        pb('\033[?J', ('screen_erase_in_display', 0, 1))
        pb('\033[?2J', ('screen_erase_in_display', 2, 1))
        pb('\033[h')
        pb('\033[20;4h', ('screen_set_mode', 20, 0), ('screen_set_mode', 4, 0))
        pb('\033[?1000;1004h', ('screen_set_mode', 1000, 1), ('screen_set_mode', 1004, 1))
        pb('\033[20;4;20l', ('screen_reset_mode', 20, 0), ('screen_reset_mode', 4, 0), ('screen_reset_mode', 20, 0))
        pb('\033[=c', ('report_device_attributes', 0, 61))
        s.reset()

        def sgr(*params):
            return (('select_graphic_rendition', f'{x}') for x in params)

        pb('\033[1;2;3;4;7;9;34;44m', *sgr('1;2;3;4;7;9;34;44'))
        for attr in 'bold italic reverse strikethrough dim'.split():
            self.assertTrue(getattr(s.cursor, attr), attr)
        self.ae(s.cursor.decoration, 1)
        self.ae(s.cursor.fg, 4 << 8 | 1)
        self.ae(s.cursor.bg, 4 << 8 | 1)
        pb('\033[38;5;1;48;5;7m', ('select_graphic_rendition', '38:5:1'), ('select_graphic_rendition', '48:5:7'))
        self.ae(s.cursor.fg, 1 << 8 | 1)
        self.ae(s.cursor.bg, 7 << 8 | 1)
        pb('\033[38;2;1;2;3;48;2;7;8;9m', ('select_graphic_rendition', '38:2:1:2:3'), ('select_graphic_rendition', '48:2:7:8:9'))
        self.ae(s.cursor.fg, 1 << 24 | 2 << 16 | 3 << 8 | 2)
        self.ae(s.cursor.bg, 7 << 24 | 8 << 16 | 9 << 8 | 2)
        pb('\033[0;2m', *sgr('0;2'))
        pb('\033[;2m', *sgr('0;2'))
        pb('\033[m', *sgr('0'))
        pb('\033[1;;2m', *sgr('1;0;2'))
        pb('\033[38;5;1m', ('select_graphic_rendition', '38:5:1'))
        pb('\033[58;2;1;2;3m', ('select_graphic_rendition', '58:2:1:2:3'))
        pb('\033[38;2;1;2;3m', ('select_graphic_rendition', '38:2:1:2:3'))
        pb('\033[1001:2:1:2:3m', ('select_graphic_rendition', '1001:2:1:2:3'))
        pb(
            '\033[38:2:1:2:3;48:5:9;58;5;7m',
            ('select_graphic_rendition', '38:2:1:2:3'),
            ('select_graphic_rendition', '48:5:9'),
            ('select_graphic_rendition', '58:5:7'),
        )
        s.reset()
        pb('\033[1;2;3;4:5;7;9;34;44m', *sgr('1;2;3', '4:5', '7;9;34;44'))
        for attr in 'bold italic reverse strikethrough dim'.split():
            self.assertTrue(getattr(s.cursor, attr), attr)
        self.ae(s.cursor.decoration, 5)
        c = s.callbacks
        pb('\033[5n', ('report_device_status', 5, 0))
        self.ae(c.wtcbuf, b'\033[0n')
        c.clear()
        pb('\033[6n', ('report_device_status', 6, 0))
        self.ae(c.wtcbuf, b'\033[1;1R')
        pb('12345', '12345')
        c.clear()
        pb('\033[6n', ('report_device_status', 6, 0))
        self.ae(c.wtcbuf, b'\033[2;1R')
        c.clear()
        s.cursor_key_mode = True
        pb('\033[?1$p', ('report_mode_status', 1, 1))
        self.ae(c.wtcbuf, b'\033[?1;1$y')
        pb('\033[?1l', ('screen_reset_mode', 1, 1))
        self.assertFalse(s.cursor_key_mode)
        c.clear()
        pb('\033[?1$p', ('report_mode_status', 1, 1))
        self.ae(c.wtcbuf, b'\033[?1;2$y')
        pb('\033[2;4r', ('screen_set_margins', 2, 4))
        c.clear()
        pb('\033[14t', ('screen_report_size', 14, 0))
        pb('\033[14;2t', ('screen_report_size', 14, 2))
        self.ae(c.wtcbuf, b'\033[4;100;50t\033[4;100;50t')
        self.ae(s.margin_top, 1), self.ae(s.margin_bottom, 3)
        pb('\033[r', ('screen_set_margins', 0, 0))
        self.ae(s.margin_top, 0), self.ae(s.margin_bottom, 4)
        pb('\033[1 q', ('screen_set_cursor', 1, ord(' ')))
        self.assertTrue(s.cursor.blink)
        self.ae(s.cursor.shape, CURSOR_BLOCK)

        s.reset()
        pb('\033[3 @', ('Shift left escape code not implemented',))
        pb('\033[3 A', ('Shift right escape code not implemented',))
        pb('\033[3;4 S', ('Select presentation directions escape code not implemented',))
        pb('\033[1T', ('screen_reverse_scroll', 1))
        pb('\033[T', ('screen_reverse_scroll', 1))
        pb('\033[+T', ('screen_reverse_scroll_and_fill_from_scrollback', 1))

        c.clear()
        pb('\033[?2026$p', ('report_mode_status', 2026, 1))
        self.ae(c.wtcbuf, b'\x1b[?2026;2$y')
        c.clear()
        pb('\033[?2026h', ('screen_set_mode', 2026, 1))
        pb('\033[?2026$p', ('report_mode_status', 2026, 1))
        self.ae(c.wtcbuf, b'\x1b[?2026;1$y')
        pb('\033[?2026l', ('screen_reset_mode', 2026, 1))
        c.clear()
        pb('\033[?2026$p', ('report_mode_status', 2026, 1))
        self.ae(c.wtcbuf, b'\x1b[?2026;2$y')

        c.clear()
        pb('\033[?998n', ('report_device_status', 998, 1))
        self.ae(c.wtcbuf, b'\x1b[?999;1n')
        c.clear()
        pb('\033[?2033h', ('screen_set_mode', 2033, 1))
        self.ae(c.wtcbuf, b'\x1b[?999;1n')
        c.clear()
        pb('\033[?2033$p', ('report_mode_status', 2033, 1))
        self.ae(c.wtcbuf, b'\x1b[?2033;1$y')
        pb('\033[?2033l', ('screen_reset_mode', 2033, 1))

    def test_csi_code_rep(self):
        s = self.create_screen(8)
        pb = partial(self.parse_bytes_dump, s)
        pb('\033[1b', ('screen_repeat_character', 1))
        self.ae(str(s.line(0)), '')
        pb('x\033[7b', 'x', ('screen_repeat_character', 7))
        self.ae(str(s.line(0)), 'xxxxxxxx')
        pb('\033[1;3H', ('screen_cursor_position', 1, 3))
        pb('\033[byz\033[b', ('screen_repeat_character', 1), 'yz', ('screen_repeat_character', 1))
        # repeat 'x' at 3, then 'yz' at 4-5, then repeat 'z' at 6
        self.ae(str(s.line(0)), 'xxxyzzxx')
        s.reset()
        pb(' \033[3b', ' ', ('screen_repeat_character', 3))
        self.ae(str(s.line(0)), '    ')
        s.reset()
        pb('\t\033[b', ('screen_tab',), ('screen_repeat_character', 1))
        self.ae(str(s.line(0)), '\t')
        s.reset()
        b']]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]'

    def test_osc_codes(self):
        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)
        c = s.callbacks
        pb('a\033]2;x\\ryz\033\\bcde', 'a', ('set_title', 'x\\ryz'), 'bcde')
        self.ae(str(s.line(0)), 'abcde')
        self.ae(c.titlebuf, ['x\\ryz'])
        c.clear()
        pb('\033]\x07', ('set_title', ''), ('set_icon', ''))
        self.ae(c.titlebuf, ['']), self.ae(c.iconbuf, '')
        pb('1\033]ab\x072', '1', ('set_title', 'ab'), ('set_icon', 'ab'), '2')
        self.ae(c.titlebuf, ['', 'ab']), self.ae(c.iconbuf, 'ab')
        c.clear()
        pb('\033]2;;;;\x07', ('set_title', ';;;'))
        self.ae(c.titlebuf, [';;;'])
        c.clear()
        pb('\033]2;\x07', ('set_title', ''))
        self.ae(c.titlebuf, [''])
        pb('\033]110\x07', ('set_dynamic_color', 110, ''))
        self.ae(c.colorbuf, '')
        c.clear()
        pb('\033]9;\x07', ('desktop_notify', 9, ''))
        pb('\033]9;test it with a nice long string\x07', ('desktop_notify', 9, 'test it with a nice long string'))
        pb('\033]99;moo=foo;test it\x07', ('desktop_notify', 99, 'moo=foo;test it'))
        self.ae(c.notifications, [(9, ''), (9, 'test it with a nice long string'), (99, 'moo=foo;test it')])
        c.clear()
        pb('\033]8;;\x07', ('set_active_hyperlink', None, None))
        pb('\033]8moo\x07', ('Ignoring malformed OSC 8 code',))
        pb('\033]8;moo\x07', ('Ignoring malformed OSC 8 code',))
        pb('\033]8;id=xyz;\x07', ('set_active_hyperlink', 'xyz', None))
        pb('\033]8;moo:x=z:id=xyz:id=abc;http://yay;.com\x07', ('set_active_hyperlink', 'xyz', 'http://yay;.com'))
        c.clear()
        payload = '1' * 1024
        pb(f'\033]52;p;{payload}\x07', ('clipboard_control', 52, f'p;{payload}'))
        c.clear()
        pb('\033]52;p;xyz\x07', ('clipboard_control', 52, 'p;xyz'))
        c.clear()
        pb('\033]22;?__current__\x07', ('set_dynamic_color', 22, '?__current__'))

    def test_dcs_codes(self):
        s = self.create_screen()
        c = s.callbacks
        pb = partial(self.parse_bytes_dump, s)
        q = hexlify(b'kind').decode('ascii')
        pb(f'a\033P+q{q}\033\\bcde', 'a', ('screen_request_capabilities', 43, q), 'bcde')
        self.ae(str(s.line(0)), 'abcde')
        self.ae(c.wtcbuf, '1+r{}={}'.format(q, '1b5b313b3242').encode('ascii'))
        c.clear()
        pb('\033P$q q\033\\', ('screen_request_capabilities', ord('$'), ' q'))
        self.ae(c.wtcbuf, b'\033P1$r1 q\033\\')
        c.clear()
        pb('\033P$qm\033\\', ('screen_request_capabilities', ord('$'), 'm'))
        self.ae(c.wtcbuf, b'\033P1$rm\033\\')
        for sgr in '0;34;102;1;2;3;4 0;38:5:200;58:2:10:11:12'.split():
            expected = set(sgr.split(';'))
            c.clear()
            parse_bytes(s, f'\033[{sgr}m\033P$qm\033\\'.encode('ascii'))
            r = c.wtcbuf.decode('ascii').partition('r')[2].partition('m')[0]
            self.ae(expected, set(r.split(';')))
        c.clear()
        pb('\033P$qr\033\\', ('screen_request_capabilities', ord('$'), 'r'))
        self.ae(c.wtcbuf, f'\033P1$r{s.margin_top + 1};{s.margin_bottom + 1}r\033\\'.encode('ascii'))
        pb('\033P@kitty-cmd{abc\033\\', ('handle_remote_cmd', '{abc'))
        p = base64_encode('abcd').decode()
        pb(f'\033P@kitty-print|{p}\033\\', ('handle_remote_print', p))
        self.ae(['abcd'], s.callbacks.printbuf)

        c.clear()
        pb('\033[?2026$p', ('report_mode_status', 2026, 1))
        self.ae(c.wtcbuf, b'\x1b[?2026;2$y')
        pb('\033P=1s\033\\', ('screen_start_pending_mode',))
        c.clear()
        pb('\033[?2026$p', ('report_mode_status', 2026, 1))
        self.ae(c.wtcbuf, b'\x1b[?2026;1$y')
        pb('\033P=2s\033\\', ('screen_stop_pending_mode',))
        c.clear()
        pb('\033[?2026$p', ('report_mode_status', 2026, 1))
        self.ae(c.wtcbuf, b'\x1b[?2026;2$y')

    def test_oth_codes(self):
        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)
        pb('a\033_+\\+\033\\bcde', ('draw', 'a'), ('Unrecognized APC code: 0x2b',), ('draw', 'bcde'))
        pb('a\033^+\\+\033\\bcde', ('draw', 'a'), ('Unrecognized PM code: 0x2b',), ('draw', 'bcde'))
        pb('a\033X+\\+\033\\bcde', ('draw', 'a'), ('Unrecognized SOS code: 0x2b',), ('draw', 'bcde'))

    def test_graphics_command(self):
        from base64 import standard_b64encode

        def enc(x):
            return standard_b64encode(x.encode('utf-8') if isinstance(x, str) else x).decode('ascii')

        def c(**k):
            for p, v in tuple(k.items()):
                if isinstance(v, str) and p != 'payload':
                    k[p] = v.encode('ascii')
            for f in 'action delete_action transmission_type compressed'.split():
                k.setdefault(f, b'\0')
            for f in (
                'format more id data_sz data_offset width height x_offset y_offset data_height data_width cursor_movement'
                ' num_cells num_lines cell_x_offset cell_y_offset z_index placement_id image_number quiet unicode_placement'
                ' parent_id parent_placement_id usage_hints offset_from_parent_x offset_from_parent_y'
            ).split():
                k.setdefault(f, 0)
            p = k.pop('payload', '')
            k[''] = p
            return ('graphics_command', k)

        def t(cmd, **kw):
            pb('\033_G{};{}\033\\'.format(cmd, enc(kw.get('payload', ''))), c(**kw))

        def e(cmd, err):
            pb(f'\033_G{cmd}\033\\', (err,))

        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)
        uint32_max = 2**32 - 1
        t('i=%d' % uint32_max, id=uint32_max)
        t('i=3,p=4', id=3, placement_id=4)
        e('i=%d' % (uint32_max + 1), 'Malformed GraphicsCommand control block, number is too large')
        pb('\033_Gi=12\033\\', c(id=12))
        t('a=t,t=d,s=100,z=-9', payload='X', action='t', transmission_type='d', data_width=100, z_index=-9)
        t('a=t,t=d,s=100,z=9', payload='payload', action='t', transmission_type='d', data_width=100, z_index=9)
        t('a=t,t=d,s=100,z=9,q=2', action='t', transmission_type='d', data_width=100, z_index=9, quiet=2)
        t('N=1', usage_hints=1)
        e(',s=1', 'Malformed GraphicsCommand control block, invalid key character: 0x2c')
        e('W=1', 'Malformed GraphicsCommand control block, invalid key character: 0x57')
        e('1=1', 'Malformed GraphicsCommand control block, invalid key character: 0x31')
        e('a=t,,w=2', 'Malformed GraphicsCommand control block, invalid key character: 0x2c')
        e('s', 'Malformed GraphicsCommand control block, no = after key')
        e('s=', 'Malformed GraphicsCommand control block, expecting an integer value')
        e('s==', 'Malformed GraphicsCommand control block, expecting an integer value for key: s')
        e('s=1=', 'Malformed GraphicsCommand control block, expecting a , or semi-colon after a value, found: 0x3d')

    def test_deccara(self):
        s = self.create_screen()
        pb = partial(self.parse_bytes_dump, s)
        pb('\033[$r', ('deccara', '0;0;0;0;0'))
        pb(
            '\033[;;;;4:3;38:5:10;48:2:1:2:3;1$r',
            ('deccara', '0;0;0;0;4:3'),
            ('deccara', '0;0;0;0;38:5:10'),
            ('deccara', '0;0;0;0;48:2:1:2:3'),
            ('deccara', '0;0;0;0;1'),
        )
        for y in range(s.lines):
            line = s.line(y)
            for x in range(s.columns):
                c = line.cursor_from(x)
                self.ae(c.bold, True)
                self.ae(c.italic, False)
                self.ae(c.decoration, 3)
                self.ae(c.fg, (10 << 8) | 1)
                self.ae(c.bg, (1 << 24 | 2 << 16 | 3 << 8 | 2))
        self.ae(s.line(0).cursor_from(0).bold, True)
        pb('\033[1;2;2;3;22;39$r', ('deccara', '1;2;2;3;22;39'))
        self.ae(s.line(0).cursor_from(0).bold, True)
        line = s.line(0)
        for x in range(1, s.columns):
            c = line.cursor_from(x)
            self.ae(c.bold, False)
            self.ae(c.fg, 0)
        line = s.line(1)
        for x in range(0, 3):
            c = line.cursor_from(x)
            self.ae(c.bold, False)
        self.ae(line.cursor_from(3).bold, True)
        pb('\033[2*x\033[3;2;4;3;34$r\033[*x', ('screen_decsace', 2), ('deccara', '3;2;4;3;34'), ('screen_decsace', 0))
        for y in range(2, 4):
            line = s.line(y)
            for x in range(s.columns):
                self.ae(line.cursor_from(x).fg, (10 << 8 | 1) if x < 1 or x > 2 else (4 << 8) | 1)