"""Print the two golden digests from the local build."""
import hashlib
import os
import sys

ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, ROOT)
sys.path.insert(0, os.path.join(ROOT, "tests"))

import load_local

svr = load_local.load()

import types

try:
    import kernels
except ImportError:
    kernels = types.ModuleType("kernels")
    sys.modules["kernels"] = kernels
kernels.get_kernel = lambda *a, **k: svr

import numpy as np
import torch
from test_svr import hash_rgba_u8


def digest(t):
    return hashlib.sha256(t.detach().cpu().numpy().tobytes()).hexdigest()


u8 = hash_rgba_u8(2, 384, 416, salt=101).cuda()
out8 = svr.upscale(u8, out_hw=(704, 768), center=[[0.46, 0.52], [0.54, 0.52]],
                   radii=(0.28, 0.66), sharp=0.45)
print(f'"eyes_uint8": "{digest(out8)}",')
f32 = (hash_rgba_u8(1, 256, 288, salt=202).cuda().float() / 255.0).contiguous()
outf = svr.upscale(f32, out_hw=(480, 512), center=(0.5, 0.47),
                   radii=(0.3, 0.7), sharp=0.35)
print(f'"eyes_f32": "{digest(outf)}",')