"""Burst capture vs the browser renderer: chunk-level input, game state on every frame.

In a burst all of a chunk's input enters at its first frame, so both sides get
each chunk's input on the first drain only. No press and release share a chunk,
so nothing is split across frames on either side.
"""
import json, os, sys
import numpy as np
sys.path.insert(0, '/root/autodl-tmp/edenlive')
os.environ['EDEN_NATIVE_CAPTURE'] = '1'
from eden.apps.game2v import game_renderer, session
from eden.apps.game2v.game_renderer import GameCaptureWorker, GameRenderer
from eden.apps.game2v.game_timeline import GameTimeline
from capture_compat import install_capture_context
import native_capture
install_capture_context(game_renderer)
native_capture.install(game_renderer, session)
_l = GameRenderer._gpu_launch_options
GameRenderer._gpu_launch_options = lambda r: {**_l(r), 'executable_path': os.environ['GBDATA_CHROMIUM']}
P = lambda x, y, **k: dict(action='pointer', phase=k.get('phase', 'move'), x=x, y=y, button=k.get('button'), pressed=k.get('pressed'))
SCRIPT = {0: [{'action': 'enter', 'phase': 'down'}], 1: [{'action': 'enter', 'phase': 'up'}],
          3: [{'action': 'w', 'phase': 'down'}, P(0.5, 0.5)], 5: [P(0.57, 0.48)], 6: [{'action': 'w', 'phase': 'up'}, P(0.62, 0.46)],
          8: [P(0.5, 0.5, phase='button', button=0, pressed=True)], 9: [P(0.5, 0.5, phase='button', button=0, pressed=False)],
          11: [{'action': 'k', 'phase': 'down'}], 13: [{'action': 'k', 'phase': 'up'}, P(0.44, 0.52)],
          15: [{'action': 'a', 'phase': 'down'}], 17: [{'action': 'a', 'phase': 'up'}]}
def run(native):
    os.environ['EDEN_NATIVE_CAPTURE'] = '1' if native else '0'
    worker = GameCaptureWorker('eiffel-assault', width=960, height=544)
    tl, out = GameTimeline('eiffel-assault'), []
    for c in range(20):
        events = list(SCRIPT.get(c, []))
        def drain():
            nonlocal events
            batch, events = events, []
            return batch
        out.append(worker.capture(tl.begin_chunk(9 if c == 0 else 12, 24), duration_s=0.0, drain_events=drain))
    kind = type(worker._renderer).__name__
    worker.close()
    return kind, out
kb, browser = run(False)
kn, native = run(True)
keys = ('x', 'z', 'y', 'yaw', 'pitch')
diffs = []
for bp, np_ in zip(browser, native):
    for bf, nf in zip(bp.frames, np_.frames):
        a, b = bf.ui_state or {}, nf.ui_state or {}
        diffs.append(max([abs((a.get('player') or {}).get(k, 0) - (b.get('player') or {}).get(k, 0)) for k in keys] +
                         [float(a.get(k) != b.get(k)) for k in ('state', 'shots', 'ads', 'health', 'ammo', 'roundId', 'kills')]))
last = lambda p: {k: p[-1].frames[-1].ui_state.get(k) for k in ('state', 'shots', 'ads', 'health', 'ammo')}
print(json.dumps(dict(browser=kb, native=kn, frames=len(diffs), max_state_diff=max(diffs), browser_end=last(browser), native_end=last(native))))