"""Fail before rollout if legacy SAPIEN maps CUDA to the wrong physical GPU."""
import argparse
import os
import subprocess


def main():
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('--expected-uuid', required=True)
    args = parser.parse_args()
    import sapien.core as sapien
    engine = sapien.Engine()
    renderer = sapien.SapienRenderer(device='cuda:0', offscreen_only=True)
    engine.set_renderer(renderer)
    lines = subprocess.check_output([
        'nvidia-smi', '--query-compute-apps=gpu_uuid,pid', '--format=csv,noheader'
    ], text=True).splitlines()
    actual = {uuid.strip() for uuid, pid in (line.split(',') for line in lines)
              if pid.strip() == str(os.getpid())}
    if actual != {args.expected_uuid}:
        raise RuntimeError(
            f'SAPIEN physical GPU mismatch: expected {args.expected_uuid}, got {sorted(actual)}. '
            'This legacy renderer can confuse equal PCI bus numbers in different domains. '
            'No rollout was started; choose a verified free renderer GPU.'
        )
    print(f'SAPIEN_GPU_VERIFIED uuid={args.expected_uuid}', flush=True)


if __name__ == '__main__':
    main()