"""Test setting Zoom to 0.5 (zoom out) vs 2.0 (zoom in).
"""
import ctypes
import ctypes.wintypes as w
import struct
import time
import subprocess

k32 = ctypes.WinDLL("kernel32", use_last_error=True)
k32.OpenProcess.restype = w.HANDLE
k32.ReadProcessMemory.argtypes = [
    w.HANDLE, ctypes.c_void_p, ctypes.c_void_p, ctypes.c_size_t, ctypes.POINTER(ctypes.c_size_t)
]
k32.ReadProcessMemory.restype = w.BOOL
k32.WriteProcessMemory.argtypes = [
    w.HANDLE, ctypes.c_void_p, ctypes.c_void_p, ctypes.c_size_t, ctypes.POINTER(ctypes.c_size_t)
]
k32.WriteProcessMemory.restype = w.BOOL

def rpm(h, addr, size):
    buf = (ctypes.c_ubyte * size)()
    got = ctypes.c_size_t(0)
    if not k32.ReadProcessMemory(h, ctypes.c_void_p(addr), buf, size, ctypes.byref(got)):
        return None
    return bytes(buf[: got.value])

def rpm_f32(h, addr):
    b = rpm(h, addr, 4)
    return struct.unpack("<f", b)[0] if b else None

def wpm_f32(h, addr, val):
    buf = (ctypes.c_ubyte * 4).from_buffer_copy(struct.pack("<f", val))
    got = ctypes.c_size_t(0)
    return bool(k32.WriteProcessMemory(h, ctypes.c_void_p(addr), buf, 4, ctypes.byref(got)))

out = subprocess.check_output(['powershell', '-Command', '(Get-Process -Name *PathOfExile*).Id'])
pid = int(out.decode().split()[0])
h = k32.OpenProcess(0x1F0FFF, False, pid)

cam_addr = 0x48514f60e00
zoom_addr = cam_addr + 0x528

print(f"Current Zoom: {rpm_f32(h, zoom_addr)}")
print("Testing Zoom = 0.5 (should decrease M and zoom OUT)...")
wpm_f32(h, zoom_addr, 0.5)

for i in range(5):
    time.sleep(0.1)
    m = [struct.unpack_from("<f", rpm(h, cam_addr + 0x1A0, 64), j*4)[0] for j in range(16)]
    print(f"[{i*100}ms] Zoom={rpm_f32(h, zoom_addr):.2f}, M[0]={m[0]:.4f}, M[5]={m[5]:.4f}")

# Restore
wpm_f32(h, zoom_addr, 1.0)
print("Restored Zoom to 1.0")