"""Pretty-print the host-accessible MMIO / register map for this card.

Sources, all authoritative:
  * PCI BARs as the bridge/driver actually mapped them
  * the card's own IP discovery table (base addresses per IP instance)
  * tinygrad's register-window bookkeeping (regs_offset)
SAFETY: read-only, and deliberately no high-offset BAR0 pokes (those raise a
TLP error that wedges PCIe until a power cycle).
"""
import sys
sys.path.insert(0,"/Users/user/src/tinygrad")
from tinygrad import Device
from tinygrad.runtime.autogen.am import am
d=Device["AMD"]; adev=d.iface.dev_impl
pci=getattr(d.iface,'pci_dev',None) or getattr(adev,'pci_dev',None)

def hdr(t):
    print("\n"+"="*104); print(t); print("="*104)

hdr("1. PCI BARs  (what the host can touch directly)")
print("%-6s %-18s %-14s %-12s %s"%("BAR","base","size","mapped?","use"))
USE={0:"VRAM aperture (frame buffer, 24 GB) - high offsets WEDGE PCIe",
     2:"Doorbell aperture (ring doorbells)",
     5:"MMIO register aperture (all IP registers, incl. PSP/SMU mailboxes)"}
for i in range(6):
    try: info=pci.bar_info(i)
    except Exception as e: continue
    base,size = (info[0],info[1]) if isinstance(info,(tuple,list)) and len(info)>=2 else (info,0)
    if not base and not size: continue
    print("  %-4d %-18s %-14s %-12s %s"%(i,hex(base) if isinstance(base,int) else str(base),
          hex(size) if isinstance(size,int) else str(size),"yes",USE.get(i,"")))
print("  vram_bar index = %s   large_bar=%s"%(getattr(d.iface,'vram_bar','?'),getattr(adev,'large_bar','?')))
print("\n  BAR5 is addressed by tinygrad as DWORD indices (reg.addr), not byte offsets")
print("  vram_size = %#x (%d MB)"%(adev.vram_size, adev.vram_size>>20))

hdr("2. Indirect windows reachable THROUGH BAR5 (widen the reach beyond the aperture)")
rows=[("SMN (System Management Network)","PCIE_INDEX2 / PCIE_DATA2","BAR5 dword 14 / 15",
       "32-bit SMN address space: MP0/MP1/THM/SMUIO/BIF..."),
      ("RSMU","regBIF_BX_PF0_RSMU_INDEX/_DATA","(same decode as PCIE_INDEX2)",
       "verified byte-identical to the PCIe window - NOT a second path"),
      ("PCIE config","PCIE_INDEX/PCIE_DATA","BAR5","PCIe core config space"),
      ("GPUVM apertures","MC_VM_* regs","BAR5","translate VRAM/system memory")]
print("%-34s %-34s %-22s %s"%("window","index/data regs","location","reaches"))
for a,b,c,e in rows: print("  %-32s %-34s %-22s %s"%(a,b,c,e))

hdr("3. Mailboxes the host can drive (confirmed reachable)")
mb=[("PSP (MP0)","C2PMSG_N = BAR5 + 0x58224 + 4*(N-73)","ring doorbell 0x58100.., resp 0x582CC/0x582D0",
     "LOAD_TA/INVOKE/SETUP_TMR... fuzzed extensively"),
    ("SMU (MP1)","MSG=C2PMSG_66 ARG=C2PMSG_82 RESP=C2PMSG_90","via SMN MP1 base",
     "79 PPSMC msgs + 13 tables; audited"),
    ("TSU","C2PMSG_115/116 + doorbell 73","BAR5","OUT OF SCOPE (flash) per CONSTRAINTS.md")]
print("%-12s %-42s %-44s"%("block","registers","notes"))
for a,b,c,e in mb:
    print("  %-10s %-42s %s"%(a,b,c)); print("  %-10s %-42s -> %s"%("","",e))


hdr("4. IP blocks and their register bases, from the card's OWN discovery table")
ipn={}
try:
    ipn=adev.ip_ver
except Exception: pass
NAME={}
for k,v in vars(am).items():
    if k.endswith('_HWIP') and isinstance(v,int): NAME[v]=k[:-5]
seen={}
off=getattr(adev,'regs_offset',None)
if isinstance(off,dict):
    for hwip,inst in sorted(off.items(), key=lambda t:(str(t[0]))):
        nm=NAME.get(hwip, str(hwip))
        try: bases=[hex(x) for x in (inst if isinstance(inst,(list,tuple)) else [inst])]
        except Exception: bases=[str(inst)]
        seen[nm]=bases
print("%-16s %-10s %s"%("IP block","ip_ver","register base(s) (SMN / aperture addresses)"))
for nm in sorted(seen):
    hw=[k for k,v in NAME.items() if v==nm]
    ver=ipn.get(hw[0]) if hw and isinstance(ipn,dict) else None
    b=seen[nm]
    flat=[]
    for x in b:
        flat.append(x)
    print("  %-14s %-10s %s"%(nm, str(ver) if ver else "-", ", ".join(flat[:6])+(" ..." if len(flat)>6 else "")))
print("\n  total IP entries: %d"%len(seen))