import os, time, cv2
from optimized_swapper import FaceSwapEngine

src = cv2.imread("benchmark_source.jpg")
dst = cv2.imread("benchmark_target.jpg")
if src is None or dst is None:
    raise SystemExit("Put benchmark_source.jpg and benchmark_target.jpg in the root.")

engine = FaceSwapEngine()
for _ in range(2):
    engine.swap_image(src, 1, dst, 1, 320)

n = 20
t0 = time.perf_counter()
for _ in range(n):
    engine.swap_prepared_source(dst, 1, 320, 1)
dt = time.perf_counter() - t0
print(f"{n} swaps: {dt:.3f}s")
print(f"Average: {dt/n*1000:.2f} ms/image")
print(f"Throughput: {n/dt:.2f} images/sec")